sqstate.cpp 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759
  1. /*
  2. see copyright notice in squirrel.h
  3. */
  4. #include "sqpcheader.h"
  5. #include "sqopcodes.h"
  6. #include "sqvm.h"
  7. #include "sqfuncproto.h"
  8. #include "sqclosure.h"
  9. #include "sqstring.h"
  10. #include "sqtable.h"
  11. #include "sqarray.h"
  12. #include "squserdata.h"
  13. #include "sqclass.h"
  14. //SQObjectPtr _null_;
  15. //SQObjectPtr _true_(true);
  16. //SQObjectPtr _false_(false);
  17. //SQObjectPtr _one_((SQInteger)1);
  18. //SQObjectPtr _minusone_((SQInteger)-1);
  19. SQSharedState::SQSharedState()
  20. {
  21. _compilererrorhandler = NULL;
  22. _printfunc = NULL;
  23. _errorfunc = NULL;
  24. _debuginfo = false;
  25. _notifyallexceptions = false;
  26. _foreignptr = NULL;
  27. _releasehook = NULL;
  28. #ifdef SQ_WITH_DELAYED_RELEASE_HOOKS
  29. _already_in_CallDelayedReleaseHooks = false;
  30. #endif // SQ_WITH_DELAYED_RELEASE_HOOKS
  31. #ifdef SQ_JIT_LLVM
  32. //llvm::StructType::create(
  33. #endif
  34. }
  35. #define newsysstring(s) { \
  36. _systemstrings->push_back(SQString::Create(this,s)); \
  37. }
  38. #define newmetamethod(s) { \
  39. _metamethods->push_back(SQString::Create(this,s)); \
  40. _table(_metamethodsmap)->NewSlot(_metamethods->back(),(SQInteger)(_metamethods->size()-1)); \
  41. }
  42. bool CompileTypemask(SQIntVec &res,const SQChar *typemask)
  43. {
  44. SQInteger i = 0;
  45. SQInteger mask = 0;
  46. while(typemask[i] != 0) {
  47. switch(typemask[i]){
  48. case 'o': mask |= _RT_NULL; break;
  49. case 'i': mask |= _RT_INTEGER; break;
  50. case 'f': mask |= _RT_FLOAT; break;
  51. case 'n': mask |= (_RT_FLOAT | _RT_INTEGER); break;
  52. case 's': mask |= _RT_STRING; break;
  53. case 't': mask |= _RT_TABLE; break;
  54. case 'a': mask |= _RT_ARRAY; break;
  55. case 'u': mask |= _RT_USERDATA; break;
  56. case 'c': mask |= (_RT_CLOSURE | _RT_NATIVECLOSURE); break;
  57. case 'b': mask |= _RT_BOOL; break;
  58. case 'g': mask |= _RT_GENERATOR; break;
  59. case 'p': mask |= _RT_USERPOINTER; break;
  60. case 'v': mask |= _RT_THREAD; break;
  61. case 'x': mask |= _RT_INSTANCE; break;
  62. case 'y': mask |= _RT_CLASS; break;
  63. case 'r': mask |= _RT_WEAKREF; break;
  64. case '.': mask = -1; res.push_back(mask); i++; mask = 0; continue;
  65. case ' ': i++; continue; //ignores spaces
  66. default:
  67. return false;
  68. }
  69. i++;
  70. if(typemask[i] == '|') {
  71. i++;
  72. if(typemask[i] == 0)
  73. return false;
  74. continue;
  75. }
  76. res.push_back(mask);
  77. mask = 0;
  78. }
  79. return true;
  80. }
  81. static SQTable *CreateDefaultDelegate(SQSharedState *ss,const SQRegFunction *funcz)
  82. {
  83. SQInteger i=0;
  84. SQTable *t=SQTable::Create(ss,0);
  85. while(funcz[i].name!=0){
  86. SQNativeClosure *nc = SQNativeClosure::Create(ss,funcz[i].f,0);
  87. nc->_nparamscheck = funcz[i].nparamscheck;
  88. nc->_name = SQString::Create(ss,funcz[i].name);
  89. if(funcz[i].typemask && !CompileTypemask(nc->_typecheck,funcz[i].typemask))
  90. return NULL;
  91. t->NewSlot(SQString::Create(ss,funcz[i].name),nc);
  92. i++;
  93. }
  94. return t;
  95. }
  96. void SQSharedState::Init()
  97. {
  98. _scratchpad=NULL;
  99. _scratchpadsize=0;
  100. #ifndef NO_GARBAGE_COLLECTOR
  101. _gc_chain=NULL;
  102. #endif
  103. _stringtable = (SQStringTable*)SQ_MALLOC(sizeof(SQStringTable));
  104. new (_stringtable) SQStringTable(this);
  105. sq_new(_metamethods,SQObjectPtrVec);
  106. sq_new(_systemstrings,SQObjectPtrVec);
  107. sq_new(_types,SQObjectPtrVec);
  108. _defined_names = SQTable::Create(this,0);
  109. #define SQUILU_NAME_BASE "__SQUILU_"
  110. _define_squilu = SQString::Create(this,_SC(SQUILU_NAME_BASE "_"));
  111. _define_squilu_int_sz = SQString::Create(this,
  112. _SC((sizeof(SQInteger) == 8) ? SQUILU_NAME_BASE "INT_SZ8__" :
  113. SQUILU_NAME_BASE "INT_SZ4__"));
  114. _define_squilu_float_sz = SQString::Create(this,
  115. _SC((sizeof(SQFloat) == 8) ? SQUILU_NAME_BASE "FLOAT_SZ8__" :
  116. SQUILU_NAME_BASE "FLOAT_SZ4__"));
  117. _define_squilu_ptr_sz = SQString::Create(this,
  118. _SC((sizeof(void *) == 8) ? SQUILU_NAME_BASE "PTR_SZ8__" :
  119. SQUILU_NAME_BASE "PTR_SZ4__"));
  120. #undef SQUILU_NAME_BASE
  121. SQObjectPtr value(true);
  122. _table(_defined_names)->NewSlot(_define_squilu, value);
  123. _table(_defined_names)->NewSlot(_define_squilu_int_sz, value);
  124. _table(_defined_names)->NewSlot(_define_squilu_float_sz, value);
  125. _table(_defined_names)->NewSlot(_define_squilu_ptr_sz, value);
  126. _metamethodsmap = SQTable::Create(this,MT_LAST-1);
  127. //adding type strings to avoid memory trashing
  128. //types names
  129. newsysstring(_SC("null"));
  130. newsysstring(_SC("table"));
  131. newsysstring(_SC("array"));
  132. newsysstring(_SC("closure"));
  133. newsysstring(_SC("string"));
  134. newsysstring(_SC("userdata"));
  135. newsysstring(_SC("integer"));
  136. newsysstring(_SC("float"));
  137. newsysstring(_SC("userpointer"));
  138. newsysstring(_SC("function"));
  139. newsysstring(_SC("generator"));
  140. newsysstring(_SC("thread"));
  141. newsysstring(_SC("class"));
  142. newsysstring(_SC("instance"));
  143. newsysstring(_SC("bool"));
  144. //meta methods
  145. newmetamethod(MM_ADD);
  146. newmetamethod(MM_SUB);
  147. newmetamethod(MM_MUL);
  148. newmetamethod(MM_DIV);
  149. newmetamethod(MM_UNM);
  150. newmetamethod(MM_MODULO);
  151. newmetamethod(MM_SET);
  152. newmetamethod(MM_GET);
  153. newmetamethod(MM_TYPEOF);
  154. newmetamethod(MM_NEXTI);
  155. newmetamethod(MM_CMP);
  156. newmetamethod(MM_CALL);
  157. newmetamethod(MM_CLONED);
  158. newmetamethod(MM_NEWSLOT);
  159. newmetamethod(MM_DELSLOT);
  160. newmetamethod(MM_TOSTRING);
  161. newmetamethod(MM_NEWMEMBER);
  162. newmetamethod(MM_INHERITED);
  163. _constructoridx = SQString::Create(this,_SC("constructor"));
  164. _destructoridx = SQString::Create(this,_SC("destructor"));
  165. _registry = SQTable::Create(this,0);
  166. _consts = SQTable::Create(this,0);
  167. _table_default_delegate = CreateDefaultDelegate(this,_table_default_delegate_funcz);
  168. _array_default_delegate = CreateDefaultDelegate(this,_array_default_delegate_funcz);
  169. _string_default_delegate = CreateDefaultDelegate(this,_string_default_delegate_funcz);
  170. _number_default_delegate = CreateDefaultDelegate(this,_number_default_delegate_funcz);
  171. _closure_default_delegate = CreateDefaultDelegate(this,_closure_default_delegate_funcz);
  172. _generator_default_delegate = CreateDefaultDelegate(this,_generator_default_delegate_funcz);
  173. _thread_default_delegate = CreateDefaultDelegate(this,_thread_default_delegate_funcz);
  174. _class_default_delegate = CreateDefaultDelegate(this,_class_default_delegate_funcz);
  175. _instance_default_delegate = CreateDefaultDelegate(this,_instance_default_delegate_funcz);
  176. _weakref_default_delegate = CreateDefaultDelegate(this,_weakref_default_delegate_funcz);
  177. }
  178. SQSharedState::~SQSharedState()
  179. {
  180. if(_releasehook) { _releasehook(_foreignptr,0,0); _releasehook = NULL; }
  181. _constructoridx.Null();
  182. _destructoridx.Null();
  183. _define_squilu.Null();
  184. _define_squilu_int_sz.Null();
  185. _define_squilu_float_sz.Null();
  186. _define_squilu_ptr_sz.Null();
  187. _table(_registry)->Finalize();
  188. _table(_consts)->Finalize();
  189. _table(_metamethodsmap)->Finalize();
  190. _table(_defined_names)->Finalize();
  191. _registry.Null();
  192. _consts.Null();
  193. _metamethodsmap.Null();
  194. _defined_names.Null();
  195. while(!_systemstrings->empty()) {
  196. _systemstrings->back().Null();
  197. _systemstrings->pop_back();
  198. }
  199. if(sq_type(_root_vm) != OT_NULL)
  200. {
  201. _thread(_root_vm)->Finalize();
  202. _root_vm.Null();
  203. }
  204. _table_default_delegate.Null();
  205. _array_default_delegate.Null();
  206. _string_default_delegate.Null();
  207. _number_default_delegate.Null();
  208. _closure_default_delegate.Null();
  209. _generator_default_delegate.Null();
  210. _thread_default_delegate.Null();
  211. _class_default_delegate.Null();
  212. _instance_default_delegate.Null();
  213. _weakref_default_delegate.Null();
  214. _refs_table.Finalize();
  215. #ifndef NO_GARBAGE_COLLECTOR
  216. SQCollectable *t = _gc_chain;
  217. SQCollectable *nx = NULL;
  218. if(t) {
  219. t->_uiRef++;
  220. while(t) {
  221. t->Finalize();
  222. nx = t->_next;
  223. if(nx) nx->_uiRef++;
  224. if(--t->_uiRef == 0)
  225. t->Release();
  226. t = nx;
  227. }
  228. }
  229. assert(_gc_chain==NULL); //just to proove a theory
  230. while(_gc_chain){
  231. _gc_chain->_uiRef++;
  232. _gc_chain->Release();
  233. }
  234. #endif
  235. sq_delete(_types,SQObjectPtrVec);
  236. sq_delete(_systemstrings,SQObjectPtrVec);
  237. sq_delete(_metamethods,SQObjectPtrVec);
  238. sq_delete(_stringtable,SQStringTable);
  239. if(_scratchpad)SQ_FREE(_scratchpad,_scratchpadsize);
  240. }
  241. SQInteger SQSharedState::GetMetaMethodIdxByName(const SQObjectPtr &name)
  242. {
  243. if(sq_type(name) != OT_STRING)
  244. return -1;
  245. SQObjectPtr ret;
  246. if(_table(_metamethodsmap)->Get(name,ret)) {
  247. return _integer(ret);
  248. }
  249. return -1;
  250. }
  251. #ifdef SQ_WITH_DELAYED_RELEASE_HOOKS
  252. void SQSharedState::AddDelayedReleaseHook(SQRELEASEHOOK hook, SQUserPointer ptr, SQInteger size)
  253. {
  254. SQDelayedReleseHook dh;
  255. dh.hook = hook;
  256. dh.ptr = ptr;
  257. dh.size = size;
  258. _delayed_release_hook.push_back(dh);
  259. }
  260. void SQSharedState::CallDelayedReleaseHooks(SQVM *vm, int count)
  261. {
  262. if(_already_in_CallDelayedReleaseHooks) return;
  263. if(_delayed_release_hook.size()){
  264. _already_in_CallDelayedReleaseHooks = true;
  265. //get the count of release hooks before start
  266. //and only work on that number because while
  267. //calling each release hook new ones can be added
  268. //to the list, the new ones will be processed on next call
  269. if(count == 0) count = _delayed_release_hook.size();
  270. for(SQInteger i=count-1; i >= 0; --i){
  271. SQDelayedReleseHook &dh = _delayed_release_hook[i];
  272. dh.hook(dh.ptr, dh.size, vm);
  273. }
  274. _delayed_release_hook.removeFromBegining(count);
  275. _already_in_CallDelayedReleaseHooks = false;
  276. }
  277. }
  278. #endif // SQ_WITH_DELAYED_RELEASE_HOOKS
  279. #ifndef NO_GARBAGE_COLLECTOR
  280. void SQSharedState::MarkObject(SQObjectPtr &o,SQCollectable **chain)
  281. {
  282. switch(sq_type(o)){
  283. case OT_TABLE:_table(o)->Mark(chain);break;
  284. case OT_ARRAY:_array(o)->Mark(chain);break;
  285. case OT_USERDATA:_userdata(o)->Mark(chain);break;
  286. case OT_CLOSURE:_closure(o)->Mark(chain);break;
  287. case OT_NATIVECLOSURE:_nativeclosure(o)->Mark(chain);break;
  288. case OT_GENERATOR:_generator(o)->Mark(chain);break;
  289. case OT_THREAD:_thread(o)->Mark(chain);break;
  290. case OT_CLASS:_class(o)->Mark(chain);break;
  291. case OT_INSTANCE:_instance(o)->Mark(chain);break;
  292. case OT_OUTER:_outer(o)->Mark(chain);break;
  293. case OT_FUNCPROTO:_funcproto(o)->Mark(chain);break;
  294. default: break; //shutup compiler
  295. }
  296. }
  297. void SQSharedState::RunMark(SQVM* SQ_UNUSED_ARG(vm),SQCollectable **tchain)
  298. {
  299. SQVM *vms = _thread(_root_vm);
  300. vms->Mark(tchain);
  301. _refs_table.Mark(tchain);
  302. MarkObject(_registry,tchain);
  303. MarkObject(_consts,tchain);
  304. MarkObject(_metamethodsmap,tchain);
  305. MarkObject(_table_default_delegate,tchain);
  306. MarkObject(_array_default_delegate,tchain);
  307. MarkObject(_string_default_delegate,tchain);
  308. MarkObject(_number_default_delegate,tchain);
  309. MarkObject(_generator_default_delegate,tchain);
  310. MarkObject(_thread_default_delegate,tchain);
  311. MarkObject(_closure_default_delegate,tchain);
  312. MarkObject(_class_default_delegate,tchain);
  313. MarkObject(_instance_default_delegate,tchain);
  314. MarkObject(_weakref_default_delegate,tchain);
  315. }
  316. SQInteger SQSharedState::ResurrectUnreachable(SQVM *vm)
  317. {
  318. SQInteger n=0;
  319. SQCollectable *tchain=NULL;
  320. RunMark(vm,&tchain);
  321. SQCollectable *resurrected = _gc_chain;
  322. SQCollectable *t = resurrected;
  323. //SQCollectable *nx = NULL;
  324. _gc_chain = tchain;
  325. SQArray *ret = NULL;
  326. if(resurrected) {
  327. ret = SQArray::Create(this,0);
  328. SQCollectable *rlast = NULL;
  329. while(t) {
  330. rlast = t;
  331. SQObjectType type = t->GetType();
  332. if(type != OT_FUNCPROTO && type != OT_OUTER) {
  333. SQObject sqo;
  334. sqo._type = type;
  335. sqo._unVal.pRefCounted = t;
  336. ret->Append(sqo);
  337. }
  338. t = t->_next;
  339. n++;
  340. }
  341. assert(rlast->_next == NULL);
  342. rlast->_next = _gc_chain;
  343. if(_gc_chain)
  344. {
  345. _gc_chain->_prev = rlast;
  346. }
  347. _gc_chain = resurrected;
  348. }
  349. t = _gc_chain;
  350. while(t) {
  351. t->UnMark();
  352. t = t->_next;
  353. }
  354. if(ret) {
  355. SQObjectPtr temp = ret;
  356. vm->Push(temp);
  357. }
  358. else {
  359. vm->PushNull();
  360. }
  361. return n;
  362. }
  363. SQInteger SQSharedState::CollectGarbage(SQVM *vm)
  364. {
  365. SQInteger n = 0;
  366. SQCollectable *tchain = NULL;
  367. RunMark(vm,&tchain);
  368. SQCollectable *t = _gc_chain;
  369. SQCollectable *nx = NULL;
  370. if(t) {
  371. t->_uiRef++;
  372. while(t) {
  373. t->Finalize();
  374. nx = t->_next;
  375. if(nx) nx->_uiRef++;
  376. if(--t->_uiRef == 0)
  377. t->Release();
  378. t = nx;
  379. n++;
  380. }
  381. }
  382. t = tchain;
  383. while(t) {
  384. t->UnMark();
  385. t = t->_next;
  386. }
  387. _gc_chain = tchain;
  388. return n;
  389. }
  390. #endif
  391. #ifndef NO_GARBAGE_COLLECTOR
  392. void SQCollectable::AddToChain(SQCollectable **chain,SQCollectable *c)
  393. {
  394. c->_prev = NULL;
  395. c->_next = *chain;
  396. if(*chain) (*chain)->_prev = c;
  397. *chain = c;
  398. }
  399. void SQCollectable::RemoveFromChain(SQCollectable **chain,SQCollectable *c)
  400. {
  401. if(c->_prev) c->_prev->_next = c->_next;
  402. else *chain = c->_next;
  403. if(c->_next)
  404. c->_next->_prev = c->_prev;
  405. c->_next = NULL;
  406. c->_prev = NULL;
  407. }
  408. #endif
  409. SQChar* SQSharedState::GetScratchPad(SQInteger size)
  410. {
  411. SQInteger newsize;
  412. if(size>0) {
  413. if(_scratchpadsize < size) {
  414. newsize = size + (size>>1);
  415. _scratchpad = (SQChar *)SQ_REALLOC(_scratchpad,_scratchpadsize,newsize);
  416. _scratchpadsize = newsize;
  417. }else if(_scratchpadsize >= (size<<5)) {
  418. newsize = _scratchpadsize >> 1;
  419. _scratchpad = (SQChar *)SQ_REALLOC(_scratchpad,_scratchpadsize,newsize);
  420. _scratchpadsize = newsize;
  421. }
  422. }
  423. return _scratchpad;
  424. }
  425. RefTable::RefTable()
  426. {
  427. AllocNodes(4);
  428. }
  429. void RefTable::Finalize()
  430. {
  431. RefNode *nodes = _nodes;
  432. for(SQUnsignedInteger n = 0; n < _numofslots; n++) {
  433. nodes->obj.Null();
  434. nodes++;
  435. }
  436. }
  437. RefTable::~RefTable()
  438. {
  439. SQ_FREE(_buckets,(_numofslots * sizeof(RefNode *)) + (_numofslots * sizeof(RefNode)));
  440. }
  441. #ifndef NO_GARBAGE_COLLECTOR
  442. void RefTable::Mark(SQCollectable **chain)
  443. {
  444. RefNode *nodes = (RefNode *)_nodes;
  445. for(SQUnsignedInteger n = 0; n < _numofslots; n++) {
  446. if(sq_type(nodes->obj) != OT_NULL) {
  447. SQSharedState::MarkObject(nodes->obj,chain);
  448. }
  449. nodes++;
  450. }
  451. }
  452. #endif
  453. void RefTable::AddRef(SQObject &obj)
  454. {
  455. SQHash mainpos;
  456. RefNode *prev;
  457. RefNode *ref = Get(obj,mainpos,&prev,true);
  458. ref->refs++;
  459. }
  460. SQUnsignedInteger RefTable::GetRefCount(SQObject &obj)
  461. {
  462. SQHash mainpos;
  463. RefNode *prev;
  464. RefNode *ref = Get(obj,mainpos,&prev,true);
  465. return ref->refs;
  466. }
  467. SQBool RefTable::Release(SQObject &obj)
  468. {
  469. SQHash mainpos;
  470. RefNode *prev;
  471. RefNode *ref = Get(obj,mainpos,&prev,false);
  472. if(ref) {
  473. if(--ref->refs == 0) {
  474. SQObjectPtr o = ref->obj;
  475. if(prev) {
  476. prev->next = ref->next;
  477. }
  478. else {
  479. _buckets[mainpos] = ref->next;
  480. }
  481. ref->next = _freelist;
  482. _freelist = ref;
  483. _slotused--;
  484. ref->obj.Null();
  485. //<<FIXME>>test for shrink?
  486. return SQTrue;
  487. }
  488. }
  489. else {
  490. assert(0);
  491. }
  492. return SQFalse;
  493. }
  494. void RefTable::Resize(SQUnsignedInteger size)
  495. {
  496. RefNode **oldbucks = _buckets;
  497. RefNode *t = _nodes;
  498. SQUnsignedInteger oldnumofslots = _numofslots;
  499. AllocNodes(size);
  500. //rehash
  501. SQUnsignedInteger nfound = 0;
  502. for(SQUnsignedInteger n = 0; n < oldnumofslots; n++) {
  503. if(sq_type(t->obj) != OT_NULL) {
  504. //add back;
  505. assert(t->refs != 0);
  506. RefNode *nn = Add(::HashObj(t->obj)&(_numofslots-1),t->obj);
  507. nn->refs = t->refs;
  508. t->obj.Null();
  509. nfound++;
  510. }
  511. t++;
  512. }
  513. assert(nfound == oldnumofslots);
  514. SQ_FREE(oldbucks,(oldnumofslots * sizeof(RefNode *)) + (oldnumofslots * sizeof(RefNode)));
  515. }
  516. RefTable::RefNode *RefTable::Add(SQHash mainpos,SQObject &obj)
  517. {
  518. RefNode *t = _buckets[mainpos];
  519. RefNode *newnode = _freelist;
  520. newnode->obj = obj;
  521. _buckets[mainpos] = newnode;
  522. _freelist = _freelist->next;
  523. newnode->next = t;
  524. assert(newnode->refs == 0);
  525. _slotused++;
  526. return newnode;
  527. }
  528. RefTable::RefNode *RefTable::Get(SQObject &obj,SQHash &mainpos,RefNode **prev,bool add)
  529. {
  530. RefNode *ref;
  531. mainpos = ::HashObj(obj)&(_numofslots-1);
  532. *prev = NULL;
  533. for (ref = _buckets[mainpos]; ref; ) {
  534. if(_rawval(ref->obj) == _rawval(obj) && sq_type(ref->obj) == sq_type(obj))
  535. break;
  536. *prev = ref;
  537. ref = ref->next;
  538. }
  539. if(ref == NULL && add) {
  540. if(_numofslots == _slotused) {
  541. assert(_freelist == 0);
  542. Resize(_numofslots*2);
  543. mainpos = ::HashObj(obj)&(_numofslots-1);
  544. }
  545. ref = Add(mainpos,obj);
  546. }
  547. return ref;
  548. }
  549. void RefTable::AllocNodes(SQUnsignedInteger size)
  550. {
  551. RefNode **bucks;
  552. RefNode *nodes;
  553. bucks = (RefNode **)SQ_MALLOC((size * sizeof(RefNode *)) + (size * sizeof(RefNode)));
  554. nodes = (RefNode *)&bucks[size];
  555. RefNode *temp = nodes;
  556. SQUnsignedInteger n;
  557. for(n = 0; n < size - 1; n++) {
  558. bucks[n] = NULL;
  559. temp->refs = 0;
  560. new (&temp->obj) SQObjectPtr;
  561. temp->next = temp+1;
  562. temp++;
  563. }
  564. bucks[n] = NULL;
  565. temp->refs = 0;
  566. new (&temp->obj) SQObjectPtr;
  567. temp->next = NULL;
  568. _freelist = nodes;
  569. _nodes = nodes;
  570. _buckets = bucks;
  571. _slotused = 0;
  572. _numofslots = size;
  573. }
  574. //////////////////////////////////////////////////////////////////////////
  575. //SQStringTable
  576. /*
  577. * The following code is based on Lua 4.0 (Copyright 1994-2002 Tecgraf, PUC-Rio.)
  578. * http://www.lua.org/copyright.html#4
  579. * http://www.lua.org/source/4.0.1/src_lstring.c.html
  580. */
  581. SQStringTable::SQStringTable(SQSharedState *ss):_slotused(0), _sharedstate(ss)
  582. {
  583. AllocNodes(4);
  584. }
  585. SQStringTable::~SQStringTable()
  586. {
  587. SQ_FREE(_strings,sizeof(SQString*)*_numofslots);
  588. _strings = NULL;
  589. }
  590. void SQStringTable::AllocNodes(SQInteger size)
  591. {
  592. _numofslots = size;
  593. _strings = (SQString**)SQ_MALLOC(sizeof(SQString*)*_numofslots);
  594. memset(_strings,0,sizeof(SQString*)*_numofslots);
  595. }
  596. SQString *SQStringTable::Contains(const SQChar *news, SQInteger &len, SQHash &newhash, SQHash &h)
  597. {
  598. if(len<0)
  599. len = (SQInteger)scstrlen(news);
  600. newhash = ::_hashstr(news,len);
  601. h = newhash&(_numofslots-1);
  602. SQString *s;
  603. SQInteger calculated_len = rsl(len);
  604. for (s = _strings[h]; s; s = s->_next){
  605. if(s->_len == len && (!memcmp(news,s->_val,calculated_len)))
  606. return s; //found
  607. }
  608. return 0;
  609. }
  610. SQString *SQStringTable::Add(const SQChar *news,SQInteger len)
  611. {
  612. SQHash newhash, h;
  613. SQString *t = Contains(news, len, newhash, h);
  614. if(t) return t;
  615. t = NewStrBuf(len);
  616. memcpy(t->_val,news,rsl(len));
  617. return Add(t, newhash, h);
  618. }
  619. SQString *SQStringTable::Add(SQString *strBuf)
  620. {
  621. SQHash newhash, h;
  622. #ifdef SQ_PACKED_STRUCT
  623. SQInteger blen = strBuf->_len;
  624. SQString *t = Contains(strBuf->_val, blen, newhash, h);
  625. #else
  626. SQString *t = Contains(strBuf->_val, strBuf->_len, newhash, h);
  627. #endif
  628. if(t) {
  629. SQ_FREE(strBuf,SQSTRING_CALCULATED_SIZE(rsl(strBuf->_len)));
  630. return t;
  631. }
  632. return Add(strBuf, newhash, h);
  633. }
  634. SQString *SQStringTable::Add(SQString *strBuf, SQHash newhash, SQHash h)
  635. {
  636. strBuf->_sharedstate = _sharedstate;
  637. strBuf->_val[strBuf->_len] = _SC('\0');
  638. strBuf->_hash = newhash;
  639. strBuf->_next = _strings[h];
  640. _strings[h] = strBuf;
  641. _slotused++;
  642. if (_slotused > _numofslots) /* too crowded? */
  643. Resize(_numofslots*2);
  644. return strBuf;
  645. }
  646. SQString *SQStringTable::NewStrBuf(SQInteger len)
  647. {
  648. //Take in account padding
  649. SQInteger calculated_size = SQSTRING_CALCULATED_SIZE(rsl(len));
  650. SQString *t = (SQString *)SQ_MALLOC(calculated_size);
  651. memset(t, 0, calculated_size);
  652. new (t) SQString;
  653. t->_len = len;
  654. return t;
  655. }
  656. void SQStringTable::Resize(SQInteger size)
  657. {
  658. SQInteger oldsize=_numofslots;
  659. SQString **oldtable=_strings;
  660. AllocNodes(size);
  661. for (SQInteger i=0; i<oldsize; i++){
  662. SQString *p = oldtable[i];
  663. while(p){
  664. SQString *next = p->_next;
  665. SQHash h = p->_hash&(_numofslots-1);
  666. p->_next = _strings[h];
  667. _strings[h] = p;
  668. p = next;
  669. }
  670. }
  671. SQ_FREE(oldtable,oldsize*sizeof(SQString*));
  672. }
  673. void SQStringTable::Remove(SQString *bs)
  674. {
  675. SQString *s;
  676. SQString *prev=NULL;
  677. SQHash h = bs->_hash&(_numofslots - 1);
  678. for (s = _strings[h]; s; ){
  679. if(s == bs){
  680. if(prev)
  681. prev->_next = s->_next;
  682. else
  683. _strings[h] = s->_next;
  684. _slotused--;
  685. SQInteger slen = s->_len;
  686. s->~SQString();
  687. //printf("Free str %p : %d : %.*s\n", s, (int)SQSTRING_CALCULATED_SIZE(rsl(slen))), (int)slen, s->_val);
  688. SQ_FREE(s,SQSTRING_CALCULATED_SIZE(rsl(slen)));
  689. return;
  690. }
  691. prev = s;
  692. s = s->_next;
  693. }
  694. assert(0);//if this fail something is wrong
  695. }