sqapi.cpp 40 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681
  1. /*
  2. see copyright notice in squirrel.h
  3. */
  4. #include "sqpcheader.h"
  5. #include "sqvm.h"
  6. #include "sqstring.h"
  7. #include "sqtable.h"
  8. #include "sqarray.h"
  9. #include "sqfuncproto.h"
  10. #include "sqclosure.h"
  11. #include "squserdata.h"
  12. #include "sqcompiler.h"
  13. #include "sqfuncstate.h"
  14. #include "sqclass.h"
  15. #include <stdarg.h>
  16. bool sq_aux_gettypedarg(HSQUIRRELVM v,SQInteger idx,SQObjectType type,SQObjectPtr **o)
  17. {
  18. *o = &stack_get(v,idx);
  19. if(type(**o) != type){
  20. SQObjectPtr oval = v->PrintObjVal(**o);
  21. v->Raise_Error(_SC("wrong argument type, expected '%s' got '%.50s'"),IdType2Name(type),_stringval(oval));
  22. return false;
  23. }
  24. return true;
  25. }
  26. #define _GETSAFE_OBJ(v,idx,type,o) { if(!sq_aux_gettypedarg(v,idx,type,&o)) return SQ_ERROR; }
  27. #define sq_aux_paramscheck(v,count) \
  28. { \
  29. if(sq_gettop(v) < count){ v->Raise_Error(_SC("not enough params in the stack")); return SQ_ERROR; }\
  30. }
  31. SQInteger sq_aux_invalidtype(HSQUIRRELVM v,SQObjectType type)
  32. {
  33. SQChar *buf = _ss(v)->GetScratchPad(100);
  34. scsprintf(buf, _SC("unexpected type %s"), IdType2Name(type));
  35. return sq_throwerror(v, buf);
  36. }
  37. HSQUIRRELVM sq_open(SQInteger initialstacksize)
  38. {
  39. SQSharedState *ss;
  40. SQVM *v;
  41. sq_new(ss, SQSharedState);
  42. ss->Init();
  43. v = (SQVM *)SQ_MALLOC(sizeof(SQVM));
  44. new (v) SQVM(ss);
  45. ss->_root_vm = v;
  46. if(v->Init(NULL, initialstacksize)) {
  47. return v;
  48. } else {
  49. sq_delete(v, SQVM);
  50. return NULL;
  51. }
  52. return v;
  53. }
  54. HSQUIRRELVM sq_newthread(HSQUIRRELVM friendvm, SQInteger initialstacksize)
  55. {
  56. SQSharedState *ss;
  57. SQVM *v;
  58. ss=_ss(friendvm);
  59. v= (SQVM *)SQ_MALLOC(sizeof(SQVM));
  60. new (v) SQVM(ss);
  61. if(v->Init(friendvm, initialstacksize)) {
  62. friendvm->Push(v);
  63. return v;
  64. } else {
  65. sq_delete(v, SQVM);
  66. return NULL;
  67. }
  68. }
  69. SQInteger sq_getvmstate(HSQUIRRELVM v)
  70. {
  71. if(v->_suspended)
  72. return SQ_VMSTATE_SUSPENDED;
  73. else {
  74. if(v->_callsstacksize != 0) return SQ_VMSTATE_RUNNING;
  75. else return SQ_VMSTATE_IDLE;
  76. }
  77. }
  78. void sq_seterrorhandler(HSQUIRRELVM v)
  79. {
  80. SQObject o = stack_get(v, -1);
  81. if(sq_isclosure(o) || sq_isnativeclosure(o) || sq_isnull(o)) {
  82. v->_errorhandler = o;
  83. v->Pop();
  84. }
  85. }
  86. void sq_setnativedebughook(HSQUIRRELVM v,SQDEBUGHOOK hook)
  87. {
  88. v->_debughook_native = hook;
  89. v->_debughook_closure.Null();
  90. v->_debughook = hook?true:false;
  91. }
  92. void sq_setdebughook(HSQUIRRELVM v)
  93. {
  94. SQObject o = stack_get(v,-1);
  95. if(sq_isclosure(o) || sq_isnativeclosure(o) || sq_isnull(o)) {
  96. v->_debughook_closure = o;
  97. v->_debughook_native = NULL;
  98. v->_debughook = !sq_isnull(o);
  99. v->Pop();
  100. }
  101. }
  102. void sq_close(HSQUIRRELVM v)
  103. {
  104. SQSharedState *ss = _ss(v);
  105. _thread(ss->_root_vm)->Finalize();
  106. sq_delete(ss, SQSharedState);
  107. }
  108. SQInteger sq_getversion()
  109. {
  110. return SQUIRREL_VERSION_NUMBER;
  111. }
  112. SQRESULT sq_compile(HSQUIRRELVM v,SQLEXREADFUNC read,SQUserPointer p,const SQChar *sourcename,SQBool raiseerror)
  113. {
  114. SQObjectPtr o;
  115. #ifndef NO_COMPILER
  116. if(Compile(v, read, p, sourcename, o, raiseerror?true:false, _ss(v)->_debuginfo)) {
  117. v->Push(SQClosure::Create(_ss(v), _funcproto(o)));
  118. return SQ_OK;
  119. }
  120. return SQ_ERROR;
  121. #else
  122. return sq_throwerror(v,_SC("this is a no compiler build"));
  123. #endif
  124. }
  125. void sq_enabledebuginfo(HSQUIRRELVM v, SQBool enable)
  126. {
  127. _ss(v)->_debuginfo = enable?true:false;
  128. }
  129. void sq_notifyallexceptions(HSQUIRRELVM v, SQBool enable)
  130. {
  131. _ss(v)->_notifyallexceptions = enable?true:false;
  132. }
  133. void sq_addref(HSQUIRRELVM v,HSQOBJECT *po)
  134. {
  135. if(!ISREFCOUNTED(type(*po))) return;
  136. #ifdef NO_GARBAGE_COLLECTOR
  137. __AddRef(po->_type,po->_unVal);
  138. #else
  139. _ss(v)->_refs_table.AddRef(*po);
  140. #endif
  141. }
  142. SQUnsignedInteger sq_getrefcount(HSQUIRRELVM v,HSQOBJECT *po)
  143. {
  144. if(!ISREFCOUNTED(type(*po))) return 0;
  145. #ifdef NO_GARBAGE_COLLECTOR
  146. return po->_unVal.pRefCounted->_uiRef;
  147. #else
  148. return _ss(v)->_refs_table.GetRefCount(*po);
  149. #endif
  150. }
  151. SQBool sq_release(HSQUIRRELVM v,HSQOBJECT *po)
  152. {
  153. if(!ISREFCOUNTED(type(*po))) return SQTrue;
  154. #ifdef NO_GARBAGE_COLLECTOR
  155. bool ret = (po->_unVal.pRefCounted->_uiRef <= 1) ? SQTrue : SQFalse;
  156. __Release(po->_type,po->_unVal);
  157. return ret; //the ret val doesn't work(and cannot be fixed)
  158. #else
  159. return _ss(v)->_refs_table.Release(*po);
  160. #endif
  161. }
  162. const SQChar *sq_objtostring(const HSQOBJECT *o)
  163. {
  164. if(sq_type(*o) == OT_STRING) {
  165. return _stringval(*o);
  166. }
  167. return NULL;
  168. }
  169. SQInteger sq_objtointeger(const HSQOBJECT *o)
  170. {
  171. if(sq_isnumeric(*o)) {
  172. return tointeger(*o);
  173. }
  174. return 0;
  175. }
  176. SQFloat sq_objtofloat(const HSQOBJECT *o)
  177. {
  178. if(sq_isnumeric(*o)) {
  179. return tofloat(*o);
  180. }
  181. return 0;
  182. }
  183. SQBool sq_objtobool(const HSQOBJECT *o)
  184. {
  185. if(sq_isbool(*o)) {
  186. return _integer(*o);
  187. }
  188. return SQFalse;
  189. }
  190. SQUserPointer sq_objtouserpointer(const HSQOBJECT *o)
  191. {
  192. if(sq_isuserpointer(*o)) {
  193. return _userpointer(*o);
  194. }
  195. return 0;
  196. }
  197. void sq_pushnull(HSQUIRRELVM v)
  198. {
  199. v->PushNull();
  200. }
  201. void sq_pushstring(HSQUIRRELVM v,const SQChar *s,SQInteger len)
  202. {
  203. if(s)
  204. v->Push(SQObjectPtr(SQString::Create(_ss(v), s, len)));
  205. else v->PushNull();
  206. }
  207. void sq_pushinteger(HSQUIRRELVM v,SQInteger n)
  208. {
  209. v->Push(n);
  210. }
  211. void sq_pushbool(HSQUIRRELVM v,SQBool b)
  212. {
  213. v->Push(b?true:false);
  214. }
  215. void sq_pushfloat(HSQUIRRELVM v,SQFloat n)
  216. {
  217. v->Push(n);
  218. }
  219. void sq_pushuserpointer(HSQUIRRELVM v,SQUserPointer p)
  220. {
  221. v->Push(p);
  222. }
  223. SQUserPointer sq_newuserdata(HSQUIRRELVM v,SQUnsignedInteger size)
  224. {
  225. SQUserData *ud = SQUserData::Create(_ss(v), size + SQ_ALIGNMENT);
  226. v->Push(ud);
  227. return (SQUserPointer)sq_aligning(ud + 1);
  228. }
  229. void sq_newtable(HSQUIRRELVM v)
  230. {
  231. v->Push(SQTable::Create(_ss(v), 0));
  232. }
  233. void sq_newtableex(HSQUIRRELVM v,SQInteger initialcapacity)
  234. {
  235. v->Push(SQTable::Create(_ss(v), initialcapacity));
  236. }
  237. void sq_newarray(HSQUIRRELVM v,SQInteger size)
  238. {
  239. v->Push(SQArray::Create(_ss(v), size));
  240. }
  241. SQRESULT sq_newclass(HSQUIRRELVM v,SQBool hasbase)
  242. {
  243. SQClass *baseclass = NULL;
  244. if(hasbase) {
  245. SQObjectPtr &base = stack_get(v,-1);
  246. if(type(base) != OT_CLASS)
  247. return sq_throwerror(v,_SC("invalid base type"));
  248. baseclass = _class(base);
  249. }
  250. SQClass *newclass = SQClass::Create(_ss(v), baseclass);
  251. if(baseclass) v->Pop();
  252. v->Push(newclass);
  253. return SQ_OK;
  254. }
  255. SQRESULT sq_pushnewclass(HSQUIRRELVM v, const SQChar *className,
  256. const SQChar *parentName,
  257. void *classTag, SQRegFunction *methods){
  258. if(!className || !classTag)
  259. return sq_throwerror(v, _SC("Missing base class name or class tag."));
  260. sq_pushstring(v,className,-1);
  261. if(parentName){
  262. sq_pushstring(v, parentName,-1);
  263. if (SQ_FAILED(sq_get(v, -3)))
  264. return sq_throwerror(v, _SC("Missing base class \"%s\" for \"%s\"."), parentName, className);
  265. }
  266. sq_newclass(v,SQTrue);
  267. sq_settypetag(v,-1,classTag);
  268. if(methods) sq_insert_reg_funcs(v, methods);
  269. return sq_newslot(v, -3, parentName ? SQTrue : SQFalse);
  270. }
  271. SQBool sq_instanceof(HSQUIRRELVM v)
  272. {
  273. SQObjectPtr &inst = stack_get(v,-1);
  274. SQObjectPtr &cl = stack_get(v,-2);
  275. if(type(inst) != OT_INSTANCE || type(cl) != OT_CLASS)
  276. return sq_throwerror(v,_SC("invalid param type"));
  277. return _instance(inst)->InstanceOf(_class(cl))?SQTrue:SQFalse;
  278. }
  279. SQRESULT sq_arrayappend(HSQUIRRELVM v,SQInteger idx)
  280. {
  281. sq_aux_paramscheck(v,2);
  282. SQObjectPtr *arr;
  283. _GETSAFE_OBJ(v, idx, OT_ARRAY,arr);
  284. _array(*arr)->Append(v->GetUp(-1));
  285. v->Pop();
  286. return SQ_OK;
  287. }
  288. SQRESULT sq_arraypop(HSQUIRRELVM v,SQInteger idx,SQBool pushval)
  289. {
  290. sq_aux_paramscheck(v, 1);
  291. SQObjectPtr *arr;
  292. _GETSAFE_OBJ(v, idx, OT_ARRAY,arr);
  293. if(_array(*arr)->Size() > 0) {
  294. if(pushval != 0){ v->Push(_array(*arr)->Top()); }
  295. _array(*arr)->Pop();
  296. return SQ_OK;
  297. }
  298. return sq_throwerror(v, _SC("empty array"));
  299. }
  300. SQRESULT sq_arrayresize(HSQUIRRELVM v,SQInteger idx,SQInteger newsize)
  301. {
  302. sq_aux_paramscheck(v,1);
  303. SQObjectPtr *arr;
  304. _GETSAFE_OBJ(v, idx, OT_ARRAY,arr);
  305. if(newsize >= 0) {
  306. _array(*arr)->Resize(newsize);
  307. return SQ_OK;
  308. }
  309. return sq_throwerror(v,_SC("negative size"));
  310. }
  311. SQRESULT sq_arrayreverse(HSQUIRRELVM v,SQInteger idx)
  312. {
  313. sq_aux_paramscheck(v, 1);
  314. SQObjectPtr *o;
  315. _GETSAFE_OBJ(v, idx, OT_ARRAY,o);
  316. SQArray *arr = _array(*o);
  317. if(arr->Size() > 0) {
  318. SQObjectPtr t;
  319. SQInteger size = arr->Size();
  320. SQInteger n = size >> 1; size -= 1;
  321. for(SQInteger i = 0; i < n; i++) {
  322. t = arr->_values[i];
  323. arr->_values[i] = arr->_values[size-i];
  324. arr->_values[size-i] = t;
  325. }
  326. return SQ_OK;
  327. }
  328. return SQ_OK;
  329. }
  330. SQRESULT sq_arrayremove(HSQUIRRELVM v,SQInteger idx,SQInteger itemidx)
  331. {
  332. sq_aux_paramscheck(v, 1);
  333. SQObjectPtr *arr;
  334. _GETSAFE_OBJ(v, idx, OT_ARRAY,arr);
  335. return _array(*arr)->Remove(itemidx) ? SQ_OK : sq_throwerror(v,_SC("index out of range"));
  336. }
  337. SQRESULT sq_arrayinsert(HSQUIRRELVM v,SQInteger idx,SQInteger destpos)
  338. {
  339. sq_aux_paramscheck(v, 1);
  340. SQObjectPtr *arr;
  341. _GETSAFE_OBJ(v, idx, OT_ARRAY,arr);
  342. SQRESULT ret = _array(*arr)->Insert(destpos, v->GetUp(-1)) ? SQ_OK : sq_throwerror(v,_SC("index out of range"));
  343. v->Pop();
  344. return ret;
  345. }
  346. void sq_newclosure(HSQUIRRELVM v,SQFUNCTION func,SQUnsignedInteger nfreevars)
  347. {
  348. SQNativeClosure *nc = SQNativeClosure::Create(_ss(v), func,nfreevars);
  349. nc->_nparamscheck = 0;
  350. for(SQUnsignedInteger i = 0; i < nfreevars; i++) {
  351. nc->_outervalues[i] = v->Top();
  352. v->Pop();
  353. }
  354. v->Push(SQObjectPtr(nc));
  355. }
  356. SQRESULT sq_getclosureinfo(HSQUIRRELVM v,SQInteger idx,SQUnsignedInteger *nparams,SQUnsignedInteger *nfreevars)
  357. {
  358. SQObject o = stack_get(v, idx);
  359. if(type(o) == OT_CLOSURE) {
  360. SQClosure *c = _closure(o);
  361. SQFunctionProto *proto = c->_function;
  362. *nparams = (SQUnsignedInteger)proto->_nparameters;
  363. *nfreevars = (SQUnsignedInteger)proto->_noutervalues;
  364. return SQ_OK;
  365. }
  366. else if(type(o) == OT_NATIVECLOSURE)
  367. {
  368. SQNativeClosure *c = _nativeclosure(o);
  369. *nparams = (SQUnsignedInteger)c->_nparamscheck;
  370. *nfreevars = c->_noutervalues;
  371. return SQ_OK;
  372. }
  373. return sq_throwerror(v,_SC("the object is not a closure"));
  374. }
  375. SQRESULT sq_setnativeclosurename(HSQUIRRELVM v,SQInteger idx,const SQChar *name)
  376. {
  377. SQObject o = stack_get(v, idx);
  378. if(sq_isnativeclosure(o)) {
  379. SQNativeClosure *nc = _nativeclosure(o);
  380. nc->_name = SQString::Create(_ss(v),name);
  381. return SQ_OK;
  382. }
  383. return sq_throwerror(v,_SC("the object is not a nativeclosure"));
  384. }
  385. SQRESULT sq_setparamscheck(HSQUIRRELVM v,SQInteger nparamscheck,const SQChar *typemask)
  386. {
  387. SQObject o = stack_get(v, -1);
  388. if(!sq_isnativeclosure(o))
  389. return sq_throwerror(v, _SC("native closure expected"));
  390. SQNativeClosure *nc = _nativeclosure(o);
  391. nc->_nparamscheck = nparamscheck;
  392. if(typemask) {
  393. SQIntVec res;
  394. if(!CompileTypemask(res, typemask))
  395. return sq_throwerror(v, _SC("invalid typemask"));
  396. nc->_typecheck.copy(res);
  397. }
  398. else {
  399. nc->_typecheck.resize(0);
  400. }
  401. if(nparamscheck == SQ_MATCHTYPEMASKSTRING) {
  402. nc->_nparamscheck = nc->_typecheck.size();
  403. }
  404. return SQ_OK;
  405. }
  406. SQRESULT sq_bindenv(HSQUIRRELVM v,SQInteger idx)
  407. {
  408. SQObjectPtr &o = stack_get(v,idx);
  409. if(!sq_isnativeclosure(o) &&
  410. !sq_isclosure(o))
  411. return sq_throwerror(v,_SC("the target is not a closure"));
  412. SQObjectPtr &env = stack_get(v,-1);
  413. if(!sq_istable(env) &&
  414. !sq_isclass(env) &&
  415. !sq_isinstance(env))
  416. return sq_throwerror(v,_SC("invalid environment"));
  417. SQWeakRef *w = _refcounted(env)->GetWeakRef(type(env));
  418. SQObjectPtr ret;
  419. if(sq_isclosure(o)) {
  420. SQClosure *c = _closure(o)->Clone();
  421. __ObjRelease(c->_env);
  422. c->_env = w;
  423. __ObjAddRef(c->_env);
  424. if(_closure(o)->_base) {
  425. c->_base = _closure(o)->_base;
  426. __ObjAddRef(c->_base);
  427. }
  428. ret = c;
  429. }
  430. else { //then must be a native closure
  431. SQNativeClosure *c = _nativeclosure(o)->Clone();
  432. __ObjRelease(c->_env);
  433. c->_env = w;
  434. __ObjAddRef(c->_env);
  435. ret = c;
  436. }
  437. v->Pop();
  438. v->Push(ret);
  439. return SQ_OK;
  440. }
  441. SQRESULT sq_getclosurename(HSQUIRRELVM v,SQInteger idx)
  442. {
  443. SQObjectPtr &o = stack_get(v,idx);
  444. if(!sq_isnativeclosure(o) &&
  445. !sq_isclosure(o))
  446. return sq_throwerror(v,_SC("the target is not a closure"));
  447. if(sq_isnativeclosure(o))
  448. {
  449. v->Push(_nativeclosure(o)->_name);
  450. }
  451. else { //closure
  452. v->Push(_closure(o)->_function->_name);
  453. }
  454. return SQ_OK;
  455. }
  456. SQRESULT sq_clear(HSQUIRRELVM v,SQInteger idx)
  457. {
  458. SQObject &o=stack_get(v,idx);
  459. switch(type(o)) {
  460. case OT_TABLE: _table(o)->Clear(); break;
  461. case OT_ARRAY: _array(o)->Resize(0); break;
  462. default:
  463. return sq_throwerror(v, _SC("clear only works on table and array"));
  464. break;
  465. }
  466. return SQ_OK;
  467. }
  468. void sq_pushroottable(HSQUIRRELVM v)
  469. {
  470. v->Push(v->_roottable);
  471. }
  472. SQRESULT sq_getonroottable(HSQUIRRELVM v)
  473. {
  474. SQObjectPtr &obj = v->GetUp(-1);
  475. if(_table(v->_roottable)->Get(obj,obj))
  476. return SQ_OK;
  477. v->Pop();
  478. return SQ_ERROR;
  479. }
  480. SQRESULT sq_setonroottable(HSQUIRRELVM v)
  481. {
  482. SQObjectPtr &key = v->GetUp(-2);
  483. if(type(key) == OT_NULL) {
  484. v->Pop(2);
  485. return sq_throwerror(v, _SC("null key"));
  486. }
  487. _table(v->_roottable)->NewSlot(key, v->GetUp(-1));
  488. v->Pop(2);
  489. return SQ_OK;
  490. }
  491. void sq_pushregistrytable(HSQUIRRELVM v)
  492. {
  493. v->Push(_ss(v)->_registry);
  494. }
  495. SQRESULT sq_getonregistrytable(HSQUIRRELVM v)
  496. {
  497. SQObjectPtr &obj = v->GetUp(-1);
  498. if(_table(_ss(v)->_registry)->Get(obj,obj))
  499. return SQ_OK;
  500. v->Pop();
  501. return SQ_ERROR;
  502. }
  503. SQRESULT sq_setonregistrytable(HSQUIRRELVM v)
  504. {
  505. SQObjectPtr &key = v->GetUp(-2);
  506. if(type(key) == OT_NULL) {
  507. v->Pop(2);
  508. return sq_throwerror(v, _SC("null key"));
  509. }
  510. _table(_ss(v)->_registry)->NewSlot(key, v->GetUp(-1));
  511. v->Pop(2);
  512. return SQ_OK;
  513. }
  514. void sq_pushconsttable(HSQUIRRELVM v)
  515. {
  516. v->Push(_ss(v)->_consts);
  517. }
  518. SQRESULT sq_setroottable(HSQUIRRELVM v)
  519. {
  520. SQObject o = stack_get(v, -1);
  521. if(sq_istable(o) || sq_isnull(o)) {
  522. v->_roottable = o;
  523. v->Pop();
  524. return SQ_OK;
  525. }
  526. return sq_throwerror(v, _SC("ivalid type"));
  527. }
  528. SQRESULT sq_setconsttable(HSQUIRRELVM v)
  529. {
  530. SQObject o = stack_get(v, -1);
  531. if(sq_istable(o)) {
  532. _ss(v)->_consts = o;
  533. v->Pop();
  534. return SQ_OK;
  535. }
  536. return sq_throwerror(v, _SC("ivalid type, expected table"));
  537. }
  538. void sq_setforeignptr(HSQUIRRELVM v,SQUserPointer p)
  539. {
  540. v->_foreignptr = p;
  541. }
  542. SQUserPointer sq_getforeignptr(HSQUIRRELVM v)
  543. {
  544. return v->_foreignptr;
  545. }
  546. void sq_push(HSQUIRRELVM v,SQInteger idx)
  547. {
  548. v->Push(stack_get(v, idx));
  549. }
  550. SQObjectType sq_gettype(HSQUIRRELVM v,SQInteger idx)
  551. {
  552. return type(stack_get(v, idx));
  553. }
  554. SQUIRREL_API const SQChar *sq_gettypename(HSQUIRRELVM v,SQInteger idx){
  555. SQObjectPtr &o = stack_get(v, idx);
  556. return GetTypeName(o);
  557. }
  558. SQRESULT sq_typeof(HSQUIRRELVM v,SQInteger idx)
  559. {
  560. SQObjectPtr &o = stack_get(v, idx);
  561. SQObjectPtr res;
  562. if(!v->TypeOf(o,res)) {
  563. return SQ_ERROR;
  564. }
  565. v->Push(res);
  566. return SQ_OK;
  567. }
  568. SQRESULT sq_tostring(HSQUIRRELVM v,SQInteger idx)
  569. {
  570. SQObjectPtr &o = stack_get(v, idx);
  571. SQObjectPtr res;
  572. if(!v->ToString(o,res)) {
  573. return SQ_ERROR;
  574. }
  575. v->Push(res);
  576. return SQ_OK;
  577. }
  578. void sq_tobool(HSQUIRRELVM v, SQInteger idx, SQBool *b)
  579. {
  580. SQObjectPtr &o = stack_get(v, idx);
  581. *b = SQVM::IsFalse(o)?SQFalse:SQTrue;
  582. }
  583. SQRESULT sq_getinteger(HSQUIRRELVM v,SQInteger idx,SQInteger *i)
  584. {
  585. SQObjectPtr &o = stack_get(v, idx);
  586. if(sq_isnumeric(o)) {
  587. *i = tointeger(o);
  588. return SQ_OK;
  589. }
  590. return SQ_ERROR;
  591. }
  592. SQRESULT sq_getfloat(HSQUIRRELVM v,SQInteger idx,SQFloat *f)
  593. {
  594. SQObjectPtr &o = stack_get(v, idx);
  595. if(sq_isnumeric(o)) {
  596. *f = tofloat(o);
  597. return SQ_OK;
  598. }
  599. return SQ_ERROR;
  600. }
  601. SQRESULT sq_getbool(HSQUIRRELVM v,SQInteger idx,SQBool *b)
  602. {
  603. SQObjectPtr &o = stack_get(v, idx);
  604. if(sq_isbool(o)) {
  605. *b = _integer(o);
  606. return SQ_OK;
  607. }
  608. return SQ_ERROR;
  609. }
  610. SQRESULT sq_getstring(HSQUIRRELVM v,SQInteger idx,const SQChar **c)
  611. {
  612. SQObjectPtr *o = NULL;
  613. _GETSAFE_OBJ(v, idx, OT_STRING,o);
  614. *c = _stringval(*o);
  615. return SQ_OK;
  616. }
  617. SQRESULT sq_getstr_and_size(HSQUIRRELVM v,SQInteger idx,const SQChar **c, SQInteger *size)
  618. {
  619. SQObjectPtr *o = NULL;
  620. _GETSAFE_OBJ(v, idx, OT_STRING,o);
  621. *c = _stringval(*o);
  622. *size = _string(*o)->_len;
  623. return SQ_OK;
  624. }
  625. SQRESULT sq_getthread(HSQUIRRELVM v,SQInteger idx,HSQUIRRELVM *thread)
  626. {
  627. SQObjectPtr *o = NULL;
  628. _GETSAFE_OBJ(v, idx, OT_THREAD,o);
  629. *thread = _thread(*o);
  630. return SQ_OK;
  631. }
  632. SQRESULT sq_clone(HSQUIRRELVM v,SQInteger idx)
  633. {
  634. SQObjectPtr &o = stack_get(v,idx);
  635. v->PushNull();
  636. if(!v->Clone(o, stack_get(v, -1))){
  637. v->Pop();
  638. return SQ_ERROR;
  639. }
  640. return SQ_OK;
  641. }
  642. SQInteger sq_getsize(HSQUIRRELVM v, SQInteger idx)
  643. {
  644. SQObjectPtr &o = stack_get(v, idx);
  645. SQObjectType type = type(o);
  646. switch(type) {
  647. case OT_STRING: return _string(o)->_len;
  648. case OT_TABLE: return _table(o)->CountUsed();
  649. case OT_ARRAY: return _array(o)->Size();
  650. case OT_USERDATA: return _userdata(o)->_size;
  651. case OT_INSTANCE: return _instance(o)->_class->_udsize;
  652. case OT_CLASS: return _class(o)->_udsize;
  653. default:
  654. return sq_aux_invalidtype(v, type);
  655. }
  656. }
  657. SQHash sq_gethash(HSQUIRRELVM v, SQInteger idx)
  658. {
  659. SQObjectPtr &o = stack_get(v, idx);
  660. return HashObj(o);
  661. }
  662. SQRESULT sq_getuserdata(HSQUIRRELVM v,SQInteger idx,SQUserPointer *p,SQUserPointer *typetag)
  663. {
  664. SQObjectPtr *o = NULL;
  665. _GETSAFE_OBJ(v, idx, OT_USERDATA,o);
  666. (*p) = _userdataval(*o);
  667. if(typetag) *typetag = _userdata(*o)->_typetag;
  668. return SQ_OK;
  669. }
  670. SQRESULT sq_settypetag(HSQUIRRELVM v,SQInteger idx,SQUserPointer typetag)
  671. {
  672. SQObjectPtr &o = stack_get(v,idx);
  673. switch(type(o)) {
  674. case OT_USERDATA: _userdata(o)->_typetag = typetag; break;
  675. case OT_CLASS: _class(o)->_typetag = typetag; break;
  676. default: return sq_throwerror(v,_SC("invalid object type"));
  677. }
  678. return SQ_OK;
  679. }
  680. SQRESULT sq_getobjtypetag(const HSQOBJECT *o,SQUserPointer * typetag)
  681. {
  682. switch(type(*o)) {
  683. case OT_INSTANCE: *typetag = _instance(*o)->_class->_typetag; break;
  684. case OT_USERDATA: *typetag = _userdata(*o)->_typetag; break;
  685. case OT_CLASS: *typetag = _class(*o)->_typetag; break;
  686. default: return SQ_ERROR;
  687. }
  688. return SQ_OK;
  689. }
  690. SQRESULT sq_gettypetag(HSQUIRRELVM v,SQInteger idx,SQUserPointer *typetag)
  691. {
  692. SQObjectPtr &o = stack_get(v,idx);
  693. if(SQ_FAILED(sq_getobjtypetag(&o,typetag)))
  694. return sq_throwerror(v,_SC("invalid object type"));
  695. return SQ_OK;
  696. }
  697. SQRESULT sq_getuserpointer(HSQUIRRELVM v, SQInteger idx, SQUserPointer *p)
  698. {
  699. SQObjectPtr *o = NULL;
  700. _GETSAFE_OBJ(v, idx, OT_USERPOINTER,o);
  701. (*p) = _userpointer(*o);
  702. return SQ_OK;
  703. }
  704. SQRESULT sq_setinstanceup(HSQUIRRELVM v, SQInteger idx, SQUserPointer p)
  705. {
  706. SQObjectPtr &o = stack_get(v,idx);
  707. if(type(o) != OT_INSTANCE) return sq_throwerror(v,_SC("the object is not a class instance"));
  708. _instance(o)->_userpointer = p;
  709. return SQ_OK;
  710. }
  711. SQRESULT sq_setclassudsize(HSQUIRRELVM v, SQInteger idx, SQInteger udsize)
  712. {
  713. SQObjectPtr &o = stack_get(v,idx);
  714. if(type(o) != OT_CLASS) return sq_throwerror(v,_SC("the object is not a class"));
  715. if(_class(o)->_locked) return sq_throwerror(v,_SC("the class is locked"));
  716. _class(o)->_udsize = udsize;
  717. return SQ_OK;
  718. }
  719. SQRESULT sq_getinstanceup(HSQUIRRELVM v, SQInteger idx, SQUserPointer *p,SQUserPointer typetag)
  720. {
  721. SQObjectPtr &o = stack_get(v,idx);
  722. if(type(o) != OT_INSTANCE) return sq_throwerror(v,_SC("the object is not a class instance"));
  723. (*p) = _instance(o)->_userpointer;
  724. if(typetag != 0) {
  725. SQClass *cl = _instance(o)->_class;
  726. do{
  727. if(cl->_typetag == typetag)
  728. return SQ_OK;
  729. cl = cl->_base;
  730. }while(cl != NULL);
  731. return sq_throwerror(v,_SC("invalid type tag"));
  732. }
  733. return SQ_OK;
  734. }
  735. SQInteger sq_getfulltop(HSQUIRRELVM v)
  736. {
  737. return v->_top;
  738. }
  739. SQInteger sq_gettop(HSQUIRRELVM v)
  740. {
  741. return (v->_top) - v->_stackbase;
  742. }
  743. void sq_settop(HSQUIRRELVM v, SQInteger newtop)
  744. {
  745. SQInteger top = sq_gettop(v);
  746. if(top > newtop)
  747. sq_pop(v, top - newtop);
  748. else
  749. while(top++ < newtop) sq_pushnull(v);
  750. }
  751. void sq_pop(HSQUIRRELVM v, SQInteger nelemstopop)
  752. {
  753. assert(v->_top >= nelemstopop);
  754. v->Pop(nelemstopop);
  755. }
  756. void sq_poptop(HSQUIRRELVM v)
  757. {
  758. assert(v->_top >= 1);
  759. v->Pop();
  760. }
  761. void sq_remove(HSQUIRRELVM v, SQInteger idx)
  762. {
  763. v->Remove(idx);
  764. }
  765. void sq_insert(HSQUIRRELVM v, SQInteger idx)
  766. {
  767. v->Insert(idx);
  768. }
  769. void sq_replace(HSQUIRRELVM v, SQInteger idx)
  770. {
  771. v->Replace(idx);
  772. }
  773. SQInteger sq_cmp(HSQUIRRELVM v)
  774. {
  775. SQInteger res;
  776. v->ObjCmp(stack_get(v, -1), stack_get(v, -2),res);
  777. return res;
  778. }
  779. SQRESULT sq_newslot(HSQUIRRELVM v, SQInteger idx, SQBool bstatic)
  780. {
  781. sq_aux_paramscheck(v, 3);
  782. SQObjectPtr &self = stack_get(v, idx);
  783. if(type(self) == OT_TABLE || type(self) == OT_CLASS) {
  784. SQObjectPtr &key = v->GetUp(-2);
  785. if(type(key) == OT_NULL) return sq_throwerror(v, _SC("null is not a valid key"));
  786. v->NewSlot(self, key, v->GetUp(-1),bstatic?true:false);
  787. v->Pop(2);
  788. }
  789. return SQ_OK;
  790. }
  791. SQRESULT sq_deleteslot(HSQUIRRELVM v,SQInteger idx,SQBool pushval)
  792. {
  793. sq_aux_paramscheck(v, 2);
  794. SQObjectPtr *self;
  795. _GETSAFE_OBJ(v, idx, OT_TABLE,self);
  796. SQObjectPtr &key = v->GetUp(-1);
  797. if(type(key) == OT_NULL) return sq_throwerror(v, _SC("null is not a valid key"));
  798. SQObjectPtr res;
  799. if(!v->DeleteSlot(*self, key, res)){
  800. v->Pop();
  801. return SQ_ERROR;
  802. }
  803. if(pushval) v->GetUp(-1) = res;
  804. else v->Pop();
  805. return SQ_OK;
  806. }
  807. SQRESULT sq_set(HSQUIRRELVM v,SQInteger idx)
  808. {
  809. SQObjectPtr &self = stack_get(v, idx);
  810. if(v->Set(self, v->GetUp(-2), v->GetUp(-1),DONT_FALL_BACK)) {
  811. v->Pop(2);
  812. return SQ_OK;
  813. }
  814. return SQ_ERROR;
  815. }
  816. SQRESULT sq_rawset(HSQUIRRELVM v,SQInteger idx)
  817. {
  818. SQObjectPtr &self = stack_get(v, idx);
  819. SQObjectPtr &key = v->GetUp(-2);
  820. if(type(key) == OT_NULL) {
  821. v->Pop(2);
  822. return sq_throwerror(v, _SC("null key"));
  823. }
  824. switch(type(self)) {
  825. case OT_TABLE:
  826. _table(self)->NewSlot(key, v->GetUp(-1));
  827. v->Pop(2);
  828. return SQ_OK;
  829. break;
  830. case OT_CLASS:
  831. _class(self)->NewSlot(_ss(v), key, v->GetUp(-1),false);
  832. v->Pop(2);
  833. return SQ_OK;
  834. break;
  835. case OT_INSTANCE:
  836. if(_instance(self)->Set(key, v->GetUp(-1))) {
  837. v->Pop(2);
  838. return SQ_OK;
  839. }
  840. break;
  841. case OT_ARRAY:
  842. if(v->Set(self, key, v->GetUp(-1),false)) {
  843. v->Pop(2);
  844. return SQ_OK;
  845. }
  846. break;
  847. default:
  848. v->Pop(2);
  849. return sq_throwerror(v, _SC("rawset works only on array/table/class and instance"));
  850. }
  851. v->Raise_IdxError(v->GetUp(-2));return SQ_ERROR;
  852. }
  853. SQRESULT sq_newmember(HSQUIRRELVM v,SQInteger idx,SQBool bstatic)
  854. {
  855. SQObjectPtr &self = stack_get(v, idx);
  856. if(type(self) != OT_CLASS) return sq_throwerror(v, _SC("new member only works with classes"));
  857. SQObjectPtr &key = v->GetUp(-3);
  858. if(type(key) == OT_NULL) return sq_throwerror(v, _SC("null key"));
  859. if(!v->NewSlotA(self,key,v->GetUp(-2),v->GetUp(-1),bstatic?true:false,false))
  860. return SQ_ERROR;
  861. return SQ_OK;
  862. }
  863. SQRESULT sq_rawnewmember(HSQUIRRELVM v,SQInteger idx,SQBool bstatic)
  864. {
  865. SQObjectPtr &self = stack_get(v, idx);
  866. if(type(self) != OT_CLASS) return sq_throwerror(v, _SC("new member only works with classes"));
  867. SQObjectPtr &key = v->GetUp(-3);
  868. if(type(key) == OT_NULL) return sq_throwerror(v, _SC("null key"));
  869. if(!v->NewSlotA(self,key,v->GetUp(-2),v->GetUp(-1),bstatic?true:false,true))
  870. return SQ_ERROR;
  871. return SQ_OK;
  872. }
  873. SQRESULT sq_setdelegate(HSQUIRRELVM v,SQInteger idx)
  874. {
  875. SQObjectPtr &self = stack_get(v, idx);
  876. SQObjectPtr &mt = v->GetUp(-1);
  877. SQObjectType type = type(self);
  878. switch(type) {
  879. case OT_TABLE:
  880. if(type(mt) == OT_TABLE) {
  881. if(!_table(self)->SetDelegate(_table(mt))) return sq_throwerror(v, _SC("delagate cycle")); v->Pop();}
  882. else if(type(mt)==OT_NULL) {
  883. _table(self)->SetDelegate(NULL); v->Pop(); }
  884. else return sq_aux_invalidtype(v,type);
  885. break;
  886. case OT_USERDATA:
  887. if(type(mt)==OT_TABLE) {
  888. _userdata(self)->SetDelegate(_table(mt)); v->Pop(); }
  889. else if(type(mt)==OT_NULL) {
  890. _userdata(self)->SetDelegate(NULL); v->Pop(); }
  891. else return sq_aux_invalidtype(v, type);
  892. break;
  893. default:
  894. return sq_aux_invalidtype(v, type);
  895. break;
  896. }
  897. return SQ_OK;
  898. }
  899. SQRESULT sq_rawdeleteslot(HSQUIRRELVM v,SQInteger idx,SQBool pushval)
  900. {
  901. sq_aux_paramscheck(v, 2);
  902. SQObjectPtr *self;
  903. _GETSAFE_OBJ(v, idx, OT_TABLE,self);
  904. SQObjectPtr &key = v->GetUp(-1);
  905. SQObjectPtr t;
  906. if(_table(*self)->Get(key,t)) {
  907. _table(*self)->Remove(key);
  908. }
  909. if(pushval != 0)
  910. v->GetUp(-1) = t;
  911. else
  912. v->Pop();
  913. return SQ_OK;
  914. }
  915. SQRESULT sq_getdelegate(HSQUIRRELVM v,SQInteger idx)
  916. {
  917. SQObjectPtr &self=stack_get(v,idx);
  918. switch(type(self)){
  919. case OT_TABLE:
  920. case OT_USERDATA:
  921. if(!_delegable(self)->_delegate){
  922. v->PushNull();
  923. break;
  924. }
  925. v->Push(SQObjectPtr(_delegable(self)->_delegate));
  926. break;
  927. default: return sq_throwerror(v,_SC("wrong type")); break;
  928. }
  929. return SQ_OK;
  930. }
  931. SQRESULT sq_get(HSQUIRRELVM v,SQInteger idx)
  932. {
  933. SQObjectPtr &self=stack_get(v,idx);
  934. SQObjectPtr &obj = v->GetUp(-1);
  935. if(v->Get(self,obj,obj,false,DONT_FALL_BACK))
  936. return SQ_OK;
  937. v->Pop();
  938. return SQ_ERROR;
  939. }
  940. SQRESULT sq_getbyname(HSQUIRRELVM v,SQInteger idx, const SQChar *key, SQInteger key_len)
  941. {
  942. SQObjectPtr &self=stack_get(v,idx);
  943. sq_pushstring(v, key, key_len);
  944. SQObjectPtr &obj = v->GetUp(-1);
  945. switch(type(self)) {
  946. case OT_TABLE:
  947. if(_table(self)->Get(obj,obj))
  948. return SQ_OK;
  949. break;
  950. case OT_CLASS:
  951. if(_class(self)->Get(obj,obj))
  952. return SQ_OK;
  953. break;
  954. case OT_INSTANCE:
  955. if(_instance(self)->Get(obj,obj))
  956. return SQ_OK;
  957. break;
  958. default:
  959. sq_poptop(v);
  960. return sq_throwerror(v,_SC("sq_getbyname works only on table/instance and class"));
  961. }
  962. sq_poptop(v);
  963. return sq_throwerror(v,_SC("the index doesn't exist"));
  964. }
  965. SQRESULT sq_rawget(HSQUIRRELVM v,SQInteger idx)
  966. {
  967. SQObjectPtr &self=stack_get(v,idx);
  968. SQObjectPtr &obj = v->GetUp(-1);
  969. switch(type(self)) {
  970. case OT_TABLE:
  971. if(_table(self)->Get(obj,obj))
  972. return SQ_OK;
  973. break;
  974. case OT_CLASS:
  975. if(_class(self)->Get(obj,obj))
  976. return SQ_OK;
  977. break;
  978. case OT_INSTANCE:
  979. if(_instance(self)->Get(obj,obj))
  980. return SQ_OK;
  981. break;
  982. case OT_ARRAY:{
  983. if(sq_isnumeric(obj)){
  984. if(_array(self)->Get(tointeger(obj),obj)) {
  985. return SQ_OK;
  986. }
  987. }
  988. else {
  989. v->Pop();
  990. return sq_throwerror(v,_SC("invalid index type for an array"));
  991. }
  992. }
  993. break;
  994. default:
  995. v->Pop();
  996. return sq_throwerror(v,_SC("rawget works only on array/table/instance and class"));
  997. }
  998. v->Pop();
  999. return sq_throwerror(v,_SC("the index doesn't exist"));
  1000. }
  1001. SQRESULT sq_getstackobj(HSQUIRRELVM v,SQInteger idx,HSQOBJECT *po)
  1002. {
  1003. *po=stack_get(v,idx);
  1004. return SQ_OK;
  1005. }
  1006. const SQChar *sq_getlocal(HSQUIRRELVM v,SQUnsignedInteger level,SQUnsignedInteger idx)
  1007. {
  1008. SQUnsignedInteger cstksize=v->_callsstacksize;
  1009. SQUnsignedInteger lvl=(cstksize-level)-1;
  1010. SQInteger stackbase=v->_stackbase;
  1011. if(lvl<cstksize){
  1012. for(SQUnsignedInteger i=0;i<level;i++){
  1013. SQVM::CallInfo &ci=v->_callsstack[(cstksize-i)-1];
  1014. stackbase-=ci._prevstkbase;
  1015. }
  1016. SQVM::CallInfo &ci=v->_callsstack[lvl];
  1017. if(type(ci._closure)!=OT_CLOSURE)
  1018. return NULL;
  1019. SQClosure *c=_closure(ci._closure);
  1020. SQFunctionProto *func=c->_function;
  1021. if(func->_noutervalues > (SQInteger)idx) {
  1022. v->Push(*_outer(c->_outervalues[idx])->_valptr);
  1023. return _stringval(func->_outervalues[idx]._name);
  1024. }
  1025. idx -= func->_noutervalues;
  1026. return func->GetLocal(v,stackbase,idx,(SQInteger)(ci._ip-func->_instructions)-1);
  1027. }
  1028. return NULL;
  1029. }
  1030. void sq_pushobject(HSQUIRRELVM v,HSQOBJECT obj)
  1031. {
  1032. v->Push(SQObjectPtr(obj));
  1033. }
  1034. void sq_resetobject(HSQOBJECT *po)
  1035. {
  1036. po->_unVal.pUserPointer=NULL;po->_type=OT_NULL;
  1037. }
  1038. SQRESULT sq_throwerror(HSQUIRRELVM v,const SQChar *fmt, ...)
  1039. {
  1040. static SQChar err[256];
  1041. va_list vl;
  1042. va_start(vl, fmt);
  1043. scvsnprintf(err, sizeof(err), fmt, vl);
  1044. va_end(vl);
  1045. v->_lasterror=SQString::Create(_ss(v),err);
  1046. return SQ_ERROR;
  1047. }
  1048. SQRESULT sq_throwobject(HSQUIRRELVM v)
  1049. {
  1050. v->_lasterror = v->GetUp(-1);
  1051. v->Pop();
  1052. return SQ_ERROR;
  1053. }
  1054. void sq_reseterror(HSQUIRRELVM v)
  1055. {
  1056. v->_lasterror.Null();
  1057. }
  1058. void sq_getlasterror(HSQUIRRELVM v)
  1059. {
  1060. v->Push(v->_lasterror);
  1061. }
  1062. const SQChar *sq_getlasterror_str(HSQUIRRELVM v)
  1063. {
  1064. return _string(v->_lasterror)->_val;
  1065. }
  1066. SQRESULT sq_reservestack(HSQUIRRELVM v,SQInteger nsize)
  1067. {
  1068. if (((SQUnsignedInteger)v->_top + nsize) > v->_stack.size()) {
  1069. if(v->_nmetamethodscall) {
  1070. return sq_throwerror(v,_SC("cannot resize stack while in a metamethod"));
  1071. }
  1072. v->_stack.resize(v->_stack.size() + ((v->_top + nsize) - v->_stack.size()));
  1073. }
  1074. return SQ_OK;
  1075. }
  1076. SQRESULT sq_resume(HSQUIRRELVM v,SQBool retval,SQBool raiseerror)
  1077. {
  1078. SQObjectPtr &obj = v->GetUp(-1);
  1079. if(type(obj)==OT_GENERATOR){
  1080. v->PushNull(); //retval
  1081. if(!v->Execute(v->GetUp(-2),0,v->_top,obj,raiseerror,SQVM::ET_RESUME_GENERATOR))
  1082. {v->Raise_Error(v->_lasterror); return SQ_ERROR;}
  1083. if(!retval)
  1084. v->Pop();
  1085. return SQ_OK;
  1086. }
  1087. return sq_throwerror(v,_SC("only generators can be resumed"));
  1088. }
  1089. SQRESULT sq_call(HSQUIRRELVM v,SQInteger params,SQBool retval,SQBool raiseerror)
  1090. {
  1091. SQObjectPtr res;
  1092. if(v->Call(v->GetUp(-(params+1)),params,v->_top-params,res,raiseerror?true:false)){
  1093. if(!v->_suspended) {
  1094. v->Pop(params);//pop closure and args
  1095. }
  1096. if(retval){
  1097. v->Push(res); return SQ_OK;
  1098. }
  1099. return SQ_OK;
  1100. }
  1101. else {
  1102. v->Pop(params);
  1103. return SQ_ERROR;
  1104. }
  1105. if(!v->_suspended)
  1106. v->Pop(params);
  1107. return sq_throwerror(v,_SC("call failed"));
  1108. }
  1109. SQRESULT sq_suspendvm(HSQUIRRELVM v)
  1110. {
  1111. return v->Suspend();
  1112. }
  1113. SQRESULT sq_wakeupvm(HSQUIRRELVM v,SQBool wakeupret,SQBool retval,SQBool raiseerror,SQBool throwerror)
  1114. {
  1115. SQObjectPtr ret;
  1116. if(!v->_suspended)
  1117. return sq_throwerror(v,_SC("cannot resume a vm that is not running any code"));
  1118. SQInteger target = v->_suspended_target;
  1119. if(wakeupret) {
  1120. if(target != -1) {
  1121. v->GetAt(v->_stackbase+v->_suspended_target)=v->GetUp(-1); //retval
  1122. }
  1123. v->Pop();
  1124. } else if(target != -1) { v->GetAt(v->_stackbase+v->_suspended_target).Null(); }
  1125. SQObjectPtr dummy;
  1126. if(!v->Execute(dummy,-1,-1,ret,raiseerror,throwerror?SQVM::ET_RESUME_THROW_VM : SQVM::ET_RESUME_VM)) {
  1127. return SQ_ERROR;
  1128. }
  1129. if(retval)
  1130. v->Push(ret);
  1131. return SQ_OK;
  1132. }
  1133. void sq_setreleasehook(HSQUIRRELVM v,SQInteger idx,SQRELEASEHOOK hook)
  1134. {
  1135. if(sq_gettop(v) >= 1){
  1136. SQObjectPtr &ud=stack_get(v,idx);
  1137. switch( type(ud) ) {
  1138. case OT_USERDATA: _userdata(ud)->_hook = hook; break;
  1139. case OT_INSTANCE: _instance(ud)->_hook = hook; break;
  1140. case OT_CLASS: _class(ud)->_hook = hook; break;
  1141. default: break; //shutup compiler
  1142. }
  1143. }
  1144. }
  1145. void sq_setcompilererrorhandler(HSQUIRRELVM v,SQCOMPILERERROR f)
  1146. {
  1147. _ss(v)->_compilererrorhandler = f;
  1148. }
  1149. SQRESULT sq_writeclosure(HSQUIRRELVM v,SQWRITEFUNC w,SQUserPointer up)
  1150. {
  1151. SQObjectPtr *o = NULL;
  1152. _GETSAFE_OBJ(v, -1, OT_CLOSURE,o);
  1153. unsigned short tag = SQ_BYTECODE_STREAM_TAG;
  1154. if(_closure(*o)->_function->_noutervalues)
  1155. return sq_throwerror(v,_SC("a closure with free valiables bound it cannot be serialized"));
  1156. if(w(up,&tag,2) != 2)
  1157. return sq_throwerror(v,_SC("io error"));
  1158. if(!_closure(*o)->Save(v,up,w))
  1159. return SQ_ERROR;
  1160. return SQ_OK;
  1161. }
  1162. SQRESULT sq_readclosure(HSQUIRRELVM v,SQREADFUNC r,SQUserPointer up)
  1163. {
  1164. SQObjectPtr closure;
  1165. unsigned short tag;
  1166. if(r(up,&tag,2) != 2)
  1167. return sq_throwerror(v,_SC("io error"));
  1168. if(tag != SQ_BYTECODE_STREAM_TAG)
  1169. return sq_throwerror(v,_SC("invalid stream"));
  1170. if(!SQClosure::Load(v,up,r,closure))
  1171. return SQ_ERROR;
  1172. v->Push(closure);
  1173. return SQ_OK;
  1174. }
  1175. SQChar *sq_getscratchpad(HSQUIRRELVM v,SQInteger minsize)
  1176. {
  1177. return _ss(v)->GetScratchPad(minsize);
  1178. }
  1179. SQRESULT sq_resurrectunreachable(HSQUIRRELVM v)
  1180. {
  1181. #ifndef NO_GARBAGE_COLLECTOR
  1182. _ss(v)->ResurrectUnreachable(v);
  1183. return SQ_OK;
  1184. #else
  1185. return sq_throwerror(v,_SC("sq_resurrectunreachable requires a garbage collector build"));
  1186. #endif
  1187. }
  1188. SQInteger sq_collectgarbage(HSQUIRRELVM v)
  1189. {
  1190. #ifndef NO_GARBAGE_COLLECTOR
  1191. return _ss(v)->CollectGarbage(v);
  1192. #else
  1193. return -1;
  1194. #endif
  1195. }
  1196. SQRESULT sq_getcallee(HSQUIRRELVM v)
  1197. {
  1198. if(v->_callsstacksize > 1)
  1199. {
  1200. v->Push(v->_callsstack[v->_callsstacksize - 2]._closure);
  1201. return SQ_OK;
  1202. }
  1203. return sq_throwerror(v,_SC("no closure in the calls stack"));
  1204. }
  1205. const SQChar *sq_getfreevariable(HSQUIRRELVM v,SQInteger idx,SQUnsignedInteger nval)
  1206. {
  1207. SQObjectPtr &self=stack_get(v,idx);
  1208. const SQChar *name = NULL;
  1209. switch(type(self))
  1210. {
  1211. case OT_CLOSURE:{
  1212. SQClosure *clo = _closure(self);
  1213. SQFunctionProto *fp = clo->_function;
  1214. if(((SQUnsignedInteger)fp->_noutervalues) > nval) {
  1215. v->Push(*(_outer(clo->_outervalues[nval])->_valptr));
  1216. SQOuterVar &ov = fp->_outervalues[nval];
  1217. name = _stringval(ov._name);
  1218. }
  1219. }
  1220. break;
  1221. case OT_NATIVECLOSURE:{
  1222. SQNativeClosure *clo = _nativeclosure(self);
  1223. if(clo->_noutervalues > nval) {
  1224. v->Push(clo->_outervalues[nval]);
  1225. name = _SC("@NATIVE");
  1226. }
  1227. }
  1228. break;
  1229. default: break; //shutup compiler
  1230. }
  1231. return name;
  1232. }
  1233. SQRESULT sq_setfreevariable(HSQUIRRELVM v,SQInteger idx,SQUnsignedInteger nval)
  1234. {
  1235. SQObjectPtr &self=stack_get(v,idx);
  1236. switch(type(self))
  1237. {
  1238. case OT_CLOSURE:{
  1239. SQFunctionProto *fp = _closure(self)->_function;
  1240. if(((SQUnsignedInteger)fp->_noutervalues) > nval){
  1241. *(_outer(_closure(self)->_outervalues[nval])->_valptr) = stack_get(v,-1);
  1242. }
  1243. else return sq_throwerror(v,_SC("invalid free var index"));
  1244. }
  1245. break;
  1246. case OT_NATIVECLOSURE:
  1247. if(_nativeclosure(self)->_noutervalues > nval){
  1248. _nativeclosure(self)->_outervalues[nval] = stack_get(v,-1);
  1249. }
  1250. else return sq_throwerror(v,_SC("invalid free var index"));
  1251. break;
  1252. default:
  1253. return sq_aux_invalidtype(v,type(self));
  1254. }
  1255. v->Pop();
  1256. return SQ_OK;
  1257. }
  1258. SQRESULT sq_setattributes(HSQUIRRELVM v,SQInteger idx)
  1259. {
  1260. SQObjectPtr *o = NULL;
  1261. _GETSAFE_OBJ(v, idx, OT_CLASS,o);
  1262. SQObjectPtr &key = stack_get(v,-2);
  1263. SQObjectPtr &val = stack_get(v,-1);
  1264. SQObjectPtr attrs;
  1265. if(type(key) == OT_NULL) {
  1266. attrs = _class(*o)->_attributes;
  1267. _class(*o)->_attributes = val;
  1268. v->Pop(2);
  1269. v->Push(attrs);
  1270. return SQ_OK;
  1271. }else if(_class(*o)->GetAttributes(key,attrs)) {
  1272. _class(*o)->SetAttributes(key,val);
  1273. v->Pop(2);
  1274. v->Push(attrs);
  1275. return SQ_OK;
  1276. }
  1277. return sq_throwerror(v,_SC("wrong index"));
  1278. }
  1279. SQRESULT sq_getattributes(HSQUIRRELVM v,SQInteger idx)
  1280. {
  1281. SQObjectPtr *o = NULL;
  1282. _GETSAFE_OBJ(v, idx, OT_CLASS,o);
  1283. SQObjectPtr &key = stack_get(v,-1);
  1284. SQObjectPtr attrs;
  1285. if(type(key) == OT_NULL) {
  1286. attrs = _class(*o)->_attributes;
  1287. v->Pop();
  1288. v->Push(attrs);
  1289. return SQ_OK;
  1290. }
  1291. else if(_class(*o)->GetAttributes(key,attrs)) {
  1292. v->Pop();
  1293. v->Push(attrs);
  1294. return SQ_OK;
  1295. }
  1296. return sq_throwerror(v,_SC("wrong index"));
  1297. }
  1298. SQRESULT sq_getmemberhandle(HSQUIRRELVM v,SQInteger idx,HSQMEMBERHANDLE *handle)
  1299. {
  1300. SQObjectPtr *o = NULL;
  1301. _GETSAFE_OBJ(v, idx, OT_CLASS,o);
  1302. SQObjectPtr &key = stack_get(v,-1);
  1303. SQTable *m = _class(*o)->_members;
  1304. SQObjectPtr val;
  1305. if(m->Get(key,val)) {
  1306. handle->_static = _isfield(val) ? SQFalse : SQTrue;
  1307. handle->_index = _member_idx(val);
  1308. v->Pop();
  1309. return SQ_OK;
  1310. }
  1311. return sq_throwerror(v,_SC("wrong index"));
  1312. }
  1313. SQRESULT _getmemberbyhandle(HSQUIRRELVM v,SQObjectPtr &self,const HSQMEMBERHANDLE *handle,SQObjectPtr *&val)
  1314. {
  1315. switch(type(self)) {
  1316. case OT_INSTANCE: {
  1317. SQInstance *i = _instance(self);
  1318. if(handle->_static) {
  1319. SQClass *c = i->_class;
  1320. val = &c->_methods[handle->_index].val;
  1321. }
  1322. else {
  1323. val = &i->_values[handle->_index];
  1324. }
  1325. }
  1326. break;
  1327. case OT_CLASS: {
  1328. SQClass *c = _class(self);
  1329. if(handle->_static) {
  1330. val = &c->_methods[handle->_index].val;
  1331. }
  1332. else {
  1333. val = &c->_defaultvalues[handle->_index].val;
  1334. }
  1335. }
  1336. break;
  1337. default:
  1338. return sq_throwerror(v,_SC("wrong type(expected class or instance)"));
  1339. }
  1340. return SQ_OK;
  1341. }
  1342. SQRESULT sq_getbyhandle(HSQUIRRELVM v,SQInteger idx,const HSQMEMBERHANDLE *handle)
  1343. {
  1344. SQObjectPtr &self = stack_get(v,idx);
  1345. SQObjectPtr *val = NULL;
  1346. if(SQ_FAILED(_getmemberbyhandle(v,self,handle,val))) {
  1347. return SQ_ERROR;
  1348. }
  1349. v->Push(_realval(*val));
  1350. return SQ_OK;
  1351. }
  1352. SQRESULT sq_setbyhandle(HSQUIRRELVM v,SQInteger idx,const HSQMEMBERHANDLE *handle)
  1353. {
  1354. SQObjectPtr &self = stack_get(v,idx);
  1355. SQObjectPtr &newval = stack_get(v,-1);
  1356. SQObjectPtr *val = NULL;
  1357. if(SQ_FAILED(_getmemberbyhandle(v,self,handle,val))) {
  1358. return SQ_ERROR;
  1359. }
  1360. *val = newval;
  1361. v->Pop();
  1362. return SQ_OK;
  1363. }
  1364. SQRESULT sq_getbase(HSQUIRRELVM v,SQInteger idx)
  1365. {
  1366. SQObjectPtr *o = NULL;
  1367. _GETSAFE_OBJ(v, idx, OT_CLASS,o);
  1368. if(_class(*o)->_base)
  1369. v->Push(SQObjectPtr(_class(*o)->_base));
  1370. else
  1371. v->PushNull();
  1372. return SQ_OK;
  1373. }
  1374. SQRESULT sq_getclass(HSQUIRRELVM v,SQInteger idx)
  1375. {
  1376. SQObjectPtr *o = NULL;
  1377. _GETSAFE_OBJ(v, idx, OT_INSTANCE,o);
  1378. v->Push(SQObjectPtr(_instance(*o)->_class));
  1379. return SQ_OK;
  1380. }
  1381. SQRESULT sq_createinstance(HSQUIRRELVM v,SQInteger idx)
  1382. {
  1383. SQObjectPtr *o = NULL;
  1384. _GETSAFE_OBJ(v, idx, OT_CLASS,o);
  1385. v->Push(_class(*o)->CreateInstance());
  1386. return SQ_OK;
  1387. }
  1388. void sq_weakref(HSQUIRRELVM v,SQInteger idx)
  1389. {
  1390. SQObject &o=stack_get(v,idx);
  1391. if(ISREFCOUNTED(type(o))) {
  1392. v->Push(_refcounted(o)->GetWeakRef(type(o)));
  1393. return;
  1394. }
  1395. v->Push(o);
  1396. }
  1397. SQRESULT sq_getweakrefval(HSQUIRRELVM v,SQInteger idx)
  1398. {
  1399. SQObjectPtr &o = stack_get(v,idx);
  1400. if(type(o) != OT_WEAKREF) {
  1401. return sq_throwerror(v,_SC("the object must be a weakref"));
  1402. }
  1403. v->Push(_weakref(o)->_obj);
  1404. return SQ_OK;
  1405. }
  1406. SQRESULT sq_getdefaultdelegate(HSQUIRRELVM v,SQObjectType t)
  1407. {
  1408. SQSharedState *ss = _ss(v);
  1409. switch(t) {
  1410. case OT_TABLE: v->Push(ss->_table_default_delegate); break;
  1411. case OT_ARRAY: v->Push(ss->_array_default_delegate); break;
  1412. case OT_STRING: v->Push(ss->_string_default_delegate); break;
  1413. case OT_INTEGER: case OT_FLOAT: v->Push(ss->_number_default_delegate); break;
  1414. case OT_GENERATOR: v->Push(ss->_generator_default_delegate); break;
  1415. case OT_CLOSURE: case OT_NATIVECLOSURE: v->Push(ss->_closure_default_delegate); break;
  1416. case OT_THREAD: v->Push(ss->_thread_default_delegate); break;
  1417. case OT_CLASS: v->Push(ss->_class_default_delegate); break;
  1418. case OT_INSTANCE: v->Push(ss->_instance_default_delegate); break;
  1419. case OT_WEAKREF: v->Push(ss->_weakref_default_delegate); break;
  1420. default: return sq_throwerror(v,_SC("the type doesn't have a default delegate"));
  1421. }
  1422. return SQ_OK;
  1423. }
  1424. SQRESULT sq_next(HSQUIRRELVM v,SQInteger idx)
  1425. {
  1426. SQObjectPtr o=stack_get(v,idx),&refpos = stack_get(v,-1),realkey,val;
  1427. if(type(o) == OT_GENERATOR) {
  1428. return sq_throwerror(v,_SC("cannot iterate a generator"));
  1429. }
  1430. int faketojump;
  1431. if(!v->FOREACH_OP(o,realkey,val,refpos,0,666,faketojump))
  1432. return SQ_ERROR;
  1433. if(faketojump != 666) {
  1434. v->Push(realkey);
  1435. v->Push(val);
  1436. return SQ_OK;
  1437. }
  1438. return SQ_ERROR;
  1439. }
  1440. struct BufState{
  1441. const SQChar *buf;
  1442. SQInteger ptr;
  1443. SQInteger size;
  1444. };
  1445. SQInteger buf_lexfeed(SQUserPointer file)
  1446. {
  1447. BufState *buf=(BufState*)file;
  1448. if(buf->size<(buf->ptr+1))
  1449. return 0;
  1450. return buf->buf[buf->ptr++];
  1451. }
  1452. SQRESULT sq_compilebuffer(HSQUIRRELVM v,const SQChar *s,SQInteger size,const SQChar *sourcename,SQBool raiseerror) {
  1453. BufState buf;
  1454. buf.buf = s;
  1455. buf.size = size;
  1456. buf.ptr = 0;
  1457. return sq_compile(v, buf_lexfeed, &buf, sourcename, raiseerror);
  1458. }
  1459. void sq_move(HSQUIRRELVM dest,HSQUIRRELVM src,SQInteger idx)
  1460. {
  1461. dest->Push(stack_get(src,idx));
  1462. }
  1463. void sq_setprintfunc(HSQUIRRELVM v, SQPRINTFUNCTION printfunc,SQPRINTFUNCTION errfunc)
  1464. {
  1465. _ss(v)->_printfunc = printfunc;
  1466. _ss(v)->_errorfunc = errfunc;
  1467. }
  1468. SQPRINTFUNCTION sq_getprintfunc(HSQUIRRELVM v)
  1469. {
  1470. return _ss(v)->_printfunc;
  1471. }
  1472. SQPRINTFUNCTION sq_geterrorfunc(HSQUIRRELVM v)
  1473. {
  1474. return _ss(v)->_errorfunc;
  1475. }
  1476. void *sq_malloc(SQUnsignedInteger size)
  1477. {
  1478. return SQ_MALLOC(size);
  1479. }
  1480. void *sq_realloc(void* p,SQUnsignedInteger oldsize,SQUnsignedInteger newsize)
  1481. {
  1482. return SQ_REALLOC(p,oldsize,newsize);
  1483. }
  1484. void sq_free(void *p,SQUnsignedInteger size)
  1485. {
  1486. SQ_FREE(p,size);
  1487. }
  1488. SQRESULT sq_optinteger(HSQUIRRELVM sqvm, SQInteger idx, SQInteger *value, SQInteger default_value){
  1489. if(sq_gettop(sqvm) >= idx){
  1490. return sq_getinteger(sqvm, idx, value);
  1491. }
  1492. *value = default_value;
  1493. return SQ_OK;
  1494. }
  1495. SQRESULT sq_optstr_and_size(HSQUIRRELVM sqvm, SQInteger idx, const SQChar **value, const SQChar *dflt, SQInteger *size){
  1496. if(sq_gettop(sqvm) >= idx){
  1497. return sq_getstr_and_size(sqvm, idx, value, size);
  1498. }
  1499. *value = dflt;
  1500. *size = strlen(dflt);
  1501. return SQ_OK;
  1502. }
  1503. void sq_insert_reg_funcs(HSQUIRRELVM sqvm, SQRegFunction *obj_funcs){
  1504. SQInteger i = 0;
  1505. while(obj_funcs[i].name != 0) {
  1506. SQRegFunction &f = obj_funcs[i];
  1507. sq_pushstring(sqvm,f.name,-1);
  1508. sq_newclosure(sqvm,f.f,0);
  1509. sq_setparamscheck(sqvm,f.nparamscheck,f.typemask);
  1510. sq_setnativeclosurename(sqvm,-1,f.name);
  1511. sq_newslot(sqvm,-3,SQFalse);
  1512. i++;
  1513. }
  1514. }