DEADSOFTWARE

GIT / REPO / FRED-BOY
Projects / dsw-obn.git / blob
summary | shortlog | log | commit | commitdiff | tree
history | raw | HEAD
JVM: Добавлено копирование массивов
[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>
8
9 #include <gc.h>
10
11 #include "../../../include/oberon.h"
12 #include "../../oberon-internals.h"
13 #include "generator-jvm.h"
14 #include "generator-jvm-basic.h"
15
16 // ==========================================
17 // ==========================================
18 // ==========================================
19
20 gen_proc_t *
21 jvm_create_proc(struct gen_class * class)
22 {
23 gen_proc_t * p = GC_MALLOC(sizeof *p);
24 memset(p, 0, sizeof *p);
25
26 struct gen_register_file * rf = GC_MALLOC(sizeof *rf);
27 memset(rf, 0, sizeof *rf);
28
29 struct gen_stack * stack = GC_MALLOC(sizeof *stack);
30 memset(rf, 0, sizeof *stack);
31
32 p -> rf = rf;
33 p -> stack = stack;
34 p -> class = class;
35 p -> label_id = 0;
36
37 return p;
38 }
39
40 void
41 jvm_generate_function_header(gen_proc_t * p, char * access, char * name, char * signature)
42 {
43 struct gen_class * class;
44 class = p -> class;
45
46 /* Делаем процедуру текущей в этом классе */
47 assert(class -> p == NULL);
48 class -> p = p;
49
50 fprintf(class -> fp, ".method %s %s%s\n", access, name, signature);
51 fprintf(class -> fp, " start:\n");
52 }
53
54 void
55 jvm_generate_function_end(gen_proc_t * p)
56 {
57 struct gen_class * class;
58 class = p -> class;
59
60 assert(class -> p);
61 class -> p = NULL;
62
63 int pointer = p -> stack -> pointer;
64 int max_pointer = p -> stack -> max_pointer;
65 int locals = p -> rf -> num_used;
66 int max_locals = p -> rf -> max_used;
67
68 fprintf(class -> fp, " .limit stack %i \t; current(%i)\n", max_pointer, pointer);
69 fprintf(class -> fp, " .limit locals %i \t; current(%i)\n", max_locals, locals);
70 fprintf(class -> fp, " end:\n");
71 fprintf(class -> fp, ".end method\n\n");
72 }
73
74 struct gen_class *
75 jvm_create_class(char * full_name)
76 {
77 struct gen_class * class = GC_MALLOC(sizeof *class);
78 memset(class, 0, sizeof *class);
79
80 char * fname = new_string("%s.j", full_name);
81 class -> full_name = new_string(full_name);
82 class -> fp = fopen(fname, "w");
83 assert(class -> fp);
84
85 return class;
86 }
87
88 static void
89 jvm_destroy_class(struct gen_class * class)
90 {
91 assert(class -> p == NULL);
92 fclose(class -> fp);
93 }
94
95 static void
96 jvm_stack_push(gen_proc_t * p, unsigned size)
97 {
98 p -> stack -> pointer += size;
99 if(p -> stack -> pointer > p -> stack -> max_pointer)
100 {
101 p -> stack -> max_pointer = p -> stack -> pointer;
102 }
103 }
104
105 static void
106 jvm_stack_pop(gen_proc_t * p, unsigned size)
107 {
108 p -> stack -> pointer -= size;
109
110 if(p -> stack -> pointer < 0)
111 {
112 printf("WARING: stack pointer %i\n", p -> stack -> pointer);
113 }
114 }
115
116 static void
117 jvm_generate(gen_proc_t * p, unsigned get, unsigned push, char * format, ...)
118 {
119 va_list ptr;
120 va_start(ptr, format);
121
122 jvm_stack_pop(p, get);
123 fprintf(p -> class -> fp, " ");
124 vfprintf(p -> class -> fp, format, ptr);
125 jvm_stack_push(p, push);
126 fprintf(p -> class -> fp, " \t ;>>>> %i -- %i : current_stack(%i)\n", get, push, p -> stack -> pointer);
127
128 va_end(ptr);
129 }
130
131 static void
132 jvm_generate_push_int(gen_proc_t * p, int64_t i)
133 {
134 if(i == -1)
135 {
136 jvm_generate(p, 0, 1, "iconst_m1");
137 }
138 else if(i >= 0 && i <= 5)
139 {
140 jvm_generate(p, 0, 1, "iconst_%li", i);
141 }
142 else if(i >= -128 && i <= 127)
143 {
144 jvm_generate(p, 0, 1, "bipush %li", i);
145 }
146 else if(i >= -32768 && i <= 32767)
147 {
148 jvm_generate(p, 0, 1, "sipush %li", i);
149 }
150 else if(i >= -2147483648 && i <= 2147483647)
151 {
152 jvm_generate(p, 0, 1, "ldc %li", i);
153 }
154 else
155 {
156 jvm_generate(p, 0, 2, "ldc2 %li", i);
157 }
158 }
159
160 static void
161 jvm_generate_push_float(gen_proc_t * p, double f, int size)
162 {
163 if(size <= 4)
164 {
165 if(f == 0.0)
166 {
167 jvm_generate(p, 0, 1, "fconst_0");
168 }
169 if(f == 1.0)
170 {
171 jvm_generate(p, 0, 1, "fconst_1");
172 }
173 if(f == 2.0)
174 {
175 jvm_generate(p, 0, 1, "fconst_2");
176 }
177 else
178 {
179 jvm_generate(p, 0, 1, "ldc %lf", f);
180 }
181 }
182 else
183 {
184 if(f == 0.0)
185 {
186 jvm_generate(p, 0, 2, "dconst_0");
187 }
188 if(f == 1.0)
189 {
190 jvm_generate(p, 0, 2, "dconst_1");
191 }
192 else
193 {
194 jvm_generate(p, 0, 2, "ldc2 %lf", f);
195 }
196 }
197 }
198
199 static gen_var_t *
200 oberon_generator_new_var()
201 {
202 gen_var_t * v = GC_MALLOC(sizeof *v);
203 memset(v, 0, sizeof *v);
204 return v;
205 }
206
207 static void
208 jvm_generate_and_init_global_var(gen_var_t * v, struct gen_class * class, char * access, char * name, char * desc)
209 {
210 assert(class -> p == NULL);
211 fprintf(class -> fp, ".field %s static %s %s\n\n", access, name, desc);
212
213 v -> storage = JVM_STORAGE_STATIC;
214 v -> full_name = new_string("%s/%s", class -> full_name, name);
215 v -> desc = new_string(desc);
216 }
217
218 static void
219 jvm_generate_and_init_field(gen_var_t * v, struct gen_class * class, char * access, char * name, char * desc)
220 {
221 assert(class -> p == NULL);
222 fprintf(class -> fp, ".field %s %s %s\n\n", access, name, desc);
223
224 v -> storage = JVM_STORAGE_FIELD;
225 v -> full_name = new_string("%s/%s", class -> full_name, name);
226 v -> desc = new_string(desc);
227 }
228
229 static void
230 jvm_generate_and_init_local_var(gen_var_t * v, gen_proc_t * p, bool wide)
231 {
232 v -> storage = JVM_STORAGE_REGISTER;
233 v -> reg = jvm_alloc_register_untyped(p -> rf, wide);
234 }
235
236 static void
237 jvm_generate_and_init_named_local_var(gen_var_t * v, gen_proc_t * p, bool wide, char * name, char * desc)
238 {
239 jvm_generate_and_init_local_var(v, p, wide);
240 v -> desc = new_string(desc);
241 jvm_generate(p, 0, 0, ".var %i is %s %s from start to end", v -> reg, name, desc);
242 }
243
244 static int
245 jvm_new_label_id(gen_proc_t * p)
246 {
247 int label_id = p -> label_id;
248 p -> label_id += 1;
249 return label_id;
250 }
251
252 static char *
253 jvm_get_label_name(int label_id)
254 {
255 return new_string("L%i", label_id);
256 }
257
258 static void
259 jvm_generate_label(gen_proc_t * p, int label_id)
260 {
261 jvm_generate(p, 0, 0, "L%i:", label_id);
262 }
263
264 static void
265 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num);
266
267 /*
268 * Функция jvm_generate_static_array_initialization генерирует код для
269 * статической инициализации массива. На входе массив, на выходе тот же массив.
270 * (arrayref -- arrayref)
271 */
272
273 static void
274 jvm_generate_static_array_initialization(gen_proc_t * p, oberon_type_t * type)
275 {
276 /* for(int i = 0; i < a.length; i++) */
277 /* a[i] := new(base); */
278 /* init(a[i]) */
279
280 /* a := pop(array) */
281 gen_var_t * va = oberon_generator_new_var();
282 jvm_generate_and_init_local_var(va, p, false);
283 int reg_a = va -> reg;
284 /* ссылка уже на стеке */
285 jvm_generate(p, 1, 0, "astore %i", reg_a);
286
287 /* int i = 0; */
288 gen_var_t * vi = oberon_generator_new_var();
289 jvm_generate_and_init_local_var(vi, p, false);
290 int reg_i = vi -> reg;
291 jvm_generate(p, 0, 1, "iconst_0");
292 jvm_generate(p, 1, 0, "istore %i", reg_i);
293
294 /* int l := a.length */
295 gen_var_t * vl = oberon_generator_new_var();
296 jvm_generate_and_init_local_var(vl, p, false);
297 int reg_l = vl -> reg;
298 jvm_generate(p, 0, 1, "aload %i", reg_a);
299 jvm_generate(p, 1, 1, "arraylength");
300 jvm_generate(p, 1, 0, "istore %i", reg_l);
301
302 int label_start = jvm_new_label_id(p);
303 int label_end = jvm_new_label_id(p);
304
305 /* start: */
306 /* if (i >= l) goto end; */
307 /* body*/
308 /* i += 1;*/
309 /* goto start;*/
310 /* end: */
311 /* push a */
312
313 jvm_generate_label(p, label_start);
314 jvm_generate(p, 0, 1, "iload %i", reg_i);
315 jvm_generate(p, 0, 1, "iload %i", reg_l);
316 jvm_generate(p, 2, 0, "if_icmpge L%i", label_end);
317
318 if(type -> base -> class == OBERON_TYPE_ARRAY)
319 {
320 /* Инициализируем следующую размерность */
321 jvm_generate(p, 0, 1, "aload %i", reg_a);
322 jvm_generate(p, 0, 1, "iload %i", reg_i);
323 jvm_generate(p, 0, 1, "aaload");
324 jvm_generate_static_array_initialization(p, type -> base);
325 jvm_generate(p, 1, 0, "pop");
326 }
327 else if(type -> base -> class == OBERON_TYPE_RECORD)
328 {
329 jvm_generate(p, 0, 1, "aload %i", reg_a);
330 jvm_generate(p, 0, 1, "iload %i", reg_i);
331 jvm_generate_new(p, type -> base, 0);
332 jvm_generate(p, 3, 0, "aastore");
333 }
334 else
335 {
336 assert(0);
337 }
338
339 /* i += 1; */
340 /* goto start */
341
342 jvm_generate(p, 0, 0, "iinc %i %i", reg_i, 1);
343 jvm_generate(p, 0, 0, "goto L%i", label_start);
344 jvm_generate_label(p, label_end);
345
346 /* push(a) */
347 jvm_generate(p, 0, 1, "aload %i", reg_a);
348
349 /* TODO освобождение регистров */
350 }
351
352 static void
353 jvm_generate_new(gen_proc_t * p, oberon_type_t * type, int num)
354 {
355 int dim;
356 char * cname;
357 char * desc;
358 oberon_type_t * base;
359
360 switch(type -> class)
361 {
362 case OBERON_TYPE_INTEGER:
363 case OBERON_TYPE_BOOLEAN:
364 case OBERON_TYPE_PROCEDURE:
365 case OBERON_TYPE_REAL:
366 case OBERON_TYPE_POINTER:
367 gen_error("jvm_generate_new_static: static alocation not allowed");
368 break;
369 case OBERON_TYPE_RECORD:
370 cname = jvm_get_class_full_name(type);
371 jvm_generate(p, 0, 1, "new %s", cname);
372 jvm_generate(p, 1, 2, "dup");
373 jvm_generate(p, 1, 0, "invokespecial %s/<init>()V", cname);
374 break;
375 case OBERON_TYPE_ARRAY:
376 dim = 0;
377 base = type;
378 desc = jvm_get_descriptor(type);
379 while(base -> class == OBERON_TYPE_ARRAY)
380 {
381 if(num > 0)
382 {
383 assert(base -> size == 0);
384 num -= 1;
385 }
386 else
387 {
388 assert(base -> size > 0);
389 jvm_generate_push_int(p, base -> size);
390 }
391 dim += 1;
392 base = base -> base;
393 }
394
395 assert(num == 0);
396 jvm_generate(p, dim, 1, "multianewarray %s %i", desc, dim);
397
398 if(base -> class == OBERON_TYPE_RECORD)
399 {
400 jvm_generate_static_array_initialization(p, type);
401 }
402 break;
403 default:
404 gen_error("jvm_generate_new_static: unk type class %i", type -> class);
405 break;
406 }
407 }
408
409 static void
410 jvm_generate_load(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * src)
411 {
412 char prefix;
413 int cell_size;
414
415 cell_size = jvm_cell_size_for_type(src_type);
416
417 switch(src -> storage)
418 {
419 case JVM_STORAGE_REGISTER:
420 prefix = jvm_get_prefix(src_type);
421 jvm_generate(p, 0, cell_size, "%cload %i", prefix, src -> reg);
422 break;
423 case JVM_STORAGE_STATIC:
424 jvm_generate(p, 0, cell_size, "getstatic %s %s", src -> full_name, src -> desc);
425 break;
426 case JVM_STORAGE_FIELD:
427 jvm_generate(p, 1, cell_size, "getfield %s %s", src -> full_name, src -> desc);
428 break;
429 default:
430 gen_error("jvm_generate_load: unknow storage type %i", src -> storage);
431 break;
432 }
433 }
434
435 static void
436 jvm_generate_store(gen_proc_t * p, oberon_type_t * src_type, gen_var_t * dst)
437 {
438 char prefix;
439 int cell_size;
440
441 cell_size = jvm_cell_size_for_type(src_type);
442
443 switch(dst -> storage)
444 {
445 case JVM_STORAGE_REGISTER:
446 prefix = jvm_get_prefix(src_type);
447 jvm_generate(p, cell_size, 0, "%cstore %i", prefix, dst -> reg);
448 break;
449 case JVM_STORAGE_STATIC:
450 jvm_generate(p, cell_size, 0, "putstatic %s %s", dst -> full_name, dst -> desc);
451 break;
452 case JVM_STORAGE_FIELD:
453 jvm_generate(p, 1 + cell_size, 0, "putfield %s %s", dst -> full_name, dst -> desc);
454 break;
455 default:
456 gen_error("jvm_generate_store: unknow storage type %i", dst -> storage);
457 break;
458 }
459 }
460
461 /*
462 * Генерирует код для копирования полей из первой записи во вторую.
463 * ( aref_dst aref_src -- )
464 * dst := src;
465 */
466
467 static void
468 jvm_generate_copy_record(gen_proc_t * p, oberon_type_t * rec)
469 {
470 assert(rec -> class == OBERON_TYPE_RECORD);
471 char * desc = jvm_get_descriptor(rec);
472 char * cname = jvm_get_class_full_name(rec);
473 jvm_generate(p, 1 + 1, 0, "invokestatic %s/$COPY$(%s%s)V", cname, desc, desc);
474 }
475
476 /*
477 * Генерирует кода для копирования массивов.
478 * ( aref_dst aref_src -- )
479 * dst := src;
480 */
481
482 static void
483 jvm_generate_copy_array(gen_proc_t * p, oberon_type_t * arr)
484 {
485 int dim = 0;
486 oberon_type_t * base = arr;
487 while(base -> class == OBERON_TYPE_ARRAY)
488 {
489 dim += 1;
490 base = base -> base;
491 }
492
493 struct {
494 gen_var_t * index;
495 gen_var_t * length;
496 int start;
497 int end;
498 } loop[dim];
499
500 gen_var_t * dst = oberon_generator_new_var();
501 gen_var_t * src = oberon_generator_new_var();
502 jvm_generate_and_init_local_var(dst, p, false);
503 jvm_generate_and_init_local_var(src, p, false);
504
505 jvm_generate_store(p, arr, src);
506 jvm_generate_store(p, arr, dst);
507
508 /* Входящие параметры заграблены
509 * Теперь генерируем эквивалентный код:
510 * int i = 0;
511 * int len = dst.length
512 * while(i < len)
513 * {
514 * ...
515 * {
516 * copy from src[i, ...] to dst[i, ...];
517 * }
518 * ...
519 * i += 1;
520 * }
521 * Где "..." такой же код (начало и конец) для следующей размерности.
522 */
523
524 for(int i = 0; i < dim; i++)
525 {
526 loop[i].index = oberon_generator_new_var();
527 loop[i].length = oberon_generator_new_var();
528 loop[i].start = jvm_new_label_id(p);
529 loop[i].end = jvm_new_label_id(p);
530 jvm_generate_and_init_local_var(loop[i].index, p, false);
531 jvm_generate_and_init_local_var(loop[i].length, p, false);
532
533 /* TODO преределать через jvm_generate_load{{store}} */
534
535 jvm_generate(p, 0, 1, "iconst_0");
536 jvm_generate(p, 1, 0, "istore %i", loop[i].index -> reg);
537
538 jvm_generate_load(p, arr, dst);
539 jvm_generate(p, 1, 1, "arraylength");
540 jvm_generate(p, 1, 0, "istore %i", loop[i].length -> reg);
541
542 /* if(i >= len) goto end; */
543 jvm_generate_label(p, loop[i].start);
544 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
545 jvm_generate(p, 0, 1, "iload %i", loop[i].length -> reg);
546 jvm_generate(p, 2, 0, "if_icmpge L%i", loop[i].end);
547 }
548
549 if(base -> class == OBERON_TYPE_ARRAY)
550 {
551 // Вся эта шляпа уже должна знать о всех размерностях
552 gen_error("jvm_generate_copy_array: wat");
553 }
554 else if(base -> class == OBERON_TYPE_RECORD)
555 {
556 /* Получаем записи по индексам ( -- dst src ) */
557
558 jvm_generate_load(p, arr, dst);
559 for(int i = 0; i < dim; i++)
560 {
561 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
562 jvm_generate(p, 2, 1, "aaload");
563 }
564
565 jvm_generate_load(p, arr, src);
566 for(int i = 0; i < dim; i++)
567 {
568 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
569 jvm_generate(p, 2, 1, "aaload");
570 }
571
572 /* Копируем записи ( dst src -- ) */
573 jvm_generate_copy_record(p, base);
574 }
575 else
576 {
577 char postfix = jvm_get_postfix(base);
578 int cell_size = jvm_cell_size_for_postfix(postfix);
579
580 /* Получаем массивы и индексы ( -- dst i src i ) */
581
582 jvm_generate_load(p, arr, dst);
583 jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg);
584 for(int i = 1; i < dim; i++)
585 {
586 jvm_generate(p, 2, 1, "aaload");
587 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
588 }
589
590 jvm_generate_load(p, arr, src);
591 jvm_generate(p, 0, 1, "iload %i", loop[0].index -> reg);
592 for(int i = 1; i < dim; i++)
593 {
594 jvm_generate(p, 2, 1, "aaload");
595 jvm_generate(p, 0, 1, "iload %i", loop[i].index -> reg);
596 }
597
598 /* Копируем значения ( dst i src i -- ) */
599 jvm_generate(p, 2, cell_size, "%caload", postfix);
600 jvm_generate(p, 2 + cell_size, 0, "%castore", postfix);
601 }
602
603 for(int i = dim - 1; i >= 0; i--)
604 {
605 jvm_generate(p, 0, 0, "iinc %i 1", loop[i].index -> reg);
606 jvm_generate(p, 0, 0, "goto L%i", loop[i].start);
607 jvm_generate_label(p, loop[i].end);
608 }
609 }
610
611 // ==========================================
612 // ==========================================
613 // ==========================================
614
615 void
616 oberon_generator_init_context(oberon_context_t * ctx)
617 {
618 gen_context_t * gen_context = GC_MALLOC(sizeof *gen_context);
619 memset(gen_context, 0, sizeof *gen_context);
620
621 ctx -> gen_context = gen_context;
622 }
623
624 void
625 oberon_generator_destroy_context(oberon_context_t * ctx)
626 {
627 }
628
629 static void
630 oberon_generate_procedure_class(oberon_type_t * proc)
631 {
632 FILE * fp;
633 char * cname;
634 char * signature;
635 struct gen_class * class;
636
637 cname = jvm_get_class_full_name(proc);
638 class = jvm_create_class(cname);
639 fp = class -> fp;
640
641 fprintf(fp, ".source SYSTEM\n");
642 fprintf(fp, ".class public abstract %s\n", cname);
643 fprintf(fp, ".super java/lang/Object\n\n");
644
645 fprintf(fp, ".method public <init>()V\n");
646 fprintf(fp, " aload_0\n");
647 fprintf(fp, " invokespecial java/lang/Object/<init>()V\n");
648 fprintf(fp, " return\n");
649 fprintf(fp, ".end method\n\n");
650
651 signature = jvm_get_procedure_signature(proc);
652
653 fprintf(fp, ".method public abstract invoke%s\n", signature);
654 fprintf(fp, ".end method\n\n");
655
656 jvm_destroy_class(class);
657 }
658
659 static void
660 oberon_generate_record_class(gen_module_t * m, oberon_type_t * rec)
661 {
662 char * cname;
663 struct gen_class * class;
664
665 /* Устанавливаем новоый id */
666 rec -> gen_type -> rec_id = m -> rec_id;
667 m -> rec_id += 1;
668
669 cname = jvm_get_class_full_name(rec);
670 class = jvm_create_class(cname);
671
672 fprintf(class -> fp, ".source %s\n", rec -> module -> name);
673 fprintf(class -> fp, ".class public %s\n", cname);
674 fprintf(class -> fp, ".super java/lang/Object\n\n");
675
676 rec -> gen_type -> class = class;
677 }
678
679 void
680 oberon_generator_init_type(oberon_context_t * ctx, oberon_type_t * type)
681 {
682 gen_type_t * t = GC_MALLOC(sizeof *t);
683 memset(t, 0, sizeof *t);
684 type -> gen_type = t;
685
686 switch(type -> class)
687 {
688 case OBERON_TYPE_VOID:
689 case OBERON_TYPE_INTEGER:
690 case OBERON_TYPE_BOOLEAN:
691 case OBERON_TYPE_ARRAY:
692 case OBERON_TYPE_REAL:
693 break;
694 case OBERON_TYPE_RECORD:
695 ;
696 gen_module_t * m;
697 m = type -> module -> gen_mod;
698 oberon_generate_record_class(m, type);
699 break;
700 case OBERON_TYPE_PROCEDURE:
701 oberon_generate_procedure_class(type);
702 break;
703 case OBERON_TYPE_POINTER:
704 break;
705 default:
706 gen_error("oberon_generator_init_type: unk calss %i", type -> class);
707 break;
708 }
709 }
710
711 void
712 oberon_generator_init_record(oberon_context_t * ctx, oberon_type_t * rec)
713 {
714 struct gen_class * class;
715 class = rec -> gen_type -> class;
716
717 int num = rec -> num_decl;
718 oberon_object_t * field = rec -> decl;
719 for(int i = 0; i < num; i++)
720 {
721 gen_var_t * v = field -> gen_var;
722 char * name = field -> name;
723 char * desc = jvm_get_descriptor(field -> type);
724 jvm_generate_and_init_field(v, class, "public", name, desc);
725 field = field -> next;
726 }
727
728 /* Стандартный конструктор класса */
729 /* Инициализирует внутренние статические записи и массивы */
730 gen_proc_t * p = jvm_create_proc(class);
731 jvm_generate_function_header(p, "public", "<init>", "()V");
732 jvm_alloc_register_untyped(p -> rf, false);
733 jvm_generate(p, 0, 1, "aload_0");
734 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
735 num = rec -> num_decl;
736 field = rec -> decl;
737 for(int i = 0; i < num; i++)
738 {
739 if(field -> type -> class == OBERON_TYPE_RECORD
740 || field -> type -> class == OBERON_TYPE_ARRAY)
741 {
742 jvm_generate(p, 0, 1, "aload_0");
743 jvm_generate_new(p, field -> type, 0);
744 jvm_generate_store(p, field -> type, field -> gen_var);
745 }
746 field = field -> next;
747 }
748 jvm_generate(p, 0, 0, "return");
749 jvm_generate_function_end(p);
750
751 /* Метод для копирования полей класса */
752 /* reg0 == src -> reg1 == dst */
753 p = jvm_create_proc(class);
754 gen_var_t * copy_dst = oberon_generator_new_var();
755 gen_var_t * copy_src = oberon_generator_new_var();
756 char * desc = jvm_get_descriptor(rec);
757 char * signature = new_string("(%s%s)V", desc, desc);
758 jvm_generate_function_header(p, "public static", "$COPY$", signature);
759 jvm_generate_and_init_named_local_var(copy_dst, p, false, "dst", desc);
760 jvm_generate_and_init_named_local_var(copy_src, p, false, "src", desc);
761 num = rec -> num_decl;
762 field = rec -> decl;
763 for(int i = 0; i < num; i++)
764 {
765 if(field -> type -> class == OBERON_TYPE_RECORD)
766 {
767 jvm_generate_load(p, rec, copy_dst);
768 jvm_generate_load(p, field -> type, field -> gen_var);
769 jvm_generate_load(p, rec, copy_src);
770 jvm_generate_load(p, field -> type, field -> gen_var);
771 jvm_generate_copy_record(p, field -> type);
772 }
773 else if(field -> type -> class == OBERON_TYPE_ARRAY)
774 {
775 jvm_generate_load(p, rec, copy_dst);
776 jvm_generate_load(p, field -> type, field -> gen_var);
777 jvm_generate_load(p, rec, copy_src);
778 jvm_generate_load(p, field -> type, field -> gen_var);
779 jvm_generate_copy_array(p, field -> type);
780 }
781 else
782 {
783 jvm_generate_load(p, rec, copy_dst);
784 jvm_generate_load(p, rec, copy_src);
785 jvm_generate_load(p, field -> type, field -> gen_var);
786 jvm_generate_store(p, field -> type, field -> gen_var);
787 }
788 field = field -> next;
789 }
790 jvm_generate(p, 0, 0, "return");
791 jvm_generate_function_end(p);
792
793 jvm_destroy_class(class);
794 }
795
796 void
797 oberon_generator_init_var(oberon_context_t * ctx, oberon_object_t * var)
798 {
799 gen_var_t * v = oberon_generator_new_var();
800 var -> gen_var = v;
801
802 gen_module_t * m;
803 m = ctx -> mod -> gen_mod;
804
805 struct gen_class * class;
806 class = m -> class;
807
808 char * desc;
809 switch(var -> class)
810 {
811 case OBERON_CLASS_VAR_PARAM:
812 gen_error("generator: VAR-parameters not implemented");
813 break;
814 case OBERON_CLASS_PARAM:
815 /* Заполняется при генерации функции */
816 /* смотри jvm_init_local_object() */
817 break;
818 case OBERON_CLASS_FIELD:
819 /* Заполняются при инициализации структуры */
820 /* смотри oberon_generator_init_record() */
821 break;
822 case OBERON_CLASS_VAR:
823 /* Локальные заполняются при генерации функции */
824 /* смотри jvm_init_local_object() */
825 if(var -> local == 0)
826 {
827 desc = jvm_get_descriptor(var -> type);
828 jvm_generate_and_init_global_var(v, class, "public", var -> name, desc);
829 }
830 break;
831 default:
832 gen_error("oberon_generator_init_var: unk var class %i", var -> class);
833 break;
834 }
835 }
836
837 void
838 oberon_generator_init_proc(oberon_context_t * ctx, oberon_object_t * proc)
839 {
840 struct gen_class * class = proc -> module -> gen_mod -> class;
841 proc -> gen_proc = jvm_create_proc(class);
842
843 if(proc -> local)
844 {
845 gen_error("generator: local procedures not implemented");
846 }
847 }
848
849 void
850 oberon_generator_init_module(oberon_context_t * ctx, oberon_module_t * mod)
851 {
852 gen_module_t * m = GC_MALLOC(sizeof *m);
853 memset(m, 0, sizeof *m);
854 mod -> gen_mod = m;
855
856 struct gen_class * class;
857 class = jvm_create_class(mod -> name);
858
859 fprintf(class -> fp, ".source %s\n", mod -> name);
860 fprintf(class -> fp, ".class %s\n", mod -> name);
861 fprintf(class -> fp, ".super java/lang/Object\n\n");
862
863 m -> class = class;
864
865 ctx -> gen_context -> current_m = m;
866 }
867
868 void
869 oberon_generator_fini_module(oberon_module_t * mod)
870 {
871 jvm_destroy_class(mod -> gen_mod -> class);
872 }
873
874 static void
875 push_expr(gen_proc_t * p, oberon_expr_t * expr);
876
877 void
878 oberon_generate_begin_module(oberon_context_t * ctx)
879 {
880 struct gen_class * class = ctx -> mod -> gen_mod -> class;
881 gen_proc_t * p = jvm_create_proc(class);
882
883 jvm_generate_function_header(p, "public", "<init>", "()V");
884 jvm_alloc_register_untyped(p -> rf, false);
885 jvm_generate(p, 0, 1, "aload_0");
886 jvm_generate(p, 1, 0, "invokespecial java/lang/Object/<init>()V");
887
888 /* Инициализация переменных объявленных в модуле */
889 oberon_object_t * x = ctx -> mod -> decl -> list -> next;
890 while(x != NULL)
891 {
892 if(x -> class == OBERON_CLASS_VAR)
893 {
894 if(x -> type -> class == OBERON_TYPE_ARRAY
895 || x -> type -> class == OBERON_TYPE_RECORD)
896 {
897 jvm_generate_new(p, x -> type, 0);
898 jvm_generate_store(p, x -> type, x -> gen_var);
899 }
900 }
901 x = x -> next;
902 }
903 }
904
905 void
906 oberon_generate_end_module(oberon_context_t * ctx)
907 {
908 struct gen_class * class = ctx -> mod -> gen_mod -> class;
909 gen_proc_t * p = class -> p;
910
911 jvm_generate(p, 0, 0, "return");
912 jvm_generate_function_end(class -> p);
913 }
914
915 static void
916 jvm_init_local_object(gen_proc_t * p, oberon_object_t * x)
917 {
918 gen_var_t * v;
919 bool wide;
920 char * desc;
921 struct gen_class * class;
922
923 v = x -> gen_var;
924 class = p -> class;
925 wide = jvm_is_wide_type(x -> type);
926 desc = jvm_get_descriptor(x -> type);
927
928 // Убеждаемся что сейчас находимся в функции
929 assert(class -> p);
930 assert(x -> local);
931
932 switch(x -> class)
933 {
934 case OBERON_CLASS_VAR:
935 case OBERON_CLASS_PARAM:
936 jvm_generate_and_init_named_local_var(v, p, wide, x -> name, desc);
937 break;
938 default:
939 gen_error("jvm_init_local_object: wat");
940 break;
941 }
942 }
943
944 void
945 oberon_generate_begin_proc(oberon_context_t * ctx, oberon_object_t * proc)
946 {
947 gen_proc_t * p;
948 char * signature;
949
950 p = proc -> gen_proc;
951
952 signature = jvm_get_procedure_signature(proc -> type);
953
954 jvm_generate_function_header(p, "public static", proc -> name, signature);
955
956 oberon_object_t * var = proc -> type -> decl;
957 while(var)
958 {
959 jvm_init_local_object(p, var);
960 var = var -> next;
961 }
962 }
963
964 void
965 oberon_generate_end_proc(oberon_context_t * ctx)
966 {
967 gen_module_t * m;
968 gen_proc_t * p;
969
970 m = ctx -> mod -> gen_mod;
971 p = m -> class -> p;
972
973 jvm_generate_function_end(p);
974 }
975
976 static void
977 jvm_generate_call_proc(gen_proc_t * p, oberon_expr_t * desig)
978 {
979 assert(desig -> is_item);
980 assert(desig -> item.mode == MODE_CALL);
981
982 oberon_object_t * proc;
983 char * fullname;
984 char * signature;
985
986 proc = desig -> item.var;
987 fullname = jvm_get_field_full_name(proc);
988 signature = jvm_get_procedure_signature(proc -> type);
989
990 int args_cells = 0;
991 int result_cells = jvm_cell_size_for_type(proc -> type -> base);
992
993 int num = desig -> item.num_args;
994 oberon_expr_t * arg = desig -> item.args;
995 for(int i = 0; i < num; i++)
996 {
997 args_cells += jvm_cell_size_for_type(arg -> result);
998 push_expr(p, arg);
999 arg = arg -> next;
1000 }
1001
1002 jvm_generate(p, args_cells, result_cells, "invokestatic %s%s", fullname, signature);
1003 }
1004
1005 void
1006 oberon_generate_call_proc(oberon_context_t * ctx, oberon_expr_t * desig)
1007 {
1008 jvm_generate_call_proc(ctx -> gen_context -> current_m -> class -> p, desig);
1009 }
1010
1011 void
1012 oberon_generate_return(oberon_context_t * ctx, oberon_expr_t * expr)
1013 {
1014 gen_module_t * m;
1015 gen_proc_t * p;
1016 char prefix;
1017 int cell_size;
1018
1019 m = ctx -> mod -> gen_mod;
1020 p = m -> class -> p;
1021
1022 if(expr)
1023 {
1024 push_expr(p, expr);
1025 prefix = jvm_get_prefix(expr -> result);
1026 cell_size = jvm_cell_size_for_type(expr -> result);
1027 jvm_generate(p, cell_size, 0, "%creturn", prefix);
1028 }
1029 else
1030 {
1031 jvm_generate(p, 0, 0, "return");
1032 }
1033 }
1034
1035 static void
1036 jvm_generate_expr_new_pointer(gen_proc_t * p, oberon_type_t * type, int num, oberon_expr_t * arg)
1037 {
1038 assert(type -> class == OBERON_TYPE_POINTER);
1039
1040 for(int i = 0; i < num; i++)
1041 {
1042 push_expr(p, arg);
1043 arg = arg -> next;
1044 }
1045
1046 jvm_generate_new(p, type -> base, num);
1047 }
1048
1049 static void
1050 push_item(gen_proc_t * p, oberon_item_t * item)
1051 {
1052 switch(item -> mode)
1053 {
1054 case MODE_VAR:
1055 jvm_generate_load(p, item -> result, item -> var -> gen_var);
1056 break;
1057 case MODE_INTEGER:
1058 jvm_generate_push_int(p, item -> integer);
1059 break;
1060 case MODE_BOOLEAN:
1061 jvm_generate_push_int(p, item -> boolean);
1062 break;
1063 case MODE_CALL:
1064 jvm_generate_call_proc(p, (oberon_expr_t *) item);
1065 break;
1066 case MODE_INDEX:
1067 ;
1068 char postfix = jvm_get_postfix(item -> result);
1069 int cell_size = jvm_cell_size_for_postfix(postfix);
1070 push_item(p, item -> parent);
1071 push_expr(p, item -> args);
1072 jvm_generate(p, 1 + 1, cell_size, "%caload", postfix);
1073 break;
1074 case MODE_FIELD:
1075 push_item(p, item -> parent);
1076 jvm_generate_load(p, item -> result, item -> var -> gen_var);
1077 break;
1078 case MODE_DEREF:
1079 /* Все объекты в jvm представляются как указатели */
1080 push_item(p, item -> parent);
1081 break;
1082 case MODE_NIL:
1083 jvm_generate(p, 0, 1, "aconst_null");
1084 break;
1085 case MODE_NEW:
1086 jvm_generate_expr_new_pointer(p, item -> result, item -> num_args, item -> args);
1087 break;
1088 case MODE_REAL:
1089 jvm_generate_push_float(p, item -> real, item -> result -> size);
1090 break;
1091 default:
1092 gen_error("push_item: unk mode %i", item -> mode);
1093 break;
1094 }
1095 }
1096
1097 static void
1098 jvm_generate_logical_not(gen_proc_t * p)
1099 {
1100 int label_done = jvm_new_label_id(p);
1101 int label_false = jvm_new_label_id(p);
1102 char * label_name_done = jvm_get_label_name(label_done);
1103 char * label_name_false = jvm_get_label_name(label_false);
1104
1105 jvm_generate(p, 1, 0, "ifne %s", label_name_false);
1106 jvm_generate(p, 0, 1, "iconst_1");
1107 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1108 jvm_generate_label(p, label_false);
1109 jvm_generate(p, 0, 1, "iconst_0");
1110 jvm_generate_label(p, label_done);
1111 }
1112
1113 static void
1114 jvm_generate_abs(gen_proc_t * p, char prefix)
1115 {
1116 char t = jvm_get_type_of_prefix(prefix);
1117 int cell_size = jvm_cell_size_for_postfix(prefix);
1118 jvm_generate(p, cell_size, cell_size, "invokestatic java/lang/Math/abs(%c)%c", t, t);
1119 }
1120
1121 static void
1122 jvm_generate_compare_op(gen_proc_t * p, char prefix, int op)
1123 {
1124 int label_true = jvm_new_label_id(p);
1125 int label_done = jvm_new_label_id(p);
1126 char * label_name_true = jvm_get_label_name(label_true);
1127 char * label_name_done = jvm_get_label_name(label_done);
1128 int cell_size = 2 * jvm_cell_size_for_postfix(prefix);
1129
1130 assert(prefix == 'i' || prefix == 'a');
1131
1132 const char * cmpop = "";
1133 switch(op)
1134 {
1135 case OP_EQ:
1136 cmpop = "eq";
1137 break;
1138 case OP_NEQ:
1139 cmpop = "ne";
1140 break;
1141 case OP_LSS:
1142 cmpop = "lt";
1143 break;
1144 case OP_LEQ:
1145 cmpop = "le";
1146 break;
1147 case OP_GRT:
1148 cmpop = "gt";
1149 break;
1150 case OP_GEQ:
1151 cmpop = "ge";
1152 break;
1153 default:
1154 gen_error("jvm_generate_compare_op: wat");
1155 break;
1156 }
1157
1158 jvm_generate(p, cell_size, 0, "if_%ccmp%s %s", prefix, cmpop, label_name_true);
1159 jvm_generate(p, 0, 1, "iconst_0");
1160 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1161 jvm_generate_label(p, label_true);
1162 jvm_generate(p, 0, 1, "iconst_1");
1163 jvm_generate_label(p, label_done);
1164 }
1165
1166 static void
1167 jvm_generate_operator(gen_proc_t * p, char prefix, int op)
1168 {
1169 int cell_size = jvm_cell_size_for_postfix(prefix);
1170 switch(op)
1171 {
1172 case OP_UNARY_MINUS:
1173 jvm_generate(p, cell_size, cell_size, "%cneg", prefix);
1174 break;
1175 case OP_BITWISE_NOT:
1176 jvm_generate_push_int(p, -1);
1177 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1178 break;
1179 case OP_LOGIC_NOT:
1180 jvm_generate_logical_not(p);
1181 break;
1182 case OP_ABS:
1183 jvm_generate_abs(p, prefix);
1184 break;
1185
1186 case OP_ADD:
1187 jvm_generate(p, 2 * cell_size, cell_size, "%cadd", prefix);
1188 break;
1189 case OP_SUB:
1190 jvm_generate(p, 2 * cell_size, cell_size, "%csub", prefix);
1191 break;
1192 case OP_MUL:
1193 jvm_generate(p, 2 * cell_size, cell_size, "%cmul", prefix);
1194 break;
1195 case OP_DIV:
1196 jvm_generate(p, 2 * cell_size, cell_size, "%cdiv", prefix);
1197 break;
1198 case OP_MOD:
1199 jvm_generate(p, 2 * cell_size, cell_size, "%crem", prefix);
1200 break;
1201 case OP_BITWISE_AND:
1202 jvm_generate(p, 2 * cell_size, cell_size, "%cand", prefix);
1203 break;
1204 case OP_BITWISE_XOR:
1205 jvm_generate(p, 2 * cell_size, cell_size, "%cxor", prefix);
1206 break;
1207 case OP_BITWISE_OR:
1208 jvm_generate(p, 2 * cell_size, cell_size, "%cor", prefix);
1209 break;
1210
1211 case OP_EQ:
1212 case OP_NEQ:
1213 case OP_LSS:
1214 case OP_LEQ:
1215 case OP_GRT:
1216 case OP_GEQ:
1217 jvm_generate_compare_op(p, prefix, op);
1218 break;
1219 default:
1220 gen_error("jvm_generate_operator: unk op %i", op);
1221 break;
1222 }
1223 }
1224
1225 static void
1226 jvm_generate_logical_or(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1227 {
1228 int label_calc_b = jvm_new_label_id(p);
1229 int label_done = jvm_new_label_id(p);
1230 char * label_name_calc_b = jvm_get_label_name(label_calc_b);
1231 char * label_name_done = jvm_get_label_name(label_done);
1232
1233 /* a OR b -- если a, то TRUE, иначе b */
1234
1235 push_expr(p, a);
1236 jvm_generate(p, 1, 0, "ifne %s", label_name_calc_b);
1237 jvm_generate(p, 0, 1, "iconst_1");
1238 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1239 jvm_generate_label(p, label_calc_b);
1240 push_expr(p, b);
1241 jvm_generate_label(p, label_done);
1242 }
1243
1244 static void
1245 jvm_generate_logical_and(gen_proc_t * p, oberon_expr_t * a, oberon_expr_t * b)
1246 {
1247 int label_false = jvm_new_label_id(p);
1248 int label_done = jvm_new_label_id(p);
1249 char * label_name_false = jvm_get_label_name(label_false);
1250 char * label_name_done = jvm_get_label_name(label_done);
1251
1252 /* a AND b -- если a, то b, иначе FALSE */
1253
1254 push_expr(p, a);
1255 jvm_generate(p, 1, 0, "ifne %s", label_name_false);
1256 push_expr(p, b);
1257 jvm_generate(p, 0, 0, "goto %s", label_name_done);
1258 jvm_generate_label(p, label_false);
1259 jvm_generate(p, 0, 1, "iconst_0");
1260 jvm_generate_label(p, label_done);
1261 }
1262
1263 static void
1264 push_operator(gen_proc_t * p, oberon_oper_t * oper)
1265 {
1266 char prefix = jvm_get_prefix(oper -> result);
1267 int op = oper -> op;
1268 switch(op)
1269 {
1270 case OP_UNARY_MINUS:
1271 case OP_BITWISE_NOT:
1272 case OP_LOGIC_NOT:
1273 case OP_ABS:
1274 push_expr(p, oper -> left);
1275 jvm_generate_operator(p, prefix, op);
1276 break;
1277
1278 case OP_ADD:
1279 case OP_SUB:
1280 case OP_MUL:
1281 case OP_DIV:
1282 case OP_MOD:
1283 case OP_BITWISE_AND:
1284 case OP_BITWISE_XOR:
1285 case OP_BITWISE_OR:
1286
1287 case OP_EQ:
1288 case OP_NEQ:
1289 case OP_LSS:
1290 case OP_LEQ:
1291 case OP_GRT:
1292 case OP_GEQ:
1293 push_expr(p, oper -> left);
1294 push_expr(p, oper -> right);
1295 jvm_generate_operator(p, prefix, op);
1296 break;
1297
1298 case OP_LOGIC_OR:
1299 jvm_generate_logical_or(p, oper -> left, oper -> right);
1300 break;
1301 case OP_LOGIC_AND:
1302 jvm_generate_logical_and(p, oper -> left, oper -> right);
1303 break;
1304 default:
1305 gen_error("push_oper: unk op %i", op);
1306 break;
1307 }
1308 }
1309
1310 static void
1311 push_expr(gen_proc_t * p, oberon_expr_t * expr)
1312 {
1313 if(expr -> is_item)
1314 {
1315 push_item(p, (oberon_item_t *) expr);
1316 }
1317 else
1318 {
1319 push_operator(p, (oberon_oper_t *) expr);
1320 }
1321 }
1322
1323 static void
1324 store_expr(gen_proc_t * p, oberon_expr_t * dst, oberon_expr_t * src)
1325 {
1326 assert(dst -> is_item);
1327 oberon_item_t * item = (oberon_item_t *) dst;
1328
1329 if(dst -> result -> class == OBERON_TYPE_ARRAY
1330 || src -> result -> class == OBERON_TYPE_ARRAY)
1331 {
1332 push_expr(p, dst);
1333 push_expr(p, src);
1334 jvm_generate_copy_array(p, dst -> result);
1335 }
1336 else if(dst -> result -> class == OBERON_TYPE_RECORD
1337 || src -> result -> class == OBERON_TYPE_RECORD)
1338 {
1339 push_expr(p, dst);
1340 push_expr(p, src);
1341 jvm_generate_copy_record(p, dst -> result);
1342 }
1343 else switch(item -> mode)
1344 {
1345 case MODE_VAR:
1346 push_expr(p, src);
1347 jvm_generate_store(p, src -> result, item -> var -> gen_var);
1348 break;
1349 case MODE_INDEX:
1350 ;
1351 char postfix = jvm_get_postfix(src -> result);
1352 int cell_size = jvm_cell_size_for_postfix(postfix);
1353 push_item(p, item -> parent);
1354 push_expr(p, item -> args);
1355 push_expr(p, src);
1356 jvm_generate(p, 1 + 1 + cell_size, 0, "%castore", postfix);
1357 break;
1358 case MODE_FIELD:
1359 push_item(p, item -> parent);
1360 push_expr(p, src);
1361 jvm_generate_store(p, src -> result, item -> var -> gen_var);
1362 break;
1363 default:
1364 gen_error("store_expr: unk mode %i", item -> mode);
1365 break;
1366 }
1367 }
1368
1369 void
1370 oberon_generate_assign(oberon_context_t * ctx, oberon_expr_t * src, oberon_expr_t * dst)
1371 {
1372 gen_module_t * m;
1373 gen_proc_t * p;
1374
1375 m = ctx -> mod -> gen_mod;
1376 p = m -> class -> p;
1377
1378 store_expr(p, dst, src);
1379 }
1380
1381 void
1382 oberon_generate_code(oberon_context_t * ctx)
1383 {
1384 printf("generate code\n");
1385 }
1386
1387 void
1388 oberon_generator_dump(oberon_context_t * ctx, char * path)
1389 {
1390 printf("jit: dump code\n");
1391 }
1392
1393 void *
1394 oberon_generator_get_procedure(oberon_context_t * ctx, const char * name)
1395 {
1396 printf("jit: get pointer to procedure %s\n", name);
1397 return NULL;
1398 }
1399
1400 void *
1401 oberon_generator_get_var(oberon_context_t * ctx, const char * name)
1402 {
1403 printf("jit: get pointer to var %s\n", name);
1404 return NULL;
1405 }
Another attempt to make Oberon compiler
[25/02/2024 19:55:35]
DEADSOFTWARE 2012-2024