DEADSOFTWARE

Добавлен счёт строк
[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-jvm.h"
13 #include "generator-jvm-abi.h"
14 #include "generator-jvm-asm.h"
15 #include "generator-jvm-basic.h"
17 static void
18 push_item(gen_proc_t * p, oberon_item_t * item);
20 static void
21 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num);
23 /*
24 * Генерирует код для инициализации массива со статическим базовым типом
25 * ( aref -- )
26 */
28 static void
29 jvm_generate_array_initialization(gen_proc_t * p, oberon_type_t * arr)
30 {
31 int dim = 0;
32 oberon_type_t * base = arr;
33 while(base -> class == OBERON_TYPE_ARRAY)
34 {
35 dim += 1;
36 base = base -> base;
37 }
39 if(base -> class != OBERON_TYPE_RECORD)
40 {
41 jvm_generate(p, 1, 0, "pop");
42 return;
43 }
45 struct {
46 int reg_index;
47 int reg_length;
48 int start;
49 int end;
50 } loop[dim];
52 int reg_dst;
53 reg_dst = jvm_alloc_register_untyped(p -> rf, false);
54 jvm_generate(p, 1, 0, "astore %i", reg_dst);
56 /*
57 * Входящие параметры заграблены.
58 * Теперь генерируем эквивалентный код:
59 * int i = 0;
60 * int len = dst.length
61 * while(i < len)
62 * {
63 * ...
64 * {
65 * dst[i, ...] = new record;
66 * }
67 * ...
68 * i += 1;
69 * }
70 * Где "..." такой же код (начало и конец) для следующей размерности.
71 */
73 for(int i = 0; i < dim; i++)
74 {
75 loop[i].reg_index = jvm_alloc_register_untyped(p -> rf, false);
76 loop[i].reg_length = jvm_alloc_register_untyped(p -> rf, false);
77 loop[i].start = jvm_new_label_id(p);
78 loop[i].end = jvm_new_label_id(p);
80 jvm_generate(p, 0, 1, "iconst_0");
81 jvm_generate(p, 1, 0, "istore %i", loop[i].reg_index);
83 jvm_generate(p, 0, 1, "aload %i", reg_dst);
84 jvm_generate(p, 1, 1, "arraylength");
85 jvm_generate(p, 1, 0, "istore %i", loop[i].reg_length);
87 /* if(i >= len) goto end; */
88 jvm_generate_label(p, loop[i].start);
89 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
90 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_length);
91 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
92 }
94 jvm_generate(p, 0, 1, "aload %i", reg_dst);
95 jvm_generate(p, 0, 1, "iload %i", loop[0].reg_index);
96 for(int i = 1; i < dim; i++)
97 {
98 jvm_generate(p, 2, 1, "aaload");
99 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
101 jvm_generate_new(p, base, 0);
102 jvm_generate(p, 3, 0, "aastore");
104 for(int i = dim - 1; i >= 0; i--)
106 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].reg_index);
107 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
108 jvm_generate_label(p, loop[i].end);
112 static void
113 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num)
115 int dim;
116 char * cname;
117 char * desc;
118 oberon_type_t * base;
120 switch(type -> class)
122 case OBERON_TYPE_INTEGER:
123 case OBERON_TYPE_BOOLEAN:
124 case OBERON_TYPE_PROCEDURE:
125 case OBERON_TYPE_REAL:
126 case OBERON_TYPE_POINTER:
127 gen_error("jvm_generate_new_static: static alocation not allowed");
128 break;
129 case OBERON_TYPE_RECORD:
130 assert(num == 0);
131 cname = jvm_get_class_full_name(type);
132 jvm_generate(p, 0, 1, "new %s", cname);
133 jvm_generate(p, 1, 2, "dup");
134 jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", cname);
135 break;
136 case OBERON_TYPE_ARRAY:
137 dim = 0;
138 base = type;
139 desc = jvm_get_descriptor(type);
140 while(base -> class == OBERON_TYPE_ARRAY)
142 if(num > 0)
144 assert(base -> size == 0);
145 num -= 1;
147 else
149 assert(base -> size > 0);
150 jvm_generate_push_int(p, base -> size);
152 dim += 1;
153 base = base -> base;
156 assert(num == 0);
157 jvm_generate(p, dim, 1, "multianewarray %s %i", desc, dim);
158 jvm_generate(p, 1, 2, "dup");
159 jvm_generate_array_initialization(p, type);
160 break;
161 default:
162 gen_error("jvm_generate_new_static: unk type class %i", type -> class);
163 break;
167 /*
168 * Генерирует код для копирования полей из первой записи во вторую.
169 * ( aref_dst aref_src -- )
170 * dst := src;
171 */
173 static void
174 jvm_generate_copy_record(gen_proc_t * p, oberon_type_t * rec)
176 assert(rec -> class == OBERON_TYPE_RECORD);
177 char * desc = jvm_get_descriptor(rec);
178 char * cname = jvm_get_class_full_name(rec);
179 jvm_generate(p, 1 + 1, 0, "invokestatic %s/$COPY$(%s%s)V", cname, desc, desc);
182 /*
183 * Генерирует кода для копирования массивов.
184 * ( aref_dst aref_src -- )
185 * dst := src;
186 */
188 static void
189 jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
191 int dim = 0;
192 oberon_type_t * base = arr;
193 while(base -> class == OBERON_TYPE_ARRAY)
195 dim += 1;
196 base = base -> base;
199 struct {
200 int reg_index;
201 int reg_length;
202 int start;
203 int end;
204 } loop[dim];
206 int reg_dst = jvm_alloc_register_untyped(p -> rf, false);
207 int reg_src = jvm_alloc_register_untyped(p -> rf, false);
208 jvm_generate(p, 1, 0, "astore %i", reg_src);
209 jvm_generate(p, 1, 0, "astore %i", reg_dst);
211 /*
212 * Входящие параметры заграблены.
213 * Теперь генерируем эквивалентный код:
214 * int i = 0;
215 * int len = src.length
216 * while(i < len)
217 * {
218 * ...
219 * {
220 * copy from src[i, ...] to dst[i, ...];
221 * }
222 * ...
223 * i += 1;
224 * }
225 * Где "..." такой же код (начало и конец) для следующей размерности.
226 */
228 for(int i = 0; i < dim; i++)
230 loop[i].reg_index = jvm_alloc_register_untyped(p -> rf, false);
231 loop[i].reg_length = jvm_alloc_register_untyped(p -> rf, false);
232 loop[i].start = jvm_new_label_id(p);
233 loop[i].end = jvm_new_label_id(p);
235 jvm_generate(p, 0, 1, "iconst_0");
236 jvm_generate(p, 1, 0, "istore %i", loop[i].reg_index);
238 jvm_generate(p, 0, 1, "aload %i", reg_src);
239 jvm_generate(p, 1, 1, "arraylength");
240 jvm_generate(p, 1, 0, "istore %i", loop[i].reg_length);
242 /* if(i >= len) goto end; */
243 jvm_generate_label(p, loop[i].start);
244 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
245 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_length);
246 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
249 if(base -> class == OBERON_TYPE_RECORD)
251 /* Получаем записи по индексам ( -- dst src ) */
253 jvm_generate(p, 0, 1, "aload %i", reg_dst);
254 for(int i = 0; i < dim; i++)
256 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
257 jvm_generate(p, 2, 1, "aaload");
260 jvm_generate(p, 0, 1, "aload %i", reg_src);
261 for(int i = 0; i < dim; i++)
263 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
264 jvm_generate(p, 2, 1, "aaload");
267 /* Копируем записи ( dst src -- ) */
268 jvm_generate_copy_record(p, base);
270 else
272 char postfix = jvm_get_postfix(base);
273 int cell_size = jvm_cell_size_for_postfix(postfix);
275 /* Получаем массивы и индексы ( -- dst i src i ) */
277 jvm_generate(p, 0, 1, "aload %i", reg_dst);
278 jvm_generate(p, 0, 1, "iload %i", loop[0].reg_index);
279 for(int i = 1; i < dim; i++)
281 jvm_generate(p, 2, 1, "aaload");
282 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
285 jvm_generate(p, 0, 1, "aload %i", reg_src);
286 jvm_generate(p, 0, 1, "iload %i", loop[0].reg_index);
287 for(int i = 1; i < dim; i++)
289 jvm_generate(p, 2, 1, "aaload");
290 jvm_generate(p, 0, 1, "iload %i", loop[i].reg_index);
293 /* Копируем значения ( dst i src i -- ) */
294 jvm_generate(p, 2, cell_size, "%caload", postfix);
295 jvm_generate(p, 2 + cell_size, 0, "%castore", postfix);
298 for(int i = dim - 1; i >= 0; i--)
300 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].reg_index);
301 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
302 jvm_generate_label(p, loop[i].end);
306 static void
307 jvm_generate_push_procedure_pointer(gen_proc_t * p, oberon_object_t * proc)
309 char * full_name = jvm_get_field_full_name_safe(proc);
310 char * desc = jvm_get_descriptor(proc -> type);
311 jvm_generate(p, 0, 1, "getstatic %s/pointer %s", full_name, desc);
314 void
315 oberon_generator_init_context(oberon_context_t * ctx)
317 gen_context_t * gen_context = GC_MALLOC(sizeof *gen_context);
318 memset(gen_context, 0, sizeof *gen_context);
320 ctx -> gen_context = gen_context;
321 gen_context -> dir = ".";
324 void
325 oberon_generator_destroy_context(oberon_context_t * ctx)
329 static void
330 oberon_generate_procedure_class(const char * dir, oberon_type_t * proc)
332 FILE * fp;
333 char * cname;
334 char * signature;
335 struct gen_class * class;
337 cname = jvm_get_class_full_name(proc);
338 class = jvm_create_class(dir, cname);
339 fp = class -> fp;
341 fprintf(fp, ".source SYSTEM\n");
342 fprintf(fp, ".class public abstract %s\n", cname);
343 fprintf(fp, ".super java/lang/Object\n\n");
345 fprintf(fp, ".method <init>()V\n");
346 fprintf(fp, " aload_0\n");
347 fprintf(fp, " invokespecial java/lang/Object/<init>()V\n");
348 fprintf(fp, " return\n");
349 fprintf(fp, ".end method\n\n");
351 signature = jvm_get_procedure_signature(proc);
353 fprintf(fp, ".method public abstract invoke%s\n", signature);
354 fprintf(fp, ".end method\n\n");
356 jvm_destroy_class(class);
359 static void
360 oberon_generate_procedure_pointer_class(const char * dir, oberon_object_t * proc)
362 FILE * fp;
363 char * cname;
364 char * abscname;
365 char * absdesc;
366 char * signature;
367 struct gen_class * class;
369 cname = jvm_get_field_full_name_safe(proc);
370 class = jvm_create_class(dir, cname);
371 abscname = jvm_get_class_full_name(proc -> type);
372 absdesc = jvm_get_descriptor(proc -> type);
373 fp = class -> fp;
375 fprintf(fp, ".source %s\n", proc -> module -> name);
376 fprintf(fp, ".class public %s\n", cname);
377 fprintf(fp, ".super %s\n\n", abscname);
379 fprintf(fp, ".field public static pointer %s\n\n", absdesc);
381 fprintf(fp, ".method private <init>()V\n");
382 fprintf(fp, " aload_0\n");
383 fprintf(fp, " invokespecial %s/<init>()V\n", abscname);
384 fprintf(fp, " return\n");
385 fprintf(fp, ".end method\n\n");
387 fprintf(fp, ".method static <clinit>()V\n");
388 fprintf(fp, " .limit stack 2\n");
389 fprintf(fp, " new %s\n", cname);
390 fprintf(fp, " dup\n");
391 fprintf(fp, " invokespecial %s/<init>()V\n", cname);
392 fprintf(fp, " putstatic %s/pointer %s\n", cname, absdesc);
393 fprintf(fp, " return\n");
394 fprintf(fp, ".end method\n\n");
396 signature = jvm_get_procedure_signature(proc -> type);
398 gen_proc_t * p = jvm_create_proc(class);
399 jvm_generate_function_header(p, "public", "invoke", signature);
400 jvm_alloc_register_untyped(p -> rf, false);
402 int use_size = 0;
403 int num = proc -> type -> num_decl;
404 oberon_object_t * param = proc -> type -> decl;
405 for(int i = 0; i < num; i++)
407 gen_type_t * t = param -> type -> gen_type;
408 if(param -> class == OBERON_CLASS_VAR_PARAM)
410 int reg = jvm_alloc_register_untyped(p -> rf, true);
411 jvm_generate(p, 0, 1, "aload %i", reg);
412 jvm_generate(p, 0, 1, "iload %i", reg + 1);
414 else
416 int reg = jvm_alloc_register_untyped(p -> rf, t -> wide);
417 jvm_generate(p, 0, t -> cell_size, "%cload %i", t -> prefix, reg);
419 use_size += t -> cell_size;
420 param = param -> next;
423 char * full_name = jvm_get_field_full_name(proc);
424 int cell_size = jvm_cell_size_for_type(proc -> type -> base);
425 jvm_generate(p, use_size, cell_size, "invokestatic %s%s", full_name, signature);
427 if(proc -> type -> base -> class == OBERON_TYPE_NOTYPE)
429 jvm_generate(p, 0, 0, "return");
431 else
433 char prefix = jvm_get_prefix(proc -> type -> base);
434 jvm_generate(p, cell_size, 0, "%creturn", prefix);
437 jvm_generate_function_end(p);
439 jvm_destroy_class(class);
442 static void
443 oberon_generate_record_class(const char * dir, gen_module_t * m, oberon_type_t * rec)
445 char * cname;
446 struct gen_class * class;
448 /* Устанавливаем новоый id */
449 rec -> gen_type -> rec_id = m -> rec_id;
450 m -> rec_id += 1;
452 cname = jvm_get_class_full_name(rec);
453 class = jvm_create_class(dir, cname);
455 fprintf(class -> fp, ".source %s\n", rec -> module -> name);
456 fprintf(class -> fp, ".class public %s\n", cname);
458 if(rec -> base == NULL)
460 fprintf(class -> fp, ".super java/lang/Object\n\n");
462 else
464 class -> base = rec -> base -> gen_type -> class;
465 fprintf(class -> fp, ".super %s\n\n", class -> base -> full_name);
468 rec -> gen_type -> class = class;
471 void
472 oberon_generator_init_type(oberon_context_t * ctx, oberon_type_t * type)
474 gen_type_t * t = GC_MALLOC(sizeof *t);
475 memset(t, 0, sizeof *t);
476 type -> gen_type = t;
478 gen_context_t * c;
479 c = ctx -> gen_context;
481 gen_module_t * m;
482 switch(type -> class)
484 case OBERON_TYPE_NOTYPE:
485 case OBERON_TYPE_INTEGER:
486 case OBERON_TYPE_BOOLEAN:
487 case OBERON_TYPE_ARRAY:
488 case OBERON_TYPE_POINTER:
489 case OBERON_TYPE_REAL:
490 case OBERON_TYPE_CHAR:
491 case OBERON_TYPE_STRING:
492 case OBERON_TYPE_SET:
493 case OBERON_TYPE_NIL:
494 case OBERON_TYPE_SYSTEM_BYTE:
495 case OBERON_TYPE_SYSTEM_PTR:
496 break;
497 case OBERON_TYPE_RECORD:
498 m = type -> module -> gen_mod;
499 oberon_generate_record_class(c -> dir, m, type);
500 break;
501 case OBERON_TYPE_PROCEDURE:
502 oberon_generate_procedure_class(c -> dir, type);
503 break;
504 default:
505 gen_error("oberon_generator_init_type: unk calss %i", type -> class);
506 break;
509 if(type -> class != OBERON_TYPE_NOTYPE)
511 t -> wide = jvm_is_wide_type(type);
512 t -> prefix = jvm_get_prefix(type);
513 t -> postfix = jvm_get_postfix(type);
516 t -> cell_size = jvm_cell_size_for_type(type);
518 if(type -> class != OBERON_TYPE_NIL)
520 t -> desc = jvm_get_descriptor(type);
524 void
525 oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
527 struct gen_class * class;
528 class = rec -> gen_type -> class;
530 int num = rec -> num_decl;
531 oberon_object_t * field = rec -> decl;
532 for(int i = 0; i < num; i++)
534 jvm_generate_var(field -> gen_var);
535 field = field -> next;
538 /* Стандартный конструктор класса */
539 /* Инициализирует внутренние статические записи и массивы */
540 gen_proc_t * p = jvm_create_proc(class);
541 jvm_generate_function_header(p, "public", "<init>", "()V");
542 jvm_alloc_register_untyped(p -> rf, false);
543 jvm_generate(p, 0, 1, "aload_0");
544 if(class -> base)
546 jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", class -> base -> full_name);
548 else
550 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
552 num = rec -> num_decl;
553 field = rec -> decl;
554 for(int i = 0; i < num; i++)
556 jvm_generate(p, 0, 1, "aload_0");
557 jvm_generate_var_initialization(p, field -> gen_var);
559 if(field -> type -> class == OBERON_TYPE_RECORD
560 || field -> type -> class == OBERON_TYPE_ARRAY)
562 jvm_generate(p, 0, 1, "aload_0");
563 jvm_generate_ldst_prepare(p, field -> gen_var);
564 jvm_generate_new(p, field -> type, 0);
565 jvm_generate_store(p, field -> gen_var);
567 field = field -> next;
569 jvm_generate(p, 0, 0, "return");
570 jvm_generate_function_end(p);
572 /* Метод для копирования полей класса */
573 /* reg0 == src -> reg1 == dst */
574 p = jvm_create_proc(class);
575 char * signature = new_string("(%s%s)V", rec -> gen_type -> desc, rec -> gen_type -> desc);
576 jvm_generate_function_header(p, "public static", "$COPY$", signature);
578 gen_var_t * copy_dst = jvm_create_function_var(p, JVM_STORAGE_REGISTER, "dst", rec -> gen_type);
579 gen_var_t * copy_src = jvm_create_function_var(p, JVM_STORAGE_REGISTER, "src", rec -> gen_type);
580 jvm_generate_var(copy_dst);
581 jvm_generate_var(copy_src);
582 jvm_generate_var_initialization(p, copy_dst);
583 jvm_generate_var_initialization(p, copy_src);
585 num = rec -> num_decl;
586 field = rec -> decl;
587 for(int i = 0; i < num; i++)
589 if(field -> type -> class == OBERON_TYPE_RECORD)
591 jvm_generate_load(p, copy_dst);
592 jvm_generate_load(p, field -> gen_var);
593 jvm_generate_load(p, copy_src);
594 jvm_generate_load(p, field -> gen_var);
595 jvm_generate_copy_record(p, field -> type);
597 else if(field -> type -> class == OBERON_TYPE_ARRAY)
599 jvm_generate_load(p, copy_dst);
600 jvm_generate_load(p, field -> gen_var);
601 jvm_generate_load(p, copy_src);
602 jvm_generate_load(p, field -> gen_var);
603 jvm_generate_copy_array(p, field -> type);
605 else
607 jvm_generate_load(p, copy_dst);
608 jvm_generate_ldst_prepare(p, field -> gen_var);
609 jvm_generate_load(p, copy_src);
610 jvm_generate_load(p, field -> gen_var);
611 jvm_generate_store(p, field -> gen_var);
613 field = field -> next;
615 jvm_generate(p, 0, 0, "return");
616 jvm_generate_function_end(p);
618 jvm_destroy_class(class);
621 void
622 oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var)
624 gen_module_t * m;
625 m = ctx -> mod -> gen_mod;
627 struct gen_class * class;
628 class = m -> class;
630 gen_proc_t * p;
631 struct gen_class * c;
632 char * name = var -> name;
633 gen_type_t * t = var -> type -> gen_type;
634 switch(var -> class)
636 case OBERON_CLASS_VAR_PARAM:
637 p = var -> parent -> gen_proc;
638 var -> gen_var = jvm_create_function_var(p, JVM_STORAGE_FRAME_PARAM_VARPTR, name, t);
639 break;
640 case OBERON_CLASS_PARAM:
641 p = var -> parent -> gen_proc;
642 var -> gen_var = jvm_create_function_var(p, JVM_STORAGE_FRAME_PARAM_VAR, name, t);
643 break;
644 case OBERON_CLASS_FIELD:
645 c = var -> parent_type -> gen_type -> class;
646 var -> gen_var = jvm_create_class_var(c, JVM_STORAGE_FIELD_VAR, name, t);
647 break;
648 case OBERON_CLASS_VAR:
649 if(var -> local)
651 p = var -> parent -> gen_proc;
652 var -> gen_var = jvm_create_function_var(p, JVM_STORAGE_FRAME_VAR, name, t);
654 else
656 var -> gen_var = jvm_create_class_var(class, JVM_STORAGE_STATIC_VAR, name, t);
657 jvm_generate_var(var -> gen_var);
659 break;
660 default:
661 gen_error("oberon_generator_init_var: unk var class %i", var -> class);
662 break;
666 void
667 oberon_generator_init_temp_var(oberon_context_t * ctx, oberon_object_t * var)
669 assert(var -> class == OBERON_CLASS_VAR);
671 gen_module_t * m;
672 m = ctx -> mod -> gen_mod;
674 gen_proc_t * p;
675 p = m -> class -> p;
677 gen_type_t * t;
678 t = var -> type -> gen_type;
680 var -> gen_var = jvm_create_function_temp_var(p, t);
681 jvm_generate_var(var -> gen_var);
682 jvm_generate_var_initialization(p, var -> gen_var);
685 void
686 oberon_generator_init_proc(oberon_context_t * ctx, oberon_object_t * proc)
688 struct gen_class * class = proc -> module -> gen_mod -> class;
689 proc -> gen_proc = jvm_create_proc(class);
690 proc -> gen_proc -> full_name = jvm_get_field_full_name_safe(proc);
691 jvm_create_static_links(proc);
694 void
695 oberon_generator_init_module(oberon_context_t * ctx, oberon_module_t * mod)
697 gen_module_t * m = GC_MALLOC(sizeof *m);
698 memset(m, 0, sizeof *m);
699 mod -> gen_mod = m;
701 gen_context_t * c;
702 c = ctx -> gen_context;
704 struct gen_class * class;
705 class = jvm_create_class(c -> dir, mod -> name);
707 fprintf(class -> fp, ".source %s\n", mod -> name);
708 fprintf(class -> fp, ".class %s\n", mod -> name);
709 fprintf(class -> fp, ".super java/lang/Object\n\n");
711 fprintf(class -> fp, ".field private static $INITIALIZED$ Z\n\n");
713 m -> class = class;
715 c -> current_m = m;
718 void
719 oberon_generator_fini_module(oberon_module_t * mod)
721 jvm_destroy_class(mod -> gen_mod -> class);
724 static void
725 push_expr(gen_proc_t * p, oberon_expr_t * expr);
727 void
728 oberon_generate_begin_module(oberon_context_t * ctx)
730 gen_proc_t * p;
731 int label_cont;
732 struct gen_class * class;
734 class = ctx -> mod -> gen_mod -> class;
736 p = jvm_create_proc(class);
737 jvm_generate_function_header(p, "public static", "BEGIN", "()V");
739 label_cont = jvm_new_label_id(p);
740 jvm_generate(p, 0, 1, "getstatic %s/$INITIALIZED$ Z", class -> full_name);
741 jvm_generate(p, 1, 0, "ifeq L%i", label_cont);
742 jvm_generate(p, 0, 0, "return");
743 jvm_generate_label(p, label_cont);
745 jvm_generate(p, 0, 1, "iconst_1");
746 jvm_generate(p, 1, 0, "putstatic %s/$INITIALIZED$ Z", class -> full_name);
748 /* Инициализация переменных объявленных в модуле */
749 oberon_object_t * x = ctx -> mod -> decl -> list -> next;
750 while(x != NULL)
752 if(x -> class == OBERON_CLASS_MODULE)
754 if(!x -> module -> intrinsic)
756 jvm_generate(p, 0, 0, "invokestatic %s/BEGIN()V", x -> module -> gen_mod -> class -> full_name);
759 else if(x -> class == OBERON_CLASS_VAR)
761 jvm_generate_var_initialization(p, x -> gen_var);
762 if(x -> type -> class == OBERON_TYPE_ARRAY
763 || x -> type -> class == OBERON_TYPE_RECORD)
765 jvm_generate_ldst_prepare(p, x -> gen_var);
766 jvm_generate_new(p, x -> type, 0);
767 jvm_generate_store(p, x -> gen_var);
771 x = x -> next;
775 void
776 oberon_generate_end_module(oberon_context_t * ctx)
778 struct gen_class * class = ctx -> mod -> gen_mod -> class;
779 gen_proc_t * p = class -> p;
781 jvm_generate(p, 0, 0, "return");
782 jvm_generate_function_end(class -> p);
784 p = jvm_create_proc(class);
785 jvm_generate_function_header(p, "private", "<init>", "()V");
786 jvm_alloc_register_untyped(p -> rf, false);
787 jvm_generate(p, 0, 1, "aload_0");
788 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
789 jvm_generate(p, 0, 0, "return");
790 jvm_generate_function_end(class -> p);
793 /*
794 * Генерирует код для получения размера измерения массива
795 * Аналог Обероновского LEN(v, n); где n = 0 - первое измерение.
796 * ( aref -- int )
797 */
799 static void
800 jvm_generate_array_len(gen_proc_t * p, int dim)
802 while(dim > 0)
804 jvm_generate(p, 0, 1, "iconst_0");
805 jvm_generate(p, 2, 1, "aaload");
806 dim -= 1;
808 jvm_generate(p, 1, 1, "arraylength");
811 static void
812 jvm_generate_array_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * arr)
814 jvm_generate_ldst_prepare(p, v);
816 int dim = 0;
817 oberon_type_t * base = arr;
818 while(base -> class == OBERON_TYPE_ARRAY)
820 if(base -> size == 0)
822 jvm_generate_load(p, v);
823 jvm_generate_array_len(p, dim);
824 dim += 1;
826 base = base -> base;
829 jvm_generate_new(p, arr, dim);
830 jvm_generate(p, 1, 2, "dup");
831 jvm_generate_load(p, v);
832 jvm_generate_copy_array(p, arr);
833 jvm_generate_store(p, v);
836 static void
837 jvm_generate_record_duplicate_and_replace(gen_proc_t * p, gen_var_t * v, oberon_type_t * rec)
839 jvm_generate_ldst_prepare(p, v);
840 jvm_generate_new(p, rec, 0);
841 jvm_generate(p, 1, 2, "dup");
842 jvm_generate_load(p, v);
843 jvm_generate_copy_record(p, rec);
844 jvm_generate_store(p, v);
847 static void
848 jvm_generate_local_object(gen_proc_t * p, oberon_object_t * x)
850 gen_var_t * v;
851 struct gen_class * class;
853 v = x -> gen_var;
854 class = p -> class;
856 // Убеждаемся что сейчас находимся в функции
857 assert(class -> p);
858 assert(x -> local);
860 switch(x -> class)
862 case OBERON_CLASS_VAR_PARAM:
863 case OBERON_CLASS_PARAM:
864 case OBERON_CLASS_VAR:
865 jvm_generate_var(v);
866 break;
867 case OBERON_CLASS_PROC:
868 break;
869 default:
870 gen_error("jvm_generate_local_initialization: wat");
871 break;
875 static void
876 jvm_generate_local_initialization(gen_proc_t * p, oberon_object_t * x)
878 gen_var_t * v;
879 struct gen_class * class;
881 v = x -> gen_var;
882 class = p -> class;
884 // Убеждаемся что сейчас находимся в функции
885 assert(class -> p);
886 assert(x -> local);
888 switch(x -> class)
890 case OBERON_CLASS_VAR_PARAM:
891 case OBERON_CLASS_PARAM:
892 case OBERON_CLASS_VAR:
893 jvm_generate_var_initialization(p, v);
894 break;
895 case OBERON_CLASS_PROC:
896 break;
897 default:
898 gen_error("jvm_generate_local_initialization: wat");
899 break;
902 if(x -> class == OBERON_CLASS_PARAM)
904 if(x -> type -> class == OBERON_TYPE_ARRAY)
906 jvm_generate_array_duplicate_and_replace(p, v, x -> type);
908 else if(x -> type -> class == OBERON_TYPE_RECORD)
909 {
910 jvm_generate_record_duplicate_and_replace(p, v, x -> type);
915 void
916 oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc)
918 gen_proc_t * p;
919 char * name;
920 char * signature;
921 oberon_object_t * var;
923 p = proc -> gen_proc;
925 if(proc -> local)
927 signature = jvm_get_local_procedure_signature(proc);
929 else
931 oberon_generate_procedure_pointer_class(p -> class -> dir, proc);
932 signature = jvm_get_procedure_signature(proc -> type);
935 name = jvm_get_name(proc);
936 jvm_generate_function_header(p, "public static", name, signature);
938 if(proc -> local)
940 jvm_generate_staticlinks(proc);
943 jvm_generate_procedure_frame(proc);
945 /* Создание параметров и переменных */
946 var = proc -> scope -> list -> next;
947 while(var)
949 jvm_generate_local_object(p, var);
950 var = var -> next;
953 jvm_generate_frame_initialization(p);
955 /* Инициализация парамеров и переменных */
956 var = proc -> scope -> list -> next;
957 while(var)
959 jvm_generate_local_initialization(p, var);
960 var = var -> next;
964 void
965 oberon_generate_end_proc(oberon_context_t * ctx)
967 gen_module_t * m;
968 gen_proc_t * p;
970 m = ctx -> mod -> gen_mod;
971 p = m -> class -> p;
973 jvm_generate_function_end(p);
976 gen_label_t *
977 oberon_generator_reserve_label(oberon_context_t * ctx)
979 gen_module_t * m;
980 gen_proc_t * p;
981 gen_label_t * l;
983 m = ctx -> mod -> gen_mod;
984 p = m -> class -> p;
986 l = GC_MALLOC(sizeof *l);
987 memset(l, 0, sizeof *l);
989 l -> id = jvm_new_label_id(p);
990 return l;
993 void
994 oberon_generate_label(oberon_context_t * ctx, gen_label_t * l)
996 gen_module_t * m;
997 gen_proc_t * p;
999 m = ctx -> mod -> gen_mod;
1000 p = m -> class -> p;
1002 jvm_generate_label(p, l -> id);
1005 void
1006 oberon_generate_goto(oberon_context_t * ctx, gen_label_t * l)
1008 gen_module_t * m;
1009 gen_proc_t * p;
1011 m = ctx -> mod -> gen_mod;
1012 p = m -> class -> p;
1014 jvm_generate(p, 0, 0, "goto L%i", l -> id);
1017 void
1018 oberon_generate_branch(oberon_context_t * ctx, oberon_expr_t * cond, bool gotoif, gen_label_t * l)
1020 gen_module_t * m;
1021 gen_proc_t * p;
1023 m = ctx -> mod -> gen_mod;
1024 p = m -> class -> p;
1026 push_expr(p, cond);
1028 if(gotoif == false)
1030 /* переход если false */
1031 jvm_generate(p, 1, 0, "ifeq L%i", l -> id);
1033 else
1035 /* переход если true */
1036 jvm_generate(p, 1, 0, "ifne L%i", l -> id);
1040 static void
1041 push_varptr(gen_proc_t * p, oberon_expr_t * expr)
1043 assert(expr -> is_item);
1045 switch(expr -> item.mode)
1047 case MODE_VAR:
1048 jvm_generate_ldst_prepare(p, expr -> item.var -> gen_var);
1049 break;
1050 case MODE_INDEX:
1051 push_item(p, expr -> item.parent);
1052 push_expr(p, expr -> item.args);
1053 break;
1054 case MODE_FIELD:
1055 push_item(p, expr -> item.parent);
1056 jvm_generate_ldst_prepare(p, expr -> item.var -> gen_var);
1057 break;
1058 case MODE_DEREF:
1059 push_varptr(p, (oberon_expr_t *) expr -> item.parent);
1060 break;
1061 default:
1062 gen_error("push_varptr: wat %i", expr -> item.mode);
1063 break;
1067 static void
1068 jvm_generate_call_proc(gen_proc_t * p, oberon_item_t * desig)
1070 assert(desig -> var == NULL);
1071 assert(desig -> mode == MODE_CALL);
1073 bool direct_call = false;
1074 if(desig -> parent -> mode == MODE_VAR)
1076 if(desig -> parent -> var -> class == OBERON_CLASS_PROC)
1078 direct_call = true;
1082 oberon_type_t * procsig;
1083 procsig = desig -> parent -> result;
1085 if(direct_call == false)
1087 /* Загружаем указатель на процедуру */
1088 push_item(p, desig -> parent);
1091 if(direct_call)
1093 jvm_generate_push_static_links(p, desig -> parent -> var);
1096 int args_cells = 0;
1097 int result_cells = jvm_cell_size_for_type(procsig -> base);
1099 int num = desig -> num_args;
1100 oberon_expr_t * arg = desig -> args;
1101 oberon_object_t * param = procsig -> decl;
1102 for(int i = 0; i < num; i++)
1104 if(param -> class == OBERON_CLASS_VAR_PARAM)
1106 args_cells += 2;
1107 push_varptr(p, arg);
1109 else
1111 args_cells += jvm_cell_size_for_type(arg -> result);
1112 push_expr(p, arg);
1114 arg = arg -> next;
1115 param = param -> next;
1118 if(direct_call)
1120 char * full_name = jvm_get_field_full_name(desig -> parent -> var);
1121 char * signature = jvm_get_local_procedure_signature(desig -> parent -> var);
1122 jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", full_name, signature);
1124 else
1126 char * cname = jvm_get_class_full_name(procsig);
1127 char * signature = jvm_get_procedure_signature(procsig);
1128 jvm_generate(p, 1 + args_cells, result_cells, "invokevirtual %s/invoke%s", cname, signature);
1132 void
1133 oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig)
1135 assert(desig -> is_item);
1136 jvm_generate_call_proc(ctx -> mod -> gen_mod -> class -> p, (oberon_item_t *) desig);
1139 void
1140 oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr)
1142 gen_module_t * m;
1143 gen_proc_t * p;
1144 char prefix;
1145 int cell_size;
1147 m = ctx -> mod -> gen_mod;
1148 p = m -> class -> p;
1150 if(expr)
1152 push_expr(p, expr);
1153 prefix = jvm_get_prefix(expr -> result);
1154 cell_size = jvm_cell_size_for_type(expr -> result);
1155 jvm_generate(p, cell_size, 0, "%creturn", prefix);
1157 else
1159 jvm_generate(p, 0, 0, "return");
1163 static void
1164 jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
1166 assert(type -> class == OBERON_TYPE_POINTER);
1168 for(int i = 0; i < num; i++)
1170 push_expr(p, arg);
1171 arg = arg -> next;
1174 jvm_generate_new(p, type -> base, num);
1177 static void
1178 jvm_generate_cast_prefix(gen_proc_t * p, char prefix, char postfix)
1180 if(prefix == postfix)
1182 return;
1185 int from_cell_size = jvm_cell_size_for_postfix(prefix);
1186 int to_cell_size = jvm_cell_size_for_postfix(postfix);
1187 jvm_generate(p, from_cell_size, to_cell_size, "%c2%c", prefix, postfix);
1190 static void
1191 jvm_generate_cast_type(gen_proc_t * p, oberon_type_t * from, oberon_type_t * to)
1193 if(to -> class == OBERON_TYPE_RECORD || to -> class == OBERON_TYPE_POINTER)
1195 if(to -> class == OBERON_TYPE_POINTER && to -> base -> class == OBERON_TYPE_RECORD)
1197 char * full_name = jvm_get_class_full_name(to);
1198 jvm_generate(p, 1, 1, "checkcast %s", full_name);
1201 else
1203 char prefix = jvm_get_prefix(from);
1204 char postfix = jvm_get_postfix(to);
1205 jvm_generate_cast_prefix(p, prefix, postfix);
1209 static void
1210 jvm_generate_hard_cast_type(gen_proc_t * p, oberon_type_t * from, oberon_type_t * to)
1212 if(from -> class == OBERON_TYPE_REAL
1213 && (to -> class == OBERON_TYPE_INTEGER || to -> class == OBERON_TYPE_SYSTEM_BYTE))
1215 char postfix = jvm_get_postfix(to);
1216 if(from -> size <= 4)
1218 jvm_generate(p, 1, 1, "invokestatic java/lang/Float/floatToIntBits(F)I");
1219 jvm_generate_cast_prefix(p, 'i', postfix);
1221 else
1223 jvm_generate(p, 2, 2, "invokestatic java/lang/Double/doubleToIntBits(D)J");
1224 jvm_generate_cast_prefix(p, 'l', postfix);
1227 else if((from -> class == OBERON_TYPE_INTEGER || from -> class == OBERON_TYPE_SYSTEM_BYTE)
1228 && to -> class == OBERON_TYPE_REAL)
1230 char prefix = jvm_get_prefix(from);
1231 if(to -> size <= 4)
1233 jvm_generate_cast_prefix(p, prefix, 'i');
1234 jvm_generate(p, 1, 1, "invokestatic java/lang/Float/intBitsToFloat(I)F");
1236 else
1238 jvm_generate_cast_prefix(p, prefix, 'l');
1239 jvm_generate(p, 2, 2, "invokestatic java/lang/Double/longBitsToDouble(J)D");
1242 else
1244 jvm_generate_cast_type(p, from, to);
1248 static void
1249 push_item(gen_proc_t * p, oberon_item_t * item)
1251 switch(item -> mode)
1253 case MODE_VAR:
1254 if(item -> var -> class == OBERON_CLASS_PROC)
1256 jvm_generate_push_procedure_pointer(p, item -> var);
1258 else
1260 jvm_generate_load(p, item -> var -> gen_var);
1262 break;
1263 case MODE_INTEGER:
1264 case MODE_BOOLEAN:
1265 case MODE_CHAR:
1266 case MODE_SET:
1267 jvm_generate_push_int_size(p, item -> integer, item -> result -> size);
1268 break;
1269 case MODE_CALL:
1270 jvm_generate_call_proc(p, item);
1271 break;
1272 case MODE_INDEX:
1274 char postfix = jvm_get_postfix(item -> result);
1275 int cell_size = jvm_cell_size_for_postfix(postfix);
1276 push_item(p, item -> parent);
1277 push_expr(p, item -> args);
1278 jvm_generate(p, 1 + 1, cell_size, "%caload", postfix);
1279 break;
1280 case MODE_FIELD:
1281 push_item(p, item -> parent);
1282 jvm_generate_load(p, item -> var -> gen_var);
1283 break;
1284 case MODE_DEREF:
1285 /* Все объекты в jvm представляются как указатели */
1286 push_item(p, item -> parent);
1287 break;
1288 case MODE_NIL:
1289 jvm_generate(p, 0, 1, "aconst_null");
1290 break;
1291 case MODE_NEW:
1292 jvm_generate_expr_new_pointer(p, item -> result, item -> num_args, item -> args);
1293 break;
1294 case MODE_REAL:
1295 jvm_generate_push_float(p, item -> real, item -> result -> size);
1296 break;
1297 case MODE_STRING:
1298 jvm_generate_push_string(p, item -> string, item -> result -> size);
1299 break;
1300 case MODE_LEN:
1301 push_item(p, item -> parent);
1302 jvm_generate_array_len(p, item -> integer);
1303 jvm_generate_cast_prefix(p, 'i', jvm_get_postfix(item -> result));
1304 break;
1305 default:
1306 gen_error("push_item: unk mode %i", item -> mode);
1307 break;
1311 static void
1312 jvm_generate_logical_not(gen_proc_t * p)
1314 int label_done = jvm_new_label_id(p);
1315 int label_false = jvm_new_label_id(p);
1317 jvm_generate(p, 1, 0, "ifne L%i", label_false);
1318 jvm_generate(p, 0, 1, "iconst_1");
1319 jvm_generate(p, 0, 0, "goto L%i", label_done);
1320 jvm_generate_label(p, label_false);
1321 jvm_generate(p, 0, 1, "iconst_0");
1322 jvm_generate_label(p, label_done);
1325 static void
1326 jvm_generate_abs(gen_proc_t * p, char prefix)
1328 char t = jvm_get_type_of_prefix(prefix);
1329 int cell_size = jvm_cell_size_for_postfix(prefix);
1330 jvm_generate(p, cell_size, cell_size, "invokestatic java/lang/Math/abs(%c)%c", t, t);
1333 static char *
1334 jvm_get_compare_postfix(int op)
1336 char * cmpop = "";
1337 switch(op)
1339 case OP_EQ:
1340 cmpop = "eq";
1341 break;
1342 case OP_NEQ:
1343 cmpop = "ne";
1344 break;
1345 case OP_LSS:
1346 cmpop = "lt";
1347 break;
1348 case OP_LEQ:
1349 cmpop = "le";
1350 break;
1351 case OP_GRT:
1352 cmpop = "gt";
1353 break;
1354 case OP_GEQ:
1355 cmpop = "ge";
1356 break;
1357 default:
1358 gen_error("jvm_generate_compare_op: wat");
1359 break;
1361 return cmpop;
1364 static void
1365 jvm_generate_compare_op(gen_proc_t * p, oberon_type_t * t, int op)
1367 char prefix = jvm_get_prefix(t);
1368 int label_true = jvm_new_label_id(p);
1369 int label_done = jvm_new_label_id(p);
1370 int cell_size = jvm_cell_size_for_type(t);
1371 char * cmpop = jvm_get_compare_postfix(op);
1373 if(prefix == 'l')
1375 jvm_generate(p, 2 * cell_size, 1, "lcmp");
1376 jvm_generate(p, 1, 1, "if%s L%i", cmpop, label_true);
1378 else if(prefix == 'f' || prefix == 'd')
1380 char fop;
1381 if(op == OP_EQ || op == OP_NEQ || op == OP_GRT || op == OP_GEQ)
1383 fop = 'l';
1385 else
1387 fop = 'g';
1389 jvm_generate(p, 2 * cell_size, 1, "%ccmp%c", prefix, fop);
1390 jvm_generate(p, 1, 1, "if%s L%i", cmpop, label_true);
1392 else if(prefix == 'a')
1394 if(t -> class == OBERON_TYPE_STRING
1395 || (t -> class == OBERON_TYPE_POINTER && t -> base -> class == OBERON_TYPE_CHAR))
1397 jvm_generate(p, 2, 1, "invokestatic SYSTEM/STRCMP([B[B)I");
1398 jvm_generate(p, 1, 0, "if%s L%i", cmpop, label_true);
1400 else
1402 jvm_generate(p, 1, 0, "if_acmp%s L%i", cmpop, label_true);
1405 else
1407 jvm_generate(p, 2 * cell_size, 0, "if_%ccmp%s L%i", prefix, cmpop, label_true);
1410 jvm_generate(p, 0, 1, "iconst_0");
1411 jvm_generate(p, 0, 0, "goto L%i", label_done);
1412 jvm_generate_label(p, label_true);
1413 jvm_generate(p, 0, 1, "iconst_1");
1414 jvm_generate_label(p, label_done);
1417 static void
1418 jvm_generate_operator(gen_proc_t * p, oberon_type_t * t, int op)
1420 char prefix = jvm_get_prefix(t);
1421 int cell_size = jvm_cell_size_for_type(t);
1422 switch(op)
1424 case OP_UNARY_MINUS:
1425 jvm_generate(p, cell_size, cell_size, "%cneg", prefix);
1426 break;
1427 case OP_COMPLEMENTATION:
1428 jvm_generate_push_int_size(p, -1, t -> size);
1429 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1430 break;
1431 case OP_LOGIC_NOT:
1432 jvm_generate_logical_not(p);
1433 break;
1434 case OP_ABS:
1435 jvm_generate_abs(p, prefix);
1436 break;
1437 case OP_CAP:
1438 jvm_generate(p, cell_size, cell_size, "invokestatic java/lang/Character/toUpperCase(I)I");
1439 break;
1441 case OP_ADD:
1442 jvm_generate(p, 2 * cell_size, cell_size, "%cadd", prefix);
1443 break;
1444 case OP_SUB:
1445 jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix);
1446 break;
1447 case OP_MUL:
1448 jvm_generate(p, 2 * cell_size, cell_size, "%cmul", prefix);
1449 break;
1450 case OP_DIV:
1451 jvm_generate(p, 2 * cell_size, cell_size, "%cdiv", prefix);
1452 break;
1453 case OP_MOD:
1454 jvm_generate(p, 2 * cell_size, cell_size, "%crem", prefix);
1455 break;
1456 case OP_UNION:
1457 jvm_generate(p, 2 * cell_size, cell_size, "%cor", prefix);
1458 break;
1459 case OP_INTERSECTION:
1460 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1461 break;
1462 case OP_DIFFERENCE:
1463 /* (a - b) == a & ~b */
1464 jvm_generate_push_int_size(p, -1, t -> size);
1465 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1466 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1467 break;
1468 case OP_SYM_DIFFERENCE:
1469 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1470 break;
1472 case OP_EQ:
1473 case OP_NEQ:
1474 case OP_LSS:
1475 case OP_LEQ:
1476 case OP_GRT:
1477 case OP_GEQ:
1478 jvm_generate_compare_op(p, t, op);
1479 break;
1480 default:
1481 gen_error("jvm_generate_operator: unk op %i", op);
1482 break;
1483 }
1486 static void
1487 jvm_generate_logical_or(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1489 int label_calc_b = jvm_new_label_id(p);
1490 int label_done = jvm_new_label_id(p);
1492 /* a OR b -- если a, то TRUE, иначе b */
1494 push_expr(p, a);
1495 jvm_generate(p, 1, 0, "ifeq L%i", label_calc_b);
1496 jvm_generate(p, 0, 1, "iconst_1");
1497 jvm_generate(p, 0, 0, "goto L%i", label_done);
1498 jvm_generate_label(p, label_calc_b);
1499 push_expr(p, b);
1500 jvm_generate_label(p, label_done);
1503 static void
1504 jvm_generate_logical_and(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1506 int label_false = jvm_new_label_id(p);
1507 int label_done = jvm_new_label_id(p);
1509 /* a AND b -- если a, то b, иначе FALSE */
1511 push_expr(p, a);
1512 jvm_generate(p, 1, 0, "ifeq L%i", label_false);
1513 push_expr(p, b);
1514 jvm_generate(p, 0, 0, "goto L%i", label_done);
1515 jvm_generate_label(p, label_false);
1516 jvm_generate(p, 0, 1, "iconst_0");
1517 jvm_generate_label(p, label_done);
1520 static void
1521 jvm_generate_range(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1523 /* { a } == 1 << a */
1524 /* { a..b } == (a <= b) ? ((2 << b) - (1 << a)) : (0); */
1526 char prefix;
1527 int cell_size;
1528 oberon_type_t * type;
1529 gen_var_t * ra;
1530 gen_var_t * rb;
1531 int label_else;
1532 int label_end;
1534 type = a -> result;
1535 cell_size = jvm_cell_size_for_type(type);
1536 prefix = jvm_get_prefix(type);
1538 if(b == NULL)
1540 jvm_generate_push_int_size(p, 1, type -> size);
1541 push_expr(p, a);
1542 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1544 else
1546 ra = jvm_create_function_temp_var(p, type -> gen_type);
1547 rb = jvm_create_function_temp_var(p, type -> gen_type);
1548 jvm_generate_var(ra);
1549 jvm_generate_var(rb);
1550 jvm_generate_var_initialization(p, ra);
1551 jvm_generate_var_initialization(p, rb);
1552 label_else = jvm_new_label_id(p);
1553 label_end = jvm_new_label_id(p);
1555 push_expr(p, a);
1556 jvm_generate_store(p, ra);
1557 push_expr(p, b);
1558 jvm_generate_store(p, rb);
1560 jvm_generate_load(p, ra);
1561 jvm_generate_load(p, rb);
1562 jvm_generate(p, 2 * cell_size, 0, "if_%ccmpgt L%i", prefix, label_else);
1564 jvm_generate_push_int_size(p, 2, type -> size);
1565 jvm_generate_load(p, rb);
1566 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1567 jvm_generate_push_int_size(p, 2, type -> size);
1568 jvm_generate_load(p, ra);
1569 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1570 jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix);
1571 jvm_generate(p, 0, 0, "goto L%i", label_end);
1573 jvm_generate_label(p, label_else);
1574 jvm_generate_push_int_size(p, 0, type -> size);
1575 jvm_generate_label(p, label_end);
1578 /* TODO free registers */
1581 static void
1582 jvm_generate_in(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1584 oberon_type_t * t = a -> result;
1585 int cell_size = jvm_cell_size_for_type(t);
1586 char prefix = jvm_get_prefix(t);
1587 int label_else = jvm_new_label_id(p);
1588 int label_end = jvm_new_label_id(p);
1590 /* (a IN b) == (1 << a) & b */
1591 jvm_generate_push_int_size(p, 1, t -> size);
1592 push_expr(p, a);
1593 jvm_generate(p, 2 * cell_size, cell_size, "%cshl", prefix);
1594 push_expr(p, b);
1595 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1597 if(cell_size > 1)
1599 jvm_generate(p, cell_size, 0, "lcmp");
1602 jvm_generate(p, 1, 0, "ifeq L%i", label_else);
1603 jvm_generate(p, 0, 1, "iconst_1");
1604 jvm_generate(p, 0, 0, "goto L%i", label_end);
1605 jvm_generate_label(p, label_else);
1606 jvm_generate(p, 0, 1, "iconst_0");
1607 jvm_generate_label(p, label_end);
1610 static void
1611 jvm_generate_shift(gen_proc_t * p, int op, oberon_expr_t * a, oberon_expr_t * b)
1613 oberon_type_t * t = a -> result;
1614 int cell_size = jvm_cell_size_for_type(t);
1615 char prefix = jvm_get_prefix(t);
1616 char dt = jvm_get_type_of_prefix(prefix);
1618 push_expr(p, a);
1619 jvm_generate_cast_type(p, a -> result, t);
1620 push_expr(p, b);
1621 jvm_generate_cast_type(p, b -> result, t);
1623 char * opname;
1624 switch(op)
1626 case OP_ASH:
1627 opname = "ASH";
1628 break;
1629 case OP_LSH:
1630 opname = "LSH";
1631 break;
1632 case OP_ROT:
1633 opname = "ROT";
1634 break;
1635 default:
1636 gen_error("jvm_generate_shift: invalid op %i", op);
1639 jvm_generate(p, 2 * cell_size, cell_size, "invokestatic SYSTEM/%s(%c%c)%c", opname, dt, dt, dt);
1642 static void
1643 jvm_generate_entier(gen_proc_t * p, oberon_expr_t * x, oberon_type_t * res)
1645 char prefix = jvm_get_prefix(x -> result);
1646 char postfix = jvm_get_postfix(res);
1648 push_expr(p, x);
1649 jvm_generate_cast_prefix(p, prefix, 'd');
1650 jvm_generate(p, 2, 2, "invokestatic java/lang/Math/floor(D)D");
1651 jvm_generate_cast_prefix(p, 'd', postfix);
1654 static void
1655 push_operator(gen_proc_t * p, oberon_oper_t * oper)
1657 oberon_type_t * preq = oper -> left -> result;
1658 int op = oper -> op;
1659 switch(op)
1661 case OP_CAST:
1662 push_expr(p, oper -> left);
1663 jvm_generate_cast_type(p, oper -> left -> result, oper -> result);
1664 break;
1665 case OP_HARDCAST:
1666 push_expr(p, oper -> left);
1667 jvm_generate_hard_cast_type(p, oper -> left -> result, oper -> result);
1668 break;
1669 case OP_COMPLEMENTATION:
1670 case OP_UNARY_MINUS:
1671 case OP_LOGIC_NOT:
1672 case OP_ABS:
1673 case OP_CAP:
1674 push_expr(p, oper -> left);
1675 jvm_generate_operator(p, preq, op);
1676 break;
1677 case OP_ENTIER:
1678 jvm_generate_entier(p, oper -> left, oper -> result);
1679 break;
1681 case OP_ADD:
1682 case OP_SUB:
1683 case OP_MUL:
1684 case OP_DIV:
1685 case OP_MOD:
1687 case OP_UNION:
1688 case OP_INTERSECTION:
1689 case OP_DIFFERENCE:
1690 case OP_SYM_DIFFERENCE:
1692 case OP_EQ:
1693 case OP_NEQ:
1694 case OP_LSS:
1695 case OP_LEQ:
1696 case OP_GRT:
1697 case OP_GEQ:
1698 push_expr(p, oper -> left);
1699 push_expr(p, oper -> right);
1700 jvm_generate_operator(p, preq, op);
1701 break;
1703 case OP_LOGIC_OR:
1704 jvm_generate_logical_or(p, oper -> left, oper -> right);
1705 break;
1706 case OP_LOGIC_AND:
1707 jvm_generate_logical_and(p, oper -> left, oper -> right);
1708 break;
1710 case OP_IS:
1711 preq = oper -> right -> result;
1712 char * cname = jvm_get_class_full_name(preq);
1713 push_expr(p, oper -> left);
1714 jvm_generate(p, 1, 1, "instanceof %s", cname);
1715 break;
1716 case OP_RANGE:
1717 jvm_generate_range(p, oper -> left, oper -> right);
1718 break;
1719 case OP_IN:
1720 jvm_generate_in(p, oper -> left, oper -> right);
1721 break;
1723 case OP_ASH:
1724 case OP_LSH:
1725 case OP_ROT:
1726 jvm_generate_shift(p, op, oper -> left, oper -> right);
1727 break;
1728 default:
1729 gen_error("push_oper: unk op %i", op);
1730 break;
1734 static void
1735 push_expr(gen_proc_t * p, oberon_expr_t * expr)
1737 if(expr -> is_item)
1739 push_item(p, (oberon_item_t *) expr);
1741 else
1743 push_operator(p, (oberon_oper_t *) expr);
1747 static void
1748 store_expr(gen_proc_t * p, oberon_expr_t * dst, oberon_expr_t * src)
1750 assert(dst -> is_item);
1751 oberon_item_t * item = (oberon_item_t *) dst;
1753 if(dst -> result -> class == OBERON_TYPE_ARRAY
1754 || src -> result -> class == OBERON_TYPE_ARRAY)
1756 push_expr(p, dst);
1757 push_expr(p, src);
1758 jvm_generate_copy_array(p, dst -> result);
1760 else if(dst -> result -> class == OBERON_TYPE_RECORD
1761 || src -> result -> class == OBERON_TYPE_RECORD)
1763 push_expr(p, dst);
1764 push_expr(p, src);
1765 jvm_generate_copy_record(p, dst -> result);
1767 else switch(item -> mode)
1769 case MODE_VAR:
1770 jvm_generate_ldst_prepare(p, item -> var -> gen_var);
1771 push_expr(p, src);
1772 jvm_generate_store(p, item -> var -> gen_var);
1773 break;
1774 case MODE_INDEX:
1776 char postfix = jvm_get_postfix(src -> result);
1777 int cell_size = jvm_cell_size_for_postfix(postfix);
1778 assert(item -> parent -> is_item);
1779 push_item(p, (oberon_item_t *) item -> parent);
1780 push_expr(p, item -> args);
1781 push_expr(p, src);
1782 jvm_generate(p, 1 + 1 + cell_size, 0, "%castore", postfix);
1783 break;
1784 case MODE_FIELD:
1785 assert(item -> parent -> is_item);
1786 push_item(p, (oberon_item_t *) item -> parent);
1787 jvm_generate_ldst_prepare(p, item -> var -> gen_var);
1788 push_expr(p, src);
1789 jvm_generate_store(p, item -> var -> gen_var);
1790 break;
1791 default:
1792 gen_error("store_expr: unk mode %i", item -> mode);
1793 break;
1797 void
1798 oberon_generate_assign(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_t * dst)
1800 gen_module_t * m;
1801 gen_proc_t * p;
1803 m = ctx -> mod -> gen_mod;
1804 p = m -> class -> p;
1806 store_expr(p, dst, src);
1809 void
1810 oberon_generate_copy(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_t * dst)
1812 gen_module_t * m;
1813 gen_proc_t * p;
1814 char * desc;
1816 m = ctx -> mod -> gen_mod;
1817 p = m -> class -> p;
1819 push_expr(p, src);
1820 push_expr(p, dst);
1822 desc = jvm_get_descriptor(dst -> result);
1824 jvm_generate(p, 2, 0, "invokestatic SYSTEM/COPY(%s%s)V", desc, desc);
1827 void
1828 oberon_generate_assert(oberon_context_t * ctx, oberon_expr_t * cond)
1830 gen_module_t * m;
1831 gen_proc_t * p;
1833 m = ctx -> mod -> gen_mod;
1834 p = m -> class -> p;
1836 push_expr(p, cond);
1837 jvm_generate(p, 1, 0, "invokestatic SYSTEM/ASSERT(Z)V");
1840 void
1841 oberon_generate_assert_n(oberon_context_t * ctx, oberon_expr_t * cond, int64_t n)
1843 gen_module_t * m;
1844 gen_proc_t * p;
1846 m = ctx -> mod -> gen_mod;
1847 p = m -> class -> p;
1849 push_expr(p, cond);
1850 jvm_generate_push_int_size(p, n, 8);
1851 jvm_generate(p, 1 + 2, 0, "invokestatic SYSTEM/ASSERT(ZJ)V");
1854 void
1855 oberon_generate_trap(oberon_context_t * ctx, int64_t n)
1857 gen_module_t * m;
1858 gen_proc_t * p;
1860 m = ctx -> mod -> gen_mod;
1861 p = m -> class -> p;
1863 jvm_generate_push_int_size(p, n, 8);
1864 jvm_generate(p, 2, 0, "invokestatic SYSTEM/TRAP(J)V");
1867 void
1868 oberon_generate_halt(oberon_context_t * ctx, int64_t n)
1870 gen_module_t * m;
1871 gen_proc_t * p;
1873 m = ctx -> mod -> gen_mod;
1874 p = m -> class -> p;
1876 jvm_generate_push_int_size(p, n, 8);
1877 jvm_generate(p, 2, 0, "invokestatic SYSTEM/HALT(J)V");
1880 void
1881 oberon_set_out_directory(oberon_context_t * ctx, const char * path)
1883 gen_context_t * c;
1885 c = ctx -> gen_context;
1887 c -> dir = new_string(path);