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

#include "SummaryFile.h"

#include "util/FileSys.h"

#include <boost/property_tree/ptree.hpp>

namespace stride {
namespace output {

using namespace std;
using namespace stride::util;

// FIXME Some of the parameters written to this file have become optional over time.
// Seeding rate + immmunity rate

SummaryFile::SummaryFile(const string& output_prefix) : m_fstream() { Initialize(output_prefix); }

SummaryFile::~SummaryFile() { m_fstream.close(); }

void SummaryFile::Initialize(const string& output_prefix)
{
        const auto p = FileSys::BuildPath(output_prefix, "summary.csv");
        m_fstream.open(p.c_str());

        // add header
        m_fstream << "population_file,num_days,population_size,seeding_rate,r0,transmission_probability,"
                     "immunity_rate,num_threads,rng_seed,"
                     "run_time,total_time,num_cases,AR,output_prefix,start_date,age_"
                     "contact_matrix_file,num_"
                     "participants_survey,disease_config_file"
                  << endl;
}

void SummaryFile::Print(const boost::property_tree::ptree& config_pt, unsigned int population_size,
                        unsigned int num_cases, double transmission_probability, unsigned int run_time,
                        unsigned int total_time)
{
        m_fstream << config_pt.get<string>("run.population_file") << "," << config_pt.get<unsigned int>("run.num_days")
                  << "," << population_size << "," << config_pt.get<double>("run.seeding_rate") << ","
                  << config_pt.get<double>("run.r0") << "," << transmission_probability << ","
                  << config_pt.get<double>("run.immunity_rate") << "," << config_pt.get<unsigned int>("run.num_threads")
                  << "," << config_pt.get<unsigned int>("run.rng_seed") << "," << run_time << "," << total_time << ","
                  << num_cases << "," << static_cast<double>(num_cases) / population_size << ","
                  << config_pt.get<string>("run.output_prefix") << "," << config_pt.get<string>("run.start_date") << ","
                  << config_pt.get<string>("run.age_contact_matrix_file") << ","
                  << config_pt.get<unsigned int>("run.num_participants_survey") << ","
                  << config_pt.get<string>("run.disease_config_file") << endl;
}

} // namespace output
} // namespace stride