| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691 |
- #include "pocketpy/interpreter/vm.h"
- #include "pocketpy/common/memorypool.h"
- #include "pocketpy/common/sstream.h"
- #include "pocketpy/common/utils.h"
- #include "pocketpy/interpreter/generator.h"
- #include "pocketpy/interpreter/modules.h"
- #include "pocketpy/objects/base.h"
- #include "pocketpy/common/_generated.h"
- #include "pocketpy/pocketpy.h"
- #include <stdbool.h>
- static char* pk_default_import_file(const char* path) {
- #if PK_ENABLE_OS
- FILE* f = fopen(path, "rb");
- if(f == NULL) return NULL;
- fseek(f, 0, SEEK_END);
- long size = ftell(f);
- fseek(f, 0, SEEK_SET);
- char* buffer = malloc(size + 1);
- size = fread(buffer, 1, size, f);
- buffer[size] = 0;
- fclose(f);
- return buffer;
- #else
- return NULL;
- #endif
- }
- static void pk_default_print(const char* data) { printf("%s", data); }
- static void py_TypeInfo__ctor(py_TypeInfo* self,
- py_Name name,
- py_Type index,
- py_Type base,
- py_TValue module) {
- memset(self, 0, sizeof(py_TypeInfo));
- self->name = name;
- self->base = base;
- // create type object with __dict__
- ManagedHeap* heap = &pk_current_vm->heap;
- PyObject* typeobj = ManagedHeap__new(heap, tp_type, -1, sizeof(py_Type));
- *(py_Type*)PyObject__userdata(typeobj) = index;
- self->self = (py_TValue){
- .type = typeobj->type,
- .is_ptr = true,
- ._obj = typeobj,
- };
- self->module = module;
- c11_vector__ctor(&self->annotated_fields, sizeof(py_Name));
- }
- static void py_TypeInfo__dtor(py_TypeInfo* self) { c11_vector__dtor(&self->annotated_fields); }
- void VM__ctor(VM* self) {
- self->top_frame = NULL;
- ModuleDict__ctor(&self->modules, NULL, *py_NIL);
- c11_vector__ctor(&self->types, sizeof(py_TypeInfo));
- self->builtins = *py_NIL;
- self->main = *py_NIL;
- self->ceval_on_step = NULL;
- self->import_file = pk_default_import_file;
- self->print = pk_default_print;
- self->last_retval = *py_NIL;
- self->curr_exception = *py_NIL;
- self->is_curr_exc_handled = false;
- self->is_stopiteration = false;
- self->__curr_class = NULL;
- ManagedHeap__ctor(&self->heap, self);
- ValueStack__ctor(&self->stack);
- /* Init Builtin Types */
- // 0: unused
- void* placeholder = c11_vector__emplace(&self->types);
- memset(placeholder, 0, sizeof(py_TypeInfo));
- #define validate(t, expr) \
- if(t != (expr)) abort()
- validate(tp_object, pk_newtype("object", 0, NULL, NULL, true, false));
- validate(tp_type, pk_newtype("type", 1, NULL, NULL, false, true));
- pk_object__register();
- validate(tp_int, pk_newtype("int", tp_object, NULL, NULL, false, true));
- validate(tp_float, pk_newtype("float", tp_object, NULL, NULL, false, true));
- validate(tp_bool, pk_newtype("bool", tp_object, NULL, NULL, false, true));
- pk_number__register();
- validate(tp_str, pk_str__register());
- validate(tp_str_iterator, pk_str_iterator__register());
- validate(tp_list, pk_list__register());
- validate(tp_tuple, pk_tuple__register());
- validate(tp_array_iterator, pk_array_iterator__register());
- validate(tp_slice, pk_slice__register());
- validate(tp_range, pk_range__register());
- validate(tp_range_iterator, pk_range_iterator__register());
- validate(tp_module, pk_newtype("module", tp_object, NULL, NULL, false, true));
- validate(tp_function, pk_function__register());
- validate(tp_nativefunc, pk_nativefunc__register());
- validate(tp_boundmethod, pk_boundmethod__register());
- validate(tp_super, pk_super__register());
- validate(tp_BaseException, pk_BaseException__register());
- validate(tp_Exception, pk_Exception__register());
- validate(tp_bytes, pk_bytes__register());
- validate(tp_namedict, pk_namedict__register());
- validate(tp_locals, pk_locals__register());
- validate(tp_code, pk_code__register());
- validate(tp_dict, pk_dict__register());
- validate(tp_dict_items, pk_dict_items__register());
- validate(tp_property, pk_property__register());
- validate(tp_star_wrapper, pk_newtype("star_wrapper", tp_object, NULL, NULL, false, true));
- validate(tp_staticmethod, pk_staticmethod__register());
- validate(tp_classmethod, pk_classmethod__register());
- validate(tp_NoneType, pk_newtype("NoneType", tp_object, NULL, NULL, false, true));
- validate(tp_NotImplementedType,
- pk_newtype("NotImplementedType", tp_object, NULL, NULL, false, true));
- validate(tp_ellipsis, pk_newtype("ellipsis", tp_object, NULL, NULL, false, true));
- validate(tp_generator, pk_generator__register());
- self->builtins = pk_builtins__register();
- // inject some builtin expections
- #define INJECT_BUILTIN_EXC(name, TBase) \
- do { \
- py_Type type = pk_newtype(#name, TBase, &self->builtins, NULL, false, true); \
- py_setdict(&self->builtins, py_name(#name), py_tpobject(type)); \
- validate(tp_##name, type); \
- } while(0)
- INJECT_BUILTIN_EXC(SystemExit, tp_BaseException);
- INJECT_BUILTIN_EXC(KeyboardInterrupt, tp_BaseException);
- INJECT_BUILTIN_EXC(StopIteration, tp_Exception);
- INJECT_BUILTIN_EXC(SyntaxError, tp_Exception);
- INJECT_BUILTIN_EXC(StackOverflowError, tp_Exception);
- INJECT_BUILTIN_EXC(IOError, tp_Exception);
- INJECT_BUILTIN_EXC(OSError, tp_Exception);
- INJECT_BUILTIN_EXC(NotImplementedError, tp_Exception);
- INJECT_BUILTIN_EXC(TypeError, tp_Exception);
- INJECT_BUILTIN_EXC(IndexError, tp_Exception);
- INJECT_BUILTIN_EXC(ValueError, tp_Exception);
- INJECT_BUILTIN_EXC(RuntimeError, tp_Exception);
- INJECT_BUILTIN_EXC(ZeroDivisionError, tp_Exception);
- INJECT_BUILTIN_EXC(NameError, tp_Exception);
- INJECT_BUILTIN_EXC(UnboundLocalError, tp_Exception);
- INJECT_BUILTIN_EXC(AttributeError, tp_Exception);
- INJECT_BUILTIN_EXC(ImportError, tp_Exception);
- INJECT_BUILTIN_EXC(AssertionError, tp_Exception);
- INJECT_BUILTIN_EXC(KeyError, tp_Exception);
- #undef INJECT_BUILTIN_EXC
- #undef validate
- /* Setup Public Builtin Types */
- py_Type public_types[] = {
- tp_object,
- tp_type,
- tp_int,
- tp_float,
- tp_bool,
- tp_str,
- tp_list,
- tp_tuple,
- tp_slice,
- tp_range,
- tp_bytes,
- tp_dict,
- tp_property,
- tp_staticmethod,
- tp_classmethod,
- tp_super,
- tp_BaseException,
- tp_Exception,
- };
- for(int i = 0; i < c11__count_array(public_types); i++) {
- py_TypeInfo* ti = c11__at(py_TypeInfo, &self->types, public_types[i]);
- py_setdict(&self->builtins, ti->name, &ti->self);
- }
- py_newnotimplemented(py_emplacedict(&self->builtins, py_name("NotImplemented")));
- // add modules
- pk__add_module_pkpy();
- pk__add_module_os();
- pk__add_module_sys();
- pk__add_module_math();
- pk__add_module_dis();
- pk__add_module_random();
- pk__add_module_json();
- pk__add_module_gc();
- pk__add_module_time();
- pk__add_module_easing();
- pk__add_module_traceback();
- // add python builtins
- do {
- bool ok;
- ok = py_exec(kPythonLibs_builtins, "<builtins>", EXEC_MODE, &self->builtins);
- if(!ok) goto __ABORT;
- break;
- __ABORT:
- py_printexc();
- c11__abort("failed to load python builtins!");
- } while(0);
- self->main = *py_newmodule("__main__");
- }
- void VM__dtor(VM* self) {
- // destroy all objects
- ManagedHeap__dtor(&self->heap);
- // clear frames
- while(self->top_frame)
- VM__pop_frame(self);
- ModuleDict__dtor(&self->modules);
- c11__foreach(py_TypeInfo, &self->types, ti) py_TypeInfo__dtor(ti);
- c11_vector__dtor(&self->types);
- ValueStack__clear(&self->stack);
- }
- void VM__push_frame(VM* self, Frame* frame) {
- frame->f_back = self->top_frame;
- self->top_frame = frame;
- }
- void VM__pop_frame(VM* self) {
- assert(self->top_frame);
- Frame* frame = self->top_frame;
- // reset stack pointer
- self->stack.sp = frame->p0;
- // pop frame and delete
- self->top_frame = frame->f_back;
- Frame__delete(frame);
- }
- static void _clip_int(int* value, int min, int max) {
- if(*value < min) *value = min;
- if(*value > max) *value = max;
- }
- bool pk__parse_int_slice(py_Ref slice, int length, int* start, int* stop, int* step) {
- py_Ref s_start = py_getslot(slice, 0);
- py_Ref s_stop = py_getslot(slice, 1);
- py_Ref s_step = py_getslot(slice, 2);
- if(py_isnone(s_step))
- *step = 1;
- else {
- if(!py_checkint(s_step)) return false;
- *step = py_toint(s_step);
- }
- if(*step == 0) return ValueError("slice step cannot be zero");
- if(*step > 0) {
- if(py_isnone(s_start))
- *start = 0;
- else {
- if(!py_checkint(s_start)) return false;
- *start = py_toint(s_start);
- if(*start < 0) *start += length;
- _clip_int(start, 0, length);
- }
- if(py_isnone(s_stop))
- *stop = length;
- else {
- if(!py_checkint(s_stop)) return false;
- *stop = py_toint(s_stop);
- if(*stop < 0) *stop += length;
- _clip_int(stop, 0, length);
- }
- } else {
- if(py_isnone(s_start))
- *start = length - 1;
- else {
- if(!py_checkint(s_start)) return false;
- *start = py_toint(s_start);
- if(*start < 0) *start += length;
- _clip_int(start, -1, length - 1);
- }
- if(py_isnone(s_stop))
- *stop = -1;
- else {
- if(!py_checkint(s_stop)) return false;
- *stop = py_toint(s_stop);
- if(*stop < 0) *stop += length;
- _clip_int(stop, -1, length - 1);
- }
- }
- return true;
- }
- bool pk__normalize_index(int* index, int length) {
- if(*index < 0) *index += length;
- if(*index < 0 || *index >= length) { return IndexError("%d not in [0, %d)", *index, length); }
- return true;
- }
- py_Type pk_newtype(const char* name,
- py_Type base,
- const py_GlobalRef module,
- void (*dtor)(void*),
- bool is_python,
- bool is_sealed) {
- c11_vector* types = &pk_current_vm->types;
- py_Type index = types->length;
- py_TypeInfo* ti = c11_vector__emplace(types);
- py_TypeInfo* base_ti = base ? c11__at(py_TypeInfo, types, base) : NULL;
- if(base_ti && base_ti->is_sealed) {
- c11__abort("type '%s' is not an acceptable base type", py_name2str(base_ti->name));
- }
- py_TypeInfo__ctor(ti, py_name(name), index, base, module ? *module : *py_NIL);
- if(!dtor && base) dtor = base_ti->dtor;
- ti->dtor = dtor;
- ti->is_python = is_python;
- ti->is_sealed = is_sealed;
- return index;
- }
- py_Type py_newtype(const char* name, py_Type base, const py_GlobalRef module, void (*dtor)(void*)) {
- py_Type type = pk_newtype(name, base, module, dtor, false, false);
- if(module) py_setdict(module, py_name(name), py_tpobject(type));
- return type;
- }
- static bool
- prepare_py_call(py_TValue* buffer, py_Ref argv, py_Ref p1, int kwargc, const FuncDecl* decl) {
- const CodeObject* co = &decl->code;
- int decl_argc = decl->args.length;
- if(p1 - argv < decl_argc) {
- return TypeError("%s() takes %d positional arguments but %d were given",
- co->name->data,
- decl_argc,
- p1 - argv);
- }
- py_TValue* t = argv;
- // prepare args
- memset(buffer, 0, co->nlocals * sizeof(py_TValue));
- c11__foreach(int, &decl->args, index) buffer[*index] = *t++;
- // prepare kwdefaults
- c11__foreach(FuncDeclKwArg, &decl->kwargs, kv) buffer[kv->index] = kv->value;
- // handle *args
- if(decl->starred_arg != -1) {
- int exceed_argc = p1 - t;
- py_Ref vargs = &buffer[decl->starred_arg];
- py_newtuple(vargs, exceed_argc);
- for(int j = 0; j < exceed_argc; j++) {
- py_tuple_setitem(vargs, j, t++);
- }
- } else {
- // kwdefaults override
- // def f(a, b, c=None)
- // f(1, 2, 3) -> c=3
- c11__foreach(FuncDeclKwArg, &decl->kwargs, kv) {
- if(t >= p1) break;
- buffer[kv->index] = *t++;
- }
- // not able to consume all args
- if(t < p1) return TypeError("too many arguments (%s)", co->name->data);
- }
- if(decl->starred_kwarg != -1) py_newdict(&buffer[decl->starred_kwarg]);
- for(int j = 0; j < kwargc; j++) {
- py_Name key = py_toint(&p1[2 * j]);
- int index = c11_smallmap_n2i__get(&decl->kw_to_index, key, -1);
- // if key is an explicit key, set as local variable
- if(index >= 0) {
- buffer[index] = p1[2 * j + 1];
- } else {
- // otherwise, set as **kwargs if possible
- if(decl->starred_kwarg == -1) {
- return TypeError("'%n' is an invalid keyword argument for %s()",
- key,
- co->name->data);
- } else {
- // add to **kwargs
- bool ok = py_dict_setitem_by_str(&buffer[decl->starred_kwarg],
- py_name2str(key),
- &p1[2 * j + 1]);
- if(!ok) return false;
- }
- }
- }
- return true;
- }
- FrameResult VM__vectorcall(VM* self, uint16_t argc, uint16_t kwargc, bool opcall) {
- pk_print_stack(self, self->top_frame, (Bytecode){0});
- py_Ref p1 = self->stack.sp - kwargc * 2;
- py_Ref p0 = p1 - argc - 2;
- // [callable, <self>, args..., kwargs...]
- // ^p0 ^p1 ^_sp
- // handle boundmethod, do a patch
- if(p0->type == tp_boundmethod) {
- assert(py_isnil(p0 + 1)); // self must be NULL
- py_TValue* slots = PyObject__slots(p0->_obj);
- p0[0] = slots[1]; // callable
- p0[1] = slots[0]; // self
- // [unbound, self, args..., kwargs...]
- }
- py_Ref argv = py_isnil(p0 + 1) ? p0 + 2 : p0 + 1;
- if(p0->type == tp_function) {
- /*****************_py_call*****************/
- // check stack overflow
- if(self->stack.sp > self->stack.end) {
- py_exception(tp_StackOverflowError, "");
- return RES_ERROR;
- }
- Function* fn = py_touserdata(p0);
- const CodeObject* co = &fn->decl->code;
- switch(fn->decl->type) {
- case FuncType_NORMAL: {
- bool ok = prepare_py_call(self->__vectorcall_buffer, argv, p1, kwargc, fn->decl);
- if(!ok) return RES_ERROR;
- // copy buffer back to stack
- self->stack.sp = argv + co->nlocals;
- memcpy(argv, self->__vectorcall_buffer, co->nlocals * sizeof(py_TValue));
- // submit the call
- if(!fn->cfunc) {
- VM__push_frame(self, Frame__new(co, &fn->module, p0, argv, true));
- return opcall ? RES_CALL : VM__run_top_frame(self);
- } else {
- bool ok = py_callcfunc(fn->cfunc, co->nlocals, argv);
- self->stack.sp = p0;
- return ok ? RES_RETURN : RES_ERROR;
- }
- }
- case FuncType_SIMPLE:
- if(p1 - argv != fn->decl->args.length) {
- const char* fmt = "%s() takes %d positional arguments but %d were given";
- TypeError(fmt, co->name->data, fn->decl->args.length, p1 - argv);
- return RES_ERROR;
- }
- if(kwargc) {
- TypeError("%s() takes no keyword arguments", co->name->data);
- return RES_ERROR;
- }
- // [callable, <self>, args..., local_vars...]
- // ^p0 ^p1 ^_sp
- self->stack.sp = argv + co->nlocals;
- // initialize local variables to py_NIL
- memset(p1, 0, (char*)self->stack.sp - (char*)p1);
- // submit the call
- VM__push_frame(self, Frame__new(co, &fn->module, p0, argv, true));
- return opcall ? RES_CALL : VM__run_top_frame(self);
- case FuncType_GENERATOR: {
- bool ok = prepare_py_call(self->__vectorcall_buffer, argv, p1, kwargc, fn->decl);
- if(!ok) return RES_ERROR;
- Frame* frame = Frame__new(co, &fn->module, p0, argv, false);
- pk_newgenerator(py_retval(), frame, self->__vectorcall_buffer, co->nlocals);
- self->stack.sp = p0;
- return RES_RETURN;
- }
- default: c11__unreachedable();
- };
- c11__unreachedable();
- /*****************_py_call*****************/
- }
- if(p0->type == tp_nativefunc) {
- bool ok = py_callcfunc(p0->_cfunc, p1 - argv, argv);
- self->stack.sp = p0;
- return ok ? RES_RETURN : RES_ERROR;
- }
- if(p0->type == tp_type) {
- // [cls, NULL, args..., kwargs...]
- py_Ref new_f = py_tpfindmagic(py_totype(p0), __new__);
- assert(new_f && py_isnil(p0 + 1));
- // prepare a copy of args and kwargs
- int span = self->stack.sp - argv;
- *self->stack.sp++ = *new_f; // push __new__
- *self->stack.sp++ = *p0; // push cls
- memcpy(self->stack.sp, argv, span * sizeof(py_TValue));
- self->stack.sp += span;
- // [new_f, cls, args..., kwargs...]
- if(VM__vectorcall(self, argc, kwargc, false) == RES_ERROR) return RES_ERROR;
- // by recursively using vectorcall, args and kwargs are consumed
- // try __init__
- // NOTE: previously we use `get_unbound_method` but here we just use `tpfindmagic`
- // >> [cls, NULL, args..., kwargs...]
- // >> py_retval() is the new instance
- py_Ref init_f = py_tpfindmagic(py_totype(p0), __init__);
- if(init_f) {
- // do an inplace patch
- *p0 = *init_f; // __init__
- p0[1] = self->last_retval; // self
- // [__init__, self, args..., kwargs...]
- if(VM__vectorcall(self, argc, kwargc, false) == RES_ERROR) return RES_ERROR;
- *py_retval() = p0[1]; // restore the new instance
- }
- // reset the stack
- self->stack.sp = p0;
- return RES_RETURN;
- }
- // handle `__call__` overload
- if(pk_loadmethod(p0, __call__)) {
- // [__call__, self, args..., kwargs...]
- return VM__vectorcall(self, argc, kwargc, opcall);
- }
- TypeError("'%t' object is not callable", p0->type);
- return RES_ERROR;
- }
- /****************************************/
- void PyObject__delete(PyObject* self) {
- py_TypeInfo* ti = c11__at(py_TypeInfo, &pk_current_vm->types, self->type);
- if(ti->dtor) ti->dtor(PyObject__userdata(self));
- if(self->slots == -1) NameDict__dtor(PyObject__dict(self));
- if(self->gc_is_large) {
- free(self);
- } else {
- PoolObject_dealloc(self);
- }
- }
- static void mark_object(PyObject* obj);
- void pk__mark_value(py_TValue* val) {
- if(val->is_ptr) mark_object(val->_obj);
- }
- void pk__mark_namedict(NameDict* dict) {
- for(int i = 0; i < dict->length; i++) {
- NameDict_KV* kv = c11__at(NameDict_KV, dict, i);
- pk__mark_value(&kv->value);
- }
- }
- void pk__tp_set_marker(py_Type type, void (*gc_mark)(void*)) {
- py_TypeInfo* ti = c11__at(py_TypeInfo, &pk_current_vm->types, type);
- assert(ti->gc_mark == NULL);
- ti->gc_mark = gc_mark;
- }
- static void mark_object(PyObject* obj) {
- if(obj->gc_marked) return;
- obj->gc_marked = true;
- if(obj->slots > 0) {
- py_TValue* p = PyObject__slots(obj);
- for(int i = 0; i < obj->slots; i++)
- pk__mark_value(p + i);
- } else if(obj->slots == -1) {
- NameDict* dict = PyObject__dict(obj);
- pk__mark_namedict(dict);
- }
- py_TypeInfo* types = c11__at(py_TypeInfo, &pk_current_vm->types, obj->type);
- if(types->gc_mark) types->gc_mark(PyObject__userdata(obj));
- }
- void CodeObject__gc_mark(const CodeObject* self) {
- c11__foreach(py_TValue, &self->consts, i) { pk__mark_value(i); }
- c11__foreach(FuncDecl_, &self->func_decls, i) { CodeObject__gc_mark(&(*i)->code); }
- }
- void ManagedHeap__mark(ManagedHeap* self) {
- VM* vm = self->vm;
- // mark heap objects
- for(int i = 0; i < self->no_gc.length; i++) {
- PyObject* obj = c11__getitem(PyObject*, &self->no_gc, i);
- mark_object(obj);
- }
- // mark value stack
- for(py_TValue* p = vm->stack.begin; p != vm->stack.end; p++) {
- pk__mark_value(p);
- }
- // mark magic slots
- py_TypeInfo* types = vm->types.data;
- int types_length = vm->types.length;
- // 0-th type is placeholder
- for(int i = 1; i < types_length; i++) {
- for(int j = 0; j <= __missing__; j++) {
- py_TValue* slot = types[i].magic + j;
- if(py_isnil(slot)) continue;
- pk__mark_value(slot);
- }
- }
- // mark frame
- for(Frame* frame = vm->top_frame; frame; frame = frame->f_back) {
- Frame__gc_mark(frame);
- }
- // mark vm's registers
- pk__mark_value(&vm->last_retval);
- pk__mark_value(&vm->curr_exception);
- for(int i = 0; i < c11__count_array(vm->reg); i++) {
- pk__mark_value(&vm->reg[i]);
- }
- }
- void pk_print_stack(VM* self, Frame* frame, Bytecode byte) {
- return;
- if(frame == NULL || py_isnil(&self->main)) return;
- py_TValue* sp = self->stack.sp;
- c11_sbuf buf;
- c11_sbuf__ctor(&buf);
- for(py_Ref p = self->stack.begin; p != sp; p++) {
- switch(p->type) {
- case 0: c11_sbuf__write_cstr(&buf, "nil"); break;
- case tp_int: c11_sbuf__write_i64(&buf, p->_i64); break;
- case tp_float: c11_sbuf__write_f64(&buf, p->_f64, -1); break;
- case tp_bool: c11_sbuf__write_cstr(&buf, p->_bool ? "True" : "False"); break;
- case tp_NoneType: c11_sbuf__write_cstr(&buf, "None"); break;
- case tp_list: {
- pk_sprintf(&buf, "list(%d)", py_list_len(p));
- break;
- }
- case tp_tuple: {
- pk_sprintf(&buf, "tuple(%d)", py_tuple_len(p));
- break;
- }
- case tp_function: {
- Function* ud = py_touserdata(p);
- c11_sbuf__write_cstr(&buf, ud->decl->code.name->data);
- c11_sbuf__write_cstr(&buf, "()");
- break;
- }
- case tp_type: {
- pk_sprintf(&buf, "<class '%t'>", py_totype(p));
- break;
- }
- case tp_str: {
- pk_sprintf(&buf, "%q", py_tosv(p));
- break;
- }
- case tp_module: {
- py_Ref path = py_getdict(p, __path__);
- pk_sprintf(&buf, "<module '%v'>", py_tosv(path));
- break;
- }
- default: {
- pk_sprintf(&buf, "(%t)", p->type);
- break;
- }
- }
- if(p != &sp[-1]) c11_sbuf__write_cstr(&buf, ", ");
- }
- c11_string* stack_str = c11_sbuf__submit(&buf);
- printf("%s:%-3d: %-25s %-6d [%s]\n",
- frame->co->src->filename->data,
- Frame__lineno(frame),
- pk_opname(byte.op),
- byte.arg,
- stack_str->data);
- c11_string__delete(stack_str);
- }
- bool pk_wrapper__self(int argc, py_Ref argv) {
- PY_CHECK_ARGC(1);
- py_assign(py_retval(), argv);
- return true;
- }
- bool pk_wrapper__NotImplementedError(int argc, py_Ref argv) {
- return py_exception(tp_NotImplementedError, "");
- }
|