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

use memset

Browse Source
This commit is contained in:
blueloveTH 2024-05-25 00:10:00 +08:00
parent 30d968dea5
commit 70f2a92b9f
4 changed files with 49 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/pocketpy/common.h
View File

@ -216,7 +216,10 @@ struct PyVar final{
explicit operator bool() const { return (bool)type; } explicit operator bool() const { return (bool)type; }
void set_null() { _qword(0) = 0; _qword(1) = 0; }
i64 _qword(int i) const { return ((const i64*)this)[i]; } i64 _qword(int i) const { return ((const i64*)this)[i]; }
i64& _qword(int i) { return ((i64*)this)[i]; }
bool operator==(const PyVar& other) const { bool operator==(const PyVar& other) const {
return _qword(0) == other._qword(0) && _qword(1) == other._qword(1); return _qword(0) == other._qword(0) && _qword(1) == other._qword(1);

2
include/pocketpy/vm.h
View File

@ -261,7 +261,7 @@ public:
#endif #endif
#if PK_REGION("Name Lookup Methods") #if PK_REGION("Name Lookup Methods")
PyVar find_name_in_mro(Type cls, StrName name); PyVar* find_name_in_mro(Type cls, StrName name);
PyVar get_unbound_method(PyVar obj, StrName name, PyVar* self, bool throw_err=true, bool fallback=false); PyVar get_unbound_method(PyVar obj, StrName name, PyVar* self, bool throw_err=true, bool fallback=false);
PyVar getattr(PyVar obj, StrName name, bool throw_err=true); PyVar getattr(PyVar obj, StrName name, bool throw_err=true);
void delattr(PyVar obj, StrName name); void delattr(PyVar obj, StrName name);

4
src/ceval.cpp
View File

@ -323,14 +323,14 @@ __NEXT_STEP:
case OP_DELETE_FAST:{ case OP_DELETE_FAST:{
PyVar _0 = frame->_locals[byte.arg]; PyVar _0 = frame->_locals[byte.arg];
if(_0 == PY_NULL) vm->UnboundLocalError(frame->co->varnames[byte.arg]); if(_0 == PY_NULL) vm->UnboundLocalError(frame->co->varnames[byte.arg]);
frame->_locals[byte.arg] = PY_NULL; frame->_locals[byte.arg].set_null();
}DISPATCH() }DISPATCH()
case OP_DELETE_NAME:{ case OP_DELETE_NAME:{
StrName _name(byte.arg); StrName _name(byte.arg);
if(frame->_callable != nullptr){ if(frame->_callable != nullptr){
PyVar* slot = frame->_locals.try_get_name(_name); PyVar* slot = frame->_locals.try_get_name(_name);
if(slot != nullptr){ if(slot != nullptr){
*slot = PY_NULL; slot->set_null();
}else{ }else{
Function& func = PK_OBJ_GET(Function, frame->_callable); Function& func = PK_OBJ_GET(Function, frame->_callable);
if(func.decl == __dynamic_func_decl){ if(func.decl == __dynamic_func_decl){

87
src/vm.cpp
View File

@ -143,10 +143,10 @@ namespace pkpy{
mod->attr().set("argv", VAR(std::move(argv_))); mod->attr().set("argv", VAR(std::move(argv_)));
} }
PyVar VM::find_name_in_mro(Type cls, StrName name){ PyVar* VM::find_name_in_mro(Type cls, StrName name){
PyVar val; PyVar* val;
do{ do{
val = _t(cls)->attr().try_get(name); val = _t(cls)->attr().try_get_2(name);
if(val != nullptr) return val; if(val != nullptr) return val;
cls = _all_types[cls].base; cls = _all_types[cls].base;
if(!cls) break; if(!cls) break;
@ -960,9 +960,8 @@ void VM::__prepare_py_call(PyVar* buffer, ArgsView args, ArgsView kwargs, const
int i = 0; int i = 0;
// prepare args // prepare args
memset(buffer, 0, co_nlocals * sizeof(PyVar));
for(int index: decl->args) buffer[index] = args[i++]; for(int index: decl->args) buffer[index] = args[i++];
// set extra varnames to PY_NULL
for(int j=i; j<co_nlocals; j++) buffer[j] = PY_NULL;
// prepare kwdefaults // prepare kwdefaults
for(auto& kv: decl->kwargs) buffer[kv.index] = kv.value; for(auto& kv: decl->kwargs) buffer[kv.index] = kv.value;
@ -1055,7 +1054,7 @@ PyVar VM::vectorcall(int ARGC, int KWARGC, bool op_call){
// ^p0 ^p1 ^_sp // ^p0 ^p1 ^_sp
s_data.reset(_base + co_nlocals); s_data.reset(_base + co_nlocals);
// initialize local variables to PY_NULL // initialize local variables to PY_NULL
for(PyVar* p=p1; p!=s_data._sp; p++) *p = PY_NULL; memset(p1, 0, (char*)s_data._sp - (char*)p1);
break; break;
case FuncType::EMPTY: case FuncType::EMPTY:
if(args.size() != fn.decl->args.size()) TypeError(_S(co->name, "() takes ", fn.decl->args.size(), " positional arguments but ", args.size(), " were given")); if(args.size() != fn.decl->args.size()) TypeError(_S(co->name, "() takes ", fn.decl->args.size(), " positional arguments but ", args.size(), " were given"));
@ -1099,7 +1098,7 @@ PyVar VM::vectorcall(int ARGC, int KWARGC, bool op_call){
if(callable_t == tp_type){ if(callable_t == tp_type){
// [type, NULL, args..., kwargs...] // [type, NULL, args..., kwargs...]
PyVar new_f = find_name_in_mro(PK_OBJ_GET(Type, callable), __new__); PyVar new_f = *find_name_in_mro(PK_OBJ_GET(Type, callable), __new__);
PyVar obj; PyVar obj;
PK_DEBUG_ASSERT(new_f != nullptr && p0[1]==PY_NULL); PK_DEBUG_ASSERT(new_f != nullptr && p0[1]==PY_NULL);
if(new_f == __cached_object_new) { if(new_f == __cached_object_new) {
@ -1165,45 +1164,45 @@ PyVar VM::getattr(PyVar obj, StrName name, bool throw_err){
}else{ }else{
objtype = _tp(obj); objtype = _tp(obj);
} }
PyVar cls_var = find_name_in_mro(objtype, name); PyVar* cls_var = find_name_in_mro(objtype, name);
if(cls_var != nullptr){ if(cls_var != nullptr){
// handle descriptor // handle descriptor
if(is_type(cls_var, tp_property)){ if(is_type(*cls_var, tp_property)){
const Property& prop = PK_OBJ_GET(Property, cls_var); const Property& prop = PK_OBJ_GET(Property, *cls_var);
return call(prop.getter, obj); return call(prop.getter, obj);
} }
} }
// handle instance __dict__ // handle instance __dict__
if(!is_tagged(obj) && obj->is_attr_valid()){ if(!is_tagged(obj) && obj->is_attr_valid()){
PyVar val; PyVar* val;
if(obj.type == tp_type){ if(obj.type == tp_type){
val = find_name_in_mro(PK_OBJ_GET(Type, obj), name); val = find_name_in_mro(PK_OBJ_GET(Type, obj), name);
if(val != nullptr){ if(val != nullptr){
if(is_tagged(val)) return val; if(is_tagged(*val)) return *val;
if(val.type == tp_staticmethod) return PK_OBJ_GET(StaticMethod, val).func; if(val->type == tp_staticmethod) return PK_OBJ_GET(StaticMethod, *val).func;
if(val.type == tp_classmethod) return VAR(BoundMethod(obj, PK_OBJ_GET(ClassMethod, val).func)); if(val->type == tp_classmethod) return VAR(BoundMethod(obj, PK_OBJ_GET(ClassMethod, *val).func));
return val; return *val;
} }
}else{ }else{
val = obj->attr().try_get_likely_found(name); val = obj->attr().try_get_2_likely_found(name);
if(val != nullptr) return val; if(val != nullptr) return *val;
} }
} }
if(cls_var != nullptr){ if(cls_var != nullptr){
// bound method is non-data descriptor // bound method is non-data descriptor
if(!is_tagged(cls_var)){ if(!is_tagged(*cls_var)){
switch(cls_var.type.index){ switch(cls_var->type.index){
case tp_function.index: case tp_function.index:
return VAR(BoundMethod(obj, cls_var)); return VAR(BoundMethod(obj, *cls_var));
case tp_native_func.index: case tp_native_func.index:
return VAR(BoundMethod(obj, cls_var)); return VAR(BoundMethod(obj, *cls_var));
case tp_staticmethod.index: case tp_staticmethod.index:
return PK_OBJ_GET(StaticMethod, cls_var).func; return PK_OBJ_GET(StaticMethod, *cls_var).func;
case tp_classmethod.index: case tp_classmethod.index:
return VAR(BoundMethod(_t(objtype), PK_OBJ_GET(ClassMethod, cls_var).func)); return VAR(BoundMethod(_t(objtype), PK_OBJ_GET(ClassMethod, *cls_var).func));
} }
} }
return cls_var; return *cls_var;
} }
const PyTypeInfo* ti = &_all_types[objtype]; const PyTypeInfo* ti = &_all_types[objtype];
@ -1219,7 +1218,7 @@ PyVar VM::getattr(PyVar obj, StrName name, bool throw_err){
// used by OP_LOAD_METHOD // used by OP_LOAD_METHOD
// try to load a unbound method (fallback to `getattr` if not found) // try to load a unbound method (fallback to `getattr` if not found)
PyVar VM::get_unbound_method(PyVar obj, StrName name, PyVar* self, bool throw_err, bool fallback){ PyVar VM::get_unbound_method(PyVar obj, StrName name, PyVar* self, bool throw_err, bool fallback){
*self = PY_NULL; self->set_null();
Type objtype(0); Type objtype(0);
// handle super() proxy // handle super() proxy
if(is_type(obj, tp_super)){ if(is_type(obj, tp_super)){
@ -1229,37 +1228,37 @@ PyVar VM::get_unbound_method(PyVar obj, StrName name, PyVar* self, bool throw_er
}else{ }else{
objtype = _tp(obj); objtype = _tp(obj);
} }
PyVar cls_var = find_name_in_mro(objtype, name); PyVar* cls_var = find_name_in_mro(objtype, name);
if(fallback){ if(fallback){
if(cls_var != nullptr){ if(cls_var != nullptr){
// handle descriptor // handle descriptor
if(is_type(cls_var, tp_property)){ if(is_type(*cls_var, tp_property)){
const Property& prop = PK_OBJ_GET(Property, cls_var); const Property& prop = PK_OBJ_GET(Property, *cls_var);
return call(prop.getter, obj); return call(prop.getter, obj);
} }
} }
// handle instance __dict__ // handle instance __dict__
if(!is_tagged(obj) && obj->is_attr_valid()){ if(!is_tagged(obj) && obj->is_attr_valid()){
PyVar val; PyVar* val;
if(obj.type == tp_type){ if(obj.type == tp_type){
val = find_name_in_mro(PK_OBJ_GET(Type, obj), name); val = find_name_in_mro(PK_OBJ_GET(Type, obj), name);
if(val != nullptr){ if(val != nullptr){
if(is_tagged(val)) return val; if(is_tagged(*val)) return *val;
if(val.type == tp_staticmethod) return PK_OBJ_GET(StaticMethod, val).func; if(val->type == tp_staticmethod) return PK_OBJ_GET(StaticMethod, *val).func;
if(val.type == tp_classmethod) return VAR(BoundMethod(obj, PK_OBJ_GET(ClassMethod, val).func)); if(val->type == tp_classmethod) return VAR(BoundMethod(obj, PK_OBJ_GET(ClassMethod, *val).func));
return val; return *val;
} }
}else{ }else{
val = obj->attr().try_get_likely_found(name); val = obj->attr().try_get_2_likely_found(name);
if(val != nullptr) return val; if(val != nullptr) return *val;
} }
} }
} }
if(cls_var != nullptr){ if(cls_var != nullptr){
if(!is_tagged(cls_var)){ if(!is_tagged(*cls_var)){
switch(cls_var.type.index){ switch(cls_var->type.index){
case tp_function.index: case tp_function.index:
*self = obj; *self = obj;
break; break;
@ -1267,14 +1266,14 @@ PyVar VM::get_unbound_method(PyVar obj, StrName name, PyVar* self, bool throw_er
*self = obj; *self = obj;
break; break;
case tp_staticmethod.index: case tp_staticmethod.index:
*self = PY_NULL; self->set_null();
return PK_OBJ_GET(StaticMethod, cls_var).func; return PK_OBJ_GET(StaticMethod, *cls_var).func;
case tp_classmethod.index: case tp_classmethod.index:
*self = _t(objtype); *self = _t(objtype);
return PK_OBJ_GET(ClassMethod, cls_var).func; return PK_OBJ_GET(ClassMethod, *cls_var).func;
} }
} }
return cls_var; return *cls_var;
} }
const PyTypeInfo* ti = &_all_types[objtype]; const PyTypeInfo* ti = &_all_types[objtype];
@ -1297,11 +1296,11 @@ void VM::setattr(PyVar obj, StrName name, PyVar value){
}else{ }else{
objtype = _tp(obj); objtype = _tp(obj);
} }
PyVar cls_var = find_name_in_mro(objtype, name); PyVar* cls_var = find_name_in_mro(objtype, name);
if(cls_var != nullptr){ if(cls_var != nullptr){
// handle descriptor // handle descriptor
if(is_type(cls_var, tp_property)){ if(is_type(*cls_var, tp_property)){
const Property& prop = _CAST(Property&, cls_var); const Property& prop = _CAST(Property&, *cls_var);
if(prop.setter != vm->None){ if(prop.setter != vm->None){
call(prop.setter, obj, value); call(prop.setter, obj, value);
}else{ }else{