1003 implement ode seir metapopulation model

cpp/CMakeLists.txt

@@ -163,6 +163,7 @@ if(MEMILIO_BUILD_MODELS)
     add_subdirectory(models/ode_seir)
     add_subdirectory(models/ode_seair)
     add_subdirectory(models/ode_sir)
+    add_subdirectory(models/ode_seir_metapop)
     add_subdirectory(models/sde_sir)
     add_subdirectory(models/sde_sirs)
     add_subdirectory(models/sde_seirvv)

cpp/examples/CMakeLists.txt

@@ -96,6 +96,10 @@ add_executable(graph_stochastic_mobility_example graph_stochastic_mobility.cpp)
 target_link_libraries(graph_stochastic_mobility_example PRIVATE memilio ode_secir)
 target_compile_options(graph_stochastic_mobility_example PRIVATE ${MEMILIO_CXX_FLAGS_ENABLE_WARNING_ERRORS})
 
+add_executable(ode_metapop_seir_example ode_seir_metapop.cpp)
+target_link_libraries(ode_metapop_seir_example PRIVATE memilio ode_seir_metapop)
+target_compile_options(ode_metapop_seir_example PRIVATE ${MEMILIO_CXX_FLAGS_ENABLE_WARNING_ERRORS})
+
 add_executable(abm_minimal_example abm_minimal.cpp)
 target_link_libraries(abm_minimal_example PRIVATE memilio abm)
 target_compile_options(abm_minimal_example PRIVATE ${MEMILIO_CXX_FLAGS_ENABLE_WARNING_ERRORS})

cpp/examples/ode_seir_metapop.cpp

@@ -0,0 +1,42 @@
+#include "memilio/compartments/simulation.h"
+#include "memilio/utils/custom_index_array.h"
+#include "models/ode_seir_metapop/infection_state.h"
+#include "models/ode_seir_metapop/model.h"
+#include "models/ode_seir_metapop/parameters.h"
+#include "models/ode_seir_metapop/regions.h"
+
+int main()
+{
+    const ScalarType t0   = 0.;
+    const ScalarType tmax = 10;
+    ScalarType dt         = 0.1;
+
+    mio::oseirmetapop::Model<ScalarType> model(3, 1);
+
+    for (size_t i = 0; i < (size_t)model.parameters.get_num_regions(); i++) {
+        model.populations[{mio::oseirmetapop::Region(i), mio::AgeGroup(0),
+                           mio::oseirmetapop::InfectionState::Susceptible}] = 10000;
+    }
+
+    model.populations[{mio::oseirmetapop::Region(0), mio::AgeGroup(0), mio::oseirmetapop::InfectionState::Exposed}] +=
+        100;
+    model.populations[{mio::oseirmetapop::Region(0), mio::AgeGroup(0),
+                       mio::oseirmetapop::InfectionState::Susceptible}] -= 100;
+
+    Eigen::MatrixXd mobility_data_commuter(3, 3);
+    mobility_data_commuter << 0.4, 0.3, 0.3, 0.2, 0.7, 0.1, 0.4, 0.1, 0.5;
+
+    model.set_commuting_strengths(mobility_data_commuter);
+
+    model.parameters.template get<mio::oseirmetapop::ContactPatterns<>>()
+        .get_cont_freq_mat()[0]
+        .get_baseline()
+        .setConstant(2.7);
+
+    model.parameters.set<mio::oseirmetapop::TimeExposed<>>(3.335);
+    model.parameters.set<mio::oseirmetapop::TimeInfected<>>(8.097612257);
+    model.parameters.set<mio::oseirmetapop::TransmissionProbabilityOnContact<>>(0.07333);
+
+    auto result = simulate(t0, tmax, dt, model);
+    return 0;
+}

cpp/models/ode_seir_metapop/CMakeLists.txt

@@ -0,0 +1,13 @@
+add_library(ode_seir_metapop
+    infection_state.h
+    model.h
+    model.cpp
+    parameters.h
+    regions.h
+)
+target_link_libraries(ode_seir_metapop PUBLIC memilio)
+target_include_directories(ode_seir_metapop PUBLIC
+    $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/..>
+    $<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>
+)
+target_compile_options(ode_seir_metapop PRIVATE ${MEMILIO_CXX_FLAGS_ENABLE_WARNING_ERRORS})

cpp/models/ode_seir_metapop/infection_state.h

@@ -0,0 +1,44 @@
+/* 
+* Copyright (C) 2020-2025 MEmilio
+*
+* Authors: Carlotta Gerstein
+*
+* 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 ODESEIRMETAPOP_INFECTIONSTATE_H
+#define ODESEIRMETAPOP_INFECTIONSTATE_H
+
+namespace mio
+{
+namespace oseirmetapop
+{
+
+/**
+ * @brief The InfectionState enum describes the possible
+ * categories for the infectious state of persons
+ */
+enum class InfectionState
+{
+    Susceptible,
+    Exposed,
+    Infected,
+    Recovered,
+    Count
+};
+
+} // namespace oseirmetapop
+} // namespace mio
+
+#endif // ODESEIR_INFECTIONSTATE_H

cpp/models/ode_seir_metapop/model.cpp

@@ -0,0 +1,28 @@
+/* 
+* Copyright (C) 2020-2025 MEmilio
+*
+* Authors: Carlotta Gerstein
+*
+* 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.
+*/
+#include "ode_seir_metapop/model.h"
+
+namespace mio
+{
+namespace oseirmetapop
+{
+
+} // namespace oseirmetapop
+} // namespace mio