30#include "system/System.hpp"
31#include "thermostat.hpp"
54static auto is_solver_set(std::unique_ptr<Solver::Implementation>
const &ptr) {
55 return ptr !=
nullptr and ptr->solver.has_value();
58static void check_solver(std::unique_ptr<Solver::Implementation>
const &ptr) {
73 std::visit([](
auto &ptr) { ptr->propagate(); }, *impl->solver);
78 std::visit([](
auto &ptr) { ptr->ghost_communication(); }, *impl->solver);
83 std::visit([](
auto &ptr) { ptr->ghost_communication_pdf(); }, *impl->solver);
88 std::visit([](
auto &ptr) { ptr->ghost_communication_vel(); }, *impl->solver);
94 std::visit([&](
auto &ptr) { ptr->sanity_checks(system); }, *impl->solver);
100 std::visit([=](
auto &ptr) { ptr->veto_time_step(time_step); },
107 std::visit([=](
auto &ptr) { ptr->veto_kT(kT); }, *impl->solver);
112 unsigned int shear_plane_normal)
const {
114 auto const callback = [=](
auto &ptr) {
115 ptr->lebc_sanity_checks(shear_direction, shear_plane_normal);
117 std::visit(callback, *impl->solver);
123 auto &solver = *impl->solver;
124 std::visit([](
auto &ptr) { ptr->on_cell_structure_change(); }, solver);
130 std::visit([](
auto const &ptr) { ptr->veto_boxl_change(); }, *impl->solver);
136 std::visit([](
auto &ptr) { ptr->on_boxl_change(); }, *impl->solver);
142 std::visit([](
auto &ptr) { ptr->on_node_grid_change(); }, *impl->solver);
148 std::visit([](
auto &ptr) { ptr->on_timestep_change(); }, *impl->solver);
154 std::visit([](
auto &ptr) { ptr->on_temperature_change(); }, *impl->solver);
160 std::visit([](
auto &ptr) { ptr->on_lees_edwards_change(); }, *impl->solver);
166 std::visit([](
auto &ptr) { ptr->update_collision_model(); }, *impl->solver);
172 return std::visit([](
auto &ptr) {
return ptr->is_gpu(); }, *impl->solver);
177 return std::visit([](
auto &ptr) {
return ptr->get_agrid(); }, *impl->solver);
182 return std::visit([](
auto &ptr) {
return ptr->get_tau(); }, *impl->solver);
187 return std::visit([](
auto &ptr) {
return ptr->get_kT(); }, *impl->solver);
192 return std::visit([](
auto &ptr) {
return ptr->get_pressure_tensor(); },
196std::optional<Utils::Vector3d>
204 auto const lb_pos = box_geo.folded_position(pos) * m_conv.
pos_to_lb;
205 return ptr->get_velocity_at_pos(lb_pos,
false);
215 auto const lb_pos = box_geo.folded_position(pos) * m_conv.
pos_to_lb;
216 return ptr->get_density_at_pos(lb_pos,
false);
225 auto const res = ptr->get_velocity_at_pos(pos * m_conv.
pos_to_lb,
true);
233 std::vector<Utils::Vector3d>
const &pos)
const {
236 std::vector<Utils::Vector3d> pos_lb;
237 pos_lb.reserve(pos.size());
238 for (
auto const &pos_md : pos) {
239 pos_lb.emplace_back(pos_md * m_conv.
pos_to_lb);
241 auto res = ptr->get_velocities_at_pos(pos_lb);
242 for (
auto &v : res) {
251 std::vector<Utils::Vector3d>
const &forces) {
254 std::vector<Utils::Vector3d> pos_lb;
255 std::vector<Utils::Vector3d> force_lb;
256 pos_lb.reserve(pos.size());
257 force_lb.reserve(pos.size());
258 for (
auto const &pos_md : pos) {
259 pos_lb.emplace_back(pos_md * m_conv.
pos_to_lb);
261 for (
auto const &force_md : forces) {
262 force_lb.emplace_back(force_md * m_conv.
force_to_lb);
264 ptr->add_forces_at_pos(pos_lb, force_lb);
273 if (not ptr->add_force_at_pos(pos * m_conv.
pos_to_lb,
275 throw std::runtime_error(
"Cannot apply force to LB");
283 return std::visit([](
auto const &ptr) {
return ptr->get_momentum(); },
287template <>
void Solver::set<LBNone>(std::shared_ptr<LBNone> lb_instance) {
289 assert(not impl->solver.has_value());
290 impl->solver = lb_instance;
295void Solver::set<LBWalberla>(std::shared_ptr<LBWalberlaBase> lb_fluid,
296 std::shared_ptr<LBWalberlaParams> lb_params) {
298 assert(not impl->solver.has_value());
300 auto lb_instance = std::make_shared<LBWalberla>(lb_fluid, lb_params);
301 lb_instance->sanity_checks(system);
302 auto const &lebc = system.box_geo->lees_edwards_bc();
303 lb_fluid->check_lebc(lebc.shear_direction, lebc.shear_plane_normal);
304 impl->solver = lb_instance;
305 auto const agrid = lb_instance->get_agrid();
306 auto const tau = lb_instance->get_tau();
307 m_conv =
Conversions{1. / agrid, agrid / tau, tau * tau / agrid};
LBWalberlaBase provides the public interface of the LB waLBerla bridge.
Vector implementation and trait types for boost qvm interoperability.
std::shared_ptr< BoxGeometry > box_geo
This file contains the defaults for ESPResSo.
static auto is_solver_set(std::unique_ptr< Solver::Implementation > const &ptr)
static void check_solver(std::unique_ptr< Solver::Implementation > const &ptr)
void veto_boxl_change() const
void lebc_sanity_checks(unsigned int shear_direction, unsigned int shear_plane_normal) const
Perform LB LEbc parameter checks.
void add_force_density(Utils::Vector3d const &pos, Utils::Vector3d const &force_density)
Add a force density to the fluid at the given position.
Utils::Vector3d get_coupling_interpolated_velocity(Utils::Vector3d const &pos) const
Calculate the interpolated fluid velocity in MD units.
void sanity_checks() const
Perform LB parameter and boundary velocity checks.
void on_timestep_change()
void ghost_communication()
Perform a full ghost communication.
void propagate()
Propagate the LB fluid.
void reset()
Remove the LB solver.
std::optional< Utils::Vector3d > get_interpolated_velocity(Utils::Vector3d const &pos) const
Calculate the interpolated fluid velocity in LB units.
bool is_solver_set() const
Return true if a LB solver is active.
void ghost_communication_pdf()
Perform a ghost communication of the PDF field.
double get_tau() const
Get the LB time step.
void on_temperature_change()
std::optional< double > get_interpolated_density(Utils::Vector3d const &pos) const
Calculate the interpolated fluid density in LB units.
void add_forces_at_pos(std::vector< Utils::Vector3d > const &pos, std::vector< Utils::Vector3d > const &forces)
void veto_kT(double kT) const
Check if a thermostat is compatible with the LB temperature.
double get_agrid() const
Get the LB grid spacing.
Utils::VectorXd< 9 > get_pressure_tensor() const
void update_collision_model()
void veto_time_step(double time_step) const
Check if a MD time step is compatible with the LB tau.
void ghost_communication_vel()
Perform a ghost communication of the velocity field.
void on_lees_edwards_change()
void on_node_grid_change()
Utils::Vector3d get_momentum() const
void on_cell_structure_change()
std::vector< Utils::Vector3d > get_coupling_interpolated_velocities(std::vector< Utils::Vector3d > const &pos) const
double get_kT() const
Get the thermal energy parameter of the LB fluid.