summaryrefslogtreecommitdiff
path: root/src/oldpool.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/oldpool.cpp')
-rw-r--r--src/oldpool.cpp86
1 files changed, 86 insertions, 0 deletions
diff --git a/src/oldpool.cpp b/src/oldpool.cpp
new file mode 100644
index 000000000..2b2d188e8
--- /dev/null
+++ b/src/oldpool.cpp
@@ -0,0 +1,86 @@
+/* $Id$ */
+
+#include "stdafx.h"
+#include "openttd.h"
+#include "debug.h"
+#include "functions.h"
+#include "oldpool.h"
+
+/**
+ * Clean a pool in a safe way (does free all blocks)
+ */
+void CleanPool(OldMemoryPool *pool)
+{
+ uint i;
+
+ DEBUG(misc, 4, "[Pool] (%s) cleaning pool..", pool->name);
+
+ /* Free all blocks */
+ for (i = 0; i < pool->current_blocks; i++) {
+ if (pool->clean_block_proc != NULL) {
+ pool->clean_block_proc(i * (1 << pool->block_size_bits), (i + 1) * (1 << pool->block_size_bits) - 1);
+ }
+ free(pool->blocks[i]);
+ }
+
+ /* Free the block itself */
+ free(pool->blocks);
+
+ /* Clear up some critical data */
+ pool->total_items = 0;
+ pool->current_blocks = 0;
+ pool->blocks = NULL;
+}
+
+/**
+ * This function tries to increase the size of array by adding
+ * 1 block too it
+ *
+ * @return Returns false if the pool could not be increased
+ */
+bool AddBlockToPool(OldMemoryPool *pool)
+{
+ /* Is the pool at his max? */
+ if (pool->max_blocks == pool->current_blocks)
+ return false;
+
+ pool->total_items = (pool->current_blocks + 1) * (1 << pool->block_size_bits);
+
+ DEBUG(misc, 4, "[Pool] (%s) increasing size of pool to %d items (%d bytes)", pool->name, pool->total_items, pool->total_items * pool->item_size);
+
+ /* Increase the poolsize */
+ pool->blocks = realloc(pool->blocks, sizeof(pool->blocks[0]) * (pool->current_blocks + 1));
+ if (pool->blocks == NULL) error("Pool: (%s) could not allocate memory for blocks", pool->name);
+
+ /* Allocate memory to the new block item */
+ pool->blocks[pool->current_blocks] = malloc(pool->item_size * (1 << pool->block_size_bits));
+ if (pool->blocks[pool->current_blocks] == NULL)
+ error("Pool: (%s) could not allocate memory for blocks", pool->name);
+
+ /* Clean the content of the new block */
+ memset(pool->blocks[pool->current_blocks], 0, pool->item_size * (1 << pool->block_size_bits));
+
+ /* Call a custom function if defined (e.g. to fill indexes) */
+ if (pool->new_block_proc != NULL)
+ pool->new_block_proc(pool->current_blocks * (1 << pool->block_size_bits));
+
+ /* We have a new block */
+ pool->current_blocks++;
+
+ return true;
+}
+
+/**
+ * Adds blocks to the pool if needed (and possible) till index fits inside the pool
+ *
+ * @return Returns false if adding failed
+ */
+bool AddBlockIfNeeded(OldMemoryPool *pool, uint index)
+{
+ while (index >= pool->total_items) {
+ if (!AddBlockToPool(pool))
+ return false;
+ }
+
+ return true;
+}