progress

espressomd · Dec 20, 2023 · f6179ec · f6179ec
1 parent 5a937e5
commit f6179ec
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/src/script_interface/system/System.cpp b/src/script_interface/system/System.cpp
@@ -73,6 +73,7 @@ struct System::Leaves {
   Leaves() = default;
   std::shared_ptr<CellSystem::CellSystem> cell_system;
   std::shared_ptr<Integrators::IntegratorHandle> integrator;
+  std::shared_ptr<Thermostats::Thermostats> thermostats;
   std::shared_ptr<Analysis::Analysis> analysis;
   std::shared_ptr<Galilei::ComFixed> comfixed;
   std::shared_ptr<Galilei::Galilei> galilei;
@@ -84,7 +85,6 @@ struct System::Leaves {
 #ifdef DIPOLES
   std::shared_ptr<Dipoles::Container> magnetostatics;
 #endif
-  std::shared_ptr<Thermostats::Thermostats> thermostats;
 };
 
 System::System() : m_instance{}, m_leaves{std::make_shared<Leaves>()} {

diff --git a/src/script_interface/thermostat/thermostats.hpp b/src/script_interface/thermostat/thermostats.hpp
@@ -237,6 +237,14 @@ std::shared_ptr<T> make_new_thermostat(char const* so_name, VariantMap const &pa
 }
 
 public:
+  Thermostats() {
+    add_parameters({
+      {"langevin",
+       AutoParameter::read_only,
+       [this]() { return langevin? Variant{langevin} : Variant{None{}}; }},
+    });
+  }
+
   Variant do_call_method(std::string const &name,
                          VariantMap const &params) override {
     if (name == "set_brownian") {

diff --git a/testsuite/python/propagation_langevin.py b/testsuite/python/propagation_langevin.py
@@ -208,6 +208,8 @@ def test_07__virtual(self):
 
         system.thermostat.set_langevin(
             kT=0, gamma=1, gamma_rotation=1., seed=41)
+        print(system.thermostat)
+        print(system.thermostats.langevin.get_params())
 
         system.integrator.run(0)