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

...

Browse Source
This commit is contained in:
blueloveTH 2024-01-17 15:39:09 +08:00
parent 75233f9579
commit 81cf28f685
1 changed files with 7 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
src/ceval.cpp
View File

@ -370,17 +370,6 @@ __NEXT_STEP:;
PUSH(_0); PUSH(_0);
} DISPATCH(); } DISPATCH();
/*****************************************/ /*****************************************/
#define PREDICT_INT_OP(op) \
if(is_small_int(TOP()) && is_small_int(SECOND())){ \
_1 = POPX(); \
_0 = TOP(); \
if constexpr(#op[0] == '/' || #op[0] == '%'){ \
if(_py_sint(_1) == 0) ZeroDivisionError(); \
} \
TOP() = VAR(_py_sint(_0) op _py_sint(_1)); \
DISPATCH(); \
}
#define BINARY_OP_SPECIAL(func) \ #define BINARY_OP_SPECIAL(func) \
_1 = POPX(); \ _1 = POPX(); \
_0 = TOP(); \ _0 = TOP(); \
@ -415,13 +404,11 @@ __NEXT_STEP:;
} DISPATCH(); } DISPATCH();
TARGET(BINARY_ADD){ TARGET(BINARY_ADD){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(+);
BINARY_OP_SPECIAL(__add__); BINARY_OP_SPECIAL(__add__);
BINARY_OP_RSPECIAL("+", __radd__); BINARY_OP_RSPECIAL("+", __radd__);
} DISPATCH() } DISPATCH()
TARGET(BINARY_SUB){ TARGET(BINARY_SUB){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(-);
BINARY_OP_SPECIAL(__sub__); BINARY_OP_SPECIAL(__sub__);
BINARY_OP_RSPECIAL("-", __rsub__); BINARY_OP_RSPECIAL("-", __rsub__);
} DISPATCH() } DISPATCH()
@ -432,13 +419,11 @@ __NEXT_STEP:;
} DISPATCH() } DISPATCH()
TARGET(BINARY_FLOORDIV){ TARGET(BINARY_FLOORDIV){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(/);
BINARY_OP_SPECIAL(__floordiv__); BINARY_OP_SPECIAL(__floordiv__);
if(TOP() == NotImplemented) BinaryOptError("//"); if(TOP() == NotImplemented) BinaryOptError("//");
} DISPATCH() } DISPATCH()
TARGET(BINARY_MOD){ TARGET(BINARY_MOD){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(%);
BINARY_OP_SPECIAL(__mod__); BINARY_OP_SPECIAL(__mod__);
if(TOP() == NotImplemented) BinaryOptError("%"); if(TOP() == NotImplemented) BinaryOptError("%");
} DISPATCH() } DISPATCH()
@ -474,31 +459,26 @@ __NEXT_STEP:;
} DISPATCH() } DISPATCH()
TARGET(BITWISE_LSHIFT){ TARGET(BITWISE_LSHIFT){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(<<);
BINARY_OP_SPECIAL(__lshift__); BINARY_OP_SPECIAL(__lshift__);
if(TOP() == NotImplemented) BinaryOptError("<<"); if(TOP() == NotImplemented) BinaryOptError("<<");
} DISPATCH() } DISPATCH()
TARGET(BITWISE_RSHIFT){ TARGET(BITWISE_RSHIFT){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(>>);
BINARY_OP_SPECIAL(__rshift__); BINARY_OP_SPECIAL(__rshift__);
if(TOP() == NotImplemented) BinaryOptError(">>"); if(TOP() == NotImplemented) BinaryOptError(">>");
} DISPATCH() } DISPATCH()
TARGET(BITWISE_AND){ TARGET(BITWISE_AND){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(&);
BINARY_OP_SPECIAL(__and__); BINARY_OP_SPECIAL(__and__);
if(TOP() == NotImplemented) BinaryOptError("&"); if(TOP() == NotImplemented) BinaryOptError("&");
} DISPATCH() } DISPATCH()
TARGET(BITWISE_OR){ TARGET(BITWISE_OR){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(|);
BINARY_OP_SPECIAL(__or__); BINARY_OP_SPECIAL(__or__);
if(TOP() == NotImplemented) BinaryOptError("|"); if(TOP() == NotImplemented) BinaryOptError("|");
} DISPATCH() } DISPATCH()
TARGET(BITWISE_XOR){ TARGET(BITWISE_XOR){
PyObject* _0; PyObject* _1; const PyTypeInfo* _ti; PyObject* _0; PyObject* _1; const PyTypeInfo* _ti;
PREDICT_INT_OP(^);
BINARY_OP_SPECIAL(__xor__); BINARY_OP_SPECIAL(__xor__);
if(TOP() == NotImplemented) BinaryOptError("^"); if(TOP() == NotImplemented) BinaryOptError("^");
} DISPATCH() } DISPATCH()
@ -509,7 +489,6 @@ __NEXT_STEP:;
} DISPATCH(); } DISPATCH();
#undef BINARY_OP_SPECIAL #undef BINARY_OP_SPECIAL
#undef PREDICT_INT_OP
TARGET(IS_OP){ TARGET(IS_OP){
PyObject* _1 = POPX(); // rhs PyObject* _1 = POPX(); // rhs
@ -536,26 +515,23 @@ __NEXT_STEP:;
frame->jump_abs(_CAST(uint16_t, POPX())); frame->jump_abs(_CAST(uint16_t, POPX()));
DISPATCH(); DISPATCH();
TARGET(POP_JUMP_IF_FALSE){ TARGET(POP_JUMP_IF_FALSE){
PyObject* _0 = POPX(); if(!py_bool(TOP())) frame->jump_abs(byte.arg);
if(_0==False || !py_bool(_0)) frame->jump_abs(byte.arg); POP();
} DISPATCH(); } DISPATCH();
TARGET(POP_JUMP_IF_TRUE){ TARGET(POP_JUMP_IF_TRUE){
PyObject* _0 = POPX(); if(py_bool(TOP())) frame->jump_abs(byte.arg);
if(_0==True || py_bool(_0)) frame->jump_abs(byte.arg); POP();
} DISPATCH(); } DISPATCH();
TARGET(JUMP_IF_TRUE_OR_POP){ TARGET(JUMP_IF_TRUE_OR_POP){
PyObject* _0 = TOP(); if(py_bool(TOP())) frame->jump_abs(byte.arg);
if(_0==True || py_bool(_0)) frame->jump_abs(byte.arg);
else POP(); else POP();
} DISPATCH(); } DISPATCH();
TARGET(JUMP_IF_FALSE_OR_POP){ TARGET(JUMP_IF_FALSE_OR_POP){
PyObject* _0 = TOP(); if(!py_bool(TOP())) frame->jump_abs(byte.arg);
if(_0==False || !py_bool(_0)) frame->jump_abs(byte.arg);
else POP(); else POP();
} DISPATCH(); } DISPATCH();
TARGET(SHORTCUT_IF_FALSE_OR_POP){ TARGET(SHORTCUT_IF_FALSE_OR_POP){
PyObject* _0 = TOP(); if(!py_bool(TOP())){ // [b, False]
if(_0==False || !py_bool(_0)){ // [b, False]
STACK_SHRINK(2); // [] STACK_SHRINK(2); // []
PUSH(vm->False); // [False] PUSH(vm->False); // [False]
frame->jump_abs(byte.arg); frame->jump_abs(byte.arg);