InertialModel: Init of actual instantiation

Co-authored-by: Moritz Sallermann <[email protected]>
seldon-code · Mar 23, 2024 · dee1ae6 · dee1ae6
1 parent 1d7e838
commit dee1ae6
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/include/config_parser.hpp b/include/config_parser.hpp
@@ -25,6 +25,7 @@ enum class Model
 {
     DeGroot,
     ActivityDrivenModel,
+    InertialModel,
     DeffuantModel
 };
 
@@ -81,6 +82,11 @@ struct ActivityDrivenSettings
     double covariance_factor          = 0.0;
 };
 
+struct InertialSettings : public ActivityDrivenSettings
+{
+    double friction_coefficient = 1.0;
+};
+
 struct InitialNetworkSettings
 {
     std::optional<std::string> file;

diff --git a/include/models/InertialModel.hpp b/include/models/InertialModel.hpp
@@ -24,6 +24,8 @@ class InertialModel : public ActivityDrivenModelAbstract<InertialAgent>
     using NetworkT = Network<AgentT>;
     using WeightT  = typename NetworkT::WeightT;
 
+    InertialModel( const Config::InertialSettings & settings, NetworkT & network, std::mt19937 & gen );
+
     void iteration() override;
 
 private:

diff --git a/src/models/InertialModel.cpp b/src/models/InertialModel.cpp
@@ -9,6 +9,12 @@
 namespace Seldon
 {
 
+InertialModel::InertialModel( const Config::InertialSettings & settings, NetworkT & network, std::mt19937 & gen )
+        : ActivityDrivenModelAbstract<InertialAgent>( settings, network, gen ),
+          friction_coefficient( settings.friction_coefficient )
+{
+}
+
 // X(t+dt)
 // Also updates the position
 void InertialModel::calc_position()