icode: remove method info from ICodeBuilder
[nit.git] / src / compiling / compiling_global.nit
index d2217f8..745a523 100644 (file)
@@ -17,9 +17,7 @@
 # Compute and generate tables for classes and modules.
 package compiling_global
 
-#import compiling_base
-private import compiling_methods
-private import syntax
+private import compiling_icode
 
 # Something that store color of table elements
 class ColorContext
@@ -61,7 +59,7 @@ special ColorContext
        # FIXME: do something better.
        readable writable var _max_class_table_length: Int = 0
 
-       init(module: MMSrcModule)
+       init(module: MMModule)
        do
                _module = module
        end
@@ -71,7 +69,7 @@ class GlobalCompilerVisitor
 special CompilerVisitor
        # The global analysis result
        readable var _global_analysis: GlobalAnalysis
-       init(m: MMSrcModule, tc: ToolContext, ga: GlobalAnalysis)
+       init(m: MMModule, tc: ToolContext, ga: GlobalAnalysis)
        do
                super(m, tc)
                _global_analysis = ga
@@ -102,7 +100,7 @@ special ColorContext
        init(c: MMLocalClass) do _local_class = c
 end
 
-redef class MMSrcLocalClass
+redef class MMConcreteClass
        # The table element of the subtype check
        fun class_color_pos: TableEltClassColor do return _class_color_pos.as(not null)
        var _class_color_pos: nullable TableEltClassColor
@@ -126,16 +124,16 @@ redef class MMSrcLocalClass
                        module_table.add(cpp)
                        clt.add(new TableEltClassInitTable(self))
                end
-               for p in src_local_properties do
+               for p in local_local_properties do
                        var pg = p.global
                        if pg.intro == p then
-                               if p isa MMSrcAttribute then
+                               if p isa MMAttribute then
                                        ilt.add(new TableEltAttr(p))
-                               else if p isa MMSrcMethod then
+                               else if p isa MMMethod then
                                        clt.add(new TableEltMeth(p))
                                end
                        end
-                       if p isa MMSrcMethod and p.need_super then
+                       if p isa MMMethod and p.need_super then
                                clt.add(new TableEltSuper(p))
                        end
                end
@@ -154,15 +152,17 @@ redef class MMSrcLocalClass
        end
 end
 
-redef class MMSrcModule
+redef class MMModule
        # The local table of the module (refers things introduced in the module)
        var _local_table: Array[ModuleTableElt] = new Array[ModuleTableElt]
 
        # Builds the local tables and local classes layouts
        fun local_analysis(tc: ToolContext)
        do
-               for c in src_local_classes do
-                       c.build_layout_in(tc, _local_table)
+               for c in local_classes do
+                       if c isa MMConcreteClass then
+                               c.build_layout_in(tc, _local_table)
+                       end
                end
        end
 
@@ -218,7 +218,7 @@ redef class MMSrcModule
 
                        # Store the colortableelt in the class table pool
                        var bc = c.global.intro
-                       assert bc isa MMSrcLocalClass
+                       assert bc isa MMConcreteClass
                        ctab.add(bc.class_color_pos)
                end
 
@@ -252,7 +252,7 @@ redef class MMSrcModule
                                assert cc.class_table.is_empty
                                cc.class_table.add_all(scc.class_table)
                                var bc = c.global.intro
-                               assert bc isa MMSrcLocalClass
+                               assert bc isa MMConcreteClass
                                var colpos = bc.class_color_pos
                                var colposcolor = cc.class_table.length
                                ga.color(colpos) = colposcolor
@@ -268,7 +268,7 @@ redef class MMSrcModule
                        var cte = cc.class_layout
                        var ite = cc.instance_layout
                        for sc in c.crhe.greaters_and_self do
-                               if sc isa MMSrcLocalClass then
+                               if sc isa MMConcreteClass then
                                        cte.add(sc, sc.class_layout)
                                        ite.add(sc, sc.instance_layout)
                                end
@@ -303,7 +303,7 @@ redef class MMSrcModule
                                cc.class_table.clear
                                cc.class_table.add_all(scc.class_table)
                                var bc = c.global.intro
-                               assert bc isa MMSrcLocalClass
+                               assert bc isa MMConcreteClass
                                var colpos = bc.class_color_pos
                                cc.class_table[ga.color(colpos)] = colpos
                                while cc.class_table.length <= maxcolor do
@@ -415,7 +415,6 @@ redef class MMSrcModule
        fun compile_tables_to_c(v: GlobalCompilerVisitor)
        do
                for m in mhe.greaters_and_self do
-                       assert m isa MMSrcModule
                        m.compile_local_table_to_c(v)
                end
 
@@ -488,17 +487,18 @@ redef class MMSrcModule
                        end
                        e.compile_macros(v, value)
                end
-               for c in src_local_classes do
+               for c in local_classes do
+                       if not c isa MMConcreteClass then continue
                        for pg in c.global_properties do
                                var p = c[pg]
