LBWalberlaBase.hpp

Go to the documentation of this file.

/*
 * Copyright (C) 2019-2023 The ESPResSo project
 *
 * This file is part of ESPResSo.
 *
 * ESPResSo is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * ESPResSo is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#pragma once
 
/**
 * @file
 * @ref LBWalberlaBase provides the public interface of the LB
 * waLBerla bridge. It relies on type erasure to hide the waLBerla
 * implementation details from the ESPResSo core. It is implemented
 * by @ref walberla::LBWalberlaImpl.
 */
 
#include <walberla_bridge/LatticeModel.hpp>
#include <walberla_bridge/lattice_boltzmann/LeesEdwardsPack.hpp>
 
#include <utils/Vector.hpp>
 
#include <cstddef>
#include <functional>
#include <memory>
#include <optional>
#include <vector>
 
/** @brief Interface of a lattice-based fluid model. */
class LBWalberlaBase : public LatticeModel {
public:
  ~LBWalberlaBase() override = default;
 
  /**
   * @brief Integrate LB for one time step.
   * The ghost layer may be out-of-date after integration.
   * Call @ref ghost_communication() to refresh them before
   * calling any getter function that reads from the halo region.
   */
  virtual void integrate() = 0;
 
  /** @brief Perform a full ghost communication. */
  virtual void ghost_communication() = 0;
 
  /** @brief Perform a ghost communication of the PDF field. */
  virtual void ghost_communication_pdf() = 0;
 
  /** @brief Perform a ghost communication of the velocity field. */
  virtual void ghost_communication_vel() = 0;
 
  /** @brief Perform a ghost communication of the last applied forces field. */
  virtual void ghost_communication_laf() = 0;
 
  /** @brief Number of discretized velocities in the PDF. */
  virtual std::size_t stencil_size() const noexcept = 0;
 
  /** @brief Whether kernels use double-precision floating point numbers. */
  [[nodiscard]] virtual bool is_double_precision() const noexcept = 0;
 
  /** @brief Make a functor to check if a position is in the local domain. */
  virtual std::function<bool(Utils::Vector3d const &)>
  make_lattice_position_checker(bool consider_points_in_halo) const = 0;
 
  /** @brief Get interpolated velocities at a position. */
  virtual std::optional<Utils::Vector3d>
  get_velocity_at_pos(Utils::Vector3d const &position,
                      bool consider_points_in_halo = false) const = 0;
 
  /** @brief Get interpolated velocities at positions. */
  virtual std::vector<Utils::Vector3d>
  get_velocities_at_pos(std::vector<Utils::Vector3d> const &pos) = 0;
 
  /** @brief Get interpolated densities at a position. */
  virtual std::optional<double>
  get_density_at_pos(Utils::Vector3d const &position,
                     bool consider_points_in_halo = false) const = 0;
 
  /** @brief Get interpolated densities at positions. */
  virtual std::vector<double>
  get_densities_at_pos(std::vector<Utils::Vector3d> const &pos) = 0;
 
  /**
   * @brief Interpolate a force to the stored forces to be applied on nodes
   * in the next time step.
   */
  virtual bool add_force_at_pos(Utils::Vector3d const &position,
                                Utils::Vector3d const &force) = 0;
 
  /**
   * @brief Interpolate forces to the stored forces to be applied on nodes
   * in the next time step.
   */
  virtual void
  add_forces_at_pos(std::vector<Utils::Vector3d> const &positions,
                    std::vector<Utils::Vector3d> const &forces) = 0;
 
  /** @brief Get stored force to be applied on node in the next time step. */
  virtual std::optional<Utils::Vector3d>
  get_node_force_to_be_applied(Utils::Vector3i const &node) const = 0;
 
  /** @brief Get stored force that was applied on node in the last time step. */
  virtual std::optional<Utils::Vector3d>
  get_node_last_applied_force(Utils::Vector3i const &node,
                              bool consider_ghosts = false) const = 0;
 
  /** @brief Set stored force that was applied on node in the last time step. */
  virtual bool set_node_last_applied_force(Utils::Vector3i const &node,
                                           Utils::Vector3d const &force) = 0;
 
