Expose init_interaction_const and move to correct header/source
[alexxy/gromacs.git] / src / gromacs / mdtypes / interaction_const.cpp
index 230dca2bff54cdb586c950b73cdd783ee7e76f59..9353a24047aa105f840968e450cd0f975652f118 100644 (file)
@@ -1,7 +1,7 @@
 /*
  * This file is part of the GROMACS molecular simulation package.
  *
- * Copyright (c) 2020, by the GROMACS development team, led by
+ * Copyright (c) 2020,2021, by the GROMACS development team, led by
  * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
  * and including many others, as listed in the AUTHORS file in the
  * top-level source directory and at http://www.gromacs.org.
 
 #include <cstdio>
 
+#include "gromacs/ewald/ewald_utils.h"
 #include "gromacs/math/functions.h"
+#include "gromacs/math/units.h"
+#include "gromacs/mdlib/rf_util.h"
+#include "gromacs/mdtypes/forcerec.h"
 #include "gromacs/mdtypes/inputrec.h"
+#include "gromacs/topology/topology.h"
+#include "gromacs/utility/fatalerror.h"
+#include "gromacs/utility/pleasecite.h"
 
 interaction_const_t::SoftCoreParameters::SoftCoreParameters(const t_lambda& fepvals) :
     alphaVdw(fepvals.sc_alpha),
@@ -51,3 +58,320 @@ interaction_const_t::SoftCoreParameters::SoftCoreParameters(const t_lambda& fepv
     // This is checked during tpr reading, so we can assert here
     GMX_RELEASE_ASSERT(fepvals.sc_r_power == 6.0, "We only support soft-core r-power 6");
 }
+
+/*! \brief Print Coulomb Ewald citations and set ewald coefficients */
+static void initCoulombEwaldParameters(FILE*                fp,
+                                       const t_inputrec&    ir,
+                                       bool                 systemHasNetCharge,
+                                       interaction_const_t* ic)
+{
+    if (!EEL_PME_EWALD(ir.coulombtype))
+    {
+        return;
+    }
+
+    if (fp)
+    {
+        fprintf(fp, "Will do PME sum in reciprocal space for electrostatic interactions.\n");
+
+        if (ir.coulombtype == CoulombInteractionType::P3mAD)
+        {
+            please_cite(fp, "Hockney1988");
+            please_cite(fp, "Ballenegger2012");
+        }
+        else
+        {
+            please_cite(fp, "Essmann95a");
+        }
+
+        if (ir.ewald_geometry == EwaldGeometry::ThreeDC)
+        {
+            if (fp)
+            {
+                fprintf(fp,
+                        "Using the Ewald3DC correction for systems with a slab geometry%s.\n",
+                        systemHasNetCharge ? " and net charge" : "");
+            }
+            please_cite(fp, "In-Chul99a");
+            if (systemHasNetCharge)
+            {
+                please_cite(fp, "Ballenegger2009");
+            }
+        }
+    }
+
+    ic->ewaldcoeff_q = calc_ewaldcoeff_q(ir.rcoulomb, ir.ewald_rtol);
+    if (fp)
+    {
+        fprintf(fp, "Using a Gaussian width (1/beta) of %g nm for Ewald\n", 1 / ic->ewaldcoeff_q);
+    }
+
+    if (ic->coulomb_modifier == InteractionModifiers::PotShift)
+    {
+        GMX_RELEASE_ASSERT(ic->rcoulomb != 0, "Cutoff radius cannot be zero");
+        ic->sh_ewald = std::erfc(ic->ewaldcoeff_q * ic->rcoulomb) / ic->rcoulomb;
+    }
+    else
+    {
+        ic->sh_ewald = 0;
+    }
+}
+
+/*! \brief Print Van der Waals Ewald citations and set ewald coefficients */
+static void initVdwEwaldParameters(FILE* fp, const t_inputrec& ir, interaction_const_t* ic)
+{
+    if (!EVDW_PME(ir.vdwtype))
+    {
+        return;
+    }
+
+    if (fp)
+    {
+        fprintf(fp, "Will do PME sum in reciprocal space for LJ dispersion interactions.\n");
+        please_cite(fp, "Essmann95a");
+    }
+    ic->ewaldcoeff_lj = calc_ewaldcoeff_lj(ir.rvdw, ir.ewald_rtol_lj);
+    if (fp)
+    {
+        fprintf(fp, "Using a Gaussian width (1/beta) of %g nm for LJ Ewald\n", 1 / ic->ewaldcoeff_lj);
+    }
+
+    if (ic->vdw_modifier == InteractionModifiers::PotShift)
+    {
+        real crc2       = gmx::square(ic->ewaldcoeff_lj * ic->rvdw);
+        ic->sh_lj_ewald = (std::exp(-crc2) * (1 + crc2 + 0.5 * crc2 * crc2) - 1) / gmx::power6(ic->rvdw);
+    }
+    else
+    {
+        ic->sh_lj_ewald = 0;
+    }
+}
+
+static real calcBuckinghamBMax(FILE* fplog, const gmx_mtop_t& mtop)
+{
+    const t_atoms *at1, *at2;
+    int            i, j, tpi, tpj, ntypes;
+    real           b, bmin;
+
+    if (fplog)
+    {
+        fprintf(fplog, "Determining largest Buckingham b parameter for table\n");
+    }
+    ntypes = mtop.ffparams.atnr;
+
+    bmin            = -1;
+    real bham_b_max = 0;
+    for (size_t mt1 = 0; mt1 < mtop.moltype.size(); mt1++)
+    {
+        at1 = &mtop.moltype[mt1].atoms;
+        for (i = 0; (i < at1->nr); i++)
+        {
+            tpi = at1->atom[i].type;
+            if (tpi >= ntypes)
+            {
+                gmx_fatal(FARGS, "Atomtype[%d] = %d, maximum = %d", i, tpi, ntypes);
+            }
+
+            for (size_t mt2 = mt1; mt2 < mtop.moltype.size(); mt2++)
+            {
+                at2 = &mtop.moltype[mt2].atoms;
+                for (j = 0; (j < at2->nr); j++)
+                {
+                    tpj = at2->atom[j].type;
+                    if (tpj >= ntypes)
+                    {
+                        gmx_fatal(FARGS, "Atomtype[%d] = %d, maximum = %d", j, tpj, ntypes);
+                    }
+                    b = mtop.ffparams.iparams[tpi * ntypes + tpj].bham.b;
+                    if (b > bham_b_max)
+                    {
+                        bham_b_max = b;
+                    }
+                    if ((b < bmin) || (bmin == -1))
+                    {
+                        bmin = b;
+                    }
+                }
+            }
+        }
+    }
+    if (fplog)
+    {
+        fprintf(fplog, "Buckingham b parameters, min: %g, max: %g\n", bmin, bham_b_max);
+    }
+
+    return bham_b_max;
+}
+
+static void clear_force_switch_constants(shift_consts_t* sc)
+{
+    sc->c2   = 0;
+    sc->c3   = 0;
+    sc->cpot = 0;
+}
+
+static void force_switch_constants(real p, real rsw, real rc, shift_consts_t* sc)
+{
+    /* Here we determine the coefficient for shifting the force to zero
+     * between distance rsw and the cut-off rc.
+     * For a potential of r^-p, we have force p*r^-(p+1).
+     * But to save flops we absorb p in the coefficient.
+     * Thus we get:
+     * force/p   = r^-(p+1) + c2*r^2 + c3*r^3
+     * potential = r^-p + c2/3*r^3 + c3/4*r^4 + cpot
+     */
+    sc->c2   = ((p + 1) * rsw - (p + 4) * rc) / (pow(rc, p + 2) * gmx::square(rc - rsw));
+    sc->c3   = -((p + 1) * rsw - (p + 3) * rc) / (pow(rc, p + 2) * gmx::power3(rc - rsw));
+    sc->cpot = -pow(rc, -p) + p * sc->c2 / 3 * gmx::power3(rc - rsw)
+               + p * sc->c3 / 4 * gmx::power4(rc - rsw);
+}
+
+static void potential_switch_constants(real rsw, real rc, switch_consts_t* sc)
+{
+    /* The switch function is 1 at rsw and 0 at rc.
+     * The derivative and second derivate are zero at both ends.
+     * rsw        = max(r - r_switch, 0)
+     * sw         = 1 + c3*rsw^3 + c4*rsw^4 + c5*rsw^5
+     * dsw        = 3*c3*rsw^2 + 4*c4*rsw^3 + 5*c5*rsw^4
+     * force      = force*dsw - potential*sw
+     * potential *= sw
+     */
+    sc->c3 = -10 / gmx::power3(rc - rsw);
+    sc->c4 = 15 / gmx::power4(rc - rsw);
+    sc->c5 = -6 / gmx::power5(rc - rsw);
+}
+
+
+interaction_const_t init_interaction_const(FILE* fp, const t_inputrec& ir, const gmx_mtop_t& mtop, bool systemHasNetCharge)
+{
+    interaction_const_t interactionConst;
+
+    interactionConst.coulombEwaldTables = std::make_unique<EwaldCorrectionTables>();
+    interactionConst.vdwEwaldTables     = std::make_unique<EwaldCorrectionTables>();
+
+    /* Lennard-Jones */
+    interactionConst.vdwtype         = ir.vdwtype;
+    interactionConst.vdw_modifier    = ir.vdw_modifier;
+    interactionConst.reppow          = mtop.ffparams.reppow;
+    interactionConst.rvdw            = cutoff_inf(ir.rvdw);
+    interactionConst.rvdw_switch     = ir.rvdw_switch;
+    interactionConst.ljpme_comb_rule = ir.ljpme_combination_rule;
+    interactionConst.useBuckingham   = (mtop.ffparams.functype[0] == F_BHAM);
+    if (interactionConst.useBuckingham)
+    {
+        interactionConst.buckinghamBMax = calcBuckinghamBMax(fp, mtop);
+    }
+
+    initVdwEwaldParameters(fp, ir, &interactionConst);
+
+    clear_force_switch_constants(&interactionConst.dispersion_shift);
+    clear_force_switch_constants(&interactionConst.repulsion_shift);
+
+    switch (interactionConst.vdw_modifier)
+    {
+        case InteractionModifiers::PotShift:
+            /* Only shift the potential, don't touch the force */
+            interactionConst.dispersion_shift.cpot = -1.0 / gmx::power6(interactionConst.rvdw);
+            interactionConst.repulsion_shift.cpot  = -1.0 / gmx::power12(interactionConst.rvdw);
+            break;
+        case InteractionModifiers::ForceSwitch:
+            /* Switch the force, switch and shift the potential */
+            force_switch_constants(
+                    6.0, interactionConst.rvdw_switch, interactionConst.rvdw, &interactionConst.dispersion_shift);
+            force_switch_constants(
+                    12.0, interactionConst.rvdw_switch, interactionConst.rvdw, &interactionConst.repulsion_shift);
+            break;
+        case InteractionModifiers::PotSwitch:
+            /* Switch the potential and force */
+            potential_switch_constants(
+                    interactionConst.rvdw_switch, interactionConst.rvdw, &interactionConst.vdw_switch);
+            break;
+        case InteractionModifiers::None:
+        case InteractionModifiers::ExactCutoff:
+            /* Nothing to do here */
+            break;
+        default: gmx_incons("unimplemented potential modifier");
+    }
+
+    /* Electrostatics */
+    interactionConst.eeltype          = ir.coulombtype;
+    interactionConst.coulomb_modifier = ir.coulomb_modifier;
+    interactionConst.rcoulomb         = cutoff_inf(ir.rcoulomb);
+    interactionConst.rcoulomb_switch  = ir.rcoulomb_switch;
+    interactionConst.epsilon_r        = ir.epsilon_r;
+
+    /* Set the Coulomb energy conversion factor */
+    if (interactionConst.epsilon_r != 0)
+    {
+        interactionConst.epsfac = gmx::c_one4PiEps0 / interactionConst.epsilon_r;
+    }
+    else
+    {
+        /* eps = 0 is infinite dieletric: no Coulomb interactions */
+        interactionConst.epsfac = 0;
+    }
+
+    /* Reaction-field */
+    if (EEL_RF(interactionConst.eeltype))
+    {
+        GMX_RELEASE_ASSERT(interactionConst.eeltype != CoulombInteractionType::GRFNotused,
+                           "GRF is no longer supported");
+        interactionConst.reactionFieldPermitivity = ir.epsilon_rf;
+        calc_rffac(fp,
+                   interactionConst.epsilon_r,
+                   interactionConst.reactionFieldPermitivity,
+                   interactionConst.rcoulomb,
+                   &interactionConst.reactionFieldCoefficient,
+                   &interactionConst.reactionFieldShift);
+    }
+    else
+    {
+        /* For plain cut-off we might use the reaction-field kernels */
+        interactionConst.reactionFieldPermitivity = interactionConst.epsilon_r;
+        interactionConst.reactionFieldCoefficient = 0;
+        if (ir.coulomb_modifier == InteractionModifiers::PotShift)
+        {
+            interactionConst.reactionFieldShift = 1 / interactionConst.rcoulomb;
+        }
+        else
+        {
+            interactionConst.reactionFieldShift = 0;
+        }
+    }
+
+    initCoulombEwaldParameters(fp, ir, systemHasNetCharge, &interactionConst);
+
+    if (fp != nullptr)
+    {
+        real dispersion_shift;
+
+        dispersion_shift = interactionConst.dispersion_shift.cpot;
+        if (EVDW_PME(interactionConst.vdwtype))
+        {
+            dispersion_shift -= interactionConst.sh_lj_ewald;
+        }
+        fprintf(fp,
+                "Potential shift: LJ r^-12: %.3e r^-6: %.3e",
+                interactionConst.repulsion_shift.cpot,
+                dispersion_shift);
+
+        if (interactionConst.eeltype == CoulombInteractionType::Cut)
+        {
+            fprintf(fp, ", Coulomb %.e", -interactionConst.reactionFieldShift);
+        }
+        else if (EEL_PME(interactionConst.eeltype))
+        {
+            fprintf(fp, ", Ewald %.3e", -interactionConst.sh_ewald);
+        }
+        fprintf(fp, "\n");
+    }
+
+    if (ir.efep != FreeEnergyPerturbationType::No)
+    {
+        GMX_RELEASE_ASSERT(ir.fepvals, "ir.fepvals should be set wth free-energy");
+        interactionConst.softCoreParameters =
+                std::make_unique<interaction_const_t::SoftCoreParameters>(*ir.fepvals);
+    }
+
+    return interactionConst;
+}