void as_3_function::compile_stack_resize( int count )
{
#ifdef __GAMESWF_ENABLE_JIT__
jit_load( m_compiled_code, jit_this_pointer, jit_getarg( 1 ) );
jit_this_call( m_compiled_code, &array<as_value>::size );
jit_subi( m_compiled_code, jit_eax, count );
jit_pusharg( m_compiled_code, jit_eax );
jit_load( m_compiled_code, jit_this_pointer, jit_getarg( 1 ) );
jit_this_call( m_compiled_code, &array<as_value>::resize );
jit_popargs( m_compiled_code, 1 );
#endif
}
jit_value pop_reg(Frame* frame) {
struct jit* jit = frame->jit;
jit_movr(jit, frame->accum[0], frame->null_ptr);
jit_op* underflow_jump = jit_beqi(jit, (intptr_t) JIT_FORWARD, frame->stack_offset, 0);
jit_subi(jit, frame->stack_offset, frame->stack_offset, 1);
jit_muli(jit, frame->accum[ACCUM_COUNT-1], frame->stack_offset, sizeof(YValue*));
jit_ldxr(jit, frame->accum[0], frame->stack_ptr, frame->accum[ACCUM_COUNT-1], sizeof(YValue*));
jit_patch(jit, underflow_jump);
jit_prepare(jit);
jit_putargr(jit, frame->accum[0]);
jit_call(jit, dec_linkc);
return frame->accum[0];
}
jit_pointer_t compile(const std::vector<Oper>& ops, jit_word_t *memory, const bool flush = true)
{
jit_prolog();
jit_movi(JIT_V0, reinterpret_cast<jit_word_t>(memory));
jit_movi(JIT_V1, 0);
std::stack<Loop> loops;
jit_node_t* start = jit_note(__FILE__, __LINE__);
for ( size_t n=0; n<ops.size(); ++n ) {
switch ( ops[n].code ) {
case '<':
jit_str(JIT_V0, JIT_V1);
jit_subi(JIT_V0, JIT_V0, ops[n].count * sizeof(jit_word_t));
jit_ldr(JIT_V1, JIT_V0);
break;
case '>':
jit_str(JIT_V0, JIT_V1);
jit_addi(JIT_V0, JIT_V0, ops[n].count * sizeof(jit_word_t));
jit_ldr(JIT_V1, JIT_V0);
break;
case 'z':
jit_movi(JIT_V1, 0);
break;
case '+':
jit_addi(JIT_V1, JIT_V1, ops[n].count);
break;
case '-':
jit_subi(JIT_V1, JIT_V1, ops[n].count);
break;
case '.':
jit_prepare();
jit_pushargr(JIT_V1);
jit_finishi(reinterpret_cast<jit_pointer_t>(putchar));
if ( flush ) {
jit_prepare();
jit_pushargi(reinterpret_cast<jit_word_t>(stdout));
jit_finishi(reinterpret_cast<jit_pointer_t>(fflush));
}
break;
case ',':
jit_prepare();
jit_finishi(reinterpret_cast<jit_pointer_t>(getchar));
jit_retval(JIT_V1);
break;
case '[': {
Loop loop;
loop.end = jit_forward();
jit_node_t *j = jit_beqi(JIT_V1, 0);
jit_patch_at(j, loop.end);
loop.body = jit_label();
loops.push(loop);
} break;
case ']': {
Loop loop = loops.top();
jit_node_t *j = jit_bnei(JIT_V1, 0);
jit_patch_at(j, loop.body);
jit_link(loop.end);
loops.pop();
break;
}
default:
break;
}
}
jit_node_t* stop = jit_note(__FILE__, __LINE__);
jit_ret();
jit_epilog();
jit_pointer_t r = jit_emit();
fprintf(stderr, "compiled to %zu bytes\n", (char*)jit_address(stop) -
(char*)jit_address(start));
return r;
}
void
outrun(int ch, int count)
{
jit_node_t *argp;
switch(ch) {
case '!':
if (bytecell) tape_step = 1; else
tape_step = sizeof(int);
init_jit(NULL); // argv[0]);
_jit = jit_new_state();
start = jit_note(__FILE__, __LINE__);
jit_prolog();
/* Get the data area pointer */
argp = jit_arg();
jit_getarg(REG_P, argp);
break;
case '~':
jit_ret();
jit_epilog();
end = jit_note(__FILE__, __LINE__);
codeptr = jit_emit();
jit_clear_state();
codeptr(calloc(tape_step, tapelen));
jit_destroy_state();
finish_jit();
codeptr = 0;
if (loopstack) { free(loopstack); loopstack = 0; }
break;
case '>':
jit_addi(REG_P, REG_P, count * tape_step);
break;
case '<':
jit_subi(REG_P, REG_P, count * tape_step);
break;
case '+':
if (tape_step>1)
jit_ldr_i(REG_ACC, REG_P);
else
jit_ldr_uc(REG_ACC, REG_P);
jit_addi(REG_ACC, REG_ACC, count);
if (tape_step>1)
jit_str_i(REG_P, REG_ACC);
else
jit_str_c(REG_P, REG_ACC);
break;
case '-':
if (tape_step>1)
jit_ldr_i(REG_ACC, REG_P);
else
jit_ldr_uc(REG_ACC, REG_P);
jit_subi(REG_ACC, REG_ACC, count);
if (tape_step>1)
jit_str_i(REG_P, REG_ACC);
else
jit_str_c(REG_P, REG_ACC);
break;
case '[':
if (tape_step>1)
jit_ldr_i(REG_ACC, REG_P);
else
jit_ldr_uc(REG_ACC, REG_P);
if (stackptr >= maxstack) {
loopstack = realloc(loopstack,
((maxstack+=32)+2)*sizeof(*loopstack));
if (loopstack == 0) {
perror("loop stack realloc failure");
exit(1);
}
}
loopstack[stackptr] = jit_beqi(REG_ACC, 0);
loopstack[stackptr+1] = jit_label();
stackptr += 2;
break;
case ']':
if (tape_step>1)
jit_ldr_i(REG_ACC, REG_P);
else
jit_ldr_uc(REG_ACC, REG_P);
stackptr -= 2;
if (stackptr < 0) {
fprintf(stderr, "Code gen failure: Stack pointer negative.\n");
exit(1);
}
{
jit_node_t *ref;
ref = jit_bnei(REG_ACC, 0);
jit_patch_at(ref, loopstack[stackptr+1]);
jit_patch(loopstack[stackptr]);
}
break;
case '.':
if (tape_step>1)
jit_ldr_i(REG_ACC, REG_P);
else
jit_ldr_uc(REG_ACC, REG_P);
jit_prepare();
jit_pushargr(REG_ACC);
jit_finishi(putchar);
break;
case ',':
jit_prepare();
jit_pushargr(REG_ACC);
jit_finishi(getchar);
jit_retval(REG_ACC);
if (tape_step>1)
jit_str_i(REG_P, REG_ACC);
else
jit_str_c(REG_P, REG_ACC);
break;
}
}