  /** @brief Get stored force that was applied on slice in the last time step.
   */
  virtual std::vector<double>
  get_slice_last_applied_force(Utils::Vector3i const &lower_corner,
                               Utils::Vector3i const &upper_corner) const = 0;
 
  /** @brief Set stored force that was applied on slice in the last time step.
   */
  virtual void
  set_slice_last_applied_force(Utils::Vector3i const &lower_corner,
                               Utils::Vector3i const &upper_corner,
                               std::vector<double> const &force) = 0;
 
  /** @brief Get node population. */
  virtual std::optional<std::vector<double>>
  get_node_population(Utils::Vector3i const &node,
                      bool consider_ghosts = false) const = 0;
 
  /** @brief Set node population. */
  virtual bool set_node_population(Utils::Vector3i const &node,
                                   std::vector<double> const &population) = 0;
 
  /** @brief Get slice population. */
  virtual std::vector<double>
  get_slice_population(Utils::Vector3i const &lower_corner,
                       Utils::Vector3i const &upper_corner) const = 0;
 
  /** @brief Set slice population. */
  virtual void set_slice_population(Utils::Vector3i const &lower_corner,
                                    Utils::Vector3i const &upper_corner,
                                    std::vector<double> const &population) = 0;
 
  /** @brief Get node velocity. */
  virtual std::optional<Utils::Vector3d>
  get_node_velocity(Utils::Vector3i const &node,
                    bool consider_ghosts = false) const = 0;
 
  /** @brief Set node velocity. */
  virtual bool set_node_velocity(Utils::Vector3i const &node,
                                 Utils::Vector3d const &v) = 0;
 
  /** @brief Get slice velocity. */
  virtual std::vector<double>
  get_slice_velocity(Utils::Vector3i const &lower_corner,
                     Utils::Vector3i const &upper_corner) const = 0;
 
  /** @brief Set slice velocity. */
  virtual void set_slice_velocity(Utils::Vector3i const &lower_corner,
                                  Utils::Vector3i const &upper_corner,
                                  std::vector<double> const &velocity) = 0;
 
  /** @brief Get node density. */
  virtual std::optional<double>
  get_node_density(Utils::Vector3i const &node,
                   bool consider_ghosts = false) const = 0;
 
  /** @brief Set node density. */
  virtual bool set_node_density(Utils::Vector3i const &node,
                                double density) = 0;
 
  /** @brief Get slice density. */
  virtual std::vector<double>
  get_slice_density(Utils::Vector3i const &lower_corner,
                    Utils::Vector3i const &upper_corner) const = 0;
 
  /** @brief Set slice density. */
  virtual void set_slice_density(Utils::Vector3i const &lower_corner,
                                 Utils::Vector3i const &upper_corner,
                                 std::vector<double> const &density) = 0;
 
  /** @brief Get node velocity boundary conditions. */
  virtual std::optional<Utils::Vector3d>
  get_node_velocity_at_boundary(Utils::Vector3i const &node,
                                bool consider_ghosts = false) const = 0;
 
  /** @brief Set node velocity boundary conditions. */
  virtual bool
  set_node_velocity_at_boundary(Utils::Vector3i const &node,
                                Utils::Vector3d const &velocity) = 0;
 
  /** @brief Get slice velocity boundary conditions. */
  virtual std::vector<std::optional<Utils::Vector3d>>
  get_slice_velocity_at_boundary(Utils::Vector3i const &lower_corner,
                                 Utils::Vector3i const &upper_corner) const = 0;
 
  /** @brief Set slice velocity boundary conditions. */
  virtual void set_slice_velocity_at_boundary(
      Utils::Vector3i const &lower_corner, Utils::Vector3i const &upper_corner,
      std::vector<std::optional<Utils::Vector3d>> const &velocity) = 0;
 
  /** @brief Get (stored) force applied on node due to boundary condition. */
  virtual std::optional<Utils::Vector3d>
  get_node_boundary_force(Utils::Vector3i const &node) const = 0;
 
