DEADSOFTWARE

JVM: Добавлено копирование массивов
[dsw-obn.git] / src / backends / jvm / generator-jvm.c
index 9408eca32328248fd37b8454c30b99d663ad7ebd..4b5b9474c0a2f006c99e15c5e4251e37763785b1 100644 (file)
@@ -48,6 +48,7 @@ jvm_generate_function_header(gen_proc_t * p, char * access, char * name, char *
        class -> p = p;
 
        fprintf(class -> fp, ".method %s %s%s\n", access, name, signature);
+       fprintf(class -> fp, "  start:\n");
 }
 
 void
@@ -66,6 +67,7 @@ jvm_generate_function_end(gen_proc_t * p)
 
        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");
 }
 
@@ -104,7 +106,11 @@ static void
 jvm_stack_pop(gen_proc_t * p, unsigned size)
 {
        p -> stack -> pointer -= size;
-       assert(p -> stack -> pointer >= 0);
+
+       if(p -> stack -> pointer < 0)
+       {
+               printf("WARING: stack pointer %i\n", p -> stack -> pointer);
+       }
 }
 
 static void
@@ -256,12 +262,12 @@ jvm_generate_label(gen_proc_t * p, int label_id)
 }
 
 static void
-jvm_generate_new_static(gen_proc_t * p, oberon_type_t * type);
+jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num);
 
 /*
  * Функция jvm_generate_static_array_initialization генерирует код для
  * статической инициализации массива. На входе массив, на выходе тот же массив.
- * arrayref -- arrayref
+ * (arrayref -- arrayref)
  */
 
 static void
@@ -322,7 +328,7 @@ jvm_generate_static_array_initialization(gen_proc_t * p, oberon_type_t * type)
        {
                jvm_generate(p, 0, 1, "aload %i", reg_a);
                jvm_generate(p, 0, 1, "iload %i", reg_i);
-               jvm_generate_new_static(p, type -> base);
+               jvm_generate_new(p, type -> base, 0);
                jvm_generate(p, 3, 0, "aastore");
        }
        else
@@ -344,7 +350,7 @@ jvm_generate_static_array_initialization(gen_proc_t * p, oberon_type_t * type)
 }
 
 static void
-jvm_generate_new_static(gen_proc_t * p, oberon_type_t * type)
+jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num)
 {
        int dim;
        char * cname;
@@ -372,10 +378,21 @@ jvm_generate_new_static(gen_proc_t * p, oberon_type_t * type)
                        desc = jvm_get_descriptor(type);
                        while(base -> class == OBERON_TYPE_ARRAY)
                        {
+                               if(num > 0)
+                               {
+                                       assert(base -> size == 0);
+                                       num -= 1;
+                               }
+                               else
+                               {
+                                       assert(base -> size > 0);
+                                       jvm_generate_push_int(p, base -> size);
+                               }
                                dim += 1;
-                               jvm_generate_push_int(p, base -> size);
                                base = base -> base;
                        }
+
+                       assert(num == 0);
                        jvm_generate(p, dim, 1, "multianewarray %s %i", desc, dim);
 
                        if(base -> class == OBERON_TYPE_RECORD)
@@ -389,6 +406,32 @@ jvm_generate_new_static(gen_proc_t * p, oberon_type_t * type)
        }
 }
 
+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 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:
+                       gen_error("jvm_generate_load: unknow storage type %i", src -> storage);
+                       break;
+       }
+}
+
 static void
 jvm_generate_store(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * dst)
 {
@@ -415,6 +458,156 @@ jvm_generate_store(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * dst)
        }
 }
 
+/*
+ * Генерирует код для копирования полей из первой записи во вторую.
+ *  ( aref_dst aref_src -- )
+ *  dst := src;
+ */
+
+static void
+jvm_generate_copy_record(gen_proc_t * p, oberon_type_t * rec)
+{
+       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);
+}
+
+/*
+ * Генерирует кода для копирования массивов.
+ * ( aref_dst aref_src -- )
+ * dst := src;
+ */
+
+static void
+jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
+{
+       int dim = 0;
+       oberon_type_t * base = arr;
+       while(base -> class == OBERON_TYPE_ARRAY)
+       {
+               dim += 1;
+               base = base -> base;
+       }
+
+       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++)
+       {
+               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);
+       }
+
+       if(base -> class == OBERON_TYPE_ARRAY)
+       {
+               // Вся эта шляпа уже должна знать о всех размерностях
+               gen_error("jvm_generate_copy_array: wat");
+       }
+       else if(base -> class == OBERON_TYPE_RECORD)
+       {
+               /* Получаем записи по индексам ( -- dst src ) */
+
+               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");
+               }
+
+               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);
+
+               /* Получаем массивы и индексы ( -- dst i src i ) */
+
+               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);
+               }
+
+               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);
+               }
+
+               /* Копируем значения ( 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--)
+       {
+               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);
+       }
+}
+
 // ==========================================
 // ==========================================
 // ==========================================
@@ -532,13 +725,13 @@ oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
                field = field -> next;
        }
 
+       /* Стандартный конструктор класса */
+       /* Инициализирует внутренние статические записи и массивы */
        gen_proc_t * p = jvm_create_proc(class);
