DEADSOFTWARE

Исправлены выражения с операторами
[dsw-obn.git] / src / backends / jvm / generator-jvm.c
index 895a5ae3e524c2ef30063d1de38a316b2e3d23b1..581faf92c302bab84ce601e5087eeb39e480d4ef 100644 (file)
@@ -128,6 +128,19 @@ jvm_generate(gen_proc_t * p, unsigned get, unsigned push, char * format, ...)
        va_end(ptr);
 }
 
+static void
+jvm_generate_comment(gen_proc_t * p, char * format, ...)
+{
+       va_list ptr;
+       va_start(ptr, format);
+
+       fprintf(p -> class -> fp, "  ;;;; ");
+       vfprintf(p -> class -> fp, format, ptr);
+       fprintf(p -> class -> fp, "\n");
+
+       va_end(ptr);
+}
+
 static void
 jvm_generate_push_int(gen_proc_t * p, int64_t i)
 {
@@ -157,6 +170,60 @@ jvm_generate_push_int(gen_proc_t * p, int64_t i)
        }
 }
 
+static void
+jvm_generate_push_int_size(gen_proc_t * p, int64_t i, int size)
+{
+       int pushed_cell = 1;
+
+       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
+       {
+               pushed_cell = 2;
+               jvm_generate(p, 0, 2, "ldc2 %li", i);
+       }
+
+       assert(size <= 8);
+       if(size > 4 && pushed_cell == 1)
+       {
+               jvm_generate(p, pushed_cell, 2, "i2l");
+       }
+       else if(size <= 4)
+       {
+               if(pushed_cell > 1)
+               {
+                       jvm_generate(p, 2, 1, "l2i");
+               }
+
+               if(size == 2)
+               {
+                       jvm_generate(p, 1, 1, "i2s");
+               }
+               else if(size == 1)
+               {
+                       jvm_generate(p, 1, 1, "i2b");
+               }
+       }
+}
+
 static void
 jvm_generate_push_float(gen_proc_t * p, double f, int size)
 {
@@ -349,7 +416,8 @@ jvm_generate_array_initialization(gen_proc_t * p, oberon_type_t * arr)
        jvm_generate_and_init_local_var(dst, p, false);
        jvm_generate_store(p, arr, dst);
 
-       /* Входящие параметры заграблены.
+       /*
+        * Входящие параметры заграблены.
         * Теперь генерируем эквивалентный код:
         *   int i = 0;
         *   int len = dst.length
@@ -510,7 +578,8 @@ jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
        jvm_generate_store(p, arr, src);
        jvm_generate_store(p, arr, dst);
 
-       /* Входящие параметры заграблены.
+       /*
+        * Входящие параметры заграблены.
         * Теперь генерируем эквивалентный код:
         *   int i = 0;
         *   int len = dst.length
@@ -551,12 +620,7 @@ jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
                jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
        }
 
-       if(base -> class == OBERON_TYPE_ARRAY)
-       {
-               // Вся эта шляпа уже должна знать о всех размерностях
-               gen_error("jvm_generate_copy_array: wat");
-       }
-       else if(base -> class == OBERON_TYPE_RECORD)
+       if(base -> class == OBERON_TYPE_RECORD)
        {
                /* Получаем записи по индексам ( -- dst src ) */
 
@@ -613,6 +677,14 @@ jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
        }
 }
 
+static void
+jvm_generate_push_procedure_pointer(gen_proc_t * p, oberon_object_t * proc)
+{
+       char * full_name = jvm_get_field_full_name_safe(proc);
+       char * desc = jvm_get_descriptor(proc -> type);
+       jvm_generate(p, 0, 1, "getstatic %s/pointer %s", full_name, desc);
+}
+
 // ==========================================
 // ==========================================
 // ==========================================
@@ -647,7 +719,7 @@ oberon_generate_procedure_class(oberon_type_t * proc)
        fprintf(fp, ".class  public abstract %s\n", cname);
        fprintf(fp, ".super  java/lang/Object\n\n");
 
-       fprintf(fp, ".method public <init>()V\n");
+       fprintf(fp, ".method <init>()V\n");
        fprintf(fp, "  aload_0\n");
        fprintf(fp, "  invokespecial java/lang/Object/<init>()V\n");
        fprintf(fp, "  return\n");
