1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2008 Jean Privat <jean@pryen.org>
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
17 # Compile method bodies, statments and expressions to C.
18 package compiling_methods
23 redef class CompilerVisitor
24 # Compile a statment node
25 meth compile_stmt
(n
: PExpr)
27 n
.prepare_compile_stmt
(self)
28 var i
= _variable_index
33 # Compile is expression node
34 meth compile_expr
(n
: PExpr): String
36 var i
= _variable_index
37 var s
= n
.compile_expr
(self)
42 if s
== variable
(_variable_index-1
) then
50 # Ensure that a c expression is a var
51 meth ensure_var
(s
: String): String
53 if s
.substring
(0,3) == "variable" then
61 # Add a assignment between a variable and an expression
62 meth add_assignment
(v
: String, s
: String)
65 add_instr
("{v} = {s};")
69 # Return the ith variable
70 protected meth variable
(i
: Int): String
75 # Next available variable number
76 attr _variable_index
: Int
78 # Total number of variable
79 attr _variable_index_max
: Int
81 # Return the next available variable
84 var v
= variable
(_variable_index
)
85 _variable_index
= _variable_index
+ 1
86 if _variable_index
> _variable_index_max
then
87 add_decl
("val_t {v};")
88 _variable_index_max
= _variable_index
93 # Mark the variable available
94 meth free_var
(v
: String)
97 if v
== variable
(_variable_index-1
) then
98 _variable_index
= _variable_index
- 1
102 # Clear all status related to a method body
108 _variable_index_max
= 0
111 # Association between nit variable and the corrsponding c variable
112 readable attr _varnames
: Map[Variable, String] = new HashMap[Variable, String]
114 # Is a "return" found in the method body
115 readable writable attr _has_return
: Bool
117 # Association between parameters and the corresponding c variables
118 readable writable attr _method_params
: Array[String]
120 # Current method compiled
121 readable writable attr _method
: MMSrcMethod
123 # Where a nit return must branch
124 readable writable attr _return_label
: String
126 # Where a nit break must branch
127 readable writable attr _break_label
: String
129 # Where a nit continue must branch
130 readable writable attr _continue_label
: String
132 # Variable where a functionnal nit return must store its value
133 readable writable attr _return_value
: String
135 # Generate an fprintf to display an error location
136 meth printf_locate_error
(node
: PNode): String
138 var s
= "fprintf(stderr, \""
139 if method != null then s.append(" in %s
")
140 s.append(" (%s
:%d
)\\n\
", ")
141 if method
!= null then s
.append
("LOCATE_{method.cname}, ")
142 s
.append
("LOCATE_{module.name}, {node.line_number});")
146 redef init(module: MMSrcModule)
152 meth invoke_super_init_calls_after
(start_prop
: MMMethod)
155 assert n
isa AConcreteInitPropdef
157 if n
.super_init_calls
.is_empty
then return
161 if start_prop
!= null then
162 while n
.super_init_calls
[i
] != start_prop
do
163 #s.append(" {n.super_init_calls[i]}")
167 #s.append(" {start_prop}")
169 while n
.explicit_super_init_calls
[j
] != start_prop
do
174 var stop_prop
: MMMethod = null
175 if j
< n
.explicit_super_init_calls
.length
then
176 stop_prop
= n
.explicit_super_init_calls
[j
]
178 var l
= n
.super_init_calls
.length
181 var p
= n
.super_init_calls
[i
]
182 if p
== stop_prop
then break
183 var cargs
= method_params
184 if p
.signature
.arity
== 0 then
185 cargs
= [method_params
[0]]
188 p
.compile_call
(self, cargs
)
193 # s.append(" {n.super_init_calls[i]}")
196 #if stop_prop != null then s.append(" (stop at {stop_prop})")
197 #n.printl("implicit calls in {n.method}: {s}")
201 ###############################################################################
204 # Compile a call on self for given arguments
205 # Most calls are compiled with a table access,
206 # primitive calles are inlined
207 # == and != are guarded and possibly inlined
208 meth compile_call
(v
: CompilerVisitor, cargs
: Array[String]): String
211 if i
isa MMSrcMethod then
212 if i
isa MMMethSrcMethod and i
.node
isa AInternMethPropdef or
213 (i
.local_class
.name
== (once
"Array".to_symbol
) and name
== (once
"[]".to_symbol
))
215 var e
= i
.do_compile_inside
(v
, cargs
)
219 var ee
= once
"==".to_symbol
220 var ne
= once
"!=".to_symbol
222 var eqp
= signature
.recv
.local_class
.select_method
(ee
)
223 var eqcall
= eqp
.compile_call
(v
, cargs
)
224 return "TAG_Bool(!UNTAG_Bool({eqcall}))"
226 if global
.is_init
then
228 cargs
.add
("init_table /*YYY*/")
231 var m
= "(({cname}_t)CALL({cargs[0]},{global.color_id}))"
232 var vcall
= "{m}({cargs.join(", ")}) /*{local_class}::{name}*/"
234 vcall
= "UNTAG_Bool({vcall})"
235 var obj
= once
"Object".to_symbol
236 if i
.local_class
.name
== obj
then
237 vcall
= "(({m}=={i.cname})?(IS_EQUAL_NN({cargs[0]},{cargs[1]})):({vcall}))"
239 vcall
= "TAG_Bool(({cargs.first} == {cargs[1]}) || (({cargs.first} != NIT_NULL) && {vcall}))"
241 if signature
.return_type
!= null then
244 v
.add_instr
(vcall
+ ";")
249 # Compile a call as constructor with given args
250 meth compile_constructor_call
(v
: CompilerVisitor, recvtype
: MMType, cargs
: Array[String]): String
253 v
.add_instr
("{recv} = NEW_{recvtype.local_class}_{global.intro.cname}({cargs.join(", ")}); /*new {recvtype}*/")
257 # Compile a call as call-next-method on self with given args
258 meth compile_super_call
(v
: CompilerVisitor, cargs
: Array[String]): String
260 var m
= "(({cname}_t)CALL({cargs[0]},{color_id_for_super}))"
261 var vcall
= "{m}({cargs.join(", ")}) /*super {local_class}::{name}*/"
266 redef class MMAttribute
267 # Compile an acces on selffor a given reciever.
268 # Result is a valid C left-value for assigment
269 meth compile_access
(v
: CompilerVisitor, recv
: String): String
271 return "{global.attr_access}({recv}) /*{local_class}::{name}*/"
275 redef class MMLocalProperty
276 # Compile the property as a C property
277 meth compile_property_to_c
(v
: CompilerVisitor) do end
280 redef class MMSrcMethod
281 # Compile and declare the signature to C
282 protected meth decl_csignature
(v
: CompilerVisitor, args
: Array[String]): String
284 var params
= new Array[String]
285 var params_new
: Array[String] = null
286 if global
.is_init
then
287 params_new
= new Array[String]
289 params
.add
("val_t {args[0]}")
290 for i
in [0..signature
.arity
[ do
291 var p
= "val_t {args[i+1]}"
293 if params_new
!= null then params_new
.add
(p
)
295 if global
.is_init
then
296 params
.add
("int* init_table")
299 if signature
.return_type
!= null then
304 var p
= params
.join
(", ")
305 var s
= "{ret} {cname}({p})"
306 v
.add_decl
("typedef {ret} (* {cname}_t)({p});")
308 if params_new
!= null then
309 v
.add_decl
("val_t NEW_{cname}({params_new.join(", ")});")
314 redef meth compile_property_to_c
(v
)
317 var args
= new Array[String]
319 for i
in [0..signature
.arity
[ do
320 args
.add
(" param{i}")
322 var cs
= decl_csignature
(v
, args
)
323 v
.add_decl
("#define LOCATE_{cname} \"{full_name}\
"")
325 v
.add_instr
("{cs} \{")
332 if s
.node
!= null then ln
= s
.node
.line_number
333 v
.add_decl
("struct trace_t trace = \{NULL, NULL, {ln}, LOCATE_{cname}};")
334 v
.add_instr
("trace.prev = tracehead; tracehead = &trace;")
335 v
.add_instr
("trace.file = LOCATE_{module.name};")
336 var s
= do_compile_inside
(v
, args
)
337 v
.add_instr
("tracehead = trace.prev;")
339 v
.add_instr
("return;")
341 v
.add_instr
("return {s};")
344 ctx_old
.append
(v
.ctx
)
350 # Compile the method body inline
351 meth do_compile_inside
(v
: CompilerVisitor, params
: Array[String]): String is abstract
354 redef class MMReadImplementationMethod
355 redef meth do_compile_inside
(v
, params
)
357 return node
.prop
.compile_access
(v
, params
[0])
361 redef class MMWriteImplementationMethod
362 redef meth do_compile_inside
(v
, params
)
364 v
.add_assignment
(node
.prop
.compile_access
(v
, params
[0]), params
[1])
369 redef class MMMethSrcMethod
370 redef meth do_compile_inside
(v
, params
)
372 return node
.do_compile_inside
(v
, self, params
)
377 # Compile a subtype check to self
379 meth compile_cast
(v
: CompilerVisitor, recv
: String): String
381 # Fixme: handle formaltypes
382 var g
= local_class
.global
383 return "TAG_Bool(({recv}==NIT_NULL) || VAL_ISA({recv}, {g.color_id}, {g.id_id})) /*cast {self}*/"
386 # Compile a cast assertion
387 meth compile_type_check
(v
: CompilerVisitor, recv
: String, n
: PNode)
389 # Fixme: handle formaltypes
390 var g
= local_class
.global
391 v
.add_instr
("if (({recv}!=NIT_NULL) && !VAL_ISA({recv}, {g.color_id}, {g.id_id})) \{ fprintf(stderr, \"Cast failled\"); {v.printf_locate_error(n)} nit_exit(1); } /*cast {self}*/;")
395 ###############################################################################
397 redef class AMethPropdef
398 # Compile the method body
399 meth do_compile_inside
(v
: CompilerVisitor, method
: MMSrcMethod, params
: Array[String]): String is abstract
402 redef class AConcreteMethPropdef
403 redef meth do_compile_inside
(v
, method
, params
)
405 var orig_meth
: MMLocalProperty = method
.global
.intro
406 var orig_sig
= orig_meth
.signature_for
(method
.signature
.recv
)
407 if n_signature
!= null then
408 var sig
= n_signature
409 assert sig
isa ASignature
410 for ap
in sig
.n_params
do
411 var cname
= v
.get_var
412 v
.varnames
[ap
.variable
] = cname
413 var orig_type
= orig_sig
[ap
.position
]
414 if not orig_type
< ap
.variable
.stype
then
415 # FIXME: do not test always
416 # FIXME: handle formal types
417 v
.add_instr
("/* check if p<{ap.variable.stype} with p:{orig_type} */")
418 ap
.variable
.stype
.compile_type_check
(v
, params
[ap
.position
+ 1], ap
)
420 v
.add_assignment
(cname
, params
[ap
.position
+ 1])
423 var old_method_params
= v
.method_params
424 var old_return_label
= v
.return_label
425 var old_return_value
= v
.return_value
426 var old_has_return
= v
.has_return
428 var itpos
: String = null
429 if self isa AConcreteInitPropdef then
430 itpos
= "VAL2OBJ({params[0]})->vft[{method.local_class.global.init_table_pos_id}].i"
431 # v.add_instr("printf(\"{method.full_name}: inittable[%d] = %d\\n\", {itpos}, init_table[{itpos}]);")
432 v
.add_instr
("if (init_table[{itpos}]) return;")
435 v
.method_params
= params
437 v
.return_label
= "return_label{v.new_number}"
438 if method
.signature
.return_type
!= null then
439 v
.return_value
= v
.get_var
440 v
.free_var
(v
.return_value
)
442 v
.return_value
= null
445 if self isa AConcreteInitPropdef then
446 v
.invoke_super_init_calls_after
(null)
448 if n_block
!= null then
449 v
.compile_stmt
(n_block
)
452 v
.add_instr
("{v.return_label}: while(false);")
454 if self isa AConcreteInitPropdef then
455 v
.add_instr
("init_table[{itpos}] = 1;")
457 var ret
= v
.return_value
458 v
.method_params
= old_method_params
459 v
.return_label
= old_return_label
460 v
.return_value
= old_return_value
461 v
.has_return
= old_has_return
466 redef class ADeferredMethPropdef
467 redef meth do_compile_inside
(v
, method
, params
)
469 v
.add_instr
("fprintf(stderr, \"Deferred method
%s called\
");")
470 v
.add_instr
(v
.printf_locate_error
(self))
471 v
.add_instr
("nit_exit(1);")
472 if method
.signature
.return_type
!= null then
480 redef class AExternMethPropdef
481 redef meth do_compile_inside
(v
, method
, params
)
483 var ename
= "{method.module.name}_{method.local_class.name}_{method.local_class.name}_{method.name}_{method.signature.arity}"
484 if n_extern
!= null then
485 ename
= n_extern
.text
486 ename
= ename
.substring
(1, ename
.length-2
)
488 var sig
= method
.signature
489 if params
.length
!= sig
.arity
+ 1 then
490 printl
("par:{params.length} sig:{sig.arity}")
492 var args
= new Array[String]
493 args
.add
(sig
.recv
.unboxtype
(params
[0]))
494 for i
in [0..sig
.arity
[ do
495 args
.add
(sig
[i
].unboxtype
(params
[i
+1]))
497 var s
= "{ename}({args.join(", ")})"
498 if sig
.return_type
!= null then
499 return sig
.return_type
.boxtype
(s
)
507 redef class AInternMethPropdef
508 redef meth do_compile_inside
(v
, method
, p
)
510 var c
= method
.local_class
.name
513 if c
== once
"Int".to_symbol
then
514 if n
== once
"object_id".to_symbol
then
516 else if n
== once
"unary -".to_symbol
then
517 s
= "TAG_Int(-UNTAG_Int({p[0]}))"
518 else if n
== once
"output".to_symbol
then
519 v
.add_instr
("printf(\"%d\\n\
", UNTAG_Int({p[0]}));")
520 else if n
== once
"ascii".to_symbol
then
521 s
= "TAG_Char(UNTAG_Int({p[0]}))"
522 else if n
== once
"succ".to_symbol
then
523 s
= "TAG_Int(UNTAG_Int({p[0]})+1)"
524 else if n
== once
"prec".to_symbol
then
525 s
= "TAG_Int(UNTAG_Int({p[0]})-1)"
526 else if n
== once
"to_f".to_symbol
then
527 s
= "BOX_Float((float)UNTAG_Int({p[0]}))"
528 else if n
== once
"+".to_symbol
then
529 s
= "TAG_Int(UNTAG_Int({p[0]})+UNTAG_Int({p[1]}))"
530 else if n
== once
"-".to_symbol
then
531 s
= "TAG_Int(UNTAG_Int({p[0]})-UNTAG_Int({p[1]}))"
532 else if n
== once
"*".to_symbol
then
533 s
= "TAG_Int(UNTAG_Int({p[0]})*UNTAG_Int({p[1]}))"
534 else if n
== once
"/".to_symbol
then
535 s
= "TAG_Int(UNTAG_Int({p[0]})/UNTAG_Int({p[1]}))"
536 else if n
== once
"%".to_symbol
then
537 s
= "TAG_Int(UNTAG_Int({p[0]})%UNTAG_Int({p[1]}))"
538 else if n
== once
"<".to_symbol
then
539 s
= "TAG_Bool(UNTAG_Int({p[0]})<UNTAG_Int({p[1]}))"
540 else if n
== once
">".to_symbol
then
541 s
= "TAG_Bool(UNTAG_Int({p[0]})>UNTAG_Int({p[1]}))"
542 else if n
== once
"<=".to_symbol
then
543 s
= "TAG_Bool(UNTAG_Int({p[0]})<=UNTAG_Int({p[1]}))"
544 else if n
== once
">=".to_symbol
then
545 s
= "TAG_Bool(UNTAG_Int({p[0]})>=UNTAG_Int({p[1]}))"
546 else if n
== once
"lshift".to_symbol
then
547 s
= "TAG_Int(UNTAG_Int({p[0]})<<UNTAG_Int({p[1]}))"
548 else if n
== once
"rshift".to_symbol
then
549 s
= "TAG_Int(UNTAG_Int({p[0]})>>UNTAG_Int({p[1]}))"
550 else if n
== once
"==".to_symbol
then
551 s
= "TAG_Bool(({p[0]})==({p[1]}))"
552 else if n
== once
"!=".to_symbol
then
553 s
= "TAG_Bool(({p[0]})!=({p[1]}))"
555 else if c
== once
"Float".to_symbol
then
556 if n
== once
"object_id".to_symbol
then
557 s
= "TAG_Int((bigint)UNBOX_Float({p[0]}))"
558 else if n
== once
"unary -".to_symbol
then
559 s
= "BOX_Float(-UNBOX_Float({p[0]}))"
560 else if n
== once
"output".to_symbol
then
561 v
.add_instr
("printf(\"%f\\n\
", UNBOX_Float({p[0]}));")
562 else if n
== once
"to_i".to_symbol
then
563 s
= "TAG_Int((bigint)UNBOX_Float({p[0]}))"
564 else if n
== once
"+".to_symbol
then
565 s
= "BOX_Float(UNBOX_Float({p[0]})+UNBOX_Float({p[1]}))"
566 else if n
== once
"-".to_symbol
then
567 s
= "BOX_Float(UNBOX_Float({p[0]})-UNBOX_Float({p[1]}))"
568 else if n
== once
"*".to_symbol
then
569 s
= "BOX_Float(UNBOX_Float({p[0]})*UNBOX_Float({p[1]}))"
570 else if n
== once
"/".to_symbol
then
571 s
= "BOX_Float(UNBOX_Float({p[0]})/UNBOX_Float({p[1]}))"
572 else if n
== once
"<".to_symbol
then
573 s
= "TAG_Bool(UNBOX_Float({p[0]})<UNBOX_Float({p[1]}))"
574 else if n
== once
">".to_symbol
then
575 s
= "TAG_Bool(UNBOX_Float({p[0]})>UNBOX_Float({p[1]}))"
576 else if n
== once
"<=".to_symbol
then
577 s
= "TAG_Bool(UNBOX_Float({p[0]})<=UNBOX_Float({p[1]}))"
578 else if n
== once
">=".to_symbol
then
579 s
= "TAG_Bool(UNBOX_Float({p[0]})>=UNBOX_Float({p[1]}))"
581 else if c
== once
"Char".to_symbol
then
582 if n
== once
"object_id".to_symbol
then
583 s
= "TAG_Int(UNTAG_Char({p[0]}))"
584 else if n
== once
"unary -".to_symbol
then
585 s
= "TAG_Char(-UNTAG_Char({p[0]}))"
586 else if n
== once
"output".to_symbol
then
587 v
.add_instr
("printf(\"%c\
", (unsigned char)UNTAG_Char({p[0]}));")
588 else if n
== once
"ascii".to_symbol
then
589 s
= "TAG_Int((unsigned char)UNTAG_Char({p[0]}))"
590 else if n
== once
"succ".to_symbol
then
591 s
= "TAG_Char(UNTAG_Char({p[0]})+1)"
592 else if n
== once
"prec".to_symbol
then
593 s
= "TAG_Char(UNTAG_Char({p[0]})-1)"
594 else if n
== once
"to_i".to_symbol
then
595 s
= "TAG_Int(UNTAG_Char({p[0]})-'0')"
596 else if n
== once
"+".to_symbol
then
597 s
= "TAG_Char(UNTAG_Char({p[0]})+UNTAG_Char({p[1]}))"
598 else if n
== once
"-".to_symbol
then
599 s
= "TAG_Char(UNTAG_Char({p[0]})-UNTAG_Char({p[1]}))"
600 else if n
== once
"*".to_symbol
then
601 s
= "TAG_Char(UNTAG_Char({p[0]})*UNTAG_Char({p[1]}))"
602 else if n
== once
"/".to_symbol
then
603 s
= "TAG_Char(UNTAG_Char({p[0]})/UNTAG_Char({p[1]}))"
604 else if n
== once
"%".to_symbol
then
605 s
= "TAG_Char(UNTAG_Char({p[0]})%UNTAG_Char({p[1]}))"
606 else if n
== once
"<".to_symbol
then
607 s
= "TAG_Bool(UNTAG_Char({p[0]})<UNTAG_Char({p[1]}))"
608 else if n
== once
">".to_symbol
then
609 s
= "TAG_Bool(UNTAG_Char({p[0]})>UNTAG_Char({p[1]}))"
610 else if n
== once
"<=".to_symbol
then
611 s
= "TAG_Bool(UNTAG_Char({p[0]})<=UNTAG_Char({p[1]}))"
612 else if n
== once
">=".to_symbol
then
613 s
= "TAG_Bool(UNTAG_Char({p[0]})>=UNTAG_Char({p[1]}))"
614 else if n
== once
"==".to_symbol
then
615 s
= "TAG_Bool(({p[0]})==({p[1]}))"
616 else if n
== once
"!=".to_symbol
then
617 s
= "TAG_Bool(({p[0]})!=({p[1]}))"
619 else if c
== once
"Bool".to_symbol
then
620 if n
== once
"object_id".to_symbol
then
621 s
= "TAG_Int(UNTAG_Bool({p[0]}))"
622 else if n
== once
"unary -".to_symbol
then
623 s
= "TAG_Bool(-UNTAG_Bool({p[0]}))"
624 else if n
== once
"output".to_symbol
then
625 v
.add_instr
("(void)printf(UNTAG_Bool({p[0]})?\"true\\n\
":\"false\\n\
");")
626 else if n
== once
"ascii".to_symbol
then
627 s
= "TAG_Bool(UNTAG_Bool({p[0]}))"
628 else if n
== once
"to_i".to_symbol
then
629 s
= "TAG_Int(UNTAG_Bool({p[0]}))"
630 else if n
== once
"==".to_symbol
then
631 s
= "TAG_Bool(({p[0]})==({p[1]}))"
632 else if n
== once
"!=".to_symbol
then
633 s
= "TAG_Bool(({p[0]})!=({p[1]}))"
635 else if c
== once
"NativeArray".to_symbol
then
636 if n
== once
"object_id".to_symbol
then
637 s
= "TAG_Int(UNBOX_NativeArray({p[0]}))"
638 else if n
== once
"[]".to_symbol
then
639 s
= "UNBOX_NativeArray({p[0]})[UNTAG_Int({p[1]})]"
640 else if n
== once
"[]=".to_symbol
then
641 v
.add_instr
("UNBOX_NativeArray({p[0]})[UNTAG_Int({p[1]})]={p[2]};")
642 else if n
== once
"copy_to".to_symbol
then
643 v
.add_instr
("(void)memcpy(UNBOX_NativeArray({p[1]}), UNBOX_NativeArray({p[0]}), UNTAG_Int({p[2]})*sizeof(val_t));")
645 else if c
== once
"NativeString".to_symbol
then
646 if n
== once
"object_id".to_symbol
then
647 s
= "TAG_Int(UNBOX_NativeString({p[0]}))"
648 else if n
== once
"atoi".to_symbol
then
649 s
= "TAG_Int(atoi(UNBOX_NativeString({p[0]})))"
650 else if n
== once
"[]".to_symbol
then
651 s
= "TAG_Char(UNBOX_NativeString({p[0]})[UNTAG_Int({p[1]})])"
652 else if n
== once
"[]=".to_symbol
then
653 v
.add_instr
("UNBOX_NativeString({p[0]})[UNTAG_Int({p[1]})]=UNTAG_Char({p[2]});")
654 else if n
== once
"copy_to".to_symbol
then
655 v
.add_instr
("(void)memcpy(UNBOX_NativeString({p[1]})+UNTAG_Int({p[4]}), UNBOX_NativeString({p[0]})+UNTAG_Int({p[3]}), UNTAG_Int({p[2]}));")
657 else if n
== once
"object_id".to_symbol
then
658 s
= "TAG_Int((bigint){p[0]})"
659 else if n
== once
"sys".to_symbol
then
661 else if n
== once
"is_same_type".to_symbol
then
662 s
= "TAG_Bool((VAL2VFT({p[0]})==VAL2VFT({p[1]})))"
663 else if n
== once
"exit".to_symbol
then
664 v
.add_instr
("exit(UNTAG_Int({p[1]}));")
665 else if n
== once
"calloc_array".to_symbol
then
666 s
= "BOX_NativeArray((val_t*)malloc((UNTAG_Int({p[1]}) * sizeof(val_t))))"
667 else if n
== once
"calloc_string".to_symbol
then
668 s
= "BOX_NativeString((char*)malloc((UNTAG_Int({p[1]}) * sizeof(char))))"
671 v
.add_instr
("fprintf(stderr, \"Intern {n}\\n\
"); nit_exit(1);")
673 if method
.signature
.return_type
!= null and s
== null then
674 s
= "NIT_NULL /*stub*/"
680 ###############################################################################
683 # Compile the node as an expression
684 # Only the visitor should call it
685 meth compile_expr
(v
: CompilerVisitor): String is abstract
687 # Prepare a call of node as a statement
688 # Only the visitor should call it
689 # It's used for local variable managment
690 meth prepare_compile_stmt
(v
: CompilerVisitor) do end
692 # Compile the node as a statement
693 # Only the visitor should call it
694 meth compile_stmt
(v
: CompilerVisitor) do printl
("Error!")
697 redef class ABlockExpr
698 redef meth compile_stmt
(v
)
706 redef class AVardeclExpr
707 redef meth prepare_compile_stmt
(v
)
709 var cname
= v
.get_var
710 v
.varnames
[variable
] = cname
713 redef meth compile_stmt
(v
)
715 var cname
= v
.varnames
[variable
]
716 if n_expr
== null then
717 var t
= variable
.stype
718 v
.add_assignment
(cname
, "{t.default_cvalue} /*decl variable {variable.name}*/")
720 var e
= v
.compile_expr
(n_expr
)
721 v
.add_assignment
(cname
, e
)
726 redef class AReturnExpr
727 redef meth compile_stmt
(v
)
730 if n_expr
!= null then
731 var e
= v
.compile_expr
(n_expr
)
732 v
.add_assignment
(v
.return_value
, e
)
734 v
.add_instr
("goto {v.return_label};")
738 redef class ABreakExpr
739 redef meth compile_stmt
(v
)
741 v
.add_instr
("goto {v.break_label};")
745 redef class AContinueExpr
746 redef meth compile_stmt
(v
)
748 v
.add_instr
("goto {v.continue_label};")
752 redef class AAbortExpr
753 redef meth compile_stmt
(v
)
755 v
.add_instr
("fprintf(stderr, \"Aborted\
"); {v.printf_locate_error(self)} nit_exit(1);")
760 redef meth compile_stmt
(v
)
762 if n_block
!= null then
763 v
.compile_stmt
(n_block
)
769 redef meth compile_stmt
(v
)
771 var e
= v
.compile_expr
(n_expr
)
772 v
.add_instr
("if (UNTAG_Bool({e})) \{ /*if*/")
774 if n_then
!= null then
776 v
.compile_stmt
(n_then
)
779 if n_else
!= null then
780 v
.add_instr
("} else \{ /*if*/")
782 v
.compile_stmt
(n_else
)
789 redef class AIfexprExpr
790 redef meth compile_expr
(v
)
792 var e
= v
.compile_expr
(n_expr
)
793 v
.add_instr
("if (UNTAG_Bool({e})) \{ /*if*/")
796 var e
= v
.ensure_var
(v
.compile_expr
(n_then
))
798 v
.add_instr
("} else \{ /*if*/")
801 var e2
= v
.ensure_var
(v
.compile_expr
(n_else
))
802 v
.add_assignment
(e
, e2
)
809 redef class AControlableBlock
810 meth compile_inside_block
(v
: CompilerVisitor) is abstract
811 redef meth compile_stmt
(v
)
813 var old_break_label
= v
.break_label
814 var old_continue_label
= v
.continue_label
815 var id
= v
.new_number
816 v
.break_label
= "break_{id}"
817 v
.continue_label
= "continue_{id}"
819 compile_inside_block
(v
)
822 v
.break_label
= old_break_label
823 v
.continue_label
= old_continue_label
827 redef class AWhileExpr
828 redef meth compile_inside_block
(v
)
830 v
.add_instr
("while (true) \{ /*while*/")
832 var e
= v
.compile_expr
(n_expr
)
833 v
.add_instr
("if (!UNTAG_Bool({e})) break; /* while*/")
835 if n_block
!= null then
836 v
.compile_stmt
(n_block
)
838 v
.add_instr
("{v.continue_label}: while(0);")
841 v
.add_instr
("{v.break_label}: while(0);")
846 redef meth compile_inside_block
(v
)
848 v
.compile_stmt
(n_vardecl
)
852 redef class AForVardeclExpr
853 redef meth compile_stmt
(v
)
855 var e
= v
.compile_expr
(n_expr
)
856 var prop
= n_expr
.stype
.local_class
.select_method
(once
"iterator".to_symbol
)
858 printl
("No iterator")
861 var ittype
= prop
.signature
.return_type
864 v
.add_assignment
(iter
, prop
.compile_call
(v
, [e
]))
865 var prop2
= ittype
.local_class
.select_method
(once
"is_ok".to_symbol
)
866 if prop2
== null then
870 var prop3
= ittype
.local_class
.select_method
(once
"item".to_symbol
)
871 if prop3
== null then
875 var prop4
= ittype
.local_class
.select_method
(once
"next".to_symbol
)
876 if prop4
== null then
880 v
.add_instr
("while (true) \{ /*for*/")
883 v
.add_assignment
(ok
, prop2
.compile_call
(v
, [iter
]))
884 v
.add_instr
("if (!UNTAG_Bool({ok})) break; /*for*/")
886 var e
= prop3
.compile_call
(v
, [iter
])
888 v
.varnames
[variable
] = e
890 assert par
isa AForExpr
891 var n_block
= par
.n_block
892 if n_block
!= null then
893 v
.compile_stmt
(n_block
)
895 v
.add_instr
("{v.continue_label}: while(0);")
896 e
= prop4
.compile_call
(v
, [iter
])
900 v
.add_instr
("{v.break_label}: while(0);")
904 redef class AAssertExpr
905 redef meth compile_stmt
(v
)
907 var e
= v
.compile_expr
(n_expr
)
910 s
= " '{n_id.text}' "
912 v
.add_instr
("if (!UNTAG_Bool({e})) \{ fprintf(stderr, \"Assert%s failed\", \"{s}\"); {v.printf_locate_error(self)} nit_exit
(1);}")
917 redef meth compile_expr(v)
919 return " {v.varnames[variable]} /*{variable.name}*/"
923 redef class AVarAssignExpr
924 redef meth compile_stmt(v)
926 var e = v.compile_expr(n_value)
927 v.add_assignment(v.varnames[variable], "{e} /*{variable.name}=*/")
931 redef class AVarReassignExpr
932 redef meth compile_stmt(v)
934 var e1 = v.varnames[variable]
935 var e2 = v.compile_expr(n_value)
936 var e3 = assign_method.compile_call(v, [e1, e2])
937 v.add_assignment(v.varnames[variable], "{e3} /*{variable.name}*/")
941 redef class ASelfExpr
942 redef meth compile_expr(v)
944 return v.method_params[0]
949 redef meth compile_expr(v)
951 var e = v.ensure_var(v.compile_expr(n_expr))
952 v.add_instr("if (!UNTAG_Bool({e})) \
{ /* or */")
955 var e2 = v.compile_expr(n_expr2)
956 v.add_assignment(e, e2)
964 redef meth compile_expr(v)
966 var e = v.ensure_var(v.compile_expr(n_expr))
967 v.add_instr("if (UNTAG_Bool({e})) \
{ /* and */")
970 var e2 = v.compile_expr(n_expr2)
971 v.add_assignment(e, e2)
979 redef meth compile_expr(v)
981 return " TAG_Bool(!UNTAG_Bool({v.compile_expr(n_expr)}))"
986 redef meth compile_expr(v)
988 var e = v.compile_expr(n_expr)
989 var e2 = v.compile_expr(n_expr2)
990 return "TAG_Bool(IS_EQUAL_NN({e},{e2}))"
995 redef meth compile_expr(v)
997 var e = v.compile_expr(n_expr)
998 return n_type.stype.compile_cast(v, e)
1002 redef class AAsCastExpr
1003 redef meth compile_expr(v)
1005 var e = v.compile_expr(n_expr)
1006 n_type.stype.compile_type_check(v, e, self)
1011 redef class ATrueExpr
1012 redef meth compile_expr(v)
1014 return " TAG_Bool(true)"
1018 redef class AFalseExpr
1019 redef meth compile_expr(v)
1021 return " TAG_Bool(false)"
1025 redef class AIntExpr
1026 redef meth compile_expr(v)
1028 return " TAG_Int({n_number.text})"
1032 redef class AFloatExpr
1033 redef meth compile_expr(v)
1035 return "BOX_Float({n_float.text})"
1039 redef class ACharExpr
1040 redef meth compile_expr(v)
1042 return " TAG_Char({n_char.text})"
1046 redef class AStringFormExpr
1047 redef meth compile_expr(v)
1049 var prop = stype.local_class.select_method(once "with_native
".to_symbol)
1051 return prop.compile_constructor_call(v, stype , ["BOX_NativeString(\
"{_cstring}\")", "TAG_Int({_cstring_length})"])
1054 # The raw string value
1055 protected meth string_text: String is abstract
1057 # The string in a C native format
1058 protected attr _cstring: String
1060 # The string length in bytes
1061 protected attr _cstring_length: Int
1063 # Compute _cstring and _cstring_length using string_text
1064 protected meth compute_string_info
1067 var str = string_text
1068 var res = new String
1070 while i < str.length do
1075 if c2 != '{' and c2 != '}' then
1085 _cstring_length = len
1089 redef class AStringExpr
1090 redef meth string_text do return n_string.text.substring(1, n_string.text.length - 2)
1092 redef class AStartStringExpr
1093 redef meth string_text do return n_string.text.substring(1, n_string.text.length - 2)
1095 redef class AMidStringExpr
1096 redef meth string_text do return n_string.text.substring(1, n_string.text.length - 2)
1098 redef class AEndStringExpr
1099 redef meth string_text do return n_string.text.substring(1, n_string.text.length - 2)
1102 redef class ASuperstringExpr
1103 redef meth compile_expr(v)
1105 var prop = stype.local_class.select_method(once "init".to_symbol)
1106 var recv = prop.compile_constructor_call(v, stype, new Array[String])
1108 var prop2 = stype.local_class.select_method(once "append
".to_symbol)
1110 var prop3 = stype.local_class.select_method(once "to_s
".to_symbol)
1111 for ne in n_exprs do
1112 var e = v.ensure_var(v.compile_expr(ne))
1113 if ne.stype != stype then
1114 v.add_assignment(e, prop3.compile_call(v, [e]))
1116 prop2.compile_call(v, [recv, e])
1123 redef class ANullExpr
1124 redef meth compile_expr(v)
1126 return " NIT_NULL /*null*/"
1130 redef class AArrayExpr
1131 redef meth compile_expr(v)
1133 var prop = stype.local_class.select_method(once "with_capacity
".to_symbol)
1134 var recv = prop.compile_constructor_call(v, stype, ["TAG_Int({n_exprs.length})"])
1136 var prop2 = stype.local_class.select_method(once "add
".to_symbol)
1137 for ne in n_exprs do
1138 var e = v.compile_expr(ne)
1139 prop2.compile_call(v, [recv, e])
1145 redef class ARangeExpr
1146 redef meth compile_expr(v)
1148 var prop = stype.local_class.select_method(propname)
1149 var e = v.compile_expr(n_expr)
1150 var e2 = v.compile_expr(n_expr2)
1151 return prop.compile_constructor_call(v, stype, [e, e2])
1153 # The constructor that must be used for the range
1154 protected meth propname: Symbol is abstract
1157 redef class ACrangeExpr
1158 redef meth propname do return once "init".to_symbol
1160 redef class AOrangeExpr
1161 redef meth propname do return once "without_last
".to_symbol
1164 redef class ASuperExpr
1165 redef meth compile_stmt(v)
1167 var e = compile_expr(v)
1168 if e != null then v.add_instr("{e};")
1171 redef meth compile_expr(v)
1173 var arity = v.method_params.length - 1
1174 if init_in_superclass != null then
1175 arity = init_in_superclass.signature.arity
1177 var args = new Array[String].with_capacity(arity + 1)
1178 args.add(v.method_params[0])
1179 if n_args.length != arity then
1180 for i in [0..arity[ do
1181 args.add(v.method_params[i + 1])
1185 args.add(v.compile_expr(na))
1188 #return "{prop.cname}({args.join(", ")}) /*super {prop.local_class}::{prop.name}*/"
1189 if init_in_superclass != null then
1190 return init_in_superclass.compile_call(v, args)
1192 if prop.global.is_init then args.add("init_table
")
1193 return prop.compile_super_call(v, args)
1198 redef class AAttrExpr
1199 redef meth compile_expr(v)
1201 var e = v.compile_expr(n_expr)
1202 return prop.compile_access(v, e)
1206 redef class AAttrAssignExpr
1207 redef meth compile_stmt(v)
1209 var e = v.compile_expr(n_expr)
1210 var e2 = v.compile_expr(n_value)
1211 v.add_assignment(prop.compile_access(v, e), e2)
1214 redef class AAttrReassignExpr
1215 redef meth compile_stmt(v)
1217 var e1 = v.compile_expr(n_expr)
1218 var e2 = prop.compile_access(v, e1)
1219 var e3 = v.compile_expr(n_value)
1220 var e4 = assign_method.compile_call(v, [e2, e3])
1221 v.add_assignment(e2, e4)
1225 redef class ASendExpr
1226 redef meth compile_expr(v)
1228 var recv = v.compile_expr(n_expr)
1229 var cargs = new Array[String]
1231 for a in arguments do
1232 cargs.add(v.compile_expr(a))
1235 var e = prop.compile_call(v, cargs)
1236 if prop.global.is_init then
1237 v.invoke_super_init_calls_after(prop)
1242 redef meth compile_stmt(v)
1244 var e = compile_expr(v)
1246 v.add_instr(e + ";")
1251 redef class ASendReassignExpr
1252 redef meth compile_expr(v)
1254 var recv = v.compile_expr(n_expr)
1255 var cargs = new Array[String]
1257 for a in arguments do
1258 cargs.add(v.compile_expr(a))
1261 var e2 = read_prop.compile_call(v, cargs)
1262 var e3 = v.compile_expr(n_value)
1263 var e4 = assign_method.compile_call(v, [e2, e3])
1265 return prop.compile_call(v, cargs)
1269 redef class ANewExpr
1270 redef meth compile_expr(v)
1272 var cargs = new Array[String]
1273 for a in arguments do
1274 cargs.add(v.compile_expr(a))
1276 return prop.compile_constructor_call(v, stype, cargs)
1280 redef class AProxyExpr
1281 redef meth compile_expr(v)
1283 return v.compile_expr(n_expr)
1287 redef class AOnceExpr
1288 redef meth compile_expr(v)
1290 var i = v.new_number
1291 var cvar = v.get_var
1292 v.add_decl("static val_t once_value_
{cvar}_
{i}; static int once_bool_
{cvar}_
{i};")
1293 v.add_instr("if (once_bool_
{cvar}_
{i}) {cvar} = once_value_
{cvar}_
{i};")
1294 v.add_instr("else \
{")
1297 var e = v.compile_expr(n_expr)
1298 v.add_assignment(cvar, e)
1299 v.add_instr("once_value_
{cvar}_
{i} = {cvar};")
1300 v.add_instr("once_bool_
{cvar}_
{i} = true;")