ModelPool.h

Go to the documentation of this file.

#ifndef MODEL_POOL_H
#define MODEL_POOL_H
 
#include <memory>
#include <vector>
 
namespace CNum::Model {
  template <typename ModelType>
  class ModelPool {
  private:
    size_t _n_models;
    ::CNum::DataStructs::ConcurrentQueue<ModelType *> _models;
    ::std::vector<ModelType *> _model_ptrs;
    
  public:

    /// @param n_models The number of models in the pool
    ModelPool(::std::string path, size_t n_models = 20);
 
    ModelPool(const ModelPool &) = delete;
    ModelPool &operator=(const ModelPool &) = delete;

    ~ModelPool();

    ModelType *pull();

    void push(ModelType *model);
  };
 
  #include "CNum/Model/ModelPool.tpp"
};
 
#endif