File entity-bones.hpp

File List > epiworld > entity-bones.hpp

Go to the documentation of this file

#ifndef EPIWORLD_ENTITY_BONES_HPP
#define EPIWORLD_ENTITY_BONES_HPP

template<typename TSeq>
class Model;

template<typename TSeq>
class Agent;

template<typename TSeq>
class AgentsSample;

template<typename TSeq>
class Entity {
    friend class Agent<TSeq>;
    friend class AgentsSample<TSeq>;
    friend class Model<TSeq>;
private:

    int id = -1;
    std::vector< size_t > agents;   

    int max_capacity = -1;
    std::string entity_name = "Unnamed entity";

    std::vector< epiworld_double > location = {0.0}; 

    epiworld_fast_int state_init = -99;
    epiworld_fast_int state_post = -99;

    epiworld_fast_int queue_init = 0; 
    epiworld_fast_int queue_post = 0; 

    EntityToAgentFun<TSeq> dist_fun = nullptr;

    void reset();

public:


    Entity(
        std::string name,
        EntityToAgentFun<TSeq> fun = nullptr
        ) :
            entity_name(name),
            dist_fun(fun)
        {};

    ~Entity() = default;

    void add_agent(Agent<TSeq> & p, Model<TSeq> & model);
    void add_agent(Agent<TSeq> * p, Model<TSeq> & model);
    void rm_agent(size_t idx, Model<TSeq> & model);
    size_t size() const noexcept;
    void set_location(std::vector< epiworld_double > loc);
    std::vector< epiworld_double > & get_location();

    typename std::vector< size_t >::iterator begin();
    typename std::vector< size_t >::iterator end();

    typename std::vector< size_t >::const_iterator begin() const;
    typename std::vector< size_t >::const_iterator end() const;

    size_t operator[](size_t i);

    int get_id() const noexcept;
    const std::string & get_name() const noexcept;

    void set_state(epiworld_fast_int init, epiworld_fast_int post);
    void set_queue(epiworld_fast_int init, epiworld_fast_int post);
    void get_state(epiworld_fast_int * init, epiworld_fast_int * post);
    void get_queue(epiworld_fast_int * init, epiworld_fast_int * post);

    bool operator==(const Entity<TSeq> & other) const;
    bool operator!=(const Entity<TSeq> & other) const {return !operator==(other);};

    void distribute(Model<TSeq> * model);

    const std::vector< size_t > & get_agents() const;
    const std::vector< size_t > & get_agents_ids() const;

    void print() const;
    void set_distribution(EntityToAgentFun<TSeq> fun);

};


#endif