lct_secir/parameters.h

Go to the documentation of this file.

/*
* Copyright (C) 2020-2026 MEmilio
*
* Authors: Lena Ploetzke
*
* Contact: Martin J. Kuehn <Martin.Kuehn@DLR.de>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
 
#ifndef LCT_SECIR_PARAMS_H
#define LCT_SECIR_PARAMS_H
 
#include "memilio/config.h"
#include "memilio/math/eigen.h"
#include "memilio/epidemiology/uncertain_matrix.h"
#include "memilio/utils/parameter_set.h"
#include "memilio/utils/logging.h"
#include "memilio/utils/uncertain_value.h"
 
namespace mio
{
namespace lsecir
{
 
/**********************************************
* Define Parameters of the LCT-SECIHURD model *
**********************************************/
 
template <typename FP>
struct TimeExposed {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 1.);
    }
    static std::string name()
    {
        return "TimeExposed";
    }
};
 
template <typename FP>
struct TimeInfectedNoSymptoms {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 1.);
    }
    static std::string name()
    {
        return "TimeInfectedNoSymptoms";
    }
};
 
template <typename FP>
struct TimeInfectedSymptoms {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 1.);
    }
    static std::string name()
    {
        return "TimeInfectedNoSymptoms";
    }
};
 
template <typename FP>
struct TimeInfectedSevere {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 1.);
    }
    static std::string name()
    {
        return "TimeInfectedSevere";
    }
};
 
template <typename FP>
struct TimeInfectedCritical {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 1.);
    }
    static std::string name()
    {
        return "TimeInfectedCritical";
    }
};
 
template <typename FP>
struct TransmissionProbabilityOnContact {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 1.);
    }
    static std::string name()
    {
        return "TransmissionProbabilityOnContact";
    }
};
 
template <typename FP>
struct ContactPatterns {
    using Type = UncertainContactMatrix<FP>;
 
    static Type get_default(size_t size)
    {
        mio::ContactMatrixGroup<FP> contact_matrix(1, (Eigen::Index)size);
        contact_matrix[0] =
            mio::ContactMatrix<FP>(Eigen::MatrixX<FP>::Constant((Eigen::Index)size, (Eigen::Index)size, 10.));
        return Type(contact_matrix);
    }
    static std::string name()
    {
        return "ContactPatterns";
    }
};
 
template <typename FP>
struct RelativeTransmissionNoSymptoms {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 1.);
    }
    static std::string name()
    {
        return "RelativeTransmissionNoSymptoms";
    }
};
 
template <typename FP>
struct RiskOfInfectionFromSymptomatic {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 1.);
    }
    static std::string name()
    {
        return "RiskOfInfectionFromSymptomatic";
    }
};
 
template <typename FP>
struct RecoveredPerInfectedNoSymptoms {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 0.5);
    }
    static std::string name()
    {
        return "RecoveredPerInfectedNoSymptoms";
    }
};
 
template <typename FP>
struct SeverePerInfectedSymptoms {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 0.5);
    }
    static std::string name()
    {
        return "SeverePerInfectedSymptoms";
    }
};
 
template <typename FP>
struct CriticalPerSevere {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 0.5);
    }
    static std::string name()
    {
        return "CriticalPerSevere";
    }
};
 
template <typename FP>
struct DeathsPerCritical {
    using Type = Eigen::VectorX<UncertainValue<FP>>;
    static Type get_default(size_t size)
    {
        return Type::Constant(size, 1, 0.1);
    }
    static std::string name()
    {
        return "DeathsPerCritical";
    }
};
 
template <typename FP>
struct StartDay {
    using Type = FP;
    static Type get_default(size_t)
    {
        return Type(0.0);
    }
    static std::string name()
    {
        return "StartDay";
    }
};
 
template <typename FP>
struct Seasonality {
    using Type = ScalarType;
    static Type get_default(size_t)
    {
        return Type(0.0);
    }
    static std::string name()
    {
        return "Seasonality";
    }
};
 
template <typename FP>
using ParametersBase =
    ParameterSet<TimeExposed<FP>, TimeInfectedNoSymptoms<FP>, TimeInfectedSymptoms<FP>, TimeInfectedSevere<FP>,
                 TimeInfectedCritical<FP>, TransmissionProbabilityOnContact<FP>, ContactPatterns<FP>,
                 RelativeTransmissionNoSymptoms<FP>, RiskOfInfectionFromSymptomatic<FP>,
                 RecoveredPerInfectedNoSymptoms<FP>, SeverePerInfectedSymptoms<FP>, CriticalPerSevere<FP>,
                 DeathsPerCritical<FP>, StartDay<FP>, Seasonality<FP>>;
 
