memilio/io/parameters_io.h

Go to the documentation of this file.

/* 
* Copyright (C) 2020-2026 MEmilio
*
* Authors: Henrik Zunker
*
* Contact: Martin J. Kuehn <Martin.Kuehn@DLR.de>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef MIO_IO_PARAMETERS_IO_H
#define MIO_IO_PARAMETERS_IO_H
 
#include "memilio/config.h"
 
#ifdef MEMILIO_HAS_JSONCPP
 
#include "memilio/io/epi_data.h"
#include "memilio/io/result_io.h"
 
#include <string>
#include <vector>
 
namespace mio
{
template <class EpiDataEntry>
int get_region_id(const EpiDataEntry& data_entry)
{
    if (data_entry.county_id) {
        return data_entry.county_id->get();
    }
    else if (data_entry.state_id) {
        return data_entry.state_id->get();
    }
    else if (data_entry.district_id) {
        return data_entry.district_id->get();
    }
    else {
        return 0;
    }
}
 
template <typename FP = ScalarType>
IOResult<void> compute_divi_data(const std::vector<DiviEntry>& divi_data, const std::vector<int>& vregion, Date date,
                                 std::vector<FP>& vnum_icu)
{
    auto max_date_entry = std::max_element(divi_data.begin(), divi_data.end(), [](auto&& a, auto&& b) {
        return a.date < b.date;
    });
    if (max_date_entry == divi_data.end()) {
        log_error("DIVI data is empty.");
        return failure(StatusCode::InvalidValue, "DIVI data is empty.");
    }
    auto max_date = max_date_entry->date;
    if (max_date < date) {
        log_error("DIVI data does not contain the specified date.");
        return failure(StatusCode::OutOfRange, "DIVI data does not contain the specified date.");
    }
 
    for (auto&& entry : divi_data) {
        auto it      = std::find_if(vregion.begin(), vregion.end(), [&entry](auto r) {
            return r == 0 || r == get_region_id(entry);
        });
        auto date_df = entry.date;
        if (it != vregion.end() && date_df == date) {
            auto region_idx      = size_t(it - vregion.begin());
            vnum_icu[region_idx] = entry.num_icu;
        }
    }
 
    return success();
}
 
template <typename FP = ScalarType>
IOResult<void> read_divi_data(const std::string& path, const std::vector<int>& vregion, Date date,
                              std::vector<FP>& vnum_icu)
{
    BOOST_OUTCOME_TRY(auto&& divi_data, mio::read_divi_data(path));
    return compute_divi_data(divi_data, vregion, date, vnum_icu);
}
 
IOResult<std::vector<std::vector<ScalarType>>>
read_population_data(const std::vector<PopulationDataEntry>& population_data, const std::vector<int>& vregion);
 
IOResult<std::vector<std::vector<ScalarType>>> read_population_data(const std::string& path,
                                                                    const std::vector<int>& vregion);
 
} // namespace mio
 
#endif // MEMILIO_HAS_JSONCPP
 
#endif // MIO_IO_PARAMETERS_IO_H