state.hh

Go to the documentation of this file.

#ifndef UTOPIA_MODELS_CONTDISEASE_STATE_HH
#define UTOPIA_MODELS_CONTDISEASE_STATE_HH
 
#include <utopia/core/types.hh>
 
namespace Utopia::Models::ContDisease {
 
enum class Kind : char {
    empty = 0,
    tree = 1,
    infected = 2,
    source = 3,
    stone = 4
};
 
struct State {
    Kind kind;
 
    unsigned age;
 
    unsigned int cluster_id;
 
    template<class RNG>
    State (const DataIO::Config& cfg, const std::shared_ptr<RNG>& rng)
    :
        kind(Kind::empty),
        age(0),
        cluster_id(0)
    {
        // Check if p_tree is available to set up cell state
        if (cfg["p_tree"]) {
            const auto init_density = get_as<double>("p_tree", cfg);
 
            // With this probability, the cell state is a tree
            if (std::uniform_real_distribution<double>(0., 1.)(*rng) 
                < init_density) 
            {
                kind = Kind::tree;
            }
        }
    }
};
 
    
} // namespace Utopia::Models::ContDisease
 
 
#endif // UTOPIA_MODELS_CONTDISEASE_STATE_HH