include/pocketpy/objects/codeobject.h

@@ -12,7 +12,6 @@
 
 #define BC_NOARG 0
 #define BC_KEEPLINE -1
-#define BC_RETURN_VIRTUAL 5
 
 typedef enum FuncType {
     FuncType_UNSET,
@@ -63,6 +62,7 @@ typedef struct CodeBlock {
 
 typedef struct BytecodeEx {
     int lineno;       // line number for each bytecode
+    bool is_virtual;  // whether this bytecode is virtual (not in source code)
     int iblock;       // block index
 } BytecodeEx;
 

scripts/98_profiler.py

@@ -9,13 +9,3 @@ y = 2
 costly_func(2)      # 3s
 
 time.sleep(1)       # 1s
-
-def new_func(a, b, c):
-    x = a + b
-    x = x + c
-    return a
-
-def new_func2(a, b, c):
-    x = a + b
-    x = x + c
-    return a

src/compiler/compiler.c

@@ -76,6 +76,7 @@ static int Ctx__prepare_loop_divert(Ctx* self, int line, bool is_break);
 static int Ctx__enter_block(Ctx* self, CodeBlockType type);
 static void Ctx__exit_block(Ctx* self);
 static int Ctx__emit_(Ctx* self, Opcode opcode, uint16_t arg, int line);
+static int Ctx__emit_virtual(Ctx* self, Opcode opcode, uint16_t arg, int line, bool virtual);
 // static void Ctx__revert_last_emit_(Ctx* self);
 static int Ctx__emit_int(Ctx* self, int64_t value, int line);
 static void Ctx__patch_jump(Ctx* self, int index);
@@ -1176,9 +1177,9 @@ static void Ctx__s_emit_decorators(Ctx* self, int count) {
     }
 }
 
-static int Ctx__emit_(Ctx* self, Opcode opcode, uint16_t arg, int line) {
+static int Ctx__emit_virtual(Ctx* self, Opcode opcode, uint16_t arg, int line, bool is_virtual) {
     Bytecode bc = {(uint8_t)opcode, arg};
-    BytecodeEx bcx = {line, self->curr_iblock};
+    BytecodeEx bcx = {line, is_virtual, self->curr_iblock};
     c11_vector__push(Bytecode, &self->co->codes, bc);
     c11_vector__push(BytecodeEx, &self->co->codes_ex, bcx);
     int i = self->co->codes.length - 1;
@@ -1187,6 +1188,10 @@ static int Ctx__emit_(Ctx* self, Opcode opcode, uint16_t arg, int line) {
     return i;
 }
 
+static int Ctx__emit_(Ctx* self, Opcode opcode, uint16_t arg, int line) {
+    return Ctx__emit_virtual(self, opcode, arg, line, false);
+}
+
 // static void Ctx__revert_last_emit_(Ctx* self) {
 //     c11_vector__pop(&self->co->codes);
 //     c11_vector__pop(&self->co->codes_ex);
@@ -1507,7 +1512,7 @@ static Error* pop_context(Compiler* self) {
     // previously, we only do this if the last opcode is not a return
     // however, this is buggy...since there may be a jump to the end (out of bound) even if the last
     // opcode is a return
-    Ctx__emit_(ctx(), OP_RETURN_VALUE, BC_RETURN_VIRTUAL, BC_KEEPLINE);
+    Ctx__emit_virtual(ctx(), OP_RETURN_VALUE, 1, BC_KEEPLINE, true);
 
     CodeObject* co = ctx()->co;
     // find the last valid token
@@ -2416,7 +2421,6 @@ static Error* consume_pep695_py312(Compiler* self) {
 
 static Error* compile_function(Compiler* self, int decorators) {
     Error* err;
-    int def_line = prev()->line;
     consume(TK_ID);
     c11_sv decl_name_sv = Token__sv(prev());
     int decl_index;
@@ -2446,7 +2450,7 @@ static Error* compile_function(Compiler* self, int decorators) {
         }
     }
 
-    Ctx__emit_(ctx(), OP_LOAD_FUNCTION, decl_index, def_line);
+    Ctx__emit_(ctx(), OP_LOAD_FUNCTION, decl_index, prev()->line);
     Ctx__s_emit_decorators(ctx(), decorators);
 
     py_Name decl_name = py_namev(decl_name_sv);
@@ -2459,9 +2463,9 @@ static Error* compile_function(Compiler* self, int decorators) {
             }
         }
 
-        Ctx__emit_(ctx(), OP_STORE_CLASS_ATTR, Ctx__add_name(ctx(), decl_name), def_line);
+        Ctx__emit_(ctx(), OP_STORE_CLASS_ATTR, Ctx__add_name(ctx(), decl_name), prev()->line);
     } else {
-        NameExpr* e = NameExpr__new(def_line, decl_name, name_scope(self));
+        NameExpr* e = NameExpr__new(prev()->line, decl_name, name_scope(self));
         vtemit_store((Expr*)e, ctx());
         vtdelete((Expr*)e);
     }

src/interpreter/ceval.c

@@ -105,16 +105,13 @@ __NEXT_STEP:
     byte = co_codes[frame->ip];
 
     if(self->trace_info.func) {
-        bool is_virtual = byte.op == OP_RETURN_VALUE && byte.arg == BC_RETURN_VIRTUAL;
-        if(!is_virtual) {
-            SourceLocation loc = Frame__source_location(frame);
-            SourceLocation prev_loc = self->trace_info.prev_loc;
-            if(loc.lineno != prev_loc.lineno || loc.src != prev_loc.src) {
-                if(prev_loc.src) PK_DECREF(prev_loc.src);
-                PK_INCREF(loc.src);
-                self->trace_info.prev_loc = loc;
-                self->trace_info.func(frame, TRACE_EVENT_LINE);
-            }
+        SourceLocation loc = Frame__source_location(frame);
+        SourceLocation prev_loc = self->trace_info.prev_loc;
+        if(loc.lineno != prev_loc.lineno || loc.src != prev_loc.src) {
+            if(prev_loc.src) PK_DECREF(prev_loc.src);
+            PK_INCREF(loc.src);
+            self->trace_info.prev_loc = loc;
+            self->trace_info.func(frame, TRACE_EVENT_LINE);
         }
     }
 
@@ -1222,15 +1219,14 @@ __NEXT_STEP:
     c11__unreachable();
 
 __ERROR:
-    py_BaseException__stpush(frame,
-                             &self->curr_exception,
+    py_BaseException__stpush(frame, &self->curr_exception,
                              frame->co->src,
                              Frame__lineno(frame),
                              !frame->is_locals_special ? frame->co->name->data : NULL);
 __ERROR_RE_RAISE:
     do {
     } while(0);
-
+    
     int target = Frame__prepare_jump_exception_handler(frame, &self->stack);
     if(target >= 0) {
         // 1. Exception can be handled inside the current frame
@@ -1310,10 +1306,7 @@ bool pk_stack_binaryop(VM* self, py_Name op, py_Name rop) {
 
     py_Type lhs_t = rop ? TOP()->type : SECOND()->type;
     py_Type rhs_t = rop ? SECOND()->type : TOP()->type;
-    return TypeError("unsupported operand type(s) for '%s': '%t' and '%t'",
-                     pk_op2str(op),
-                     lhs_t,
-                     rhs_t);
+    return TypeError("unsupported operand type(s) for '%s': '%t' and '%t'", pk_op2str(op), lhs_t, rhs_t);
 }
 
 bool py_binaryop(py_Ref lhs, py_Ref rhs, py_Name op, py_Name rop) {

src/modules/dis.c

@@ -44,7 +44,7 @@ static bool disassemble(CodeObject* co) {
         c11_sbuf__write_cstr(&ss, buf);
 
         c11_sbuf__write_cstr(&ss, pk_opname(byte.op));
-        c11_sbuf__write_char(&ss, ' ');
+        c11_sbuf__write_char(&ss, ex.is_virtual ? '*' : ' ');
         int padding = 24 - strlen(pk_opname(byte.op));
         for(int j = 0; j < padding; j++)
             c11_sbuf__write_char(&ss, ' ');