X-Git-Url: http://deadsoftware.ru/gitweb?a=blobdiff_plain;f=src%2Fbackends%2Fjvm%2Fgenerator-jvm.c;h=4b5b9474c0a2f006c99e15c5e4251e37763785b1;hb=5f233e3ecee9f82dcf0db94e525f1a70caa67e6f;hp=cbc834e58d5cf919926aae41423023dc4e259013;hpb=fef434efd26d259ed8a857e28af092df93282880;p=dsw-obn.git

diff --git a/src/backends/jvm/generator-jvm.c b/src/backends/jvm/generator-jvm.c
index cbc834e..4b5b947 100644
--- a/src/backends/jvm/generator-jvm.c
+++ b/src/backends/jvm/generator-jvm.c
@@ -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
@@ -261,7 +267,7 @@ jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num);
 /*
  * Функция jvm_generate_static_array_initialization генерирует код для
  * статической инициализации массива. На входе массив, на выходе тот же массив.
- * arrayref -- arrayref
+ * (arrayref -- arrayref)
  */
 
 static void
@@ -400,6 +406,32 @@ jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num)
 	}
 }
 
+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)
 {
@@ -426,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);
+	}
+}
+
 // ==========================================
 // ==========================================
 // ==========================================
@@ -543,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++)
@@ -562,8 +744,49 @@ oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
 			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);
 
@@ -809,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)
 {
@@ -848,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);
@@ -869,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:
@@ -1127,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);