riskfactor.h

Go to the documentation of this file.

#pragma once
 
#include "hierarchical_model.h"
#include "modelinput.h"
#include "repository.h"
#include "riskfactor_adjustment.h"
 
namespace hgps {
 
class RiskFactorModule final : public RiskFactorHostModule {
  public:
    RiskFactorModule() = delete;
 
    RiskFactorModule(
        std::map<HierarchicalModelType, std::unique_ptr<HierarchicalLinearModel>> &&models,
        RiskfactorAdjustmentModel &&adjustments);
 
    SimulationModuleType type() const noexcept override;
 
    const std::string &name() const noexcept override;
 
    std::size_t size() const noexcept override;
 
    bool contains(const HierarchicalModelType &modelType) const noexcept override;
 
    HierarchicalLinearModel &at(const HierarchicalModelType &model_type) const;
 
    void initialise_population(RuntimeContext &context) override;
 
    void update_population(RuntimeContext &context) override;
 
    void apply_baseline_adjustments(RuntimeContext &context) override;
 
  private:
    std::map<HierarchicalModelType, std::unique_ptr<HierarchicalLinearModel>> models_;
    RiskfactorAdjustmentModel adjustment_;
    std::string name_{"RiskFactor"};
};
 
std::unique_ptr<RiskFactorModule> build_risk_factor_module(Repository &repository,
                                                           const ModelInput &config);
} // namespace hgps