  /** @brief Remove a node from the boundaries. */
  virtual bool remove_node_from_boundary(Utils::Vector3i const &node) = 0;
 
  /** @brief Check if node has velocity boundary conditions. */
  virtual std::optional<bool>
  get_node_is_boundary(Utils::Vector3i const &node,
                       bool consider_ghosts = false) const = 0;
 
  /** @brief Check if slice has velocity boundary conditions. */
  virtual std::vector<bool>
  get_slice_is_boundary(Utils::Vector3i const &lower_corner,
                        Utils::Vector3i const &upper_corner) const = 0;
 
  /** @brief Rebuild the UBB field. This is an expensive operation. */
  virtual void reallocate_ubb_field() = 0;
 
  /** @brief Clear the boundary flag field and the UBB field. */
  virtual void clear_boundaries() = 0;
 
  /** @brief Update boundary conditions from a rasterized shape. */
  virtual void update_boundary_from_shape(std::vector<int> const &,
                                          std::vector<double> const &) = 0;
 
  /** @brief Configure the default collision model. */
  virtual void set_collision_model(double kT, unsigned int seed) = 0;
 
  /** @brief Configure a thermalized collision model for Lees-Edwards. */
  virtual void
  set_collision_model(std::unique_ptr<LeesEdwardsPack> &&lees_edwards_pack) = 0;
 
  /** @brief Check Lees-Edwards boundary conditions. */
  virtual void check_lebc(unsigned int shear_direction,
                          unsigned int shear_plane_normal) const = 0;
 
  /** @brief Get node pressure tensor. */
  virtual std::optional<Utils::VectorXd<9>>
  get_node_pressure_tensor(Utils::Vector3i const &node) const = 0;
 
  /** @brief Get slice pressure tensor. */
  virtual std::vector<double>
  get_slice_pressure_tensor(Utils::Vector3i const &lower_corner,
                            Utils::Vector3i const &upper_corner) const = 0;
 
  /** @brief Calculate boundary force from a rasterized shape. */
  virtual Utils::Vector3d
  get_boundary_force_from_shape(std::vector<int> const &raster_flat) const = 0;
 
  /** @brief Calculate boundary force of the local domain. */
  virtual Utils::Vector3d get_boundary_force() const = 0;
 
  /** @brief Calculate average pressure tensor of the local domain. */
  virtual Utils::VectorXd<9> get_pressure_tensor() const = 0;
 
  /** @brief Calculate momentum of the local domain. */
  virtual Utils::Vector3d get_momentum() const = 0;
 
  /** @brief Set a global external force. */
  virtual void set_external_force(Utils::Vector3d const &ext_force) = 0;
 
  /** @brief Get the global external force. */
  virtual Utils::Vector3d get_external_force() const noexcept = 0;
 
  /** @brief Set the fluid viscosity. */
  virtual void set_viscosity(double viscosity) = 0;
 
  /** @brief Get the fluid viscosity. */
  virtual double get_viscosity() const noexcept = 0;
 
  /** @brief Get the fluid density. */
  virtual double get_density() const noexcept = 0;
 
  /** @brief Get the fluid temperature (if thermalized). */
  virtual double get_kT() const noexcept = 0;
 
  /** @brief Get the RNG seed (if thermalized). */
  virtual unsigned int get_seed() const noexcept = 0;
 
  /** @brief Set the RNG counter (if thermalized). */
  [[nodiscard]] virtual std::optional<uint64_t> get_rng_state() const = 0;
 
  /** @brief Set the RNG counter (if thermalized). */
  virtual void set_rng_state(uint64_t counter) = 0;
 
  /** @brief Get the velocity field id */
  [[nodiscard]] virtual std::size_t get_velocity_field_id() const noexcept = 0;
 
  /** @brief Get the force field id */
  [[nodiscard]] virtual std::size_t get_force_field_id() const noexcept = 0;
 
  /** @brief Get whether the kernels run on GPUs. */
  [[nodiscard]] virtual bool is_gpu() const noexcept = 0;
};