external/squirrel/squirrel/sqarray.h

   1 /*      see copyright notice in squirrel.h */\r
   2 #ifndef _SQARRAY_H_\r
   3 #define _SQARRAY_H_\r
   4 \r
   5 struct SQArray : public CHAINABLE_OBJ\r
   6 {\r
   7 private:\r
   8         SQArray(SQSharedState *ss,SQInteger nsize){_values.resize(nsize); INIT_CHAIN();ADD_TO_CHAIN(&_ss(this)->_gc_chain,this);}\r
   9         ~SQArray()\r
  10         {\r
  11                 REMOVE_FROM_CHAIN(&_ss(this)->_gc_chain,this);\r
  12         }\r
  13 public:\r
  14         static SQArray* Create(SQSharedState *ss,SQInteger nInitialSize){\r
  15                 SQArray *newarray=(SQArray*)SQ_MALLOC(sizeof(SQArray));\r
  16                 new (newarray) SQArray(ss,nInitialSize);\r
  17                 return newarray;\r
  18         }\r
  19 #ifndef NO_GARBAGE_COLLECTOR\r
  20         void Mark(SQCollectable **chain);\r
  21         SQObjectType GetType() {return OT_ARRAY;}\r
  22 #endif\r
  23         void Finalize(){\r
  24                 _values.resize(0);\r
  25         }\r
  26         bool Get(const SQInteger nidx,SQObjectPtr &val)\r
  27         {\r
  28                 if(nidx>=0 && nidx<(SQInteger)_values.size()){\r
  29                         SQObjectPtr &o = _values[nidx];\r
  30                         val = _realval(o);\r
  31                         return true;\r
  32                 }\r
  33                 else return false;\r
  34         }\r
  35         bool Set(const SQInteger nidx,const SQObjectPtr &val)\r
  36         {\r
  37                 if(nidx>=0 && nidx<(SQInteger)_values.size()){\r
  38                         _values[nidx]=val;\r
  39                         return true;\r
  40                 }\r
  41                 else return false;\r
  42         }\r
  43         SQInteger Next(const SQObjectPtr &refpos,SQObjectPtr &outkey,SQObjectPtr &outval)\r
  44         {\r
  45                 SQUnsignedInteger idx=TranslateIndex(refpos);\r
  46                 while(idx<_values.size()){\r
  47                         //first found\r
  48                         outkey=(SQInteger)idx;\r
  49                         SQObjectPtr &o = _values[idx];\r
  50                         outval = _realval(o);\r
  51                         //return idx for the next iteration\r
  52                         return ++idx;\r
  53                 }\r
  54                 //nothing to iterate anymore\r
  55                 return -1;\r
  56         }\r
  57         SQArray *Clone(){SQArray *anew=Create(_opt_ss(this),0); anew->_values.copy(_values); return anew; }\r
  58         SQInteger Size() const {return _values.size();}\r
  59         void Resize(SQInteger size)\r
  60         {\r
  61                 SQObjectPtr _null;\r
  62                 Resize(size,_null);\r
  63         }\r
  64         void Resize(SQInteger size,SQObjectPtr &fill) { _values.resize(size,fill); ShrinkIfNeeded(); }\r
  65         void Reserve(SQInteger size) { _values.reserve(size); }\r
  66         void Append(const SQObject &o){_values.push_back(o);}\r
  67         void Extend(const SQArray *a);\r
  68         SQObjectPtr &Top(){return _values.top();}\r
  69         void Pop(){_values.pop_back(); ShrinkIfNeeded(); }\r
  70         bool Insert(SQInteger idx,const SQObject &val){\r
  71                 if(idx < 0 || idx > (SQInteger)_values.size())\r
  72                         return false;\r
  73                 _values.insert(idx,val);\r
  74                 return true;\r
  75         }\r
  76         void ShrinkIfNeeded() {\r
  77                 if(_values.size() <= _values.capacity()>>2) //shrink the array\r
  78                         _values.shrinktofit();\r
  79         }\r
  80         bool Remove(SQInteger idx){\r
  81                 if(idx < 0 || idx >= (SQInteger)_values.size())\r
  82                         return false;\r
  83                 _values.remove(idx);\r
  84                 ShrinkIfNeeded();\r
  85                 return true;\r
  86         }\r
  87         void Release()\r
  88         {\r
  89                 sq_delete(this,SQArray);\r
  90         }\r
  91         \r
  92         SQObjectPtrVec _values;\r
  93 };\r
  94 #endif //_SQARRAY_H_\r