| 1 | #include <stdio.h> |
| 2 | #include <lightning.h> |
| 3 | |
| 4 | typedef int (*pifi)(int); /* Pointer to Int Function of Int */ |
| 5 | |
| 6 | static jit_state_t *_jit; |
| 7 | |
| 8 | void stack_push(int reg, int *sp) |
| 9 | { |
| 10 | jit_stxi_i (*sp, JIT_FP, reg); |
| 11 | *sp += sizeof (int); |
| 12 | } |
| 13 | |
| 14 | void stack_pop(int reg, int *sp) |
| 15 | { |
| 16 | *sp -= sizeof (int); |
| 17 | jit_ldxi_i (reg, JIT_FP, *sp); |
| 18 | } |
| 19 | |
| 20 | jit_node_t *compile_rpn(char *expr) |
| 21 | { |
| 22 | jit_node_t *in, *fn; |
| 23 | int stack_base, stack_ptr; |
| 24 | |
| 25 | fn = jit_note(NULL, 0); |
| 26 | jit_prolog(); |
| 27 | in = jit_arg_i(); |
| 28 | stack_ptr = stack_base = jit_allocai (32 * sizeof (int)); |
| 29 | |
| 30 | jit_getarg_i(JIT_R2, in); |
| 31 | |
| 32 | while (*expr) { |
| 33 | char buf[32]; |
| 34 | int n; |
| 35 | if (sscanf(expr, "%[0-9]%n", buf, &n)) { |
| 36 | expr += n - 1; |
| 37 | stack_push(JIT_R0, &stack_ptr); |
| 38 | jit_movi(JIT_R0, atoi(buf)); |
| 39 | } else if (*expr == 'x') { |
| 40 | stack_push(JIT_R0, &stack_ptr); |
| 41 | jit_movr(JIT_R0, JIT_R2); |
| 42 | } else if (*expr == '+') { |
| 43 | stack_pop(JIT_R1, &stack_ptr); |
| 44 | jit_addr(JIT_R0, JIT_R1, JIT_R0); |
| 45 | } else if (*expr == '-') { |
| 46 | stack_pop(JIT_R1, &stack_ptr); |
| 47 | jit_subr(JIT_R0, JIT_R1, JIT_R0); |
| 48 | } else if (*expr == '*') { |
| 49 | stack_pop(JIT_R1, &stack_ptr); |
| 50 | jit_mulr(JIT_R0, JIT_R1, JIT_R0); |
| 51 | } else if (*expr == '/') { |
| 52 | stack_pop(JIT_R1, &stack_ptr); |
| 53 | jit_divr(JIT_R0, JIT_R1, JIT_R0); |
| 54 | } else { |
| 55 | fprintf(stderr, "cannot compile: %s\n", expr); |
| 56 | abort(); |
| 57 | } |
| 58 | ++expr; |
| 59 | } |
| 60 | jit_retr(JIT_R0); |
| 61 | jit_epilog(); |
| 62 | return fn; |
| 63 | } |
| 64 | |
| 65 | int main(int argc, char *argv[]) |
| 66 | { |
| 67 | jit_node_t *nc, *nf; |
| 68 | pifi c2f, f2c; |
| 69 | int i; |
| 70 | |
| 71 | init_jit(argv[0]); |
| 72 | _jit = jit_new_state(); |
| 73 | |
| 74 | nc = compile_rpn("32x9*5/+"); |
| 75 | nf = compile_rpn("x32-5*9/"); |
| 76 | (void)jit_emit(); |
| 77 | c2f = (pifi)jit_address(nc); |
| 78 | f2c = (pifi)jit_address(nf); |
| 79 | jit_clear_state(); |
| 80 | |
| 81 | printf("\nC:"); |
| 82 | for (i = 0; i <= 100; i += 10) printf("%3d ", i); |
| 83 | printf("\nF:"); |
| 84 | for (i = 0; i <= 100; i += 10) printf("%3d ", c2f(i)); |
| 85 | printf("\n"); |
| 86 | |
| 87 | printf("\nF:"); |
| 88 | for (i = 32; i <= 212; i += 18) printf("%3d ", i); |
| 89 | printf("\nC:"); |
| 90 | for (i = 32; i <= 212; i += 18) printf("%3d ", f2c(i)); |
| 91 | printf("\n"); |
| 92 | |
| 93 | jit_destroy_state(); |
| 94 | finish_jit(); |
| 95 | return 0; |
| 96 | } |