ResourcePool.h

#pragma once

#include <geneial/namespaces.h>
#include <geneial/core/population/builder/BaseChromosomeFactory.h>
#include <geneial/core/population/builder/MultiValueBuilderSettings.h>
#include <mutex>


geneial_private_namespace(geneial)
{
geneial_private_namespace(population)
{
geneial_private_namespace(management)
{

geneial_export_namespace
{

template<typename RESOURCE>
class ResourcePool
{
    std::mutex _poolMutex;

    std::vector<RESOURCE*> _freePool;
    //TODO (bewo): Think about pruning when having large amounts of resources.

public:

    bool inline retrieve(RESOURCE* &raw_ptr)
    {
        std::lock_guard<std::mutex> lock(_poolMutex);
        if (_freePool.size() > 0)
        {
            raw_ptr = _freePool.back();
            _freePool.pop_back();
            return true;
        }
        else
        {
            return false;
        }
    }

    void inline free(RESOURCE*&& raw_ptr)
    {
#ifndef NALLOCREUSE
        std::lock_guard<std::mutex> lock(_poolMutex);
        _freePool.emplace_back(raw_ptr);
#else
        delete raw_ptr;
#endif
    }

    size_t inline size()
    {
        return _freePool.size();
    }

    virtual ~ResourcePool()
    {
        //The used pool is out in the wild should be guarded by some sharedptr.
        for (auto & ptr : _freePool)
        {
            delete ptr;
        }
    }
};


} /* geneial_export_namespace */
} /* private namespace management */
} /* private namespace population */
} /* private namespace geneial */