X-Git-Url: https://deadsoftware.ru/gitweb?a=blobdiff_plain;f=src%2Fbackends%2Fjvm%2Fgenerator-jvm.c;h=f111fd16eff755440676a4b6a6e433713cc43b4b;hb=86c0ca1aafd465a3e0d4a9d6b1af661eba483ae1;hp=895a5ae3e524c2ef30063d1de38a316b2e3d23b1;hpb=31b71df908daa914198e78a0f43cf0a3965878dd;p=dsw-obn.git

diff --git a/src/backends/jvm/generator-jvm.c b/src/backends/jvm/generator-jvm.c
index 895a5ae..f111fd1 100644
--- a/src/backends/jvm/generator-jvm.c
+++ b/src/backends/jvm/generator-jvm.c
@@ -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)
 {
@@ -153,7 +166,61 @@ jvm_generate_push_int(gen_proc_t * p, int64_t i)
 	}
 	else
 	{
-		jvm_generate(p, 0, 2, "ldc2 %li", i);
+		jvm_generate(p, 0, 2, "ldc2_w %li", 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_w %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");
+		}
 	}
 }
 
@@ -191,11 +258,29 @@ jvm_generate_push_float(gen_proc_t * p, double f, int size)
 		}
 		else
 		{
-			jvm_generate(p, 0, 2, "ldc2 %lf", f);
+			jvm_generate(p, 0, 2, "ldc2_w %lf", f);
 		}
 	}
 }
 
+static void
+jvm_generate_push_string(gen_proc_t * p, char * str, int char_size)
+{
+	assert(char_size == 1);
+	int len = strlen(str);
+
+	jvm_generate_push_int(p, len + 1);
+	jvm_generate(p, 1, 1, "newarray byte");
+
+	for(int i = 0; i < len; i++)
+	{
+		jvm_generate(p, 1, 2, "dup");
+		jvm_generate_push_int(p, i);
+		jvm_generate_push_int(p, str[i]);
+		jvm_generate(p, 3, 0, "bastore");
+	}
+}
+
 static gen_var_t *
 oberon_generator_new_var()
 {
@@ -349,7 +434,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,10 +596,11 @@ 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
+	 *   int len = src.length
 	 *   while(i < len)
 	 *   {
 	 *     ...
@@ -540,7 +627,7 @@ jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
 		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_load(p, arr, src);
 		jvm_generate(p, 1, 1, "arraylength");
 		jvm_generate(p, 1, 0, "istore %i", loop[i].length -> reg);
 
@@ -551,12 +638,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,10 +695,21 @@ 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);
+}
+
 // ==========================================
 // ==========================================
 // ==========================================
 
+static void
+push_item(gen_proc_t * p, oberon_item_t * item);
+
 void
 oberon_generator_init_context(oberon_context_t * ctx)
 {
@@ -647,7 +740,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 +754,92 @@ 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, "  return\n");
+	fprintf(fp, ".end method\n\n");
+
+	fprintf(fp, ".method static <clinit>()V\n");
+	fprintf(fp, "  .limit stack 2\n");
+	fprintf(fp, "  new %s\n", cname);
+	fprintf(fp, "  dup\n");
+	fprintf(fp, "  invokespecial %s/<init>()V\n", cname);
+	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)
 {
@@ -676,7 +855,16 @@ oberon_generate_record_class(gen_module_t * m, oberon_type_t * rec)
 
 	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");
+
+	if(rec -> base == NULL)
+	{
+		fprintf(class -> fp, ".super  java/lang/Object\n\n");
+	}
+	else
+	{
+		class -> base = rec -> base -> gen_type -> class;
+		fprintf(class -> fp, ".super  %s\n\n", class -> base -> full_name);
+	}
 
 	rec -> gen_type -> class = class;
 }
@@ -695,6 +883,8 @@ oberon_generator_init_type(oberon_context_t * ctx, oberon_type_t * type)
 		case OBERON_TYPE_BOOLEAN:
 		case OBERON_TYPE_ARRAY:
 		case OBERON_TYPE_REAL:
+		case OBERON_TYPE_CHAR:
+		case OBERON_TYPE_STRING:
 			break;
 		case OBERON_TYPE_RECORD:
 			;
