steepest_descent.cpp

Go to the documentation of this file.

/*
 * Copyright (C) 2010-2022 The ESPResSo project
 * Copyright (C) 2002,2003,2004,2005,2006,2007,2008,2009,2010
 *   Max-Planck-Institute for Polymer Research, Theory Group
 *
 * 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/>.
 */
 
#include "integrators/steepest_descent.hpp"
 
#include "Particle.hpp"
#include "ParticleRange.hpp"
#include "cell_system/CellStructure.hpp"
#include "communication.hpp"
#include "config/config.hpp"
#include "rotation.hpp"
#include "system/System.hpp"
 
#include <utils/Vector.hpp>
#include <utils/math/sqr.hpp>
 
#include <boost/mpi/collectives/all_reduce.hpp>
#include <boost/mpi/operations.hpp>
 
#include <algorithm>
#include <cmath>
#include <limits>
#include <stdexcept>
 
/** Currently active steepest descent instance */
static SteepestDescentParameters params{0., 0., 0.};
 
bool steepest_descent_step(const ParticleRange &particles) {
  // Maximal force encountered on node
  auto f_max = -std::numeric_limits<double>::max();
 
  // Iteration over all local particles
  for (auto &p : particles) {
    auto f = 0.0;
 
    // For all Cartesian coordinates
    for (auto j = 0u; j < 3u; ++j) {
      // Skip, if coordinate is fixed
      if (!p.is_fixed_along(j)) {
        // Square of force on particle
        f += Utils::sqr(p.force()[j]);
 
        // Positional increment, crop to maximum allowed by user
        auto const dp =
            std::clamp(params.gamma * p.force()[j], -params.max_displacement,
                       params.max_displacement);
 
        // Move particle
        p.pos()[j] += dp;
      }
    }
#ifdef ROTATION
    {
      // Rotational increment
      auto const dq = params.gamma * p.torque(); // Vector parallel to torque
      auto const t = p.torque().norm2();
 
      // Normalize rotation axis and compute amount of rotation
      auto const l = dq.norm();
      if (l > 0.0) {
        auto const axis = dq / l;
        auto const angle =
            std::clamp(l, -params.max_displacement, params.max_displacement);
 
        // Rotate the particle around axis dq by amount l
        local_rotate_particle(p, axis, angle);
      }
 
      f_max = std::max(f_max, t);
    }
#endif
    // Note maximum force/torque encountered
    f_max = std::max(f_max, f);
  }
 
  auto &cell_structure = *System::get_system().cell_structure;
  cell_structure.set_resort_particles(Cells::RESORT_LOCAL);
 
  // Synchronize maximum force/torque encountered
  auto const f_max_global =
      boost::mpi::all_reduce(comm_cart, f_max, boost::mpi::maximum<double>());
 
  return sqrt(f_max_global) < params.f_max;
}
 
void register_integrator(SteepestDescentParameters const &obj) {
  ::params = obj;
}
 
SteepestDescentParameters::SteepestDescentParameters(
    const double f_max, const double gamma, const double max_displacement)
    : f_max{f_max}, gamma{gamma}, max_displacement{max_displacement} {
  if (f_max < 0.0) {
    throw std::runtime_error("The maximal force must be positive.");
  }
  if (gamma < 0.0) {
    throw std::runtime_error("The dampening constant must be positive.");
  }
  if (max_displacement < 0.0) {
    throw std::runtime_error("The maximal displacement must be positive.");
  }
}