models/smm/simulation.h

Go to the documentation of this file.

/*
* Copyright (C) 2020-2026 MEmilio
*
* Authors: René Schmieding, Julia Bicker
*
* 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 MIO_SMM_SIMULATION_H
#define MIO_SMM_SIMULATION_H
 
#include "smm/model.h"
#include "smm/parameters.h"
#include "memilio/utils/time_series.h"
 
namespace mio
{
 
namespace smm
{
 
template <typename FP, class Comp, class Status = Comp, class Region = mio::regions::Region>
class Simulation
{
public:
    using Model = smm::Model<FP, Comp, Status, Region>;
 
    Simulation(Model const& model, FP t0 = 0., FP dt = 1.)
        : m_dt(dt)
        , m_model(std::make_unique<Model>(model))
        , m_result(t0, m_model->get_initial_values())
        , m_internal_time(adoption_rates().size() + transition_rates().size(), t0)
        , m_tp_next_event(adoption_rates().size() + transition_rates().size(), t0)
        , m_waiting_times(adoption_rates().size() + transition_rates().size(), 0)
        , m_current_rates(adoption_rates().size() + transition_rates().size(), 0)
    {
        assert(dt > 0);
        assert(m_waiting_times.size() > 0);
        assert(std::all_of(adoption_rates().begin(), adoption_rates().end(),
                           [regions = reduce_index<Region>(model.populations.size())](auto&& r) {
                               return r.region < regions;
                           }));
        assert(std::all_of(transition_rates().begin(), transition_rates().end(),
                           [regions = reduce_index<Region>(model.populations.size())](auto&& r) {
                               return r.from < regions && r.to < regions;
                           }));
        // initialize (internal) next event times by random values
        for (size_t i = 0; i < m_tp_next_event.size(); i++) {
            m_tp_next_event[i] += mio::ExponentialDistribution<FP>::get_instance()(m_model->get_rng(), 1.0);
        }
    }
 
    Eigen::Ref<Eigen::VectorX<FP>> advance(FP tmax)
    {
        update_current_rates_and_waiting_times();
        size_t next_event = determine_next_event(); // index of the next event
        FP current_time   = m_result.get_last_time();
        // set next result time; system state is saved every dt time step
        FP next_result_time = current_time + m_dt;
        // iterate over time
        while (current_time + m_waiting_times[next_event] < tmax) {
            // update time
            current_time += m_waiting_times[next_event];
            // Save results until the next event time point
            // do not save current time, as it does not yet include the next event
            while (next_result_time < current_time) {
                m_result.add_time_point(next_result_time);
                m_result.get_last_value() = m_model->populations.get_compartments();
                next_result_time += m_dt;
            }
            // decide event type by index and perform it
            if (next_event < adoption_rates().size()) {
                // perform adoption event
                const auto& rate = adoption_rates()[next_event];
                m_model->populations[{rate.region, rate.from}] -= 1.0;
                m_model->populations[{rate.region, rate.to}] += 1.0;
            }
            else {
                // perform transition event
                const auto& rate = transition_rates()[next_event - adoption_rates().size()];
                m_model->populations[{rate.from, rate.status}] -= 1.0;
                m_model->populations[{rate.to, rate.status}] += 1.0;
            }
            // update internal times
            for (size_t i = 0; i < m_internal_time.size(); i++) {
                m_internal_time[i] += m_current_rates[i] * m_waiting_times[next_event];
            }
            // draw new "next event" time for the occured event
            m_tp_next_event[next_event] += mio::ExponentialDistribution<FP>::get_instance()(m_model->get_rng(), 1.0);
            // precalculate next event
            update_current_rates_and_waiting_times();
            next_event = determine_next_event();
        }
        // copy last result, if no event occurs between last result time point and tmax
        if (m_result.get_last_time() < tmax) {
            while (next_result_time <= tmax) {
                m_result.add_time_point(next_result_time);
                m_result.get_last_value() = m_model->populations.get_compartments();
                next_result_time += m_dt;
            }
            // update internal times
            for (size_t i = 0; i < m_internal_time.size(); i++) {
                m_internal_time[i] += m_current_rates[i] * (tmax - current_time);
            }
        }
        return m_result.get_last_value();
    }
 
    TimeSeries<FP>& get_result()
    {
        return m_result;
    }
    const TimeSeries<FP>& get_result() const
    {
        return m_result;
    }
 
    const Model& get_model() const
    {
        return *m_model;
    }
    Model& get_model()
    {
        return *m_model;
    }
 
private:
    inline constexpr const typename smm::TransitionRates<FP, Status, Region>::Type& transition_rates()
    {
        return m_model->parameters.template get<smm::TransitionRates<FP, Status, Region>>();
    }
 
    inline constexpr const typename smm::AdoptionRates<FP, Status, Region>::Type& adoption_rates()
    {
        return m_model->parameters.template get<smm::AdoptionRates<FP, Status, Region>>();
    }
 
    inline void update_current_rates_and_waiting_times()
    {
        size_t i               = 0; // shared index for iterating both rates
        const auto last_values = m_model->populations.get_compartments();
        for (const auto& rate : adoption_rates()) {
            m_current_rates[i] = m_model->evaluate(rate, last_values);
            m_waiting_times[i] = (m_current_rates[i] > 0)
                                     ? (m_tp_next_event[i] - m_internal_time[i]) / m_current_rates[i]
                                     : std::numeric_limits<FP>::max();
            i++;
        }
        for (const auto& rate : transition_rates()) {
            m_current_rates[i] = m_model->evaluate(rate, last_values);
            m_waiting_times[i] = (m_current_rates[i] > 0)
                                     ? (m_tp_next_event[i] - m_internal_time[i]) / m_current_rates[i]
                                     : std::numeric_limits<FP>::max();
            i++;
        }
    }
 
    inline size_t determine_next_event()
    {
        return std::distance(m_waiting_times.begin(), std::min_element(m_waiting_times.begin(), m_waiting_times.end()));
    }
 
    FP m_dt; 
    std::unique_ptr<Model> m_model; 
    mio::TimeSeries<FP> m_result; 
 
    std::vector<FP> m_internal_time; 
    std::vector<FP> m_tp_next_event; 
    std::vector<FP> m_waiting_times; 
    std::vector<FP> m_current_rates; 
};
 
} // namespace smm
} // namespace mio
 
#endif