energy_balance_model.h

Go to the documentation of this file.

#pragma once
 
#include "interfaces.h"
#include "mapping.h"
 
#include <optional>
 
namespace hgps {
 
class EnergyBalanceModel final : public HierarchicalLinearModel {
  public:
    EnergyBalanceModel(
        const std::unordered_map<core::Identifier, double> &energy_equation,
        const std::unordered_map<core::Identifier, std::pair<double, double>> &nutrient_ranges,
        const std::unordered_map<core::Identifier, std::map<core::Identifier, double>>
            &nutrient_equations,
        const std::unordered_map<core::Identifier, std::optional<double>> &food_prices,
        const std::unordered_map<core::Gender, std::vector<double>> &age_mean_height);
 
    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::unordered_map<core::Identifier, double> &energy_equation_;
    const std::unordered_map<core::Identifier, std::pair<double, double>> &nutrient_ranges_;
    const std::unordered_map<core::Identifier, std::map<core::Identifier, double>>
        &nutrient_equations_;
    const std::unordered_map<core::Identifier, std::optional<double>> &food_prices_;
    const std::unordered_map<core::Gender, std::vector<double>> &age_mean_height_;
 
    double bounded_nutrient_value(const core::Identifier &nutrient, double value) const;
 
    std::map<core::Identifier, double> get_current_risk_factors(const HierarchicalMapping &mapping,
                                                                Person &entity,
                                                                int time_year) const;
};
 
class EnergyBalanceModelDefinition final : public RiskFactorModelDefinition {
  public:
    EnergyBalanceModelDefinition(
        std::unordered_map<core::Identifier, double> energy_equation,
        std::unordered_map<core::Identifier, std::pair<double, double>> nutrient_ranges,
        std::unordered_map<core::Identifier, std::map<core::Identifier, double>> nutrient_equations,
        std::unordered_map<core::Identifier, std::optional<double>> food_prices,
        std::unordered_map<core::Gender, std::vector<double>> age_mean_height);
 
    std::unique_ptr<HierarchicalLinearModel> create_model() const override;
 
  private:
    std::unordered_map<core::Identifier, double> energy_equation_;
    std::unordered_map<core::Identifier, std::pair<double, double>> nutrient_ranges_;
    std::unordered_map<core::Identifier, std::map<core::Identifier, double>> nutrient_equations_;
    std::unordered_map<core::Identifier, std::optional<double>> food_prices_;
    std::unordered_map<core::Gender, std::vector<double>> age_mean_height_;
};
 
} // namespace hgps