WIP on_lees_edwards_change

src/core/lb/LBNone.hpp

@@ -67,6 +67,7 @@ struct LBNone {
   void on_node_grid_change() const { throw NoLBActive{}; }
   void on_timestep_change() const { throw NoLBActive{}; }
   void on_temperature_change() const { throw NoLBActive{}; }
+  void on_lees_edwards_change() const { throw NoLBActive{}; }
 };
 
 } // namespace LB

src/core/lb/LBWalberla.cpp

@@ -30,6 +30,8 @@
 #include "system/System.hpp"
 #include "thermostat.hpp"
 
+#include "lees_edwards/lees_edwards.hpp"
+
 #include <walberla_bridge/lattice_boltzmann/LBWalberlaBase.hpp>
 
 #include <utils/Vector.hpp>
@@ -112,6 +114,13 @@ void LBWalberla::sanity_checks(System::System const &system) const {
                              system.get_time_step());
 }
 
+void LBWalberla::on_lees_edwards_change() {
+  std::cout << "Changed LEBC." << std::endl;
+  auto &system = System::get_system();
+  auto le_protocol = system.lees_edwards->get_protocol();
+  auto const &le_bc = system.box_geo->lees_edwards_bc();
+}
+
 } // namespace LB
 
 #endif // WALBERLA

src/core/lb/LBWalberla.hpp

@@ -88,6 +88,7 @@ struct LBWalberla {
   }
   void on_timestep_change() const {}
   void on_temperature_change() const {}
+  void on_lees_edwards_change();
 };
 
 } // namespace LB

src/core/lb/Solver.cpp

@@ -140,6 +140,12 @@ void Solver::on_temperature_change() {
   }
 }
 
+void Solver::on_lees_edwards_change() {
+  if (impl->solver) {
+    std::visit([](auto &ptr) { ptr->on_lees_edwards_change(); }, *impl->solver);
+  }
+}
+
 bool Solver::is_gpu() const {
   check_solver(impl);
   return std::visit([](auto &ptr) { return ptr->is_gpu(); }, *impl->solver);

src/core/lb/Solver.hpp

@@ -168,6 +168,7 @@ struct Solver : public System::Leaf<Solver> {
   void on_cell_structure_change();
   void on_timestep_change();
   void on_temperature_change();
+  void on_lees_edwards_change();
   void veto_boxl_change() const;
 
 private:

src/core/system/System.cpp

@@ -377,6 +377,10 @@ void System::on_observable_calc() {
   clear_particle_node();
 }
 
+void System::on_lees_edwards_change() {
+  lb.on_lees_edwards_change();
+}
+
 void System::update_local_geo() {
   *local_geo = LocalBox::make_regular_decomposition(
       box_geo->length(), ::communicator.calc_node_index(),

src/core/system/System.hpp

@@ -251,6 +251,7 @@ class System : public std::enable_shared_from_this<System> {
    *  initialized immediately (P3M etc.).
    */
   void on_observable_calc();
+  void on_lees_edwards_change();
   void veto_boxl_change(bool skip_particle_checks = false) const;
   /**@}*/
 

src/script_interface/lees_edwards/LeesEdwards.hpp

@@ -103,12 +103,13 @@ class LeesEdwards : public AutoParameters<LeesEdwards, System::Leaf> {
           throw std::invalid_argument("Parameters 'shear_direction' and "
                                       "'shear_plane_normal' must differ");
         }
-        auto const &system = get_system();
+        auto &system = get_system();
         system.lb.lebc_sanity_checks(shear_direction, shear_plane_normal);
         // update box geometry and cell structure
         system.box_geo->set_lees_edwards_bc(
             LeesEdwardsBC{0., 0., shear_direction, shear_plane_normal});
         m_lees_edwards->set_protocol(m_protocol->protocol());
+	system.on_lees_edwards_change();
       });
     }
     return {};