source: src/base/has_tempgrids.hpp@ 48b662

/**
 * @file   has_tempgrids.hpp
 * @author Julian Iseringhausen <isering@ins.uni-bonn.de>
 * @date   Tue Apr  5 21:00:56 2011
 * 
 * @brief  Provides an arbitrary number of temporary grids,
 *         accessible by an integer key.
 * 
 */

#ifndef HAS_TEMPGRIDS_HPP_
#define HAS_TEMPGRIDS_HPP_

#include <map>

#include "grid/tempgrid.hpp"
#include "mg.hpp"

namespace VMG
{

class HasTempGrids
{
public:
  virtual ~HasTempGrids()
  {
    for (std::map<int, VMG::TempGrid*>::iterator iter = grids.begin(); iter != grids.end(); ++iter)
      delete iter->second;
  }

protected:
  VMG::TempGrid* GetTempGrid(int key)
  {
    std::map<int, VMG::TempGrid*>::iterator iter = grids.find(key);

    if (iter != grids.end())
      return iter->second;
    else {
      VMG::TempGrid* temp = new VMG::TempGrid();
      grids.insert(std::pair<int,VMG::TempGrid*>(key, temp));
      return temp;
    }
  }

private:
  std::map<int,VMG::TempGrid*> grids;
};

}

#endif /* HAS_TEMPGRIDS_HPP_ */