-                               if p.local_class == c then
+                               if p.local_class == c and p isa MMMethod then
                                        p.compile_property_to_c(v)
                                end
                                if pg.is_init_for(c) then
                                        # Declare constructors
                                        var params = new Array[String]
-                                       for i in [0..p.signature.arity[ do
-                                               params.add("val_t p{i}")
+                                       for j in [0..p.signature.arity[ do
+                                               params.add("val_t p{j}")
                                        end
                                        v.add_decl("val_t NEW_{c}_{p.global.intro.cname}({params.join(", ")});")
                                end
@@ -509,7 +509,7 @@ redef class MMSrcModule
        # Compile module file for the current module
        fun compile_local_table_to_c(v: GlobalCompilerVisitor)
        do
-               v.add_instr("const char *LOCATE_{name} = \"{filename}\";")
+               v.add_instr("const char *LOCATE_{name} = \"{location.file}\";")
 
                if v.tc.global or _local_table.is_empty then
                        return
@@ -865,61 +865,65 @@ redef class MMLocalClass
 
                var pi = primitive_info
                if pi == null then
-                       v.cfc = new CFunctionContext(v)
-                       v.nmc = new NitMethodContext(null)
-                       var s = "val_t NEW_{name}(void)"
-                       v.add_instr(s + " \{")
-                       v.indent
-                       var ctx_old = v.ctx
-                       v.ctx = new CContext
-
-                       var self_var = new ParamVariable(once ("self".to_symbol), null)
-                       var self_var_cname = v.cfc.register_variable(self_var)
-                       v.nmc.method_params = [self_var]
-
-                       v.add_instr("obj_t obj;")
-                       v.add_instr("obj = alloc(sizeof(val_t) * {itab.length});")
-                       v.add_instr("obj->vft = (classtable_elt_t*)VFT_{name};")
-                       v.add_assignment(self_var_cname, "OBJ2VAL(obj)")
-
-                       for g in global_properties do
-                               var p = self[g]
-                               var t = p.signature.return_type
-                               if p isa MMAttribute and t != null then
-                                       # FIXME: Not compatible with sep compilation
-                                       assert p isa MMSrcAttribute
-                                       var np = p.node
-                                       var ne = np.n_expr
-                                       if ne != null then
-                                               var e = ne.compile_expr(v)
-                                               v.add_instr("{p.global.attr_access}(obj) = {e};")
+                       do
+                               var iself = new IRegister(get_type)
+                               var iselfa = [iself]
+                               var iroutine = new IRoutine(new Array[IRegister], iself)
+                               var icb = new ICodeBuilder(module, iroutine)
+                               var obj = new INative("OBJ2VAL(obj)", null)
+                               obj.result = iself
+                               icb.stmt(obj)
+
+                               for g in global_properties do
+                                       var p = self[g]
+                                       var t = p.signature.return_type
+                                       if p isa MMAttribute and t != null then
+                                               var ir = p.iroutine
+                                               if ir == null then continue
+                                               # FIXME: Not compatible with sep compilation
+                                               var e = ir.inline_in_seq(icb.seq, iselfa).as(not null)
+                                               icb.stmt(new IAttrWrite(p, iself, e))
                                        end
                                end
+
+                               var cname = "NEW_{name}"
+                               var args = iroutine.compile_signature_to_c(v, cname, "new {name}", null, null)
+                               var ctx_old = v.ctx
+                               v.ctx = new CContext
+                               v.add_decl("obj_t obj;")
+                               v.add_instr("obj = alloc(sizeof(val_t) * {itab.length});")
+                               v.add_instr("obj->vft = (classtable_elt_t*)VFT_{name};")
+                               var r = iroutine.compile_to_c(v, cname, args).as(not null)
+                               v.add_instr("return {r};")
+                               ctx_old.append(v.ctx)
+                               v.ctx = ctx_old
+                               v.unindent
+                               v.add_instr("}")
                        end
-                       v.add_instr("return OBJ2VAL(obj);")
-                       v.cfc.generate_var_decls
-                       ctx_old.append(v.ctx)
-                       v.ctx = ctx_old
-                       v.unindent
-                       v.add_instr("}")
 
-                       # Compile CHECKNAME
-                       var s = "void CHECKNEW_{name}(val_t self, char *from)"
-                       v.add_instr(s + " \{")
-                       v.indent
-                       var ctx_old = v.ctx
-                       v.ctx = new CContext
-                       for g in global_properties do
-                               var p = self[g]
-                               var t = p.signature.return_type
-                               if p isa MMAttribute and t != null and not t.is_nullable then
-                                       v.add_instr("if ({p.global.attr_access}(self) == NIT_NULL) \{ fprintf(stderr, \"Uninitialized attribute %s at %s.\\n\", \"{p.full_name}\", from); nit_exit(1); \}")
+                       do
+                               # Compile CHECKNAME
+                               var iself = new IRegister(get_type)
+                               var iselfa = [iself]
+                               var iroutine = new IRoutine(iselfa, null)
+                               var icb = new ICodeBuilder(module, iroutine)
+                               for g in global_properties do
+                                       var p = self[g]
+                                       var t = p.signature.return_type
+                                       if p isa MMAttribute and t != null and not t.is_nullable then
+                                               icb.add_attr_check(p, iself)
+                                       end
                                end
+                               var cname = "CHECKNEW_{name}"
+                               var args = iroutine.compile_signature_to_c(v, cname, "check new {name}", null, null)
+                               var ctx_old = v.ctx
+                               v.ctx = new CContext
+                               iroutine.compile_to_c(v, cname, args)
+                               ctx_old.append(v.ctx)
+                               v.ctx = ctx_old
+                               v.unindent
+                               v.add_instr("}")
                        end
-                       ctx_old.append(v.ctx)
-                       v.ctx = ctx_old
-                       v.unindent
-                       v.add_instr("}")
 
                        var init_table_size = cshe.greaters.length + 1
                        var init_table_decl = "int init_table[{init_table_size}] = \{0{", 0" * (init_table_size-1)}};"
@@ -928,21 +932,32 @@ redef class MMLocalClass
                                var p = self[g]
                                # FIXME skip invisible constructors
                                if not p.global.is_init_for(self) then continue
-                               var params = new Array[String]
-                               var args = ["self"]
-                               for i in [0..p.signature.arity[ do
-                                       params.add("val_t p{i}")
-                                       args.add("p{i}")
-                               end
-                               args.add("init_table")
-                               var s = "val_t NEW_{self}_{p.global.intro.cname}({params.join(", ")}) \{"
-                               v.add_instr(s)
-                               v.indent
+                               assert p isa MMMethod
+
+                               var iself = new IRegister(get_type)
+                               var iparams = new Array[IRegister]
+                               for i in [0..p.signature.arity[ do iparams.add(new IRegister(p.signature[i]))
+                               var iroutine = new IRoutine(iparams, iself)
+                               iroutine.location = p.iroutine.location
+                               var icb = new ICodeBuilder(module, iroutine)
+
+                               var inew = new INative("NEW_{name}()", null)
+                               inew.result = iself
+                               icb.stmt(inew)
+                               var iargs = [iself]
+                               iargs.add_all(iparams)
+                               icb.stmt(new INative("{p.cname}(@@@{", @@@"*iparams.length}, init_table)", iargs))
+                               icb.stmt(new INative("CHECKNEW_{name}(@@@)", [iself]))
+
+                               var cname = "NEW_{self}_{p.global.intro.cname}"
+                               var new_args = iroutine.compile_signature_to_c(v, cname, "new {self} {p.full_name}", null, null)
+                               var ctx_old = v.ctx
+                               v.ctx = new CContext
                                v.add_instr(init_table_decl)
-                               v.add_instr("val_t self = NEW_{name}();")
-                               v.add_instr("{p.cname}({args.join(", ")});")
-                               v.add_instr("CHECKNEW_{name}(self, \"{p.full_name} for {self}\");")
-                               v.add_instr("return self;")
+                               var e = iroutine.compile_to_c(v, cname, new_args).as(not null)
+                               v.add_instr("return {e};")
+                               ctx_old.append(v.ctx)
+                               v.ctx = ctx_old
                                v.unindent
                                v.add_instr("}")
                        end
@@ -961,3 +976,44 @@ redef class MMLocalClass
        end
 end
 
+redef class MMMethod
+       fun compile_property_to_c(v: CompilerVisitor)
+       do
+               var ir = iroutine
+               assert ir != null
+
+               var more_params: nullable String = null
+               if global.is_init then more_params = "int* init_table"
+               var args = ir.compile_signature_to_c(v, cname, full_name, null, more_params)
+               var ctx_old = v.ctx
+               v.ctx = new CContext
+
+               v.out_contexts.clear
+
+               var itpos: nullable String = null
+               if global.is_init then
+                       itpos = "itpos{v.new_number}"
+                       v.add_decl("int {itpos} = VAL2OBJ({args.first})->vft[{local_class.global.init_table_pos_id}].i;")
+                       v.add_instr("if (init_table[{itpos}]) return;")
+               end
+
+               var s = ir.compile_to_c(v, cname, args)
+
+               if itpos != null then
+                       v.add_instr("init_table[{itpos}] = 1;")
+               end
+               if s == null then
+                       v.add_instr("return;")
+               else
+                       v.add_instr("return ", s, ";")
+               end
+
+               ctx_old.append(v.ctx)
+               v.ctx = ctx_old
+               v.unindent
+               v.add_instr("}")
+
+               for ctx in v.out_contexts do v.ctx.merge(ctx)
+       end
+end
+