@@ -661,6 +733,85 @@ oberon_generate_procedure_class(oberon_type_t * proc)
        jvm_destroy_class(class);
 }
 
+static void
+oberon_generate_procedure_pointer_class(oberon_object_t * proc)
+{
+       FILE * fp;
+       char * cname;
+       char * abscname;
+       char * absdesc;
+       char * signature;
+       struct gen_class * class;
+
+       cname = jvm_get_field_full_name_safe(proc);
+       class = jvm_create_class(cname);
+       abscname = jvm_get_class_full_name(proc -> type);
+       absdesc = jvm_get_descriptor(proc -> type);
+       fp = class -> fp;
+
+       fprintf(fp, ".source %s\n", proc -> module -> name);
+       fprintf(fp, ".class  public %s\n", cname);
+       fprintf(fp, ".super  %s\n\n", abscname);
+
+       fprintf(fp, ".field public static pointer %s\n\n", absdesc);
+
+       fprintf(fp, ".method <init>()V\n");
+       fprintf(fp, "  aload_0\n");
+       fprintf(fp, "  invokespecial %s/<init>()V\n", abscname);
+       fprintf(fp, "  aload_0\n");
+       fprintf(fp, "  putstatic %s/pointer %s\n", cname, absdesc);
+       fprintf(fp, "  return\n");
+       fprintf(fp, ".end method\n\n");
+
+       signature = jvm_get_procedure_signature(proc -> type);
+
+       gen_proc_t * p;
+       gen_var_t * this_v;
+
+       p = jvm_create_proc(class);
+       jvm_generate_function_header(p, "public", "invoke", signature);
+
+       this_v = oberon_generator_new_var();
+       jvm_generate_and_init_local_var(this_v, p, false);
+
+       gen_var_t * v;
+       int use_size = 0;
+       int num = proc -> type -> num_decl;
+       oberon_object_t * arg = proc -> type -> decl;
+       for(int i = 0; i < num; i++)
+       {
+               v = oberon_generator_new_var();
+               bool wide = jvm_is_wide_type(arg -> type);
+               char * desc = jvm_get_descriptor(arg -> type);
+               jvm_generate_and_init_named_local_var(v, p, wide, arg -> name, desc);
+
+               jvm_generate_load(p, arg -> type, v);
+
+               use_size += (wide) ? (2) : (1);
+
+               arg = arg -> next;
+       }
+
+       char * full_name = jvm_get_field_full_name(proc);
+       int cell_size = jvm_cell_size_for_type(proc -> type -> base);
+
+       jvm_generate(p, use_size, cell_size, "invokestatic %s%s", full_name, signature);
+
+       if(proc -> type -> base -> class == OBERON_TYPE_VOID)
+       {
+               jvm_generate(p, 0, 0, "return");
+       }
+       else
+       {
+               char prefix = jvm_get_prefix(proc -> type -> base);
+               jvm_generate(p, cell_size, 0, "%creturn", prefix);
+       }
+
+       jvm_generate_function_end(p);
+
+       jvm_destroy_class(class);
+}
+
 static void
 oberon_generate_record_class(gen_module_t * m, oberon_type_t * rec)
 {
@@ -825,7 +976,7 @@ oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var)
                        /* смотри oberon_generator_init_record() */
                        break;
                case OBERON_CLASS_VAR:
-                       /* Локальные заполняются при генерации функции */                        
+                       /* Локальные заполняются при генерации функции */
                        /* смотри jvm_init_local_object() */
                        if(var -> local == 0)
                        {
@@ -849,6 +1000,8 @@ oberon_generator_init_proc(oberon_context_t * ctx, oberon_object_t * proc)
        {
                gen_error("generator: local procedures not implemented");
        }
+
+       oberon_generate_procedure_pointer_class(proc);
 }
 
 void
@@ -903,6 +1056,13 @@ oberon_generate_begin_module(oberon_context_t * ctx)
                                jvm_generate_store(p, x -> type, x -> gen_var);
                        }
                }
+               else if(x -> class == OBERON_CLASS_PROC)
+               {
+                       char * cname = jvm_get_field_full_name_safe(x);
+                       jvm_generate(p, 0, 1, "new %s", cname);
+                       jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", cname);
+               }
+
                x = x -> next;
        }
 }
