datamanager.h

Go to the documentation of this file.

#pragma once
 
#include "HealthGPS.Core/api.h"
 
#include <filesystem>
#include <nlohmann/json.hpp>
 
namespace hgps::data {
 
using namespace hgps::core;
 
class DataManager : public Datastore {
  public:
    DataManager() = delete;
 
    explicit DataManager(const std::filesystem::path root_directory,
                         VerboseMode verbosity = VerboseMode::none);
 
    std::vector<Country> get_countries() const override;
 
    std::optional<Country> get_country(std::string alpha) const override;
 
    std::vector<PopulationItem> get_population(Country country) const;
 
    std::vector<PopulationItem>
    get_population(Country country,
                   const std::function<bool(const unsigned int &)> time_filter) const override;
 
    std::vector<MortalityItem> get_mortality(Country country) const;
 
    std::vector<MortalityItem>
    get_mortality(Country country,
                  const std::function<bool(const unsigned int &)> time_filter) const override;
 
    std::vector<DiseaseInfo> get_diseases() const override;
 
    std::optional<DiseaseInfo> get_disease_info(core::Identifier code) const override;
 
    DiseaseEntity get_disease(DiseaseInfo code, Country country) const override;
 
    RelativeRiskEntity get_relative_risk_to_disease(DiseaseInfo source,
                                                    DiseaseInfo target) const override;
 
    RelativeRiskEntity
    get_relative_risk_to_risk_factor(DiseaseInfo source, Gender gender,
                                     core::Identifier risk_factor_key) const override;
 
    CancerParameterEntity get_disease_parameter(DiseaseInfo info, Country country) const override;
 
    DiseaseAnalysisEntity get_disease_analysis(const Country country) const override;
 
    std::vector<BirthItem> get_birth_indicators(const Country country) const;
 
    std::vector<BirthItem> get_birth_indicators(
        const Country country,
        const std::function<bool(const unsigned int &)> time_filter) const override;
 
    std::vector<LmsDataRow> get_lms_parameters() const override;
 
  private:
    std::filesystem::path root_;
    VerboseMode verbosity_;
    nlohmann::json index_;
 
    RelativeRiskEntity generate_default_relative_risk_to_disease() const;
 
    std::map<int, std::map<Gender, double>>
    load_cost_of_diseases(Country country, nlohmann::json node,
                          std::filesystem::path parent_path) const;
 
    std::vector<LifeExpectancyItem> load_life_expectancy(const Country &country) const;
 
    std::string replace_string_tokens(std::string source, std::vector<std::string> tokens) const;
 
    std::map<std::string, std::size_t>
    create_fields_index_mapping(const std::vector<std::string> &column_names,
                                const std::vector<std::string> fields) const;
 
    void notify_warning(const std::string_view message) const;
};
} // namespace hgps::data