DEADSOFTWARE

dc79ea11cd0cbb656036a37d67d9a512c46517ae
[dsw-obn.git] / src / backends / jvm / generator-jvm.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <stdarg.h>
4 #include <stdint.h>
5 #include <stdbool.h>
6 #include <string.h>
7 #include <assert.h>
9 #include <gc.h>
11 #include "../../../include/oberon.h"
12 #include "../../oberon-internals.h"
13 #include "generator-jvm.h"
14 #include "generator-jvm-basic.h"
16 // ==========================================
17 // ==========================================
18 // ==========================================
20 gen_proc_t *
21 jvm_create_proc(struct gen_class * class)
22 {
23 gen_proc_t * p = GC_MALLOC(sizeof *p);
24 memset(p, 0, sizeof *p);
26 struct gen_register_file * rf = GC_MALLOC(sizeof *rf);
27 memset(rf, 0, sizeof *rf);
29 struct gen_stack * stack = GC_MALLOC(sizeof *stack);
30 memset(rf, 0, sizeof *stack);
32 p -> rf = rf;
33 p -> stack = stack;
34 p -> class = class;
35 p -> label_id = 0;
37 return p;
38 }
40 void
41 jvm_generate_function_header(gen_proc_t * p, char * access, char * name, char * signature)
42 {
43 struct gen_class * class;
44 class = p -> class;
46 /* Делаем процедуру текущей в этом классе */
47 assert(class -> p == NULL);
48 class -> p = p;
50 fprintf(class -> fp, ".method %s %s%s\n", access, name, signature);
51 fprintf(class -> fp, " start:\n");
52 }
54 void
55 jvm_generate_function_end(gen_proc_t * p)
56 {
57 struct gen_class * class;
58 class = p -> class;
60 assert(class -> p);
61 class -> p = NULL;
63 int pointer = p -> stack -> pointer;
64 int max_pointer = p -> stack -> max_pointer;
65 int locals = p -> rf -> num_used;
66 int max_locals = p -> rf -> max_used;
68 fprintf(class -> fp, " .limit stack %i \t; current(%i)\n", max_pointer, pointer);
69 fprintf(class -> fp, " .limit locals %i \t; current(%i)\n", max_locals, locals);
70 fprintf(class -> fp, " end:\n");
71 fprintf(class -> fp, ".end method\n\n");
72 }
74 struct gen_class *
75 jvm_create_class(char * full_name)
76 {
77 struct gen_class * class = GC_MALLOC(sizeof *class);
78 memset(class, 0, sizeof *class);
80 char * fname = new_string("%s.j", full_name);
81 class -> full_name = new_string(full_name);
82 class -> fp = fopen(fname, "w");
83 assert(class -> fp);
85 return class;
86 }
88 static void
89 jvm_destroy_class(struct gen_class * class)
90 {
91 assert(class -> p == NULL);
92 fclose(class -> fp);
93 }
95 static void
96 jvm_stack_push(gen_proc_t * p, unsigned size)
97 {
98 p -> stack -> pointer += size;
99 if(p -> stack -> pointer > p -> stack -> max_pointer)
101 p -> stack -> max_pointer = p -> stack -> pointer;
105 static void
106 jvm_stack_pop(gen_proc_t * p, unsigned size)
108 p -> stack -> pointer -= size;
110 if(p -> stack -> pointer < 0)
112 printf("WARING: stack pointer %i\n", p -> stack -> pointer);
116 static void
117 jvm_generate(gen_proc_t * p, unsigned get, unsigned push, char * format, ...)
119 va_list ptr;
120 va_start(ptr, format);
122 jvm_stack_pop(p, get);
123 fprintf(p -> class -> fp, " ");
124 vfprintf(p -> class -> fp, format, ptr);
125 jvm_stack_push(p, push);
126 fprintf(p -> class -> fp, " \t ;>>>> %i -- %i : current_stack(%i)\n", get, push, p -> stack -> pointer);
128 va_end(ptr);
131 static void
132 jvm_generate_push_int(gen_proc_t * p, int64_t i)
134 if(i == -1)
136 jvm_generate(p, 0, 1, "iconst_m1");
138 else if(i >= 0 && i <= 5)
140 jvm_generate(p, 0, 1, "iconst_%li", i);
142 else if(i >= -128 && i <= 127)
144 jvm_generate(p, 0, 1, "bipush %li", i);
146 else if(i >= -32768 && i <= 32767)
148 jvm_generate(p, 0, 1, "sipush %li", i);
150 else if(i >= -2147483648 && i <= 2147483647)
152 jvm_generate(p, 0, 1, "ldc %li", i);
154 else
156 jvm_generate(p, 0, 2, "ldc2 %li", i);
160 static void
161 jvm_generate_push_float(gen_proc_t * p, double f, int size)
163 if(size <= 4)
165 if(f == 0.0)
167 jvm_generate(p, 0, 1, "fconst_0");
169 if(f == 1.0)
171 jvm_generate(p, 0, 1, "fconst_1");
173 if(f == 2.0)
175 jvm_generate(p, 0, 1, "fconst_2");
177 else
179 jvm_generate(p, 0, 1, "ldc %lf", f);
182 else
184 if(f == 0.0)
186 jvm_generate(p, 0, 2, "dconst_0");
188 if(f == 1.0)
190 jvm_generate(p, 0, 2, "dconst_1");
192 else
194 jvm_generate(p, 0, 2, "ldc2 %lf", f);
199 static gen_var_t *
200 oberon_generator_new_var()
202 gen_var_t * v = GC_MALLOC(sizeof *v);
203 memset(v, 0, sizeof *v);
204 return v;
207 static void
208 jvm_generate_and_init_global_var(gen_var_t * v, struct gen_class * class, char * access, char * name, char * desc)
210 assert(class -> p == NULL);
211 fprintf(class -> fp, ".field %s static %s %s\n\n", access, name, desc);
213 v -> storage = JVM_STORAGE_STATIC;
214 v -> full_name = new_string("%s/%s", class -> full_name, name);
215 v -> desc = new_string(desc);
218 static void
219 jvm_generate_and_init_field(gen_var_t * v, struct gen_class * class, char * access, char * name, char * desc)
221 assert(class -> p == NULL);
222 fprintf(class -> fp, ".field %s %s %s\n\n", access, name, desc);
224 v -> storage = JVM_STORAGE_FIELD;
225 v -> full_name = new_string("%s/%s", class -> full_name, name);
226 v -> desc = new_string(desc);
229 static void
230 jvm_generate_and_init_local_var(gen_var_t * v, gen_proc_t * p, bool wide)
232 v -> storage = JVM_STORAGE_REGISTER;
233 v -> reg = jvm_alloc_register_untyped(p -> rf, wide);
236 static void
237 jvm_generate_and_init_named_local_var(gen_var_t * v, gen_proc_t * p, bool wide, char * name, char * desc)
239 jvm_generate_and_init_local_var(v, p, wide);
240 v -> desc = new_string(desc);
241 jvm_generate(p, 0, 0, ".var %i is %s %s from start to end", v -> reg, name, desc);
244 static int
245 jvm_new_label_id(gen_proc_t * p)
247 int label_id = p -> label_id;
248 p -> label_id += 1;
249 return label_id;
252 static char *
253 jvm_get_label_name(int label_id)
255 return new_string("L%i", label_id);
258 static void
259 jvm_generate_label(gen_proc_t * p, int label_id)
261 jvm_generate(p, 0, 0, "L%i:", label_id);
264 static void
265 jvm_generate_load(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * src)
267 char prefix;
268 int cell_size;
270 cell_size = jvm_cell_size_for_type(src_type);
272 switch(src -> storage)
274 case JVM_STORAGE_REGISTER:
275 prefix = jvm_get_prefix(src_type);
276 jvm_generate(p, 0, cell_size, "%cload %i", prefix, src -> reg);
277 break;
278 case JVM_STORAGE_STATIC:
279 jvm_generate(p, 0, cell_size, "getstatic %s %s", src -> full_name, src -> desc);
280 break;
281 case JVM_STORAGE_FIELD:
282 jvm_generate(p, 1, cell_size, "getfield %s %s", src -> full_name, src -> desc);
283 break;
284 default:
285 gen_error("jvm_generate_load: unknow storage type %i", src -> storage);
286 break;
290 static void
291 jvm_generate_store(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * dst)
293 char prefix;
294 int cell_size;
296 cell_size = jvm_cell_size_for_type(src_type);
298 switch(dst -> storage)
300 case JVM_STORAGE_REGISTER:
301 prefix = jvm_get_prefix(src_type);
302 jvm_generate(p, cell_size, 0, "%cstore %i", prefix, dst -> reg);
303 break;
304 case JVM_STORAGE_STATIC:
305 jvm_generate(p, cell_size, 0, "putstatic %s %s", dst -> full_name, dst -> desc);
306 break;
307 case JVM_STORAGE_FIELD:
308 jvm_generate(p, 1 + cell_size, 0, "putfield %s %s", dst -> full_name, dst -> desc);
309 break;
310 default:
311 gen_error("jvm_generate_store: unknow storage type %i", dst -> storage);
312 break;
316 static void
317 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num);
319 /*
320 * Генерирует код для инициализации массива со статическим базовым типом
321 * ( aref -- )
322 */
324 static void
325 jvm_generate_array_initialization(gen_proc_t * p, oberon_type_t * arr)
327 int dim = 0;
328 oberon_type_t * base = arr;
329 while(base -> class == OBERON_TYPE_ARRAY)
331 dim += 1;
332 base = base -> base;
335 if(base -> class != OBERON_TYPE_RECORD)
337 jvm_generate(p, 1, 0, "pop");
338 return;
341 struct {
342 gen_var_t * index;
343 gen_var_t * length;
344 int start;
345 int end;
346 } loop[dim];
348 gen_var_t * dst = oberon_generator_new_var();
349 jvm_generate_and_init_local_var(dst, p, false);
350 jvm_generate_store(p, arr, dst);
352 /*
353 * Входящие параметры заграблены.
354 * Теперь генерируем эквивалентный код:
355 * int i = 0;
356 * int len = dst.length
357 * while(i < len)
358 * {
359 * ...
360 * {
361 * dst[i, ...] = new record;
362 * }
363 * ...
364 * i += 1;
365 * }
366 * Где "..." такой же код (начало и конец) для следующей размерности.
367 */
369 for(int i = 0; i < dim; i++)
371 loop[i].index = oberon_generator_new_var();
372 loop[i].length = oberon_generator_new_var();
373 loop[i].start = jvm_new_label_id(p);
374 loop[i].end = jvm_new_label_id(p);
375 jvm_generate_and_init_local_var(loop[i].index, p, false);
376 jvm_generate_and_init_local_var(loop[i].length, p, false);
378 /* TODO преределать через jvm_generate_load{{store}} */
380 jvm_generate(p, 0, 1, "iconst_0");
381 jvm_generate(p, 1, 0, "istore %i", loop[i].index -> reg);
383 jvm_generate_load(p, arr, dst);
384 jvm_generate(p, 1, 1, "arraylength");
385 jvm_generate(p, 1, 0, "istore %i", loop[i].length -> reg);
387 /* if(i >= len) goto end; */
388 jvm_generate_label(p, loop[i].start);
389 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
390 jvm_generate(p, 0, 1, "iload %i", loop[i].length -> reg);
391 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
394 jvm_generate_load(p, arr, dst);
395 jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg);
396 for(int i = 1; i < dim; i++)
398 jvm_generate(p, 2, 1, "aaload");
399 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
401 jvm_generate_new(p, base, 0);
402 jvm_generate(p, 3, 0, "aastore");
404 for(int i = dim - 1; i >= 0; i--)
406 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].index -> reg);
407 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
408 jvm_generate_label(p, loop[i].end);
412 static void
413 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num)
415 int dim;
416 char * cname;
417 char * desc;
418 oberon_type_t * base;
420 switch(type -> class)
422 case OBERON_TYPE_INTEGER:
423 case OBERON_TYPE_BOOLEAN:
424 case OBERON_TYPE_PROCEDURE:
425 case OBERON_TYPE_REAL:
426 case OBERON_TYPE_POINTER:
427 gen_error("jvm_generate_new_static: static alocation not allowed");
428 break;
429 case OBERON_TYPE_RECORD:
430 assert(num == 0);
431 cname = jvm_get_class_full_name(type);
432 jvm_generate(p, 0, 1, "new %s", cname);
433 jvm_generate(p, 1, 2, "dup");
434 jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", cname);
435 break;
436 case OBERON_TYPE_ARRAY:
437 dim = 0;
438 base = type;
439 desc = jvm_get_descriptor(type);
440 while(base -> class == OBERON_TYPE_ARRAY)
442 if(num > 0)
444 assert(base -> size == 0);
445 num -= 1;
447 else
449 assert(base -> size > 0);
450 jvm_generate_push_int(p, base -> size);
452 dim += 1;
453 base = base -> base;
456 assert(num == 0);
457 jvm_generate(p, dim, 1, "multianewarray %s %i", desc, dim);
458 jvm_generate(p, 1, 2, "dup");
459 jvm_generate_array_initialization(p, type);
460 break;
461 default:
462 gen_error("jvm_generate_new_static: unk type class %i", type -> class);
463 break;
467 /*
468 * Генерирует код для копирования полей из первой записи во вторую.
469 * ( aref_dst aref_src -- )
470 * dst := src;
471 */
473 static void
474 jvm_generate_copy_record(gen_proc_t * p, oberon_type_t * rec)
476 assert(rec -> class == OBERON_TYPE_RECORD);
477 char * desc = jvm_get_descriptor(rec);
478 char * cname = jvm_get_class_full_name(rec);
479 jvm_generate(p, 1 + 1, 0, "invokestatic %s/$COPY$(%s%s)V", cname, desc, desc);
482 /*
483 * Генерирует кода для копирования массивов.
484 * ( aref_dst aref_src -- )
485 * dst := src;
486 */
488 static void
489 jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
491 int dim = 0;
492 oberon_type_t * base = arr;
493 while(base -> class == OBERON_TYPE_ARRAY)
495 dim += 1;
496 base = base -> base;
499 struct {
500 gen_var_t * index;
501 gen_var_t * length;
502 int start;
503 int end;
504 } loop[dim];
506 gen_var_t * dst = oberon_generator_new_var();
507 gen_var_t * src = oberon_generator_new_var();
508 jvm_generate_and_init_local_var(dst, p, false);
509 jvm_generate_and_init_local_var(src, p, false);
511 jvm_generate_store(p, arr, src);
512 jvm_generate_store(p, arr, dst);
514 /*
515 * Входящие параметры заграблены.
516 * Теперь генерируем эквивалентный код:
517 * int i = 0;
518 * int len = dst.length
519 * while(i < len)
520 * {
521 * ...
522 * {
523 * copy from src[i, ...] to dst[i, ...];
524 * }
525 * ...
526 * i += 1;
527 * }
528 * Где "..." такой же код (начало и конец) для следующей размерности.
529 */
531 for(int i = 0; i < dim; i++)
533 loop[i].index = oberon_generator_new_var();
534 loop[i].length = oberon_generator_new_var();
535 loop[i].start = jvm_new_label_id(p);
536 loop[i].end = jvm_new_label_id(p);
537 jvm_generate_and_init_local_var(loop[i].index, p, false);
538 jvm_generate_and_init_local_var(loop[i].length, p, false);
540 /* TODO преределать через jvm_generate_load{{store}} */
542 jvm_generate(p, 0, 1, "iconst_0");
543 jvm_generate(p, 1, 0, "istore %i", loop[i].index -> reg);
545 jvm_generate_load(p, arr, dst);
546 jvm_generate(p, 1, 1, "arraylength");
547 jvm_generate(p, 1, 0, "istore %i", loop[i].length -> reg);
549 /* if(i >= len) goto end; */
550 jvm_generate_label(p, loop[i].start);
551 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
552 jvm_generate(p, 0, 1, "iload %i", loop[i].length -> reg);
553 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
556 if(base -> class == OBERON_TYPE_RECORD)
558 /* Получаем записи по индексам ( -- dst src ) */
560 jvm_generate_load(p, arr, dst);
561 for(int i = 0; i < dim; i++)
563 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
564 jvm_generate(p, 2, 1, "aaload");
567 jvm_generate_load(p, arr, src);
568 for(int i = 0; i < dim; i++)
570 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
571 jvm_generate(p, 2, 1, "aaload");
574 /* Копируем записи ( dst src -- ) */
575 jvm_generate_copy_record(p, base);
577 else
579 char postfix = jvm_get_postfix(base);
580 int cell_size = jvm_cell_size_for_postfix(postfix);
582 /* Получаем массивы и индексы ( -- dst i src i ) */
584 jvm_generate_load(p, arr, dst);
585 jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg);
586 for(int i = 1; i < dim; i++)
588 jvm_generate(p, 2, 1, "aaload");
589 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
592 jvm_generate_load(p, arr, src);
593 jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg);
594 for(int i = 1; i < dim; i++)
596 jvm_generate(p, 2, 1, "aaload");
597 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
600 /* Копируем значения ( dst i src i -- ) */
601 jvm_generate(p, 2, cell_size, "%caload", postfix);
602 jvm_generate(p, 2 + cell_size, 0, "%castore", postfix);
605 for(int i = dim - 1; i >= 0; i--)
607 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].index -> reg);
608 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
609 jvm_generate_label(p, loop[i].end);
613 // ==========================================
614 // ==========================================
615 // ==========================================
617 void
618 oberon_generator_init_context(oberon_context_t * ctx)
620 gen_context_t * gen_context = GC_MALLOC(sizeof *gen_context);
621 memset(gen_context, 0, sizeof *gen_context);
623 ctx -> gen_context = gen_context;
626 void
627 oberon_generator_destroy_context(oberon_context_t * ctx)
631 static void
632 oberon_generate_procedure_class(oberon_type_t * proc)
634 FILE * fp;
635 char * cname;
636 char * signature;
637 struct gen_class * class;
639 cname = jvm_get_class_full_name(proc);
640 class = jvm_create_class(cname);
641 fp = class -> fp;
643 fprintf(fp, ".source SYSTEM\n");
644 fprintf(fp, ".class public abstract %s\n", cname);
645 fprintf(fp, ".super java/lang/Object\n\n");
647 fprintf(fp, ".method public <init>()V\n");
648 fprintf(fp, " aload_0\n");
649 fprintf(fp, " invokespecial java/lang/Object/<init>()V\n");
650 fprintf(fp, " return\n");
651 fprintf(fp, ".end method\n\n");
653 signature = jvm_get_procedure_signature(proc);
655 fprintf(fp, ".method public abstract invoke%s\n", signature);
656 fprintf(fp, ".end method\n\n");
658 jvm_destroy_class(class);
661 static void
662 oberon_generate_record_class(gen_module_t * m, oberon_type_t * rec)
664 char * cname;
665 struct gen_class * class;
667 /* Устанавливаем новоый id */
668 rec -> gen_type -> rec_id = m -> rec_id;
669 m -> rec_id += 1;
671 cname = jvm_get_class_full_name(rec);
672 class = jvm_create_class(cname);
674 fprintf(class -> fp, ".source %s\n", rec -> module -> name);
675 fprintf(class -> fp, ".class public %s\n", cname);
676 fprintf(class -> fp, ".super java/lang/Object\n\n");
678 rec -> gen_type -> class = class;
681 void
682 oberon_generator_init_type(oberon_context_t * ctx, oberon_type_t * type)
684 gen_type_t * t = GC_MALLOC(sizeof *t);
685 memset(t, 0, sizeof *t);
686 type -> gen_type = t;
688 switch(type -> class)
690 case OBERON_TYPE_VOID:
691 case OBERON_TYPE_INTEGER:
692 case OBERON_TYPE_BOOLEAN:
693 case OBERON_TYPE_ARRAY:
694 case OBERON_TYPE_REAL:
695 break;
696 case OBERON_TYPE_RECORD:
698 gen_module_t * m;
699 m = type -> module -> gen_mod;
700 oberon_generate_record_class(m, type);
701 break;
702 case OBERON_TYPE_PROCEDURE:
703 oberon_generate_procedure_class(type);
704 break;
705 case OBERON_TYPE_POINTER:
706 break;
707 default:
708 gen_error("oberon_generator_init_type: unk calss %i", type -> class);
709 break;
713 void
714 oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
716 struct gen_class * class;
717 class = rec -> gen_type -> class;
719 int num = rec -> num_decl;
720 oberon_object_t * field = rec -> decl;
721 for(int i = 0; i < num; i++)
723 gen_var_t * v = field -> gen_var;
724 char * name = field -> name;
725 char * desc = jvm_get_descriptor(field -> type);
726 jvm_generate_and_init_field(v, class, "public", name, desc);
727 field = field -> next;
730 /* Стандартный конструктор класса */
731 /* Инициализирует внутренние статические записи и массивы */
732 gen_proc_t * p = jvm_create_proc(class);
733 jvm_generate_function_header(p, "public", "<init>", "()V");
734 jvm_alloc_register_untyped(p -> rf, false);
735 jvm_generate(p, 0, 1, "aload_0");
736 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
737 num = rec -> num_decl;
738 field = rec -> decl;
739 for(int i = 0; i < num; i++)
741 if(field -> type -> class == OBERON_TYPE_RECORD
742 || field -> type -> class == OBERON_TYPE_ARRAY)
744 jvm_generate(p, 0, 1, "aload_0");
745 jvm_generate_new(p, field -> type, 0);
746 jvm_generate_store(p, field -> type, field -> gen_var);
748 field = field -> next;
750 jvm_generate(p, 0, 0, "return");
751 jvm_generate_function_end(p);
753 /* Метод для копирования полей класса */
754 /* reg0 == src -> reg1 == dst */
755 p = jvm_create_proc(class);
756 gen_var_t * copy_dst = oberon_generator_new_var();
757 gen_var_t * copy_src = oberon_generator_new_var();
758 char * desc = jvm_get_descriptor(rec);
759 char * signature = new_string("(%s%s)V", desc, desc);
760 jvm_generate_function_header(p, "public static", "$COPY$", signature);
761 jvm_generate_and_init_named_local_var(copy_dst, p, false, "dst", desc);
762 jvm_generate_and_init_named_local_var(copy_src, p, false, "src", desc);
763 num = rec -> num_decl;
764 field = rec -> decl;
765 for(int i = 0; i < num; i++)
767 if(field -> type -> class == OBERON_TYPE_RECORD)
769 jvm_generate_load(p, rec, copy_dst);
770 jvm_generate_load(p, field -> type, field -> gen_var);
771 jvm_generate_load(p, rec, copy_src);
772 jvm_generate_load(p, field -> type, field -> gen_var);
773 jvm_generate_copy_record(p, field -> type);
775 else if(field -> type -> class == OBERON_TYPE_ARRAY)
777 jvm_generate_load(p, rec, copy_dst);
778 jvm_generate_load(p, field -> type, field -> gen_var);
779 jvm_generate_load(p, rec, copy_src);
780 jvm_generate_load(p, field -> type, field -> gen_var);
781 jvm_generate_copy_array(p, field -> type);
783 else
785 jvm_generate_load(p, rec, copy_dst);
786 jvm_generate_load(p, rec, copy_src);
787 jvm_generate_load(p, field -> type, field -> gen_var);
788 jvm_generate_store(p, field -> type, field -> gen_var);
790 field = field -> next;
792 jvm_generate(p, 0, 0, "return");
793 jvm_generate_function_end(p);
795 jvm_destroy_class(class);
798 void
799 oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var)
801 gen_var_t * v = oberon_generator_new_var();
802 var -> gen_var = v;
804 gen_module_t * m;
805 m = ctx -> mod -> gen_mod;
807 struct gen_class * class;
808 class = m -> class;
810 char * desc;
811 switch(var -> class)
813 case OBERON_CLASS_VAR_PARAM:
814 gen_error("generator: VAR-parameters not implemented");
815 break;
816 case OBERON_CLASS_PARAM:
817 /* Заполняется при генерации функции */
818 /* смотри jvm_init_local_object() */
819 break;
820 case OBERON_CLASS_FIELD:
821 /* Заполняются при инициализации структуры */
822 /* смотри oberon_generator_init_record() */
823 break;
824 case OBERON_CLASS_VAR:
825 /* Локальные заполняются при генерации функции */
826 /* смотри jvm_init_local_object() */
827 if(var -> local == 0)
829 desc = jvm_get_descriptor(var -> type);
830 jvm_generate_and_init_global_var(v, class, "public", var -> name, desc);
832 break;
833 default:
834 gen_error("oberon_generator_init_var: unk var class %i", var -> class);
835 break;
839 void
840 oberon_generator_init_proc(oberon_context_t * ctx, oberon_object_t * proc)
842 struct gen_class * class = proc -> module -> gen_mod -> class;
843 proc -> gen_proc = jvm_create_proc(class);
845 if(proc -> local)
847 gen_error("generator: local procedures not implemented");
851 void
852 oberon_generator_init_module(oberon_context_t * ctx, oberon_module_t * mod)
854 gen_module_t * m = GC_MALLOC(sizeof *m);
855 memset(m, 0, sizeof *m);
856 mod -> gen_mod = m;
858 struct gen_class * class;
859 class = jvm_create_class(mod -> name);
861 fprintf(class -> fp, ".source %s\n", mod -> name);
862 fprintf(class -> fp, ".class %s\n", mod -> name);
863 fprintf(class -> fp, ".super java/lang/Object\n\n");
865 m -> class = class;
867 ctx -> gen_context -> current_m = m;
870 void
871 oberon_generator_fini_module(oberon_module_t * mod)
873 jvm_destroy_class(mod -> gen_mod -> class);
876 static void
877 push_expr(gen_proc_t * p, oberon_expr_t * expr);
879 void
880 oberon_generate_begin_module(oberon_context_t * ctx)
882 struct gen_class * class = ctx -> mod -> gen_mod -> class;
883 gen_proc_t * p = jvm_create_proc(class);
885 jvm_generate_function_header(p, "public", "<init>", "()V");
886 jvm_alloc_register_untyped(p -> rf, false);
887 jvm_generate(p, 0, 1, "aload_0");
888 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
890 /* Инициализация переменных объявленных в модуле */
891 oberon_object_t * x = ctx -> mod -> decl -> list -> next;
892 while(x != NULL)
894 if(x -> class == OBERON_CLASS_VAR)
896 if(x -> type -> class == OBERON_TYPE_ARRAY
897 || x -> type -> class == OBERON_TYPE_RECORD)
899 jvm_generate_new(p, x -> type, 0);
900 jvm_generate_store(p, x -> type, x -> gen_var);
903 x = x -> next;
907 void
908 oberon_generate_end_module(oberon_context_t * ctx)
910 struct gen_class * class = ctx -> mod -> gen_mod -> class;
911 gen_proc_t * p = class -> p;
913 jvm_generate(p, 0, 0, "return");
914 jvm_generate_function_end(class -> p);
917 static void
918 jvm_init_local_object(gen_proc_t * p, oberon_object_t * x)
920 gen_var_t * v;
921 bool wide;
922 char * desc;
923 struct gen_class * class;
925 v = x -> gen_var;
926 class = p -> class;
927 wide = jvm_is_wide_type(x -> type);
928 desc = jvm_get_descriptor(x -> type);
930 // Убеждаемся что сейчас находимся в функции
931 assert(class -> p);
932 assert(x -> local);
934 switch(x -> class)
936 case OBERON_CLASS_VAR:
937 case OBERON_CLASS_PARAM:
938 jvm_generate_and_init_named_local_var(v, p, wide, x -> name, desc);
939 break;
940 default:
941 gen_error("jvm_init_local_object: wat");
942 break;
946 /*
947 * Генерирует код для получения размера измерения массива
948 * Аналог Обероновского LEN(v, n); где n = 0 - первое измерение.
949 * ( aref -- int )
950 */
952 static void
953 jvm_generate_array_len(gen_proc_t * p, int dim)
955 while(dim > 0)
957 jvm_generate(p, 0, 1, "iconst_0");
958 jvm_generate(p, 2, 1, "aaload");
959 dim -= 1;
961 jvm_generate(p, 1, 1, "arraylength");
964 static void
965 jvm_generate_array_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * arr)
967 int dim = 0;
968 oberon_type_t * base = arr;
969 while(base -> class == OBERON_TYPE_ARRAY)
971 if(base -> size == 0)
973 jvm_generate_load(p, arr, v);
974 jvm_generate_array_len(p, dim);
975 dim += 1;
977 base = base -> base;
980 jvm_generate_new(p, arr, dim);
981 jvm_generate(p, 1, 2, "dup");
982 jvm_generate_load(p, arr, v);
983 jvm_generate_copy_array(p, arr);
984 jvm_generate_store(p, arr, v);
987 static void
988 jvm_generate_record_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * rec)
990 jvm_generate_new(p, rec, 0);
991 jvm_generate(p, 1, 2, "dup");
992 jvm_generate_load(p, rec, v);
993 jvm_generate_copy_record(p, rec);
994 jvm_generate_store(p, rec, v);
997 void
998 oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc)
1000 gen_proc_t * p;
1001 char * signature;
1003 p = proc -> gen_proc;
1005 signature = jvm_get_procedure_signature(proc -> type);
1006 jvm_generate_function_header(p, "public static", proc -> name, signature);
1008 /* Выделение регистров под параметры и переменные */
1009 oberon_object_t * var = proc -> type -> decl;
1010 while(var)
1012 jvm_init_local_object(p, var);
1013 var = var -> next;
1016 /* Копирование статических/открытых массивов и записей */
1017 var = proc -> type -> decl;
1018 while(var)
1020 if(var -> class == OBERON_CLASS_PARAM || var -> class == OBERON_CLASS_VAR_PARAM)
1022 if(var -> type -> class == OBERON_TYPE_ARRAY)
1024 jvm_generate_array_duplicate_and_replace(p, var -> gen_var, var -> type);
1026 else if(var -> type -> class == OBERON_TYPE_RECORD)
1027 {
1028 jvm_generate_record_duplicate_and_replace(p, var -> gen_var, var -> type);
1031 var = var -> next;
1032 }
1035 void
1036 oberon_generate_end_proc(oberon_context_t * ctx)
1038 gen_module_t * m;
1039 gen_proc_t * p;
1041 m = ctx -> mod -> gen_mod;
1042 p = m -> class -> p;
1044 jvm_generate_function_end(p);
1047 static void
1048 jvm_generate_call_proc(gen_proc_t * p, oberon_expr_t * desig)
1050 assert(desig -> is_item);
1051 assert(desig -> item.mode == MODE_CALL);
1053 oberon_object_t * proc;
1054 char * fullname;
1055 char * signature;
1057 proc = desig -> item.var;
1058 fullname = jvm_get_field_full_name(proc);
1059 signature = jvm_get_procedure_signature(proc -> type);
1061 int args_cells = 0;
1062 int result_cells = jvm_cell_size_for_type(proc -> type -> base);
1064 int num = desig -> item.num_args;
1065 oberon_expr_t * arg = desig -> item.args;
1066 for(int i = 0; i < num; i++)
1068 args_cells += jvm_cell_size_for_type(arg -> result);
1069 push_expr(p, arg);
1070 arg = arg -> next;
1073 jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", fullname, signature);
1076 void
1077 oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig)
1079 jvm_generate_call_proc(ctx -> gen_context -> current_m -> class -> p, desig);
1082 void
1083 oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr)
1085 gen_module_t * m;
1086 gen_proc_t * p;
1087 char prefix;
1088 int cell_size;
1090 m = ctx -> mod -> gen_mod;
1091 p = m -> class -> p;
1093 if(expr)
1095 push_expr(p, expr);
1096 prefix = jvm_get_prefix(expr -> result);
1097 cell_size = jvm_cell_size_for_type(expr -> result);
1098 jvm_generate(p, cell_size, 0, "%creturn", prefix);
1100 else
1102 jvm_generate(p, 0, 0, "return");
1106 static void
1107 jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
1109 assert(type -> class == OBERON_TYPE_POINTER);
1111 for(int i = 0; i < num; i++)
1113 push_expr(p, arg);
1114 arg = arg -> next;
1117 jvm_generate_new(p, type -> base, num);
1120 static void
1121 push_item(gen_proc_t * p, oberon_item_t * item)
1123 switch(item -> mode)
1125 case MODE_VAR:
1126 jvm_generate_load(p, item -> result, item -> var -> gen_var);
1127 break;
1128 case MODE_INTEGER:
1129 jvm_generate_push_int(p, item -> integer);
1130 break;
1131 case MODE_BOOLEAN:
1132 jvm_generate_push_int(p, item -> boolean);
1133 break;
1134 case MODE_CALL:
1135 jvm_generate_call_proc(p, (oberon_expr_t *) item);
1136 break;
1137 case MODE_INDEX:
1139 char postfix = jvm_get_postfix(item -> result);
1140 int cell_size = jvm_cell_size_for_postfix(postfix);
1141 push_item(p, item -> parent);
1142 push_expr(p, item -> args);
1143 jvm_generate(p, 1 + 1, cell_size, "%caload", postfix);
1144 break;
1145 case MODE_FIELD:
1146 push_item(p, item -> parent);
1147 jvm_generate_load(p, item -> result, item -> var -> gen_var);
1148 break;
1149 case MODE_DEREF:
1150 /* Все объекты в jvm представляются как указатели */
1151 push_item(p, item -> parent);
1152 break;
1153 case MODE_NIL:
1154 jvm_generate(p, 0, 1, "aconst_null");
1155 break;
1156 case MODE_NEW:
1157 jvm_generate_expr_new_pointer(p, item -> result, item -> num_args, item -> args);
1158 break;
1159 case MODE_REAL:
1160 jvm_generate_push_float(p, item -> real, item -> result -> size);
1161 break;
1162 default:
1163 gen_error("push_item: unk mode %i", item -> mode);
1164 break;
1168 static void
1169 jvm_generate_logical_not(gen_proc_t * p)
1171 int label_done = jvm_new_label_id(p);
1172 int label_false = jvm_new_label_id(p);
1173 char * label_name_done = jvm_get_label_name(label_done);
1174 char * label_name_false = jvm_get_label_name(label_false);
1176 jvm_generate(p, 1, 0, "ifne %s", label_name_false);
1177 jvm_generate(p, 0, 1, "iconst_1");
1178 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1179 jvm_generate_label(p, label_false);
1180 jvm_generate(p, 0, 1, "iconst_0");
1181 jvm_generate_label(p, label_done);
1184 static void
1185 jvm_generate_abs(gen_proc_t * p, char prefix)
1187 char t = jvm_get_type_of_prefix(prefix);
1188 int cell_size = jvm_cell_size_for_postfix(prefix);
1189 jvm_generate(p, cell_size, cell_size, "invokestatic java/lang/Math/abs(%c)%c", t, t);
1192 static void
1193 jvm_generate_compare_op(gen_proc_t * p, char prefix, int op)
1195 int label_true = jvm_new_label_id(p);
1196 int label_done = jvm_new_label_id(p);
1197 char * label_name_true = jvm_get_label_name(label_true);
1198 char * label_name_done = jvm_get_label_name(label_done);
1199 int cell_size = 2 * jvm_cell_size_for_postfix(prefix);
1201 assert(prefix == 'i' || prefix == 'a');
1203 const char * cmpop = "";
1204 switch(op)
1206 case OP_EQ:
1207 cmpop = "eq";
1208 break;
1209 case OP_NEQ:
1210 cmpop = "ne";
1211 break;
1212 case OP_LSS:
1213 cmpop = "lt";
1214 break;
1215 case OP_LEQ:
1216 cmpop = "le";
1217 break;
1218 case OP_GRT:
1219 cmpop = "gt";
1220 break;
1221 case OP_GEQ:
1222 cmpop = "ge";
1223 break;
1224 default:
1225 gen_error("jvm_generate_compare_op: wat");
1226 break;
1229 jvm_generate(p, cell_size, 0, "if_%ccmp%s %s", prefix, cmpop, label_name_true);
1230 jvm_generate(p, 0, 1, "iconst_0");
1231 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1232 jvm_generate_label(p, label_true);
1233 jvm_generate(p, 0, 1, "iconst_1");
1234 jvm_generate_label(p, label_done);
1237 static void
1238 jvm_generate_operator(gen_proc_t * p, char prefix, int op)
1240 int cell_size = jvm_cell_size_for_postfix(prefix);
1241 switch(op)
1243 case OP_UNARY_MINUS:
1244 jvm_generate(p, cell_size, cell_size, "%cneg", prefix);
1245 break;
1246 case OP_BITWISE_NOT:
1247 jvm_generate_push_int(p, -1);
1248 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1249 break;
1250 case OP_LOGIC_NOT:
1251 jvm_generate_logical_not(p);
1252 break;
1253 case OP_ABS:
1254 jvm_generate_abs(p, prefix);
1255 break;
1257 case OP_ADD:
1258 jvm_generate(p, 2 * cell_size, cell_size, "%cadd", prefix);
1259 break;
1260 case OP_SUB:
1261 jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix);
1262 break;
1263 case OP_MUL:
1264 jvm_generate(p, 2 * cell_size, cell_size, "%cmul", prefix);
1265 break;
1266 case OP_DIV:
1267 jvm_generate(p, 2 * cell_size, cell_size, "%cdiv", prefix);
1268 break;
1269 case OP_MOD:
1270 jvm_generate(p, 2 * cell_size, cell_size, "%crem", prefix);
1271 break;
1272 case OP_BITWISE_AND:
1273 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1274 break;
1275 case OP_BITWISE_XOR:
1276 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1277 break;
1278 case OP_BITWISE_OR:
1279 jvm_generate(p, 2 * cell_size, cell_size, "%cor", prefix);
1280 break;
1282 case OP_EQ:
1283 case OP_NEQ:
1284 case OP_LSS:
1285 case OP_LEQ:
1286 case OP_GRT:
1287 case OP_GEQ:
1288 jvm_generate_compare_op(p, prefix, op);
1289 break;
1290 default:
1291 gen_error("jvm_generate_operator: unk op %i", op);
1292 break;
1293 }
1296 static void
1297 jvm_generate_logical_or(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1299 int label_calc_b = jvm_new_label_id(p);
1300 int label_done = jvm_new_label_id(p);
1301 char * label_name_calc_b = jvm_get_label_name(label_calc_b);
1302 char * label_name_done = jvm_get_label_name(label_done);
1304 /* a OR b -- если a, то TRUE, иначе b */
1306 push_expr(p, a);
1307 jvm_generate(p, 1, 0, "ifne %s", label_name_calc_b);
1308 jvm_generate(p, 0, 1, "iconst_1");
1309 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1310 jvm_generate_label(p, label_calc_b);
1311 push_expr(p, b);
1312 jvm_generate_label(p, label_done);
1315 static void
1316 jvm_generate_logical_and(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1318 int label_false = jvm_new_label_id(p);
1319 int label_done = jvm_new_label_id(p);
1320 char * label_name_false = jvm_get_label_name(label_false);
1321 char * label_name_done = jvm_get_label_name(label_done);
1323 /* a AND b -- если a, то b, иначе FALSE */
1325 push_expr(p, a);
1326 jvm_generate(p, 1, 0, "ifne %s", label_name_false);
1327 push_expr(p, b);
1328 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1329 jvm_generate_label(p, label_false);
1330 jvm_generate(p, 0, 1, "iconst_0");
1331 jvm_generate_label(p, label_done);
1334 static void
1335 push_operator(gen_proc_t * p, oberon_oper_t * oper)
1337 char prefix = jvm_get_prefix(oper -> result);
1338 int op = oper -> op;
1339 switch(op)
1341 case OP_UNARY_MINUS:
1342 case OP_BITWISE_NOT:
1343 case OP_LOGIC_NOT:
1344 case OP_ABS:
1345 push_expr(p, oper -> left);
1346 jvm_generate_operator(p, prefix, op);
1347 break;
1349 case OP_ADD:
1350 case OP_SUB:
1351 case OP_MUL:
1352 case OP_DIV:
1353 case OP_MOD:
1354 case OP_BITWISE_AND:
1355 case OP_BITWISE_XOR:
1356 case OP_BITWISE_OR:
1358 case OP_EQ:
1359 case OP_NEQ:
1360 case OP_LSS:
1361 case OP_LEQ:
1362 case OP_GRT:
1363 case OP_GEQ:
1364 push_expr(p, oper -> left);
1365 push_expr(p, oper -> right);
1366 jvm_generate_operator(p, prefix, op);
1367 break;
1369 case OP_LOGIC_OR:
1370 jvm_generate_logical_or(p, oper -> left, oper -> right);
1371 break;
1372 case OP_LOGIC_AND:
1373 jvm_generate_logical_and(p, oper -> left, oper -> right);
1374 break;
1375 default:
1376 gen_error("push_oper: unk op %i", op);
1377 break;
1381 static void
1382 push_expr(gen_proc_t * p, oberon_expr_t * expr)
1384 if(expr -> is_item)
1386 push_item(p, (oberon_item_t *) expr);
1388 else
1390 push_operator(p, (oberon_oper_t *) expr);
1394 static void
1395 store_expr(gen_proc_t * p, oberon_expr_t * dst, oberon_expr_t * src)
1397 assert(dst -> is_item);
1398 oberon_item_t * item = (oberon_item_t *) dst;
1400 if(dst -> result -> class == OBERON_TYPE_ARRAY
1401 || src -> result -> class == OBERON_TYPE_ARRAY)
1403 push_expr(p, dst);
1404 push_expr(p, src);
1405 jvm_generate_copy_array(p, dst -> result);
1407 else if(dst -> result -> class == OBERON_TYPE_RECORD
1408 || src -> result -> class == OBERON_TYPE_RECORD)
1410 push_expr(p, dst);
1411 push_expr(p, src);
1412 jvm_generate_copy_record(p, dst -> result);
1414 else switch(item -> mode)
1416 case MODE_VAR:
1417 push_expr(p, src);
1418 jvm_generate_store(p, src -> result, item -> var -> gen_var);
1419 break;
1420 case MODE_INDEX:
1422 char postfix = jvm_get_postfix(src -> result);
1423 int cell_size = jvm_cell_size_for_postfix(postfix);
1424 push_item(p, item -> parent);
1425 push_expr(p, item -> args);
1426 push_expr(p, src);
1427 jvm_generate(p, 1 + 1 + cell_size, 0, "%castore", postfix);
1428 break;
1429 case MODE_FIELD:
1430 push_item(p, item -> parent);
1431 push_expr(p, src);
1432 jvm_generate_store(p, src -> result, item -> var -> gen_var);
1433 break;
1434 default:
1435 gen_error("store_expr: unk mode %i", item -> mode);
1436 break;
1440 void
1441 oberon_generate_assign(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_t * dst)
1443 gen_module_t * m;
1444 gen_proc_t * p;
1446 m = ctx -> mod -> gen_mod;
1447 p = m -> class -> p;
1449 store_expr(p, dst, src);
1452 void
1453 oberon_generate_code(oberon_context_t * ctx)
1455 printf("generate code\n");
1458 void
1459 oberon_generator_dump(oberon_context_t * ctx, char * path)
1461 printf("jit: dump code\n");
1464 void *
1465 oberon_generator_get_procedure(oberon_context_t * ctx, const char * name)
1467 printf("jit: get pointer to procedure %s\n", name);
1468 return NULL;
1471 void *
1472 oberon_generator_get_var(oberon_context_t * ctx, const char * name)
1474 printf("jit: get pointer to var %s\n", name);
1475 return NULL;