PopulationSizeHashMapImplementation.hpp

// Copyright 2021 Arnaud Becheler    <abechele@umich.edu>
 
/*********************************************************************** * This program 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; either version 2 of the License, or    * (at your option) any later version. *
 *                                                                      *
 ***************************************************************************/
 
#ifndef __POPULATIONS_H_INCLUDED__
#define __POPULATIONS_H_INCLUDED__
 
#include "assert.h"
#include <unordered_map>
#include <unordered_set>
#include <vector>
 
namespace quetzal
{
namespace demography
{
template <typename Space, typename Time, typename Value> class PopulationSizeHashMapImplementation
{
    std::unordered_map<Time, std::unordered_map<Space, Value>> m_populations;
 
  public:
    using time_type = Time;
    using coord_type = Space;
    using value_type = Value;
    PopulationSizeHashMapImplementation() = default;
    void set(coord_type const &x, time_type const &t, value_type N)
    {
        assert(N >= 0);
        m_populations[t][x] = N;
    }
    value_type get(coord_type const &x, time_type const &t) const
    {
        assert(is_defined(x, t));
        return m_populations.at(t).at(x);
    }
    value_type operator()(coord_type const &x, time_type const &t) const
    {
        return get(x, t);
    }
    value_type &operator()(coord_type const &x, time_type const &t)
    {
        return m_populations[t][x];
    }
    std::vector<coord_type> definition_space(time_type const &t) const
    {
        std::vector<coord_type> v;
        for (auto const &it : m_populations.at(t))
        {
            if (it.second > 0)
            {
                v.push_back(it.first);
            }
        }
        return v;
    }
    bool is_defined(coord_type const &x, time_type const &t) const
    {
        return (!m_populations.empty()) && (m_populations.find(t) != m_populations.end()) &&
               (m_populations.at(t).find(x) != m_populations.at(t).end());
    }
};
} // namespace demography
} // namespace quetzal
 
#endif