DUMP: ObjectPool

Author: ivan-mogilko

scripts/_dump/ObjectPool.asc

@@ -0,0 +1,110 @@
+
+int[] Array_ExpandIf(int arr[], int valid_count, int need_capacity, int elem_sz)
+{
+    if (valid_count >= need_capacity) { return arr; }
+    int want_space = need_capacity * elem_sz;
+    int new_arr[] = new int[want_space];
+    int took_space = valid_count * elem_sz;
+    for (int i = 0; i < took_space; i++) {
+        new_arr[i] = arr[i];
+    }
+    return new_arr;
+}
+
+
+void ObjectPool::AddObjects(int from, int to) {
+    if (this._capacity <= to) {
+        int new_capacity = to + 1;
+        this._usingObj = Array_ExpandIf(this._usingObj, this._capacity, new_capacity);
+        this._isFree = Array_ExpandIf(this._isFree, this._capacity, new_capacity);
+        this._freeObj = Array_ExpandIf(this._freeObj, this._capacity, new_capacity);
+        for (int i = this._capacity; i <= from; i++) {
+            this._usingObj[i] = false;
+            this._isFree[i] = false;
+        }
+        this._capacity = new_capacity;
+    }
+    for (int i = from; i <= to; i++) {
+        if (!this._usingObj[i]) {
+            this._freeObj[this._numFreeObj] = i;
+            this._numFreeObj++;
+            this._usingObj[i] = true;
+            this._isFree[i] = true;
+            this._numUsed++;
+        }
+    }
+}
+
+void ObjectPool::RemoveAll() {
+    this._numFreeObj = 0;
+    for (int i = 0; i < this._capacity; i++) {
+        this._usingObj[i] = false;
+        this._isFree[i] = false;
+    }
+    this._usingObj = null;
+    this._isFree = null;
+    this._freeObj = null;
+    this._capacity = 0;
+    this._numUsed = 0;
+}
+
+void ObjectPool::RemoveObjects(int from, int to) {
+    for (int i = from; i <= to; i++) {
+        if (this._usingObj[i]) {
+            for (int j = 0; j < this._numFreeObj; j++) {
+                if (this._freeObj[j] == i) {
+                    for (; j < this._numFreeObj - 1; j++) {
+                        this._freeObj[j] = this._freeObj[j + 1];
+                    }
+                    this._numFreeObj--;
+                }
+            }
+            this._isFree[i] = false;
+            this._usingObj[i] = false;
+            this._numUsed--;
+        }
+    }
+}
+
+int ObjectPool::Acquire() {
+    if (this._numFreeObj == 0) { return -1; }
+    this._numFreeObj--;
+    int id = this._freeObj[this._numFreeObj];
+    this._isFree[id] = false;
+    return id;
+}
+
+void ObjectPool::Release(int id) {
+    if (!this._usingObj[id] || this._isFree[id]) { return; }
+    this._freeObj[this._numFreeObj] = id;
+    this._numFreeObj++;
+    this._isFree[id] = true;
+}
+
+void ObjectPool::ReleaseAll() {
+    this._numFreeObj = 0;
+    for (int i = 0; i < this._capacity; i++) {
+        if (this._usingObj[i]) {
+            this._freeObj[this._numFreeObj] = i;
+            this._numFreeObj++;
+            this._isFree[i] = true;
+        }
+    }
+}
+
+int ObjectPool::GetAcquiredNum() {
+    return this._numUsed - this._numFreeObj;
+}
+
+int ObjectPool::GetFreeNum() {
+    return this._numFreeObj;
+}
+
+int ObjectPool::GetTotalNum() {
+    return this._numUsed;
+}
+
+
+int ObjectPool::GetPoolSize() {
+    return this._capacity;
+}

scripts/_dump/ObjectPool.ash

@@ -0,0 +1,35 @@
+
+//
+// ObjectPool keeps record of available game objects, which may be currrently in use or free.
+//
+struct ObjectPool {
+    import void AddObjects(int from, int to);
+    import void RemoveAll();
+    import void RemoveObjects(int from, int to);
+
+    import int  Acquire();
+    import void Release(int id);
+    import void ReleaseAll();
+    
+    // Gets number of acquired objects
+    import int  GetAcquiredNum();
+    // Gets number of available free objects
+    import int  GetFreeNum();
+    // Gets total number of registered objects
+    import int  GetTotalNum();
+    // Gets pool capacity (may include empty slots!)
+    import int  GetPoolSize();
+
+    protected int _capacity;
+    protected int _numUsed;
+    protected bool _usingObj[];
+    protected bool _isFree[];
+    protected int _freeObj[];
+    protected int _numFreeObj;
+};
+
+
+//
+// Helper functions which perform operations on arrays of ints
+//
+import int[] Array_ExpandIf(int arr[], int valid_count, int need_capacity, int elem_sz = 1);