energy_balance_hierarchical_model.h

Go to the documentation of this file.

#pragma once
 
#include "interfaces.h"
#include "mapping.h"
#include "random_algorithm.h"
 
namespace hgps {
 
struct FactorDynamicEquation {
    std::string name;
 
    std::map<core::Identifier, double> coefficients{};
 
    double residuals_standard_deviation{};
};
 
struct AgeGroupGenderEquation {
    core::IntegerInterval age_group;
 
    std::map<core::Identifier, FactorDynamicEquation> male{};
 
    std::map<core::Identifier, FactorDynamicEquation> female{};
};
 
class EnergyBalanceHierarchicalModel final : public HierarchicalLinearModel {
  public:
    EnergyBalanceHierarchicalModel(
        const std::map<core::IntegerInterval, AgeGroupGenderEquation> &equations,
        const std::map<core::Identifier, core::Identifier> &variables,
        const double boundary_percentage);
 
    HierarchicalModelType type() const noexcept override;
 
    std::string name() const noexcept override;
 
    void generate_risk_factors(RuntimeContext &context) override;
 
    void update_risk_factors(RuntimeContext &context) override;
 
  private:
    const std::map<core::IntegerInterval, AgeGroupGenderEquation> &equations_;
    const std::map<core::Identifier, core::Identifier> &variables_;
    double boundary_percentage_;
 
    const AgeGroupGenderEquation &equations_at(const int &age) const;
 
    void update_risk_factors_exposure(
        RuntimeContext &context, Person &entity,
        const std::map<core::Identifier, double> &current_risk_factors,
        const std::map<core::Identifier, FactorDynamicEquation> &equations);
 
    std::map<core::Identifier, double> get_current_risk_factors(const HierarchicalMapping &mapping,
                                                                Person &entity,
                                                                int time_year) const;
 
    double sample_normal_with_boundary(Random &random, double mean, double standard_deviation,
                                       double boundary) const;
};
 
class LiteHierarchicalModelDefinition final : public RiskFactorModelDefinition {
  public:
    LiteHierarchicalModelDefinition(
        std::map<core::IntegerInterval, AgeGroupGenderEquation> equations,
        std::map<core::Identifier, core::Identifier> variables,
        const double boundary_percentage = 0.05);
 
    std::unique_ptr<HierarchicalLinearModel> create_model() const override;
 
  private:
    std::map<core::IntegerInterval, AgeGroupGenderEquation> equations_;
    std::map<core::Identifier, core::Identifier> variables_;
    double boundary_percentage_;
};
 
} // namespace hgps