DEADSOFTWARE

bf3b07b52aee0c728cf194e2fbb521b635500557
[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 "../../oberon-internals.h"
12 #include "../../generator.h"
13 #include "generator-jvm.h"
14 #include "generator-jvm-abi.h"
15 #include "generator-jvm-asm.h"
16 #include "generator-jvm-basic.h"
18 static void
19 push_item(gen_proc_t * p, oberon_item_t * item);
21 static void
22 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num);
24 /*
25 * Генерирует код для инициализации массива со статическим базовым типом
26 * ( aref -- )
27 */
29 static void
30 jvm_generate_array_initialization(gen_proc_t * p, oberon_type_t * arr)
31 {
32 int dim = 0;
33 oberon_type_t * base = arr;
34 while(base -> class == OBERON_TYPE_ARRAY)
35 {
36 dim += 1;
37 base = base -> base;
38 }
40 if(base -> class != OBERON_TYPE_RECORD)
41 {
42 jvm_generate(p, 1, 0, "pop");
43 return;
44 }
46 struct {
47 int reg_index;
48 int reg_length;
49 int start;
50 int end;
51 } loop[dim];
53 int reg_dst;
54 reg_dst = jvm_alloc_register_untyped(p -> rf, false);
55 jvm_generate(p, 1, 0, "astore %i", reg_dst);
57 /*
58 * Входящие параметры заграблены.
59 * Теперь генерируем эквивалентный код:
60 * int i = 0;
61 * int len = dst.length
62 * while(i < len)
63 * {
64 * ...
65 * {
66 * dst[i, ...] = new record;
67 * }
68 * ...
69 * i += 1;
70 * }
71 * Где "..." такой же код (начало и конец) для следующей размерности.
72 */
74 for(int i = 0; i < dim; i++)
75 {
76 loop[i].reg_index = jvm_alloc_register_untyped(p -> rf, false);
77 loop[i].reg_length = jvm_alloc_register_untyped(p -> rf, false);
78 loop[i].start = jvm_new_label_id(p);
79 loop[i].end = jvm_new_label_id(p);
81 jvm_generate(p, 0, 1, "iconst_0");
82 jvm_generate(p, 1, 0, "istore %i", loop[i].reg_index);
84 jvm_generate(p, 0, 1, "aload %i", reg_dst);
85 jvm_generate(p, 1, 1, "arraylength");
86 jvm_generate(p, 1, 0, "istore %i", loop[i].reg_length);
88 /* if(i >= len) goto end; */
89 jvm_generate_label(p, loop[i].start);
90 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
91 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_length);
92 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
93 }
95 jvm_generate(p, 0, 1, "aload %i", reg_dst);
96 jvm_generate(p, 0, 1, "iload %i", loop[0].reg_index);
97 for(int i = 1; i < dim; i++)
98 {
99 jvm_generate(p, 2, 1, "aaload");
100 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
102 jvm_generate_new(p, base, 0);
103 jvm_generate(p, 3, 0, "aastore");
105 for(int i = dim - 1; i >= 0; i--)
107 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].reg_index);
108 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
109 jvm_generate_label(p, loop[i].end);
113 static void
114 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num)
116 int dim;
117 char * cname;
118 char * desc;
119 oberon_type_t * base;
121 switch(type -> class)
123 case OBERON_TYPE_INTEGER:
124 case OBERON_TYPE_BOOLEAN:
125 case OBERON_TYPE_PROCEDURE:
126 case OBERON_TYPE_REAL:
127 case OBERON_TYPE_POINTER:
128 gen_error("jvm_generate_new_static: static alocation not allowed");
129 break;
130 case OBERON_TYPE_RECORD:
131 assert(num == 0);
132 cname = jvm_get_class_full_name(type);
133 jvm_generate(p, 0, 1, "new %s", cname);
134 jvm_generate(p, 1, 2, "dup");
135 jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", cname);
136 break;
137 case OBERON_TYPE_ARRAY:
138 dim = 0;
139 base = type;
140 desc = jvm_get_descriptor(type);
141 while(base -> class == OBERON_TYPE_ARRAY)
143 if(num > 0)
145 assert(base -> size == 0);
146 num -= 1;
148 else
150 assert(base -> size > 0);
151 jvm_generate_push_int(p, base -> size);
153 dim += 1;
154 base = base -> base;
157 assert(num == 0);
158 jvm_generate(p, dim, 1, "multianewarray %s %i", desc, dim);
159 jvm_generate(p, 1, 2, "dup");
160 jvm_generate_array_initialization(p, type);
161 break;
162 default:
163 gen_error("jvm_generate_new_static: unk type class %i", type -> class);
164 break;
168 /*
169 * Генерирует код для копирования полей из первой записи во вторую.
170 * ( aref_dst aref_src -- )
171 * dst := src;
172 */
174 static void
175 jvm_generate_copy_record(gen_proc_t * p, oberon_type_t * rec)
177 assert(rec -> class == OBERON_TYPE_RECORD);
178 char * desc = jvm_get_descriptor(rec);
179 char * cname = jvm_get_class_full_name(rec);
180 jvm_generate(p, 1 + 1, 0, "invokestatic %s/$COPY$(%s%s)V", cname, desc, desc);
183 /*
184 * Генерирует кода для копирования массивов.
185 * ( aref_dst aref_src -- )
186 * dst := src;
187 */
189 static void
190 jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
192 int dim = 0;
193 oberon_type_t * base = arr;
194 while(base -> class == OBERON_TYPE_ARRAY)
196 dim += 1;
197 base = base -> base;
200 struct {
201 int reg_index;
202 int reg_length;
203 int start;
204 int end;
205 } loop[dim];
207 int reg_dst = jvm_alloc_register_untyped(p -> rf, false);
208 int reg_src = jvm_alloc_register_untyped(p -> rf, false);
209 jvm_generate(p, 1, 0, "astore %i", reg_src);
210 jvm_generate(p, 1, 0, "astore %i", reg_dst);
212 /*
213 * Входящие параметры заграблены.
214 * Теперь генерируем эквивалентный код:
215 * int i = 0;
216 * int len = src.length
217 * while(i < len)
218 * {
219 * ...
220 * {
221 * copy from src[i, ...] to dst[i, ...];
222 * }
223 * ...
224 * i += 1;
225 * }
226 * Где "..." такой же код (начало и конец) для следующей размерности.
227 */
229 for(int i = 0; i < dim; i++)
231 loop[i].reg_index = jvm_alloc_register_untyped(p -> rf, false);
232 loop[i].reg_length = jvm_alloc_register_untyped(p -> rf, false);
233 loop[i].start = jvm_new_label_id(p);
234 loop[i].end = jvm_new_label_id(p);
236 jvm_generate(p, 0, 1, "iconst_0");
237 jvm_generate(p, 1, 0, "istore %i", loop[i].reg_index);
239 jvm_generate(p, 0, 1, "aload %i", reg_src);
240 jvm_generate(p, 1, 1, "arraylength");
241 jvm_generate(p, 1, 0, "istore %i", loop[i].reg_length);
243 /* if(i >= len) goto end; */
244 jvm_generate_label(p, loop[i].start);
245 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
246 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_length);
247 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
250 if(base -> class == OBERON_TYPE_RECORD)
252 /* Получаем записи по индексам ( -- dst src ) */
254 jvm_generate(p, 0, 1, "aload %i", reg_dst);
255 for(int i = 0; i < dim; i++)
257 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
258 jvm_generate(p, 2, 1, "aaload");
261 jvm_generate(p, 0, 1, "aload %i", reg_src);
262 for(int i = 0; i < dim; i++)
264 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
265 jvm_generate(p, 2, 1, "aaload");
268 /* Копируем записи ( dst src -- ) */
269 jvm_generate_copy_record(p, base);
271 else
273 char postfix = jvm_get_postfix(base);
274 int cell_size = jvm_cell_size_for_postfix(postfix);
276 /* Получаем массивы и индексы ( -- dst i src i ) */
278 jvm_generate(p, 0, 1, "aload %i", reg_dst);
279 jvm_generate(p, 0, 1, "iload %i", loop[0].reg_index);
280 for(int i = 1; i < dim; i++)
282 jvm_generate(p, 2, 1, "aaload");
283 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
286 jvm_generate(p, 0, 1, "aload %i", reg_src);
287 jvm_generate(p, 0, 1, "iload %i", loop[0].reg_index);
288 for(int i = 1; i < dim; i++)
290 jvm_generate(p, 2, 1, "aaload");
291 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
294 /* Копируем значения ( dst i src i -- ) */
295 jvm_generate(p, 2, cell_size, "%caload", postfix);
296 jvm_generate(p, 2 + cell_size, 0, "%castore", postfix);
299 for(int i = dim - 1; i >= 0; i--)
301 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].reg_index);
302 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
303 jvm_generate_label(p, loop[i].end);
307 static void
308 jvm_generate_push_procedure_pointer(gen_proc_t * p, oberon_object_t * proc)
310 char * full_name = jvm_get_field_full_name_safe(proc);
311 char * desc = jvm_get_descriptor(proc -> type);
312 jvm_generate(p, 0, 1, "getstatic %s/pointer %s", full_name, desc);
315 void
316 oberon_generator_init_context(oberon_context_t * ctx)
318 gen_context_t * gen_context = GC_MALLOC(sizeof *gen_context);
319 memset(gen_context, 0, sizeof *gen_context);
321 ctx -> gen_context = gen_context;
322 gen_context -> dir = ".";
325 void
326 oberon_generator_destroy_context(oberon_context_t * ctx)
330 static void
331 oberon_generate_procedure_class(const char * dir, oberon_type_t * proc)
333 FILE * fp;
334 char * cname;
335 char * signature;
336 struct gen_class * class;
338 cname = jvm_get_class_full_name(proc);
339 class = jvm_create_class(dir, cname);
340 fp = class -> fp;
342 fprintf(fp, ".source SYSTEM\n");
343 fprintf(fp, ".class public abstract %s\n", cname);
344 fprintf(fp, ".super java/lang/Object\n\n");
346 fprintf(fp, ".method <init>()V\n");
347 fprintf(fp, " aload_0\n");
348 fprintf(fp, " invokespecial java/lang/Object/<init>()V\n");
349 fprintf(fp, " return\n");
350 fprintf(fp, ".end method\n\n");
352 signature = jvm_get_procedure_signature(proc);
354 fprintf(fp, ".method public abstract invoke%s\n", signature);
355 fprintf(fp, ".end method\n\n");
357 jvm_destroy_class(class);
360 static void
361 oberon_generate_procedure_pointer_class(const char * dir, oberon_object_t * proc)
363 FILE * fp;
364 char * cname;
365 char * abscname;
366 char * absdesc;
367 char * signature;
368 struct gen_class * class;
370 cname = jvm_get_field_full_name_safe(proc);
371 class = jvm_create_class(dir, cname);
372 abscname = jvm_get_class_full_name(proc -> type);
373 absdesc = jvm_get_descriptor(proc -> type);
374 fp = class -> fp;
376 fprintf(fp, ".source %s\n", proc -> module -> name);
377 fprintf(fp, ".class public %s\n", cname);
378 fprintf(fp, ".super %s\n\n", abscname);
380 fprintf(fp, ".field public static pointer %s\n\n", absdesc);
382 fprintf(fp, ".method private <init>()V\n");
383 fprintf(fp, " aload_0\n");
384 fprintf(fp, " invokespecial %s/<init>()V\n", abscname);
385 fprintf(fp, " return\n");
386 fprintf(fp, ".end method\n\n");
388 fprintf(fp, ".method static <clinit>()V\n");
389 fprintf(fp, " .limit stack 2\n");
390 fprintf(fp, " new %s\n", cname);
391 fprintf(fp, " dup\n");
392 fprintf(fp, " invokespecial %s/<init>()V\n", cname);
393 fprintf(fp, " putstatic %s/pointer %s\n", cname, absdesc);
394 fprintf(fp, " return\n");
395 fprintf(fp, ".end method\n\n");
397 signature = jvm_get_procedure_signature(proc -> type);
399 gen_proc_t * p = jvm_create_proc(class);
400 jvm_generate_function_header(p, "public", "invoke", signature);
401 jvm_alloc_register_untyped(p -> rf, false);
403 int use_size = 0;
404 int num = proc -> type -> num_decl;
405 oberon_object_t * param = proc -> type -> decl;
406 for(int i = 0; i < num; i++)
408 gen_type_t * t = param -> type -> gen_type;
409 if(param -> class == OBERON_CLASS_VAR_PARAM)
411 int reg = jvm_alloc_register_untyped(p -> rf, true);
412 jvm_generate(p, 0, 1, "aload %i", reg);
413 jvm_generate(p, 0, 1, "iload %i", reg + 1);
415 else
417 int reg = jvm_alloc_register_untyped(p -> rf, t -> wide);
418 jvm_generate(p, 0, t -> cell_size, "%cload %i", t -> prefix, reg);
420 use_size += t -> cell_size;
421 param = param -> next;
424 char * full_name = jvm_get_field_full_name(proc);
425 int cell_size = jvm_cell_size_for_type(proc -> type -> base);
426 jvm_generate(p, use_size, cell_size, "invokestatic %s%s", full_name, signature);
428 if(proc -> type -> base -> class == OBERON_TYPE_NOTYPE)
430 jvm_generate(p, 0, 0, "return");
432 else
434 char prefix = jvm_get_prefix(proc -> type -> base);
435 jvm_generate(p, cell_size, 0, "%creturn", prefix);
438 jvm_generate_function_end(p);
440 jvm_destroy_class(class);
443 static void
444 oberon_generate_record_class(const char * dir, gen_module_t * m, oberon_type_t * rec)
446 char * cname;
447 struct gen_class * class;
449 /* Устанавливаем новоый id */
450 rec -> gen_type -> rec_id = m -> rec_id;
451 m -> rec_id += 1;
453 cname = jvm_get_class_full_name(rec);
454 class = jvm_create_class(dir, cname);
456 fprintf(class -> fp, ".source %s\n", rec -> module -> name);
457 fprintf(class -> fp, ".class public %s\n", cname);
459 if(rec -> base == NULL)
461 fprintf(class -> fp, ".super java/lang/Object\n\n");
463 else
465 class -> base = rec -> base -> gen_type -> class;
466 fprintf(class -> fp, ".super %s\n\n", class -> base -> full_name);
469 rec -> gen_type -> class = class;
472 void
473 oberon_generator_init_type(oberon_context_t * ctx, oberon_type_t * type)
475 gen_type_t * t = GC_MALLOC(sizeof *t);
476 memset(t, 0, sizeof *t);
477 type -> gen_type = t;
479 gen_context_t * c;
480 c = ctx -> gen_context;
482 gen_module_t * m;
483 switch(type -> class)
485 case OBERON_TYPE_NOTYPE:
486 case OBERON_TYPE_INTEGER:
487 case OBERON_TYPE_BOOLEAN:
488 case OBERON_TYPE_ARRAY:
489 case OBERON_TYPE_POINTER:
490 case OBERON_TYPE_REAL:
491 case OBERON_TYPE_CHAR:
492 case OBERON_TYPE_STRING:
493 case OBERON_TYPE_SET:
494 case OBERON_TYPE_NIL:
495 case OBERON_TYPE_SYSTEM_BYTE:
496 case OBERON_TYPE_SYSTEM_PTR:
497 break;
498 case OBERON_TYPE_RECORD:
499 m = type -> module -> gen_mod;
500 oberon_generate_record_class(c -> dir, m, type);
501 break;
502 case OBERON_TYPE_PROCEDURE:
503 oberon_generate_procedure_class(c -> dir, type);
504 break;
505 default:
506 gen_error("oberon_generator_init_type: unk calss %i", type -> class);
507 break;
510 if(type -> class != OBERON_TYPE_NOTYPE)
512 t -> wide = jvm_is_wide_type(type);
513 t -> prefix = jvm_get_prefix(type);
514 t -> postfix = jvm_get_postfix(type);
517 t -> cell_size = jvm_cell_size_for_type(type);
519 if(type -> class != OBERON_TYPE_NIL)
521 t -> desc = jvm_get_descriptor(type);
525 void
526 oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
528 struct gen_class * class;
529 class = rec -> gen_type -> class;
531 int num = rec -> num_decl;
532 oberon_object_t * field = rec -> decl;
533 for(int i = 0; i < num; i++)
535 jvm_generate_var(field -> gen_var);
536 field = field -> next;
539 /* Стандартный конструктор класса */
540 /* Инициализирует внутренние статические записи и массивы */
541 gen_proc_t * p = jvm_create_proc(class);
542 jvm_generate_function_header(p, "public", "<init>", "()V");
543 jvm_alloc_register_untyped(p -> rf, false);
544 jvm_generate(p, 0, 1, "aload_0");
545 if(class -> base)
547 jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", class -> base -> full_name);
549 else
551 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
553 num = rec -> num_decl;
554 field = rec -> decl;
555 for(int i = 0; i < num; i++)
557 jvm_generate(p, 0, 1, "aload_0");
558 jvm_generate_var_initialization(p, field -> gen_var);
560 if(field -> type -> class == OBERON_TYPE_RECORD
561 || field -> type -> class == OBERON_TYPE_ARRAY)
563 jvm_generate(p, 0, 1, "aload_0");
564 jvm_generate_ldst_prepare(p, field -> gen_var);
565 jvm_generate_new(p, field -> type, 0);
566 jvm_generate_store(p, field -> gen_var);
568 field = field -> next;
570 jvm_generate(p, 0, 0, "return");
571 jvm_generate_function_end(p);
573 /* Метод для копирования полей класса */
574 /* reg0 == src -> reg1 == dst */
575 p = jvm_create_proc(class);
576 char * signature = new_string("(%s%s)V", rec -> gen_type -> desc, rec -> gen_type -> desc);
577 jvm_generate_function_header(p, "public static", "$COPY$", signature);
579 gen_var_t * copy_dst = jvm_create_function_var(p, JVM_STORAGE_REGISTER, "dst", rec -> gen_type);
580 gen_var_t * copy_src = jvm_create_function_var(p, JVM_STORAGE_REGISTER, "src", rec -> gen_type);
581 jvm_generate_var(copy_dst);
582 jvm_generate_var(copy_src);
583 jvm_generate_var_initialization(p, copy_dst);
584 jvm_generate_var_initialization(p, copy_src);
586 num = rec -> num_decl;
587 field = rec -> decl;
588 for(int i = 0; i < num; i++)
590 if(field -> type -> class == OBERON_TYPE_RECORD)
592 jvm_generate_load(p, copy_dst);
593 jvm_generate_load(p, field -> gen_var);
594 jvm_generate_load(p, copy_src);
595 jvm_generate_load(p, field -> gen_var);
596 jvm_generate_copy_record(p, field -> type);
598 else if(field -> type -> class == OBERON_TYPE_ARRAY)
600 jvm_generate_load(p, copy_dst);
601 jvm_generate_load(p, field -> gen_var);
602 jvm_generate_load(p, copy_src);
603 jvm_generate_load(p, field -> gen_var);
604 jvm_generate_copy_array(p, field -> type);
606 else
608 jvm_generate_load(p, copy_dst);
609 jvm_generate_ldst_prepare(p, field -> gen_var);
610 jvm_generate_load(p, copy_src);
611 jvm_generate_load(p, field -> gen_var);
612 jvm_generate_store(p, field -> gen_var);
614 field = field -> next;
616 jvm_generate(p, 0, 0, "return");
617 jvm_generate_function_end(p);
619 jvm_destroy_class(class);
622 void
623 oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var)
625 gen_module_t * m;
626 m = ctx -> mod -> gen_mod;
628 struct gen_class * class;
629 class = m -> class;
631 gen_proc_t * p;
632 struct gen_class * c;
633 char * name = var -> name;
634 gen_type_t * t = var -> type -> gen_type;
636 assert(name);
637 if(t == NULL)
639 gen_error("uninitialized type class %i", var -> type -> class);
642 switch(var -> class)
644 case OBERON_CLASS_VAR_PARAM:
645 p = var -> parent -> gen_proc;
646 var -> gen_var = jvm_create_function_var(p, JVM_STORAGE_FRAME_PARAM_VARPTR, name, t);
647 break;
648 case OBERON_CLASS_PARAM:
649 p = var -> parent -> gen_proc;
650 var -> gen_var = jvm_create_function_var(p, JVM_STORAGE_FRAME_PARAM_VAR, name, t);
651 break;
652 case OBERON_CLASS_FIELD:
653 c = var -> parent_type -> gen_type -> class;
654 var -> gen_var = jvm_create_class_var(c, JVM_STORAGE_FIELD_VAR, name, t);
655 break;
656 case OBERON_CLASS_VAR:
657 if(var -> local)
659 p = var -> parent -> gen_proc;
660 var -> gen_var = jvm_create_function_var(p, JVM_STORAGE_FRAME_VAR, name, t);
662 else
664 var -> gen_var = jvm_create_class_var(class, JVM_STORAGE_STATIC_VAR, name, t);
665 jvm_generate_var(var -> gen_var);
667 break;
668 default:
669 gen_error("oberon_generator_init_var: unk var class %i", var -> class);
670 break;
674 void
675 oberon_generator_init_temp_var(oberon_context_t * ctx, oberon_object_t * var)
677 assert(var -> class == OBERON_CLASS_VAR);
679 gen_module_t * m;
680 m = ctx -> mod -> gen_mod;
682 gen_proc_t * p;
683 p = m -> class -> p;
685 gen_type_t * t;
686 t = var -> type -> gen_type;
688 var -> gen_var = jvm_create_function_temp_var(p, t);
689 jvm_generate_var(var -> gen_var);
690 jvm_generate_var_initialization(p, var -> gen_var);
693 void
694 oberon_generator_init_proc(oberon_context_t * ctx, oberon_object_t * proc)
696 struct gen_class * class = proc -> module -> gen_mod -> class;
697 proc -> gen_proc = jvm_create_proc(class);
698 proc -> gen_proc -> full_name = jvm_get_field_full_name_safe(proc);
699 jvm_create_static_links(proc);
702 void
703 oberon_generator_init_module(oberon_context_t * ctx, oberon_module_t * mod)
705 gen_module_t * m = GC_MALLOC(sizeof *m);
706 memset(m, 0, sizeof *m);
707 mod -> gen_mod = m;
709 gen_context_t * c;
710 c = ctx -> gen_context;
712 struct gen_class * class;
713 class = jvm_create_class(c -> dir, mod -> name);
715 fprintf(class -> fp, ".source %s\n", mod -> name);
716 fprintf(class -> fp, ".class %s\n", mod -> name);
717 fprintf(class -> fp, ".super java/lang/Object\n\n");
719 fprintf(class -> fp, ".field private static $INITIALIZED$ Z\n\n");
721 m -> class = class;
723 c -> current_m = m;
726 void
727 oberon_generator_fini_module(oberon_module_t * mod)
729 jvm_destroy_class(mod -> gen_mod -> class);
732 static void
733 push_expr(gen_proc_t * p, oberon_expr_t * expr);
735 static void
736 oberon_generate_static_initialization(gen_proc_t * p, oberon_object_t * x)
738 if(x -> type -> class == OBERON_TYPE_ARRAY
739 || x -> type -> class == OBERON_TYPE_RECORD)
741 jvm_generate_ldst_prepare(p, x -> gen_var);
742 jvm_generate_new(p, x -> type, 0);
743 jvm_generate_store(p, x -> gen_var);
747 void
748 oberon_generate_begin_module(oberon_context_t * ctx)
750 gen_proc_t * p;
751 int label_cont;
752 struct gen_class * class;
754 class = ctx -> mod -> gen_mod -> class;
756 p = jvm_create_proc(class);
757 jvm_generate_function_header(p, "public static", "BEGIN", "()V");
759 label_cont = jvm_new_label_id(p);
760 jvm_generate(p, 0, 1, "getstatic %s/$INITIALIZED$ Z", class -> full_name);
761 jvm_generate(p, 1, 0, "ifeq L%i", label_cont);
762 jvm_generate(p, 0, 0, "return");
763 jvm_generate_label(p, label_cont);
765 jvm_generate(p, 0, 1, "iconst_1");
766 jvm_generate(p, 1, 0, "putstatic %s/$INITIALIZED$ Z", class -> full_name);
768 /* Инициализация переменных объявленных в модуле */
769 oberon_object_t * x = ctx -> mod -> decl -> list -> next;
770 while(x != NULL)
772 if(x -> class == OBERON_CLASS_MODULE)
774 if(!x -> module -> intrinsic)
776 jvm_generate(p, 0, 0, "invokestatic %s/BEGIN()V", x -> module -> gen_mod -> class -> full_name);
779 else if(x -> class == OBERON_CLASS_VAR)
781 jvm_generate_var_initialization(p, x -> gen_var);
782 oberon_generate_static_initialization(p, x);
785 x = x -> next;
789 void
790 oberon_generate_end_module(oberon_context_t * ctx)
792 struct gen_class * class = ctx -> mod -> gen_mod -> class;
793 gen_proc_t * p = class -> p;
795 jvm_generate(p, 0, 0, "return");
796 jvm_generate_function_end(class -> p);
798 p = jvm_create_proc(class);
799 jvm_generate_function_header(p, "private", "<init>", "()V");
800 jvm_alloc_register_untyped(p -> rf, false);
801 jvm_generate(p, 0, 1, "aload_0");
802 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
803 jvm_generate(p, 0, 0, "return");
804 jvm_generate_function_end(class -> p);
807 /*
808 * Генерирует код для получения размера измерения массива
809 * Аналог Обероновского LEN(v, n); где n = 0 - первое измерение.
810 * ( aref -- int )
811 */
813 static void
814 jvm_generate_array_len(gen_proc_t * p, int dim)
816 while(dim > 0)
818 jvm_generate(p, 0, 1, "iconst_0");
819 jvm_generate(p, 2, 1, "aaload");
820 dim -= 1;
822 jvm_generate(p, 1, 1, "arraylength");
825 static void
826 jvm_generate_array_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * arr)
828 jvm_generate_ldst_prepare(p, v);
830 int dim = 0;
831 oberon_type_t * base = arr;
832 while(base -> class == OBERON_TYPE_ARRAY)
834 if(base -> size == 0)
836 jvm_generate_load(p, v);
837 jvm_generate_array_len(p, dim);
838 dim += 1;
840 base = base -> base;
843 jvm_generate_new(p, arr, dim);
844 jvm_generate(p, 1, 2, "dup");
845 jvm_generate_load(p, v);
846 jvm_generate_copy_array(p, arr);
847 jvm_generate_store(p, v);
850 static void
851 jvm_generate_record_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * rec)
853 jvm_generate_ldst_prepare(p, v);
854 jvm_generate_new(p, rec, 0);
855 jvm_generate(p, 1, 2, "dup");
856 jvm_generate_load(p, v);
857 jvm_generate_copy_record(p, rec);
858 jvm_generate_store(p, v);
861 static void
862 jvm_generate_local_object(gen_proc_t * p, oberon_object_t * x)
864 gen_var_t * v;
865 struct gen_class * class;
867 v = x -> gen_var;
868 class = p -> class;
870 // Убеждаемся что сейчас находимся в функции
871 assert(class -> p);
872 assert(x -> local);
874 switch(x -> class)
876 case OBERON_CLASS_VAR_PARAM:
877 case OBERON_CLASS_PARAM:
878 case OBERON_CLASS_VAR:
879 jvm_generate_var(v);
880 break;
881 case OBERON_CLASS_CONST:
882 case OBERON_CLASS_TYPE:
883 case OBERON_CLASS_PROC:
884 break;
885 default:
886 gen_error("jvm_generate_local_initialization: wat class %i", x -> class);
887 break;
891 static void
892 jvm_generate_local_initialization(gen_proc_t * p, oberon_object_t * x)
894 gen_var_t * v;
895 struct gen_class * class;
897 v = x -> gen_var;
898 class = p -> class;
900 // Убеждаемся что сейчас находимся в функции
901 assert(class -> p);
902 assert(x -> local);
904 switch(x -> class)
906 case OBERON_CLASS_VAR_PARAM:
907 jvm_generate_var_initialization(p, v);
908 break;
909 case OBERON_CLASS_PARAM:
910 jvm_generate_var_initialization(p, v);
911 if(x -> type -> class == OBERON_TYPE_ARRAY)
913 jvm_generate_array_duplicate_and_replace(p, v, x -> type);
915 else if(x -> type -> class == OBERON_TYPE_RECORD)
916 {
917 jvm_generate_record_duplicate_and_replace(p, v, x -> type);
919 break;
920 case OBERON_CLASS_VAR:
921 jvm_generate_var_initialization(p, v);
922 oberon_generate_static_initialization(p, x);
923 break;
924 case OBERON_CLASS_CONST:
925 case OBERON_CLASS_TYPE:
926 case OBERON_CLASS_PROC:
927 break;
928 default:
929 gen_error("jvm_generate_local_initialization: wat class %i", x -> class);
930 break;
934 void
935 oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc)
937 gen_proc_t * p;
938 char * name;
939 char * signature;
940 oberon_object_t * var;
942 p = proc -> gen_proc;
944 if(proc -> local)
946 signature = jvm_get_local_procedure_signature(proc);
948 else
950 oberon_generate_procedure_pointer_class(p -> class -> dir, proc);
951 signature = jvm_get_procedure_signature(proc -> type);
954 name = jvm_get_name(proc);
955 jvm_generate_function_header(p, "public static", name, signature);
957 if(proc -> local)
959 jvm_generate_staticlinks(proc);
962 jvm_generate_procedure_frame(proc);
964 /* Создание параметров и переменных */
965 var = proc -> scope -> list -> next;
966 while(var)
968 jvm_generate_local_object(p, var);
969 var = var -> next;
972 jvm_generate_frame_initialization(p);
974 /* Инициализация парамеров и переменных */
975 var = proc -> scope -> list -> next;
976 while(var)
978 jvm_generate_local_initialization(p, var);
979 var = var -> next;
983 void
984 oberon_generate_end_proc(oberon_context_t * ctx)
986 gen_module_t * m;
987 gen_proc_t * p;
989 m = ctx -> mod -> gen_mod;
990 p = m -> class -> p;
992 oberon_generate_trap(ctx, -4);
993 jvm_generate_function_end(p);
996 gen_label_t *
997 oberon_generator_reserve_label(oberon_context_t * ctx)
999 gen_module_t * m;
1000 gen_proc_t * p;
1001 gen_label_t * l;
1003 m = ctx -> mod -> gen_mod;
1004 p = m -> class -> p;
1006 l = GC_MALLOC(sizeof *l);
1007 memset(l, 0, sizeof *l);
1009 l -> id = jvm_new_label_id(p);
1010 return l;
1013 void
1014 oberon_generate_label(oberon_context_t * ctx, gen_label_t * l)
1016 gen_module_t * m;
1017 gen_proc_t * p;
1019 m = ctx -> mod -> gen_mod;
1020 p = m -> class -> p;
1022 jvm_generate_label(p, l -> id);
1025 void
1026 oberon_generate_goto(oberon_context_t * ctx, gen_label_t * l)
1028 gen_module_t * m;
1029 gen_proc_t * p;
1031 m = ctx -> mod -> gen_mod;
1032 p = m -> class -> p;
1034 jvm_generate(p, 0, 0, "goto L%i", l -> id);
1037 void
1038 oberon_generate_branch(oberon_context_t * ctx, oberon_expr_t * cond, bool gotoif, gen_label_t * l)
1040 gen_module_t * m;
1041 gen_proc_t * p;
1043 m = ctx -> mod -> gen_mod;
1044 p = m -> class -> p;
1046 push_expr(p, cond);
1048 if(gotoif == false)
1050 /* переход если false */
1051 jvm_generate(p, 1, 0, "ifeq L%i", l -> id);
1053 else
1055 /* переход если true */
1056 jvm_generate(p, 1, 0, "ifne L%i", l -> id);
1060 static void
1061 push_varptr(gen_proc_t * p, oberon_expr_t * expr)
1063 assert(expr -> is_item);
1065 switch(expr -> item.mode)
1067 case MODE_VAR:
1068 jvm_generate_ldst_prepare(p, expr -> item.var -> gen_var);
1069 break;
1070 case MODE_INDEX:
1071 push_item(p, expr -> item.parent);
1072 push_expr(p, expr -> item.args);
1073 break;
1074 case MODE_FIELD:
1075 push_item(p, expr -> item.parent);
1076 jvm_generate_ldst_prepare(p, expr -> item.var -> gen_var);
1077 break;
1078 case MODE_DEREF:
1079 push_varptr(p, (oberon_expr_t *) expr -> item.parent);
1080 break;
1081 default:
1082 gen_error("push_varptr: wat %i", expr -> item.mode);
1083 break;
1087 static void
1088 jvm_generate_call_proc(gen_proc_t * p, oberon_item_t * desig)
1090 assert(desig -> var == NULL);
1091 assert(desig -> mode == MODE_CALL);
1093 bool direct_call = false;
1094 if(desig -> parent -> mode == MODE_VAR)
1096 if(desig -> parent -> var -> class == OBERON_CLASS_PROC)
1098 direct_call = true;
1102 oberon_type_t * procsig;
1103 procsig = desig -> parent -> result;
1105 if(direct_call == false)
1107 /* Загружаем указатель на процедуру */
1108 push_item(p, desig -> parent);
1111 if(direct_call)
1113 jvm_generate_push_static_links(p, desig -> parent -> var);
1116 int args_cells = 0;
1117 int result_cells = jvm_cell_size_for_type(procsig -> base);
1119 int num = desig -> num_args;
1120 oberon_expr_t * arg = desig -> args;
1121 oberon_object_t * param = procsig -> decl;
1122 for(int i = 0; i < num; i++)
1124 if(param -> class == OBERON_CLASS_VAR_PARAM)
1126 args_cells += 2;
1127 push_varptr(p, arg);
1129 else
1131 args_cells += jvm_cell_size_for_type(arg -> result);
1132 push_expr(p, arg);
1134 arg = arg -> next;
1135 param = param -> next;
1138 if(direct_call)
1140 char * full_name = jvm_get_field_full_name(desig -> parent -> var);
1141 char * signature = jvm_get_local_procedure_signature(desig -> parent -> var);
1142 jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", full_name, signature);
1144 else
1146 char * cname = jvm_get_class_full_name(procsig);
1147 char * signature = jvm_get_procedure_signature(procsig);
1148 jvm_generate(p, 1 + args_cells, result_cells, "invokevirtual %s/invoke%s", cname, signature);
1152 void
1153 oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig)
1155 assert(desig -> is_item);
1156 jvm_generate_call_proc(ctx -> mod -> gen_mod -> class -> p, (oberon_item_t *) desig);
1159 void
1160 oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr)
1162 gen_module_t * m;
1163 gen_proc_t * p;
1164 char prefix;
1165 int cell_size;
1167 m = ctx -> mod -> gen_mod;
1168 p = m -> class -> p;
1170 if(expr)
1172 push_expr(p, expr);
1173 prefix = jvm_get_prefix(expr -> result);
1174 cell_size = jvm_cell_size_for_type(expr -> result);
1175 jvm_generate(p, cell_size, 0, "%creturn", prefix);
1177 else
1179 jvm_generate(p, 0, 0, "return");
1183 static void
1184 jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
1186 assert(type -> class == OBERON_TYPE_POINTER);
1188 for(int i = 0; i < num; i++)
1190 push_expr(p, arg);
1191 arg = arg -> next;
1194 jvm_generate_new(p, type -> base, num);
1197 static void
1198 jvm_generate_cast_prefix(gen_proc_t * p, char prefix, char postfix)
1200 if(prefix == postfix)
1202 return;
1205 int from_cell_size = jvm_cell_size_for_postfix(prefix);
1206 int to_cell_size = jvm_cell_size_for_postfix(postfix);
1207 jvm_generate(p, from_cell_size, to_cell_size, "%c2%c", prefix, postfix);
1210 static void
1211 jvm_generate_cast_type(gen_proc_t * p, oberon_type_t * from, oberon_type_t * to)
1213 if(to -> class == OBERON_TYPE_RECORD || to -> class == OBERON_TYPE_POINTER)
1215 if(to -> class == OBERON_TYPE_POINTER && to -> base -> class == OBERON_TYPE_RECORD)
1217 char * full_name = jvm_get_class_full_name(to);
1218 jvm_generate(p, 1, 1, "checkcast %s", full_name);
1221 else
1223 char prefix = jvm_get_prefix(from);
1224 char postfix = jvm_get_postfix(to);
1225 jvm_generate_cast_prefix(p, prefix, postfix);
1229 static void
1230 jvm_generate_hard_cast_type(gen_proc_t * p, oberon_type_t * from, oberon_type_t * to)
1232 if(from -> class == OBERON_TYPE_REAL
1233 && (to -> class == OBERON_TYPE_INTEGER || to -> class == OBERON_TYPE_SYSTEM_BYTE))
1235 char postfix = jvm_get_postfix(to);
1236 if(from -> size <= 4)
1238 jvm_generate(p, 1, 1, "invokestatic java/lang/Float/floatToIntBits(F)I");
1239 jvm_generate_cast_prefix(p, 'i', postfix);
1241 else
1243 jvm_generate(p, 2, 2, "invokestatic java/lang/Double/doubleToIntBits(D)J");
1244 jvm_generate_cast_prefix(p, 'l', postfix);
1247 else if((from -> class == OBERON_TYPE_INTEGER || from -> class == OBERON_TYPE_SYSTEM_BYTE)
1248 && to -> class == OBERON_TYPE_REAL)
1250 char prefix = jvm_get_prefix(from);
1251 if(to -> size <= 4)
1253 jvm_generate_cast_prefix(p, prefix, 'i');
1254 jvm_generate(p, 1, 1, "invokestatic java/lang/Float/intBitsToFloat(I)F");
1256 else
1258 jvm_generate_cast_prefix(p, prefix, 'l');
1259 jvm_generate(p, 2, 2, "invokestatic java/lang/Double/longBitsToDouble(J)D");
1262 else
1264 jvm_generate_cast_type(p, from, to);
1268 static void
1269 push_item(gen_proc_t * p, oberon_item_t * item)
1271 switch(item -> mode)
1273 case MODE_VAR:
1274 if(item -> var -> class == OBERON_CLASS_PROC)
1276 jvm_generate_push_procedure_pointer(p, item -> var);
1278 else
1280 jvm_generate_load(p, item -> var -> gen_var);
1282 break;
1283 case MODE_INTEGER:
1284 case MODE_BOOLEAN:
1285 case MODE_CHAR:
1286 case MODE_SET:
1287 jvm_generate_push_int_size(p, item -> integer, item -> result -> size);
1288 break;
1289 case MODE_CALL:
1290 jvm_generate_call_proc(p, item);
1291 break;
1292 case MODE_INDEX:
1294 char postfix = jvm_get_postfix(item -> result);
1295 int cell_size = jvm_cell_size_for_postfix(postfix);
1296 push_item(p, item -> parent);
1297 push_expr(p, item -> args);
1298 jvm_generate(p, 1 + 1, cell_size, "%caload", postfix);
1299 break;
1300 case MODE_FIELD:
1301 push_item(p, item -> parent);
1302 jvm_generate_load(p, item -> var -> gen_var);
1303 break;
1304 case MODE_DEREF:
1305 /* Все объекты в jvm представляются как указатели */
1306 push_item(p, item -> parent);
1307 break;
1308 case MODE_NIL:
1309 jvm_generate(p, 0, 1, "aconst_null");
1310 break;
1311 case MODE_NEW:
1312 jvm_generate_expr_new_pointer(p, item -> result, item -> num_args, item -> args);
1313 break;
1314 case MODE_REAL:
1315 jvm_generate_push_float(p, item -> real, item -> result -> size);
1316 break;
1317 case MODE_STRING:
1318 jvm_generate_push_string(p, item -> string, item -> result -> size);
1319 break;
1320 case MODE_LEN:
1321 push_item(p, item -> parent);
1322 jvm_generate_array_len(p, item -> integer);
1323 jvm_generate_cast_prefix(p, 'i', jvm_get_postfix(item -> result));
1324 break;
1325 default:
1326 gen_error("push_item: unk mode %i", item -> mode);
1327 break;
1331 static void
1332 jvm_generate_logical_not(gen_proc_t * p)
1334 int label_done = jvm_new_label_id(p);
1335 int label_false = jvm_new_label_id(p);
1337 jvm_generate(p, 1, 0, "ifne L%i", label_false);
1338 jvm_generate(p, 0, 1, "iconst_1");
1339 jvm_generate(p, 0, 0, "goto L%i", label_done);
1340 jvm_generate_label(p, label_false);
1341 jvm_generate(p, 0, 1, "iconst_0");
1342 jvm_generate_label(p, label_done);
1345 static void
1346 jvm_generate_abs(gen_proc_t * p, char prefix)
1348 char t = jvm_get_type_of_prefix(prefix);
1349 int cell_size = jvm_cell_size_for_postfix(prefix);
1350 jvm_generate(p, cell_size, cell_size, "invokestatic java/lang/Math/abs(%c)%c", t, t);
1353 static char *
1354 jvm_get_compare_postfix(int op)
1356 char * cmpop = "";
1357 switch(op)
1359 case OP_EQ:
1360 cmpop = "eq";
1361 break;
1362 case OP_NEQ:
1363 cmpop = "ne";
1364 break;
1365 case OP_LSS:
1366 cmpop = "lt";
1367 break;
1368 case OP_LEQ:
1369 cmpop = "le";
1370 break;
1371 case OP_GRT:
1372 cmpop = "gt";
1373 break;
1374 case OP_GEQ:
1375 cmpop = "ge";
1376 break;
1377 default:
1378 gen_error("jvm_generate_compare_op: wat");
1379 break;
1381 return cmpop;
1384 static void
1385 jvm_generate_compare_op(gen_proc_t * p, oberon_type_t * t, int op)
1387 char prefix = jvm_get_prefix(t);
1388 int label_true = jvm_new_label_id(p);
1389 int label_done = jvm_new_label_id(p);
1390 int cell_size = jvm_cell_size_for_type(t);
1391 char * cmpop = jvm_get_compare_postfix(op);
1393 if(prefix == 'l')
1395 jvm_generate(p, 2 * cell_size, 1, "lcmp");
1396 jvm_generate(p, 1, 1, "if%s L%i", cmpop, label_true);
1398 else if(prefix == 'f' || prefix == 'd')
1400 char fop;
1401 if(op == OP_EQ || op == OP_NEQ || op == OP_GRT || op == OP_GEQ)
1403 fop = 'l';
1405 else
1407 fop = 'g';
1409 jvm_generate(p, 2 * cell_size, 1, "%ccmp%c", prefix, fop);
1410 jvm_generate(p, 1, 1, "if%s L%i", cmpop, label_true);
1412 else if(prefix == 'a')
1414 if(t -> class == OBERON_TYPE_STRING
1415 || (t -> class == OBERON_TYPE_POINTER && t -> base -> class == OBERON_TYPE_CHAR))
1417 jvm_generate(p, 2, 1, "invokestatic SYSTEM/STRCMP([B[B)I");
1418 jvm_generate(p, 1, 0, "if%s L%i", cmpop, label_true);
1420 else
1422 jvm_generate(p, 1, 0, "if_acmp%s L%i", cmpop, label_true);
1425 else
1427 jvm_generate(p, 2 * cell_size, 0, "if_%ccmp%s L%i", prefix, cmpop, label_true);
1430 jvm_generate(p, 0, 1, "iconst_0");
1431 jvm_generate(p, 0, 0, "goto L%i", label_done);
1432 jvm_generate_label(p, label_true);
1433 jvm_generate(p, 0, 1, "iconst_1");
1434 jvm_generate_label(p, label_done);
1437 static void
1438 jvm_generate_operator(gen_proc_t * p, oberon_type_t * t, int op)
1440 char prefix = jvm_get_prefix(t);
1441 int cell_size = jvm_cell_size_for_type(t);
1442 switch(op)
1444 case OP_UNARY_MINUS:
1445 jvm_generate(p, cell_size, cell_size, "%cneg", prefix);
1446 break;
1447 case OP_COMPLEMENTATION:
1448 jvm_generate_push_int_size(p, -1, t -> size);
1449 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1450 break;
1451 case OP_LOGIC_NOT:
1452 jvm_generate_logical_not(p);
1453 break;
1454 case OP_ABS:
1455 jvm_generate_abs(p, prefix);
1456 break;
1457 case OP_CAP:
1458 jvm_generate(p, cell_size, cell_size, "invokestatic java/lang/Character/toUpperCase(I)I");
1459 break;
1461 case OP_ADD:
1462 jvm_generate(p, 2 * cell_size, cell_size, "%cadd", prefix);
1463 break;
1464 case OP_SUB:
1465 jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix);
1466 break;
1467 case OP_MUL:
1468 jvm_generate(p, 2 * cell_size, cell_size, "%cmul", prefix);
1469 break;
1470 case OP_DIV:
1471 jvm_generate(p, 2 * cell_size, cell_size, "%cdiv", prefix);
1472 break;
1473 case OP_MOD:
1474 jvm_generate(p, 2 * cell_size, cell_size, "%crem", prefix);
1475 break;
1476 case OP_UNION:
1477 jvm_generate(p, 2 * cell_size, cell_size, "%cor", prefix);
1478 break;
1479 case OP_INTERSECTION:
1480 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1481 break;
1482 case OP_DIFFERENCE:
1483 /* (a - b) == a & ~b */
1484 jvm_generate_push_int_size(p, -1, t -> size);
1485 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1486 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1487 break;
1488 case OP_SYM_DIFFERENCE:
1489 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1490 break;
1492 case OP_EQ:
1493 case OP_NEQ:
1494 case OP_LSS:
1495 case OP_LEQ:
1496 case OP_GRT:
1497 case OP_GEQ:
1498 jvm_generate_compare_op(p, t, op);
1499 break;
1500 default:
1501 gen_error("jvm_generate_operator: unk op %i", op);
1502 break;
1503 }
1506 static void
1507 jvm_generate_logical_or(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1509 int label_calc_b = jvm_new_label_id(p);
1510 int label_done = jvm_new_label_id(p);
1512 /* a OR b -- если a, то TRUE, иначе b */
1514 push_expr(p, a);
1515 jvm_generate(p, 1, 0, "ifeq L%i", label_calc_b);
1516 jvm_generate(p, 0, 1, "iconst_1");
1517 jvm_generate(p, 0, 0, "goto L%i", label_done);
1518 jvm_generate_label(p, label_calc_b);
1519 push_expr(p, b);
1520 jvm_generate_label(p, label_done);
1523 static void
1524 jvm_generate_logical_and(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1526 int label_false = jvm_new_label_id(p);
1527 int label_done = jvm_new_label_id(p);
1529 /* a AND b -- если a, то b, иначе FALSE */
1531 push_expr(p, a);
1532 jvm_generate(p, 1, 0, "ifeq L%i", label_false);
1533 push_expr(p, b);
1534 jvm_generate(p, 0, 0, "goto L%i", label_done);
1535 jvm_generate_label(p, label_false);
1536 jvm_generate(p, 0, 1, "iconst_0");
1537 jvm_generate_label(p, label_done);
1540 static void
1541 jvm_generate_range(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1543 /* { a } == 1 << a */
1544 /* { a..b } == (a <= b) ? ((2 << b) - (1 << a)) : (0); */
1546 char prefix;
1547 int cell_size;
1548 oberon_type_t * type;
1549 gen_var_t * ra;
1550 gen_var_t * rb;
1551 int label_else;
1552 int label_end;
1554 type = a -> result;
1555 cell_size = jvm_cell_size_for_type(type);
1556 prefix = jvm_get_prefix(type);
1558 if(b == NULL)
1560 jvm_generate_push_int_size(p, 1, type -> size);
1561 push_expr(p, a);
1562 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1564 else
1566 ra = jvm_create_function_temp_var(p, type -> gen_type);
1567 rb = jvm_create_function_temp_var(p, type -> gen_type);
1568 jvm_generate_var(ra);
1569 jvm_generate_var(rb);
1570 jvm_generate_var_initialization(p, ra);
1571 jvm_generate_var_initialization(p, rb);
1572 label_else = jvm_new_label_id(p);
1573 label_end = jvm_new_label_id(p);
1575 push_expr(p, a);
1576 jvm_generate_store(p, ra);
1577 push_expr(p, b);
1578 jvm_generate_store(p, rb);
1580 jvm_generate_load(p, ra);
1581 jvm_generate_load(p, rb);
1582 jvm_generate(p, 2 * cell_size, 0, "if_%ccmpgt L%i", prefix, label_else);
1584 jvm_generate_push_int_size(p, 2, type -> size);
1585 jvm_generate_load(p, rb);
1586 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1587 jvm_generate_push_int_size(p, 2, type -> size);
1588 jvm_generate_load(p, ra);
1589 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1590 jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix);
1591 jvm_generate(p, 0, 0, "goto L%i", label_end);
1593 jvm_generate_label(p, label_else);
1594 jvm_generate_push_int_size(p, 0, type -> size);
1595 jvm_generate_label(p, label_end);
1598 /* TODO free registers */
1601 static void
1602 jvm_generate_in(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1604 oberon_type_t * t = a -> result;
1605 int cell_size = jvm_cell_size_for_type(t);
1606 char prefix = jvm_get_prefix(t);
1607 int label_else = jvm_new_label_id(p);
1608 int label_end = jvm_new_label_id(p);
1610 /* (a IN b) == (1 << a) & b */
1611 jvm_generate_push_int_size(p, 1, t -> size);
1612 push_expr(p, a);
1613 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1614 push_expr(p, b);
1615 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1617 if(cell_size > 1)
1619 jvm_generate(p, cell_size, 0, "lcmp");
1622 jvm_generate(p, 1, 0, "ifeq L%i", label_else);
1623 jvm_generate(p, 0, 1, "iconst_1");
1624 jvm_generate(p, 0, 0, "goto L%i", label_end);
1625 jvm_generate_label(p, label_else);
1626 jvm_generate(p, 0, 1, "iconst_0");
1627 jvm_generate_label(p, label_end);
1630 static void
1631 jvm_generate_shift(gen_proc_t * p, int op, oberon_expr_t * a, oberon_expr_t * b)
1633 oberon_type_t * t = a -> result;
1634 int cell_size = jvm_cell_size_for_type(t);
1635 char prefix = jvm_get_prefix(t);
1636 char dt = jvm_get_type_of_prefix(prefix);
1638 push_expr(p, a);
1639 jvm_generate_cast_type(p, a -> result, t);
1640 push_expr(p, b);
1641 jvm_generate_cast_type(p, b -> result, t);
1643 char * opname;
1644 switch(op)
1646 case OP_ASH:
1647 opname = "ASH";
1648 break;
1649 case OP_LSH:
1650 opname = "LSH";
1651 break;
1652 case OP_ROT:
1653 opname = "ROT";
1654 break;
1655 default:
1656 gen_error("jvm_generate_shift: invalid op %i", op);
1659 jvm_generate(p, 2 * cell_size, cell_size, "invokestatic SYSTEM/%s(%c%c)%c", opname, dt, dt, dt);
1662 static void
1663 jvm_generate_entier(gen_proc_t * p, oberon_expr_t * x, oberon_type_t * res)
1665 char prefix = jvm_get_prefix(x -> result);
1666 char postfix = jvm_get_postfix(res);
1668 push_expr(p, x);
1669 jvm_generate_cast_prefix(p, prefix, 'd');
1670 jvm_generate(p, 2, 2, "invokestatic java/lang/Math/floor(D)D");
1671 jvm_generate_cast_prefix(p, 'd', postfix);
1674 static void
1675 push_operator(gen_proc_t * p, oberon_oper_t * oper)
1677 oberon_type_t * preq = oper -> left -> result;
1678 int op = oper -> op;
1679 switch(op)
1681 case OP_CAST:
1682 push_expr(p, oper -> left);
1683 jvm_generate_cast_type(p, oper -> left -> result, oper -> result);
1684 break;
1685 case OP_HARDCAST:
1686 push_expr(p, oper -> left);
1687 jvm_generate_hard_cast_type(p, oper -> left -> result, oper -> result);
1688 break;
1689 case OP_COMPLEMENTATION:
1690 case OP_UNARY_MINUS:
1691 case OP_LOGIC_NOT:
1692 case OP_ABS:
1693 case OP_CAP:
1694 push_expr(p, oper -> left);
1695 jvm_generate_operator(p, preq, op);
1696 break;
1697 case OP_ENTIER:
1698 jvm_generate_entier(p, oper -> left, oper -> result);
1699 break;
1701 case OP_ADD:
1702 case OP_SUB:
1703 case OP_MUL:
1704 case OP_DIV:
1705 case OP_MOD:
1707 case OP_UNION:
1708 case OP_INTERSECTION:
1709 case OP_DIFFERENCE:
1710 case OP_SYM_DIFFERENCE:
1712 case OP_EQ:
1713 case OP_NEQ:
1714 case OP_LSS:
1715 case OP_LEQ:
1716 case OP_GRT:
1717 case OP_GEQ:
1718 push_expr(p, oper -> left);
1719 push_expr(p, oper -> right);
1720 jvm_generate_operator(p, preq, op);
1721 break;
1723 case OP_LOGIC_OR:
1724 jvm_generate_logical_or(p, oper -> left, oper -> right);
1725 break;
1726 case OP_LOGIC_AND:
1727 jvm_generate_logical_and(p, oper -> left, oper -> right);
1728 break;
1730 case OP_IS:
1731 preq = oper -> right -> result;
1732 char * cname = jvm_get_class_full_name(preq);
1733 push_expr(p, oper -> left);
1734 jvm_generate(p, 1, 1, "instanceof %s", cname);
1735 break;
1736 case OP_RANGE:
1737 jvm_generate_range(p, oper -> left, oper -> right);
1738 break;
1739 case OP_IN:
1740 jvm_generate_in(p, oper -> left, oper -> right);
1741 break;
1743 case OP_ASH:
1744 case OP_LSH:
1745 case OP_ROT:
1746 jvm_generate_shift(p, op, oper -> left, oper -> right);
1747 break;
1748 default:
1749 gen_error("push_oper: unk op %i", op);
1750 break;
1754 static void
1755 push_expr(gen_proc_t * p, oberon_expr_t * expr)
1757 if(expr -> is_item)
1759 push_item(p, (oberon_item_t *) expr);
1761 else
1763 push_operator(p, (oberon_oper_t *) expr);
1767 static void
1768 store_expr(gen_proc_t * p, oberon_expr_t * dst, oberon_expr_t * src)
1770 assert(dst -> is_item);
1771 oberon_item_t * item = (oberon_item_t *) dst;
1773 if(dst -> result -> class == OBERON_TYPE_ARRAY
1774 || src -> result -> class == OBERON_TYPE_ARRAY)
1776 push_expr(p, dst);
1777 push_expr(p, src);
1778 jvm_generate_copy_array(p, dst -> result);
1780 else if(dst -> result -> class == OBERON_TYPE_RECORD
1781 || src -> result -> class == OBERON_TYPE_RECORD)
1783 push_expr(p, dst);
1784 push_expr(p, src);
1785 jvm_generate_copy_record(p, dst -> result);
1787 else switch(item -> mode)
1789 case MODE_VAR:
1790 jvm_generate_ldst_prepare(p, item -> var -> gen_var);
1791 push_expr(p, src);
1792 jvm_generate_store(p, item -> var -> gen_var);
1793 break;
1794 case MODE_INDEX:
1796 char postfix = jvm_get_postfix(src -> result);
1797 int cell_size = jvm_cell_size_for_postfix(postfix);
1798 assert(item -> parent -> is_item);
1799 push_item(p, (oberon_item_t *) item -> parent);
1800 push_expr(p, item -> args);
1801 push_expr(p, src);
1802 jvm_generate(p, 1 + 1 + cell_size, 0, "%castore", postfix);
1803 break;
1804 case MODE_FIELD:
1805 assert(item -> parent -> is_item);
1806 push_item(p, (oberon_item_t *) item -> parent);
1807 jvm_generate_ldst_prepare(p, item -> var -> gen_var);
1808 push_expr(p, src);
1809 jvm_generate_store(p, item -> var -> gen_var);
1810 break;
1811 default:
1812 gen_error("store_expr: unk mode %i", item -> mode);
1813 break;
1817 void
1818 oberon_generate_assign(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_t * dst)
1820 gen_module_t * m;
1821 gen_proc_t * p;
1823 m = ctx -> mod -> gen_mod;
1824 p = m -> class -> p;
1826 store_expr(p, dst, src);
1829 void
1830 oberon_generate_copy(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_t * dst)
1832 gen_module_t * m;
1833 gen_proc_t * p;
1834 char * desc;
1836 m = ctx -> mod -> gen_mod;
1837 p = m -> class -> p;
1839 push_expr(p, src);
1840 push_expr(p, dst);
1842 desc = jvm_get_descriptor(dst -> result);
1844 jvm_generate(p, 2, 0, "invokestatic SYSTEM/COPY(%s%s)V", desc, desc);
1847 void
1848 oberon_generate_assert(oberon_context_t * ctx, oberon_expr_t * cond)
1850 gen_module_t * m;
1851 gen_proc_t * p;
1853 m = ctx -> mod -> gen_mod;
1854 p = m -> class -> p;
1856 push_expr(p, cond);
1857 jvm_generate(p, 1, 0, "invokestatic SYSTEM/ASSERT(Z)V");
1860 void
1861 oberon_generate_assert_n(oberon_context_t * ctx, oberon_expr_t * cond, int64_t n)
1863 gen_module_t * m;
1864 gen_proc_t * p;
1866 m = ctx -> mod -> gen_mod;
1867 p = m -> class -> p;
1869 push_expr(p, cond);
1870 jvm_generate_push_int_size(p, n, 8);
1871 jvm_generate(p, 1 + 2, 0, "invokestatic SYSTEM/ASSERT(ZJ)V");
1874 void
1875 oberon_generate_trap(oberon_context_t * ctx, int64_t n)
1877 gen_module_t * m;
1878 gen_proc_t * p;
1880 m = ctx -> mod -> gen_mod;
1881 p = m -> class -> p;
1883 jvm_generate_push_int_size(p, n, 8);
1884 jvm_generate(p, 2, 0, "invokestatic SYSTEM/TRAP(J)V");
1887 void
1888 oberon_generate_halt(oberon_context_t * ctx, int64_t n)
1890 gen_module_t * m;
1891 gen_proc_t * p;
1893 m = ctx -> mod -> gen_mod;
1894 p = m -> class -> p;
1896 jvm_generate_push_int_size(p, n, 8);
1897 jvm_generate(p, 2, 0, "invokestatic SYSTEM/HALT(J)V");
1900 void
1901 oberon_set_out_directory(oberon_context_t * ctx, const char * path)
1903 gen_context_t * c;
1905 c = ctx -> gen_context;
1907 c -> dir = new_string(path);