summaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
Diffstat
-rw-r--r--py/compile.c6
-rw-r--r--py/emit.h8
-rw-r--r--py/emitbc.c11
-rw-r--r--py/emitcommon.c6
-rw-r--r--py/emitcpy.c16
-rw-r--r--py/emitnative.c8
-rw-r--r--py/scope.c1
-rw-r--r--py/scope.h1
8 files changed, 31 insertions, 26 deletions
diff --git a/py/compile.c b/py/compile.c
index d622242b62..ede09058ff 100644
--- a/py/compile.c
+++ b/py/compile.c
@@ -721,7 +721,7 @@ void close_over_variables_etc(compiler_t *comp, scope_t *this_scope, int n_dict_
for (int i = 0; i < this_scope->id_info_len; i++) {
id_info_t *id_info = &this_scope->id_info[i];
if (id_info->kind == ID_INFO_KIND_FREE) {
- EMIT(load_closure, id_info->qstr);
+ EMIT(load_closure, id_info->qstr, id_info->local_num);
nfree += 1;
}
}
@@ -2624,7 +2624,7 @@ void compile_scope(compiler_t *comp, scope_t *scope, pass_kind_t pass) {
if (id->kind == ID_INFO_KIND_LOCAL) {
EMIT(load_const_tok, PY_TOKEN_KW_NONE);
} else {
- EMIT(load_closure, comp->qstr___class__);
+ EMIT(load_closure, comp->qstr___class__, 0); // XXX check this is the correct local num
}
EMIT(return_value);
}
@@ -2729,6 +2729,8 @@ void compile_scope_compute_things(compiler_t *comp, scope_t *scope) {
}
}
+ // TODO compute the index of free and cell vars (freevars[idx] in CPython)
+
// compute flags
//scope->flags = 0; since we set some things in parameters
if (scope->kind != SCOPE_MODULE) {
diff --git a/py/emit.h b/py/emit.h
index a8bc1fca24..a04acf537d 100644
--- a/py/emit.h
+++ b/py/emit.h
@@ -47,22 +47,22 @@ typedef struct _emit_method_table_t {
void (*load_fast)(emit_t *emit, qstr qstr, int local_num);
void (*load_name)(emit_t *emit, qstr qstr);
void (*load_global)(emit_t *emit, qstr qstr);
- void (*load_deref)(emit_t *emit, qstr qstr);
- void (*load_closure)(emit_t *emit, qstr qstr);
+ void (*load_deref)(emit_t *emit, qstr qstr, int local_num);
+ void (*load_closure)(emit_t *emit, qstr qstr, int local_num);
void (*load_attr)(emit_t *emit, qstr qstr);
void (*load_method)(emit_t *emit, qstr qstr);
void (*load_build_class)(emit_t *emit);
void (*store_fast)(emit_t *emit, qstr qstr, int local_num);
void (*store_name)(emit_t *emit, qstr qstr);
void (*store_global)(emit_t *emit, qstr qstr);
- void (*store_deref)(emit_t *emit, qstr qstr);
+ void (*store_deref)(emit_t *emit, qstr qstr, int local_num);
void (*store_attr)(emit_t *emit, qstr qstr);
void (*store_subscr)(emit_t *emit);
void (*store_locals)(emit_t *emit);
void (*delete_fast)(emit_t *emit, qstr qstr, int local_num);
void (*delete_name)(emit_t *emit, qstr qstr);
void (*delete_global)(emit_t *emit, qstr qstr);
- void (*delete_deref)(emit_t *emit, qstr qstr);
+ void (*delete_deref)(emit_t *emit, qstr qstr, int local_num);
void (*delete_attr)(emit_t *emit, qstr qstr);
void (*delete_subscr)(emit_t *emit);
void (*dup_top)(emit_t *emit);
diff --git a/py/emitbc.c b/py/emitbc.c
index dbd9c3bb93..55a71cb463 100644
--- a/py/emitbc.c
+++ b/py/emitbc.c
@@ -291,12 +291,12 @@ static void emit_bc_load_global(emit_t *emit, qstr qstr) {
emit_write_byte_1_qstr(emit, PYBC_LOAD_GLOBAL, qstr);
}
-static void emit_bc_load_deref(emit_t *emit, qstr qstr) {
+static void emit_bc_load_deref(emit_t *emit, qstr qstr, int local_num) {
emit_pre(emit, 1);
assert(0);
}
-static void emit_bc_load_closure(emit_t *emit, qstr qstr) {
+static void emit_bc_load_closure(emit_t *emit, qstr qstr, int local_num) {
emit_pre(emit, 1);
assert(0);
}
@@ -337,7 +337,7 @@ static void emit_bc_store_global(emit_t *emit, qstr qstr) {
emit_write_byte_1_qstr(emit, PYBC_STORE_GLOBAL, qstr);
}
-static void emit_bc_store_deref(emit_t *emit, qstr qstr) {
+static void emit_bc_store_deref(emit_t *emit, qstr qstr, int local_num) {
emit_pre(emit, -1);
assert(0);
}
@@ -374,9 +374,10 @@ static void emit_bc_delete_global(emit_t *emit, qstr qstr) {
emit_write_byte_1_qstr(emit, PYBC_DELETE_GLOBAL, qstr);
}
-static void emit_bc_delete_deref(emit_t *emit, qstr qstr) {
+static void emit_bc_delete_deref(emit_t *emit, qstr qstr, int local_num) {
emit_pre(emit, 0);
- emit_write_byte_1_qstr(emit, PYBC_DELETE_DEREF, qstr);
+ assert(0);
+ //emit_write_byte_1_qstr(emit, PYBC_DELETE_DEREF, qstr);
}
static void emit_bc_delete_attr(emit_t *emit, qstr qstr) {
diff --git a/py/emitcommon.c b/py/emitcommon.c
index 07eb1812fd..0dfe96d891 100644
--- a/py/emitcommon.c
+++ b/py/emitcommon.c
@@ -28,7 +28,7 @@ void emit_common_load_id(emit_t *emit, const emit_method_table_t *emit_method_ta
} else if (id->kind == ID_INFO_KIND_LOCAL) {
EMIT(load_fast, qstr, id->local_num);
} else if (id->kind == ID_INFO_KIND_CELL || id->kind == ID_INFO_KIND_FREE) {
- EMIT(load_deref, qstr);
+ EMIT(load_deref, qstr, id->local_num);
} else {
assert(0);
}
@@ -48,7 +48,7 @@ void emit_common_store_id(emit_t *emit, const emit_method_table_t *emit_method_t
} else if (id->kind == ID_INFO_KIND_LOCAL) {
EMIT(store_fast, qstr, id->local_num);
} else if (id->kind == ID_INFO_KIND_CELL || id->kind == ID_INFO_KIND_FREE) {
- EMIT(store_deref, qstr);
+ EMIT(store_deref, qstr, id->local_num);
} else {
assert(0);
}
@@ -68,7 +68,7 @@ void emit_common_delete_id(emit_t *emit, const emit_method_table_t *emit_method_
} else if (id->kind == ID_INFO_KIND_LOCAL) {
EMIT(delete_fast, qstr, id->local_num);
} else if (id->kind == ID_INFO_KIND_CELL || id->kind == ID_INFO_KIND_FREE) {
- EMIT(delete_deref, qstr);
+ EMIT(delete_deref, qstr, id->local_num);
} else {
assert(0);
}
diff --git a/py/emitcpy.c b/py/emitcpy.c
index 089352c0fe..428e6fd00a 100644
--- a/py/emitcpy.c
+++ b/py/emitcpy.c
@@ -279,17 +279,17 @@ static void emit_cpy_load_global(emit_t *emit, qstr qstr) {
}
}
-static void emit_cpy_load_deref(emit_t *emit, qstr qstr) {
+static void emit_cpy_load_deref(emit_t *emit, qstr qstr, int local_num) {
emit_pre(emit, 1, 3);
if (emit->pass == PASS_3) {
- printf("LOAD_DEREF %s\n", qstr_str(qstr));
+ printf("LOAD_DEREF %d %s\n", local_num, qstr_str(qstr));
}
}
-static void emit_cpy_load_closure(emit_t *emit, qstr qstr) {
+static void emit_cpy_load_closure(emit_t *emit, qstr qstr, int local_num) {
emit_pre(emit, 1, 3);
if (emit->pass == PASS_3) {
- printf("LOAD_CLOSURE %s\n", qstr_str(qstr));
+ printf("LOAD_CLOSURE %d %s\n", local_num, qstr_str(qstr));
}
}
@@ -332,10 +332,10 @@ static void emit_cpy_store_global(emit_t *emit, qstr qstr) {
}
}
-static void emit_cpy_store_deref(emit_t *emit, qstr qstr) {
+static void emit_cpy_store_deref(emit_t *emit, qstr qstr, int local_num) {
emit_pre(emit, -1, 3);
if (emit->pass == PASS_3) {
- printf("STORE_DEREF %s\n", qstr_str(qstr));
+ printf("STORE_DEREF %d %s\n", local_num, qstr_str(qstr));
}
}
@@ -381,10 +381,10 @@ static void emit_cpy_delete_global(emit_t *emit, qstr qstr) {
}
}
-static void emit_cpy_delete_deref(emit_t *emit, qstr qstr) {
+static void emit_cpy_delete_deref(emit_t *emit, qstr qstr, int local_num) {
emit_pre(emit, 0, 3);
if (emit->pass == PASS_3) {
- printf("DELETE_DEREF %s\n", qstr_str(qstr));
+ printf("DELETE_DEREF %d %s\n", local_num, qstr_str(qstr));
}
}
diff --git a/py/emitnative.c b/py/emitnative.c
index def1a66359..b535df74b6 100644
--- a/py/emitnative.c
+++ b/py/emitnative.c
@@ -675,13 +675,13 @@ static void emit_native_load_global(emit_t *emit, qstr qstr) {
emit_post_push_reg(emit, VTYPE_PYOBJ, REG_RET);
}
-static void emit_native_load_deref(emit_t *emit, qstr qstr) {
+static void emit_native_load_deref(emit_t *emit, qstr qstr, int local_num) {
// not implemented
// in principle could support this quite easily (ldr r0, [r0, #0]) and then get closed over variables!
assert(0);
}
-static void emit_native_load_closure(emit_t *emit, qstr qstr) {
+static void emit_native_load_closure(emit_t *emit, qstr qstr, int local_num) {
// not implemented
assert(0);
}
@@ -760,7 +760,7 @@ static void emit_native_store_global(emit_t *emit, qstr qstr) {
assert(0);
}
-static void emit_native_store_deref(emit_t *emit, qstr qstr) {
+static void emit_native_store_deref(emit_t *emit, qstr qstr, int local_num) {
// not implemented
assert(0);
}
@@ -812,7 +812,7 @@ static void emit_native_delete_global(emit_t *emit, qstr qstr) {
assert(0);
}
-static void emit_native_delete_deref(emit_t *emit, qstr qstr) {
+static void emit_native_delete_deref(emit_t *emit, qstr qstr, int local_num) {
// not supported
assert(0);
}
diff --git a/py/scope.c b/py/scope.c
index c5816871c2..7f844108e3 100644
--- a/py/scope.c
+++ b/py/scope.c
@@ -103,6 +103,7 @@ id_info_t *scope_find_or_add_id(scope_t *scope, qstr qstr, bool *added) {
id_info->param = false;
id_info->kind = 0;
id_info->qstr = qstr;
+ id_info->local_num = 0;
*added = true;
return id_info;
}
diff --git a/py/scope.h b/py/scope.h
index 1b626c8ec1..9a04c56f61 100644
--- a/py/scope.h
+++ b/py/scope.h
@@ -7,6 +7,7 @@ enum {
};
typedef struct _id_info_t {
+ // TODO compress this info to make structure smaller in memory
bool param;
int kind;
qstr qstr;
generated by cgit v1.2.3 (git 2.39.1) at 2025-05-19 19:11:55 +0000