namespace factor
{

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

	explicit call_frame_code_block_visitor(factor_vm *parent_, Visitor visitor_) :
		parent(parent_), visitor(visitor_) {}

	void operator()(stack_frame *frame)
	{
		cell offset = (cell)FRAME_RETURN_ADDRESS(frame,parent) - (cell)frame->xt;

		code_block *new_block = visitor(parent->frame_code(frame));
		frame->xt = new_block->xt();

		FRAME_RETURN_ADDRESS(frame,parent) = (void *)((cell)frame->xt + offset);
	}
};

template<typename Visitor> void factor_vm::visit_object_code_block(object *obj, Visitor visitor)
{
	switch(obj->h.hi_tag())
	{
	case WORD_TYPE:
		{
			word *w = (word *)obj;
			if(w->code)
				w->code = visitor(w->code);
			if(w->profiling)
				w->code = visitor(w->profiling);

			update_word_xt(w);
			break;
		}
	case QUOTATION_TYPE:
		{
			quotation *q = (quotation *)obj;
			if(q->code)
				set_quot_xt(q,visitor(q->code));
			break;
		}
	case CALLSTACK_TYPE:
		{
			callstack *stack = (callstack *)obj;
			call_frame_code_block_visitor<Visitor> call_frame_visitor(this,visitor);
			iterate_callstack_object(stack,call_frame_visitor);
			break;
		}
	}
}

template<typename Visitor> void factor_vm::visit_context_code_blocks(Visitor visitor)
{
	call_frame_code_block_visitor<Visitor> call_frame_visitor(this,visitor);
	iterate_active_frames(call_frame_visitor);
}

template<typename Visitor> struct callback_code_block_visitor {
	callback_heap *callbacks;
	Visitor visitor;

	explicit callback_code_block_visitor(callback_heap *callbacks_, Visitor visitor_) :
		callbacks(callbacks_), visitor(visitor_) {}

	void operator()(callback *stub)
	{
		stub->compiled = visitor(stub->compiled);
		callbacks->update(stub);
	}
};

template<typename Visitor> void factor_vm::visit_callback_code_blocks(Visitor visitor)
{
	callback_code_block_visitor<Visitor> callback_visitor(callbacks,visitor);
	callbacks->iterate(callback_visitor);
}

}