Sim.h

Go to the documentation of this file.

/*
 *  This 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 3 of the License, or
 *  any later version.
 *  The software is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  You should have received a copy of the GNU General Public License
 *  along with the software. If not, see <http://www.gnu.org/licenses/>.
 *
 *  Copyright 2017, 2018, Kuylen E, Willem L, Broeckhove J
 */

#pragma once

#include "contact/AgeContactProfiles.h"
#include "contact/ContactHandler.h"
#include "contact/ContactLogMode.h"
#include "contact/InfectorExec.h"
#include "contact/TransmissionProfile.h"
#include "disease/PublicHealthAgency.h"

#include <boost/property_tree/ptree.hpp>
#include <string>

namespace stride {

class Calendar;
class Population;

namespace util {
class RnMan;
}

class Sim
{
public:
        static std::shared_ptr<Sim> Create(const boost::property_tree::ptree& config, std::shared_ptr<Population> pop,
                                           util::RnMan rnMan);

        static std::shared_ptr<Sim> Create(const std::string& configString, std::shared_ptr<Population> pop, util::RnMan rnMan);

        std::shared_ptr<Calendar> GetCalendar() const { return m_calendar; }

        std::string GetConfigValue(const std::string& attribute) const { return m_config.get<std::string>(attribute); }

        std::shared_ptr<Population> GetPopulation() { return m_population; }

        double GetTransmissionProbability() const { return m_transmission_profile.GetProbability(); }

        util::RnMan& RefRnManager() { return m_rn_man; }

        const TransmissionProfile& RefTransmissionProfile() const { return m_transmission_profile; }

        void TimeStep();

private:
        explicit Sim();

        friend class SimBuilder;

private:
        boost::property_tree::ptree m_config;                        
        ContactLogMode::Id          m_contact_log_mode;              
        unsigned int                m_num_threads;                   
        bool                        m_track_index_case;              
        bool                        m_adaptive_symptomatic_behavior; 

        std::shared_ptr<Calendar>   m_calendar;         
        AgeContactProfiles          m_contact_profiles; 
        std::vector<ContactHandler> m_handlers;         
        InfectorExec*               m_infector;         
        std::shared_ptr<Population> m_population;       
        util::RnMan                 m_rn_man;           

        TransmissionProfile m_transmission_profile; 
        PublicHealthAgency  m_public_health_agency; 
};

} // namespace stride