/*

* Copyright (C) 2020-2026 MEmilio

*

* Authors: Nils Wassmuth, Rene Schmieding, Martin J. Kuehn

*

* Contact: Martin J. Kuehn <[email protected]>

*

* 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.

*/

#include "memilio/compartments/stochastic_simulation.h"

#include "memilio/data/analyze_result.h"

#include "memilio/utils/logging.h"

#include "sde_sirs/model.h"

int main()

{

mio::set_log_level(mio::LogLevel::debug);

ScalarType t0 = 0.;

ScalarType tmax = 5.;

ScalarType dt = 0.001;

ScalarType total_population = 10000;

mio::log_info("Simulating SIR; t={} ... {} with dt = {}.", t0, tmax, dt);

mio::ssirs::Model<ScalarType> model;

model.populations[{mio::Index<mio::ssirs::InfectionState>(mio::ssirs::InfectionState::Infected)}] = 100;

model.populations[{mio::Index<mio::ssirs::InfectionState>(mio::ssirs::InfectionState::Recovered)}] = 1000;

model.populations[{mio::Index<mio::ssirs::InfectionState>(mio::ssirs::InfectionState::Susceptible)}] =

total_population -

model.populations[{mio::Index<mio::ssirs::InfectionState>(mio::ssirs::InfectionState::Infected)}] -

model.populations[{mio::Index<mio::ssirs::InfectionState>(mio::ssirs::InfectionState::Recovered)}];

model.parameters.set<mio::ssirs::TimeInfected<ScalarType>>(10);

model.parameters.set<mio::ssirs::TimeImmune<ScalarType>>(100);

model.parameters.set<mio::ssirs::TransmissionProbabilityOnContact<ScalarType>>(1);

model.parameters.get<mio::ssirs::ContactPatterns<ScalarType>>().get_baseline()(0, 0) = 20.7;

model.parameters.get<mio::ssirs::ContactPatterns<ScalarType>>().add_damping(0.6,

mio::SimulationTime<ScalarType>(12.5));

model.parameters.set<mio::ssirs::StartDay<ScalarType>>(60);

model.parameters.set<mio::ssirs::Seasonality<ScalarType>>(0.2);

model.check_constraints();

auto ssirs = mio::simulate_stochastic(t0, tmax, dt, model);

// interpolate results

auto interpolated_results = mio::interpolate_simulation_result(ssirs);

interpolated_results.print_table({"Susceptible", "Infected", "Recovered"});

}