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

add py_op

Browse Source
This commit is contained in:
blueloveTH 2024-05-11 12:44:42 +08:00
parent b1516a6ad0
commit a6a834da29
2 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/pocketpy/vm.h
View File

@ -156,6 +156,7 @@ public:
PyObject* __curr_class; PyObject* __curr_class;
PyObject* __cached_object_new; PyObject* __cached_object_new;
std::map<std::string_view, CodeObject_> __cached_codes; std::map<std::string_view, CodeObject_> __cached_codes;
std::map<std::string_view, PyObject*> __cached_op_funcs;
FuncDecl_ __dynamic_func_decl; FuncDecl_ __dynamic_func_decl;
#if PK_ENABLE_PROFILER #if PK_ENABLE_PROFILER
@ -205,10 +206,12 @@ public:
bool py_le(PyObject* lhs, PyObject* rhs); // (lhs, rhs) -> lhs <= rhs bool py_le(PyObject* lhs, PyObject* rhs); // (lhs, rhs) -> lhs <= rhs
bool py_gt(PyObject* lhs, PyObject* rhs); // (lhs, rhs) -> lhs > rhs bool py_gt(PyObject* lhs, PyObject* rhs); // (lhs, rhs) -> lhs > rhs
bool py_ge(PyObject* lhs, PyObject* rhs); // (lhs, rhs) -> lhs >= rhs bool py_ge(PyObject* lhs, PyObject* rhs); // (lhs, rhs) -> lhs >= rhs
bool py_ne(PyObject* lhs, PyObject* rhs) { // (lhs, rhs) -> lhs != rhs bool py_ne(PyObject* lhs, PyObject* rhs){ // (lhs, rhs) -> lhs != rhs
return !py_eq(lhs, rhs); return !py_eq(lhs, rhs);
} }
PyObject* py_op(std::string_view name); // (name) -> operator.name
void py_exec(std::string_view, PyObject*, PyObject*); // exec(source, globals, locals) void py_exec(std::string_view, PyObject*, PyObject*); // exec(source, globals, locals)
PyObject* py_eval(std::string_view, PyObject*, PyObject*); // eval(source, globals, locals) PyObject* py_eval(std::string_view, PyObject*, PyObject*); // eval(source, globals, locals)
#endif #endif

12
src/vm.cpp
View File

@ -244,6 +244,18 @@ namespace pkpy{
return false; return false;
} }
PyObject* VM::py_op(std::string_view name){
PyObject* func;
auto it = __cached_op_funcs.find(name);
if(it == __cached_op_funcs.end()){
func = py_import("operator")->attr(StrName::get(name));
__cached_op_funcs[name] = func;
}else{
func = it->second;
}
return func;
}
i64 VM::normalized_index(i64 index, int size){ i64 VM::normalized_index(i64 index, int size){
if(index < 0) index += size; if(index < 0) index += size;
if(index < 0 || index >= size){ if(index < 0 || index >= size){