repository.h

Go to the documentation of this file.

#pragma once
#include "disease_definition.h"
#include "energy_balance_hierarchical_model.h"
#include "energy_balance_model.h"
#include "hierarchical_model_static.h"
#include "interfaces.h"
#include "lms_definition.h"
#include "modelinput.h"
#include "riskfactor_adjustment_types.h"
#include <functional>
#include <memory>
#include <mutex>
#include <optional>
 
namespace hgps {
 
class Repository {
  public:
    Repository() = default;
    Repository(Repository &&) = delete;
    Repository(const Repository &) = delete;
    Repository &operator=(Repository &&) = delete;
    Repository &operator=(const Repository &) = delete;
    virtual ~Repository() = default;
 
    virtual core::Datastore &manager() noexcept = 0;
 
    virtual const RiskFactorModelDefinition &
    get_risk_factor_model_definition(const HierarchicalModelType &model_type) const = 0;
 
    virtual BaselineAdjustment &get_baseline_adjustment_definition() = 0;
 
    virtual const std::vector<core::DiseaseInfo> &get_diseases() = 0;
 
    virtual std::optional<core::DiseaseInfo> get_disease_info(core::Identifier code) = 0;
 
    virtual DiseaseDefinition &get_disease_definition(const core::DiseaseInfo &info,
                                                      const ModelInput &config) = 0;
 
    virtual LmsDefinition &get_lms_definition() = 0;
};
 
class CachedRepository final : public Repository {
  public:
    CachedRepository() = delete;
 
    CachedRepository(core::Datastore &manager);
 
    void
    register_risk_factor_model_definition(const HierarchicalModelType &model_type,
                                          std::unique_ptr<RiskFactorModelDefinition> definition);
 
    void register_baseline_adjustment_definition(BaselineAdjustment definition);
 
    core::Datastore &manager() noexcept override;
 
    const RiskFactorModelDefinition &
    get_risk_factor_model_definition(const HierarchicalModelType &model_type) const override;
 
    BaselineAdjustment &get_baseline_adjustment_definition() override;
 
    const std::vector<core::DiseaseInfo> &get_diseases() override;
 
    std::optional<core::DiseaseInfo> get_disease_info(core::Identifier code) override;
 
    DiseaseDefinition &get_disease_definition(const core::DiseaseInfo &info,
                                              const ModelInput &config) override;
 
    LmsDefinition &get_lms_definition() override;
 
    void clear_cache() noexcept;
 
  private:
    mutable std::mutex mutex_;
    std::reference_wrapper<core::Datastore> data_manager_;
    std::map<HierarchicalModelType, std::unique_ptr<RiskFactorModelDefinition>>
        rf_model_definition_;
    BaselineAdjustment baseline_adjustments_;
    std::vector<core::DiseaseInfo> diseases_info_;
    std::map<core::Identifier, DiseaseDefinition> diseases_;
    LmsDefinition lms_parameters_;
 
    void load_disease_definition(const core::DiseaseInfo &info, const ModelInput &config);
};
} // namespace hgps