@@ -946,6 +1106,57 @@ jvm_init_local_object(gen_proc_t * p, oberon_object_t * x)
        }
 }
 
+/*
+ * Генерирует код для получения размера измерения массива
+ * Аналог Обероновского LEN(v, n); где n = 0 - первое измерение.
+ * ( aref -- int )
+ */
+
+static void
+jvm_generate_array_len(gen_proc_t * p, int dim)
+{
+       while(dim > 0)
+       {
+               jvm_generate(p, 0, 1, "iconst_0");
+               jvm_generate(p, 2, 1, "aaload");
+               dim -= 1;
+       }
+       jvm_generate(p, 1, 1, "arraylength");
+}
+
+static void
+jvm_generate_array_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * arr)
+{
+       int dim = 0;
+       oberon_type_t * base = arr;
+       while(base -> class == OBERON_TYPE_ARRAY)
+       {
+               if(base -> size == 0)
+               {
+                       jvm_generate_load(p, arr, v);
+                       jvm_generate_array_len(p, dim);
+                       dim += 1;
+               }
+               base = base -> base;
+       }
+
+       jvm_generate_new(p, arr, dim);
+       jvm_generate(p, 1, 2, "dup");
+       jvm_generate_load(p, arr, v);
+       jvm_generate_copy_array(p, arr);
+       jvm_generate_store(p, arr, v);
+}
+
+static void
+jvm_generate_record_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * rec)
+{
+       jvm_generate_new(p, rec, 0);
+       jvm_generate(p, 1, 2, "dup");
+       jvm_generate_load(p, rec, v);
+       jvm_generate_copy_record(p, rec);
+       jvm_generate_store(p, rec, v);
+}
+
 void
 oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc)
 {
@@ -955,15 +1166,33 @@ oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc)
        p = proc -> gen_proc;
 
        signature = jvm_get_procedure_signature(proc -> type);
-
        jvm_generate_function_header(p, "public static", proc -> name, signature);
 
+       /* Выделение регистров под параметры и переменные */
        oberon_object_t * var = proc -> type -> decl;
        while(var)
        {
                jvm_init_local_object(p, var);
                var = var -> next;
        }
+
+       /* Копирование статических/открытых массивов и записей */
+       var = proc -> type -> decl;
+       while(var)
+       {
+               if(var -> class == OBERON_CLASS_PARAM || var -> class == OBERON_CLASS_VAR_PARAM)
+               {
+                       if(var -> type -> class == OBERON_TYPE_ARRAY)
+                       {
+                               jvm_generate_array_duplicate_and_replace(p, var -> gen_var, var -> type);
+                       }
+                       else if(var -> type -> class == OBERON_TYPE_RECORD)
+                       {                               
+                               jvm_generate_record_duplicate_and_replace(p, var -> gen_var, var -> type);
+                       }
+               }
+               var = var -> next;
+       }       
 }
 
 void
@@ -992,6 +1221,12 @@ jvm_generate_call_proc(gen_proc_t * p, oberon_expr_t * desig)
        fullname = jvm_get_field_full_name(proc);
        signature = jvm_get_procedure_signature(proc -> type);
 
+       if(proc -> class != OBERON_CLASS_PROC)
+       {
+               /* Загружаем указатель на процедуру */
+               jvm_generate_load(p, proc -> type, proc -> gen_var);
+       }
+
        int args_cells = 0;
        int result_cells = jvm_cell_size_for_type(proc -> type -> base);
 
@@ -1004,13 +1239,23 @@ jvm_generate_call_proc(gen_proc_t * p, oberon_expr_t * desig)
                arg = arg -> next;
        }
 
-       jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", fullname, signature);
+       if(proc -> class == OBERON_CLASS_PROC)
+       {
+               /* Обычная статическая процедура */
+               jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", fullname, signature);            
+       }
+       else
+       {
+               /* Процедура-переменная */
+               char * class = jvm_get_class_full_name(proc -> type);
+               jvm_generate(p, 1 + args_cells, result_cells, "invokevirtual %s/invoke%s", class, signature);
+       }
 }
 
 void
 oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig)
 {
-       jvm_generate_call_proc(ctx -> gen_context -> current_m -> class -> p, desig);
+       jvm_generate_call_proc(ctx -> mod -> gen_mod -> class -> p, desig);
 }
 
 void
