factor/vm/jit.cpp

149 lines
3.7 KiB
C++
Raw Normal View History

2009-05-02 05:04:19 -04:00
#include "master.hpp"
2009-05-04 02:46:13 -04:00
namespace factor
{
2009-05-02 05:04:19 -04:00
/* Simple code generator used by:
- counting_profiler (counting_profiler.cpp),
2009-05-02 10:19:09 -04:00
- quotation compiler (quotations.cpp),
- megamorphic caches (dispatch.cpp),
- polymorphic inline caches (inline_cache.cpp) */
2009-05-02 05:04:19 -04:00
/* Allocates memory */
jit::jit(code_block_type type_, cell owner_, factor_vm *vm)
2009-05-02 10:19:09 -04:00
: type(type_),
owner(owner_,vm),
code(vm),
relocation(vm),
parameters(vm),
literals(vm),
2009-05-02 10:19:09 -04:00
computing_offset_p(false),
position(0),
2009-08-17 16:37:09 -04:00
offset(0),
parent(vm)
{
fixnum old_count = atomic::fetch_add(&parent->current_jit_count, 1);
assert(old_count >= 0);
}
jit::~jit()
{
fixnum old_count = atomic::fetch_subtract(&parent->current_jit_count, 1);
assert(old_count >= 1);
}
2009-05-02 05:04:19 -04:00
void jit::emit_relocation(cell relocation_template_)
2009-05-02 05:04:19 -04:00
{
data_root<byte_array> relocation_template(relocation_template_,parent);
cell capacity = array_capacity(relocation_template.untagged())
/ sizeof(relocation_entry);
relocation_entry *relocations = relocation_template->data<relocation_entry>();
for(cell i = 0; i < capacity; i++)
2009-05-02 05:04:19 -04:00
{
relocation_entry entry = relocations[i];
relocation_entry new_entry(entry.rel_type(), entry.rel_class(),
entry.rel_offset() + code.count);
relocation.append_bytes(&new_entry,sizeof(relocation_entry));
2009-05-02 05:04:19 -04:00
}
}
/* Allocates memory */
2009-05-04 05:50:24 -04:00
void jit::emit(cell code_template_)
2009-05-02 05:04:19 -04:00
{
data_root<array> code_template(code_template_,parent);
2009-05-02 05:04:19 -04:00
emit_relocation(array_nth(code_template.untagged(),0));
2009-05-02 05:04:19 -04:00
data_root<byte_array> insns(array_nth(code_template.untagged(),1),parent);
2009-05-02 05:04:19 -04:00
2009-05-02 10:19:09 -04:00
if(computing_offset_p)
2009-05-02 05:04:19 -04:00
{
2009-05-04 05:50:24 -04:00
cell size = array_capacity(insns.untagged());
2009-05-02 05:04:19 -04:00
2009-05-02 10:19:09 -04:00
if(offset == 0)
2009-05-02 05:04:19 -04:00
{
2009-05-02 10:19:09 -04:00
position--;
computing_offset_p = false;
2009-05-02 05:04:19 -04:00
}
2009-05-02 10:19:09 -04:00
else if(offset < size)
2009-05-02 05:04:19 -04:00
{
2009-05-02 10:19:09 -04:00
position++;
computing_offset_p = false;
2009-05-02 05:04:19 -04:00
}
else
2009-05-02 10:19:09 -04:00
offset -= size;
2009-05-02 05:04:19 -04:00
}
2009-05-02 10:19:09 -04:00
code.append_byte_array(insns.value());
}
void jit::emit_with_literal(cell code_template_, cell argument_) {
data_root<array> code_template(code_template_,parent);
data_root<object> argument(argument_,parent);
2009-05-02 10:19:09 -04:00
literal(argument.value());
emit(code_template.value());
}
void jit::emit_with_parameter(cell code_template_, cell argument_) {
data_root<array> code_template(code_template_,parent);
data_root<object> argument(argument_,parent);
parameter(argument.value());
emit(code_template.value());
}
bool jit::emit_subprimitive(cell word_, bool tail_call_p, bool stack_frame_p)
{
data_root<word> word(word_,parent);
data_root<array> code_template(word->subprimitive,parent);
parameters.append(untag<array>(array_nth(code_template.untagged(),0)));
literals.append(untag<array>(array_nth(code_template.untagged(),1)));
emit(array_nth(code_template.untagged(),2));
if(array_capacity(code_template.untagged()) == 5)
{
if(tail_call_p)
{
if(stack_frame_p) emit(parent->special_objects[JIT_EPILOG]);
emit(array_nth(code_template.untagged(),4));
return true;
}
else
emit(array_nth(code_template.untagged(),3));
}
return false;
}
2009-05-02 10:19:09 -04:00
/* Facility to convert compiled code offsets to quotation offsets.
Call jit_compute_offset() with the compiled code offset, then emit
code, and at the end jit->position is the quotation position. */
2009-05-04 05:50:24 -04:00
void jit::compute_position(cell offset_)
2009-05-02 10:19:09 -04:00
{
computing_offset_p = true;
position = 0;
offset = offset_;
}
2009-05-02 05:04:19 -04:00
2009-05-02 10:19:09 -04:00
/* Allocates memory */
2009-05-04 05:50:24 -04:00
code_block *jit::to_code_block()
2009-05-02 10:19:09 -04:00
{
/* Emit dummy GC info */
code.grow_bytes(alignment_for(code.count + 4,data_alignment));
u32 dummy_gc_info = 0;
code.append_bytes(&dummy_gc_info,sizeof(u32));
2009-05-02 10:19:09 -04:00
code.trim();
relocation.trim();
parameters.trim();
2009-05-02 10:19:09 -04:00
literals.trim();
return parent->add_code_block(
2009-05-02 10:19:09 -04:00
type,
2009-05-04 05:50:24 -04:00
code.elements.value(),
false_object, /* no labels */
owner.value(),
2009-05-04 05:50:24 -04:00
relocation.elements.value(),
parameters.elements.value(),
2009-05-04 05:50:24 -04:00
literals.elements.value());
2009-05-02 05:04:19 -04:00
}
2009-05-04 02:46:13 -04:00
}