template <typename FP>
class Parameters : public ParametersBase<FP>
{
public:
    Parameters(size_t num_groups)
        : ParametersBase<FP>(num_groups)
        , m_num_groups{num_groups}
    {
    }
 
    size_t get_num_groups() const
    {
        return m_num_groups;
    }
 
    bool check_constraints() const
    {
        if (this->template get<Seasonality<FP>>() < 0.0 || this->template get<Seasonality<FP>>() > 0.5) {
            log_warning("Constraint check: Parameter Seasonality should lie between {} and {}", 0.0, 0.5);
            return true;
        }
 
        for (size_t i = 0; i < m_num_groups; ++i) {
            if (this->template get<TimeExposed<FP>>()[i] < 1.0) {
                log_error("Constraint check: Parameter TimeExposed is smaller than {}", 1.0);
                return true;
            }
 
            if (this->template get<TimeInfectedNoSymptoms<FP>>()[i] < 1.0) {
                log_error("Constraint check: Parameter TimeInfectedNoSymptoms is smaller than {}", 1.0);
                return true;
            }
 
            if (this->template get<TimeInfectedSymptoms<FP>>()[i] < 1.0) {
                log_error("Constraint check: Parameter TimeInfectedSymptoms is smaller than {}", 1.0);
                return true;
            }
 
            if (this->template get<TimeInfectedSevere<FP>>()[i] < 1.0) {
                log_error("Constraint check: Parameter TimeInfectedSevere is smaller than {}", 1.0);
                return true;
            }
 
            if (this->template get<TimeInfectedCritical<FP>>()[i] < 1.0) {
                log_error("Constraint check: Parameter TimeInfectedCritical is smaller than {}", 1.0);
                return true;
            }
 
            if (this->template get<TransmissionProbabilityOnContact<FP>>()[i] < 0.0 ||
                this->template get<TransmissionProbabilityOnContact<FP>>()[i] > 1.0) {
                log_error("Constraint check: Parameter TransmissionProbabilityOnContact smaller {} or larger {}", 0, 1);
                return true;
            }
 
            if (this->template get<RelativeTransmissionNoSymptoms<FP>>()[i] < 0.0 ||
                this->template get<RelativeTransmissionNoSymptoms<FP>>()[i] > 1.0) {
                log_error("Constraint check: Parameter RelativeTransmissionNoSymptoms smaller {} or larger {}", 0, 1);
                return true;
            }
 
            if (this->template get<RiskOfInfectionFromSymptomatic<FP>>()[i] < 0.0 ||
                this->template get<RiskOfInfectionFromSymptomatic<FP>>()[i] > 1.0) {
                log_error("Constraint check: Parameter  RiskOfInfectionFromSymptomatic smaller {} or larger {}", 0, 1);
                return true;
            }
 
            if (this->template get<RecoveredPerInfectedNoSymptoms<FP>>()[i] < 0.0 ||
                this->template get<RecoveredPerInfectedNoSymptoms<FP>>()[i] > 1.0) {
                log_error("Constraint check: Parameter RecoveredPerInfectedNoSymptoms smaller {} or larger {}", 0, 1);
                return true;
            }
 
            if (this->template get<SeverePerInfectedSymptoms<FP>>()[i] < 0.0 ||
                this->template get<SeverePerInfectedSymptoms<FP>>()[i] > 1.0) {
                log_error("Constraint check: Parameter SeverePerInfectedSymptoms smaller {} or larger {}", 0, 1);
                return true;
            }
 
            if (this->template get<CriticalPerSevere<FP>>()[i] < 0.0 ||
                this->template get<CriticalPerSevere<FP>>()[i] > 1.0) {
                log_error("Constraint check: Parameter CriticalPerSevere smaller {} or larger {}", 0, 1);
                return true;
            }
 
            if (this->template get<DeathsPerCritical<FP>>()[i] < 0.0 ||
                this->template get<DeathsPerCritical<FP>>()[i] > 1.0) {
                log_error("Constraint check: Parameter DeathsPerCritical smaller {} or larger {}", 0, 1);
                return true;
            }
        }
 
        return false;
    }
 
private:
    Parameters(ParametersBase<FP>&& base)
        : ParametersBase<FP>(std::move(base))
        , m_num_groups(this->template get<ContactPatterns<FP>>().get_cont_freq_mat().get_num_groups())
    {
    }
 
    size_t m_num_groups;
 
public:
    template <class IOContext>
    static IOResult<Parameters> deserialize(IOContext& io)
    {
        BOOST_OUTCOME_TRY(auto&& base, ParametersBase<FP>::deserialize(io));
        return success(Parameters(std::move(base)));
    }
};
 
} // namespace lsecir
} // namespace mio
 
#endif // LCT_SECIR_PARAMS_H