Population.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, Kuylen E, Willem L, Broeckhove J
 */

#pragma once

#include "contact/ContactPool.h"
#include "contact/ContactPoolSys.h"
#include "contact/ContactType.h"
#include "geopop/GeoGrid.h"
#include "pop/Person.h"
#include "util/RnMan.h"
#include "util/SegmentedVector.h"

#include <boost/property_tree/ptree_fwd.hpp>
#include <memory>
#include <spdlog/spdlog.h>

namespace stride {

namespace util {
class RnMan;
}

class Population : public util::SegmentedVector<Person, 2048>
{
public:
        static std::shared_ptr<Population> Create(const boost::property_tree::ptree& config, util::RnMan rnMan,
                                                  std::shared_ptr<spdlog::logger> strideLogger = nullptr);

        static std::shared_ptr<Population> Create(const std::string& configString, util::RnMan rnMan,
                                                  std::shared_ptr<spdlog::logger> stride_logger = nullptr);

        static std::shared_ptr<Population> Create();

public:
        Person* CreatePerson(unsigned int id, double age, unsigned int householdId, unsigned int k12SchoolId,
                             unsigned int collegeId, unsigned int workId, unsigned int primaryCommunityId,
                             unsigned int secondaryCommunityId);

        unsigned int GetInfectedCount() const;

        const ContactPoolSys& CRefPoolSys() const { return m_pool_sys; }

        const geopop::GeoGrid& CRefGeoGrid() const { return m_geo_grid; }

        std::shared_ptr<spdlog::logger>& RefContactLogger() { return m_contact_logger; }

        ContactPoolSys& RefPoolSys() { return m_pool_sys; }

        geopop::GeoGrid& RefGeoGrid() { return m_geo_grid; }

private:
        Population();

private:
        ContactPoolSys                  m_pool_sys;       
        std::shared_ptr<spdlog::logger> m_contact_logger; 
        geopop::GeoGrid                 m_geo_grid;       
};

} // namespace stride