counters.hh

Go to the documentation of this file.

#ifndef UTOPIA_MODELS_SEIRD_COUNTERS_HH
#define UTOPIA_MODELS_SEIRD_COUNTERS_HH
 
#include <array>
#include <string>
 
#include <utopia/core/types.hh>
 
 
namespace Utopia::Models::SEIRD {
 
 
template<class Counter>
struct Counters {
 
    static constexpr std::size_t size = 11;
 
private:
    std::array<Counter, size> _counts {};
 
    inline static const std::array<std::string, size> _labels {
        "empty_to_susceptible",
        "living_to_empty",
        "susceptible_to_exposed_contact",
        "susceptible_to_exposed_random",
        "susceptible_to_exposed_controlled",
        "exposed_to_infected",
        "infected_to_recovered",
        "infected_to_deceased",
        "recovered_to_susceptible",
        "move_randomly",
        "move_away_from_infected"
    };
 
public:
    Counters () {
        _counts.fill(0);
    }
 
    auto counts () {
        return _counts;
    }
 
    const auto& labels () {
        return _labels;
    }
 
    void increment_empty_to_susceptible () {
        ++_counts[0];
    }
 
    void increment_living_to_empty () {
        ++_counts[1];
    }
 
    void increment_susceptible_to_exposed_contact () {
        ++_counts[2];
    }
 
    void increment_susceptible_to_exposed_random () {
        ++_counts[3];
    }
 
    void increment_susceptible_to_exposed_controlled () {
        ++_counts[4];
    }
 
    void increment_exposed_to_infected () {
        ++_counts[5];
    }
 
    void increment_infected_to_recovered () {
        ++_counts[6];
    }
 
    void increment_infected_to_deceased () {
        ++_counts[7];
    }
 
    void increment_recovered_to_susceptible () {
        ++_counts[8];
    }
 
    void increment_move_randomly () {
        ++_counts[9];
    }
 
    void increment_move_away_from_infected () {
        ++_counts[10];
    }
};
 
 
}  // namespace Utopia::Models::SEIRD
 
#endif  // UTOPIA_MODELS_SEIRD_COUNTERS_HH