pocketpy.cpp 64 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768
  1. #include "pocketpy/pocketpy.h"
  2. #ifdef PK_USE_BOX2D
  3. #include "box2dw.hpp"
  4. #endif
  5. #if defined (_WIN32) && PK_SUPPORT_DYLIB == 1
  6. #define WIN32_LEAN_AND_MEAN
  7. #include <windows.h>
  8. #endif
  9. namespace pkpy{
  10. using dylib_entry_t = const char* (*)(void*, const char*);
  11. #if PK_ENABLE_OS
  12. #if PK_SUPPORT_DYLIB == 1
  13. // win32
  14. static dylib_entry_t load_dylib(const char* path){
  15. std::error_code ec;
  16. auto p = std::filesystem::absolute(path, ec);
  17. if(ec) return nullptr;
  18. HMODULE handle = LoadLibraryA(p.string().c_str());
  19. if(!handle){
  20. DWORD errorCode = GetLastError();
  21. // Convert the error code to text
  22. LPSTR errorMessage = nullptr;
  23. FormatMessageA(
  24. FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
  25. nullptr,
  26. errorCode,
  27. MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US),
  28. (LPSTR)&errorMessage,
  29. 0,
  30. nullptr
  31. );
  32. printf("%lu: %s\n", errorCode, errorMessage);
  33. LocalFree(errorMessage);
  34. return nullptr;
  35. }
  36. return (dylib_entry_t)GetProcAddress(handle, "pkpy_module__init__");
  37. }
  38. #elif PK_SUPPORT_DYLIB == 2
  39. // linux/darwin
  40. static dylib_entry_t load_dylib(const char* path){
  41. std::error_code ec;
  42. auto p = std::filesystem::absolute(path, ec);
  43. if(ec) return nullptr;
  44. void* handle = dlopen(p.c_str(), RTLD_LAZY);
  45. if(!handle) return nullptr;
  46. return (dylib_entry_t)dlsym(handle, "pkpy_module__init__");
  47. }
  48. #elif PK_SUPPORT_DYLIB == 3
  49. // android
  50. static dylib_entry_t load_dylib(const char* path){
  51. void* handle = dlopen(path, RTLD_LAZY);
  52. if(!handle) return nullptr;
  53. return (dylib_entry_t)dlsym(handle, "pkpy_module__init__");
  54. }
  55. #else
  56. static dylib_entry_t load_dylib(const char* path){
  57. return nullptr;
  58. }
  59. #endif
  60. #else
  61. static dylib_entry_t load_dylib(const char* path){
  62. return nullptr;
  63. }
  64. #endif
  65. void init_builtins(VM* _vm) {
  66. #define BIND_NUM_ARITH_OPT(name, op) \
  67. _vm->bind##name(_vm->tp_int, [](VM* vm, PyObject* lhs, PyObject* rhs) { \
  68. if(is_int(rhs)) return VAR(_CAST(i64, lhs) op _CAST(i64, rhs)); \
  69. if(is_float(rhs)) return VAR(_CAST(i64, lhs) op _CAST(f64, rhs)); \
  70. return vm->NotImplemented; \
  71. }); \
  72. _vm->bind##name(_vm->tp_float, [](VM* vm, PyObject* lhs, PyObject* rhs) { \
  73. if(is_float(rhs)) return VAR(_CAST(f64, lhs) op _CAST(f64, rhs)); \
  74. if(is_int(rhs)) return VAR(_CAST(f64, lhs) op _CAST(i64, rhs)); \
  75. return vm->NotImplemented; \
  76. });
  77. BIND_NUM_ARITH_OPT(__add__, +)
  78. BIND_NUM_ARITH_OPT(__sub__, -)
  79. BIND_NUM_ARITH_OPT(__mul__, *)
  80. #undef BIND_NUM_ARITH_OPT
  81. #define BIND_NUM_LOGICAL_OPT(name, op) \
  82. _vm->bind##name(_vm->tp_int, [](VM* vm, PyObject* lhs, PyObject* rhs) { \
  83. i64 val; \
  84. if(try_cast_int(rhs, &val)) return VAR(_CAST(i64, lhs) op val); \
  85. if(is_float(rhs)) return VAR(_CAST(i64, lhs) op _CAST(f64, rhs)); \
  86. return vm->NotImplemented; \
  87. }); \
  88. _vm->bind##name(_vm->tp_float, [](VM* vm, PyObject* lhs, PyObject* rhs) { \
  89. i64 val; \
  90. if(try_cast_int(rhs, &val)) return VAR(_CAST(f64, lhs) op val); \
  91. if(is_float(rhs)) return VAR(_CAST(f64, lhs) op _CAST(f64, rhs)); \
  92. return vm->NotImplemented; \
  93. });
  94. BIND_NUM_LOGICAL_OPT(__eq__, ==)
  95. BIND_NUM_LOGICAL_OPT(__lt__, <)
  96. BIND_NUM_LOGICAL_OPT(__le__, <=)
  97. BIND_NUM_LOGICAL_OPT(__gt__, >)
  98. BIND_NUM_LOGICAL_OPT(__ge__, >=)
  99. #undef BIND_NUM_ARITH_OPT
  100. #undef BIND_NUM_LOGICAL_OPT
  101. _vm->bind_builtin_func<-1>("super", [](VM* vm, ArgsView args) {
  102. PyObject* class_arg = nullptr;
  103. PyObject* self_arg = nullptr;
  104. if(args.size() == 2){
  105. class_arg = args[0];
  106. self_arg = args[1];
  107. }else if(args.size() == 0){
  108. FrameId frame = vm->top_frame();
  109. if(frame->_callable != nullptr){
  110. class_arg = frame->_callable->attr().try_get(__class__);
  111. if(frame->_locals.size() > 0) self_arg = frame->_locals[0];
  112. }
  113. if(class_arg == nullptr || self_arg == nullptr){
  114. vm->TypeError("super(): unable to determine the class context, use super(class, self) instead");
  115. }
  116. }else{
  117. vm->TypeError("super() takes 0 or 2 arguments");
  118. }
  119. vm->check_non_tagged_type(class_arg, vm->tp_type);
  120. Type type = PK_OBJ_GET(Type, class_arg);
  121. if(!vm->isinstance(self_arg, type)){
  122. Str _0 = obj_type_name(vm, PK_OBJ_GET(Type, vm->_t(self_arg)));
  123. Str _1 = obj_type_name(vm, type);
  124. vm->TypeError("super(): " + _0.escape() + " is not an instance of " + _1.escape());
  125. }
  126. return vm->heap.gcnew<Super>(vm->tp_super, self_arg, vm->_all_types[type].base);
  127. });
  128. _vm->bind_builtin_func<2>("isinstance", [](VM* vm, ArgsView args) {
  129. if(is_non_tagged_type(args[1], vm->tp_tuple)){
  130. Tuple& types = _CAST(Tuple&, args[1]);
  131. for(PyObject* type : types){
  132. vm->check_non_tagged_type(type, vm->tp_type);
  133. if(vm->isinstance(args[0], PK_OBJ_GET(Type, type))) return vm->True;
  134. }
  135. return vm->False;
  136. }
  137. vm->check_non_tagged_type(args[1], vm->tp_type);
  138. Type type = PK_OBJ_GET(Type, args[1]);
  139. return VAR(vm->isinstance(args[0], type));
  140. });
  141. _vm->bind_builtin_func<0>("globals", [](VM* vm, ArgsView args) {
  142. PyObject* mod = vm->top_frame()->_module;
  143. return VAR(MappingProxy(mod));
  144. });
  145. _vm->bind(_vm->builtins, "round(x, ndigits=0)", [](VM* vm, ArgsView args) {
  146. f64 x = CAST(f64, args[0]);
  147. int ndigits = CAST(int, args[1]);
  148. if(ndigits == 0){
  149. return x >= 0 ? VAR((i64)(x + 0.5)) : VAR((i64)(x - 0.5));
  150. }
  151. if(ndigits < 0) vm->ValueError("ndigits should be non-negative");
  152. if(x >= 0){
  153. return VAR((i64)(x * std::pow(10, ndigits) + 0.5) / std::pow(10, ndigits));
  154. }else{
  155. return VAR((i64)(x * std::pow(10, ndigits) - 0.5) / std::pow(10, ndigits));
  156. }
  157. });
  158. _vm->bind_builtin_func<1>("abs", [](VM* vm, ArgsView args) {
  159. if(is_int(args[0])) return VAR(std::abs(_CAST(i64, args[0])));
  160. if(is_float(args[0])) return VAR(std::abs(_CAST(f64, args[0])));
  161. vm->TypeError("bad operand type for abs()");
  162. return vm->None;
  163. });
  164. _vm->bind_builtin_func<1>("id", [](VM* vm, ArgsView args) {
  165. PyObject* obj = args[0];
  166. if(is_tagged(obj)) return vm->None;
  167. return VAR(PK_BITS(obj));
  168. });
  169. _vm->bind_builtin_func<1>("staticmethod", [](VM* vm, ArgsView args) {
  170. return args[0];
  171. });
  172. _vm->bind_builtin_func<1>("callable", [](VM* vm, ArgsView args) {
  173. PyObject* cls = vm->_t(args[0]);
  174. Type t = PK_OBJ_GET(Type, cls);
  175. if(t == vm->tp_function) return vm->True;
  176. if(t == vm->tp_native_func) return vm->True;
  177. if(t == vm->tp_bound_method) return vm->True;
  178. if(t == vm->tp_type) return vm->True;
  179. bool ok = vm->find_name_in_mro(cls, __call__) != nullptr;
  180. return VAR(ok);
  181. });
  182. _vm->bind_builtin_func<1>("__import__", [](VM* vm, ArgsView args) {
  183. const Str& name = CAST(Str&, args[0]);
  184. auto dot = name.sv().find_last_of(".");
  185. if(dot != std::string_view::npos){
  186. auto ext = name.sv().substr(dot);
  187. if(ext == ".so" || ext == ".dll" || ext == ".dylib"){
  188. dylib_entry_t entry = load_dylib(name.c_str());
  189. if(!entry){
  190. vm->ImportError("cannot load dynamic library: " + name.escape());
  191. }
  192. vm->_c.s_view.push(ArgsView(vm->s_data.end(), vm->s_data.end()));
  193. const char* name = entry(vm, PK_VERSION);
  194. vm->_c.s_view.pop();
  195. if(name == nullptr){
  196. vm->ImportError("module initialization failed: " + Str(name).escape());
  197. }
  198. return vm->_modules[name];
  199. }
  200. }
  201. return vm->py_import(name);
  202. });
  203. _vm->bind_builtin_func<2>("divmod", [](VM* vm, ArgsView args) {
  204. if(is_int(args[0])){
  205. i64 lhs = _CAST(i64, args[0]);
  206. i64 rhs = CAST(i64, args[1]);
  207. if(rhs == 0) vm->ZeroDivisionError();
  208. auto res = std::div(lhs, rhs);
  209. return VAR(Tuple({VAR(res.quot), VAR(res.rem)}));
  210. }else{
  211. return vm->call_method(args[0], __divmod__, args[1]);
  212. }
  213. });
  214. _vm->bind_builtin_func<1>("eval", [](VM* vm, ArgsView args) {
  215. CodeObject_ code = vm->compile(CAST(Str&, args[0]), "<eval>", EVAL_MODE, true);
  216. FrameId frame = vm->top_frame();
  217. return vm->_exec(code.get(), frame->_module, frame->_callable, frame->_locals);
  218. });
  219. _vm->bind_builtin_func<1>("exec", [](VM* vm, ArgsView args) {
  220. CodeObject_ code = vm->compile(CAST(Str&, args[0]), "<exec>", EXEC_MODE, true);
  221. FrameId frame = vm->top_frame();
  222. vm->_exec(code.get(), frame->_module, frame->_callable, frame->_locals);
  223. return vm->None;
  224. });
  225. _vm->bind_builtin_func<-1>("exit", [](VM* vm, ArgsView args) {
  226. if(args.size() == 0) std::exit(0);
  227. else if(args.size() == 1) std::exit(CAST(int, args[0]));
  228. else vm->TypeError("exit() takes at most 1 argument");
  229. return vm->None;
  230. });
  231. _vm->bind_builtin_func<1>("repr", PK_LAMBDA(vm->py_repr(args[0])));
  232. _vm->bind_builtin_func<1>("len", [](VM* vm, ArgsView args){
  233. const PyTypeInfo* ti = vm->_inst_type_info(args[0]);
  234. if(ti->m__len__) return VAR(ti->m__len__(vm, args[0]));
  235. return vm->call_method(args[0], __len__);
  236. });
  237. _vm->bind_builtin_func<1>("hash", [](VM* vm, ArgsView args){
  238. i64 value = vm->py_hash(args[0]);
  239. return VAR(value);
  240. });
  241. _vm->bind_builtin_func<1>("chr", [](VM* vm, ArgsView args) {
  242. i64 i = CAST(i64, args[0]);
  243. if (i < 0 || i > 128) vm->ValueError("chr() arg not in range(128)");
  244. return VAR(std::string(1, (char)i));
  245. });
  246. _vm->bind_builtin_func<1>("ord", [](VM* vm, ArgsView args) {
  247. const Str& s = CAST(Str&, args[0]);
  248. if (s.length()!=1) vm->TypeError("ord() expected an ASCII character");
  249. return VAR((i64)(s[0]));
  250. });
  251. _vm->bind_builtin_func<2>("hasattr", [](VM* vm, ArgsView args) {
  252. return VAR(vm->getattr(args[0], CAST(Str&, args[1]), false) != nullptr);
  253. });
  254. _vm->bind_builtin_func<3>("setattr", [](VM* vm, ArgsView args) {
  255. vm->setattr(args[0], CAST(Str&, args[1]), args[2]);
  256. return vm->None;
  257. });
  258. _vm->bind_builtin_func<2>("getattr", [](VM* vm, ArgsView args) {
  259. const Str& name = CAST(Str&, args[1]);
  260. return vm->getattr(args[0], name);
  261. });
  262. _vm->bind_builtin_func<1>("hex", [](VM* vm, ArgsView args) {
  263. std::stringstream ss;
  264. ss << std::hex << CAST(i64, args[0]);
  265. return VAR("0x" + ss.str());
  266. });
  267. _vm->bind_builtin_func<1>("iter", [](VM* vm, ArgsView args) {
  268. return vm->py_iter(args[0]);
  269. });
  270. _vm->bind_builtin_func<1>("next", [](VM* vm, ArgsView args) {
  271. return vm->py_next(args[0]);
  272. });
  273. _vm->bind_builtin_func<1>("bin", [](VM* vm, ArgsView args) {
  274. std::stringstream ss;
  275. i64 x = CAST(i64, args[0]);
  276. if(x < 0){ ss << "-"; x = -x; }
  277. ss << "0b";
  278. std::string bits;
  279. while(x){
  280. bits += (x & 1) ? '1' : '0';
  281. x >>= 1;
  282. }
  283. std::reverse(bits.begin(), bits.end());
  284. if(bits.empty()) bits = "0";
  285. ss << bits;
  286. return VAR(ss.str());
  287. });
  288. _vm->bind_builtin_func<1>("dir", [](VM* vm, ArgsView args) {
  289. std::set<StrName> names;
  290. if(!is_tagged(args[0]) && args[0]->is_attr_valid()){
  291. std::vector<StrName> keys = args[0]->attr().keys();
  292. names.insert(keys.begin(), keys.end());
  293. }
  294. const NameDict& t_attr = vm->_t(args[0])->attr();
  295. std::vector<StrName> keys = t_attr.keys();
  296. names.insert(keys.begin(), keys.end());
  297. List ret;
  298. for (StrName name : names) ret.push_back(VAR(name.sv()));
  299. return VAR(std::move(ret));
  300. });
  301. _vm->bind__repr__(_vm->tp_object, [](VM* vm, PyObject* obj) {
  302. if(is_tagged(obj)) FATAL_ERROR();
  303. std::stringstream ss;
  304. ss << "<" << OBJ_NAME(vm->_t(obj)) << " object at 0x";
  305. ss << std::hex << reinterpret_cast<intptr_t>(obj) << ">";
  306. return VAR(ss.str());
  307. });
  308. _vm->bind__eq__(_vm->tp_object, [](VM* vm, PyObject* lhs, PyObject* rhs) { return VAR(lhs == rhs); });
  309. _vm->cached_object__new__ = _vm->bind_constructor<1>("object", [](VM* vm, ArgsView args) {
  310. vm->check_non_tagged_type(args[0], vm->tp_type);
  311. Type t = PK_OBJ_GET(Type, args[0]);
  312. return vm->heap.gcnew<DummyInstance>(t);
  313. });
  314. _vm->bind_method<0>("object", "_enable_instance_dict", [](VM* vm, ArgsView args){
  315. PyObject* self = args[0];
  316. if(is_tagged(self)){
  317. vm->TypeError("object: tagged object cannot enable instance dict");
  318. }
  319. if(self->is_attr_valid()){
  320. vm->TypeError("object: instance dict is already enabled");
  321. }
  322. self->enable_instance_dict();
  323. return vm->None;
  324. });
  325. _vm->bind_constructor<2>("type", PK_LAMBDA(vm->_t(args[1])));
  326. _vm->bind_constructor<-1>("range", [](VM* vm, ArgsView args) {
  327. args._begin += 1; // skip cls
  328. Range r;
  329. switch (args.size()) {
  330. case 1: r.stop = CAST(i64, args[0]); break;
  331. case 2: r.start = CAST(i64, args[0]); r.stop = CAST(i64, args[1]); break;
  332. case 3: r.start = CAST(i64, args[0]); r.stop = CAST(i64, args[1]); r.step = CAST(i64, args[2]); break;
  333. default: vm->TypeError("expected 1-3 arguments, got " + std::to_string(args.size()));
  334. }
  335. return VAR(r);
  336. });
  337. _vm->bind__iter__(_vm->tp_range, [](VM* vm, PyObject* obj) { return VAR_T(RangeIter, PK_OBJ_GET(Range, obj)); });
  338. _vm->bind__repr__(_vm->_type("NoneType"), [](VM* vm, PyObject* obj) { return VAR("None"); });
  339. _vm->bind__json__(_vm->_type("NoneType"), [](VM* vm, PyObject* obj) { return VAR("null"); });
  340. _vm->bind__truediv__(_vm->tp_float, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  341. f64 value = CAST_F(rhs);
  342. return VAR(_CAST(f64, lhs) / value);
  343. });
  344. _vm->bind__truediv__(_vm->tp_int, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  345. f64 value = CAST_F(rhs);
  346. return VAR(_CAST(i64, lhs) / value);
  347. });
  348. auto py_number_pow = [](VM* vm, PyObject* lhs_, PyObject* rhs_) {
  349. i64 lhs, rhs;
  350. if(try_cast_int(lhs_, &lhs) && try_cast_int(rhs_, &rhs)){
  351. if(rhs < 0) {
  352. if(lhs == 0) vm->ZeroDivisionError("0.0 cannot be raised to a negative power");
  353. return VAR((f64)std::pow(lhs, rhs));
  354. }
  355. i64 ret = 1;
  356. while(rhs){
  357. if(rhs & 1) ret *= lhs;
  358. lhs *= lhs;
  359. rhs >>= 1;
  360. }
  361. return VAR(ret);
  362. }else{
  363. return VAR((f64)std::pow(CAST_F(lhs_), CAST_F(rhs_)));
  364. }
  365. };
  366. _vm->bind__pow__(_vm->tp_int, py_number_pow);
  367. _vm->bind__pow__(_vm->tp_float, py_number_pow);
  368. /************ int ************/
  369. _vm->bind_constructor<-1>("int", [](VM* vm, ArgsView args) {
  370. if(args.size() == 1+0) return VAR(0);
  371. // 1 arg
  372. if(args.size() == 1+1){
  373. if (is_type(args[1], vm->tp_float)) return VAR((i64)CAST(f64, args[1]));
  374. if (is_type(args[1], vm->tp_int)) return args[1];
  375. if (is_type(args[1], vm->tp_bool)) return VAR(_CAST(bool, args[1]) ? 1 : 0);
  376. }
  377. if(args.size() > 1+2) vm->TypeError("int() takes at most 2 arguments");
  378. // 2 args
  379. if (is_type(args[1], vm->tp_str)) {
  380. int base = 10;
  381. if(args.size() == 1+2) base = CAST(i64, args[2]);
  382. const Str& s = CAST(Str&, args[1]);
  383. i64 val;
  384. if(!parse_int(s.sv(), &val, base)){
  385. vm->ValueError("invalid literal for int(): " + s.escape());
  386. }
  387. return VAR(val);
  388. }
  389. vm->TypeError("invalid arguments for int()");
  390. return vm->None;
  391. });
  392. _vm->bind_method<0>("int", "bit_length", [](VM* vm, ArgsView args) {
  393. i64 x = _CAST(i64, args[0]);
  394. if(x < 0) x = -x;
  395. int bits = 0;
  396. while(x){ x >>= 1; bits++; }
  397. return VAR(bits);
  398. });
  399. _vm->bind__floordiv__(_vm->tp_int, [](VM* vm, PyObject* lhs_, PyObject* rhs_) {
  400. i64 rhs = CAST(i64, rhs_);
  401. if(rhs == 0) vm->ZeroDivisionError();
  402. return VAR(_CAST(i64, lhs_) / rhs);
  403. });
  404. _vm->bind__mod__(_vm->tp_int, [](VM* vm, PyObject* lhs_, PyObject* rhs_) {
  405. i64 rhs = CAST(i64, rhs_);
  406. if(rhs == 0) vm->ZeroDivisionError();
  407. return VAR(_CAST(i64, lhs_) % rhs);
  408. });
  409. _vm->bind__repr__(_vm->tp_int, [](VM* vm, PyObject* obj) { return VAR(std::to_string(_CAST(i64, obj))); });
  410. _vm->bind__json__(_vm->tp_int, [](VM* vm, PyObject* obj) { return VAR(std::to_string(_CAST(i64, obj))); });
  411. _vm->bind__neg__(_vm->tp_int, [](VM* vm, PyObject* obj) { return VAR(-_CAST(i64, obj)); });
  412. _vm->bind__hash__(_vm->tp_int, [](VM* vm, PyObject* obj) { return _CAST(i64, obj); });
  413. _vm->bind__invert__(_vm->tp_int, [](VM* vm, PyObject* obj) { return VAR(~_CAST(i64, obj)); });
  414. #define INT_BITWISE_OP(name, op) \
  415. _vm->bind##name(_vm->tp_int, [](VM* vm, PyObject* lhs, PyObject* rhs) { \
  416. return VAR(_CAST(i64, lhs) op CAST(i64, rhs)); \
  417. });
  418. INT_BITWISE_OP(__lshift__, <<)
  419. INT_BITWISE_OP(__rshift__, >>)
  420. INT_BITWISE_OP(__and__, &)
  421. INT_BITWISE_OP(__or__, |)
  422. INT_BITWISE_OP(__xor__, ^)
  423. #undef INT_BITWISE_OP
  424. /************ float ************/
  425. _vm->bind_constructor<-1>("float", [](VM* vm, ArgsView args) {
  426. if(args.size() == 1+0) return VAR(0.0);
  427. if(args.size() > 1+1) vm->TypeError("float() takes at most 1 argument");
  428. // 1 arg
  429. if (is_type(args[1], vm->tp_int)) return VAR((f64)CAST(i64, args[1]));
  430. if (is_type(args[1], vm->tp_float)) return args[1];
  431. if (is_type(args[1], vm->tp_bool)) return VAR(_CAST(bool, args[1]) ? 1.0 : 0.0);
  432. if (is_type(args[1], vm->tp_str)) {
  433. const Str& s = CAST(Str&, args[1]);
  434. if(s == "inf") return VAR(INFINITY);
  435. if(s == "-inf") return VAR(-INFINITY);
  436. double float_out;
  437. char* p_end;
  438. try{
  439. float_out = std::strtod(s.begin(), &p_end);
  440. PK_ASSERT(p_end == s.end());
  441. }catch(...){
  442. vm->ValueError("invalid literal for float(): " + s.escape());
  443. }
  444. return VAR(float_out);
  445. }
  446. vm->TypeError("invalid arguments for float()");
  447. return vm->None;
  448. });
  449. _vm->bind__hash__(_vm->tp_float, [](VM* vm, PyObject* obj) {
  450. f64 val = _CAST(f64, obj);
  451. return (i64)std::hash<f64>()(val);
  452. });
  453. _vm->bind__neg__(_vm->tp_float, [](VM* vm, PyObject* obj) { return VAR(-_CAST(f64, obj)); });
  454. _vm->bind__repr__(_vm->tp_float, [](VM* vm, PyObject* obj) {
  455. f64 val = _CAST(f64, obj);
  456. if(std::isinf(val) || std::isnan(val)) return VAR(std::to_string(val));
  457. std::stringstream ss;
  458. ss << std::setprecision(std::numeric_limits<f64>::max_digits10-1) << val;
  459. std::string s = ss.str();
  460. if(std::all_of(s.begin()+1, s.end(), isdigit)) s += ".0";
  461. return VAR(s);
  462. });
  463. _vm->bind__json__(_vm->tp_float, [](VM* vm, PyObject* obj) {
  464. f64 val = _CAST(f64, obj);
  465. if(std::isinf(val) || std::isnan(val)) vm->ValueError("cannot jsonify 'nan' or 'inf'");
  466. return VAR(std::to_string(val));
  467. });
  468. /************ str ************/
  469. _vm->bind_constructor<2>("str", PK_LAMBDA(vm->py_str(args[1])));
  470. _vm->bind__hash__(_vm->tp_str, [](VM* vm, PyObject* obj) {
  471. return (i64)_CAST(Str&, obj).hash();
  472. });
  473. _vm->bind__add__(_vm->tp_str, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  474. return VAR(_CAST(Str&, lhs) + CAST(Str&, rhs));
  475. });
  476. _vm->bind__len__(_vm->tp_str, [](VM* vm, PyObject* obj) {
  477. return (i64)_CAST(Str&, obj).u8_length();
  478. });
  479. _vm->bind__mul__(_vm->tp_str, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  480. const Str& self = _CAST(Str&, lhs);
  481. i64 n = CAST(i64, rhs);
  482. std::stringstream ss;
  483. for(i64 i = 0; i < n; i++) ss << self.sv();
  484. return VAR(ss.str());
  485. });
  486. _vm->bind_method<1>("str", "__rmul__", [](VM* vm, ArgsView args) {
  487. const Str& self = _CAST(Str&, args[0]);
  488. i64 n = CAST(i64, args[1]);
  489. std::stringstream ss;
  490. for(i64 i = 0; i < n; i++) ss << self.sv();
  491. return VAR(ss.str());
  492. });
  493. _vm->bind__contains__(_vm->tp_str, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  494. const Str& self = _CAST(Str&, lhs);
  495. return VAR(self.index(CAST(Str&, rhs)) != -1);
  496. });
  497. _vm->bind__str__(_vm->tp_str, [](VM* vm, PyObject* obj) { return obj; });
  498. _vm->bind__iter__(_vm->tp_str, [](VM* vm, PyObject* obj) { return VAR_T(StringIter, obj); });
  499. _vm->bind__repr__(_vm->tp_str, [](VM* vm, PyObject* obj) {
  500. const Str& self = _CAST(Str&, obj);
  501. return VAR(self.escape(true));
  502. });
  503. _vm->bind__json__(_vm->tp_str, [](VM* vm, PyObject* obj) {
  504. const Str& self = _CAST(Str&, obj);
  505. return VAR(self.escape(false));
  506. });
  507. #define BIND_CMP_STR(name, op) \
  508. _vm->bind##name(_vm->tp_str, [](VM* vm, PyObject* lhs, PyObject* rhs) { \
  509. if(!is_non_tagged_type(rhs, vm->tp_str)) return vm->NotImplemented; \
  510. return VAR(_CAST(Str&, lhs) op _CAST(Str&, rhs)); \
  511. });
  512. BIND_CMP_STR(__eq__, ==)
  513. BIND_CMP_STR(__lt__, <)
  514. BIND_CMP_STR(__le__, <=)
  515. BIND_CMP_STR(__gt__, >)
  516. BIND_CMP_STR(__ge__, >=)
  517. #undef BIND_CMP_STR
  518. _vm->bind__getitem__(_vm->tp_str, [](VM* vm, PyObject* obj, PyObject* index) {
  519. const Str& self = _CAST(Str&, obj);
  520. if(is_non_tagged_type(index, vm->tp_slice)){
  521. const Slice& s = _CAST(Slice&, index);
  522. int start, stop, step;
  523. vm->parse_int_slice(s, self.u8_length(), start, stop, step);
  524. return VAR(self.u8_slice(start, stop, step));
  525. }
  526. int i = CAST(int, index);
  527. i = vm->normalized_index(i, self.u8_length());
  528. return VAR(self.u8_getitem(i));
  529. });
  530. _vm->bind(_vm->_t(_vm->tp_str), "replace(self, old, new, count=-1)", [](VM* vm, ArgsView args) {
  531. const Str& self = _CAST(Str&, args[0]);
  532. const Str& old = CAST(Str&, args[1]);
  533. if(old.empty()) vm->ValueError("empty substring");
  534. const Str& new_ = CAST(Str&, args[2]);
  535. int count = CAST(int, args[3]);
  536. return VAR(self.replace(old, new_, count));
  537. });
  538. _vm->bind(_vm->_t(_vm->tp_str), "split(self, sep=' ')", [](VM* vm, ArgsView args) {
  539. const Str& self = _CAST(Str&, args[0]);
  540. std::vector<std::string_view> parts = self.split(CAST(Str&, args[1]));
  541. List ret(parts.size());
  542. for(int i=0; i<parts.size(); i++) ret[i] = VAR(Str(parts[i]));
  543. return VAR(std::move(ret));
  544. });
  545. _vm->bind(_vm->_t(_vm->tp_str), "count(self, s: str)", [](VM* vm, ArgsView args) {
  546. const Str& self = _CAST(Str&, args[0]);
  547. const Str& s = CAST(Str&, args[1]);
  548. return VAR(self.count(s));
  549. });
  550. _vm->bind_method<1>("str", "index", [](VM* vm, ArgsView args) {
  551. const Str& self = _CAST(Str&, args[0]);
  552. const Str& sub = CAST(Str&, args[1]);
  553. int index = self.index(sub);
  554. if(index == -1) vm->ValueError("substring not found");
  555. return VAR(index);
  556. });
  557. _vm->bind_method<1>("str", "find", [](VM* vm, ArgsView args) {
  558. const Str& self = _CAST(Str&, args[0]);
  559. const Str& sub = CAST(Str&, args[1]);
  560. return VAR(self.index(sub));
  561. });
  562. _vm->bind_method<1>("str", "startswith", [](VM* vm, ArgsView args) {
  563. const Str& self = _CAST(Str&, args[0]);
  564. const Str& prefix = CAST(Str&, args[1]);
  565. return VAR(self.index(prefix) == 0);
  566. });
  567. _vm->bind_method<1>("str", "endswith", [](VM* vm, ArgsView args) {
  568. const Str& self = _CAST(Str&, args[0]);
  569. const Str& suffix = CAST(Str&, args[1]);
  570. int offset = self.length() - suffix.length();
  571. if(offset < 0) return vm->False;
  572. bool ok = memcmp(self.data+offset, suffix.data, suffix.length()) == 0;
  573. return VAR(ok);
  574. });
  575. _vm->bind_method<0>("str", "encode", [](VM* vm, ArgsView args) {
  576. const Str& self = _CAST(Str&, args[0]);
  577. unsigned char* buffer = new unsigned char[self.length()];
  578. memcpy(buffer, self.data, self.length());
  579. return VAR(Bytes(buffer, self.length()));
  580. });
  581. _vm->bind_method<1>("str", "join", [](VM* vm, ArgsView args) {
  582. auto _lock = vm->heap.gc_scope_lock();
  583. const Str& self = _CAST(Str&, args[0]);
  584. FastStrStream ss;
  585. PyObject* it = vm->py_iter(args[1]); // strong ref
  586. PyObject* obj = vm->py_next(it);
  587. while(obj != vm->StopIteration){
  588. if(!ss.empty()) ss << self;
  589. ss << CAST(Str&, obj);
  590. obj = vm->py_next(it);
  591. }
  592. return VAR(ss.str());
  593. });
  594. _vm->bind_method<0>("str", "lower", [](VM* vm, ArgsView args) {
  595. const Str& self = _CAST(Str&, args[0]);
  596. return VAR(self.lower());
  597. });
  598. _vm->bind_method<0>("str", "upper", [](VM* vm, ArgsView args) {
  599. const Str& self = _CAST(Str&, args[0]);
  600. return VAR(self.upper());
  601. });
  602. /************ list ************/
  603. // list.__repr__ = lambda self: '[' + ', '.join([repr(i) for i in self]) + ']'
  604. // list.__json__ = lambda self: '[' + ', '.join([i.__json__() for i in self]) + ']'
  605. // tuple.__json__ = lambda self: '[' + ', '.join([i.__json__() for i in self]) + ']'
  606. // def __f(self):
  607. // if len(self) == 1:
  608. // return '(' + repr(self[0]) + ',)'
  609. // return '(' + ', '.join([repr(i) for i in self]) + ')'
  610. // tuple.__repr__ = __f
  611. _vm->bind__repr__(_vm->tp_list, [](VM* vm, PyObject* _0){
  612. List& iterable = _CAST(List&, _0);
  613. std::stringstream ss;
  614. ss << '[';
  615. for(int i=0; i<iterable.size(); i++){
  616. ss << CAST(Str&, vm->py_repr(iterable[i]));
  617. if(i != iterable.size()-1) ss << ", ";
  618. }
  619. ss << ']';
  620. return VAR(ss.str());
  621. });
  622. _vm->bind__json__(_vm->tp_list, [](VM* vm, PyObject* _0){
  623. List& iterable = _CAST(List&, _0);
  624. std::stringstream ss;
  625. ss << '[';
  626. for(int i=0; i<iterable.size(); i++){
  627. ss << CAST(Str&, vm->py_json(iterable[i]));
  628. if(i != iterable.size()-1) ss << ", ";
  629. }
  630. ss << ']';
  631. return VAR(ss.str());
  632. });
  633. _vm->bind__repr__(_vm->tp_tuple, [](VM* vm, PyObject* _0){
  634. Tuple& iterable = _CAST(Tuple&, _0);
  635. std::stringstream ss;
  636. ss << '(';
  637. if(iterable.size() == 1){
  638. ss << CAST(Str&, vm->py_repr(iterable[0]));
  639. ss << ',';
  640. }else{
  641. for(int i=0; i<iterable.size(); i++){
  642. ss << CAST(Str&, vm->py_repr(iterable[i]));
  643. if(i != iterable.size()-1) ss << ", ";
  644. }
  645. }
  646. ss << ')';
  647. return VAR(ss.str());
  648. });
  649. _vm->bind__json__(_vm->tp_tuple, [](VM* vm, PyObject* _0){
  650. Tuple& iterable = _CAST(Tuple&, _0);
  651. std::stringstream ss;
  652. ss << '[';
  653. for(int i=0; i<iterable.size(); i++){
  654. ss << CAST(Str&, vm->py_json(iterable[i]));
  655. if(i != iterable.size()-1) ss << ", ";
  656. }
  657. ss << ']';
  658. return VAR(ss.str());
  659. });
  660. _vm->bind_constructor<-1>("list", [](VM* vm, ArgsView args) {
  661. if(args.size() == 1+0) return VAR(List());
  662. if(args.size() == 1+1){
  663. return vm->py_list(args[1]);
  664. }
  665. vm->TypeError("list() takes 0 or 1 arguments");
  666. return vm->None;
  667. });
  668. _vm->bind__contains__(_vm->tp_list, [](VM* vm, PyObject* obj, PyObject* item) {
  669. List& self = _CAST(List&, obj);
  670. for(PyObject* i: self) if(vm->py_equals(i, item)) return vm->True;
  671. return vm->False;
  672. });
  673. _vm->bind_method<1>("list", "count", [](VM* vm, ArgsView args) {
  674. List& self = _CAST(List&, args[0]);
  675. int count = 0;
  676. for(PyObject* i: self) if(vm->py_equals(i, args[1])) count++;
  677. return VAR(count);
  678. });
  679. _vm->bind__eq__(_vm->tp_list, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  680. List& a = _CAST(List&, lhs);
  681. if(!is_non_tagged_type(rhs, vm->tp_list)) return vm->NotImplemented;
  682. List& b = _CAST(List&, rhs);
  683. if(a.size() != b.size()) return vm->False;
  684. for(int i=0; i<a.size(); i++){
  685. if(!vm->py_equals(a[i], b[i])) return vm->False;
  686. }
  687. return vm->True;
  688. });
  689. _vm->bind_method<1>("list", "index", [](VM* vm, ArgsView args) {
  690. List& self = _CAST(List&, args[0]);
  691. PyObject* obj = args[1];
  692. for(int i=0; i<self.size(); i++){
  693. if(vm->py_equals(self[i], obj)) return VAR(i);
  694. }
  695. vm->ValueError(_CAST(Str&, vm->py_repr(obj)) + " is not in list");
  696. return vm->None;
  697. });
  698. _vm->bind_method<1>("list", "remove", [](VM* vm, ArgsView args) {
  699. List& self = _CAST(List&, args[0]);
  700. PyObject* obj = args[1];
  701. for(int i=0; i<self.size(); i++){
  702. if(vm->py_equals(self[i], obj)){
  703. self.erase(i);
  704. return vm->None;
  705. }
  706. }
  707. vm->ValueError(_CAST(Str&, vm->py_repr(obj)) + " is not in list");
  708. return vm->None;
  709. });
  710. _vm->bind_method<-1>("list", "pop", [](VM* vm, ArgsView args) {
  711. List& self = _CAST(List&, args[0]);
  712. if(args.size() == 1+0){
  713. if(self.empty()) vm->IndexError("pop from empty list");
  714. return self.popx_back();
  715. }
  716. if(args.size() == 1+1){
  717. int index = CAST(int, args[1]);
  718. index = vm->normalized_index(index, self.size());
  719. PyObject* ret = self[index];
  720. self.erase(index);
  721. return ret;
  722. }
  723. vm->TypeError("pop() takes at most 1 argument");
  724. return vm->None;
  725. });
  726. _vm->bind_method<1>("list", "append", [](VM* vm, ArgsView args) {
  727. List& self = _CAST(List&, args[0]);
  728. self.push_back(args[1]);
  729. return vm->None;
  730. });
  731. _vm->bind_method<1>("list", "extend", [](VM* vm, ArgsView args) {
  732. auto _lock = vm->heap.gc_scope_lock();
  733. List& self = _CAST(List&, args[0]);
  734. PyObject* it = vm->py_iter(args[1]); // strong ref
  735. PyObject* obj = vm->py_next(it);
  736. while(obj != vm->StopIteration){
  737. self.push_back(obj);
  738. obj = vm->py_next(it);
  739. }
  740. return vm->None;
  741. });
  742. _vm->bind_method<0>("list", "reverse", [](VM* vm, ArgsView args) {
  743. List& self = _CAST(List&, args[0]);
  744. std::reverse(self.begin(), self.end());
  745. return vm->None;
  746. });
  747. _vm->bind__mul__(_vm->tp_list, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  748. const List& self = _CAST(List&, lhs);
  749. if(!is_int(rhs)) return vm->NotImplemented;
  750. int n = _CAST(int, rhs);
  751. List result;
  752. result.reserve(self.size() * n);
  753. for(int i = 0; i < n; i++) result.extend(self);
  754. return VAR(std::move(result));
  755. });
  756. _vm->bind_method<1>("list", "__rmul__", [](VM* vm, ArgsView args) {
  757. const List& self = _CAST(List&, args[0]);
  758. if(!is_int(args[1])) return vm->NotImplemented;
  759. int n = _CAST(int, args[1]);
  760. List result;
  761. result.reserve(self.size() * n);
  762. for(int i = 0; i < n; i++) result.extend(self);
  763. return VAR(std::move(result));
  764. });
  765. _vm->bind_method<2>("list", "insert", [](VM* vm, ArgsView args) {
  766. List& self = _CAST(List&, args[0]);
  767. int index = CAST(int, args[1]);
  768. if(index < 0) index += self.size();
  769. if(index < 0) index = 0;
  770. if(index > self.size()) index = self.size();
  771. self.insert(index, args[2]);
  772. return vm->None;
  773. });
  774. _vm->bind_method<0>("list", "clear", [](VM* vm, ArgsView args) {
  775. _CAST(List&, args[0]).clear();
  776. return vm->None;
  777. });
  778. _vm->bind_method<0>("list", "copy", PK_LAMBDA(VAR(_CAST(List, args[0]))));
  779. _vm->bind__add__(_vm->tp_list, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  780. const List& self = _CAST(List&, lhs);
  781. const List& other = CAST(List&, rhs);
  782. List new_list(self); // copy construct
  783. new_list.extend(other);
  784. return VAR(std::move(new_list));
  785. });
  786. _vm->bind__len__(_vm->tp_list, [](VM* vm, PyObject* obj) {
  787. return (i64)_CAST(List&, obj).size();
  788. });
  789. _vm->bind__iter__(_vm->tp_list, [](VM* vm, PyObject* obj) {
  790. List& self = _CAST(List&, obj);
  791. return VAR_T(ArrayIter, obj, self.begin(), self.end());
  792. });
  793. _vm->bind__getitem__(_vm->tp_list, PyArrayGetItem<List>);
  794. _vm->bind__setitem__(_vm->tp_list, [](VM* vm, PyObject* obj, PyObject* index, PyObject* value){
  795. List& self = _CAST(List&, obj);
  796. int i = CAST(int, index);
  797. i = vm->normalized_index(i, self.size());
  798. self[i] = value;
  799. });
  800. _vm->bind__delitem__(_vm->tp_list, [](VM* vm, PyObject* obj, PyObject* index){
  801. List& self = _CAST(List&, obj);
  802. int i = CAST(int, index);
  803. i = vm->normalized_index(i, self.size());
  804. self.erase(i);
  805. });
  806. /************ tuple ************/
  807. _vm->bind_constructor<-1>("tuple", [](VM* vm, ArgsView args) {
  808. if(args.size() == 1+0) return VAR(Tuple(0));
  809. if(args.size() == 1+1){
  810. List list = CAST(List, vm->py_list(args[1]));
  811. return VAR(Tuple(std::move(list)));
  812. }
  813. vm->TypeError("tuple() takes at most 1 argument");
  814. return vm->None;
  815. });
  816. _vm->bind__contains__(_vm->tp_tuple, [](VM* vm, PyObject* obj, PyObject* item) {
  817. Tuple& self = _CAST(Tuple&, obj);
  818. for(PyObject* i: self) if(vm->py_equals(i, item)) return vm->True;
  819. return vm->False;
  820. });
  821. _vm->bind_method<1>("tuple", "count", [](VM* vm, ArgsView args) {
  822. Tuple& self = _CAST(Tuple&, args[0]);
  823. int count = 0;
  824. for(PyObject* i: self) if(vm->py_equals(i, args[1])) count++;
  825. return VAR(count);
  826. });
  827. _vm->bind__eq__(_vm->tp_tuple, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  828. const Tuple& self = _CAST(Tuple&, lhs);
  829. if(!is_non_tagged_type(rhs, vm->tp_tuple)) return vm->NotImplemented;
  830. const Tuple& other = _CAST(Tuple&, rhs);
  831. if(self.size() != other.size()) return vm->False;
  832. for(int i = 0; i < self.size(); i++) {
  833. if(!vm->py_equals(self[i], other[i])) return vm->False;
  834. }
  835. return vm->True;
  836. });
  837. _vm->bind__hash__(_vm->tp_tuple, [](VM* vm, PyObject* obj) {
  838. i64 x = 1000003;
  839. const Tuple& items = CAST(Tuple&, obj);
  840. for (int i=0; i<items.size(); i++) {
  841. i64 y = vm->py_hash(items[i]);
  842. // recommended by Github Copilot
  843. x = x ^ (y + 0x9e3779b9 + (x << 6) + (x >> 2));
  844. }
  845. return x;
  846. });
  847. _vm->bind__iter__(_vm->tp_tuple, [](VM* vm, PyObject* obj) {
  848. Tuple& self = _CAST(Tuple&, obj);
  849. return VAR_T(ArrayIter, obj, self.begin(), self.end());
  850. });
  851. _vm->bind__getitem__(_vm->tp_tuple, PyArrayGetItem<Tuple>);
  852. _vm->bind__len__(_vm->tp_tuple, [](VM* vm, PyObject* obj) {
  853. return (i64)_CAST(Tuple&, obj).size();
  854. });
  855. /************ bool ************/
  856. _vm->bind_constructor<2>("bool", PK_LAMBDA(VAR(vm->py_bool(args[1]))));
  857. _vm->bind__hash__(_vm->tp_bool, [](VM* vm, PyObject* obj) {
  858. return (i64)_CAST(bool, obj);
  859. });
  860. _vm->bind__repr__(_vm->tp_bool, [](VM* vm, PyObject* self) {
  861. bool val = _CAST(bool, self);
  862. return VAR(val ? "True" : "False");
  863. });
  864. _vm->bind__json__(_vm->tp_bool, [](VM* vm, PyObject* self) {
  865. bool val = _CAST(bool, self);
  866. return VAR(val ? "true" : "false");
  867. });
  868. const PK_LOCAL_STATIC auto f_bool_add = [](VM* vm, PyObject* lhs, PyObject* rhs) -> PyObject* {
  869. int x = (int)_CAST(bool, lhs);
  870. if(is_int(rhs)) return VAR(x + _CAST(int, rhs));
  871. if(rhs == vm->True) return VAR(x + 1);
  872. if(rhs == vm->False) return VAR(x);
  873. return vm->NotImplemented;
  874. };
  875. const PK_LOCAL_STATIC auto f_bool_mul = [](VM* vm, PyObject* lhs, PyObject* rhs) -> PyObject* {
  876. int x = (int)_CAST(bool, lhs);
  877. if(is_int(rhs)) return VAR(x * _CAST(int, rhs));
  878. if(rhs == vm->True) return VAR(x);
  879. if(rhs == vm->False) return VAR(0);
  880. return vm->NotImplemented;
  881. };
  882. _vm->bind__add__(_vm->tp_bool, f_bool_add);
  883. _vm->bind_method<1>("bool", "__radd__", [](VM* vm, ArgsView args){
  884. return f_bool_add(vm, args[0], args[1]);
  885. });
  886. _vm->bind__mul__(_vm->tp_bool, f_bool_mul);
  887. _vm->bind_method<1>("bool", "__rmul__", [](VM* vm, ArgsView args){
  888. return f_bool_mul(vm, args[0], args[1]);
  889. });
  890. _vm->bind__and__(_vm->tp_bool, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  891. return VAR(_CAST(bool, lhs) && CAST(bool, rhs));
  892. });
  893. _vm->bind__or__(_vm->tp_bool, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  894. return VAR(_CAST(bool, lhs) || CAST(bool, rhs));
  895. });
  896. _vm->bind__xor__(_vm->tp_bool, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  897. return VAR(_CAST(bool, lhs) != CAST(bool, rhs));
  898. });
  899. _vm->bind__eq__(_vm->tp_bool, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  900. if(is_non_tagged_type(rhs, vm->tp_bool)) return VAR(lhs == rhs);
  901. if(is_int(rhs)) return VAR(_CAST(bool, lhs) == (bool)CAST(i64, rhs));
  902. return vm->NotImplemented;
  903. });
  904. _vm->bind__repr__(_vm->_type("ellipsis"), [](VM* vm, PyObject* self) {
  905. return VAR("...");
  906. });
  907. _vm->bind__repr__(_vm->_type("NotImplementedType"), [](VM* vm, PyObject* self) {
  908. return VAR("NotImplemented");
  909. });
  910. /************ bytes ************/
  911. _vm->bind_constructor<2>("bytes", [](VM* vm, ArgsView args){
  912. List& list = CAST(List&, args[1]);
  913. std::vector<unsigned char> buffer(list.size());
  914. for(int i=0; i<list.size(); i++){
  915. i64 b = CAST(i64, list[i]);
  916. if(b<0 || b>255) vm->ValueError("byte must be in range[0, 256)");
  917. buffer[i] = (char)b;
  918. }
  919. return VAR(Bytes(buffer));
  920. });
  921. _vm->bind__getitem__(_vm->tp_bytes, [](VM* vm, PyObject* obj, PyObject* index) {
  922. const Bytes& self = _CAST(Bytes&, obj);
  923. int i = CAST(int, index);
  924. i = vm->normalized_index(i, self.size());
  925. return VAR(self[i]);
  926. });
  927. _vm->bind__hash__(_vm->tp_bytes, [](VM* vm, PyObject* obj) {
  928. const Bytes& self = _CAST(Bytes&, obj);
  929. std::string_view view((char*)self.data(), self.size());
  930. return (i64)std::hash<std::string_view>()(view);
  931. });
  932. _vm->bind__repr__(_vm->tp_bytes, [](VM* vm, PyObject* obj) {
  933. const Bytes& self = _CAST(Bytes&, obj);
  934. std::stringstream ss;
  935. ss << "b'";
  936. for(int i=0; i<self.size(); i++){
  937. ss << "\\x" << std::hex << std::setw(2) << std::setfill('0') << self[i];
  938. }
  939. ss << "'";
  940. return VAR(ss.str());
  941. });
  942. _vm->bind__len__(_vm->tp_bytes, [](VM* vm, PyObject* obj) {
  943. return (i64)_CAST(Bytes&, obj).size();
  944. });
  945. _vm->bind_method<0>("bytes", "decode", [](VM* vm, ArgsView args) {
  946. const Bytes& self = _CAST(Bytes&, args[0]);
  947. // TODO: check encoding is utf-8
  948. return VAR(Str(self.str()));
  949. });
  950. _vm->bind__eq__(_vm->tp_bytes, [](VM* vm, PyObject* lhs, PyObject* rhs) {
  951. if(!is_non_tagged_type(rhs, vm->tp_bytes)) return vm->NotImplemented;
  952. return VAR(_CAST(Bytes&, lhs) == _CAST(Bytes&, rhs));
  953. });
  954. /************ slice ************/
  955. _vm->bind_constructor<4>("slice", [](VM* vm, ArgsView args) {
  956. return VAR(Slice(args[1], args[2], args[3]));
  957. });
  958. _vm->bind__repr__(_vm->tp_slice, [](VM* vm, PyObject* obj) {
  959. const Slice& self = _CAST(Slice&, obj);
  960. std::stringstream ss;
  961. ss << "slice(";
  962. ss << CAST(Str, vm->py_repr(self.start)) << ", ";
  963. ss << CAST(Str, vm->py_repr(self.stop)) << ", ";
  964. ss << CAST(Str, vm->py_repr(self.step)) << ")";
  965. return VAR(ss.str());
  966. });
  967. /************ mappingproxy ************/
  968. _vm->bind_method<0>("mappingproxy", "keys", [](VM* vm, ArgsView args) {
  969. MappingProxy& self = _CAST(MappingProxy&, args[0]);
  970. List keys;
  971. for(StrName name : self.attr().keys()) keys.push_back(VAR(name.sv()));
  972. return VAR(std::move(keys));
  973. });
  974. _vm->bind_method<0>("mappingproxy", "values", [](VM* vm, ArgsView args) {
  975. MappingProxy& self = _CAST(MappingProxy&, args[0]);
  976. List values;
  977. for(auto& item : self.attr().items()) values.push_back(item.second);
  978. return VAR(std::move(values));
  979. });
  980. _vm->bind_method<0>("mappingproxy", "items", [](VM* vm, ArgsView args) {
  981. MappingProxy& self = _CAST(MappingProxy&, args[0]);
  982. List items;
  983. for(auto& item : self.attr().items()){
  984. PyObject* t = VAR(Tuple({VAR(item.first.sv()), item.second}));
  985. items.push_back(std::move(t));
  986. }
  987. return VAR(std::move(items));
  988. });
  989. _vm->bind__len__(_vm->tp_mappingproxy, [](VM* vm, PyObject* obj) {
  990. return (i64)_CAST(MappingProxy&, obj).attr().size();
  991. });
  992. _vm->bind__eq__(_vm->tp_mappingproxy, [](VM* vm, PyObject* obj, PyObject* other){
  993. MappingProxy& a = _CAST(MappingProxy&, obj);
  994. if(!is_non_tagged_type(other, vm->tp_mappingproxy)){
  995. return vm->NotImplemented;
  996. }
  997. MappingProxy& b = _CAST(MappingProxy&, other);
  998. return VAR(a.obj == b.obj);
  999. });
  1000. _vm->bind__getitem__(_vm->tp_mappingproxy, [](VM* vm, PyObject* obj, PyObject* index) {
  1001. MappingProxy& self = _CAST(MappingProxy&, obj);
  1002. StrName key = CAST(Str&, index);
  1003. PyObject* ret = self.attr().try_get(key);
  1004. if(ret == nullptr) vm->AttributeError(key.sv());
  1005. return ret;
  1006. });
  1007. _vm->bind__repr__(_vm->tp_mappingproxy, [](VM* vm, PyObject* obj) {
  1008. MappingProxy& self = _CAST(MappingProxy&, obj);
  1009. std::stringstream ss;
  1010. ss << "mappingproxy({";
  1011. bool first = true;
  1012. for(auto& item : self.attr().items()){
  1013. if(!first) ss << ", ";
  1014. first = false;
  1015. ss << item.first.escape() << ": " << CAST(Str, vm->py_repr(item.second));
  1016. }
  1017. ss << "})";
  1018. return VAR(ss.str());
  1019. });
  1020. _vm->bind__contains__(_vm->tp_mappingproxy, [](VM* vm, PyObject* obj, PyObject* key) {
  1021. MappingProxy& self = _CAST(MappingProxy&, obj);
  1022. return VAR(self.attr().contains(CAST(Str&, key)));
  1023. });
  1024. /************ dict ************/
  1025. _vm->bind_constructor<-1>("dict", [](VM* vm, ArgsView args){
  1026. return VAR(Dict(vm));
  1027. });
  1028. _vm->bind_method<-1>("dict", "__init__", [](VM* vm, ArgsView args){
  1029. if(args.size() == 1+0) return vm->None;
  1030. if(args.size() == 1+1){
  1031. auto _lock = vm->heap.gc_scope_lock();
  1032. Dict& self = _CAST(Dict&, args[0]);
  1033. List& list = CAST(List&, args[1]);
  1034. for(PyObject* item : list){
  1035. Tuple& t = CAST(Tuple&, item);
  1036. if(t.size() != 2){
  1037. vm->ValueError("dict() takes an iterable of tuples (key, value)");
  1038. return vm->None;
  1039. }
  1040. self.set(t[0], t[1]);
  1041. }
  1042. return vm->None;
  1043. }
  1044. vm->TypeError("dict() takes at most 1 argument");
  1045. return vm->None;
  1046. });
  1047. _vm->bind__len__(_vm->tp_dict, [](VM* vm, PyObject* obj) {
  1048. return (i64)_CAST(Dict&, obj).size();
  1049. });
  1050. _vm->bind__getitem__(_vm->tp_dict, [](VM* vm, PyObject* obj, PyObject* index) {
  1051. Dict& self = _CAST(Dict&, obj);
  1052. PyObject* ret = self.try_get(index);
  1053. if(ret == nullptr) vm->KeyError(index);
  1054. return ret;
  1055. });
  1056. _vm->bind__setitem__(_vm->tp_dict, [](VM* vm, PyObject* obj, PyObject* key, PyObject* value) {
  1057. Dict& self = _CAST(Dict&, obj);
  1058. self.set(key, value);
  1059. });
  1060. _vm->bind__delitem__(_vm->tp_dict, [](VM* vm, PyObject* obj, PyObject* key) {
  1061. Dict& self = _CAST(Dict&, obj);
  1062. bool ok = self.erase(key);
  1063. if(!ok) vm->KeyError(key);
  1064. });
  1065. _vm->bind_method<-1>("dict", "pop", [](VM* vm, ArgsView args) {
  1066. if(args.size() != 2 && args.size() != 3){
  1067. vm->TypeError("pop() expected 1 or 2 arguments");
  1068. return vm->None;
  1069. }
  1070. Dict& self = _CAST(Dict&, args[0]);
  1071. PyObject* value = self.try_get(args[1]);
  1072. if(value == nullptr){
  1073. if(args.size() == 2) vm->KeyError(args[1]);
  1074. if(args.size() == 3){
  1075. return args[2];
  1076. }
  1077. }
  1078. self.erase(args[1]);
  1079. return value;
  1080. });
  1081. // _vm->bind_method<0>("dict", "_data", [](VM* vm, ArgsView args) {
  1082. // Dict& self = _CAST(Dict&, args[0]);
  1083. // std::stringstream ss;
  1084. // ss << "[\n";
  1085. // for(int i=0; i<self._capacity; i++){
  1086. // auto item = self._items[i];
  1087. // Str key("None");
  1088. // Str value("None");
  1089. // if(item.first != nullptr){
  1090. // key = CAST(Str&, vm->py_repr(item.first));
  1091. // }
  1092. // if(item.second != nullptr){
  1093. // value = CAST(Str&, vm->py_repr(item.second));
  1094. // }
  1095. // int prev = self._nodes[i].prev;
  1096. // int next = self._nodes[i].next;
  1097. // ss << " [" << key << ", " << value << ", " << prev << ", " << next << "],\n";
  1098. // }
  1099. // ss << "]\n";
  1100. // vm->stdout_write(ss.str());
  1101. // return vm->None;
  1102. // });
  1103. _vm->bind__contains__(_vm->tp_dict, [](VM* vm, PyObject* obj, PyObject* key) {
  1104. Dict& self = _CAST(Dict&, obj);
  1105. return VAR(self.contains(key));
  1106. });
  1107. _vm->bind__iter__(_vm->tp_dict, [](VM* vm, PyObject* obj) {
  1108. const Dict& self = _CAST(Dict&, obj);
  1109. return vm->py_iter(VAR(self.keys()));
  1110. });
  1111. _vm->bind_method<-1>("dict", "get", [](VM* vm, ArgsView args) {
  1112. Dict& self = _CAST(Dict&, args[0]);
  1113. if(args.size() == 1+1){
  1114. PyObject* ret = self.try_get(args[1]);
  1115. if(ret != nullptr) return ret;
  1116. return vm->None;
  1117. }else if(args.size() == 1+2){
  1118. PyObject* ret = self.try_get(args[1]);
  1119. if(ret != nullptr) return ret;
  1120. return args[2];
  1121. }
  1122. vm->TypeError("get() takes at most 2 arguments");
  1123. return vm->None;
  1124. });
  1125. _vm->bind_method<0>("dict", "keys", [](VM* vm, ArgsView args) {
  1126. const Dict& self = _CAST(Dict&, args[0]);
  1127. return VAR(self.keys());
  1128. });
  1129. _vm->bind_method<0>("dict", "values", [](VM* vm, ArgsView args) {
  1130. const Dict& self = _CAST(Dict&, args[0]);
  1131. return VAR(self.values());
  1132. });
  1133. _vm->bind_method<0>("dict", "items", [](VM* vm, ArgsView args) {
  1134. const Dict& self = _CAST(Dict&, args[0]);
  1135. Tuple items(self.size());
  1136. int j = 0;
  1137. self.apply([&](PyObject* k, PyObject* v){
  1138. items[j++] = VAR(Tuple({k, v}));
  1139. });
  1140. return VAR(std::move(items));
  1141. });
  1142. _vm->bind_method<1>("dict", "update", [](VM* vm, ArgsView args) {
  1143. Dict& self = _CAST(Dict&, args[0]);
  1144. const Dict& other = CAST(Dict&, args[1]);
  1145. self.update(other);
  1146. return vm->None;
  1147. });
  1148. _vm->bind_method<0>("dict", "copy", [](VM* vm, ArgsView args) {
  1149. const Dict& self = _CAST(Dict&, args[0]);
  1150. return VAR(self);
  1151. });
  1152. _vm->bind_method<0>("dict", "clear", [](VM* vm, ArgsView args) {
  1153. Dict& self = _CAST(Dict&, args[0]);
  1154. self.clear();
  1155. return vm->None;
  1156. });
  1157. _vm->bind__repr__(_vm->tp_dict, [](VM* vm, PyObject* obj) {
  1158. Dict& self = _CAST(Dict&, obj);
  1159. std::stringstream ss;
  1160. ss << "{";
  1161. bool first = true;
  1162. self.apply([&](PyObject* k, PyObject* v){
  1163. if(!first) ss << ", ";
  1164. first = false;
  1165. Str key = CAST(Str&, vm->py_repr(k));
  1166. Str value = CAST(Str&, vm->py_repr(v));
  1167. ss << key << ": " << value;
  1168. });
  1169. ss << "}";
  1170. return VAR(ss.str());
  1171. });
  1172. _vm->bind__json__(_vm->tp_dict, [](VM* vm, PyObject* obj) {
  1173. Dict& self = _CAST(Dict&, obj);
  1174. std::stringstream ss;
  1175. ss << "{";
  1176. bool first = true;
  1177. self.apply([&](PyObject* k, PyObject* v){
  1178. if(!first) ss << ", ";
  1179. first = false;
  1180. Str key = CAST(Str&, k).escape(false);
  1181. Str value = CAST(Str&, vm->py_json(v));
  1182. ss << key << ": " << value;
  1183. });
  1184. ss << "}";
  1185. return VAR(ss.str());
  1186. });
  1187. _vm->bind__eq__(_vm->tp_dict, [](VM* vm, PyObject* a, PyObject* b) {
  1188. Dict& self = _CAST(Dict&, a);
  1189. if(!is_non_tagged_type(b, vm->tp_dict)) return vm->NotImplemented;
  1190. Dict& other = _CAST(Dict&, b);
  1191. if(self.size() != other.size()) return vm->False;
  1192. for(int i=0; i<self._capacity; i++){
  1193. auto item = self._items[i];
  1194. if(item.first == nullptr) continue;
  1195. PyObject* value = other.try_get(item.first);
  1196. if(value == nullptr) return vm->False;
  1197. if(!vm->py_equals(item.second, value)) return vm->False;
  1198. }
  1199. return vm->True;
  1200. });
  1201. _vm->bind__repr__(_vm->tp_module, [](VM* vm, PyObject* obj) {
  1202. const Str& path = CAST(Str&, obj->attr(__path__));
  1203. return VAR(fmt("<module ", path.escape(), ">"));
  1204. });
  1205. /************ property ************/
  1206. _vm->bind_constructor<-1>("property", [](VM* vm, ArgsView args) {
  1207. if(args.size() == 1+1){
  1208. return VAR(Property(args[1], vm->None, ""));
  1209. }else if(args.size() == 1+2){
  1210. return VAR(Property(args[1], args[2], ""));
  1211. }else if(args.size() == 1+3){
  1212. return VAR(Property(args[1], args[2], CAST(Str, args[3])));
  1213. }
  1214. vm->TypeError("property() takes at most 3 arguments");
  1215. return vm->None;
  1216. });
  1217. _vm->bind_property(_vm->_t(_vm->tp_property), "__signature__", [](VM* vm, ArgsView args){
  1218. Property& self = _CAST(Property&, args[0]);
  1219. return VAR(self.signature);
  1220. });
  1221. _vm->bind_property(_vm->_t(_vm->tp_function), "__doc__", [](VM* vm, ArgsView args) {
  1222. Function& func = _CAST(Function&, args[0]);
  1223. return VAR(func.decl->docstring);
  1224. });
  1225. _vm->bind_property(_vm->_t(_vm->tp_native_func), "__doc__", [](VM* vm, ArgsView args) {
  1226. NativeFunc& func = _CAST(NativeFunc&, args[0]);
  1227. if(func.decl != nullptr) return VAR(func.decl->docstring);
  1228. return VAR("");
  1229. });
  1230. _vm->bind_property(_vm->_t(_vm->tp_function), "__signature__", [](VM* vm, ArgsView args) {
  1231. Function& func = _CAST(Function&, args[0]);
  1232. return VAR(func.decl->signature);
  1233. });
  1234. // _vm->bind_property(_vm->_t(_vm->tp_function), "__call__", [](VM* vm, ArgsView args) {
  1235. // return args[0];
  1236. // });
  1237. _vm->bind_property(_vm->_t(_vm->tp_native_func), "__signature__", [](VM* vm, ArgsView args) {
  1238. NativeFunc& func = _CAST(NativeFunc&, args[0]);
  1239. if(func.decl != nullptr) return VAR(func.decl->signature);
  1240. return VAR("");
  1241. });
  1242. // _vm->bind_property(_vm->_t(_vm->tp_native_func), "__call__", [](VM* vm, ArgsView args) {
  1243. // return args[0];
  1244. // });
  1245. RangeIter::register_class(_vm, _vm->builtins);
  1246. ArrayIter::register_class(_vm, _vm->builtins);
  1247. StringIter::register_class(_vm, _vm->builtins);
  1248. Generator::register_class(_vm, _vm->builtins);
  1249. }
  1250. void add_module_timeit(VM* vm){
  1251. PyObject* mod = vm->new_module("timeit");
  1252. vm->bind_func<2>(mod, "timeit", [](VM* vm, ArgsView args) {
  1253. PyObject* f = args[0];
  1254. i64 iters = CAST(i64, args[1]);
  1255. auto now = std::chrono::system_clock::now();
  1256. for(i64 i=0; i<iters; i++) vm->call(f);
  1257. auto end = std::chrono::system_clock::now();
  1258. f64 elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(end - now).count() / 1000.0;
  1259. return VAR(elapsed);
  1260. });
  1261. }
  1262. struct PyStructTime{
  1263. PY_CLASS(PyStructTime, time, struct_time)
  1264. int tm_year;
  1265. int tm_mon;
  1266. int tm_mday;
  1267. int tm_hour;
  1268. int tm_min;
  1269. int tm_sec;
  1270. int tm_wday;
  1271. int tm_yday;
  1272. int tm_isdst;
  1273. PyStructTime(std::time_t t){
  1274. std::tm* tm = std::localtime(&t);
  1275. tm_year = tm->tm_year + 1900;
  1276. tm_mon = tm->tm_mon + 1;
  1277. tm_mday = tm->tm_mday;
  1278. tm_hour = tm->tm_hour;
  1279. tm_min = tm->tm_min;
  1280. tm_sec = tm->tm_sec + 1;
  1281. tm_wday = (tm->tm_wday + 6) % 7;
  1282. tm_yday = tm->tm_yday + 1;
  1283. tm_isdst = tm->tm_isdst;
  1284. }
  1285. PyStructTime* _() { return this; }
  1286. static void _register(VM* vm, PyObject* mod, PyObject* type){
  1287. vm->bind_notimplemented_constructor<PyStructTime>(type);
  1288. PY_READONLY_FIELD(PyStructTime, "tm_year", _, tm_year);
  1289. PY_READONLY_FIELD(PyStructTime, "tm_mon", _, tm_mon);
  1290. PY_READONLY_FIELD(PyStructTime, "tm_mday", _, tm_mday);
  1291. PY_READONLY_FIELD(PyStructTime, "tm_hour", _, tm_hour);
  1292. PY_READONLY_FIELD(PyStructTime, "tm_min", _, tm_min);
  1293. PY_READONLY_FIELD(PyStructTime, "tm_sec", _, tm_sec);
  1294. PY_READONLY_FIELD(PyStructTime, "tm_wday", _, tm_wday);
  1295. PY_READONLY_FIELD(PyStructTime, "tm_yday", _, tm_yday);
  1296. PY_READONLY_FIELD(PyStructTime, "tm_isdst", _, tm_isdst);
  1297. }
  1298. };
  1299. void add_module_time(VM* vm){
  1300. PyObject* mod = vm->new_module("time");
  1301. PyStructTime::register_class(vm, mod);
  1302. vm->bind_func<0>(mod, "time", [](VM* vm, ArgsView args) {
  1303. auto now = std::chrono::system_clock::now();
  1304. return VAR(std::chrono::duration_cast<std::chrono::milliseconds>(now.time_since_epoch()).count() / 1000.0);
  1305. });
  1306. vm->bind_func<1>(mod, "sleep", [](VM* vm, ArgsView args) {
  1307. f64 seconds = CAST_F(args[0]);
  1308. auto begin = std::chrono::system_clock::now();
  1309. while(true){
  1310. auto now = std::chrono::system_clock::now();
  1311. f64 elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(now - begin).count() / 1000.0;
  1312. if(elapsed >= seconds) break;
  1313. }
  1314. return vm->None;
  1315. });
  1316. vm->bind_func<0>(mod, "localtime", [](VM* vm, ArgsView args) {
  1317. auto now = std::chrono::system_clock::now();
  1318. std::time_t t = std::chrono::system_clock::to_time_t(now);
  1319. return VAR_T(PyStructTime, t);
  1320. });
  1321. }
  1322. void add_module_sys(VM* vm){
  1323. PyObject* mod = vm->new_module("sys");
  1324. vm->setattr(mod, "version", VAR(PK_VERSION));
  1325. vm->setattr(mod, "platform", VAR(PK_SYS_PLATFORM));
  1326. PyObject* stdout_ = vm->heap.gcnew<DummyInstance>(vm->tp_object);
  1327. PyObject* stderr_ = vm->heap.gcnew<DummyInstance>(vm->tp_object);
  1328. vm->setattr(mod, "stdout", stdout_);
  1329. vm->setattr(mod, "stderr", stderr_);
  1330. vm->bind_func<1>(stdout_, "write", [](VM* vm, ArgsView args) {
  1331. Str& s = CAST(Str&, args[0]);
  1332. vm->_stdout(vm, s.data, s.size);
  1333. return vm->None;
  1334. });
  1335. vm->bind_func<1>(stderr_, "write", [](VM* vm, ArgsView args) {
  1336. Str& s = CAST(Str&, args[0]);
  1337. vm->_stderr(vm, s.data, s.size);
  1338. return vm->None;
  1339. });
  1340. }
  1341. void add_module_json(VM* vm){
  1342. PyObject* mod = vm->new_module("json");
  1343. vm->bind_func<1>(mod, "loads", [](VM* vm, ArgsView args) {
  1344. const Str& expr = CAST(Str&, args[0]);
  1345. CodeObject_ code = vm->compile(expr, "<json>", JSON_MODE);
  1346. return vm->_exec(code, vm->top_frame()->_module);
  1347. });
  1348. vm->bind_func<1>(mod, "dumps", [](VM* vm, ArgsView args) {
  1349. return vm->py_json(args[0]);
  1350. });
  1351. }
  1352. // https://docs.python.org/3.5/library/math.html
  1353. void add_module_math(VM* vm){
  1354. PyObject* mod = vm->new_module("math");
  1355. mod->attr().set("pi", VAR(3.1415926535897932384));
  1356. mod->attr().set("e" , VAR(2.7182818284590452354));
  1357. mod->attr().set("inf", VAR(std::numeric_limits<double>::infinity()));
  1358. mod->attr().set("nan", VAR(std::numeric_limits<double>::quiet_NaN()));
  1359. vm->bind_func<1>(mod, "ceil", PK_LAMBDA(VAR((i64)std::ceil(CAST_F(args[0])))));
  1360. vm->bind_func<1>(mod, "fabs", PK_LAMBDA(VAR(std::fabs(CAST_F(args[0])))));
  1361. vm->bind_func<1>(mod, "floor", PK_LAMBDA(VAR((i64)std::floor(CAST_F(args[0])))));
  1362. vm->bind_func<1>(mod, "fsum", [](VM* vm, ArgsView args) {
  1363. List& list = CAST(List&, args[0]);
  1364. double sum = 0;
  1365. double c = 0;
  1366. for(PyObject* arg : list){
  1367. double x = CAST_F(arg);
  1368. double y = x - c;
  1369. double t = sum + y;
  1370. c = (t - sum) - y;
  1371. sum = t;
  1372. }
  1373. return VAR(sum);
  1374. });
  1375. vm->bind_func<2>(mod, "gcd", [](VM* vm, ArgsView args) {
  1376. i64 a = CAST(i64, args[0]);
  1377. i64 b = CAST(i64, args[1]);
  1378. if(a < 0) a = -a;
  1379. if(b < 0) b = -b;
  1380. while(b != 0){
  1381. i64 t = b;
  1382. b = a % b;
  1383. a = t;
  1384. }
  1385. return VAR(a);
  1386. });
  1387. vm->bind_func<1>(mod, "isfinite", PK_LAMBDA(VAR(std::isfinite(CAST_F(args[0])))));
  1388. vm->bind_func<1>(mod, "isinf", PK_LAMBDA(VAR(std::isinf(CAST_F(args[0])))));
  1389. vm->bind_func<1>(mod, "isnan", PK_LAMBDA(VAR(std::isnan(CAST_F(args[0])))));
  1390. vm->bind_func<1>(mod, "exp", PK_LAMBDA(VAR(std::exp(CAST_F(args[0])))));
  1391. vm->bind_func<1>(mod, "log", PK_LAMBDA(VAR(std::log(CAST_F(args[0])))));
  1392. vm->bind_func<1>(mod, "log2", PK_LAMBDA(VAR(std::log2(CAST_F(args[0])))));
  1393. vm->bind_func<1>(mod, "log10", PK_LAMBDA(VAR(std::log10(CAST_F(args[0])))));
  1394. vm->bind_func<2>(mod, "pow", PK_LAMBDA(VAR(std::pow(CAST_F(args[0]), CAST_F(args[1])))));
  1395. vm->bind_func<1>(mod, "sqrt", PK_LAMBDA(VAR(std::sqrt(CAST_F(args[0])))));
  1396. vm->bind_func<1>(mod, "acos", PK_LAMBDA(VAR(std::acos(CAST_F(args[0])))));
  1397. vm->bind_func<1>(mod, "asin", PK_LAMBDA(VAR(std::asin(CAST_F(args[0])))));
  1398. vm->bind_func<1>(mod, "atan", PK_LAMBDA(VAR(std::atan(CAST_F(args[0])))));
  1399. vm->bind_func<2>(mod, "atan2", PK_LAMBDA(VAR(std::atan2(CAST_F(args[0]), CAST_F(args[1])))));
  1400. vm->bind_func<1>(mod, "cos", PK_LAMBDA(VAR(std::cos(CAST_F(args[0])))));
  1401. vm->bind_func<1>(mod, "sin", PK_LAMBDA(VAR(std::sin(CAST_F(args[0])))));
  1402. vm->bind_func<1>(mod, "tan", PK_LAMBDA(VAR(std::tan(CAST_F(args[0])))));
  1403. vm->bind_func<1>(mod, "degrees", PK_LAMBDA(VAR(CAST_F(args[0]) * 180 / 3.1415926535897932384)));
  1404. vm->bind_func<1>(mod, "radians", PK_LAMBDA(VAR(CAST_F(args[0]) * 3.1415926535897932384 / 180)));
  1405. vm->bind_func<1>(mod, "modf", [](VM* vm, ArgsView args) {
  1406. f64 i;
  1407. f64 f = std::modf(CAST_F(args[0]), &i);
  1408. return VAR(Tuple({VAR(f), VAR(i)}));
  1409. });
  1410. vm->bind_func<1>(mod, "factorial", [](VM* vm, ArgsView args) {
  1411. i64 n = CAST(i64, args[0]);
  1412. if(n < 0) vm->ValueError("factorial() not defined for negative values");
  1413. i64 r = 1;
  1414. for(i64 i=2; i<=n; i++) r *= i;
  1415. return VAR(r);
  1416. });
  1417. }
  1418. void add_module_traceback(VM* vm){
  1419. PyObject* mod = vm->new_module("traceback");
  1420. vm->bind_func<0>(mod, "print_exc", [](VM* vm, ArgsView args) {
  1421. if(vm->_last_exception==nullptr) vm->ValueError("no exception");
  1422. Exception& e = CAST(Exception&, vm->_last_exception);
  1423. Str sum = e.summary();
  1424. vm->_stdout(vm, sum.data, sum.size);
  1425. return vm->None;
  1426. });
  1427. vm->bind_func<0>(mod, "format_exc", [](VM* vm, ArgsView args) {
  1428. if(vm->_last_exception==nullptr) vm->ValueError("no exception");
  1429. Exception& e = CAST(Exception&, vm->_last_exception);
  1430. return VAR(e.summary());
  1431. });
  1432. }
  1433. void add_module_dis(VM* vm){
  1434. PyObject* mod = vm->new_module("dis");
  1435. static const auto get_code = [](VM* vm, PyObject* obj)->CodeObject_{
  1436. if(is_type(obj, vm->tp_str)){
  1437. const Str& source = CAST(Str, obj);
  1438. return vm->compile(source, "<dis>", EXEC_MODE);
  1439. }
  1440. PyObject* f = obj;
  1441. if(is_type(f, vm->tp_bound_method)) f = CAST(BoundMethod, obj).func;
  1442. return CAST(Function&, f).decl->code;
  1443. };
  1444. vm->bind_func<1>(mod, "dis", [](VM* vm, ArgsView args) {
  1445. CodeObject_ code = get_code(vm, args[0]);
  1446. Str msg = vm->disassemble(code);
  1447. vm->_stdout(vm, msg.data, msg.size);
  1448. return vm->None;
  1449. });
  1450. vm->bind_func<1>(mod, "_s", [](VM* vm, ArgsView args) {
  1451. CodeObject_ code = get_code(vm, args[0]);
  1452. return VAR(code->serialize(vm));
  1453. });
  1454. }
  1455. void add_module_gc(VM* vm){
  1456. PyObject* mod = vm->new_module("gc");
  1457. vm->bind_func<0>(mod, "collect", PK_LAMBDA(VAR(vm->heap.collect())));
  1458. }
  1459. void VM::post_init(){
  1460. init_builtins(this);
  1461. bind_property(_t(tp_object), "__class__", PK_LAMBDA(VAR(vm->_t(args[0]))));
  1462. bind_property(_t(tp_type), "__base__", [](VM* vm, ArgsView args){
  1463. const PyTypeInfo& info = vm->_all_types[PK_OBJ_GET(Type, args[0])];
  1464. return info.base.index == -1 ? vm->None : vm->_all_types[info.base].obj;
  1465. });
  1466. bind_property(_t(tp_type), "__name__", [](VM* vm, ArgsView args){
  1467. const PyTypeInfo& info = vm->_all_types[PK_OBJ_GET(Type, args[0])];
  1468. return VAR(info.name);
  1469. });
  1470. bind_property(_t(tp_type), "__module__", [](VM* vm, ArgsView args){
  1471. const PyTypeInfo& info = vm->_all_types[PK_OBJ_GET(Type, args[0])];
  1472. if(info.mod == nullptr) return vm->None;
  1473. return info.mod;
  1474. });
  1475. bind_property(_t(tp_bound_method), "__self__", [](VM* vm, ArgsView args){
  1476. return CAST(BoundMethod&, args[0]).self;
  1477. });
  1478. bind_property(_t(tp_bound_method), "__func__", [](VM* vm, ArgsView args){
  1479. return CAST(BoundMethod&, args[0]).func;
  1480. });
  1481. bind__eq__(tp_bound_method, [](VM* vm, PyObject* lhs, PyObject* rhs){
  1482. if(!is_non_tagged_type(rhs, vm->tp_bound_method)) return vm->NotImplemented;
  1483. return VAR(_CAST(BoundMethod&, lhs) == _CAST(BoundMethod&, rhs));
  1484. });
  1485. bind_property(_t(tp_slice), "start", [](VM* vm, ArgsView args){
  1486. return CAST(Slice&, args[0]).start;
  1487. });
  1488. bind_property(_t(tp_slice), "stop", [](VM* vm, ArgsView args){
  1489. return CAST(Slice&, args[0]).stop;
  1490. });
  1491. bind_property(_t(tp_slice), "step", [](VM* vm, ArgsView args){
  1492. return CAST(Slice&, args[0]).step;
  1493. });
  1494. bind_property(_t(tp_object), "__dict__", [](VM* vm, ArgsView args){
  1495. if(is_tagged(args[0]) || !args[0]->is_attr_valid()) return vm->None;
  1496. return VAR(MappingProxy(args[0]));
  1497. });
  1498. #if !PK_DEBUG_NO_BUILTINS
  1499. add_module_sys(this);
  1500. add_module_traceback(this);
  1501. add_module_time(this);
  1502. add_module_json(this);
  1503. add_module_math(this);
  1504. add_module_re(this);
  1505. add_module_dis(this);
  1506. add_module_c(this);
  1507. add_module_gc(this);
  1508. add_module_random(this);
  1509. add_module_base64(this);
  1510. add_module_timeit(this);
  1511. for(const char* name: {"this", "functools", "collections", "heapq", "bisect", "pickle", "_long", "colorsys", "typing", "datetime"}){
  1512. _lazy_modules[name] = kPythonLibs[name];
  1513. }
  1514. try{
  1515. CodeObject_ code = compile(kPythonLibs["builtins"], "<builtins>", EXEC_MODE);
  1516. this->_exec(code, this->builtins);
  1517. code = compile(kPythonLibs["_set"], "<set>", EXEC_MODE);
  1518. this->_exec(code, this->builtins);
  1519. }catch(Exception& e){
  1520. std::cerr << e.summary() << std::endl;
  1521. std::cerr << "failed to load builtins module!!" << std::endl;
  1522. exit(1);
  1523. }
  1524. if(enable_os){
  1525. add_module_io(this);
  1526. add_module_os(this);
  1527. _import_handler = _default_import_handler;
  1528. }
  1529. add_module_linalg(this);
  1530. add_module_easing(this);
  1531. #ifdef PK_USE_BOX2D
  1532. add_module_box2d(this);
  1533. #endif
  1534. #endif
  1535. }
  1536. CodeObject_ VM::compile(Str source, Str filename, CompileMode mode, bool unknown_global_scope) {
  1537. Compiler compiler(this, source, filename, mode, unknown_global_scope);
  1538. try{
  1539. return compiler.compile();
  1540. }catch(Exception& e){
  1541. #if PK_DEBUG_FULL_EXCEPTION
  1542. std::cerr << e.summary() << std::endl;
  1543. #endif
  1544. _error(e);
  1545. return nullptr;
  1546. }
  1547. }
  1548. } // namespace pkpy