factor/vm/slot_visitor.hpp

namespace factor
{

template<typename Visitor> struct slot_visitor {
	factor_vm *parent;
	Visitor visitor;

	slot_visitor<Visitor>(factor_vm *parent_, Visitor visitor_) :
		parent(parent_), visitor(visitor_) {}

	void visit_handle(cell *handle)
	{
		cell pointer = *handle;

		if(immediate_p(pointer)) return;

		object *untagged = untag<object>(pointer);
		untagged = visitor.visit_object(untagged);
		*handle = RETAG(untagged,TAG(pointer));
	}

	void visit_slots(object *ptr)
	{
		cell *slot = (cell *)ptr;
		cell *end = (cell *)((cell)ptr + ptr->binary_payload_start());

		if(slot != end)
		{
			slot++;
			for(; slot < end; slot++) visit_handle(slot);
		}
	}

	void visit_stack_elements(segment *region, cell *top)
	{
		for(cell *ptr = (cell *)region->start; ptr <= top; ptr++)
			visit_handle(ptr);
	}

	void visit_registered_locals()
	{
		std::vector<cell>::const_iterator iter = parent->gc_locals.begin();
		std::vector<cell>::const_iterator end = parent->gc_locals.end();

		for(; iter < end; iter++)
			visit_handle((cell *)(*iter));
	}

	void visit_registered_bignums()
	{
		std::vector<cell>::const_iterator iter = parent->gc_bignums.begin();
		std::vector<cell>::const_iterator end = parent->gc_bignums.end();

		for(; iter < end; iter++)
		{
			cell *handle = (cell *)(*iter);

			if(*handle)
				*handle = (cell)visitor.visit_object(*(object **)handle);
		}
	}

	void visit_roots()
	{
		visit_handle(&parent->true_object);
		visit_handle(&parent->bignum_zero);
		visit_handle(&parent->bignum_pos_one);
		visit_handle(&parent->bignum_neg_one);

		visit_registered_locals();
		visit_registered_bignums();

		for(cell i = 0; i < special_object_count; i++)
			visit_handle(&parent->special_objects[i]);
	}

	void visit_contexts()
	{
		context *ctx = parent->ctx;

		while(ctx)
		{
			visit_stack_elements(ctx->datastack_region,(cell *)ctx->datastack);
			visit_stack_elements(ctx->retainstack_region,(cell *)ctx->retainstack);

			visit_handle(&ctx->catchstack_save);
			visit_handle(&ctx->current_callback_save);

			ctx = ctx->next;
		}
	}

	void visit_literal_references(code_block *compiled)
	{
		visit_handle(&compiled->owner);
		visit_handle(&compiled->literals);
		visit_handle(&compiled->relocation);
	}
};

}