X-Git-Url: http://deadsoftware.ru/gitweb?a=blobdiff_plain;f=src%2Fbackends%2Fjvm%2Fgenerator-jvm.c;h=4b5b9474c0a2f006c99e15c5e4251e37763785b1;hb=5f233e3ecee9f82dcf0db94e525f1a70caa67e6f;hp=9812f0ab7f5e2b56c243c94c97e6a0fc8ca86037;hpb=3b56d873cc052fcbe8914dba3b9ae9ea220a801e;p=dsw-obn.git diff --git a/src/backends/jvm/generator-jvm.c b/src/backends/jvm/generator-jvm.c index 9812f0a..4b5b947 100644 --- a/src/backends/jvm/generator-jvm.c +++ b/src/backends/jvm/generator-jvm.c @@ -1,5 +1,7 @@ #include +#include #include +#include #include #include #include @@ -9,255 +11,669 @@ #include "../../../include/oberon.h" #include "../../oberon-internals.h" #include "generator-jvm.h" +#include "generator-jvm-basic.h" -char * -new_string(const char * format, ...) +// ========================================== +// ========================================== +// ========================================== + +gen_proc_t * +jvm_create_proc(struct gen_class * class) +{ + gen_proc_t * p = GC_MALLOC(sizeof *p); + memset(p, 0, sizeof *p); + + struct gen_register_file * rf = GC_MALLOC(sizeof *rf); + memset(rf, 0, sizeof *rf); + + struct gen_stack * stack = GC_MALLOC(sizeof *stack); + memset(rf, 0, sizeof *stack); + + p -> rf = rf; + p -> stack = stack; + p -> class = class; + p -> label_id = 0; + + return p; +} + +void +jvm_generate_function_header(gen_proc_t * p, char * access, char * name, char * signature) +{ + struct gen_class * class; + class = p -> class; + + /* Делаем процедуру текущей в этом классе */ + assert(class -> p == NULL); + class -> p = p; + + fprintf(class -> fp, ".method %s %s%s\n", access, name, signature); + fprintf(class -> fp, " start:\n"); +} + +void +jvm_generate_function_end(gen_proc_t * p) +{ + struct gen_class * class; + class = p -> class; + + assert(class -> p); + class -> p = NULL; + + int pointer = p -> stack -> pointer; + int max_pointer = p -> stack -> max_pointer; + int locals = p -> rf -> num_used; + int max_locals = p -> rf -> max_used; + + fprintf(class -> fp, " .limit stack %i \t; current(%i)\n", max_pointer, pointer); + fprintf(class -> fp, " .limit locals %i \t; current(%i)\n", max_locals, locals); + fprintf(class -> fp, " end:\n"); + fprintf(class -> fp, ".end method\n\n"); +} + +struct gen_class * +jvm_create_class(char * full_name) +{ + struct gen_class * class = GC_MALLOC(sizeof *class); + memset(class, 0, sizeof *class); + + char * fname = new_string("%s.j", full_name); + class -> full_name = new_string(full_name); + class -> fp = fopen(fname, "w"); + assert(class -> fp); + + return class; +} + +static void +jvm_destroy_class(struct gen_class * class) +{ + assert(class -> p == NULL); + fclose(class -> fp); +} + +static void +jvm_stack_push(gen_proc_t * p, unsigned size) +{ + p -> stack -> pointer += size; + if(p -> stack -> pointer > p -> stack -> max_pointer) + { + p -> stack -> max_pointer = p -> stack -> pointer; + } +} + +static void +jvm_stack_pop(gen_proc_t * p, unsigned size) +{ + p -> stack -> pointer -= size; + + if(p -> stack -> pointer < 0) + { + printf("WARING: stack pointer %i\n", p -> stack -> pointer); + } +} + +static void +jvm_generate(gen_proc_t * p, unsigned get, unsigned push, char * format, ...) { va_list ptr; va_start(ptr, format); - char buf[1024]; - vsnprintf(buf, 1024, format, ptr); + jvm_stack_pop(p, get); + fprintf(p -> class -> fp, " "); + vfprintf(p -> class -> fp, format, ptr); + jvm_stack_push(p, push); + fprintf(p -> class -> fp, " \t ;>>>> %i -- %i : current_stack(%i)\n", get, push, p -> stack -> pointer); va_end(ptr); +} + +static void +jvm_generate_push_int(gen_proc_t * p, int64_t i) +{ + if(i == -1) + { + jvm_generate(p, 0, 1, "iconst_m1"); + } + else if(i >= 0 && i <= 5) + { + jvm_generate(p, 0, 1, "iconst_%li", i); + } + else if(i >= -128 && i <= 127) + { + jvm_generate(p, 0, 1, "bipush %li", i); + } + else if(i >= -32768 && i <= 32767) + { + jvm_generate(p, 0, 1, "sipush %li", i); + } + else if(i >= -2147483648 && i <= 2147483647) + { + jvm_generate(p, 0, 1, "ldc %li", i); + } + else + { + jvm_generate(p, 0, 2, "ldc2 %li", i); + } +} - char * result; - int size; +static void +jvm_generate_push_float(gen_proc_t * p, double f, int size) +{ + if(size <= 4) + { + if(f == 0.0) + { + jvm_generate(p, 0, 1, "fconst_0"); + } + if(f == 1.0) + { + jvm_generate(p, 0, 1, "fconst_1"); + } + if(f == 2.0) + { + jvm_generate(p, 0, 1, "fconst_2"); + } + else + { + jvm_generate(p, 0, 1, "ldc %lf", f); + } + } + else + { + if(f == 0.0) + { + jvm_generate(p, 0, 2, "dconst_0"); + } + if(f == 1.0) + { + jvm_generate(p, 0, 2, "dconst_1"); + } + else + { + jvm_generate(p, 0, 2, "ldc2 %lf", f); + } + } +} + +static gen_var_t * +oberon_generator_new_var() +{ + gen_var_t * v = GC_MALLOC(sizeof *v); + memset(v, 0, sizeof *v); + return v; +} - size = strlen(buf); - result = GC_MALLOC(size + 1); - memset(result, 0, size); - strncpy(result, buf, size); +static void +jvm_generate_and_init_global_var(gen_var_t * v, struct gen_class * class, char * access, char * name, char * desc) +{ + assert(class -> p == NULL); + fprintf(class -> fp, ".field %s static %s %s\n\n", access, name, desc); - return result; + v -> storage = JVM_STORAGE_STATIC; + v -> full_name = new_string("%s/%s", class -> full_name, name); + v -> desc = new_string(desc); } -void -oberon_generator_init_context(oberon_context_t * ctx) +static void +jvm_generate_and_init_field(gen_var_t * v, struct gen_class * class, char * access, char * name, char * desc) { - gen_context_t * gen_context = GC_MALLOC(sizeof *gen_context); - memset(gen_context, 0, sizeof *gen_context); + assert(class -> p == NULL); + fprintf(class -> fp, ".field %s %s %s\n\n", access, name, desc); - ctx -> gen_context = gen_context; + v -> storage = JVM_STORAGE_FIELD; + v -> full_name = new_string("%s/%s", class -> full_name, name); + v -> desc = new_string(desc); } -void -oberon_generator_destroy_context(oberon_context_t * ctx) +static void +jvm_generate_and_init_local_var(gen_var_t * v, gen_proc_t * p, bool wide) { - ctx -> gen_context = NULL; + v -> storage = JVM_STORAGE_REGISTER; + v -> reg = jvm_alloc_register_untyped(p -> rf, wide); } -static char * get_class_full_name(oberon_context_t * ctx, oberon_type_t * type); +static void +jvm_generate_and_init_named_local_var(gen_var_t * v, gen_proc_t * p, bool wide, char * name, char * desc) +{ + jvm_generate_and_init_local_var(v, p, wide); + v -> desc = new_string(desc); + jvm_generate(p, 0, 0, ".var %i is %s %s from start to end", v -> reg, name, desc); +} + +static int +jvm_new_label_id(gen_proc_t * p) +{ + int label_id = p -> label_id; + p -> label_id += 1; + return label_id; +} static char * -get_descriptor(oberon_context_t * ctx, oberon_type_t * type) +jvm_get_label_name(int label_id) +{ + return new_string("L%i", label_id); +} + +static void +jvm_generate_label(gen_proc_t * p, int label_id) +{ + jvm_generate(p, 0, 0, "L%i:", label_id); +} + +static void +jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num); + +/* + * Функция jvm_generate_static_array_initialization генерирует код для + * статической инициализации массива. На входе массив, на выходе тот же массив. + * (arrayref -- arrayref) + */ + +static void +jvm_generate_static_array_initialization(gen_proc_t * p, oberon_type_t * type) +{ + /* for(int i = 0; i < a.length; i++) */ + /* a[i] := new(base); */ + /* init(a[i]) */ + + /* a := pop(array) */ + gen_var_t * va = oberon_generator_new_var(); + jvm_generate_and_init_local_var(va, p, false); + int reg_a = va -> reg; + /* ссылка уже на стеке */ + jvm_generate(p, 1, 0, "astore %i", reg_a); + + /* int i = 0; */ + gen_var_t * vi = oberon_generator_new_var(); + jvm_generate_and_init_local_var(vi, p, false); + int reg_i = vi -> reg; + jvm_generate(p, 0, 1, "iconst_0"); + jvm_generate(p, 1, 0, "istore %i", reg_i); + + /* int l := a.length */ + gen_var_t * vl = oberon_generator_new_var(); + jvm_generate_and_init_local_var(vl, p, false); + int reg_l = vl -> reg; + jvm_generate(p, 0, 1, "aload %i", reg_a); + jvm_generate(p, 1, 1, "arraylength"); + jvm_generate(p, 1, 0, "istore %i", reg_l); + + int label_start = jvm_new_label_id(p); + int label_end = jvm_new_label_id(p); + + /* start: */ + /* if (i >= l) goto end; */ + /* body*/ + /* i += 1;*/ + /* goto start;*/ + /* end: */ + /* push a */ + + jvm_generate_label(p, label_start); + jvm_generate(p, 0, 1, "iload %i", reg_i); + jvm_generate(p, 0, 1, "iload %i", reg_l); + jvm_generate(p, 2, 0, "if_icmpge L%i", label_end); + + if(type -> base -> class == OBERON_TYPE_ARRAY) + { + /* Инициализируем следующую размерность */ + jvm_generate(p, 0, 1, "aload %i", reg_a); + jvm_generate(p, 0, 1, "iload %i", reg_i); + jvm_generate(p, 0, 1, "aaload"); + jvm_generate_static_array_initialization(p, type -> base); + jvm_generate(p, 1, 0, "pop"); + } + else if(type -> base -> class == OBERON_TYPE_RECORD) + { + jvm_generate(p, 0, 1, "aload %i", reg_a); + jvm_generate(p, 0, 1, "iload %i", reg_i); + jvm_generate_new(p, type -> base, 0); + jvm_generate(p, 3, 0, "aastore"); + } + else + { + assert(0); + } + + /* i += 1; */ + /* goto start */ + + jvm_generate(p, 0, 0, "iinc %i %i", reg_i, 1); + jvm_generate(p, 0, 0, "goto L%i", label_start); + jvm_generate_label(p, label_end); + + /* push(a) */ + jvm_generate(p, 0, 1, "aload %i", reg_a); + + /* TODO освобождение регистров */ +} + +static void +jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num) { + int dim; + char * cname; char * desc; + oberon_type_t * base; switch(type -> class) { - case OBERON_TYPE_VOID: - return new_string("V"); - break; case OBERON_TYPE_INTEGER: - switch(type -> size) + case OBERON_TYPE_BOOLEAN: + case OBERON_TYPE_PROCEDURE: + case OBERON_TYPE_REAL: + case OBERON_TYPE_POINTER: + gen_error("jvm_generate_new_static: static alocation not allowed"); + break; + case OBERON_TYPE_RECORD: + cname = jvm_get_class_full_name(type); + jvm_generate(p, 0, 1, "new %s", cname); + jvm_generate(p, 1, 2, "dup"); + jvm_generate(p, 1, 0, "invokespecial %s/()V", cname); + break; + case OBERON_TYPE_ARRAY: + dim = 0; + base = type; + desc = jvm_get_descriptor(type); + while(base -> class == OBERON_TYPE_ARRAY) { - case 1: - return new_string("B"); - break; - case 2: - return new_string("S"); - break; - case 4: - return new_string("I"); - break; - case 8: - return new_string("J"); - break; - default: - oberon_error(ctx, "get_descriptor: unsupported int size %i", type -> size); - break; + if(num > 0) + { + assert(base -> size == 0); + num -= 1; + } + else + { + assert(base -> size > 0); + jvm_generate_push_int(p, base -> size); + } + dim += 1; + base = base -> base; } - break; - case OBERON_TYPE_REAL: - switch(type -> size) + + assert(num == 0); + jvm_generate(p, dim, 1, "multianewarray %s %i", desc, dim); + + if(base -> class == OBERON_TYPE_RECORD) { - case 4: - return new_string("F"); - break; - case 8: - return new_string("D"); - break; - default: - oberon_error(ctx, "get_descriptor: unsupported float size %i", type -> size); - break; + jvm_generate_static_array_initialization(p, type); } break; - case OBERON_TYPE_BOOLEAN: - return new_string("Z"); + default: + gen_error("jvm_generate_new_static: unk type class %i", type -> class); break; - case OBERON_TYPE_PROCEDURE: - case OBERON_TYPE_RECORD: - desc = get_class_full_name(ctx, type); - return new_string("L%s;", desc); + } +} + +static void +jvm_generate_load(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * src) +{ + char prefix; + int cell_size; + + cell_size = jvm_cell_size_for_type(src_type); + + switch(src -> storage) + { + case JVM_STORAGE_REGISTER: + prefix = jvm_get_prefix(src_type); + jvm_generate(p, 0, cell_size, "%cload %i", prefix, src -> reg); break; - case OBERON_TYPE_ARRAY: - desc = get_descriptor(ctx, type -> base); - return new_string("[%s", desc); + case JVM_STORAGE_STATIC: + jvm_generate(p, 0, cell_size, "getstatic %s %s", src -> full_name, src -> desc); + break; + case JVM_STORAGE_FIELD: + jvm_generate(p, 1, cell_size, "getfield %s %s", src -> full_name, src -> desc); break; default: - oberon_error(ctx, "print_descriptor: unsupported type class %i", type -> class); + gen_error("jvm_generate_load: unknow storage type %i", src -> storage); break; } - - return NULL; } -static char -get_prefix(oberon_context_t * ctx, oberon_type_t * type) +static void +jvm_generate_store(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * dst) { - int size = type -> size; - switch(type -> class) + char prefix; + int cell_size; + + cell_size = jvm_cell_size_for_type(src_type); + + switch(dst -> storage) { - case OBERON_TYPE_BOOLEAN: - case OBERON_TYPE_INTEGER: - return (size <= 4) ? ('i') : ('l'); + case JVM_STORAGE_REGISTER: + prefix = jvm_get_prefix(src_type); + jvm_generate(p, cell_size, 0, "%cstore %i", prefix, dst -> reg); break; - case OBERON_TYPE_PROCEDURE: - case OBERON_TYPE_ARRAY: - case OBERON_TYPE_RECORD: - case OBERON_TYPE_POINTER: - return 'a'; + case JVM_STORAGE_STATIC: + jvm_generate(p, cell_size, 0, "putstatic %s %s", dst -> full_name, dst -> desc); break; - case OBERON_TYPE_REAL: - return (size <= 4) ? ('f') : ('d'); + case JVM_STORAGE_FIELD: + jvm_generate(p, 1 + cell_size, 0, "putfield %s %s", dst -> full_name, dst -> desc); + break; + default: + gen_error("jvm_generate_store: unknow storage type %i", dst -> storage); break; } - - oberon_error(ctx, "get_prefix: wat"); - return '!'; } -static char * -get_field_full_name(oberon_context_t * ctx, oberon_object_t * x) +/* + * Генерирует код для копирования полей из первой записи во вторую. + * ( aref_dst aref_src -- ) + * dst := src; + */ + +static void +jvm_generate_copy_record(gen_proc_t * p, oberon_type_t * rec) { - return new_string("%s/%s", x -> module -> name, x -> name); + assert(rec -> class == OBERON_TYPE_RECORD); + char * desc = jvm_get_descriptor(rec); + char * cname = jvm_get_class_full_name(rec); + jvm_generate(p, 1 + 1, 0, "invokestatic %s/$COPY$(%s%s)V", cname, desc, desc); } -static char * -get_class_full_name(oberon_context_t * ctx, oberon_type_t * type) +/* + * Генерирует кода для копирования массивов. + * ( aref_dst aref_src -- ) + * dst := src; + */ + +static void +jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr) { - int rec_id; - char * name = NULL; + int dim = 0; + oberon_type_t * base = arr; + while(base -> class == OBERON_TYPE_ARRAY) + { + dim += 1; + base = base -> base; + } - switch(type -> class) + struct { + gen_var_t * index; + gen_var_t * length; + int start; + int end; + } loop[dim]; + + gen_var_t * dst = oberon_generator_new_var(); + gen_var_t * src = oberon_generator_new_var(); + jvm_generate_and_init_local_var(dst, p, false); + jvm_generate_and_init_local_var(src, p, false); + + jvm_generate_store(p, arr, src); + jvm_generate_store(p, arr, dst); + + /* Входящие параметры заграблены + * Теперь генерируем эквивалентный код: + * int i = 0; + * int len = dst.length + * while(i < len) + * { + * ... + * { + * copy from src[i, ...] to dst[i, ...]; + * } + * ... + * i += 1; + * } + * Где "..." такой же код (начало и конец) для следующей размерности. + */ + + for(int i = 0; i < dim; i++) { - case OBERON_TYPE_PROCEDURE: - name = new_string("SYSTEM$PROCEDURE"); + loop[i].index = oberon_generator_new_var(); + loop[i].length = oberon_generator_new_var(); + loop[i].start = jvm_new_label_id(p); + loop[i].end = jvm_new_label_id(p); + jvm_generate_and_init_local_var(loop[i].index, p, false); + jvm_generate_and_init_local_var(loop[i].length, p, false); + + /* TODO преределать через jvm_generate_load{{store}} */ + + jvm_generate(p, 0, 1, "iconst_0"); + jvm_generate(p, 1, 0, "istore %i", loop[i].index -> reg); + + jvm_generate_load(p, arr, dst); + jvm_generate(p, 1, 1, "arraylength"); + jvm_generate(p, 1, 0, "istore %i", loop[i].length -> reg); + + /* if(i >= len) goto end; */ + jvm_generate_label(p, loop[i].start); + jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg); + jvm_generate(p, 0, 1, "iload %i", loop[i].length -> reg); + jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end); + } - char * desc; - desc = get_descriptor(ctx, type -> base); - name = new_string("%s$%s", name, desc); + if(base -> class == OBERON_TYPE_ARRAY) + { + // Вся эта шляпа уже должна знать о всех размерностях + gen_error("jvm_generate_copy_array: wat"); + } + else if(base -> class == OBERON_TYPE_RECORD) + { + /* Получаем записи по индексам ( -- dst src ) */ - int num = type -> num_decl; - oberon_object_t * arg = type -> decl; - for(int i = 0; i < num; i++) - { - desc = get_descriptor(ctx, arg -> type); - name = new_string("%s%s", name, desc); - arg = arg -> next; - } + jvm_generate_load(p, arr, dst); + for(int i = 0; i < dim; i++) + { + jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg); + jvm_generate(p, 2, 1, "aaload"); + } - break; - case OBERON_TYPE_RECORD: - assert(type -> module); - assert(type -> module -> gen_mod); - rec_id = type -> gen_type -> rec_id; - name = new_string("%s$RECORD%i", type -> module -> name, rec_id); - break; - default: - oberon_error(ctx, "get_record_full_name: unk type class %i", type -> class); - break; + jvm_generate_load(p, arr, src); + for(int i = 0; i < dim; i++) + { + jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg); + jvm_generate(p, 2, 1, "aaload"); + } + + /* Копируем записи ( dst src -- ) */ + jvm_generate_copy_record(p, base); } + else + { + char postfix = jvm_get_postfix(base); + int cell_size = jvm_cell_size_for_postfix(postfix); - return name; -} + /* Получаем массивы и индексы ( -- dst i src i ) */ -static char * -get_procedure_signature(oberon_context_t * ctx, oberon_type_t * proc) -{ - char * signature; - char * desc; + jvm_generate_load(p, arr, dst); + jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg); + for(int i = 1; i < dim; i++) + { + jvm_generate(p, 2, 1, "aaload"); + jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg); + } - signature = new_string("("); + jvm_generate_load(p, arr, src); + jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg); + for(int i = 1; i < dim; i++) + { + jvm_generate(p, 2, 1, "aaload"); + jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg); + } - int num = proc -> num_decl; - oberon_object_t * arg = proc -> decl; - for(int i = 0; i < num; i++) + /* Копируем значения ( dst i src i -- ) */ + jvm_generate(p, 2, cell_size, "%caload", postfix); + jvm_generate(p, 2 + cell_size, 0, "%castore", postfix); + } + + for(int i = dim - 1; i >= 0; i--) { - desc = get_descriptor(ctx, arg -> type); - signature = new_string("%s%s", signature, desc); - arg = arg -> next; + jvm_generate(p, 0, 0, "iinc %i 1", loop[i].index -> reg); + jvm_generate(p, 0, 0, "goto L%i", loop[i].start); + jvm_generate_label(p, loop[i].end); } +} - desc = get_descriptor(ctx, proc -> base); - signature = new_string("%s)%s", signature, desc); +// ========================================== +// ========================================== +// ========================================== - return signature; +void +oberon_generator_init_context(oberon_context_t * ctx) +{ + gen_context_t * gen_context = GC_MALLOC(sizeof *gen_context); + memset(gen_context, 0, sizeof *gen_context); + + ctx -> gen_context = gen_context; +} + +void +oberon_generator_destroy_context(oberon_context_t * ctx) +{ } static void -oberon_generate_procedure_class(oberon_context_t * ctx, oberon_type_t * proc) +oberon_generate_procedure_class(oberon_type_t * proc) { FILE * fp; char * cname; - char * fname; char * signature; + struct gen_class * class; - cname = get_class_full_name(ctx, proc); - fname = new_string("%s.j", cname); - - fp = fopen(fname, "w"); + cname = jvm_get_class_full_name(proc); + class = jvm_create_class(cname); + fp = class -> fp; fprintf(fp, ".source SYSTEM\n"); fprintf(fp, ".class public abstract %s\n", cname); fprintf(fp, ".super java/lang/Object\n\n"); - signature = get_procedure_signature(ctx, proc); - fprintf(fp, ".method public ()V\n"); fprintf(fp, " aload_0\n"); fprintf(fp, " invokespecial java/lang/Object/()V\n"); fprintf(fp, " return\n"); fprintf(fp, ".end method\n\n"); + signature = jvm_get_procedure_signature(proc); + fprintf(fp, ".method public abstract invoke%s\n", signature); fprintf(fp, ".end method\n\n"); - fclose(fp); + jvm_destroy_class(class); } static void -oberon_generate_record_class(oberon_context_t * ctx, oberon_type_t * rec) +oberon_generate_record_class(gen_module_t * m, oberon_type_t * rec) { - FILE * fp; char * cname; - char * fname; + struct gen_class * class; /* Устанавливаем новоый id */ - rec -> gen_type -> rec_id = rec -> module -> gen_mod -> rec_id; - rec -> module -> gen_mod -> rec_id += 1; + rec -> gen_type -> rec_id = m -> rec_id; + m -> rec_id += 1; - cname = get_class_full_name(ctx, rec); - fname = new_string("%s.j", cname); + cname = jvm_get_class_full_name(rec); + class = jvm_create_class(cname); - fp = fopen(fname, "w"); + fprintf(class -> fp, ".source %s\n", rec -> module -> name); + fprintf(class -> fp, ".class public %s\n", cname); + fprintf(class -> fp, ".super java/lang/Object\n\n"); - fprintf(fp, ".source %s\n", rec -> module -> name); - fprintf(fp, ".class public %s\n", cname); - fprintf(fp, ".super java/lang/Object\n\n"); - - rec -> gen_type -> fp = fp; + rec -> gen_type -> class = class; } void @@ -276,38 +692,18 @@ oberon_generator_init_type(oberon_context_t * ctx, oberon_type_t * type) case OBERON_TYPE_REAL: break; case OBERON_TYPE_RECORD: - oberon_generate_record_class(ctx, type); + ; + gen_module_t * m; + m = type -> module -> gen_mod; + oberon_generate_record_class(m, type); break; case OBERON_TYPE_PROCEDURE: - oberon_generate_procedure_class(ctx, type); + oberon_generate_procedure_class(type); break; case OBERON_TYPE_POINTER: - assert(type -> base -> class == OBERON_TYPE_VOID); - break; - default: - oberon_error(ctx, "oberon_generator_init_type: unk calss %i", type -> class); - break; - } -} - -static void -oberon_generate_object(oberon_context_t * ctx, FILE * fp, oberon_object_t * x) -{ - char * name; - char * desc; - - name = x -> name; - desc = get_descriptor(ctx, x -> type); - switch(x -> class) - { - case OBERON_CLASS_VAR: - fprintf(fp, ".field public static %s %s\n\n", name, desc); - break; - case OBERON_CLASS_FIELD: - fprintf(fp, ".field public %s %s\n\n", name, desc); break; default: - oberon_error(ctx, "oberon_generate_object: unk class %i", x -> class); + gen_error("oberon_generator_init_type: unk calss %i", type -> class); break; } } @@ -315,49 +711,125 @@ oberon_generate_object(oberon_context_t * ctx, FILE * fp, oberon_object_t * x) void oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec) { - FILE * fp; - - fp = rec -> gen_type -> fp; + struct gen_class * class; + class = rec -> gen_type -> class; int num = rec -> num_decl; oberon_object_t * field = rec -> decl; for(int i = 0; i < num; i++) { - oberon_generate_object(ctx, fp, field); + gen_var_t * v = field -> gen_var; + char * name = field -> name; + char * desc = jvm_get_descriptor(field -> type); + jvm_generate_and_init_field(v, class, "public", name, desc); field = field -> next; } - fprintf(fp, ".method public ()V\n"); - fprintf(fp, " aload_0\n"); - fprintf(fp, " invokespecial java/lang/Object/()V\n"); - fprintf(fp, " return\n"); - fprintf(fp, ".end method\n"); + /* Стандартный конструктор класса */ + /* Инициализирует внутренние статические записи и массивы */ + gen_proc_t * p = jvm_create_proc(class); + jvm_generate_function_header(p, "public", "", "()V"); + jvm_alloc_register_untyped(p -> rf, false); + jvm_generate(p, 0, 1, "aload_0"); + jvm_generate(p, 1, 0, "invokespecial java/lang/Object/()V"); + num = rec -> num_decl; + field = rec -> decl; + for(int i = 0; i < num; i++) + { + if(field -> type -> class == OBERON_TYPE_RECORD + || field -> type -> class == OBERON_TYPE_ARRAY) + { + jvm_generate(p, 0, 1, "aload_0"); + jvm_generate_new(p, field -> type, 0); + jvm_generate_store(p, field -> type, field -> gen_var); + } + field = field -> next; + } + jvm_generate(p, 0, 0, "return"); + jvm_generate_function_end(p); + + /* Метод для копирования полей класса */ + /* reg0 == src -> reg1 == dst */ + p = jvm_create_proc(class); + gen_var_t * copy_dst = oberon_generator_new_var(); + gen_var_t * copy_src = oberon_generator_new_var(); + char * desc = jvm_get_descriptor(rec); + char * signature = new_string("(%s%s)V", desc, desc); + jvm_generate_function_header(p, "public static", "$COPY$", signature); + jvm_generate_and_init_named_local_var(copy_dst, p, false, "dst", desc); + jvm_generate_and_init_named_local_var(copy_src, p, false, "src", desc); + num = rec -> num_decl; + field = rec -> decl; + for(int i = 0; i < num; i++) + { + if(field -> type -> class == OBERON_TYPE_RECORD) + { + jvm_generate_load(p, rec, copy_dst); + jvm_generate_load(p, field -> type, field -> gen_var); + jvm_generate_load(p, rec, copy_src); + jvm_generate_load(p, field -> type, field -> gen_var); + jvm_generate_copy_record(p, field -> type); + } + else if(field -> type -> class == OBERON_TYPE_ARRAY) + { + jvm_generate_load(p, rec, copy_dst); + jvm_generate_load(p, field -> type, field -> gen_var); + jvm_generate_load(p, rec, copy_src); + jvm_generate_load(p, field -> type, field -> gen_var); + jvm_generate_copy_array(p, field -> type); + } + else + { + jvm_generate_load(p, rec, copy_dst); + jvm_generate_load(p, rec, copy_src); + jvm_generate_load(p, field -> type, field -> gen_var); + jvm_generate_store(p, field -> type, field -> gen_var); + } + field = field -> next; + } + jvm_generate(p, 0, 0, "return"); + jvm_generate_function_end(p); - fclose(fp); + jvm_destroy_class(class); } void oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var) { - gen_var_t * v = GC_MALLOC(sizeof *v); - memset(v, 0, sizeof *v); + gen_var_t * v = oberon_generator_new_var(); var -> gen_var = v; - gen_module_t * m = ctx -> mod -> gen_mod; + gen_module_t * m; + m = ctx -> mod -> gen_mod; + + struct gen_class * class; + class = m -> class; + char * desc; switch(var -> class) { case OBERON_CLASS_VAR_PARAM: - oberon_error(ctx, "generator: VAR-parameters not implemented"); + gen_error("generator: VAR-parameters not implemented"); break; case OBERON_CLASS_PARAM: + /* Заполняется при генерации функции */ + /* смотри jvm_init_local_object() */ + break; case OBERON_CLASS_FIELD: + /* Заполняются при инициализации структуры */ + /* смотри oberon_generator_init_record() */ break; case OBERON_CLASS_VAR: - oberon_generate_object(ctx, m -> fp, var); + /* Локальные заполняются при генерации функции */ + /* смотри jvm_init_local_object() */ + if(var -> local == 0) + { + desc = jvm_get_descriptor(var -> type); + jvm_generate_and_init_global_var(v, class, "public", var -> name, desc); + } break; default: - oberon_error(ctx, "oberon_generator_init_var: unk var class %i", var -> class); + gen_error("oberon_generator_init_var: unk var class %i", var -> class); break; } } @@ -365,13 +837,12 @@ oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var) void oberon_generator_init_proc(oberon_context_t * ctx, oberon_object_t * proc) { - gen_proc_t * p = GC_MALLOC(sizeof *p); - memset(p, 0, sizeof *p); - proc -> gen_proc = p; + struct gen_class * class = proc -> module -> gen_mod -> class; + proc -> gen_proc = jvm_create_proc(class); if(proc -> local) { - oberon_error(ctx, "generator: local procedures not implemented"); + gen_error("generator: local procedures not implemented"); } } @@ -382,298 +853,359 @@ oberon_generator_init_module(oberon_context_t * ctx, oberon_module_t * mod) memset(m, 0, sizeof *m); mod -> gen_mod = m; - int fnamesz = strlen(mod -> name) + 3; - char fname[fnamesz + 1]; - snprintf(fname, fnamesz, "%s.j", mod -> name); + struct gen_class * class; + class = jvm_create_class(mod -> name); - FILE * fp; - fp = fopen(fname, "w"); - assert(fp != NULL); + fprintf(class -> fp, ".source %s\n", mod -> name); + fprintf(class -> fp, ".class %s\n", mod -> name); + fprintf(class -> fp, ".super java/lang/Object\n\n"); - fprintf(fp, ".source %s\n", mod -> name); - fprintf(fp, ".class %s\n", mod -> name); - fprintf(fp, ".super java/lang/Object\n\n"); + m -> class = class; - m -> fp = fp; + ctx -> gen_context -> current_m = m; } -// ======================================================================= -// GENERATOR -// ======================================================================= +void +oberon_generator_fini_module(oberon_module_t * mod) +{ + jvm_destroy_class(mod -> gen_mod -> class); +} static void -push_expr(oberon_context_t * ctx, FILE * fp, oberon_expr_t * expr); +push_expr(gen_proc_t * p, oberon_expr_t * expr); void oberon_generate_begin_module(oberon_context_t * ctx) { - gen_module_t * m = ctx -> mod -> gen_mod; - fprintf(m -> fp, ".method public ()V\n"); - fprintf(m -> fp, " aload_0\n"); - fprintf(m -> fp, " invokespecial java/lang/Object/()V\n"); + struct gen_class * class = ctx -> mod -> gen_mod -> class; + gen_proc_t * p = jvm_create_proc(class); + + jvm_generate_function_header(p, "public", "", "()V"); + jvm_alloc_register_untyped(p -> rf, false); + jvm_generate(p, 0, 1, "aload_0"); + jvm_generate(p, 1, 0, "invokespecial java/lang/Object/()V"); + + /* Инициализация переменных объявленных в модуле */ + oberon_object_t * x = ctx -> mod -> decl -> list -> next; + while(x != NULL) + { + if(x -> class == OBERON_CLASS_VAR) + { + if(x -> type -> class == OBERON_TYPE_ARRAY + || x -> type -> class == OBERON_TYPE_RECORD) + { + jvm_generate_new(p, x -> type, 0); + jvm_generate_store(p, x -> type, x -> gen_var); + } + } + x = x -> next; + } } void oberon_generate_end_module(oberon_context_t * ctx) { - gen_module_t * m = ctx -> mod -> gen_mod; + struct gen_class * class = ctx -> mod -> gen_mod -> class; + gen_proc_t * p = class -> p; - fprintf(m -> fp, " .limit stack 32\n"); - fprintf(m -> fp, " .limit locals 32\n"); - fprintf(m -> fp, " return\n"); - fprintf(m -> fp, ".end method\n"); + jvm_generate(p, 0, 0, "return"); + jvm_generate_function_end(class -> p); +} + +static void +jvm_init_local_object(gen_proc_t * p, oberon_object_t * x) +{ + gen_var_t * v; + bool wide; + char * desc; + struct gen_class * class; + + v = x -> gen_var; + class = p -> class; + wide = jvm_is_wide_type(x -> type); + desc = jvm_get_descriptor(x -> type); + + // Убеждаемся что сейчас находимся в функции + assert(class -> p); + assert(x -> local); + + switch(x -> class) + { + case OBERON_CLASS_VAR: + case OBERON_CLASS_PARAM: + jvm_generate_and_init_named_local_var(v, p, wide, x -> name, desc); + break; + default: + gen_error("jvm_init_local_object: wat"); + break; + } } void oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc) { - gen_module_t * m; + gen_proc_t * p; char * signature; - m = ctx -> mod -> gen_mod; - signature = get_procedure_signature(ctx, proc -> type); + p = proc -> gen_proc; + + signature = jvm_get_procedure_signature(proc -> type); + + jvm_generate_function_header(p, "public static", proc -> name, signature); - fprintf(m -> fp, ".method public static %s%s\n", proc -> name, signature); + oberon_object_t * var = proc -> type -> decl; + while(var) + { + jvm_init_local_object(p, var); + var = var -> next; + } } void -oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig) +oberon_generate_end_proc(oberon_context_t * ctx) { - oberon_object_t * proc; gen_module_t * m; - char * fullname; - char * signature; + gen_proc_t * p; + + m = ctx -> mod -> gen_mod; + p = m -> class -> p; + + jvm_generate_function_end(p); +} +static void +jvm_generate_call_proc(gen_proc_t * p, oberon_expr_t * desig) +{ assert(desig -> is_item); assert(desig -> item.mode == MODE_CALL); + + oberon_object_t * proc; + char * fullname; + char * signature; - m = ctx -> mod -> gen_mod; proc = desig -> item.var; - fullname = get_field_full_name(ctx, proc); - signature = get_procedure_signature(ctx, proc -> type); + fullname = jvm_get_field_full_name(proc); + signature = jvm_get_procedure_signature(proc -> type); + + int args_cells = 0; + int result_cells = jvm_cell_size_for_type(proc -> type -> base); int num = desig -> item.num_args; oberon_expr_t * arg = desig -> item.args; for(int i = 0; i < num; i++) { - push_expr(ctx, m -> fp, arg); + args_cells += jvm_cell_size_for_type(arg -> result); + push_expr(p, arg); arg = arg -> next; } - fprintf(m -> fp, "invokestatic %s%s\n", fullname, signature); + jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", fullname, signature); } void -oberon_generate_end_proc(oberon_context_t * ctx) +oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig) { - gen_module_t * m; - m = ctx -> mod -> gen_mod; - - fprintf(m -> fp, " .limit stack 32\n"); - fprintf(m -> fp, " .limit locals 32\n"); - fprintf(m -> fp, ".end method\n\n"); + jvm_generate_call_proc(ctx -> gen_context -> current_m -> class -> p, desig); } void oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr) { gen_module_t * m; + gen_proc_t * p; + char prefix; + int cell_size; m = ctx -> mod -> gen_mod; + p = m -> class -> p; if(expr) { - oberon_error(ctx, "oberon_generate_return: TODO return expr"); + push_expr(p, expr); + prefix = jvm_get_prefix(expr -> result); + cell_size = jvm_cell_size_for_type(expr -> result); + jvm_generate(p, cell_size, 0, "%creturn", prefix); } else { - fprintf(m -> fp, " return\n"); + jvm_generate(p, 0, 0, "return"); } } static void -push_int(FILE * fp, long i) +jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg) { - if(i == -1) - { - fprintf(fp, "iconst_m1\n"); - } - else if(i >= 0 && i <= 5) - { - fprintf(fp, "iconst_%li\n", i); - } - else if(i >= -128 && i <= 127) - { - fprintf(fp, "bipush %li\n", i); - } - else if(i >= -32768 && i <= 32767) - { - fprintf(fp, "sipush %li\n", i); - } - else if(i >= -2147483648 && i <= 2147483647) - { - fprintf(fp, "ldc %li\n", i); - } - else - { - fprintf(fp, "ldc2 %li\n", i); - } -} + assert(type -> class == OBERON_TYPE_POINTER); -static void -push_float(FILE * fp, double f, int size) -{ - if(size <= 4) - { - if(f == 0.0) - { - fprintf(fp, "fconst_0\n"); - } - if(f == 1.0) - { - fprintf(fp, "fconst_1\n"); - } - if(f == 2.0) - { - fprintf(fp, "fconst_2\n"); - } - else - { - fprintf(fp, "ldc %lf\n", f); - } - } - else + for(int i = 0; i < num; i++) { - if(f == 0.0) - { - fprintf(fp, "dconst_0\n"); - } - if(f == 1.0) - { - fprintf(fp, "dconst_1\n"); - } - else - { - fprintf(fp, "ldc2 %lf\n", f); - } + push_expr(p, arg); + arg = arg -> next; } -} -static void push_expr(oberon_context_t * ctx, FILE * fp, oberon_expr_t * expr); - -static void -push_var(oberon_context_t * ctx, FILE * fp, oberon_object_t * var) -{ - if(var -> local) - { - int reg = var -> gen_var -> reg; - char prefix = get_prefix(ctx, var -> type); - fprintf(fp, "%cload %i\n", prefix, reg); - } - else - { - char * fullname = get_field_full_name(ctx, var); - char * desc = get_descriptor(ctx, var -> type); - fprintf(fp, "getstatic %s %s\n", fullname, desc); - } + jvm_generate_new(p, type -> base, num); } static void -push_item(oberon_context_t * ctx, FILE * fp, oberon_item_t * item) +push_item(gen_proc_t * p, oberon_item_t * item) { switch(item -> mode) { case MODE_VAR: - push_var(ctx, fp, item -> var); + jvm_generate_load(p, item -> result, item -> var -> gen_var); break; case MODE_INTEGER: - push_int(fp, item -> integer); + jvm_generate_push_int(p, item -> integer); break; case MODE_BOOLEAN: - push_int(fp, item -> boolean); + jvm_generate_push_int(p, item -> boolean); break; case MODE_CALL: - oberon_error(ctx, "push_item: TODO call"); + jvm_generate_call_proc(p, (oberon_expr_t *) item); break; case MODE_INDEX: - oberon_error(ctx, "push_item: TODO index"); + ; + char postfix = jvm_get_postfix(item -> result); + int cell_size = jvm_cell_size_for_postfix(postfix); + push_item(p, item -> parent); + push_expr(p, item -> args); + jvm_generate(p, 1 + 1, cell_size, "%caload", postfix); break; case MODE_FIELD: - oberon_error(ctx, "push_item: TODO field"); + push_item(p, item -> parent); + jvm_generate_load(p, item -> result, item -> var -> gen_var); break; case MODE_DEREF: - oberon_error(ctx, "push_item: TODO deref"); + /* Все объекты в jvm представляются как указатели */ + push_item(p, item -> parent); break; case MODE_NIL: - fprintf(fp, "aconst_null\n"); + jvm_generate(p, 0, 1, "aconst_null"); break; case MODE_NEW: - oberon_error(ctx, "push_item: TODO new"); + jvm_generate_expr_new_pointer(p, item -> result, item -> num_args, item -> args); break; case MODE_REAL: - push_float(fp, item -> real, item -> result -> size); + jvm_generate_push_float(p, item -> real, item -> result -> size); + break; + default: + gen_error("push_item: unk mode %i", item -> mode); + break; + } +} + +static void +jvm_generate_logical_not(gen_proc_t * p) +{ + int label_done = jvm_new_label_id(p); + int label_false = jvm_new_label_id(p); + char * label_name_done = jvm_get_label_name(label_done); + char * label_name_false = jvm_get_label_name(label_false); + + jvm_generate(p, 1, 0, "ifne %s", label_name_false); + jvm_generate(p, 0, 1, "iconst_1"); + jvm_generate(p, 0, 0, "goto %s", label_name_done); + jvm_generate_label(p, label_false); + jvm_generate(p, 0, 1, "iconst_0"); + jvm_generate_label(p, label_done); +} + +static void +jvm_generate_abs(gen_proc_t * p, char prefix) +{ + char t = jvm_get_type_of_prefix(prefix); + int cell_size = jvm_cell_size_for_postfix(prefix); + jvm_generate(p, cell_size, cell_size, "invokestatic java/lang/Math/abs(%c)%c", t, t); +} + +static void +jvm_generate_compare_op(gen_proc_t * p, char prefix, int op) +{ + int label_true = jvm_new_label_id(p); + int label_done = jvm_new_label_id(p); + char * label_name_true = jvm_get_label_name(label_true); + char * label_name_done = jvm_get_label_name(label_done); + int cell_size = 2 * jvm_cell_size_for_postfix(prefix); + + assert(prefix == 'i' || prefix == 'a'); + + const char * cmpop = ""; + switch(op) + { + case OP_EQ: + cmpop = "eq"; + break; + case OP_NEQ: + cmpop = "ne"; + break; + case OP_LSS: + cmpop = "lt"; + break; + case OP_LEQ: + cmpop = "le"; + break; + case OP_GRT: + cmpop = "gt"; + break; + case OP_GEQ: + cmpop = "ge"; break; default: - oberon_error(ctx, "push_item: unk mode %i", item -> mode); + gen_error("jvm_generate_compare_op: wat"); break; } + + jvm_generate(p, cell_size, 0, "if_%ccmp%s %s", prefix, cmpop, label_name_true); + jvm_generate(p, 0, 1, "iconst_0"); + jvm_generate(p, 0, 0, "goto %s", label_name_done); + jvm_generate_label(p, label_true); + jvm_generate(p, 0, 1, "iconst_1"); + jvm_generate_label(p, label_done); } static void -push_operator(oberon_context_t * ctx, FILE * fp, oberon_oper_t * oper) +jvm_generate_operator(gen_proc_t * p, char prefix, int op) { - char prefix = get_prefix(ctx, oper -> result); - switch(oper -> op) + int cell_size = jvm_cell_size_for_postfix(prefix); + switch(op) { case OP_UNARY_MINUS: - push_expr(ctx, fp, oper -> left); - fprintf(fp, "%cneg\n", prefix); + jvm_generate(p, cell_size, cell_size, "%cneg", prefix); break; case OP_BITWISE_NOT: - push_expr(ctx, fp, oper -> left); - push_int(fp, -1); - fprintf(fp, "%cxor\n", prefix); + jvm_generate_push_int(p, -1); + jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix); + break; + case OP_LOGIC_NOT: + jvm_generate_logical_not(p); + break; + case OP_ABS: + jvm_generate_abs(p, prefix); break; case OP_ADD: - push_expr(ctx, fp, oper -> left); - push_expr(ctx, fp, oper -> right); - fprintf(fp, "%cadd\n", prefix); + jvm_generate(p, 2 * cell_size, cell_size, "%cadd", prefix); break; case OP_SUB: - push_expr(ctx, fp, oper -> left); - push_expr(ctx, fp, oper -> right); - fprintf(fp, "%csub\n", prefix); + jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix); break; case OP_MUL: - push_expr(ctx, fp, oper -> left); - push_expr(ctx, fp, oper -> right); - fprintf(fp, "%cmul\n", prefix); + jvm_generate(p, 2 * cell_size, cell_size, "%cmul", prefix); break; case OP_DIV: - push_expr(ctx, fp, oper -> left); - push_expr(ctx, fp, oper -> right); - fprintf(fp, "%cdiv\n", prefix); + jvm_generate(p, 2 * cell_size, cell_size, "%cdiv", prefix); break; case OP_MOD: - push_expr(ctx, fp, oper -> left); - push_expr(ctx, fp, oper -> right); - fprintf(fp, "%crem\n", prefix); + jvm_generate(p, 2 * cell_size, cell_size, "%crem", prefix); break; case OP_BITWISE_AND: - push_expr(ctx, fp, oper -> left); - push_expr(ctx, fp, oper -> right); - fprintf(fp, "%cand\n", prefix); + jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix); break; case OP_BITWISE_XOR: - push_expr(ctx, fp, oper -> left); - push_expr(ctx, fp, oper -> right); - fprintf(fp, "%cxor\n", prefix); + jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix); break; case OP_BITWISE_OR: - push_expr(ctx, fp, oper -> left); - push_expr(ctx, fp, oper -> right); - fprintf(fp, "%cor\n", prefix); + jvm_generate(p, 2 * cell_size, cell_size, "%cor", prefix); break; case OP_EQ: @@ -682,54 +1214,154 @@ push_operator(oberon_context_t * ctx, FILE * fp, oberon_oper_t * oper) case OP_LEQ: case OP_GRT: case OP_GEQ: + jvm_generate_compare_op(p, prefix, op); + break; + default: + gen_error("jvm_generate_operator: unk op %i", op); + break; + } +} + +static void +jvm_generate_logical_or(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b) +{ + int label_calc_b = jvm_new_label_id(p); + int label_done = jvm_new_label_id(p); + char * label_name_calc_b = jvm_get_label_name(label_calc_b); + char * label_name_done = jvm_get_label_name(label_done); + + /* a OR b -- если a, то TRUE, иначе b */ + + push_expr(p, a); + jvm_generate(p, 1, 0, "ifne %s", label_name_calc_b); + jvm_generate(p, 0, 1, "iconst_1"); + jvm_generate(p, 0, 0, "goto %s", label_name_done); + jvm_generate_label(p, label_calc_b); + push_expr(p, b); + jvm_generate_label(p, label_done); +} + +static void +jvm_generate_logical_and(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b) +{ + int label_false = jvm_new_label_id(p); + int label_done = jvm_new_label_id(p); + char * label_name_false = jvm_get_label_name(label_false); + char * label_name_done = jvm_get_label_name(label_done); + + /* a AND b -- если a, то b, иначе FALSE */ + + push_expr(p, a); + jvm_generate(p, 1, 0, "ifne %s", label_name_false); + push_expr(p, b); + jvm_generate(p, 0, 0, "goto %s", label_name_done); + jvm_generate_label(p, label_false); + jvm_generate(p, 0, 1, "iconst_0"); + jvm_generate_label(p, label_done); +} + +static void +push_operator(gen_proc_t * p, oberon_oper_t * oper) +{ + char prefix = jvm_get_prefix(oper -> result); + int op = oper -> op; + switch(op) + { + case OP_UNARY_MINUS: + case OP_BITWISE_NOT: case OP_LOGIC_NOT: case OP_ABS: - oberon_error(ctx, "push_oper: TODO op %i", oper -> op); + push_expr(p, oper -> left); + jvm_generate_operator(p, prefix, op); + break; + + case OP_ADD: + case OP_SUB: + case OP_MUL: + case OP_DIV: + case OP_MOD: + case OP_BITWISE_AND: + case OP_BITWISE_XOR: + case OP_BITWISE_OR: + + case OP_EQ: + case OP_NEQ: + case OP_LSS: + case OP_LEQ: + case OP_GRT: + case OP_GEQ: + push_expr(p, oper -> left); + push_expr(p, oper -> right); + jvm_generate_operator(p, prefix, op); + break; + + case OP_LOGIC_OR: + jvm_generate_logical_or(p, oper -> left, oper -> right); + break; + case OP_LOGIC_AND: + jvm_generate_logical_and(p, oper -> left, oper -> right); break; default: - oberon_error(ctx, "push_oper: unk op %i", oper -> op); + gen_error("push_oper: unk op %i", op); break; } } static void -push_expr(oberon_context_t * ctx, FILE * fp, oberon_expr_t * expr) +push_expr(gen_proc_t * p, oberon_expr_t * expr) { if(expr -> is_item) { - push_item(ctx, fp, (oberon_item_t *) expr); + push_item(p, (oberon_item_t *) expr); } else { - push_operator(ctx, fp, (oberon_oper_t *) expr); + push_operator(p, (oberon_oper_t *) expr); } } static void -store_expr(oberon_context_t * ctx, FILE * fp, oberon_expr_t * expr) +store_expr(gen_proc_t * p, oberon_expr_t * dst, oberon_expr_t * src) { - assert(expr -> is_item); - oberon_item_t * item = (oberon_item_t *) expr; + assert(dst -> is_item); + oberon_item_t * item = (oberon_item_t *) dst; - char prefix; - switch(item -> mode) + if(dst -> result -> class == OBERON_TYPE_ARRAY + || src -> result -> class == OBERON_TYPE_ARRAY) + { + push_expr(p, dst); + push_expr(p, src); + jvm_generate_copy_array(p, dst -> result); + } + else if(dst -> result -> class == OBERON_TYPE_RECORD + || src -> result -> class == OBERON_TYPE_RECORD) + { + push_expr(p, dst); + push_expr(p, src); + jvm_generate_copy_record(p, dst -> result); + } + else switch(item -> mode) { case MODE_VAR: - if(item -> var -> local) - { - int reg = item -> var -> gen_var -> reg; - prefix = get_prefix(ctx, item -> result); - fprintf(fp, "%cstore %i\n", prefix, reg); - } - else - { - char * fullname = get_field_full_name(ctx, item -> var); - char * desc = get_descriptor(ctx, item -> result); - fprintf(fp, "putstatic %s %s\n", fullname, desc); - } + push_expr(p, src); + jvm_generate_store(p, src -> result, item -> var -> gen_var); + break; + case MODE_INDEX: + ; + char postfix = jvm_get_postfix(src -> result); + int cell_size = jvm_cell_size_for_postfix(postfix); + push_item(p, item -> parent); + push_expr(p, item -> args); + push_expr(p, src); + jvm_generate(p, 1 + 1 + cell_size, 0, "%castore", postfix); + break; + case MODE_FIELD: + push_item(p, item -> parent); + push_expr(p, src); + jvm_generate_store(p, src -> result, item -> var -> gen_var); break; default: - oberon_error(ctx, "store_expr: unk mode %i", item -> mode); + gen_error("store_expr: unk mode %i", item -> mode); break; } } @@ -738,10 +1370,12 @@ void oberon_generate_assign(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_t * dst) { gen_module_t * m; + gen_proc_t * p; + m = ctx -> mod -> gen_mod; + p = m -> class -> p; - push_expr(ctx, m -> fp, src); - store_expr(ctx, m -> fp, dst); + store_expr(p, dst, src); } void