void JitInstanceMemberFunction::compileThisAdjustementThunk( size_t a_uiThisOffset ) const
{
o_assert(m_ThisAdjustmentThunks.find(a_uiThisOffset) == m_ThisAdjustmentThunks.end());
size_t argCount = getSignature()->getParameterCount()+1; // parameters+this
jit_value_t* args = o_allocate_n(argCount, jit_value_t);
// The fixing function has the same signature as the indirected one
jit_function_t func = jit_function_create((jit_context_t)m_jit_context.context, jit_function_get_signature((jit_function_t)m_jit_function.function));
jit_value_t new_this = jit_insn_add(func, jit_value_get_param(func, 0), jit_value_create_nint_constant(func, jit_type_nint, -((int)a_uiThisOffset)));
args[0] = new_this;
size_t i = 1;
for(;i<argCount;++i)
{
args[i] = jit_value_get_param(func, i);
}
string name = "[thunk]:"+m_pSubroutine->getQualifiedDecoratedName()+":adjustor{"+boost::lexical_cast<string>(a_uiThisOffset)+"}";
jit_insn_call(func
, name.c_str()
, (jit_function_t)m_jit_function.function
, jit_function_get_signature((jit_function_t)m_jit_function.function)
, args
, argCount
, 0);
int result = jit_function_compile(func);
o_assert(result != 0);
o_deallocate_n(args, argCount, jit_value_t);
m_ThisAdjustmentThunks[a_uiThisOffset] = func;
}
jit_value jit_function::insn_call
(const char *name, jit_function_t jit_func,
jit_type_t signature, jit_value_t *args, unsigned int num_args,
int flags)
{
value_wrap(jit_insn_call
(func, name, jit_func, signature, args, num_args, flags));
}
int main(int argc, char **argv)
{
jit_context_t context;
jit_type_t params[2];
jit_type_t signature;
jit_function_t function;
jit_value_t x, y;
jit_value_t temp1, temp2;
jit_value_t temp3, temp4;
jit_value_t temp_args[2];
jit_label_t label1 = jit_label_undefined;
jit_label_t label2 = jit_label_undefined;
jit_uint arg1, arg2;
void *args[2];
jit_uint result;
/* Create a context to hold the JIT's primary state */
context = jit_context_create();
/* Lock the context while we build and compile the function */
jit_context_build_start(context);
/* Build the function signature */
params[0] = jit_type_uint;
params[1] = jit_type_uint;
signature = jit_type_create_signature
(jit_abi_cdecl, jit_type_uint, params, 2, 1);
/* Create the function object */
function = jit_function_create(context, signature);
/* Check the condition "if(x == y)" */
x = jit_value_get_param(function, 0);
y = jit_value_get_param(function, 1);
temp1 = jit_insn_eq(function, x, y);
jit_insn_branch_if_not(function, temp1, &label1);
/* Implement "return x" */
jit_insn_return(function, x);
/* Set "label1" at this position */
jit_insn_label(function, &label1);
/* Check the condition "if(x < y)" */
temp2 = jit_insn_lt(function, x, y);
jit_insn_branch_if_not(function, temp2, &label2);
/* Implement "return gcd(x, y - x)" */
temp_args[0] = x;
temp_args[1] = jit_insn_sub(function, y, x);
temp3 = jit_insn_call
(function, "gcd", function, 0, temp_args, 2, 0);
jit_insn_return(function, temp3);
/* Set "label2" at this position */
jit_insn_label(function, &label2);
/* Implement "return gcd(x - y, y)" */
temp_args[0] = jit_insn_sub(function, x, y);
temp_args[1] = y;
temp4 = jit_insn_call
(function, "gcd", function, 0, temp_args, 2, 0);
jit_insn_return(function, temp4);
/* Compile the function */
jit_function_compile(function);
/* Unlock the context */
jit_context_build_end(context);
/* Execute the function and print the result */
arg1 = 27;
arg2 = 14;
args[0] = &arg1;
args[1] = &arg2;
jit_function_apply(function, args, &result);
printf("gcd(27, 14) = %u\n", (unsigned int)result);
/* Clean up */
jit_context_destroy(context);
/* Finished */
return 0;
}
static jit_value_t parse_recursive(jit_function_t func) {
jit_value_t arg1, arg2, result;
jit_function_t func1;
double val;
char *t = token();
// Somebody, do something with this!
// It's awful monkeycoding, but I'm too lazy to rewrite it :3
if (STREQ(t, "F"))
result = val_freq;
else if (STREQ(t, "X"))
result = jit_insn_convert(func, val_sample, jit_type_float64, 0);
else if (STREQ(t, "LEN"))
result = jit_insn_convert(func, val_length, jit_type_float64, 0);
else if (STREQ(t, "RATE"))
result = const_rate;
else if (STREQ(t, "PI"))
result = const_pi;
else if (STREQ(t, "+"))
result = jit_insn_add(func, parse_recursive(func), parse_recursive(func));
else if (STREQ(t, "-")) {
arg1 = parse_recursive(func);
arg2 = parse_recursive(func);
result = jit_insn_sub(func, arg1, arg2);
}
else if (STREQ(t, "*"))
result = jit_insn_mul(func, parse_recursive(func), parse_recursive(func));
else if (STREQ(t, "/")) {
arg1 = parse_recursive(func);
arg2 = parse_recursive(func);
result = jit_insn_div(func, arg1, arg2);
}
else if (STREQ(t, "%")) {
arg1 = parse_recursive(func);
arg2 = parse_recursive(func);
result = jit_insn_rem(func, arg1, arg2);
}
else if (STREQ(t, ">")) {
arg1 = parse_recursive(func);
arg2 = parse_recursive(func);
result = jit_insn_gt(func, arg1, arg2);
}
else if (STREQ(t, "<")) {
arg1 = parse_recursive(func);
arg2 = parse_recursive(func);
result = jit_insn_lt(func, arg1, arg2);
}
else if (STREQ(t, "if")) {
jit_value_t tmpval = jit_value_create(func, jit_type_float64);
jit_label_t lb_false = jit_label_undefined,
lb_end = jit_label_undefined;
jit_insn_branch_if_not(func, jit_insn_to_bool(func, parse_recursive(func)), &lb_false);
jit_insn_store(func, tmpval, parse_recursive(func));
jit_insn_branch(func, &lb_end);
jit_insn_label(func, &lb_false);
jit_insn_store(func, tmpval, parse_recursive(func));
jit_insn_label(func, &lb_end);
result = jit_insn_load(func, tmpval);
}
else if (STREQ(t, "sin"))
result = jit_insn_sin(func, parse_recursive(func));
else if (sscanf(t, "%lf", &val) == 1)
result = jit_value_create_float64_constant(func, jit_type_float64, val);
else if ((func1 = get_function_by_name(t)) != NULL) {
arg1 = parse_recursive(func);
arg2 = parse_recursive(func);
jit_value_t args[3] = {arg1, arg2, val_length};
result = jit_insn_call(
func,
t,
func1,
NULL,
args,
3,
0);
}
else {
LOGF("Unexpected token '%s'", t);
result = NULL;
}
free(t);
return result;
}
jit_function_t GPerlJITCompiler::compile(JITParam *param)
{
GPerlVirtualMachineCode *pc = param->mtd;
jit_context_t ctx = jit_context_create();
jit_context_build_start(ctx);
int argc = param->argc;
jit_type_t _params[argc];
for (int i = 0; i < argc; i++) {
_params[i] = getJitType(param->arg_types[i]);
}
jit_type_t rtype = getJitType(param->return_type);
jit_value_t curstack[32] = {0};
jit_value_t argstack[MAX_ARGSTACK_SIZE] = {0};
jit_type_t signature = jit_type_create_signature(jit_abi_fastcall, rtype, _params, argc, 0);
jit_function_t func = jit_function_create(ctx, signature);
jit_value_t _v[MAX_REG_SIZE] = {0};
GPerlJmpStack *jmp_stack = new GPerlJmpStack();
argc = 0;
for (; pc->op != UNDEF; pc++) {
if (jmp_stack->isJmp()) {
GPerlJmpInfo *inf = jmp_stack->pop();
jit_insn_label(func, &inf->label);
}
switch (pc->op) {
case LET:
DBG_PL("COMPILE LET");
curstack[pc->dst] = _v[pc->src];
break;
case MOV:
DBG_PL("COMPILE MOV");
_v[pc->dst] = compileMOV(pc, &func);
break;
case ARRAY_ARGAT: {
DBG_PL("COMPILE ARGAT");
jit_value_t arg = jit_value_get_param(func, pc->src);
jit_nint offset = 48;
jit_value_t list = jit_insn_load_relative(func, arg, offset, object_ptr_type);
jit_value_t idx = jit_value_create_nint_constant(func, jit_type_int, pc->idx);
jit_value_t elem = jit_insn_load_elem(func, list, idx, value_type);
_v[pc->dst] = elem;
break;
}
case vMOV:
DBG_PL("COMPILE vMOV");
_v[pc->dst] = curstack[pc->src];
break;
case gMOV:
break;
case ARGMOV:
DBG_PL("COMPILE ARGMOV");
_v[pc->dst] = jit_value_get_param(func, pc->src);
break;
case ADD:
DBG_PL("COMPILE ADD");
_v[pc->dst] = jit_insn_add(func, _v[pc->dst], _v[pc->src]);
break;
case SUB:
DBG_PL("COMPILE SUB");
_v[pc->dst] = jit_insn_sub(func, _v[pc->dst], _v[pc->src]);
break;
case iADDC: {
DBG_PL("COMPILE iADDC");
jit_value_t c = jit_value_create_nint_constant(func, jit_type_int, pc->v.ivalue);
_v[pc->dst] = jit_insn_add(func, _v[pc->dst], c);
break;
}
case iSUBC: {
DBG_PL("COMPILE iSUBC");
jit_value_t c = jit_value_create_nint_constant(func, jit_type_int, pc->v.ivalue);
_v[pc->dst] = jit_insn_sub(func, _v[pc->dst], c);
break;
}
case IS: {
DBG_PL("COMPILE IS");
jit_value_t c = jit_value_create_nint_constant(func, jit_type_int, 1);
jit_value_t tmp = jit_insn_eq(func, _v[pc->dst], c);
GPerlJmpInfo *inf = new GPerlJmpInfo(pc->jmp);
jmp_stack->push(inf);
jit_insn_branch_if_not(func, tmp, &inf->label);
break;
}
case iJLC: {
DBG_PL("COMPILE iJLC");
jit_value_t c = jit_value_create_nint_constant(func, jit_type_int, pc->v.ivalue);
jit_value_t tmp = jit_insn_lt(func, _v[pc->dst], c);
GPerlJmpInfo *inf = new GPerlJmpInfo(pc->jmp);
jmp_stack->push(inf);
jit_insn_branch_if_not(func, tmp, &inf->label);
break;
}
case iJLEC: {
DBG_PL("COMPILE iJLEC");
jit_value_t c = jit_value_create_nint_constant(func, jit_type_int, pc->v.ivalue);
jit_value_t tmp = jit_insn_le(func, _v[pc->dst], c);
GPerlJmpInfo *inf = new GPerlJmpInfo(pc->jmp);
jmp_stack->push(inf);
jit_insn_branch_if_not(func, tmp, &inf->label);
break;
}
case JE: {
DBG_PL("COMPILE JE");
jit_value_t tmp = jit_insn_eq(func, _v[pc->dst], _v[pc->src]);
GPerlJmpInfo *inf = new GPerlJmpInfo(pc->jmp);
jmp_stack->push(inf);
jit_insn_branch_if_not(func, tmp, &inf->label);
break;
}
case JLE: {
DBG_PL("COMPILE JLE");
jit_value_t tmp = jit_insn_le(func, _v[pc->dst], _v[pc->src]);
GPerlJmpInfo *inf = new GPerlJmpInfo(pc->jmp);
jmp_stack->push(inf);
jit_insn_branch_if_not(func, tmp, &inf->label);
break;
}
case PUSH:
DBG_PL("COMPILE PUSH");
argstack[pc->src] = _v[pc->dst];
argc++;
break;
case SELFCALL: {
DBG_PL("COMPILE SELFCALL");
_v[pc->dst] = jit_insn_call(func, "", func, NULL, argstack, argc, 0);//JIT_CALL_TAIL);
argc = 0;
break;
}
case SELF_FASTCALL0: {
DBG_PL("COMPILE SELF_FASTCALL0");
argstack[0] = _v[pc->arg0];
_v[pc->dst] = jit_insn_call(func, "", func, NULL, argstack, 1, 0);//JIT_CALL_TAIL);
break;
}
case SELF_FASTCALL1: {
DBG_PL("COMPILE SELF_FASTCALL1");
argstack[0] = _v[pc->arg0];
argstack[1] = _v[pc->arg1];
_v[pc->dst] = jit_insn_call(func, "", func, NULL, argstack, 2, 0);//JIT_CALL_TAIL);
break;
}
case SELF_FASTCALL2: {
DBG_PL("COMPILE SELF_FASTCALL2");
argstack[0] = _v[pc->arg0];
argstack[1] = _v[pc->arg1];
argstack[2] = _v[pc->arg2];
_v[pc->dst] = jit_insn_call(func, "", func, NULL, argstack, 3, 0);//JIT_CALL_TAIL);
break;
}
case SELF_FASTCALL3: {
DBG_PL("COMPILE SELF_FASTCALL3");
argstack[0] = _v[pc->arg0];
argstack[1] = _v[pc->arg1];
argstack[2] = _v[pc->arg2];
argstack[3] = _v[pc->arg3];
_v[pc->dst] = jit_insn_call(func, "", func, NULL, argstack, 4, 0);//JIT_CALL_TAIL);
break;
}
case REF: {
//int ret = 0;
//if (TYPE_CHECK(v) > 1) {
//GPerlObject *o = (GPerlObject *)getObject(v);
//if (o->h.type == ArrayRef) {
//ret = 1;
//}
//}
//INT_init(reg[0], ret);
break;
}
case RET: case JIT_COUNTDOWN_RET:
DBG_PL("COMPILE RET");
_v[0] = _v[pc->src];
jit_insn_return(func, _v[0]);
break;
default:
DBG_PL("COMPILE DEFALUT");
break;
}
}
jit_function_compile(func);
jit_context_build_end(ctx);
return func;
}
