From 5f233e3ecee9f82dcf0db94e525f1a70caa67e6f Mon Sep 17 00:00:00 2001
From: DeaDDooMER <deaddoomer@deadsoftware.ru>
Date: Sat, 29 Jul 2017 16:11:58 +0300
Subject: [PATCH] =?utf8?q?JVM:=20=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB?=
 =?utf8?q?=D0=B5=D0=BD=D0=BE=20=D0=BA=D0=BE=D0=BF=D0=B8=D1=80=D0=BE=D0=B2?=
 =?utf8?q?=D0=B0=D0=BD=D0=B8=D0=B5=20=D0=BC=D0=B0=D1=81=D1=81=D0=B8=D0=B2?=
 =?utf8?q?=D0=BE=D0=B2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 src/backends/jvm/generator-jvm.c | 171 ++++++++++++++++++++++++++-----
 src/test.c                       |  17 +--
 2 files changed, 148 insertions(+), 40 deletions(-)

diff --git a/src/backends/jvm/generator-jvm.c b/src/backends/jvm/generator-jvm.c
index b46b27d..4b5b947 100644
--- a/src/backends/jvm/generator-jvm.c
+++ b/src/backends/jvm/generator-jvm.c
@@ -473,6 +473,140 @@ jvm_generate_copy_record(gen_proc_t * p, oberon_type_t * 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);
+	}
+}
 
 // ==========================================
 // ==========================================
@@ -638,7 +772,11 @@ oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
 		}
 		else if(field -> type -> class == OBERON_TYPE_ARRAY)
 		{
-			gen_error("copy array not implemented");
+			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
 		{
@@ -894,25 +1032,6 @@ oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr)
 	}
 }
 
-static void
-jvm_push_var(gen_proc_t * p, oberon_object_t * var)
-{
-	int cell_size = jvm_cell_size_for_type(var -> type);
-
-	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
-	{
-		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);
-	}
-}
-
 static void
 jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
 {
@@ -933,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);
@@ -954,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:
-			/* Все объекты представляются как увказатели */
+			/* Все объекты в jvm представляются как указатели */
 			push_item(p, item -> parent);
 			break;
 		case MODE_NIL:
@@ -1212,7 +1329,9 @@ 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)
diff --git a/src/test.c b/src/test.c
index b0e9bda..df0ff80 100644
--- a/src/test.c
+++ b/src/test.c
@@ -8,24 +8,13 @@ static char source_test[] =
 	"(* Main module *)"
 	"MODULE Test;"
 	"TYPE"
-	"  RecA = RECORD"
-	"    i, j, k : INTEGER;"
-	"  END;"
-	""
-	"  Rec = POINTER TO RecDesc;"
-	"  RecDesc = RECORD"
-	"    x, y, z : INTEGER;"
-	"    r, q, s : RecA;"
-	"    p, c, g : Rec;"
-	"  END;"
+	"  RecDesc = RECORD a : ARRAY 3, 5 OF INTEGER; END;"
 	""
 	"VAR"
-	"  a, b : Rec;"
+	"  a, b : ARRAY 3, 6 OF RecDesc;"
 	""
 	"BEGIN"
-	"  NEW(a);"
-	"  NEW(b);"
-	"  a^ := b^;"
+	"  a := b;"
 	"END Test."
 ;
 
-- 
2.29.2