Entities¶

Illustrates how to create structured populations using entities (e.g., groups, households) with contact matrices.

Try It Live¶

Edit the code below and click ▶ Run to compile and execute it in your browser via Compiler Explorer. No local setup needed!

// #define EPI_DEBUG
#include "epiworld.hpp"

using namespace epiworld;

template<typename TSeq = EPI_DEFAULT_TSEQ>
EntityToAgentFun<TSeq> dist_factory(size_t from, size_t to) {
    return [from, to](Entity<> & e, Model<> * m) -> void {

auto & agents = m->get_agents();
            for (size_t i = from; i < to; ++i)
            {
                e.add_agent(&agents[i], *m);
            }

return;

};
}

int main() {
    constexpr epiworld_fast_uint n_agents = 5000u;
    constexpr size_t entity1_end = n_agents * 3u / 10u;
    constexpr size_t entity2_end = n_agents * 6u / 10u;

std::vector< double > contact_matrix = {
        9.0, 1.0, 1.0,
        0.5, 8.0, 2.0,
        0.5, 1.0, 7.0
    };

epimodels::ModelSEIRMixing<> model(
        "Flu", // std::string vname,
        n_agents, // epiworld_fast_uint n,
        0.01,// epiworld_double prevalence,
        0.1,// epiworld_double transmission_rate,
        4.0,// epiworld_double avg_incubation_days,
        1.0/7.0,// epiworld_double recovery_rate,
        contact_matrix
    );

// Creating three groups
    Entity<> e1("Entity 1", dist_factory<>(0, entity1_end));
    Entity<> e2("Entity 2", dist_factory<>(entity1_end, entity2_end));
    Entity<> e3("Entity 3", dist_factory<>(entity2_end, n_agents));

model.add_entity(e1);
    model.add_entity(e2);
    model.add_entity(e3);

// Running and checking the results
    model.run(50, 123);
    model.print();

return 0;

}

Output