@@ -736,7 +926,14 @@ oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
 	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");
+	if(class -> base)
+	{
+		jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", class -> base -> full_name);
+	}
+	else
+	{
+		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++)
@@ -825,7 +1022,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 +1046,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 +1102,7 @@ oberon_generate_begin_module(oberon_context_t * ctx)
 				jvm_generate_store(p, x -> type, x -> gen_var);
 			}
 		}
+
 		x = x -> next;
 	}
 }
@@ -946,6 +1146,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 +1206,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
@@ -979,24 +1248,39 @@ oberon_generate_end_proc(oberon_context_t * ctx)
 }
 
 static void
-jvm_generate_call_proc(gen_proc_t * p, oberon_expr_t * desig)
+jvm_generate_call_proc(gen_proc_t * p, oberon_item_t * desig)
 {
-	assert(desig -> is_item);
-	assert(desig -> item.mode == MODE_CALL);
-	
-	oberon_object_t * proc;
-	char * fullname;
-	char * signature;
+	assert(desig -> var == NULL);
+	assert(desig -> mode == MODE_CALL);
 
-	proc = desig -> item.var;
-	fullname = jvm_get_field_full_name(proc);
-	signature = jvm_get_procedure_signature(proc -> type);
+	char * signature = NULL;
+
+	bool direct_call = false;
+	if(desig -> parent -> mode == MODE_VAR)
+	{
+		if(desig -> parent -> var -> class == OBERON_CLASS_PROC)
+		{
+			direct_call = true;
+		}
+	}
+
+	printf("direct_call == %i\n", direct_call);
+
+	oberon_type_t * procsig;
+	procsig = desig -> parent -> result;
+	signature = jvm_get_procedure_signature(procsig);
+
+	if(direct_call == false)
+	{
+		/* Загружаем указатель на процедуру */
+		push_item(p, desig -> parent);
+	}
 
 	int args_cells = 0;
-	int result_cells = jvm_cell_size_for_type(proc -> type -> base);
+	int result_cells = jvm_cell_size_for_type(procsig -> base);
 
-	int num = desig -> item.num_args;
-	oberon_expr_t * arg = desig -> item.args;
+	int num = desig -> num_args;
+	oberon_expr_t * arg = desig -> args;
 	for(int i = 0; i < num; i++)
 	{
 		args_cells += jvm_cell_size_for_type(arg -> result);
@@ -1004,13 +1288,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(direct_call)
+	{
+		char * full_name = jvm_get_field_full_name(desig -> parent -> var);
+		jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", full_name, signature);
+	}
+	else
+	{
+		char * cname = jvm_get_class_full_name(procsig);
+		jvm_generate(p, 1 + args_cells, result_cells, "invokevirtual %s/invoke%s", cname, 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);
+	assert(desig -> is_item);
+	jvm_generate_call_proc(ctx -> mod -> gen_mod -> class -> p, (oberon_item_t *) desig);
 }
 
 void
@@ -1051,38 +1345,75 @@ 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)
+{
+	if(to -> class == OBERON_TYPE_RECORD || to -> class == OBERON_TYPE_POINTER)
+	{
+		char * full_name = jvm_get_class_full_name(to);
+		jvm_generate(p, 1, 1, "checkcast %s", full_name);
+	}
+	else
+	{
+		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);
+		case MODE_CHAR:
+			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);
+			jvm_generate_call_proc(p, item);
 			break;
 		case MODE_INDEX:
 			;
 			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 +1424,9 @@ 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_STRING:
+			jvm_generate_push_string(p, item -> string, item -> result -> size);
+			break;
 		default:
 			gen_error("push_item: unk mode %i", item -> mode);
 			break;
@@ -1272,6 +1606,10 @@ push_operator(gen_proc_t * p, oberon_oper_t * oper)
 	int op = oper -> op;
 	switch(op)
 	{
+		case OP_CAST:
+			push_expr(p, oper -> left);
+			jvm_generate_cast_type(p, oper -> left -> result, oper -> result);
+			break;
 		case OP_UNARY_MINUS:
 		case OP_BITWISE_NOT:
 		case OP_LOGIC_NOT:
@@ -1355,13 +1693,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;