nitg-s: rename SeparateCompiler:do_global_type_coloring to do_type_coloring
[nit.git] / src / naive_interpreter.nit
index 4a07467..41fed1e 100644 (file)
@@ -21,6 +21,17 @@ import literal
 import typing
 import auto_super_init
 
+redef class ToolContext
+       # --discover-call-trace
+       var opt_discover_call_trace: OptionBool = new OptionBool("Trace calls of the first invocation of a method", "--discover-call-trace")
+
+       redef init
+       do
+               super
+               self.option_context.add_option(self.opt_discover_call_trace)
+       end
+end
+
 redef class ModelBuilder
        # Execute the program from the entry point (Sys::main) of the `mainmodule'
        # `arguments' are the command-line arguments in order
@@ -266,11 +277,28 @@ private class NaiveInterpreter
                exit(1)
        end
 
+       # Debug on the current node
+       fun debug(message: String)
+       do
+               if frames.is_empty then
+                       print message
+               else
+                       self.frame.current_node.debug(message)
+               end
+       end
+
+       # Store known method, used to trace methods as thez are reached
+       var discover_call_trace: Set[MMethodDef] = new HashSet[MMethodDef]
+
        # Execute `mpropdef' for a `args' (where args[0] is the receiver).
        # Return a falue if `mpropdef' is a function, or null if it is a procedure.
        # The call is direct/static. There is no message-seding/late-bindng.
        fun call(mpropdef: MMethodDef, args: Array[Instance]): nullable Instance
        do
+               if self.modelbuilder.toolcontext.opt_discover_call_trace.value and not self.discover_call_trace.has(mpropdef) then
+                       self.discover_call_trace.add mpropdef
+                       self.debug("Discovered {mpropdef}")
+               end
                var vararg_rank = mpropdef.msignature.vararg_rank
                if vararg_rank >= 0 then
                        assert args.length >= mpropdef.msignature.arity + 1 # because of self
@@ -452,10 +480,14 @@ abstract class Instance
        # Human readable object identity "Type#number"
        redef fun to_s do return "{mtype}#{object_id}"
 
-       # Return the integer valur is the instance is an integer.
+       # Return the integer value if the instance is an integer.
        # else aborts
        fun to_i: Int do abort
 
+       # Return the integer value if the instance is a float.
+       # else aborts
+       fun to_f: Float do abort
+
        # The real value encapsulated if the instance is primitive.
        # Else aborts.
        fun val: Object do abort
@@ -505,6 +537,8 @@ class PrimitiveInstance[E: Object]
        redef fun to_s do return "{mtype}#{val.object_id}({val})"
 
        redef fun to_i do return val.as(Int)
+
+       redef fun to_f do return val.as(Float)
 end
 
 private class ClosureInstance
@@ -542,10 +576,11 @@ redef class ANode
        private fun fatal(v: NaiveInterpreter, message: String)
        do
                if v.modelbuilder.toolcontext.opt_no_color.value == true then
-                       print("Runtime error: {message} ({location.file.filename}:{location.line_start})")
+                       stderr.write("Runtime error: {message} ({location.file.filename}:{location.line_start})\n")
                else
-                       print("{location}: Runtime error: {message}\n{location.colored_line("0;31")}")
-                       print(v.stack_trace)
+                       stderr.write("{location}: Runtime error: {message}\n{location.colored_line("0;31")}\n")
+                       stderr.write(v.stack_trace)
+                       stderr.write("\n")
                end
                exit(1)
        end
@@ -695,16 +730,18 @@ redef class AInternMethPropdef
                                return v.int_instance(recv <=> args[1].val.as(Char))
                        end
                else if cname == "Float" then
-                       if pname == "+" then
-                               return v.float_instance(args[0].val.as(Float) + args[1].val.as(Float))
+                       if pname == "unary -" then
+                               return v.float_instance(-args[0].to_f)
+                       else if pname == "+" then
+                               return v.float_instance(args[0].to_f + args[1].to_f)
                        else if pname == "-" then
-                               return v.float_instance(args[0].val.as(Float) - args[1].val.as(Float))
+                               return v.float_instance(args[0].to_f - args[1].to_f)
                        else if pname == "*" then
-                               return v.float_instance(args[0].val.as(Float) * args[1].val.as(Float))
+                               return v.float_instance(args[0].to_f * args[1].to_f)
                        else if pname == "/" then
-                               return v.float_instance(args[0].val.as(Float) / args[1].val.as(Float))
+                               return v.float_instance(args[0].to_f / args[1].to_f)
                        else if pname == "to_i" then
-                               return v.int_instance(args[0].val.as(Float).to_i)
+                               return v.int_instance(args[0].to_f.to_i)
                        end
                else if cname == "NativeString" then
                        var recvval = args.first.val.as(Buffer)
@@ -762,8 +799,9 @@ redef class AInternMethPropdef
                        end
                else if pname == "calloc_array" then
                        var recvtype = args.first.mtype.as(MClassType)
-                       var mtype: MType = recvtype.supertype_to(v.mainmodule, recvtype, v.mainmodule.get_primitive_class("ArrayCapable"))
-                       mtype = mtype.as(MGenericType).arguments.first
+                       var mtype: MType
+                       mtype = recvtype.supertype_to(v.mainmodule, recvtype, v.mainmodule.get_primitive_class("ArrayCapable"))
+                       mtype = mtype.arguments.first
                        var val = new Array[Instance].filled_with(v.null_instance, args[1].to_i)
                        return new PrimitiveInstance[Array[Instance]](v.mainmodule.get_primitive_class("NativeArray").get_mtype([mtype]), val)
                end
@@ -843,6 +881,16 @@ redef class AExternMethPropdef
                                var res = sys.system(recvval.to_s)
                                return v.int_instance(res)
                        end
+               else if cname == "Int" then
+                       if pname == "rand" then
+                               return v.int_instance(args[0].to_i.rand)
+                       end
+               else if cname == "Float" then
+                       if pname == "cos" then
+                               return v.float_instance(args[0].to_f.cos)
+                       else if pname == "sin" then
+                               return v.float_instance(args[0].to_f.sin)
+                       end
                else if pname == "native_argc" then
                        return v.int_instance(v.arguments.length)
                else if pname == "native_argv" then
@@ -850,6 +898,10 @@ redef class AExternMethPropdef
                        return v.native_string_instance(txt)
                else if pname == "get_time" then
                        return v.int_instance(get_time)
+               else if pname == "atan2" then
+                       return v.float_instance(atan2(args[1].to_f, args[2].to_f))
+               else if pname == "pi" then
+                       return v.float_instance(pi)
                else if pname == "lexer_goto" then
                        return v.int_instance(lexer_goto(args[1].to_i, args[2].to_i))
                else if pname == "lexer_accept" then
@@ -1254,7 +1306,7 @@ redef class AArrayExpr
                        if i == null then return null
                        val.add(i)
                end
-               var mtype = v.unanchor_type(self.mtype.as(not null)).as(MGenericType)
+               var mtype = v.unanchor_type(self.mtype.as(not null)).as(MClassType)
                var elttype = mtype.arguments.first
                return v.array_instance(val, elttype)
        end