@@ -1051,19 +1296,44 @@ jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, obe
        jvm_generate_new(p, type -> base, num);
 }
 
+static void
+jvm_generate_cast_type(gen_proc_t * p, oberon_type_t * from, oberon_type_t * to)
+{
+       char prefix = jvm_get_prefix(from);
+       char postfix = jvm_get_postfix(to);
+       int from_cell_size = jvm_cell_size_for_type(from);
+       int to_cell_size = jvm_cell_size_for_type(to);
+
+       jvm_generate_comment(p, "cast type class from %i(%i) to %i(%i)", from -> class, from -> size, to -> class, to -> size);
+
+       if(prefix == postfix)
+       {
+               return;
+       }
+
+       jvm_generate(p, from_cell_size, to_cell_size, "%c2%c", prefix, postfix);
+}
+
 static void
 push_item(gen_proc_t * p, oberon_item_t * item)
 {
        switch(item -> mode)
        {
                case MODE_VAR:
-                       jvm_generate_load(p, item -> result, item -> var -> gen_var);
+                       if(item -> var -> class == OBERON_CLASS_PROC)
+                       {
+                               jvm_generate_push_procedure_pointer(p, item -> var);
+                       }
+                       else
+                       {
+                               jvm_generate_load(p, item -> result, item -> var -> gen_var);
+                       }
                        break;
                case MODE_INTEGER:
-                       jvm_generate_push_int(p, item -> integer);
+                       jvm_generate_push_int_size(p, item -> integer, item -> result -> size);
                        break;
                case MODE_BOOLEAN:
-                       jvm_generate_push_int(p, item -> boolean);
+                       jvm_generate_push_int_size(p, item -> boolean, item -> result -> size);
                        break;
                case MODE_CALL:
                        jvm_generate_call_proc(p, (oberon_expr_t *) item);
@@ -1072,17 +1342,20 @@ push_item(gen_proc_t * p, oberon_item_t * item)
                        ;
                        char postfix = jvm_get_postfix(item -> result);
                        int cell_size = jvm_cell_size_for_postfix(postfix);
-                       push_item(p, item -> parent);
+                       assert(item -> parent -> is_item);
+                       push_item(p, (oberon_item_t *) item -> parent);
                        push_expr(p, item -> args);
                        jvm_generate(p, 1 + 1, cell_size, "%caload", postfix);
                        break;
                case MODE_FIELD:
-                       push_item(p, item -> parent);
+                       assert(item -> parent -> is_item);
+                       push_item(p, (oberon_item_t *) item -> parent);
                        jvm_generate_load(p, item -> result, item -> var -> gen_var);
                        break;
                case MODE_DEREF:
                        /* Все объекты в jvm представляются как указатели */
-                       push_item(p, item -> parent);
+                       assert(item -> parent -> is_item);
+                       push_item(p, (oberon_item_t *) item -> parent);
                        break;
                case MODE_NIL:
                        jvm_generate(p, 0, 1, "aconst_null");
@@ -1093,6 +1366,10 @@ push_item(gen_proc_t * p, oberon_item_t * item)
                case MODE_REAL:
                        jvm_generate_push_float(p, item -> real, item -> result -> size);
                        break;
+               case MODE_CAST:
+                       push_expr(p, item -> parent);
+                       jvm_generate_cast_type(p, item -> parent -> result, item -> result);
+                       break;
                default:
                        gen_error("push_item: unk mode %i", item -> mode);
                        break;
@@ -1355,13 +1632,15 @@ store_expr(gen_proc_t * p, oberon_expr_t * dst, oberon_expr_t * src)
                        ;
                        char postfix = jvm_get_postfix(src -> result);
                        int cell_size = jvm_cell_size_for_postfix(postfix);
-                       push_item(p, item -> parent);
+                       assert(item -> parent -> is_item);
+                       push_item(p, (oberon_item_t *) 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);
+                       assert(item -> parent -> is_item);
+                       push_item(p, (oberon_item_t *) item -> parent);
                        push_expr(p, src);
                        jvm_generate_store(p, src -> result, item -> var -> gen_var);
                        break;