-
        jvm_generate_function_header(p, "public", "<init>", "()V");
        jvm_alloc_register_untyped(p -> rf, false);
        jvm_generate(p, 0, 1, "aload_0");
        jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
-
        num = rec -> num_decl;
        field = rec -> decl;
        for(int i = 0; i < num; i++)
@@ -547,12 +740,53 @@ oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
                        || field -> type -> class == OBERON_TYPE_ARRAY)
                {
                        jvm_generate(p, 0, 1, "aload_0");
-                       jvm_generate_new_static(p, field -> type);
+                       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);
 
@@ -660,7 +894,7 @@ oberon_generate_begin_module(oberon_context_t * ctx)
                        if(x -> type -> class == OBERON_TYPE_ARRAY
                                || x -> type -> class == OBERON_TYPE_RECORD)
                        {
-                               jvm_generate_new_static(p, x -> type);
+                               jvm_generate_new(p, x -> type, 0);
                                jvm_generate_store(p, x -> type, x -> gen_var);
                        }
                }
@@ -799,36 +1033,17 @@ oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr)
 }
 
 static void
-jvm_push_var(gen_proc_t * p, oberon_object_t * var)
+jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
 {
-       int cell_size = jvm_cell_size_for_type(var -> type);
+       assert(type -> class == OBERON_TYPE_POINTER);
 
-       if(var -> local)
-       {
-               int reg = var -> gen_var -> reg;
-               char prefix = jvm_get_prefix(var -> type);
-               jvm_generate(p, 0, cell_size, "%cload %i", prefix, reg);
-       }
-       else
+       for(int i = 0; i < num; i++)
        {
-               char * fullname = jvm_get_field_full_name(var);
-               char * desc = jvm_get_descriptor(var -> type);
-               jvm_generate(p, 0, cell_size, "getstatic %s %s", fullname, desc);
+               push_expr(p, arg);
+               arg = arg -> next;
        }
-}
 
-static void
-jvm_generate_expr_new_static(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
-{
-       assert(num == 0);
-       jvm_generate_new_static(p, type);
-}
-
-static void
-jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
-{
-       assert(type -> class == OBERON_TYPE_POINTER);
-       jvm_generate_expr_new_static(p, type -> base, num, arg);
+       jvm_generate_new(p, type -> base, num);
 }
 
 static void
@@ -837,7 +1052,7 @@ push_item(gen_proc_t * p, oberon_item_t * item)
        switch(item -> mode)
        {
                case MODE_VAR:
-                       jvm_push_var(p, item -> var);
+                       jvm_generate_load(p, item -> result, item -> var -> gen_var);
                        break;
                case MODE_INTEGER:
                        jvm_generate_push_int(p, item -> integer);
@@ -858,12 +1073,10 @@ push_item(gen_proc_t * p, oberon_item_t * item)
                        break;
                case MODE_FIELD:
                        push_item(p, item -> parent);
-                       char * field = jvm_get_field_full_name(item -> var);
-                       char * desc = jvm_get_descriptor(item -> var -> type);
-                       jvm_generate(p, 1, 1, "getfield %s %s", field, desc);
+                       jvm_generate_load(p, item -> result, item -> var -> gen_var);
                        break;
                case MODE_DEREF:
-                       /* Ð\92Ñ\81е Ð¾Ð±Ñ\8aекÑ\82Ñ\8b Ð¿Ñ\80едÑ\81Ñ\82авлÑ\8fÑ\8eÑ\82Ñ\81Ñ\8f ÐºÐ°Ðº Ñ\83вказатели */
+                       /* Ð\92Ñ\81е Ð¾Ð±Ñ\8aекÑ\82Ñ\8b Ð² jvm Ð¿Ñ\80едÑ\81Ñ\82авлÑ\8fÑ\8eÑ\82Ñ\81Ñ\8f ÐºÐ°Ðº Ñ\83казатели */
                        push_item(p, item -> parent);
                        break;
                case MODE_NIL:
@@ -1116,15 +1329,18 @@ store_expr(gen_proc_t * p, oberon_expr_t * dst, oberon_expr_t * src)
        if(dst -> result -> class == OBERON_TYPE_ARRAY
                || src -> result -> class == OBERON_TYPE_ARRAY)
        {
-               gen_error("array copy not implemented");
+               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)
        {
-               gen_error("record copy not implemented");
+               push_expr(p, dst);
+               push_expr(p, src);
+               jvm_generate_copy_record(p, dst -> result);
        }
-
-       switch(item -> mode)
+       else switch(item -> mode)
        {
                case MODE_VAR:
                        push_expr(p, src);