DEADSOFTWARE

Добавлена конструкция CASE
[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_comment(gen_proc_t * p, char * format, ...)
134 va_list ptr;
135 va_start(ptr, format);
137 fprintf(p -> class -> fp, " ;;;; ");
138 vfprintf(p -> class -> fp, format, ptr);
139 fprintf(p -> class -> fp, "\n");
141 va_end(ptr);
144 static void
145 jvm_generate_push_int(gen_proc_t * p, int64_t i)
147 if(i == -1)
149 jvm_generate(p, 0, 1, "iconst_m1");
151 else if(i >= 0 && i <= 5)
153 jvm_generate(p, 0, 1, "iconst_%li", i);
155 else if(i >= -128 && i <= 127)
157 jvm_generate(p, 0, 1, "bipush %li", i);
159 else if(i >= -32768 && i <= 32767)
161 jvm_generate(p, 0, 1, "sipush %li", i);
163 else if(i >= -2147483648 && i <= 2147483647)
165 jvm_generate(p, 0, 1, "ldc %li", i);
167 else
169 jvm_generate(p, 0, 2, "ldc2_w %li", i);
173 static void
174 jvm_generate_push_int_size(gen_proc_t * p, int64_t i, int size)
176 int pushed_cell = 1;
178 if(i == -1)
180 jvm_generate(p, 0, 1, "iconst_m1");
182 else if(i >= 0 && i <= 5)
184 jvm_generate(p, 0, 1, "iconst_%li", i);
186 else if(i >= -128 && i <= 127)
188 jvm_generate(p, 0, 1, "bipush %li", i);
190 else if(i >= -32768 && i <= 32767)
192 jvm_generate(p, 0, 1, "sipush %li", i);
194 else if(i >= -2147483648 && i <= 2147483647)
196 jvm_generate(p, 0, 1, "ldc %li", i);
198 else
200 pushed_cell = 2;
201 jvm_generate(p, 0, 2, "ldc2_w %li", i);
204 assert(size <= 8);
205 if(size > 4 && pushed_cell == 1)
207 jvm_generate(p, pushed_cell, 2, "i2l");
209 else if(size <= 4)
211 if(pushed_cell > 1)
213 jvm_generate(p, 2, 1, "l2i");
216 if(size == 2)
218 jvm_generate(p, 1, 1, "i2s");
220 else if(size == 1)
222 jvm_generate(p, 1, 1, "i2b");
227 static void
228 jvm_generate_push_float(gen_proc_t * p, double f, int size)
230 if(size <= 4)
232 if(f == 0.0)
234 jvm_generate(p, 0, 1, "fconst_0");
236 if(f == 1.0)
238 jvm_generate(p, 0, 1, "fconst_1");
240 if(f == 2.0)
242 jvm_generate(p, 0, 1, "fconst_2");
244 else
246 jvm_generate(p, 0, 1, "ldc %lf", f);
249 else
251 if(f == 0.0)
253 jvm_generate(p, 0, 2, "dconst_0");
255 if(f == 1.0)
257 jvm_generate(p, 0, 2, "dconst_1");
259 else
261 jvm_generate(p, 0, 2, "ldc2_w %lf", f);
266 static void
267 jvm_generate_push_string(gen_proc_t * p, char * str, int char_size)
269 assert(char_size == 1);
270 int len = strlen(str);
272 jvm_generate_push_int(p, len + 1);
273 jvm_generate(p, 1, 1, "newarray byte");
275 for(int i = 0; i < len; i++)
277 jvm_generate(p, 1, 2, "dup");
278 jvm_generate_push_int(p, i);
279 jvm_generate_push_int(p, str[i]);
280 jvm_generate(p, 3, 0, "bastore");
284 static gen_var_t *
285 oberon_generator_new_var()
287 gen_var_t * v = GC_MALLOC(sizeof *v);
288 memset(v, 0, sizeof *v);
289 return v;
292 static void
293 jvm_generate_and_init_global_var(gen_var_t * v, struct gen_class * class, char * access, char * name, char * desc)
295 assert(class -> p == NULL);
296 fprintf(class -> fp, ".field %s static %s %s\n\n", access, name, desc);
298 v -> storage = JVM_STORAGE_STATIC;
299 v -> full_name = new_string("%s/%s", class -> full_name, name);
300 v -> desc = new_string(desc);
303 static void
304 jvm_generate_and_init_field(gen_var_t * v, struct gen_class * class, char * access, char * name, char * desc)
306 assert(class -> p == NULL);
307 fprintf(class -> fp, ".field %s %s %s\n\n", access, name, desc);
309 v -> storage = JVM_STORAGE_FIELD;
310 v -> full_name = new_string("%s/%s", class -> full_name, name);
311 v -> desc = new_string(desc);
314 static void
315 jvm_generate_and_init_local_var(gen_var_t * v, gen_proc_t * p, bool wide)
317 v -> storage = JVM_STORAGE_REGISTER;
318 v -> reg = jvm_alloc_register_untyped(p -> rf, wide);
321 static void
322 jvm_generate_and_init_named_local_var(gen_var_t * v, gen_proc_t * p, bool wide, char * name, char * desc)
324 jvm_generate_and_init_local_var(v, p, wide);
325 v -> desc = new_string(desc);
326 jvm_generate(p, 0, 0, ".var %i is %s %s from start to end", v -> reg, name, desc);
329 static int
330 jvm_new_label_id(gen_proc_t * p)
332 int label_id = p -> label_id;
333 p -> label_id += 1;
334 return label_id;
337 static char *
338 jvm_get_label_name(int label_id)
340 return new_string("L%i", label_id);
343 static void
344 jvm_generate_label(gen_proc_t * p, int label_id)
346 jvm_generate(p, 0, 0, "L%i:", label_id);
349 static void
350 jvm_generate_load(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * src)
352 char prefix;
353 int cell_size;
355 cell_size = jvm_cell_size_for_type(src_type);
357 switch(src -> storage)
359 case JVM_STORAGE_REGISTER:
360 prefix = jvm_get_prefix(src_type);
361 jvm_generate(p, 0, cell_size, "%cload %i", prefix, src -> reg);
362 break;
363 case JVM_STORAGE_STATIC:
364 jvm_generate(p, 0, cell_size, "getstatic %s %s", src -> full_name, src -> desc);
365 break;
366 case JVM_STORAGE_FIELD:
367 jvm_generate(p, 1, cell_size, "getfield %s %s", src -> full_name, src -> desc);
368 break;
369 default:
370 gen_error("jvm_generate_load: unknow storage type %i", src -> storage);
371 break;
375 static void
376 jvm_generate_store(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * dst)
378 char prefix;
379 int cell_size;
381 cell_size = jvm_cell_size_for_type(src_type);
383 switch(dst -> storage)
385 case JVM_STORAGE_REGISTER:
386 prefix = jvm_get_prefix(src_type);
387 jvm_generate(p, cell_size, 0, "%cstore %i", prefix, dst -> reg);
388 break;
389 case JVM_STORAGE_STATIC:
390 jvm_generate(p, cell_size, 0, "putstatic %s %s", dst -> full_name, dst -> desc);
391 break;
392 case JVM_STORAGE_FIELD:
393 jvm_generate(p, 1 + cell_size, 0, "putfield %s %s", dst -> full_name, dst -> desc);
394 break;
395 default:
396 gen_error("jvm_generate_store: unknow storage type %i", dst -> storage);
397 break;
401 static void
402 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num);
404 /*
405 * Генерирует код для инициализации массива со статическим базовым типом
406 * ( aref -- )
407 */
409 static void
410 jvm_generate_array_initialization(gen_proc_t * p, oberon_type_t * arr)
412 int dim = 0;
413 oberon_type_t * base = arr;
414 while(base -> class == OBERON_TYPE_ARRAY)
416 dim += 1;
417 base = base -> base;
420 if(base -> class != OBERON_TYPE_RECORD)
422 jvm_generate(p, 1, 0, "pop");
423 return;
426 struct {
427 gen_var_t * index;
428 gen_var_t * length;
429 int start;
430 int end;
431 } loop[dim];
433 gen_var_t * dst = oberon_generator_new_var();
434 jvm_generate_and_init_local_var(dst, p, false);
435 jvm_generate_store(p, arr, dst);
437 /*
438 * Входящие параметры заграблены.
439 * Теперь генерируем эквивалентный код:
440 * int i = 0;
441 * int len = dst.length
442 * while(i < len)
443 * {
444 * ...
445 * {
446 * dst[i, ...] = new record;
447 * }
448 * ...
449 * i += 1;
450 * }
451 * Где "..." такой же код (начало и конец) для следующей размерности.
452 */
454 for(int i = 0; i < dim; i++)
456 loop[i].index = oberon_generator_new_var();
457 loop[i].length = oberon_generator_new_var();
458 loop[i].start = jvm_new_label_id(p);
459 loop[i].end = jvm_new_label_id(p);
460 jvm_generate_and_init_local_var(loop[i].index, p, false);
461 jvm_generate_and_init_local_var(loop[i].length, p, false);
463 /* TODO преределать через jvm_generate_load{{store}} */
465 jvm_generate(p, 0, 1, "iconst_0");
466 jvm_generate(p, 1, 0, "istore %i", loop[i].index -> reg);
468 jvm_generate_load(p, arr, dst);
469 jvm_generate(p, 1, 1, "arraylength");
470 jvm_generate(p, 1, 0, "istore %i", loop[i].length -> reg);
472 /* if(i >= len) goto end; */
473 jvm_generate_label(p, loop[i].start);
474 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
475 jvm_generate(p, 0, 1, "iload %i", loop[i].length -> reg);
476 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
479 jvm_generate_load(p, arr, dst);
480 jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg);
481 for(int i = 1; i < dim; i++)
483 jvm_generate(p, 2, 1, "aaload");
484 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
486 jvm_generate_new(p, base, 0);
487 jvm_generate(p, 3, 0, "aastore");
489 for(int i = dim - 1; i >= 0; i--)
491 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].index -> reg);
492 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
493 jvm_generate_label(p, loop[i].end);
497 static void
498 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num)
500 int dim;
501 char * cname;
502 char * desc;
503 oberon_type_t * base;
505 switch(type -> class)
507 case OBERON_TYPE_INTEGER:
508 case OBERON_TYPE_BOOLEAN:
509 case OBERON_TYPE_PROCEDURE:
510 case OBERON_TYPE_REAL:
511 case OBERON_TYPE_POINTER:
512 gen_error("jvm_generate_new_static: static alocation not allowed");
513 break;
514 case OBERON_TYPE_RECORD:
515 assert(num == 0);
516 cname = jvm_get_class_full_name(type);
517 jvm_generate(p, 0, 1, "new %s", cname);
518 jvm_generate(p, 1, 2, "dup");
519 jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", cname);
520 break;
521 case OBERON_TYPE_ARRAY:
522 dim = 0;
523 base = type;
524 desc = jvm_get_descriptor(type);
525 while(base -> class == OBERON_TYPE_ARRAY)
527 if(num > 0)
529 assert(base -> size == 0);
530 num -= 1;
532 else
534 assert(base -> size > 0);
535 jvm_generate_push_int(p, base -> size);
537 dim += 1;
538 base = base -> base;
541 assert(num == 0);
542 jvm_generate(p, dim, 1, "multianewarray %s %i", desc, dim);
543 jvm_generate(p, 1, 2, "dup");
544 jvm_generate_array_initialization(p, type);
545 break;
546 default:
547 gen_error("jvm_generate_new_static: unk type class %i", type -> class);
548 break;
552 /*
553 * Генерирует код для копирования полей из первой записи во вторую.
554 * ( aref_dst aref_src -- )
555 * dst := src;
556 */
558 static void
559 jvm_generate_copy_record(gen_proc_t * p, oberon_type_t * rec)
561 assert(rec -> class == OBERON_TYPE_RECORD);
562 char * desc = jvm_get_descriptor(rec);
563 char * cname = jvm_get_class_full_name(rec);
564 jvm_generate(p, 1 + 1, 0, "invokestatic %s/$COPY$(%s%s)V", cname, desc, desc);
567 /*
568 * Генерирует кода для копирования массивов.
569 * ( aref_dst aref_src -- )
570 * dst := src;
571 */
573 static void
574 jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
576 int dim = 0;
577 oberon_type_t * base = arr;
578 while(base -> class == OBERON_TYPE_ARRAY)
580 dim += 1;
581 base = base -> base;
584 struct {
585 gen_var_t * index;
586 gen_var_t * length;
587 int start;
588 int end;
589 } loop[dim];
591 gen_var_t * dst = oberon_generator_new_var();
592 gen_var_t * src = oberon_generator_new_var();
593 jvm_generate_and_init_local_var(dst, p, false);
594 jvm_generate_and_init_local_var(src, p, false);
596 jvm_generate_store(p, arr, src);
597 jvm_generate_store(p, arr, dst);
599 /*
600 * Входящие параметры заграблены.
601 * Теперь генерируем эквивалентный код:
602 * int i = 0;
603 * int len = src.length
604 * while(i < len)
605 * {
606 * ...
607 * {
608 * copy from src[i, ...] to dst[i, ...];
609 * }
610 * ...
611 * i += 1;
612 * }
613 * Где "..." такой же код (начало и конец) для следующей размерности.
614 */
616 for(int i = 0; i < dim; i++)
618 loop[i].index = oberon_generator_new_var();
619 loop[i].length = oberon_generator_new_var();
620 loop[i].start = jvm_new_label_id(p);
621 loop[i].end = jvm_new_label_id(p);
622 jvm_generate_and_init_local_var(loop[i].index, p, false);
623 jvm_generate_and_init_local_var(loop[i].length, p, false);
625 /* TODO преределать через jvm_generate_load{{store}} */
627 jvm_generate(p, 0, 1, "iconst_0");
628 jvm_generate(p, 1, 0, "istore %i", loop[i].index -> reg);
630 jvm_generate_load(p, arr, src);
631 jvm_generate(p, 1, 1, "arraylength");
632 jvm_generate(p, 1, 0, "istore %i", loop[i].length -> reg);
634 /* if(i >= len) goto end; */
635 jvm_generate_label(p, loop[i].start);
636 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
637 jvm_generate(p, 0, 1, "iload %i", loop[i].length -> reg);
638 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
641 if(base -> class == OBERON_TYPE_RECORD)
643 /* Получаем записи по индексам ( -- dst src ) */
645 jvm_generate_load(p, arr, dst);
646 for(int i = 0; i < dim; i++)
648 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
649 jvm_generate(p, 2, 1, "aaload");
652 jvm_generate_load(p, arr, src);
653 for(int i = 0; i < dim; i++)
655 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
656 jvm_generate(p, 2, 1, "aaload");
659 /* Копируем записи ( dst src -- ) */
660 jvm_generate_copy_record(p, base);
662 else
664 char postfix = jvm_get_postfix(base);
665 int cell_size = jvm_cell_size_for_postfix(postfix);
667 /* Получаем массивы и индексы ( -- dst i src i ) */
669 jvm_generate_load(p, arr, dst);
670 jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg);
671 for(int i = 1; i < dim; i++)
673 jvm_generate(p, 2, 1, "aaload");
674 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
677 jvm_generate_load(p, arr, src);
678 jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg);
679 for(int i = 1; i < dim; i++)
681 jvm_generate(p, 2, 1, "aaload");
682 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
685 /* Копируем значения ( dst i src i -- ) */
686 jvm_generate(p, 2, cell_size, "%caload", postfix);
687 jvm_generate(p, 2 + cell_size, 0, "%castore", postfix);
690 for(int i = dim - 1; i >= 0; i--)
692 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].index -> reg);
693 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
694 jvm_generate_label(p, loop[i].end);
698 static void
699 jvm_generate_push_procedure_pointer(gen_proc_t * p, oberon_object_t * proc)
701 char * full_name = jvm_get_field_full_name_safe(proc);
702 char * desc = jvm_get_descriptor(proc -> type);
703 jvm_generate(p, 0, 1, "getstatic %s/pointer %s", full_name, desc);
706 // ==========================================
707 // ==========================================
708 // ==========================================
710 static void
711 push_item(gen_proc_t * p, oberon_item_t * item);
713 void
714 oberon_generator_init_context(oberon_context_t * ctx)
716 gen_context_t * gen_context = GC_MALLOC(sizeof *gen_context);
717 memset(gen_context, 0, sizeof *gen_context);
719 ctx -> gen_context = gen_context;
722 void
723 oberon_generator_destroy_context(oberon_context_t * ctx)
727 static void
728 oberon_generate_procedure_class(oberon_type_t * proc)
730 FILE * fp;
731 char * cname;
732 char * signature;
733 struct gen_class * class;
735 cname = jvm_get_class_full_name(proc);
736 class = jvm_create_class(cname);
737 fp = class -> fp;
739 fprintf(fp, ".source SYSTEM\n");
740 fprintf(fp, ".class public abstract %s\n", cname);
741 fprintf(fp, ".super java/lang/Object\n\n");
743 fprintf(fp, ".method <init>()V\n");
744 fprintf(fp, " aload_0\n");
745 fprintf(fp, " invokespecial java/lang/Object/<init>()V\n");
746 fprintf(fp, " return\n");
747 fprintf(fp, ".end method\n\n");
749 signature = jvm_get_procedure_signature(proc);
751 fprintf(fp, ".method public abstract invoke%s\n", signature);
752 fprintf(fp, ".end method\n\n");
754 jvm_destroy_class(class);
757 static void
758 oberon_generate_procedure_pointer_class(oberon_object_t * proc)
760 FILE * fp;
761 char * cname;
762 char * abscname;
763 char * absdesc;
764 char * signature;
765 struct gen_class * class;
767 cname = jvm_get_field_full_name_safe(proc);
768 class = jvm_create_class(cname);
769 abscname = jvm_get_class_full_name(proc -> type);
770 absdesc = jvm_get_descriptor(proc -> type);
771 fp = class -> fp;
773 fprintf(fp, ".source %s\n", proc -> module -> name);
774 fprintf(fp, ".class public %s\n", cname);
775 fprintf(fp, ".super %s\n\n", abscname);
777 fprintf(fp, ".field public static pointer %s\n\n", absdesc);
779 fprintf(fp, ".method <init>()V\n");
780 fprintf(fp, " aload_0\n");
781 fprintf(fp, " invokespecial %s/<init>()V\n", abscname);
782 fprintf(fp, " return\n");
783 fprintf(fp, ".end method\n\n");
785 fprintf(fp, ".method static <clinit>()V\n");
786 fprintf(fp, " .limit stack 2\n");
787 fprintf(fp, " new %s\n", cname);
788 fprintf(fp, " dup\n");
789 fprintf(fp, " invokespecial %s/<init>()V\n", cname);
790 fprintf(fp, " putstatic %s/pointer %s\n", cname, absdesc);
791 fprintf(fp, " return\n");
792 fprintf(fp, ".end method\n\n");
794 signature = jvm_get_procedure_signature(proc -> type);
796 gen_proc_t * p;
797 gen_var_t * this_v;
799 p = jvm_create_proc(class);
800 jvm_generate_function_header(p, "public", "invoke", signature);
802 this_v = oberon_generator_new_var();
803 jvm_generate_and_init_local_var(this_v, p, false);
805 gen_var_t * v;
806 int use_size = 0;
807 int num = proc -> type -> num_decl;
808 oberon_object_t * arg = proc -> type -> decl;
809 for(int i = 0; i < num; i++)
811 v = oberon_generator_new_var();
812 bool wide = jvm_is_wide_type(arg -> type);
813 char * desc = jvm_get_descriptor(arg -> type);
814 jvm_generate_and_init_named_local_var(v, p, wide, arg -> name, desc);
816 jvm_generate_load(p, arg -> type, v);
818 use_size += (wide) ? (2) : (1);
820 arg = arg -> next;
823 char * full_name = jvm_get_field_full_name(proc);
824 int cell_size = jvm_cell_size_for_type(proc -> type -> base);
826 jvm_generate(p, use_size, cell_size, "invokestatic %s%s", full_name, signature);
828 if(proc -> type -> base -> class == OBERON_TYPE_VOID)
830 jvm_generate(p, 0, 0, "return");
832 else
834 char prefix = jvm_get_prefix(proc -> type -> base);
835 jvm_generate(p, cell_size, 0, "%creturn", prefix);
838 jvm_generate_function_end(p);
840 jvm_destroy_class(class);
843 static void
844 oberon_generate_record_class(gen_module_t * m, oberon_type_t * rec)
846 char * cname;
847 struct gen_class * class;
849 /* Устанавливаем новоый id */
850 rec -> gen_type -> rec_id = m -> rec_id;
851 m -> rec_id += 1;
853 cname = jvm_get_class_full_name(rec);
854 class = jvm_create_class(cname);
856 fprintf(class -> fp, ".source %s\n", rec -> module -> name);
857 fprintf(class -> fp, ".class public %s\n", cname);
859 if(rec -> base == NULL)
861 fprintf(class -> fp, ".super java/lang/Object\n\n");
863 else
865 class -> base = rec -> base -> gen_type -> class;
866 fprintf(class -> fp, ".super %s\n\n", class -> base -> full_name);
869 rec -> gen_type -> class = class;
872 void
873 oberon_generator_init_type(oberon_context_t * ctx, oberon_type_t * type)
875 gen_type_t * t = GC_MALLOC(sizeof *t);
876 memset(t, 0, sizeof *t);
877 type -> gen_type = t;
879 switch(type -> class)
881 case OBERON_TYPE_VOID:
882 case OBERON_TYPE_INTEGER:
883 case OBERON_TYPE_BOOLEAN:
884 case OBERON_TYPE_ARRAY:
885 case OBERON_TYPE_REAL:
886 case OBERON_TYPE_CHAR:
887 case OBERON_TYPE_STRING:
888 case OBERON_TYPE_SET:
889 break;
890 case OBERON_TYPE_RECORD:
892 gen_module_t * m;
893 m = type -> module -> gen_mod;
894 oberon_generate_record_class(m, type);
895 break;
896 case OBERON_TYPE_PROCEDURE:
897 oberon_generate_procedure_class(type);
898 break;
899 case OBERON_TYPE_POINTER:
900 break;
901 default:
902 gen_error("oberon_generator_init_type: unk calss %i", type -> class);
903 break;
907 void
908 oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
910 struct gen_class * class;
911 class = rec -> gen_type -> class;
913 int num = rec -> num_decl;
914 oberon_object_t * field = rec -> decl;
915 for(int i = 0; i < num; i++)
917 gen_var_t * v = field -> gen_var;
918 char * name = field -> name;
919 char * desc = jvm_get_descriptor(field -> type);
920 jvm_generate_and_init_field(v, class, "public", name, desc);
921 field = field -> next;
924 /* Стандартный конструктор класса */
925 /* Инициализирует внутренние статические записи и массивы */
926 gen_proc_t * p = jvm_create_proc(class);
927 jvm_generate_function_header(p, "public", "<init>", "()V");
928 jvm_alloc_register_untyped(p -> rf, false);
929 jvm_generate(p, 0, 1, "aload_0");
930 if(class -> base)
932 jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", class -> base -> full_name);
934 else
936 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
938 num = rec -> num_decl;
939 field = rec -> decl;
940 for(int i = 0; i < num; i++)
942 if(field -> type -> class == OBERON_TYPE_RECORD
943 || field -> type -> class == OBERON_TYPE_ARRAY)
945 jvm_generate(p, 0, 1, "aload_0");
946 jvm_generate_new(p, field -> type, 0);
947 jvm_generate_store(p, field -> type, field -> gen_var);
949 field = field -> next;
951 jvm_generate(p, 0, 0, "return");
952 jvm_generate_function_end(p);
954 /* Метод для копирования полей класса */
955 /* reg0 == src -> reg1 == dst */
956 p = jvm_create_proc(class);
957 gen_var_t * copy_dst = oberon_generator_new_var();
958 gen_var_t * copy_src = oberon_generator_new_var();
959 char * desc = jvm_get_descriptor(rec);
960 char * signature = new_string("(%s%s)V", desc, desc);
961 jvm_generate_function_header(p, "public static", "$COPY$", signature);
962 jvm_generate_and_init_named_local_var(copy_dst, p, false, "dst", desc);
963 jvm_generate_and_init_named_local_var(copy_src, p, false, "src", desc);
964 num = rec -> num_decl;
965 field = rec -> decl;
966 for(int i = 0; i < num; i++)
968 if(field -> type -> class == OBERON_TYPE_RECORD)
970 jvm_generate_load(p, rec, copy_dst);
971 jvm_generate_load(p, field -> type, field -> gen_var);
972 jvm_generate_load(p, rec, copy_src);
973 jvm_generate_load(p, field -> type, field -> gen_var);
974 jvm_generate_copy_record(p, field -> type);
976 else if(field -> type -> class == OBERON_TYPE_ARRAY)
978 jvm_generate_load(p, rec, copy_dst);
979 jvm_generate_load(p, field -> type, field -> gen_var);
980 jvm_generate_load(p, rec, copy_src);
981 jvm_generate_load(p, field -> type, field -> gen_var);
982 jvm_generate_copy_array(p, field -> type);
984 else
986 jvm_generate_load(p, rec, copy_dst);
987 jvm_generate_load(p, rec, copy_src);
988 jvm_generate_load(p, field -> type, field -> gen_var);
989 jvm_generate_store(p, field -> type, field -> gen_var);
991 field = field -> next;
993 jvm_generate(p, 0, 0, "return");
994 jvm_generate_function_end(p);
996 jvm_destroy_class(class);
999 void
1000 oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var)
1002 gen_var_t * v = oberon_generator_new_var();
1003 var -> gen_var = v;
1005 gen_module_t * m;
1006 m = ctx -> mod -> gen_mod;
1008 struct gen_class * class;
1009 class = m -> class;
1011 char * desc;
1012 switch(var -> class)
1014 case OBERON_CLASS_VAR_PARAM:
1015 gen_error("generator: VAR-parameters not implemented");
1016 break;
1017 case OBERON_CLASS_PARAM:
1018 /* Заполняется при генерации функции */
1019 /* смотри jvm_init_local_object() */
1020 break;
1021 case OBERON_CLASS_FIELD:
1022 /* Заполняются при инициализации структуры */
1023 /* смотри oberon_generator_init_record() */
1024 break;
1025 case OBERON_CLASS_VAR:
1026 /* Локальные заполняются при генерации функции */
1027 /* смотри jvm_init_local_object() */
1028 if(var -> local == 0)
1030 desc = jvm_get_descriptor(var -> type);
1031 jvm_generate_and_init_global_var(v, class, "public", var -> name, desc);
1033 break;
1034 default:
1035 gen_error("oberon_generator_init_var: unk var class %i", var -> class);
1036 break;
1040 void
1041 oberon_generator_init_temp_var(oberon_context_t * ctx, oberon_object_t * var)
1043 assert(var -> class == OBERON_CLASS_VAR);
1045 gen_var_t * v = oberon_generator_new_var();
1046 var -> gen_var = v;
1048 gen_module_t * m;
1049 m = ctx -> mod -> gen_mod;
1051 gen_proc_t * p;
1052 p = m -> class -> p;
1054 bool wide;
1055 wide = jvm_is_wide_type(var -> type);
1057 jvm_generate_and_init_local_var(v, p, wide);
1060 void
1061 oberon_generator_init_proc(oberon_context_t * ctx, oberon_object_t * proc)
1063 struct gen_class * class = proc -> module -> gen_mod -> class;
1064 proc -> gen_proc = jvm_create_proc(class);
1066 if(proc -> local)
1068 gen_error("generator: local procedures not implemented");
1071 oberon_generate_procedure_pointer_class(proc);
1074 void
1075 oberon_generator_init_module(oberon_context_t * ctx, oberon_module_t * mod)
1077 gen_module_t * m = GC_MALLOC(sizeof *m);
1078 memset(m, 0, sizeof *m);
1079 mod -> gen_mod = m;
1081 struct gen_class * class;
1082 class = jvm_create_class(mod -> name);
1084 fprintf(class -> fp, ".source %s\n", mod -> name);
1085 fprintf(class -> fp, ".class %s\n", mod -> name);
1086 fprintf(class -> fp, ".super java/lang/Object\n\n");
1088 m -> class = class;
1090 ctx -> gen_context -> current_m = m;
1093 void
1094 oberon_generator_fini_module(oberon_module_t * mod)
1096 jvm_destroy_class(mod -> gen_mod -> class);
1099 static void
1100 push_expr(gen_proc_t * p, oberon_expr_t * expr);
1102 void
1103 oberon_generate_begin_module(oberon_context_t * ctx)
1105 struct gen_class * class = ctx -> mod -> gen_mod -> class;
1106 gen_proc_t * p = jvm_create_proc(class);
1108 jvm_generate_function_header(p, "public", "<init>", "()V");
1109 jvm_alloc_register_untyped(p -> rf, false);
1110 jvm_generate(p, 0, 1, "aload_0");
1111 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
1113 /* Инициализация переменных объявленных в модуле */
1114 oberon_object_t * x = ctx -> mod -> decl -> list -> next;
1115 while(x != NULL)
1117 if(x -> class == OBERON_CLASS_VAR)
1119 if(x -> type -> class == OBERON_TYPE_ARRAY
1120 || x -> type -> class == OBERON_TYPE_RECORD)
1122 jvm_generate_new(p, x -> type, 0);
1123 jvm_generate_store(p, x -> type, x -> gen_var);
1127 x = x -> next;
1131 void
1132 oberon_generate_end_module(oberon_context_t * ctx)
1134 struct gen_class * class = ctx -> mod -> gen_mod -> class;
1135 gen_proc_t * p = class -> p;
1137 jvm_generate(p, 0, 0, "return");
1138 jvm_generate_function_end(class -> p);
1141 static void
1142 jvm_init_local_object(gen_proc_t * p, oberon_object_t * x)
1144 gen_var_t * v;
1145 bool wide;
1146 char * desc;
1147 struct gen_class * class;
1149 v = x -> gen_var;
1150 class = p -> class;
1151 wide = jvm_is_wide_type(x -> type);
1152 desc = jvm_get_descriptor(x -> type);
1154 // Убеждаемся что сейчас находимся в функции
1155 assert(class -> p);
1156 assert(x -> local);
1158 switch(x -> class)
1160 case OBERON_CLASS_VAR:
1161 case OBERON_CLASS_PARAM:
1162 jvm_generate_and_init_named_local_var(v, p, wide, x -> name, desc);
1163 break;
1164 default:
1165 gen_error("jvm_init_local_object: wat");
1166 break;
1170 /*
1171 * Генерирует код для получения размера измерения массива
1172 * Аналог Обероновского LEN(v, n); где n = 0 - первое измерение.
1173 * ( aref -- int )
1174 */
1176 static void
1177 jvm_generate_array_len(gen_proc_t * p, int dim)
1179 while(dim > 0)
1181 jvm_generate(p, 0, 1, "iconst_0");
1182 jvm_generate(p, 2, 1, "aaload");
1183 dim -= 1;
1185 jvm_generate(p, 1, 1, "arraylength");
1188 static void
1189 jvm_generate_array_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * arr)
1191 int dim = 0;
1192 oberon_type_t * base = arr;
1193 while(base -> class == OBERON_TYPE_ARRAY)
1195 if(base -> size == 0)
1197 jvm_generate_load(p, arr, v);
1198 jvm_generate_array_len(p, dim);
1199 dim += 1;
1201 base = base -> base;
1204 jvm_generate_new(p, arr, dim);
1205 jvm_generate(p, 1, 2, "dup");
1206 jvm_generate_load(p, arr, v);
1207 jvm_generate_copy_array(p, arr);
1208 jvm_generate_store(p, arr, v);
1211 static void
1212 jvm_generate_record_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * rec)
1214 jvm_generate_new(p, rec, 0);
1215 jvm_generate(p, 1, 2, "dup");
1216 jvm_generate_load(p, rec, v);
1217 jvm_generate_copy_record(p, rec);
1218 jvm_generate_store(p, rec, v);
1221 void
1222 oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc)
1224 gen_proc_t * p;
1225 char * signature;
1227 p = proc -> gen_proc;
1229 signature = jvm_get_procedure_signature(proc -> type);
1230 jvm_generate_function_header(p, "public static", proc -> name, signature);
1232 /* Выделение регистров под параметры и переменные */
1233 oberon_object_t * var = proc -> type -> decl;
1234 while(var)
1236 jvm_init_local_object(p, var);
1237 var = var -> next;
1240 /* Копирование статических/открытых массивов и записей */
1241 var = proc -> type -> decl;
1242 while(var)
1244 if(var -> class == OBERON_CLASS_PARAM || var -> class == OBERON_CLASS_VAR_PARAM)
1246 if(var -> type -> class == OBERON_TYPE_ARRAY)
1248 jvm_generate_array_duplicate_and_replace(p, var -> gen_var, var -> type);
1250 else if(var -> type -> class == OBERON_TYPE_RECORD)
1251 {
1252 jvm_generate_record_duplicate_and_replace(p, var -> gen_var, var -> type);
1255 var = var -> next;
1256 }
1259 void
1260 oberon_generate_end_proc(oberon_context_t * ctx)
1262 gen_module_t * m;
1263 gen_proc_t * p;
1265 m = ctx -> mod -> gen_mod;
1266 p = m -> class -> p;
1268 jvm_generate_function_end(p);
1271 gen_label_t *
1272 oberon_generator_reserve_label(oberon_context_t * ctx)
1274 gen_module_t * m;
1275 gen_proc_t * p;
1276 gen_label_t * l;
1278 m = ctx -> mod -> gen_mod;
1279 p = m -> class -> p;
1281 l = GC_MALLOC(sizeof *l);
1282 memset(l, 0, sizeof *l);
1284 l -> id = jvm_new_label_id(p);
1285 return l;
1288 void
1289 oberon_generate_label(oberon_context_t * ctx, gen_label_t * l)
1291 gen_module_t * m;
1292 gen_proc_t * p;
1294 m = ctx -> mod -> gen_mod;
1295 p = m -> class -> p;
1297 jvm_generate_label(p, l -> id);
1300 void
1301 oberon_generate_goto(oberon_context_t * ctx, gen_label_t * l)
1303 gen_module_t * m;
1304 gen_proc_t * p;
1306 m = ctx -> mod -> gen_mod;
1307 p = m -> class -> p;
1309 jvm_generate(p, 0, 0, "goto L%i", l -> id);
1312 void
1313 oberon_generate_branch(oberon_context_t * ctx, oberon_expr_t * cond, bool gotoif, gen_label_t * l)
1315 gen_module_t * m;
1316 gen_proc_t * p;
1318 m = ctx -> mod -> gen_mod;
1319 p = m -> class -> p;
1321 push_expr(p, cond);
1323 if(gotoif == false)
1325 /* переход если false */
1326 jvm_generate(p, 1, 0, "ifeq L%i", l -> id);
1328 else
1330 /* переход если true */
1331 jvm_generate(p, 1, 0, "ifne L%i", l -> id);
1335 static void
1336 jvm_generate_call_proc(gen_proc_t * p, oberon_item_t * desig)
1338 assert(desig -> var == NULL);
1339 assert(desig -> mode == MODE_CALL);
1341 char * signature = NULL;
1343 bool direct_call = false;
1344 if(desig -> parent -> mode == MODE_VAR)
1346 if(desig -> parent -> var -> class == OBERON_CLASS_PROC)
1348 direct_call = true;
1352 printf("direct_call == %i\n", direct_call);
1354 oberon_type_t * procsig;
1355 procsig = desig -> parent -> result;
1356 signature = jvm_get_procedure_signature(procsig);
1358 if(direct_call == false)
1360 /* Загружаем указатель на процедуру */
1361 push_item(p, desig -> parent);
1364 int args_cells = 0;
1365 int result_cells = jvm_cell_size_for_type(procsig -> base);
1367 int num = desig -> num_args;
1368 oberon_expr_t * arg = desig -> args;
1369 for(int i = 0; i < num; i++)
1371 args_cells += jvm_cell_size_for_type(arg -> result);
1372 push_expr(p, arg);
1373 arg = arg -> next;
1376 if(direct_call)
1378 char * full_name = jvm_get_field_full_name(desig -> parent -> var);
1379 jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", full_name, signature);
1381 else
1383 char * cname = jvm_get_class_full_name(procsig);
1384 jvm_generate(p, 1 + args_cells, result_cells, "invokevirtual %s/invoke%s", cname, signature);
1388 void
1389 oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig)
1391 assert(desig -> is_item);
1392 jvm_generate_call_proc(ctx -> mod -> gen_mod -> class -> p, (oberon_item_t *) desig);
1395 void
1396 oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr)
1398 gen_module_t * m;
1399 gen_proc_t * p;
1400 char prefix;
1401 int cell_size;
1403 m = ctx -> mod -> gen_mod;
1404 p = m -> class -> p;
1406 if(expr)
1408 push_expr(p, expr);
1409 prefix = jvm_get_prefix(expr -> result);
1410 cell_size = jvm_cell_size_for_type(expr -> result);
1411 jvm_generate(p, cell_size, 0, "%creturn", prefix);
1413 else
1415 jvm_generate(p, 0, 0, "return");
1419 static void
1420 jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
1422 assert(type -> class == OBERON_TYPE_POINTER);
1424 for(int i = 0; i < num; i++)
1426 push_expr(p, arg);
1427 arg = arg -> next;
1430 jvm_generate_new(p, type -> base, num);
1433 static void
1434 jvm_generate_cast_type(gen_proc_t * p, oberon_type_t * from, oberon_type_t * to)
1436 if(to -> class == OBERON_TYPE_RECORD || to -> class == OBERON_TYPE_POINTER)
1438 char * full_name = jvm_get_class_full_name(to);
1439 jvm_generate(p, 1, 1, "checkcast %s", full_name);
1441 else
1443 char prefix = jvm_get_prefix(from);
1444 char postfix = jvm_get_postfix(to);
1445 int from_cell_size = jvm_cell_size_for_type(from);
1446 int to_cell_size = jvm_cell_size_for_type(to);
1448 jvm_generate_comment(p, "cast type class from %i(%i) to %i(%i)", from -> class, from -> size, to -> class, to -> size);
1450 if(prefix == postfix)
1452 return;
1455 jvm_generate(p, from_cell_size, to_cell_size, "%c2%c", prefix, postfix);
1459 static void
1460 push_item(gen_proc_t * p, oberon_item_t * item)
1462 switch(item -> mode)
1464 case MODE_VAR:
1465 if(item -> var -> class == OBERON_CLASS_PROC)
1467 jvm_generate_push_procedure_pointer(p, item -> var);
1469 else
1471 jvm_generate_load(p, item -> result, item -> var -> gen_var);
1473 break;
1474 case MODE_INTEGER:
1475 case MODE_CHAR:
1476 case MODE_SET:
1477 jvm_generate_push_int_size(p, item -> integer, item -> result -> size);
1478 break;
1479 case MODE_BOOLEAN:
1480 jvm_generate_push_int_size(p, item -> boolean, item -> result -> size);
1481 break;
1482 case MODE_CALL:
1483 jvm_generate_call_proc(p, item);
1484 break;
1485 case MODE_INDEX:
1487 char postfix = jvm_get_postfix(item -> result);
1488 int cell_size = jvm_cell_size_for_postfix(postfix);
1489 assert(item -> parent -> is_item);
1490 push_item(p, (oberon_item_t *) item -> parent);
1491 push_expr(p, item -> args);
1492 jvm_generate(p, 1 + 1, cell_size, "%caload", postfix);
1493 break;
1494 case MODE_FIELD:
1495 assert(item -> parent -> is_item);
1496 push_item(p, (oberon_item_t *) item -> parent);
1497 jvm_generate_load(p, item -> result, item -> var -> gen_var);
1498 break;
1499 case MODE_DEREF:
1500 /* Все объекты в jvm представляются как указатели */
1501 assert(item -> parent -> is_item);
1502 push_item(p, (oberon_item_t *) item -> parent);
1503 break;
1504 case MODE_NIL:
1505 jvm_generate(p, 0, 1, "aconst_null");
1506 break;
1507 case MODE_NEW:
1508 jvm_generate_expr_new_pointer(p, item -> result, item -> num_args, item -> args);
1509 break;
1510 case MODE_REAL:
1511 jvm_generate_push_float(p, item -> real, item -> result -> size);
1512 break;
1513 case MODE_STRING:
1514 jvm_generate_push_string(p, item -> string, item -> result -> size);
1515 break;
1516 default:
1517 gen_error("push_item: unk mode %i", item -> mode);
1518 break;
1522 static void
1523 jvm_generate_logical_not(gen_proc_t * p)
1525 int label_done = jvm_new_label_id(p);
1526 int label_false = jvm_new_label_id(p);
1527 char * label_name_done = jvm_get_label_name(label_done);
1528 char * label_name_false = jvm_get_label_name(label_false);
1530 jvm_generate(p, 1, 0, "ifne %s", label_name_false);
1531 jvm_generate(p, 0, 1, "iconst_1");
1532 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1533 jvm_generate_label(p, label_false);
1534 jvm_generate(p, 0, 1, "iconst_0");
1535 jvm_generate_label(p, label_done);
1538 static void
1539 jvm_generate_abs(gen_proc_t * p, char prefix)
1541 char t = jvm_get_type_of_prefix(prefix);
1542 int cell_size = jvm_cell_size_for_postfix(prefix);
1543 jvm_generate(p, cell_size, cell_size, "invokestatic java/lang/Math/abs(%c)%c", t, t);
1546 static char *
1547 jvm_get_compare_postfix(int op)
1549 char * cmpop = "";
1550 switch(op)
1552 case OP_EQ:
1553 cmpop = "eq";
1554 break;
1555 case OP_NEQ:
1556 cmpop = "ne";
1557 break;
1558 case OP_LSS:
1559 cmpop = "lt";
1560 break;
1561 case OP_LEQ:
1562 cmpop = "le";
1563 break;
1564 case OP_GRT:
1565 cmpop = "gt";
1566 break;
1567 case OP_GEQ:
1568 cmpop = "ge";
1569 break;
1570 default:
1571 gen_error("jvm_generate_compare_op: wat");
1572 break;
1574 return cmpop;
1577 static void
1578 jvm_generate_compare_op(gen_proc_t * p, oberon_type_t * t, int op)
1580 char prefix = jvm_get_prefix(t);
1581 int label_true = jvm_new_label_id(p);
1582 int label_done = jvm_new_label_id(p);
1583 int cell_size = jvm_cell_size_for_type(t);
1584 char * cmpop = jvm_get_compare_postfix(op);
1586 if(prefix == 'l')
1588 jvm_generate(p, 2 * cell_size, 1, "lcmp");
1589 jvm_generate(p, 1, 1, "if%s L%i", cmpop, label_true);
1591 else if(prefix == 'f' || prefix == 'd')
1593 char fop;
1594 if(op == OP_EQ || op == OP_NEQ || op == OP_GRT || op == OP_GEQ)
1596 fop = 'l';
1598 else
1600 fop = 'g';
1602 jvm_generate(p, 2 * cell_size, 1, "%ccmp%c", prefix, fop);
1603 jvm_generate(p, 1, 1, "if%s L%i", cmpop, label_true);
1605 else
1607 jvm_generate(p, 2 * cell_size, 0, "if_%ccmp%s L%i", prefix, cmpop, label_true);
1610 jvm_generate(p, 0, 1, "iconst_0");
1611 jvm_generate(p, 0, 0, "goto L%i", label_done);
1612 jvm_generate_label(p, label_true);
1613 jvm_generate(p, 0, 1, "iconst_1");
1614 jvm_generate_label(p, label_done);
1617 static void
1618 jvm_generate_operator(gen_proc_t * p, oberon_type_t * t, int op)
1620 char prefix = jvm_get_prefix(t);
1621 int cell_size = jvm_cell_size_for_type(t);
1622 switch(op)
1624 case OP_UNARY_MINUS:
1625 jvm_generate(p, cell_size, cell_size, "%cneg", prefix);
1626 break;
1627 case OP_COMPLEMENTATION:
1628 jvm_generate_push_int_size(p, -1, t -> size);
1629 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1630 break;
1631 case OP_LOGIC_NOT:
1632 jvm_generate_logical_not(p);
1633 break;
1634 case OP_ABS:
1635 jvm_generate_abs(p, prefix);
1636 break;
1638 case OP_ADD:
1639 jvm_generate(p, 2 * cell_size, cell_size, "%cadd", prefix);
1640 break;
1641 case OP_SUB:
1642 jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix);
1643 break;
1644 case OP_MUL:
1645 jvm_generate(p, 2 * cell_size, cell_size, "%cmul", prefix);
1646 break;
1647 case OP_DIV:
1648 jvm_generate(p, 2 * cell_size, cell_size, "%cdiv", prefix);
1649 break;
1650 case OP_MOD:
1651 jvm_generate(p, 2 * cell_size, cell_size, "%crem", prefix);
1652 break;
1653 case OP_UNION:
1654 jvm_generate(p, 2 * cell_size, cell_size, "%cor", prefix);
1655 break;
1656 case OP_INTERSECTION:
1657 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1658 break;
1659 case OP_DIFFERENCE:
1660 jvm_generate_push_int_size(p, -1, t -> size);
1661 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1662 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1663 break;
1664 case OP_SYM_DIFFERENCE:
1665 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1666 break;
1668 case OP_EQ:
1669 case OP_NEQ:
1670 case OP_LSS:
1671 case OP_LEQ:
1672 case OP_GRT:
1673 case OP_GEQ:
1674 jvm_generate_compare_op(p, t, op);
1675 break;
1676 default:
1677 gen_error("jvm_generate_operator: unk op %i", op);
1678 break;
1679 }
1682 static void
1683 jvm_generate_logical_or(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1685 int label_calc_b = jvm_new_label_id(p);
1686 int label_done = jvm_new_label_id(p);
1687 char * label_name_calc_b = jvm_get_label_name(label_calc_b);
1688 char * label_name_done = jvm_get_label_name(label_done);
1690 /* a OR b -- если a, то TRUE, иначе b */
1692 push_expr(p, a);
1693 jvm_generate(p, 1, 0, "ifeq %s", label_name_calc_b);
1694 jvm_generate(p, 0, 1, "iconst_1");
1695 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1696 jvm_generate_label(p, label_calc_b);
1697 push_expr(p, b);
1698 jvm_generate_label(p, label_done);
1701 static void
1702 jvm_generate_logical_and(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1704 int label_false = jvm_new_label_id(p);
1705 int label_done = jvm_new_label_id(p);
1706 char * label_name_false = jvm_get_label_name(label_false);
1707 char * label_name_done = jvm_get_label_name(label_done);
1709 /* a AND b -- если a, то b, иначе FALSE */
1711 push_expr(p, a);
1712 jvm_generate(p, 1, 0, "ifeq %s", label_name_false);
1713 push_expr(p, b);
1714 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1715 jvm_generate_label(p, label_false);
1716 jvm_generate(p, 0, 1, "iconst_0");
1717 jvm_generate_label(p, label_done);
1720 static void
1721 jvm_generate_range(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1723 /* { a } == 1 << a */
1724 /* { a..b } == (a <= b) ? ((2 << b) - (1 << a)) : (0); */
1726 char prefix;
1727 int cell_size;
1728 oberon_type_t * t;
1729 gen_var_t * ra;
1730 gen_var_t * rb;
1731 int label_else;
1732 int label_end;
1733 bool wide;
1735 t = a -> result;
1736 cell_size = jvm_cell_size_for_type(t);
1737 prefix = jvm_get_prefix(t);
1739 if(b == NULL)
1741 jvm_generate_push_int_size(p, 1, t -> size);
1742 push_expr(p, a);
1743 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1745 else
1747 wide = jvm_is_wide_type(t);
1748 ra = oberon_generator_new_var();
1749 rb = oberon_generator_new_var();
1750 jvm_generate_and_init_local_var(ra, p, wide);
1751 jvm_generate_and_init_local_var(rb, p, wide);
1752 label_else = jvm_new_label_id(p);
1753 label_end = jvm_new_label_id(p);
1755 push_expr(p, a);
1756 jvm_generate_store(p, t, ra);
1757 push_expr(p, b);
1758 jvm_generate_store(p, t, rb);
1760 jvm_generate_load(p, t, ra);
1761 jvm_generate_load(p, t, rb);
1762 jvm_generate(p, 2 * cell_size, 0, "if_%ccmpgt L%i", prefix, label_else);
1764 jvm_generate_push_int_size(p, 2, t -> size);
1765 jvm_generate_load(p, t, rb);
1766 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1767 jvm_generate_push_int_size(p, 2, t -> size);
1768 jvm_generate_load(p, t, ra);
1769 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1770 jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix);
1771 jvm_generate(p, 0, 0, "goto L%i", label_end);
1773 jvm_generate_label(p, label_else);
1774 jvm_generate_push_int_size(p, 0, t -> size);
1775 jvm_generate_label(p, label_end);
1778 /* TODO free registers */
1781 static void
1782 jvm_generate_in(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1784 oberon_type_t * t = a -> result;
1785 int cell_size = jvm_cell_size_for_type(t);
1786 char prefix = jvm_get_prefix(t);
1787 int label_else = jvm_new_label_id(p);
1788 int label_end = jvm_new_label_id(p);
1790 jvm_generate_push_int_size(p, 1, t -> size);
1791 push_expr(p, a);
1792 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1793 push_expr(p, b);
1794 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1796 if(cell_size > 1)
1798 jvm_generate(p, cell_size, 0, "lcmp");
1801 jvm_generate(p, 1, 0, "ifeq L%i", label_else);
1802 jvm_generate(p, 0, 1, "iconst_1");
1803 jvm_generate(p, 0, 0, "goto L%i", label_end);
1804 jvm_generate_label(p, label_else);
1805 jvm_generate(p, 0, 1, "iconst_0");
1806 jvm_generate_label(p, label_end);
1809 static void
1810 push_operator(gen_proc_t * p, oberon_oper_t * oper)
1812 oberon_type_t * preq = oper -> left -> result;
1813 int op = oper -> op;
1814 switch(op)
1816 case OP_CAST:
1817 push_expr(p, oper -> left);
1818 jvm_generate_cast_type(p, oper -> left -> result, oper -> result);
1819 break;
1820 case OP_COMPLEMENTATION:
1821 case OP_UNARY_MINUS:
1822 case OP_LOGIC_NOT:
1823 case OP_ABS:
1824 push_expr(p, oper -> left);
1825 jvm_generate_operator(p, preq, op);
1826 break;
1828 case OP_ADD:
1829 case OP_SUB:
1830 case OP_MUL:
1831 case OP_DIV:
1832 case OP_MOD:
1834 case OP_UNION:
1835 case OP_INTERSECTION:
1836 case OP_DIFFERENCE:
1837 case OP_SYM_DIFFERENCE:
1839 case OP_EQ:
1840 case OP_NEQ:
1841 case OP_LSS:
1842 case OP_LEQ:
1843 case OP_GRT:
1844 case OP_GEQ:
1845 push_expr(p, oper -> left);
1846 push_expr(p, oper -> right);
1847 jvm_generate_operator(p, preq, op);
1848 break;
1850 case OP_LOGIC_OR:
1851 jvm_generate_logical_or(p, oper -> left, oper -> right);
1852 break;
1853 case OP_LOGIC_AND:
1854 jvm_generate_logical_and(p, oper -> left, oper -> right);
1855 break;
1857 case OP_IS:
1858 preq = oper -> right -> result;
1859 char * cname = jvm_get_class_full_name(preq);
1860 push_expr(p, oper -> left);
1861 jvm_generate(p, 1, 1, "instanceof %s", cname);
1862 break;
1863 case OP_RANGE:
1864 jvm_generate_range(p, oper -> left, oper -> right);
1865 break;
1866 case OP_IN:
1867 jvm_generate_in(p, oper -> left, oper -> right);
1868 break;
1869 default:
1870 gen_error("push_oper: unk op %i", op);
1871 break;
1875 static void
1876 push_expr(gen_proc_t * p, oberon_expr_t * expr)
1878 if(expr -> is_item)
1880 push_item(p, (oberon_item_t *) expr);
1882 else
1884 push_operator(p, (oberon_oper_t *) expr);
1888 static void
1889 store_expr(gen_proc_t * p, oberon_expr_t * dst, oberon_expr_t * src)
1891 assert(dst -> is_item);
1892 oberon_item_t * item = (oberon_item_t *) dst;
1894 if(dst -> result -> class == OBERON_TYPE_ARRAY
1895 || src -> result -> class == OBERON_TYPE_ARRAY)
1897 push_expr(p, dst);
1898 push_expr(p, src);
1899 jvm_generate_copy_array(p, dst -> result);
1901 else if(dst -> result -> class == OBERON_TYPE_RECORD
1902 || src -> result -> class == OBERON_TYPE_RECORD)
1904 push_expr(p, dst);
1905 push_expr(p, src);
1906 jvm_generate_copy_record(p, dst -> result);
1908 else switch(item -> mode)
1910 case MODE_VAR:
1911 push_expr(p, src);
1912 jvm_generate_store(p, src -> result, item -> var -> gen_var);
1913 break;
1914 case MODE_INDEX:
1916 char postfix = jvm_get_postfix(src -> result);
1917 int cell_size = jvm_cell_size_for_postfix(postfix);
1918 assert(item -> parent -> is_item);
1919 push_item(p, (oberon_item_t *) item -> parent);
1920 push_expr(p, item -> args);
1921 push_expr(p, src);
1922 jvm_generate(p, 1 + 1 + cell_size, 0, "%castore", postfix);
1923 break;
1924 case MODE_FIELD:
1925 assert(item -> parent -> is_item);
1926 push_item(p, (oberon_item_t *) item -> parent);
1927 push_expr(p, src);
1928 jvm_generate_store(p, src -> result, item -> var -> gen_var);
1929 break;
1930 default:
1931 gen_error("store_expr: unk mode %i", item -> mode);
1932 break;
1936 void
1937 oberon_generate_assign(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_t * dst)
1939 gen_module_t * m;
1940 gen_proc_t * p;
1942 m = ctx -> mod -> gen_mod;
1943 p = m -> class -> p;
1945 store_expr(p, dst, src);
1948 void
1949 oberon_generate_code(oberon_context_t * ctx)
1951 printf("generate code\n");
1954 void
1955 oberon_generator_dump(oberon_context_t * ctx, char * path)
1957 printf("jit: dump code\n");
1960 void *
1961 oberon_generator_get_procedure(oberon_context_t * ctx, const char * name)
1963 printf("jit: get pointer to procedure %s\n", name);
1964 return NULL;
1967 void *
1968 oberon_generator_get_var(oberon_context_t * ctx, const char * name)
1970 printf("jit: get pointer to var %s\n", name);
1971 return NULL;