Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-23 13:00:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

some optimize

Browse Source
This commit is contained in:
blueloveTH 2023-02-19 23:26:57 +08:00
parent 8433d40f2c
commit ad99bd5b21
3 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/compiler.h
View File

@ -401,11 +401,18 @@ private:
} }
void exprAssign() { void exprAssign() {
int lhs = co()->codes.empty() ? -1 : co()->codes.size() - 1;
co()->_rvalue = true; co()->_rvalue = true;
TokenIndex op = parser->prev.type; TokenIndex op = parser->prev.type;
if(op == TK("=")) { // a = (expr) if(op == TK("=")) { // a = (expr)
EXPR_TUPLE(); EXPR_TUPLE();
emit(OP_STORE_REF); if(lhs!=-1 && co()->codes[lhs].op == OP_LOAD_NAME_REF){
emit(OP_STORE_NAME, co()->codes[lhs].arg);
co()->codes[lhs].op = OP_NO_OP;
co()->codes[lhs].arg = -1;
}else{
emit(OP_STORE_REF);
}
}else{ // a += (expr) -> a = a + (expr) }else{ // a += (expr) -> a = a + (expr)
EXPR(); EXPR();
switch (op) { switch (op) {

4
src/pocketpy.h
View File

@ -591,7 +591,9 @@ void add_module_math(VM* vm){
void add_module_dis(VM* vm){ void add_module_dis(VM* vm){
PyVar mod = vm->new_module("dis"); PyVar mod = vm->new_module("dis");
vm->bind_func<1>(mod, "dis", [](VM* vm, pkpy::Args& args) { vm->bind_func<1>(mod, "dis", [](VM* vm, pkpy::Args& args) {
CodeObject_ code = vm->PyFunction_AS_C(args[0]).code; PyVar f = args[0];
if(f->is_type(vm->tp_bound_method)) f = vm->PyBoundMethod_AS_C(args[0]).method;
CodeObject_ code = vm->PyFunction_AS_C(f).code;
(*vm->_stdout) << vm->disassemble(code); (*vm->_stdout) << vm->disassemble(code);
return vm->None; return vm->None;
}); });

2
src/vm.h
View File

@ -483,7 +483,7 @@ public:
if(byte.op == OP_LOAD_CONST){ if(byte.op == OP_LOAD_CONST){
argStr += " (" + PyStr_AS_C(asRepr(co->consts[byte.arg])) + ")"; argStr += " (" + PyStr_AS_C(asRepr(co->consts[byte.arg])) + ")";
} }
if(byte.op == OP_LOAD_NAME_REF || byte.op == OP_LOAD_NAME || byte.op == OP_RAISE){ if(byte.op == OP_LOAD_NAME_REF || byte.op == OP_LOAD_NAME || byte.op == OP_RAISE || byte.op == OP_STORE_NAME){
argStr += " (" + co->names[byte.arg].first.escape(true) + ")"; argStr += " (" + co->names[byte.arg].first.escape(true) + ")";
} }
if(byte.op == OP_FAST_INDEX || byte.op == OP_FAST_INDEX_REF){ if(byte.op == OP_FAST_INDEX || byte.op == OP_FAST_INDEX_REF){