rta: store real types in live_cast_type
[nit.git] / src / separate_compiler.nit
1 # This file is part of NIT ( http://www.nitlanguage.org ).
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 # http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 # Separate compilation of a Nit program
16 module separate_compiler
17
18 import abstract_compiler
19 import layout_builders
20 import rapid_type_analysis
21 import collect_super_sends
22 import compiler_ffi
23
24 # Add separate compiler specific options
25 redef class ToolContext
26 # --separate
27 var opt_separate: OptionBool = new OptionBool("Use separate compilation", "--separate")
28 # --no-inline-intern
29 var opt_no_inline_intern: OptionBool = new OptionBool("Do not inline call to intern methods", "--no-inline-intern")
30 # --no-union-attribute
31 var opt_no_union_attribute: OptionBool = new OptionBool("Put primitive attibutes in a box instead of an union", "--no-union-attribute")
32 # --no-shortcut-equate
33 var opt_no_shortcut_equate: OptionBool = new OptionBool("Always call == in a polymorphic way", "--no-shortcut-equal")
34 # --inline-coloring-numbers
35 var opt_inline_coloring_numbers: OptionBool = new OptionBool("Inline colors and ids", "--inline-coloring-numbers")
36 # --use-naive-coloring
37 var opt_bm_typing: OptionBool = new OptionBool("Colorize items incrementaly, used to simulate binary matrix typing", "--bm-typing")
38 # --use-mod-perfect-hashing
39 var opt_phmod_typing: OptionBool = new OptionBool("Replace coloration by perfect hashing (with mod operator)", "--phmod-typing")
40 # --use-and-perfect-hashing
41 var opt_phand_typing: OptionBool = new OptionBool("Replace coloration by perfect hashing (with and operator)", "--phand-typing")
42 # --tables-metrics
43 var opt_tables_metrics: OptionBool = new OptionBool("Enable static size measuring of tables used for vft, typing and resolution", "--tables-metrics")
44
45 redef init
46 do
47 super
48 self.option_context.add_option(self.opt_separate)
49 self.option_context.add_option(self.opt_no_inline_intern)
50 self.option_context.add_option(self.opt_no_union_attribute)
51 self.option_context.add_option(self.opt_no_shortcut_equate)
52 self.option_context.add_option(self.opt_inline_coloring_numbers)
53 self.option_context.add_option(self.opt_bm_typing)
54 self.option_context.add_option(self.opt_phmod_typing)
55 self.option_context.add_option(self.opt_phand_typing)
56 self.option_context.add_option(self.opt_tables_metrics)
57 end
58 end
59
60 redef class ModelBuilder
61 fun run_separate_compiler(mainmodule: MModule, runtime_type_analysis: nullable RapidTypeAnalysis)
62 do
63 var time0 = get_time
64 self.toolcontext.info("*** GENERATING C ***", 1)
65
66 var compiler = new SeparateCompiler(mainmodule, self, runtime_type_analysis)
67 compiler.compile_header
68
69 # compile class structures
70 self.toolcontext.info("Property coloring", 2)
71 compiler.new_file("{mainmodule.name}.classes")
72 compiler.do_property_coloring
73 for m in mainmodule.in_importation.greaters do
74 for mclass in m.intro_mclasses do
75 if mclass.kind == abstract_kind or mclass.kind == interface_kind then continue
76 compiler.compile_class_to_c(mclass)
77 end
78 end
79
80 # The main function of the C
81 compiler.new_file("{mainmodule.name}.main")
82 compiler.compile_main_function
83
84 # compile methods
85 for m in mainmodule.in_importation.greaters do
86 self.toolcontext.info("Generate C for module {m}", 2)
87 compiler.new_file("{m.name}.sep")
88 compiler.compile_module_to_c(m)
89 end
90
91 # compile live & cast type structures
92 self.toolcontext.info("Type coloring", 2)
93 compiler.new_file("{mainmodule.name}.types")
94 var mtypes = compiler.do_type_coloring
95 for t in mtypes do
96 compiler.compile_type_to_c(t)
97 end
98
99 compiler.display_stats
100
101 var time1 = get_time
102 self.toolcontext.info("*** END GENERATING C: {time1-time0} ***", 2)
103 write_and_make(compiler)
104 end
105 end
106
107 # Singleton that store the knowledge about the separate compilation process
108 class SeparateCompiler
109 super AbstractCompiler
110
111 redef type VISITOR: SeparateCompilerVisitor
112
113 # The result of the RTA (used to know live types and methods)
114 var runtime_type_analysis: nullable RapidTypeAnalysis
115
116 private var undead_types: Set[MType] = new HashSet[MType]
117 private var partial_types: Set[MType] = new HashSet[MType]
118 private var live_unresolved_types: Map[MClassDef, Set[MType]] = new HashMap[MClassDef, HashSet[MType]]
119
120 private var type_layout: nullable Layout[MType]
121 private var resolution_layout: nullable Layout[MType]
122 protected var method_layout: nullable Layout[PropertyLayoutElement]
123 protected var attr_layout: nullable Layout[MAttribute]
124
125 init(mainmodule: MModule, mmbuilder: ModelBuilder, runtime_type_analysis: nullable RapidTypeAnalysis) do
126 super(mainmodule, mmbuilder)
127 var file = new_file("nit.common")
128 self.header = new CodeWriter(file)
129 self.runtime_type_analysis = runtime_type_analysis
130 self.compile_box_kinds
131 end
132
133 redef fun compile_header_structs do
134 self.header.add_decl("typedef void(*nitmethod_t)(void); /* general C type representing a Nit method. */")
135 self.compile_header_attribute_structs
136 self.header.add_decl("struct class \{ int box_kind; nitmethod_t vft[]; \}; /* general C type representing a Nit class. */")
137
138 # With resolution_table_table, all live type resolution are stored in a big table: resolution_table
139 self.header.add_decl("struct type \{ int id; const char *name; int color; short int is_nullable; const struct types *resolution_table; int table_size; int type_table[]; \}; /* general C type representing a Nit type. */")
140 self.header.add_decl("struct instance \{ const struct type *type; const struct class *class; nitattribute_t attrs[]; \}; /* general C type representing a Nit instance. */")
141
142 if modelbuilder.toolcontext.opt_phmod_typing.value or modelbuilder.toolcontext.opt_phand_typing.value then
143 self.header.add_decl("struct types \{ int mask; const struct type *types[]; \}; /* a list types (used for vts, fts and unresolved lists). */")
144 else
145 self.header.add_decl("struct types \{ int dummy; const struct type *types[]; \}; /* a list types (used for vts, fts and unresolved lists). */")
146 end
147
148 if modelbuilder.toolcontext.opt_phmod_typing.value then
149 self.header.add_decl("#define HASH(mask, id) ((mask)%(id))")
150 else if modelbuilder.toolcontext.opt_phand_typing.value then
151 self.header.add_decl("#define HASH(mask, id) ((mask)&(id))")
152 end
153
154 self.header.add_decl("typedef struct instance val; /* general C type representing a Nit instance. */")
155 end
156
157 fun compile_header_attribute_structs
158 do
159 if modelbuilder.toolcontext.opt_no_union_attribute.value then
160 self.header.add_decl("typedef void* nitattribute_t; /* general C type representing a Nit attribute. */")
161 else
162 self.header.add_decl("typedef union \{")
163 self.header.add_decl("void* val;")
164 for c, v in self.box_kinds do
165 var t = c.mclass_type
166 self.header.add_decl("{t.ctype} {t.ctypename};")
167 end
168 self.header.add_decl("\} nitattribute_t; /* general C type representing a Nit attribute. */")
169 end
170 end
171
172 fun compile_box_kinds
173 do
174 # Collect all bas box class
175 # FIXME: this is not completely fine with a separate compilation scheme
176 for classname in ["Int", "Bool", "Char", "Float", "NativeString", "Pointer"] do
177 var classes = self.mainmodule.model.get_mclasses_by_name(classname)
178 if classes == null then continue
179 assert classes.length == 1 else print classes.join(", ")
180 self.box_kinds[classes.first] = self.box_kinds.length + 1
181 end
182 end
183
184 var box_kinds = new HashMap[MClass, Int]
185
186 fun box_kind_of(mclass: MClass): Int
187 do
188 if mclass.mclass_type.ctype == "val*" then
189 return 0
190 else if mclass.kind == extern_kind then
191 return self.box_kinds[self.mainmodule.get_primitive_class("Pointer")]
192 else
193 return self.box_kinds[mclass]
194 end
195
196 end
197
198 fun compile_color_consts(colors: Map[Object, Int]) do
199 var v = new_visitor
200 for m, c in colors do
201 compile_color_const(v, m, c)
202 end
203 end
204
205 fun compile_color_const(v: SeparateCompilerVisitor, m: Object, color: Int) do
206 if color_consts_done.has(m) then return
207 if m isa MProperty then
208 if modelbuilder.toolcontext.opt_inline_coloring_numbers.value then
209 self.provide_declaration(m.const_color, "#define {m.const_color} {color}")
210 else
211 self.provide_declaration(m.const_color, "extern const int {m.const_color};")
212 v.add("const int {m.const_color} = {color};")
213 end
214 else if m isa MPropDef then
215 if modelbuilder.toolcontext.opt_inline_coloring_numbers.value then
216 self.provide_declaration(m.const_color, "#define {m.const_color} {color}")
217 else
218 self.provide_declaration(m.const_color, "extern const int {m.const_color};")
219 v.add("const int {m.const_color} = {color};")
220 end
221 else if m isa MType then
222 if modelbuilder.toolcontext.opt_inline_coloring_numbers.value then
223 self.provide_declaration(m.const_color, "#define {m.const_color} {color}")
224 else
225 self.provide_declaration(m.const_color, "extern const int {m.const_color};")
226 v.add("const int {m.const_color} = {color};")
227 end
228 end
229 color_consts_done.add(m)
230 end
231
232 private var color_consts_done = new HashSet[Object]
233
234 # colorize classe properties
235 fun do_property_coloring do
236 var mclasses = new HashSet[MClass].from(modelbuilder.model.mclasses)
237
238 # Layouts
239 var method_layout_builder: PropertyLayoutBuilder[PropertyLayoutElement]
240 var attribute_layout_builder: PropertyLayoutBuilder[MAttribute]
241 #FIXME PH and BM layouts too slow for large programs
242 #if modelbuilder.toolcontext.opt_bm_typing.value then
243 # method_layout_builder = new MMethodBMizer(self.mainmodule)
244 # attribute_layout_builder = new MAttributeBMizer(self.mainmodule)
245 #else if modelbuilder.toolcontext.opt_phmod_typing.value then
246 # method_layout_builder = new MMethodHasher(new PHModOperator, self.mainmodule)
247 # attribute_layout_builder = new MAttributeHasher(new PHModOperator, self.mainmodule)
248 #else if modelbuilder.toolcontext.opt_phand_typing.value then
249 # method_layout_builder = new MMethodHasher(new PHAndOperator, self.mainmodule)
250 # attribute_layout_builder = new MAttributeHasher(new PHAndOperator, self.mainmodule)
251 #else
252
253 var class_layout_builder = new MClassColorer(self.mainmodule)
254 class_layout_builder.build_layout(mclasses)
255 method_layout_builder = new MPropertyColorer[PropertyLayoutElement](self.mainmodule, class_layout_builder)
256 attribute_layout_builder = new MPropertyColorer[MAttribute](self.mainmodule, class_layout_builder)
257 #end
258
259 # lookup properties to build layout with
260 var mmethods = new HashMap[MClass, Set[PropertyLayoutElement]]
261 var mattributes = new HashMap[MClass, Set[MAttribute]]
262 for mclass in mclasses do
263 mmethods[mclass] = new HashSet[PropertyLayoutElement]
264 mattributes[mclass] = new HashSet[MAttribute]
265 for mprop in self.mainmodule.properties(mclass) do
266 if mprop isa MMethod then
267 mmethods[mclass].add(mprop)
268 else if mprop isa MAttribute then
269 mattributes[mclass].add(mprop)
270 end
271 end
272 end
273
274 # lookup super calls and add it to the list of mmethods to build layout with
275 var super_calls
276 if runtime_type_analysis != null then
277 super_calls = runtime_type_analysis.live_super_sends
278 else
279 super_calls = modelbuilder.collect_super_sends
280 end
281 for mmethoddef in super_calls do
282 var mclass = mmethoddef.mclassdef.mclass
283 mmethods[mclass].add(mmethoddef)
284 for descendant in mclass.in_hierarchy(self.mainmodule).smallers do
285 mmethods[descendant].add(mmethoddef)
286 end
287 end
288
289 # methods coloration
290 self.method_layout = method_layout_builder.build_layout(mmethods)
291 self.method_tables = build_method_tables(mclasses, super_calls)
292 self.compile_color_consts(method_layout.pos)
293
294 # attribute null color to dead supercalls
295 for mmodule in self.mainmodule.in_importation.greaters do
296 for mclassdef in mmodule.mclassdefs do
297 for mpropdef in mclassdef.mpropdefs do
298 if mpropdef.has_supercall then
299 compile_color_const(new_visitor, mpropdef, -1)
300 end
301 end
302 end
303 end
304
305 # attributes coloration
306 self.attr_layout = attribute_layout_builder.build_layout(mattributes)
307 self.attr_tables = build_attr_tables(mclasses)
308 self.compile_color_consts(attr_layout.pos)
309 end
310
311 fun build_method_tables(mclasses: Set[MClass], super_calls: Set[MMethodDef]): Map[MClass, Array[nullable MPropDef]] do
312 var layout = self.method_layout
313 var tables = new HashMap[MClass, Array[nullable MPropDef]]
314 for mclass in mclasses do
315 var table = new Array[nullable MPropDef]
316 var supercalls = new List[MMethodDef]
317
318 # first, fill table from parents by reverse linearization order
319 var parents = new Array[MClass]
320 if mainmodule.flatten_mclass_hierarchy.has(mclass) then
321 parents = mclass.in_hierarchy(mainmodule).greaters.to_a
322 self.mainmodule.linearize_mclasses(parents)
323 end
324
325 for parent in parents do
326 if parent == mclass then continue
327 for mproperty in self.mainmodule.properties(parent) do
328 if not mproperty isa MMethod then continue
329 var color = layout.pos[mproperty]
330 if table.length <= color then
331 for i in [table.length .. color[ do
332 table[i] = null
333 end
334 end
335 for mpropdef in mproperty.mpropdefs do
336 if mpropdef.mclassdef.mclass == parent then
337 table[color] = mpropdef
338 end
339 end
340 end
341
342 # lookup for super calls in super classes
343 for mmethoddef in super_calls do
344 for mclassdef in parent.mclassdefs do
345 if mclassdef.mpropdefs.has(mmethoddef) then
346 supercalls.add(mmethoddef)
347 end
348 end
349 end
350 end
351
352 # then override with local properties
353 for mproperty in self.mainmodule.properties(mclass) do
354 if not mproperty isa MMethod then continue
355 var color = layout.pos[mproperty]
356 if table.length <= color then
357 for i in [table.length .. color[ do
358 table[i] = null
359 end
360 end
361 for mpropdef in mproperty.mpropdefs do
362 if mpropdef.mclassdef.mclass == mclass then
363 table[color] = mpropdef
364 end
365 end
366 end
367
368 # lookup for super calls in local class
369 for mmethoddef in super_calls do
370 for mclassdef in mclass.mclassdefs do
371 if mclassdef.mpropdefs.has(mmethoddef) then
372 supercalls.add(mmethoddef)
373 end
374 end
375 end
376 # insert super calls in table according to receiver
377 for supercall in supercalls do
378 var color = layout.pos[supercall]
379 if table.length <= color then
380 for i in [table.length .. color[ do
381 table[i] = null
382 end
383 end
384 var mmethoddef = supercall.lookup_next_definition(self.mainmodule, mclass.intro.bound_mtype)
385 table[color] = mmethoddef
386 end
387 tables[mclass] = table
388 end
389 return tables
390 end
391
392 fun build_attr_tables(mclasses: Set[MClass]): Map[MClass, Array[nullable MPropDef]] do
393 var layout = self.attr_layout
394 var tables = new HashMap[MClass, Array[nullable MPropDef]]
395 for mclass in mclasses do
396 var table = new Array[nullable MPropDef]
397 # first, fill table from parents by reverse linearization order
398 var parents = new Array[MClass]
399 if mainmodule.flatten_mclass_hierarchy.has(mclass) then
400 parents = mclass.in_hierarchy(mainmodule).greaters.to_a
401 self.mainmodule.linearize_mclasses(parents)
402 end
403 for parent in parents do
404 if parent == mclass then continue
405 for mproperty in self.mainmodule.properties(parent) do
406 if not mproperty isa MAttribute then continue
407 var color = layout.pos[mproperty]
408 if table.length <= color then
409 for i in [table.length .. color[ do
410 table[i] = null
411 end
412 end
413 for mpropdef in mproperty.mpropdefs do
414 if mpropdef.mclassdef.mclass == parent then
415 table[color] = mpropdef
416 end
417 end
418 end
419 end
420
421 # then override with local properties
422 for mproperty in self.mainmodule.properties(mclass) do
423 if not mproperty isa MAttribute then continue
424 var color = layout.pos[mproperty]
425 if table.length <= color then
426 for i in [table.length .. color[ do
427 table[i] = null
428 end
429 end
430 for mpropdef in mproperty.mpropdefs do
431 if mpropdef.mclassdef.mclass == mclass then
432 table[color] = mpropdef
433 end
434 end
435 end
436 tables[mclass] = table
437 end
438 return tables
439 end
440
441 # colorize live types of the program
442 private fun do_type_coloring: POSet[MType] do
443 var mtypes = new HashSet[MType]
444 mtypes.add_all(self.runtime_type_analysis.live_types)
445 mtypes.add_all(self.runtime_type_analysis.live_cast_types)
446 mtypes.add_all(self.undead_types)
447 for c in self.box_kinds.keys do
448 mtypes.add(c.mclass_type)
449 end
450
451 for mtype in mtypes do
452 retrieve_partial_types(mtype)
453 end
454 mtypes.add_all(self.partial_types)
455
456 # Typing Layout
457 var layout_builder: TypingLayoutBuilder[MType]
458 if modelbuilder.toolcontext.opt_bm_typing.value then
459 layout_builder = new MTypeBMizer(self.mainmodule)
460 else if modelbuilder.toolcontext.opt_phmod_typing.value then
461 layout_builder = new MTypeHasher(new PHModOperator, self.mainmodule)
462 else if modelbuilder.toolcontext.opt_phand_typing.value then
463 layout_builder = new MTypeHasher(new PHAndOperator, self.mainmodule)
464 else
465 layout_builder = new MTypeColorer(self.mainmodule)
466 end
467
468 # colorize types
469 self.type_layout = layout_builder.build_layout(mtypes)
470 var poset = layout_builder.poset.as(not null)
471 self.type_tables = self.build_type_tables(poset)
472
473 # VT and FT are stored with other unresolved types in the big resolution_tables
474 self.compile_resolution_tables(mtypes)
475
476 return poset
477 end
478
479 # Build type tables
480 fun build_type_tables(mtypes: POSet[MType]): Map[MType, Array[nullable MType]] do
481 var tables = new HashMap[MType, Array[nullable MType]]
482 var layout = self.type_layout
483 for mtype in mtypes do
484 var table = new Array[nullable MType]
485 for sup in mtypes[mtype].greaters do
486 var color: Int
487 if layout isa PHLayout[MType, MType] then
488 color = layout.hashes[mtype][sup]
489 else
490 color = layout.pos[sup]
491 end
492 if table.length <= color then
493 for i in [table.length .. color[ do
494 table[i] = null
495 end
496 end
497 table[color] = sup
498 end
499 tables[mtype] = table
500 end
501 return tables
502 end
503
504 protected fun compile_resolution_tables(mtypes: Set[MType]) do
505 # resolution_tables is used to perform a type resolution at runtime in O(1)
506
507 # During the visit of the body of classes, live_unresolved_types are collected
508 # and associated to
509 # Collect all live_unresolved_types (visited in the body of classes)
510
511 # Determinate fo each livetype what are its possible requested anchored types
512 var mtype2unresolved = new HashMap[MClassType, Set[MType]]
513 for mtype in self.runtime_type_analysis.live_types do
514 var set = new HashSet[MType]
515 for cd in mtype.collect_mclassdefs(self.mainmodule) do
516 if self.live_unresolved_types.has_key(cd) then
517 set.add_all(self.live_unresolved_types[cd])
518 end
519 end
520 mtype2unresolved[mtype] = set
521 end
522
523 # Compute the table layout with the prefered method
524 var resolution_builder: ResolutionLayoutBuilder
525 if modelbuilder.toolcontext.opt_bm_typing.value then
526 resolution_builder = new ResolutionBMizer
527 else if modelbuilder.toolcontext.opt_phmod_typing.value then
528 resolution_builder = new ResolutionHasher(new PHModOperator)
529 else if modelbuilder.toolcontext.opt_phand_typing.value then
530 resolution_builder = new ResolutionHasher(new PHAndOperator)
531 else
532 resolution_builder = new ResolutionColorer
533 end
534 self.resolution_layout = resolution_builder.build_layout(mtype2unresolved)
535 self.resolution_tables = self.build_resolution_tables(mtype2unresolved)
536
537 # Compile a C constant for each collected unresolved type.
538 # Either to a color, or to -1 if the unresolved type is dead (no live receiver can require it)
539 var all_unresolved = new HashSet[MType]
540 for t in self.live_unresolved_types.values do
541 all_unresolved.add_all(t)
542 end
543 var all_unresolved_types_colors = new HashMap[MType, Int]
544 for t in all_unresolved do
545 if self.resolution_layout.pos.has_key(t) then
546 all_unresolved_types_colors[t] = self.resolution_layout.pos[t]
547 else
548 all_unresolved_types_colors[t] = -1
549 end
550 end
551 self.compile_color_consts(all_unresolved_types_colors)
552
553 #print "tables"
554 #for k, v in unresolved_types_tables.as(not null) do
555 # print "{k}: {v.join(", ")}"
556 #end
557 #print ""
558 end
559
560 fun build_resolution_tables(elements: Map[MClassType, Set[MType]]): Map[MClassType, Array[nullable MType]] do
561 var tables = new HashMap[MClassType, Array[nullable MType]]
562 var layout = self.resolution_layout
563 for mclasstype, mtypes in elements do
564 var table = new Array[nullable MType]
565 for mtype in mtypes do
566 var color: Int
567 if layout isa PHLayout[MClassType, MType] then
568 color = layout.hashes[mclasstype][mtype]
569 else
570 color = layout.pos[mtype]
571 end
572 if table.length <= color then
573 for i in [table.length .. color[ do
574 table[i] = null
575 end
576 end
577 table[color] = mtype
578 end
579 tables[mclasstype] = table
580 end
581 return tables
582 end
583
584 fun retrieve_partial_types(mtype: MType) do
585 # add formal types arguments to mtypes
586 if mtype isa MGenericType then
587 for ft in mtype.arguments do
588 if ft.need_anchor then
589 print("Why do we need anchor here ?")
590 abort
591 end
592 self.partial_types.add(ft)
593 retrieve_partial_types(ft)
594 end
595 end
596 var mclass_type: MClassType
597 if mtype isa MNullableType then
598 mclass_type = mtype.mtype.as(MClassType)
599 else
600 mclass_type = mtype.as(MClassType)
601 end
602
603 # add virtual types to mtypes
604 for vt in self.mainmodule.properties(mclass_type.mclass) do
605 if vt isa MVirtualTypeProp then
606 var anchored = vt.mvirtualtype.lookup_bound(self.mainmodule, mclass_type).anchor_to(self.mainmodule, mclass_type)
607 self.partial_types.add(anchored)
608 end
609 end
610 end
611
612 # Separately compile all the method definitions of the module
613 fun compile_module_to_c(mmodule: MModule)
614 do
615 var old_module = self.mainmodule
616 self.mainmodule = mmodule
617 for cd in mmodule.mclassdefs do
618 for pd in cd.mpropdefs do
619 if not pd isa MMethodDef then continue
620 #print "compile {pd} @ {cd} @ {mmodule}"
621 var r = pd.separate_runtime_function
622 r.compile_to_c(self)
623 var r2 = pd.virtual_runtime_function
624 r2.compile_to_c(self)
625 end
626 end
627 self.mainmodule = old_module
628 end
629
630 # Globaly compile the type structure of a live type
631 fun compile_type_to_c(mtype: MType)
632 do
633 var c_name = mtype.c_name
634 var v = new SeparateCompilerVisitor(self)
635 v.add_decl("/* runtime type {mtype} */")
636
637 # extern const struct type_X
638 self.provide_declaration("type_{c_name}", "extern const struct type type_{c_name};")
639
640 # const struct type_X
641 v.add_decl("const struct type type_{c_name} = \{")
642 v.add_decl("{self.type_layout.ids[mtype]},")
643 v.add_decl("\"{mtype}\", /* class_name_string */")
644 var layout = self.type_layout
645 if layout isa PHLayout[MType, MType] then
646 v.add_decl("{layout.masks[mtype]},")
647 else
648 v.add_decl("{layout.pos[mtype]},")
649 end
650 if mtype isa MNullableType then
651 v.add_decl("1,")
652 else
653 v.add_decl("0,")
654 end
655 if compile_type_resolution_table(mtype) then
656 v.require_declaration("resolution_table_{c_name}")
657 v.add_decl("&resolution_table_{c_name},")
658 else
659 v.add_decl("NULL,")
660 end
661 v.add_decl("{self.type_tables[mtype].length},")
662 v.add_decl("\{")
663 for stype in self.type_tables[mtype] do
664 if stype == null then
665 v.add_decl("-1, /* empty */")
666 else
667 v.add_decl("{self.type_layout.ids[stype]}, /* {stype} */")
668 end
669 end
670 v.add_decl("\},")
671 v.add_decl("\};")
672 end
673
674 fun compile_type_resolution_table(mtype: MType): Bool do
675
676 var mclass_type: MClassType
677 if mtype isa MNullableType then
678 mclass_type = mtype.mtype.as(MClassType)
679 else
680 mclass_type = mtype.as(MClassType)
681 end
682 if not self.resolution_tables.has_key(mclass_type) then return false
683
684 var layout = self.resolution_layout
685
686 # extern const struct resolution_table_X resolution_table_X
687 self.provide_declaration("resolution_table_{mtype.c_name}", "extern const struct types resolution_table_{mtype.c_name};")
688
689 # const struct fts_table_X fts_table_X
690 var v = new_visitor
691 v.add_decl("const struct types resolution_table_{mtype.c_name} = \{")
692 if layout isa PHLayout[MClassType, MType] then
693 v.add_decl("{layout.masks[mclass_type]},")
694 else
695 v.add_decl("0, /* dummy */")
696 end
697 v.add_decl("\{")
698 for t in self.resolution_tables[mclass_type] do
699 if t == null then
700 v.add_decl("NULL, /* empty */")
701 else
702 # The table stores the result of the type resolution
703 # Therefore, for a receiver `mclass_type`, and a unresolved type `t`
704 # the value stored is tv.
705 var tv = t.resolve_for(mclass_type, mclass_type, self.mainmodule, true)
706 # FIXME: What typeids means here? How can a tv not be live?
707 if self.type_layout.ids.has_key(tv) then
708 v.require_declaration("type_{tv.c_name}")
709 v.add_decl("&type_{tv.c_name}, /* {t}: {tv} */")
710 else
711 v.add_decl("NULL, /* empty ({t}: {tv} not a live type) */")
712 end
713 end
714 end
715 v.add_decl("\}")
716 v.add_decl("\};")
717 return true
718 end
719
720 # Globally compile the table of the class mclass
721 # In a link-time optimisation compiler, tables are globally computed
722 # In a true separate compiler (a with dynamic loading) you cannot do this unfortnally
723 fun compile_class_to_c(mclass: MClass)
724 do
725 var mtype = mclass.intro.bound_mtype
726 var c_name = mclass.c_name
727
728 var vft = self.method_tables[mclass]
729 var attrs = self.attr_tables[mclass]
730 var v = new_visitor
731
732 var is_dead = runtime_type_analysis != null and not runtime_type_analysis.live_classes.has(mclass) and mtype.ctype == "val*" and mclass.name != "NativeArray"
733
734 v.add_decl("/* runtime class {c_name} */")
735
736 # Build class vft
737 if not is_dead then
738 self.provide_declaration("class_{c_name}", "extern const struct class class_{c_name};")
739 v.add_decl("const struct class class_{c_name} = \{")
740 v.add_decl("{self.box_kind_of(mclass)}, /* box_kind */")
741 v.add_decl("\{")
742 for i in [0 .. vft.length[ do
743 var mpropdef = vft[i]
744 if mpropdef == null then
745 v.add_decl("NULL, /* empty */")
746 else
747 assert mpropdef isa MMethodDef
748 var rf = mpropdef.virtual_runtime_function
749 v.require_declaration(rf.c_name)
750 v.add_decl("(nitmethod_t){rf.c_name}, /* pointer to {mclass.intro_mmodule}:{mclass}:{mpropdef} */")
751 end
752 end
753 v.add_decl("\}")
754 v.add_decl("\};")
755 end
756
757 if mtype.ctype != "val*" then
758 #Build instance struct
759 self.header.add_decl("struct instance_{c_name} \{")
760 self.header.add_decl("const struct type *type;")
761 self.header.add_decl("const struct class *class;")
762 self.header.add_decl("{mtype.ctype} value;")
763 self.header.add_decl("\};")
764
765 if not self.runtime_type_analysis.live_types.has(mtype) then return
766
767 #Build BOX
768 self.header.add_decl("val* BOX_{c_name}({mtype.ctype});")
769 v.add_decl("/* allocate {mtype} */")
770 v.add_decl("val* BOX_{mtype.c_name}({mtype.ctype} value) \{")
771 v.add("struct instance_{c_name}*res = nit_alloc(sizeof(struct instance_{c_name}));")
772 v.require_declaration("type_{c_name}")
773 v.add("res->type = &type_{c_name};")
774 v.require_declaration("class_{c_name}")
775 v.add("res->class = &class_{c_name};")
776 v.add("res->value = value;")
777 v.add("return (val*)res;")
778 v.add("\}")
779 return
780 else if mclass.name == "NativeArray" then
781 #Build instance struct
782 self.header.add_decl("struct instance_{c_name} \{")
783 self.header.add_decl("const struct type *type;")
784 self.header.add_decl("const struct class *class;")
785 # NativeArrays are just a instance header followed by an array of values
786 self.header.add_decl("val* values[0];")
787 self.header.add_decl("\};")
788
789 #Build NEW
790 self.provide_declaration("NEW_{c_name}", "{mtype.ctype} NEW_{c_name}(int length, const struct type* type);")
791 v.add_decl("/* allocate {mtype} */")
792 v.add_decl("{mtype.ctype} NEW_{c_name}(int length, const struct type* type) \{")
793 var res = v.new_named_var(mtype, "self")
794 res.is_exact = true
795 var mtype_elt = mtype.arguments.first
796 v.add("{res} = nit_alloc(sizeof(struct instance_{c_name}) + length*sizeof({mtype_elt.ctype}));")
797 v.add("{res}->type = type;")
798 hardening_live_type(v, "type")
799 v.require_declaration("class_{c_name}")
800 v.add("{res}->class = &class_{c_name};")
801 v.add("return {res};")
802 v.add("\}")
803 return
804 end
805
806 #Build NEW
807 self.provide_declaration("NEW_{c_name}", "{mtype.ctype} NEW_{c_name}(const struct type* type);")
808 v.add_decl("/* allocate {mtype} */")
809 v.add_decl("{mtype.ctype} NEW_{c_name}(const struct type* type) \{")
810 if is_dead then
811 v.add_abort("{mclass} is DEAD")
812 else
813 var res = v.new_named_var(mtype, "self")
814 res.is_exact = true
815 v.add("{res} = nit_alloc(sizeof(struct instance) + {attrs.length}*sizeof(nitattribute_t));")
816 v.add("{res}->type = type;")
817 hardening_live_type(v, "type")
818 v.require_declaration("class_{c_name}")
819 v.add("{res}->class = &class_{c_name};")
820 self.generate_init_attr(v, res, mtype)
821 v.add("return {res};")
822 end
823 v.add("\}")
824 end
825
826 # Add a dynamic test to ensure that the type referenced by `t` is a live type
827 fun hardening_live_type(v: VISITOR, t: String)
828 do
829 if not v.compiler.modelbuilder.toolcontext.opt_hardening.value then return
830 v.add("if({t} == NULL) \{")
831 v.add_abort("type null")
832 v.add("\}")
833 v.add("if({t}->resolution_table == NULL) \{")
834 v.add("fprintf(stderr, \"Insantiation of a dead type: %s\\n\", {t}->name);")
835 v.add_abort("type dead")
836 v.add("\}")
837 end
838
839 redef fun new_visitor do return new SeparateCompilerVisitor(self)
840
841 # Stats
842
843 private var type_tables: Map[MType, Array[nullable MType]] = new HashMap[MType, Array[nullable MType]]
844 private var resolution_tables: Map[MClassType, Array[nullable MType]] = new HashMap[MClassType, Array[nullable MType]]
845 protected var method_tables: Map[MClass, Array[nullable MPropDef]] = new HashMap[MClass, Array[nullable MPropDef]]
846 protected var attr_tables: Map[MClass, Array[nullable MPropDef]] = new HashMap[MClass, Array[nullable MPropDef]]
847
848 redef fun display_stats
849 do
850 super
851 if self.modelbuilder.toolcontext.opt_tables_metrics.value then
852 display_sizes
853 end
854 end
855
856 fun display_sizes
857 do
858 print "# size of subtyping tables"
859 print "\ttotal \tholes"
860 var total = 0
861 var holes = 0
862 for t, table in type_tables do
863 total += table.length
864 for e in table do if e == null then holes += 1
865 end
866 print "\t{total}\t{holes}"
867
868 print "# size of resolution tables"
869 print "\ttotal \tholes"
870 total = 0
871 holes = 0
872 for t, table in resolution_tables do
873 total += table.length
874 for e in table do if e == null then holes += 1
875 end
876 print "\t{total}\t{holes}"
877
878 print "# size of methods tables"
879 print "\ttotal \tholes"
880 total = 0
881 holes = 0
882 for t, table in method_tables do
883 total += table.length
884 for e in table do if e == null then holes += 1
885 end
886 print "\t{total}\t{holes}"
887
888 print "# size of attributes tables"
889 print "\ttotal \tholes"
890 total = 0
891 holes = 0
892 for t, table in attr_tables do
893 total += table.length
894 for e in table do if e == null then holes += 1
895 end
896 print "\t{total}\t{holes}"
897 end
898
899 redef fun compile_nitni_structs
900 do
901 self.header.add_decl("struct nitni_instance \{struct instance *value;\};")
902 end
903
904 redef fun finalize_ffi_for_module(nmodule)
905 do
906 self.mainmodule = nmodule.mmodule.as(not null)
907 super
908 end
909 end
910
911 # A visitor on the AST of property definition that generate the C code of a separate compilation process.
912 class SeparateCompilerVisitor
913 super AbstractCompilerVisitor
914
915 redef type COMPILER: SeparateCompiler
916
917 redef fun adapt_signature(m, args)
918 do
919 var msignature = m.msignature.resolve_for(m.mclassdef.bound_mtype, m.mclassdef.bound_mtype, m.mclassdef.mmodule, true)
920 var recv = args.first
921 if recv.mtype.ctype != m.mclassdef.mclass.mclass_type.ctype then
922 args.first = self.autobox(args.first, m.mclassdef.mclass.mclass_type)
923 end
924 for i in [0..msignature.arity[ do
925 var t = msignature.mparameters[i].mtype
926 if i == msignature.vararg_rank then
927 t = args[i+1].mtype
928 end
929 args[i+1] = self.autobox(args[i+1], t)
930 end
931 end
932
933 redef fun autobox(value, mtype)
934 do
935 if value.mtype == mtype then
936 return value
937 else if value.mtype.ctype == "val*" and mtype.ctype == "val*" then
938 return value
939 else if value.mtype.ctype == "val*" then
940 return self.new_expr("((struct instance_{mtype.c_name}*){value})->value; /* autounbox from {value.mtype} to {mtype} */", mtype)
941 else if mtype.ctype == "val*" then
942 var valtype = value.mtype.as(MClassType)
943 var res = self.new_var(mtype)
944 if compiler.runtime_type_analysis != null and not compiler.runtime_type_analysis.live_types.has(valtype) then
945 self.add("/*no autobox from {value.mtype} to {mtype}: {value.mtype} is not live! */")
946 self.add("printf(\"Dead code executed!\\n\"); show_backtrace(1);")
947 return res
948 end
949 self.add("{res} = BOX_{valtype.c_name}({value}); /* autobox from {value.mtype} to {mtype} */")
950 return res
951 else if value.mtype.cname_blind == "void*" and mtype.cname_blind == "void*" then
952 return value
953 else
954 # Bad things will appen!
955 var res = self.new_var(mtype)
956 self.add("/* {res} left unintialized (cannot convert {value.mtype} to {mtype}) */")
957 self.add("printf(\"Cast error: Cannot cast %s to %s.\\n\", \"{value.mtype}\", \"{mtype}\"); show_backtrace(1);")
958 return res
959 end
960 end
961
962 # Return a C expression returning the runtime type structure of the value
963 # The point of the method is to works also with primitives types.
964 fun type_info(value: RuntimeVariable): String
965 do
966 if value.mtype.ctype == "val*" then
967 return "{value}->type"
968 else
969 self.require_declaration("type_{value.mtype.c_name}")
970 return "(&type_{value.mtype.c_name})"
971 end
972 end
973
974 redef fun send(mmethod, arguments)
975 do
976 self.varargize(mmethod.intro, mmethod.intro.msignature.as(not null), arguments)
977
978 if arguments.first.mcasttype.ctype != "val*" then
979 # In order to shortcut the primitive, we need to find the most specific method
980 # Howverr, because of performance (no flattening), we always work on the realmainmodule
981 var m = self.compiler.mainmodule
982 self.compiler.mainmodule = self.compiler.realmainmodule
983 var res = self.monomorphic_send(mmethod, arguments.first.mcasttype, arguments)
984 self.compiler.mainmodule = m
985 return res
986 end
987
988 return table_send(mmethod, arguments, mmethod.const_color)
989 end
990
991 private fun table_send(mmethod: MMethod, arguments: Array[RuntimeVariable], const_color: String): nullable RuntimeVariable
992 do
993 var res: nullable RuntimeVariable
994 var msignature = mmethod.intro.msignature.resolve_for(mmethod.intro.mclassdef.bound_mtype, mmethod.intro.mclassdef.bound_mtype, mmethod.intro.mclassdef.mmodule, true)
995 var ret = msignature.return_mtype
996 if mmethod.is_new then
997 ret = arguments.first.mtype
998 res = self.new_var(ret)
999 else if ret == null then
1000 res = null
1001 else
1002 res = self.new_var(ret)
1003 end
1004
1005 var s = new Buffer
1006 var ss = new Buffer
1007
1008 var recv = arguments.first
1009 s.append("val*")
1010 ss.append("{recv}")
1011 for i in [0..msignature.arity[ do
1012 var a = arguments[i+1]
1013 var t = msignature.mparameters[i].mtype
1014 if i == msignature.vararg_rank then
1015 t = arguments[i+1].mcasttype
1016 end
1017 s.append(", {t.ctype}")
1018 a = self.autobox(a, t)
1019 ss.append(", {a}")
1020 end
1021
1022 var consider_null = not self.compiler.modelbuilder.toolcontext.opt_no_check_other.value or mmethod.name == "==" or mmethod.name == "!="
1023 var maybenull = recv.mcasttype isa MNullableType and consider_null
1024 if maybenull then
1025 self.add("if ({recv} == NULL) \{")
1026 if mmethod.name == "==" then
1027 assert res != null
1028 var arg = arguments[1]
1029 if arg.mcasttype isa MNullableType then
1030 self.add("{res} = ({arg} == NULL);")
1031 else if arg.mcasttype isa MNullType then
1032 self.add("{res} = 1; /* is null */")
1033 else
1034 self.add("{res} = 0; /* {arg.inspect} cannot be null */")
1035 end
1036 else if mmethod.name == "!=" then
1037 assert res != null
1038 var arg = arguments[1]
1039 if arg.mcasttype isa MNullableType then
1040 self.add("{res} = ({arg} != NULL);")
1041 else if arg.mcasttype isa MNullType then
1042 self.add("{res} = 0; /* is null */")
1043 else
1044 self.add("{res} = 1; /* {arg.inspect} cannot be null */")
1045 end
1046 else
1047 self.add_abort("Reciever is null")
1048 end
1049 self.add("\} else \{")
1050 end
1051 if not self.compiler.modelbuilder.toolcontext.opt_no_shortcut_equate.value and (mmethod.name == "==" or mmethod.name == "!=") then
1052 assert res != null
1053 # Recv is not null, thus is arg is, it is easy to conclude (and respect the invariants)
1054 var arg = arguments[1]
1055 if arg.mcasttype isa MNullType then
1056 if mmethod.name == "==" then
1057 self.add("{res} = 0; /* arg is null but recv is not */")
1058 else
1059 self.add("{res} = 1; /* arg is null and recv is not */")
1060 end
1061 if maybenull then
1062 self.add("\}")
1063 end
1064 return res
1065 end
1066 end
1067
1068 var r
1069 if ret == null then r = "void" else r = ret.ctype
1070 self.require_declaration(const_color)
1071 var call = "(({r} (*)({s}))({arguments.first}->class->vft[{const_color}]))({ss}) /* {mmethod} on {arguments.first.inspect}*/"
1072
1073 if res != null then
1074 self.add("{res} = {call};")
1075 else
1076 self.add("{call};")
1077 end
1078
1079 if maybenull then
1080 self.add("\}")
1081 end
1082
1083 return res
1084 end
1085
1086 redef fun call(mmethoddef, recvtype, arguments)
1087 do
1088 var res: nullable RuntimeVariable
1089 var ret = mmethoddef.msignature.return_mtype
1090 if mmethoddef.mproperty.is_new then
1091 ret = arguments.first.mtype
1092 res = self.new_var(ret)
1093 else if ret == null then
1094 res = null
1095 else
1096 ret = ret.resolve_for(mmethoddef.mclassdef.bound_mtype, mmethoddef.mclassdef.bound_mtype, mmethoddef.mclassdef.mmodule, true)
1097 res = self.new_var(ret)
1098 end
1099
1100 if self.compiler.modelbuilder.mpropdef2npropdef.has_key(mmethoddef) and
1101 self.compiler.modelbuilder.mpropdef2npropdef[mmethoddef] isa AInternMethPropdef and
1102 not compiler.modelbuilder.toolcontext.opt_no_inline_intern.value then
1103 var frame = new Frame(self, mmethoddef, recvtype, arguments)
1104 frame.returnlabel = self.get_name("RET_LABEL")
1105 frame.returnvar = res
1106 var old_frame = self.frame
1107 self.frame = frame
1108 self.add("\{ /* Inline {mmethoddef} ({arguments.join(",")}) */")
1109 mmethoddef.compile_inside_to_c(self, arguments)
1110 self.add("{frame.returnlabel.as(not null)}:(void)0;")
1111 self.add("\}")
1112 self.frame = old_frame
1113 return res
1114 end
1115
1116 # Autobox arguments
1117 self.adapt_signature(mmethoddef, arguments)
1118
1119 self.require_declaration(mmethoddef.c_name)
1120 if res == null then
1121 self.add("{mmethoddef.c_name}({arguments.join(", ")});")
1122 return null
1123 else
1124 self.add("{res} = {mmethoddef.c_name}({arguments.join(", ")});")
1125 end
1126
1127 return res
1128 end
1129
1130 redef fun supercall(m: MMethodDef, recvtype: MClassType, arguments: Array[RuntimeVariable]): nullable RuntimeVariable
1131 do
1132 if arguments.first.mcasttype.ctype != "val*" then
1133 # In order to shortcut the primitive, we need to find the most specific method
1134 # However, because of performance (no flattening), we always work on the realmainmodule
1135 var main = self.compiler.mainmodule
1136 self.compiler.mainmodule = self.compiler.realmainmodule
1137 var res = self.monomorphic_super_send(m, recvtype, arguments)
1138 self.compiler.mainmodule = main
1139 return res
1140 end
1141 return table_send(m.mproperty, arguments, m.const_color)
1142 end
1143
1144 redef fun vararg_instance(mpropdef, recv, varargs, elttype)
1145 do
1146 # A vararg must be stored into an new array
1147 # The trick is that the dymaic type of the array may depends on the receiver
1148 # of the method (ie recv) if the static type is unresolved
1149 # This is more complex than usual because the unresolved type must not be resolved
1150 # with the current receiver (ie self).
1151 # Therefore to isolate the resolution from self, a local Frame is created.
1152 # One can see this implementation as an inlined method of the receiver whose only
1153 # job is to allocate the array
1154 var old_frame = self.frame
1155 var frame = new Frame(self, mpropdef, mpropdef.mclassdef.bound_mtype, [recv])
1156 self.frame = frame
1157 #print "required Array[{elttype}] for recv {recv.inspect}. bound=Array[{self.resolve_for(elttype, recv)}]. selfvar={frame.arguments.first.inspect}"
1158 var res = self.array_instance(varargs, elttype)
1159 self.frame = old_frame
1160 return res
1161 end
1162
1163 redef fun isset_attribute(a, recv)
1164 do
1165 self.check_recv_notnull(recv)
1166 var res = self.new_var(bool_type)
1167
1168 # What is the declared type of the attribute?
1169 var mtype = a.intro.static_mtype.as(not null)
1170 var intromclassdef = a.intro.mclassdef
1171 mtype = mtype.resolve_for(intromclassdef.bound_mtype, intromclassdef.bound_mtype, intromclassdef.mmodule, true)
1172
1173 if mtype isa MNullableType then
1174 self.add("{res} = 1; /* easy isset: {a} on {recv.inspect} */")
1175 return res
1176 end
1177
1178 self.require_declaration(a.const_color)
1179 if self.compiler.modelbuilder.toolcontext.opt_no_union_attribute.value then
1180 self.add("{res} = {recv}->attrs[{a.const_color}] != NULL; /* {a} on {recv.inspect}*/")
1181 else
1182
1183 if mtype.ctype == "val*" then
1184 self.add("{res} = {recv}->attrs[{a.const_color}].val != NULL; /* {a} on {recv.inspect} */")
1185 else
1186 self.add("{res} = 1; /* NOT YET IMPLEMENTED: isset of primitives: {a} on {recv.inspect} */")
1187 end
1188 end
1189 return res
1190 end
1191
1192 redef fun read_attribute(a, recv)
1193 do
1194 self.check_recv_notnull(recv)
1195
1196 # What is the declared type of the attribute?
1197 var ret = a.intro.static_mtype.as(not null)
1198 var intromclassdef = a.intro.mclassdef
1199 ret = ret.resolve_for(intromclassdef.bound_mtype, intromclassdef.bound_mtype, intromclassdef.mmodule, true)
1200
1201 self.require_declaration(a.const_color)
1202 if self.compiler.modelbuilder.toolcontext.opt_no_union_attribute.value then
1203 # Get the attribute or a box (ie. always a val*)
1204 var cret = self.object_type.as_nullable
1205 var res = self.new_var(cret)
1206 res.mcasttype = ret
1207
1208 self.add("{res} = {recv}->attrs[{a.const_color}]; /* {a} on {recv.inspect} */")
1209
1210 # Check for Uninitialized attribute
1211 if not ret isa MNullableType and not self.compiler.modelbuilder.toolcontext.opt_no_check_initialization.value then
1212 self.add("if ({res} == NULL) \{")
1213 self.add_abort("Uninitialized attribute {a.name}")
1214 self.add("\}")
1215 end
1216
1217 # Return the attribute or its unboxed version
1218 # Note: it is mandatory since we reuse the box on write, we do not whant that the box escapes
1219 return self.autobox(res, ret)
1220 else
1221 var res = self.new_var(ret)
1222 self.add("{res} = {recv}->attrs[{a.const_color}].{ret.ctypename}; /* {a} on {recv.inspect} */")
1223
1224 # Check for Uninitialized attribute
1225 if ret.ctype == "val*" and not ret isa MNullableType and not self.compiler.modelbuilder.toolcontext.opt_no_check_initialization.value then
1226 self.add("if ({res} == NULL) \{")
1227 self.add_abort("Uninitialized attribute {a.name}")
1228 self.add("\}")
1229 end
1230
1231 return res
1232 end
1233 end
1234
1235 redef fun write_attribute(a, recv, value)
1236 do
1237 self.check_recv_notnull(recv)
1238
1239 # What is the declared type of the attribute?
1240 var mtype = a.intro.static_mtype.as(not null)
1241 var intromclassdef = a.intro.mclassdef
1242 mtype = mtype.resolve_for(intromclassdef.bound_mtype, intromclassdef.bound_mtype, intromclassdef.mmodule, true)
1243
1244 # Adapt the value to the declared type
1245 value = self.autobox(value, mtype)
1246
1247 self.require_declaration(a.const_color)
1248 if self.compiler.modelbuilder.toolcontext.opt_no_union_attribute.value then
1249 var attr = "{recv}->attrs[{a.const_color}]"
1250 if mtype.ctype != "val*" then
1251 assert mtype isa MClassType
1252 # The attribute is primitive, thus we store it in a box
1253 # The trick is to create the box the first time then resuse the box
1254 self.add("if ({attr} != NULL) \{")
1255 self.add("((struct instance_{mtype.c_name}*){attr})->value = {value}; /* {a} on {recv.inspect} */")
1256 self.add("\} else \{")
1257 value = self.autobox(value, self.object_type.as_nullable)
1258 self.add("{attr} = {value}; /* {a} on {recv.inspect} */")
1259 self.add("\}")
1260 else
1261 # The attribute is not primitive, thus store it direclty
1262 self.add("{attr} = {value}; /* {a} on {recv.inspect} */")
1263 end
1264 else
1265 self.add("{recv}->attrs[{a.const_color}].{mtype.ctypename} = {value}; /* {a} on {recv.inspect} */")
1266 end
1267 end
1268
1269 redef fun init_instance(mtype)
1270 do
1271 self.require_declaration("NEW_{mtype.mclass.c_name}")
1272 var compiler = self.compiler
1273 if mtype isa MGenericType and mtype.need_anchor then
1274 link_unresolved_type(self.frame.mpropdef.mclassdef, mtype)
1275 var recv = self.frame.arguments.first
1276 var recv_type_info = self.type_info(recv)
1277 self.require_declaration(mtype.const_color)
1278 if compiler.modelbuilder.toolcontext.opt_phmod_typing.value or compiler.modelbuilder.toolcontext.opt_phand_typing.value then
1279 return self.new_expr("NEW_{mtype.mclass.c_name}({recv_type_info}->resolution_table->types[HASH({recv_type_info}->resolution_table->mask, {mtype.const_color})])", mtype)
1280 else
1281 return self.new_expr("NEW_{mtype.mclass.c_name}({recv_type_info}->resolution_table->types[{mtype.const_color}])", mtype)
1282 end
1283 end
1284 compiler.undead_types.add(mtype)
1285 self.require_declaration("type_{mtype.c_name}")
1286 return self.new_expr("NEW_{mtype.mclass.c_name}(&type_{mtype.c_name})", mtype)
1287 end
1288
1289 redef fun type_test(value, mtype, tag)
1290 do
1291 self.add("/* {value.inspect} isa {mtype} */")
1292 var compiler = self.compiler
1293
1294 var recv = self.frame.arguments.first
1295 var recv_type_info = self.type_info(recv)
1296
1297 var res = self.new_var(bool_type)
1298
1299 var cltype = self.get_name("cltype")
1300 self.add_decl("int {cltype};")
1301 var idtype = self.get_name("idtype")
1302 self.add_decl("int {idtype};")
1303
1304 var maybe_null = self.maybe_null(value)
1305 var accept_null = "0"
1306 var ntype = mtype
1307 if ntype isa MNullableType then
1308 ntype = ntype.mtype
1309 accept_null = "1"
1310 end
1311
1312 if value.mcasttype.is_subtype(self.frame.mpropdef.mclassdef.mmodule, self.frame.mpropdef.mclassdef.bound_mtype, mtype) then
1313 self.add("{res} = 1; /* easy {value.inspect} isa {mtype}*/")
1314 if compiler.modelbuilder.toolcontext.opt_typing_test_metrics.value then
1315 self.compiler.count_type_test_skipped[tag] += 1
1316 self.add("count_type_test_skipped_{tag}++;")
1317 end
1318 return res
1319 end
1320
1321 if ntype.need_anchor then
1322 var type_struct = self.get_name("type_struct")
1323 self.add_decl("const struct type* {type_struct};")
1324
1325 # Either with resolution_table with a direct resolution
1326 link_unresolved_type(self.frame.mpropdef.mclassdef, ntype)
1327 self.require_declaration(ntype.const_color)
1328 if compiler.modelbuilder.toolcontext.opt_phmod_typing.value or compiler.modelbuilder.toolcontext.opt_phand_typing.value then
1329 self.add("{type_struct} = {recv_type_info}->resolution_table->types[HASH({recv_type_info}->resolution_table->mask, {ntype.const_color})];")
1330 else
1331 self.add("{type_struct} = {recv_type_info}->resolution_table->types[{ntype.const_color}];")
1332 end
1333 if compiler.modelbuilder.toolcontext.opt_typing_test_metrics.value then
1334 self.compiler.count_type_test_unresolved[tag] += 1
1335 self.add("count_type_test_unresolved_{tag}++;")
1336 end
1337 self.add("{cltype} = {type_struct}->color;")
1338 self.add("{idtype} = {type_struct}->id;")
1339 if maybe_null and accept_null == "0" then
1340 var is_nullable = self.get_name("is_nullable")
1341 self.add_decl("short int {is_nullable};")
1342 self.add("{is_nullable} = {type_struct}->is_nullable;")
1343 accept_null = is_nullable.to_s
1344 end
1345 else if ntype isa MClassType then
1346 compiler.undead_types.add(mtype)
1347 self.require_declaration("type_{mtype.c_name}")
1348 self.add("{cltype} = type_{mtype.c_name}.color;")
1349 self.add("{idtype} = type_{mtype.c_name}.id;")
1350 if compiler.modelbuilder.toolcontext.opt_typing_test_metrics.value then
1351 self.compiler.count_type_test_resolved[tag] += 1
1352 self.add("count_type_test_resolved_{tag}++;")
1353 end
1354 else
1355 self.add("printf(\"NOT YET IMPLEMENTED: type_test(%s, {mtype}).\\n\", \"{value.inspect}\"); show_backtrace(1);")
1356 end
1357
1358 # check color is in table
1359 if maybe_null then
1360 self.add("if({value} == NULL) \{")
1361 self.add("{res} = {accept_null};")
1362 self.add("\} else \{")
1363 end
1364 var value_type_info = self.type_info(value)
1365 if compiler.modelbuilder.toolcontext.opt_phmod_typing.value or compiler.modelbuilder.toolcontext.opt_phand_typing.value then
1366 self.add("{cltype} = HASH({value_type_info}->color, {idtype});")
1367 end
1368 self.add("if({cltype} >= {value_type_info}->table_size) \{")
1369 self.add("{res} = 0;")
1370 self.add("\} else \{")
1371 self.add("{res} = {value_type_info}->type_table[{cltype}] == {idtype};")
1372 self.add("\}")
1373 if maybe_null then
1374 self.add("\}")
1375 end
1376
1377 return res
1378 end
1379
1380 redef fun is_same_type_test(value1, value2)
1381 do
1382 var res = self.new_var(bool_type)
1383 # Swap values to be symetric
1384 if value2.mtype.ctype != "val*" and value1.mtype.ctype == "val*" then
1385 var tmp = value1
1386 value1 = value2
1387 value2 = tmp
1388 end
1389 if value1.mtype.ctype != "val*" then
1390 if value2.mtype == value1.mtype then
1391 self.add("{res} = 1; /* is_same_type_test: compatible types {value1.mtype} vs. {value2.mtype} */")
1392 else if value2.mtype.ctype != "val*" then
1393 self.add("{res} = 0; /* is_same_type_test: incompatible types {value1.mtype} vs. {value2.mtype}*/")
1394 else
1395 var mtype1 = value1.mtype.as(MClassType)
1396 self.require_declaration("class_{mtype1.c_name}")
1397 self.add("{res} = ({value2} != NULL) && ({value2}->class == &class_{mtype1.c_name}); /* is_same_type_test */")
1398 end
1399 else
1400 self.add("{res} = ({value1} == {value2}) || ({value1} != NULL && {value2} != NULL && {value1}->class == {value2}->class); /* is_same_type_test */")
1401 end
1402 return res
1403 end
1404
1405 redef fun class_name_string(value)
1406 do
1407 var res = self.get_name("var_class_name")
1408 self.add_decl("const char* {res};")
1409 if value.mtype.ctype == "val*" then
1410 self.add "{res} = {value} == NULL ? \"null\" : {value}->type->name;"
1411 else if value.mtype isa MClassType and value.mtype.as(MClassType).mclass.kind == extern_kind then
1412 self.add "{res} = \"{value.mtype.as(MClassType).mclass}\";"
1413 else
1414 self.require_declaration("type_{value.mtype.c_name}")
1415 self.add "{res} = type_{value.mtype.c_name}.name;"
1416 end
1417 return res
1418 end
1419
1420 redef fun equal_test(value1, value2)
1421 do
1422 var res = self.new_var(bool_type)
1423 if value2.mtype.ctype != "val*" and value1.mtype.ctype == "val*" then
1424 var tmp = value1
1425 value1 = value2
1426 value2 = tmp
1427 end
1428 if value1.mtype.ctype != "val*" then
1429 if value2.mtype == value1.mtype then
1430 self.add("{res} = {value1} == {value2};")
1431 else if value2.mtype.ctype != "val*" then
1432 self.add("{res} = 0; /* incompatible types {value1.mtype} vs. {value2.mtype}*/")
1433 else
1434 var mtype1 = value1.mtype.as(MClassType)
1435 self.require_declaration("class_{mtype1.c_name}")
1436 self.add("{res} = ({value2} != NULL) && ({value2}->class == &class_{mtype1.c_name});")
1437 self.add("if ({res}) \{")
1438 self.add("{res} = ({self.autobox(value2, value1.mtype)} == {value1});")
1439 self.add("\}")
1440 end
1441 return res
1442 end
1443 var maybe_null = true
1444 var test = new Array[String]
1445 var t1 = value1.mcasttype
1446 if t1 isa MNullableType then
1447 test.add("{value1} != NULL")
1448 t1 = t1.mtype
1449 else
1450 maybe_null = false
1451 end
1452 var t2 = value2.mcasttype
1453 if t2 isa MNullableType then
1454 test.add("{value2} != NULL")
1455 t2 = t2.mtype
1456 else
1457 maybe_null = false
1458 end
1459
1460 var incompatible = false
1461 var primitive
1462 if t1.ctype != "val*" then
1463 primitive = t1
1464 if t1 == t2 then
1465 # No need to compare class
1466 else if t2.ctype != "val*" then
1467 incompatible = true
1468 else if can_be_primitive(value2) then
1469 test.add("{value1}->class == {value2}->class")
1470 else
1471 incompatible = true
1472 end
1473 else if t2.ctype != "val*" then
1474 primitive = t2
1475 if can_be_primitive(value1) then
1476 test.add("{value1}->class == {value2}->class")
1477 else
1478 incompatible = true
1479 end
1480 else
1481 primitive = null
1482 end
1483
1484 if incompatible then
1485 if maybe_null then
1486 self.add("{res} = {value1} == {value2}; /* incompatible types {t1} vs. {t2}; but may be NULL*/")
1487 return res
1488 else
1489 self.add("{res} = 0; /* incompatible types {t1} vs. {t2}; cannot be NULL */")
1490 return res
1491 end
1492 end
1493 if primitive != null then
1494 test.add("((struct instance_{primitive.c_name}*){value1})->value == ((struct instance_{primitive.c_name}*){value2})->value")
1495 else if can_be_primitive(value1) and can_be_primitive(value2) then
1496 test.add("{value1}->class == {value2}->class")
1497 var s = new Array[String]
1498 for t, v in self.compiler.box_kinds do
1499 s.add "({value1}->class->box_kind == {v} && ((struct instance_{t.c_name}*){value1})->value == ((struct instance_{t.c_name}*){value2})->value)"
1500 end
1501 test.add("({s.join(" || ")})")
1502 else
1503 self.add("{res} = {value1} == {value2};")
1504 return res
1505 end
1506 self.add("{res} = {value1} == {value2} || ({test.join(" && ")});")
1507 return res
1508 end
1509
1510 fun can_be_primitive(value: RuntimeVariable): Bool
1511 do
1512 var t = value.mcasttype
1513 if t isa MNullableType then t = t.mtype
1514 if not t isa MClassType then return false
1515 var k = t.mclass.kind
1516 return k == interface_kind or t.ctype != "val*"
1517 end
1518
1519 fun maybe_null(value: RuntimeVariable): Bool
1520 do
1521 var t = value.mcasttype
1522 return t isa MNullableType or t isa MNullType
1523 end
1524
1525 redef fun array_instance(array, elttype)
1526 do
1527 var nclass = self.get_class("NativeArray")
1528 var arrayclass = self.get_class("Array")
1529 var arraytype = arrayclass.get_mtype([elttype])
1530 var res = self.init_instance(arraytype)
1531 self.add("\{ /* {res} = array_instance Array[{elttype}] */")
1532 var length = self.int_instance(array.length)
1533 var nat = native_array_instance(elttype, length)
1534 for i in [0..array.length[ do
1535 var r = self.autobox(array[i], self.object_type)
1536 self.add("((struct instance_{nclass.c_name}*){nat})->values[{i}] = (val*) {r};")
1537 end
1538 self.send(self.get_property("with_native", arrayclass.intro.bound_mtype), [res, nat, length])
1539 self.add("\}")
1540 return res
1541 end
1542
1543 fun native_array_instance(elttype: MType, length: RuntimeVariable): RuntimeVariable
1544 do
1545 var mtype = self.get_class("NativeArray").get_mtype([elttype])
1546 self.require_declaration("NEW_{mtype.mclass.c_name}")
1547 assert mtype isa MGenericType
1548 var compiler = self.compiler
1549 if mtype.need_anchor then
1550 link_unresolved_type(self.frame.mpropdef.mclassdef, mtype)
1551 var recv = self.frame.arguments.first
1552 var recv_type_info = self.type_info(recv)
1553 self.require_declaration(mtype.const_color)
1554 if compiler.modelbuilder.toolcontext.opt_phmod_typing.value or compiler.modelbuilder.toolcontext.opt_phand_typing.value then
1555 return self.new_expr("NEW_{mtype.mclass.c_name}({length}, {recv_type_info}->resolution_table->types[HASH({recv_type_info}->resolution_table->mask, {mtype.const_color})])", mtype)
1556 else
1557 return self.new_expr("NEW_{mtype.mclass.c_name}({length}, {recv_type_info}->resolution_table->types[{mtype.const_color}])", mtype)
1558 end
1559 end
1560 compiler.undead_types.add(mtype)
1561 self.require_declaration("type_{mtype.c_name}")
1562 return self.new_expr("NEW_{mtype.mclass.c_name}({length}, &type_{mtype.c_name})", mtype)
1563 end
1564
1565 redef fun native_array_def(pname, ret_type, arguments)
1566 do
1567 var elttype = arguments.first.mtype
1568 var nclass = self.get_class("NativeArray")
1569 var recv = "((struct instance_{nclass.c_name}*){arguments[0]})->values"
1570 if pname == "[]" then
1571 self.ret(self.new_expr("{recv}[{arguments[1]}]", ret_type.as(not null)))
1572 return
1573 else if pname == "[]=" then
1574 self.add("{recv}[{arguments[1]}]={arguments[2]};")
1575 return
1576 else if pname == "copy_to" then
1577 var recv1 = "((struct instance_{nclass.c_name}*){arguments[1]})->values"
1578 self.add("memcpy({recv1}, {recv}, {arguments[2]}*sizeof({elttype.ctype}));")
1579 return
1580 end
1581 end
1582
1583 redef fun calloc_array(ret_type, arguments)
1584 do
1585 var mclass = self.get_class("ArrayCapable")
1586 var ft = mclass.mclass_type.arguments.first.as(MParameterType)
1587 var res = self.native_array_instance(ft, arguments[1])
1588 self.ret(res)
1589 end
1590
1591 fun link_unresolved_type(mclassdef: MClassDef, mtype: MType) do
1592 assert mtype.need_anchor
1593 var compiler = self.compiler
1594 if not compiler.live_unresolved_types.has_key(self.frame.mpropdef.mclassdef) then
1595 compiler.live_unresolved_types[self.frame.mpropdef.mclassdef] = new HashSet[MType]
1596 end
1597 compiler.live_unresolved_types[self.frame.mpropdef.mclassdef].add(mtype)
1598 end
1599 end
1600
1601 redef class MMethodDef
1602 fun separate_runtime_function: AbstractRuntimeFunction
1603 do
1604 var res = self.separate_runtime_function_cache
1605 if res == null then
1606 res = new SeparateRuntimeFunction(self)
1607 self.separate_runtime_function_cache = res
1608 end
1609 return res
1610 end
1611 private var separate_runtime_function_cache: nullable SeparateRuntimeFunction
1612
1613 fun virtual_runtime_function: AbstractRuntimeFunction
1614 do
1615 var res = self.virtual_runtime_function_cache
1616 if res == null then
1617 res = new VirtualRuntimeFunction(self)
1618 self.virtual_runtime_function_cache = res
1619 end
1620 return res
1621 end
1622 private var virtual_runtime_function_cache: nullable VirtualRuntimeFunction
1623 end
1624
1625 # The C function associated to a methoddef separately compiled
1626 class SeparateRuntimeFunction
1627 super AbstractRuntimeFunction
1628
1629 redef fun build_c_name: String do return "{mmethoddef.c_name}"
1630
1631 redef fun to_s do return self.mmethoddef.to_s
1632
1633 redef fun compile_to_c(compiler)
1634 do
1635 var mmethoddef = self.mmethoddef
1636
1637 var recv = self.mmethoddef.mclassdef.bound_mtype
1638 var v = compiler.new_visitor
1639 var selfvar = new RuntimeVariable("self", recv, recv)
1640 var arguments = new Array[RuntimeVariable]
1641 var frame = new Frame(v, mmethoddef, recv, arguments)
1642 v.frame = frame
1643
1644 var msignature = mmethoddef.msignature.resolve_for(mmethoddef.mclassdef.bound_mtype, mmethoddef.mclassdef.bound_mtype, mmethoddef.mclassdef.mmodule, true)
1645
1646 var sig = new Buffer
1647 var comment = new Buffer
1648 var ret = msignature.return_mtype
1649 if ret != null then
1650 sig.append("{ret.ctype} ")
1651 else if mmethoddef.mproperty.is_new then
1652 ret = recv
1653 sig.append("{ret.ctype} ")
1654 else
1655 sig.append("void ")
1656 end
1657 sig.append(self.c_name)
1658 sig.append("({selfvar.mtype.ctype} {selfvar}")
1659 comment.append("({selfvar}: {selfvar.mtype}")
1660 arguments.add(selfvar)
1661 for i in [0..msignature.arity[ do
1662 var mtype = msignature.mparameters[i].mtype
1663 if i == msignature.vararg_rank then
1664 mtype = v.get_class("Array").get_mtype([mtype])
1665 end
1666 comment.append(", {mtype}")
1667 sig.append(", {mtype.ctype} p{i}")
1668 var argvar = new RuntimeVariable("p{i}", mtype, mtype)
1669 arguments.add(argvar)
1670 end
1671 sig.append(")")
1672 comment.append(")")
1673 if ret != null then
1674 comment.append(": {ret}")
1675 end
1676 compiler.provide_declaration(self.c_name, "{sig};")
1677
1678 v.add_decl("/* method {self} for {comment} */")
1679 v.add_decl("{sig} \{")
1680 if ret != null then
1681 frame.returnvar = v.new_var(ret)
1682 end
1683 frame.returnlabel = v.get_name("RET_LABEL")
1684
1685 if recv != arguments.first.mtype then
1686 #print "{self} {recv} {arguments.first}"
1687 end
1688 mmethoddef.compile_inside_to_c(v, arguments)
1689
1690 v.add("{frame.returnlabel.as(not null)}:;")
1691 if ret != null then
1692 v.add("return {frame.returnvar.as(not null)};")
1693 end
1694 v.add("\}")
1695 if not self.c_name.has_substring("VIRTUAL", 0) then compiler.names[self.c_name] = "{mmethoddef.mclassdef.mmodule.name}::{mmethoddef.mclassdef.mclass.name}::{mmethoddef.mproperty.name} ({mmethoddef.location.file.filename}:{mmethoddef.location.line_start})"
1696 end
1697 end
1698
1699 # The C function associated to a methoddef on a primitive type, stored into a VFT of a class
1700 # The first parameter (the reciever) is always typed by val* in order to accept an object value
1701 class VirtualRuntimeFunction
1702 super AbstractRuntimeFunction
1703
1704 redef fun build_c_name: String do return "VIRTUAL_{mmethoddef.c_name}"
1705
1706 redef fun to_s do return self.mmethoddef.to_s
1707
1708 redef fun compile_to_c(compiler)
1709 do
1710 var mmethoddef = self.mmethoddef
1711
1712 var recv = self.mmethoddef.mclassdef.bound_mtype
1713 var v = compiler.new_visitor
1714 var selfvar = new RuntimeVariable("self", v.object_type, recv)
1715 var arguments = new Array[RuntimeVariable]
1716 var frame = new Frame(v, mmethoddef, recv, arguments)
1717 v.frame = frame
1718
1719 var sig = new Buffer
1720 var comment = new Buffer
1721
1722 # Because the function is virtual, the signature must match the one of the original class
1723 var intromclassdef = self.mmethoddef.mproperty.intro.mclassdef
1724 var msignature = mmethoddef.mproperty.intro.msignature.resolve_for(intromclassdef.bound_mtype, intromclassdef.bound_mtype, intromclassdef.mmodule, true)
1725 var ret = msignature.return_mtype
1726 if ret != null then
1727 sig.append("{ret.ctype} ")
1728 else if mmethoddef.mproperty.is_new then
1729 ret = recv
1730 sig.append("{ret.ctype} ")
1731 else
1732 sig.append("void ")
1733 end
1734 sig.append(self.c_name)
1735 sig.append("({selfvar.mtype.ctype} {selfvar}")
1736 comment.append("({selfvar}: {selfvar.mtype}")
1737 arguments.add(selfvar)
1738 for i in [0..msignature.arity[ do
1739 var mtype = msignature.mparameters[i].mtype
1740 if i == msignature.vararg_rank then
1741 mtype = v.get_class("Array").get_mtype([mtype])
1742 end
1743 comment.append(", {mtype}")
1744 sig.append(", {mtype.ctype} p{i}")
1745 var argvar = new RuntimeVariable("p{i}", mtype, mtype)
1746 arguments.add(argvar)
1747 end
1748 sig.append(")")
1749 comment.append(")")
1750 if ret != null then
1751 comment.append(": {ret}")
1752 end
1753 compiler.provide_declaration(self.c_name, "{sig};")
1754
1755 v.add_decl("/* method {self} for {comment} */")
1756 v.add_decl("{sig} \{")
1757 if ret != null then
1758 frame.returnvar = v.new_var(ret)
1759 end
1760 frame.returnlabel = v.get_name("RET_LABEL")
1761
1762 var subret = v.call(mmethoddef, recv, arguments)
1763 if ret != null then
1764 assert subret != null
1765 v.assign(frame.returnvar.as(not null), subret)
1766 end
1767
1768 v.add("{frame.returnlabel.as(not null)}:;")
1769 if ret != null then
1770 v.add("return {frame.returnvar.as(not null)};")
1771 end
1772 v.add("\}")
1773 if not self.c_name.has_substring("VIRTUAL", 0) then compiler.names[self.c_name] = "{mmethoddef.mclassdef.mmodule.name}::{mmethoddef.mclassdef.mclass.name}::{mmethoddef.mproperty.name} ({mmethoddef.location.file.filename}--{mmethoddef.location.line_start})"
1774 end
1775
1776 # TODO ?
1777 redef fun call(v, arguments) do abort
1778 end
1779
1780 redef class MType
1781 fun const_color: String do return "COLOR_{c_name}"
1782 end
1783
1784 redef class MProperty
1785 fun const_color: String do return "COLOR_{c_name}"
1786 end
1787
1788 redef class MPropDef
1789 fun const_color: String do return "COLOR_{c_name}"
1790 end