core/galilei/ComFixed.hpp

Go to the documentation of this file.

/*
 * Copyright (C) 2017-2022 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
 
#include "ParticleRange.hpp"
#include "communication.hpp"
 
#include <utils/Vector.hpp>
 
#include <boost/mpi/collectives/all_reduce.hpp>
#include <boost/mpi/communicator.hpp>
 
#include <algorithm>
#include <functional>
#include <ranges>
#include <unordered_map>
#include <vector>
 
class ComFixed {
private:
  std::unordered_map<int, int> m_type_index;
 
  std::vector<Utils::Vector3d>
  local_type_forces(ParticleRange const &particles) const {
    std::vector<Utils::Vector3d> ret(m_type_index.size(), Utils::Vector3d{});
 
    for (auto const &p : particles) {
      /* Check if type is of interest */
      auto it = m_type_index.find(p.type());
      if (it != m_type_index.end()) {
        ret[it->second] += p.force();
      }
    }
 
    return ret;
  }
 
  std::vector<double> local_type_masses(ParticleRange const &particles) const {
    std::vector<double> ret(m_type_index.size(), 0.);
 
    for (auto const &p : particles) {
      /* Check if type is of interest */
      auto it = m_type_index.find(p.type());
      if (it != m_type_index.end()) {
        ret[it->second] += p.mass();
      }
    }
 
    return ret;
  }
 
public:
  ComFixed() = default;
 
  void set_fixed_types(std::vector<int> const &p_types) {
    m_type_index.clear();
 
    int i = 0;
    for (auto const &p_type : p_types) {
      m_type_index[p_type] = i++;
    }
  }
 
  std::vector<int> get_fixed_types() const {
    std::vector<int> res{};
    std::ranges::copy(std::views::keys(m_type_index), std::back_inserter(res));
    return res;
  }
 
  void apply(ParticleRange const &particles) const {
    /* Bail out early if there is nothing to do. */
    if (m_type_index.empty())
      return;
 
    auto const local_forces = local_type_forces(particles);
    auto const local_masses = local_type_masses(particles);
 
    /* Total forces and masses of the types. */
    std::vector<Utils::Vector3d> forces(m_type_index.size(), Utils::Vector3d{});
    std::vector<double> masses(m_type_index.size(), 0.0);
 
    /* Add contributions from all nodes and redistribute them to all. */
    boost::mpi::all_reduce(::comm_cart, local_forces.data(),
                           static_cast<int>(local_forces.size()), forces.data(),
                           std::plus<Utils::Vector3d>{});
    boost::mpi::all_reduce(::comm_cart, local_masses.data(),
                           static_cast<int>(local_masses.size()), masses.data(),
                           std::plus<double>{});
 
    for (auto &p : particles) {
      /* Check if type is of interest */
      auto it = m_type_index.find(p.type());
      if (it != m_type_index.end()) {
        auto const mass_frac = p.mass() / masses[it->second];
        auto const &type_force = forces[it->second];
        for (unsigned int i = 0u; i < 3u; i++) {
          p.force()[i] -= mass_frac * type_force[i];
        }
      }
    }
  }
};