matrix_pool.h

Go to the documentation of this file.

#pragma once

#ifndef MATRIX_POOL_H
#define MATRIX_POOL_H

#include <list>
#include <boost/unordered_map.hpp>

struct mzd_t;
struct clmatrix_t;
struct clm4rm_conditions;

namespace frechet { namespace data {

class MatrixPool {
private:
    typedef std::pair<int,int> Key;
    typedef std::list<mzd_t*> mzdList;
    typedef std::list<clmatrix_t*> clmatrixList;

    typedef boost::unordered_map<Key,mzdList> mzdMap;
    typedef boost::unordered_map<Key,clmatrixList> clmatrixMap;

    mzdMap free_mzd;
    clmatrixMap free_clmatrix;

public:
    MatrixPool();
    ~MatrixPool();

    mzd_t*      new_mzd_t       (int rows, int cols);
    clmatrix_t* new_clmatrix_t  (int rows, int cols, clm4rm_conditions* cond);

    void reclaim(mzd_t*);
    void reclaim(clmatrix_t*);

    void clear();

private:
    mzdList& getMzdFreeList(int rows, int cols);
    clmatrixList& getClmatrixFreeList(int rows, int cols);
};

mzd_t*      new_mzd     (int rows, int cols, MatrixPool* pool);
clmatrix_t* new_clmatrix(int rows, int cols, MatrixPool* pool, clm4rm_conditions* cond);

void reclaim(mzd_t* m, MatrixPool* pool);
void reclaim(clmatrix_t* clm, MatrixPool* pool);

} } //  namespace frechet::data

#endif // MATRIX_POOL_H