Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-12-07 02:30:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add num pool

Browse Source 
Update vm.h
This commit is contained in:
blueloveTH 2022-11-08 23:04:41 +08:00
parent 035fe636ab
commit 2f12ace827
2 changed files with 36 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.cpp
View File

@ -5,7 +5,7 @@
#include "pocketpy.h" #include "pocketpy.h"
#define PK_DEBUG #define PK_DEBUG
//#define PK_DEBUG_TIME #define PK_DEBUG_TIME
class Timer{ class Timer{
private: private:

42
src/vm.h
View File

@ -3,16 +3,21 @@
#include "codeobject.h" #include "codeobject.h"
#include "iter.h" #include "iter.h"
#define DEF_NATIVE(type, ctype, ptype) \ #define __DEF_PY_AS_C(type, ctype, ptype) \
inline PyVar Py##type(ctype value) { \
return newObject(ptype, value); \
} \
\
inline ctype& Py##type##_AS_C(const PyVar& obj) { \ inline ctype& Py##type##_AS_C(const PyVar& obj) { \
__checkType(obj, ptype); \ __checkType(obj, ptype); \
return std::get<ctype>(obj->_native); \ return std::get<ctype>(obj->_native); \
} }
#define __DEF_PY(type, ctype, ptype) \
inline PyVar Py##type(ctype value) { \
return newObject(ptype, value); \
}
#define DEF_NATIVE(type, ctype, ptype) \
__DEF_PY(type, ctype, ptype) \
__DEF_PY_AS_C(type, ctype, ptype)
#define BINARY_XXX(i) \ #define BINARY_XXX(i) \
{PyVar rhs = frame->popValue(this); \ {PyVar rhs = frame->popValue(this); \
PyVar lhs = frame->popValue(this); \ PyVar lhs = frame->popValue(this); \
@ -26,9 +31,12 @@
// TODO: we should split this into stdout and stderr // TODO: we should split this into stdout and stderr
typedef void(*PrintFn)(const char*); typedef void(*PrintFn)(const char*);
#define NUM_POOL_MAX_SIZE 1024
class VM{ class VM{
private: private:
std::stack< std::shared_ptr<Frame> > callstack; std::stack< std::shared_ptr<Frame> > callstack;
std::vector<PyObject*> numPool;
public: public:
StlDict _types; // builtin types StlDict _types; // builtin types
PyVar None, True, False; PyVar None, True, False;
@ -455,6 +463,24 @@ public:
return obj; return obj;
} }
PyVar newNumber(PyVar type, _Value _native) {
if(type != _tp_int && type != _tp_float)
_error("SystemError", "type is not a number type");
PyObject* _raw = nullptr;
if(numPool.size() > 0) {
_raw = numPool.back();
_raw->_native = _native;
numPool.pop_back();
}else{
_raw = new PyObject(_native);
}
PyVar obj = PyVar(_raw, [this](PyObject* p){
if(numPool.size() < NUM_POOL_MAX_SIZE) numPool.push_back(p);
});
setAttr(obj, __class__, type);
return obj;
}
PyVar newModule(_Str name) { PyVar newModule(_Str name) {
PyVar obj = newObject(_tp_module, 0); PyVar obj = newObject(_tp_module, 0);
setAttr(obj, "__name__", PyStr(name)); setAttr(obj, "__name__", PyStr(name));
@ -549,8 +575,8 @@ public:
PyVar _tp_function, _tp_native_function, _tp_native_iterator, _tp_bounded_method; PyVar _tp_function, _tp_native_function, _tp_native_iterator, _tp_bounded_method;
PyVar _tp_slice, _tp_range, _tp_module, _tp_pointer; PyVar _tp_slice, _tp_range, _tp_module, _tp_pointer;
DEF_NATIVE(Int, int, _tp_int) __DEF_PY_AS_C(Int, int, _tp_int)
DEF_NATIVE(Float, float, _tp_float) __DEF_PY_AS_C(Float, float, _tp_float)
DEF_NATIVE(Str, _Str, _tp_str) DEF_NATIVE(Str, _Str, _tp_str)
DEF_NATIVE(List, PyVarList, _tp_list) DEF_NATIVE(List, PyVarList, _tp_list)
DEF_NATIVE(Tuple, PyVarList, _tp_tuple) DEF_NATIVE(Tuple, PyVarList, _tp_tuple)
@ -562,6 +588,8 @@ public:
DEF_NATIVE(Slice, _Slice, _tp_slice) DEF_NATIVE(Slice, _Slice, _tp_slice)
DEF_NATIVE(Pointer, _Pointer, _tp_pointer) DEF_NATIVE(Pointer, _Pointer, _tp_pointer)
inline PyVar PyInt(int i) { return newNumber(_tp_int, i); }
inline PyVar PyFloat(float f) { return newNumber(_tp_float, f); }
inline bool PyBool_AS_C(PyVar obj){return obj == True;} inline bool PyBool_AS_C(PyVar obj){return obj == True;}
inline PyVar PyBool(bool value){return value ? True : False;} inline PyVar PyBool(bool value){return value ? True : False;}