diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/core/pool_func.cpp | 30 | ||||
-rw-r--r-- | src/core/pool_type.hpp | 45 |
2 files changed, 73 insertions, 2 deletions
diff --git a/src/core/pool_func.cpp b/src/core/pool_func.cpp new file mode 100644 index 000000000..4c55293eb --- /dev/null +++ b/src/core/pool_func.cpp @@ -0,0 +1,30 @@ +/* $Id$ */ + +/* + * This file is part of OpenTTD. + * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2. + * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. + * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>. + */ + +/** @file pool_func.cpp Implementation of PoolBase methods. */ + +#include "../stdafx.h" +#include "pool_type.hpp" + +PoolBase::~PoolBase() +{ + PoolVector *pools = PoolBase::GetPools(); + pools->Erase(pools->Find(this)); + if (pools->Length() == 0) delete pools; +} + +/* static */ void PoolBase::CleanAll() +{ + PoolVector *pools = PoolBase::GetPools(); + PoolBase **end = pools->End(); + for (PoolBase **ppool = pools->Begin(); ppool != end; ppool++) { + PoolBase *pool = *ppool; + pool->CleanPool(); + } +} diff --git a/src/core/pool_type.hpp b/src/core/pool_type.hpp index 395aa8438..e96608bdb 100644 --- a/src/core/pool_type.hpp +++ b/src/core/pool_type.hpp @@ -12,6 +12,47 @@ #ifndef POOL_TYPE_HPP #define POOL_TYPE_HPP +#include "smallvec_type.hpp" + +typedef SmallVector<struct PoolBase *, 4> PoolVector; ///< Vector of pointers to PoolBase + +/** Base class for base of all pools. */ +struct PoolBase { + /** + * Function used to access the vector of all pools. + * @return pointer to vector of all pools + */ + static PoolVector *GetPools() + { + static PoolVector *pools = new PoolVector(); + return pools; + } + + /** + * Clean all pools - calls Pool::CleanPool() + */ + static void CleanAll(); + + /** + * Contructor registers this object in the pool vector. + */ + PoolBase() + { + *PoolBase::GetPools()->Append() = this; + } + + /** + * Destructor removes this object from the pool vector and + * deletes the vector itself if this was the last item removed. + */ + ~PoolBase(); + + /** + * Virtual method that deletes all items in the pool. + */ + virtual void CleanPool() = 0; +}; + /** * Base class for all pools. * @tparam Titem Type of the class/struct that is going to be pooled @@ -23,7 +64,7 @@ * @warning when Tcache is enabled *all* instances of this pool's item must be of the same size. */ template <class Titem, typename Tindex, size_t Tgrowth_step, size_t Tmax_size, bool Tcache = false, bool Tzero = true> -struct Pool { +struct Pool : PoolBase { static const size_t MAX_SIZE = Tmax_size; ///< Make template parameter accessible from outside const char * const name; ///< Name of this pool @@ -40,7 +81,7 @@ struct Pool { Titem **data; ///< Pointer to array of pointers to Titem Pool(const char *name); - void CleanPool(); + virtual void CleanPool(); /** * Returs Titem with given index |