PublicHealthAgency.cpp

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

#include "PublicHealthAgency.h"

#include "pop/Population.h"
#include "util/RnMan.h"

#include <numeric>
#include <vector>

namespace stride {

using namespace std;
using namespace util;

void PublicHealthAgency::Exec(std::shared_ptr<Population> pop, util::RnMan& rnManager, unsigned short int simDay)
{
        PerformCaseFinding(std::move(pop), rnManager, simDay);
}

void PublicHealthAgency::Initialize(double detection_probability) { m_detection_probability = detection_probability; }

void PublicHealthAgency::PerformCaseFinding(std::shared_ptr<Population> pop, util::RnMan& rnMan,
                                            unsigned short int simDay)
{
        // perform case finding, only if the probability is > 0.0
        if (m_detection_probability <= 0.0) {
                return;
        }

        using namespace ContactType;
        auto  uniform01Gen = rnMan.GetUniform01Generator(0U);
        auto& logger       = pop->RefContactLogger();

        std::initializer_list<Id> AgencyPoolIdList{Id::Household};

        for (auto& p_case : *pop) {
                if (p_case.GetHealth().IsSymptomatic() && p_case.GetHealth().SymptomsStartedToday()) {
                        for (Id typ : AgencyPoolIdList) {

                                const auto& pools  = pop->CRefPoolSys().CRefPools(typ);
                                const auto  poolId = p_case.GetPoolId(typ);
                                if (poolId == 0) {
                                        continue;
                                }

                                for (const auto& p_member : pools[poolId].GetPool()) {
                                        if (p_case != *p_member && p_member->GetHealth().IsSusceptible() &&
                                            uniform01Gen() < m_detection_probability) {

                                                // set immune
                                                p_member->GetHealth().SetImmune();

                                                // TODO: check log_level
                                                logger->info("[VACC] {} {} {} {} {} {} {}", p_member->GetId(),
                                                             p_member->GetAge(), ToString(typ), poolId, p_case.GetId(),
                                                             p_case.GetAge(), simDay);
                                        }
                                }
                        }
                }
        }
}

} // namespace stride