SIMD acceleration for LINCS
[alexxy/gromacs.git] / src / gromacs / listed-forces / bonded.h
1 /*
2  * This file is part of the GROMACS molecular simulation package.
3  *
4  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
5  * Copyright (c) 2001-2004, The GROMACS development team.
6  * Copyright (c) 2013,2014,2015, by the GROMACS development team, led by
7  * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
8  * and including many others, as listed in the AUTHORS file in the
9  * top-level source directory and at http://www.gromacs.org.
10  *
11  * GROMACS is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU Lesser General Public License
13  * as published by the Free Software Foundation; either version 2.1
14  * of the License, or (at your option) any later version.
15  *
16  * GROMACS is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19  * Lesser General Public License for more details.
20  *
21  * You should have received a copy of the GNU Lesser General Public
22  * License along with GROMACS; if not, see
23  * http://www.gnu.org/licenses, or write to the Free Software Foundation,
24  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
25  *
26  * If you want to redistribute modifications to GROMACS, please
27  * consider that scientific software is very special. Version
28  * control is crucial - bugs must be traceable. We will be happy to
29  * consider code for inclusion in the official distribution, but
30  * derived work must not be called official GROMACS. Details are found
31  * in the README & COPYING files - if they are missing, get the
32  * official version at http://www.gromacs.org.
33  *
34  * To help us fund GROMACS development, we humbly ask that you cite
35  * the research papers on the package. Check out http://www.gromacs.org.
36  */
37 /*! \libinternal \file
38  *
39  * \brief This file contains declarations necessary for low-level
40  * functions for computing energies and forces for bonded interactions.
41  *
42  * \author Mark Abraham <mark.j.abraham@gmail.com>
43  * \inlibraryapi
44  * \ingroup module_listed-forces
45  */
46
47 #ifndef GMX_LISTED_FORCES_BONDED_H
48 #define GMX_LISTED_FORCES_BONDED_H
49
50 #include <stdio.h>
51
52 #include "gromacs/legacyheaders/types/fcdata.h"
53 #include "gromacs/legacyheaders/types/forcerec.h"
54 #include "gromacs/legacyheaders/types/ifunc.h"
55 #include "gromacs/legacyheaders/types/interaction_const.h"
56 #include "gromacs/legacyheaders/types/mdatom.h"
57 #include "gromacs/legacyheaders/types/nrnb.h"
58 #include "gromacs/math/vec.h"
59 #include "gromacs/pbcutil/pbc-simd.h"
60 #include "gromacs/simd/simd.h"
61 #include "gromacs/topology/idef.h"
62 #include "gromacs/utility/basedefinitions.h"
63
64 #ifdef __cplusplus
65 extern "C" {
66 #endif
67
68 struct gmx_wallcycle;
69 struct t_graph;
70 struct t_pbc;
71
72 /*! \brief Calculate bond-angle. No PBC is taken into account (use mol-shift) */
73 real bond_angle(const rvec xi, const rvec xj, const rvec xk,
74                 const struct t_pbc *pbc,
75                 rvec r_ij, rvec r_kj, real *costh,
76                 int *t1, int *t2);  /* out */
77
78 /*! \brief Calculate dihedral-angle. No PBC is taken into account (use mol-shift) */
79 real dih_angle(const rvec xi, const rvec xj, const rvec xk, const rvec xl,
80                const struct t_pbc *pbc,
81                rvec r_ij, rvec r_kj, rvec r_kl, rvec m, rvec n, /* out */
82                real *sign,
83                int *t1, int *t2, int *t3);
84
85 /*! \brief Do an update of the forces for dihedral potentials */
86 void do_dih_fup(int i, int j, int k, int l, real ddphi,
87                 rvec r_ij, rvec r_kj, rvec r_kl,
88                 rvec m, rvec n, rvec f[], rvec fshift[],
89                 const struct t_pbc *pbc, const struct t_graph *g,
90                 const rvec *x, int t1, int t2, int t3);
91
92 /*! \brief Make a dihedral fall in the range (-pi,pi) */
93 void make_dp_periodic(real *dp);
94
95 /*! \brief Compute CMAP dihedral energies and forces */
96 real
97     cmap_dihs(int nbonds,
98               const t_iatom forceatoms[], const t_iparams forceparams[],
99               const gmx_cmap_t *cmap_grid,
100               const rvec x[], rvec f[], rvec fshift[],
101               const struct t_pbc *pbc, const struct t_graph *g,
102               real gmx_unused lambda, real gmx_unused *dvdlambda,
103               const t_mdatoms gmx_unused *md, t_fcdata gmx_unused *fcd,
104               int  gmx_unused *global_atom_index);
105
106 //! \cond
107 /*************************************************************************
108  *
109  *  Bonded force functions
110  *
111  *************************************************************************/
112 t_ifunc bonds, g96bonds, morse_bonds, cubic_bonds, FENE_bonds, restraint_bonds;
113 t_ifunc angles, g96angles, cross_bond_bond, cross_bond_angle, urey_bradley, quartic_angles, linear_angles;
114 t_ifunc restrangles;
115 t_ifunc pdihs, idihs, rbdihs;
116 t_ifunc restrdihs, cbtdihs;
117 t_ifunc tab_bonds, tab_angles, tab_dihs;
118 t_ifunc polarize, anharm_polarize, water_pol, thole_pol, angres, angresz, dihres, unimplemented;
119
120 /* As pdihs(), but without calculating energies and shift forces */
121 void
122     pdihs_noener(int nbonds,
123                  const t_iatom forceatoms[], const t_iparams forceparams[],
124                  const rvec x[], rvec f[],
125                  const struct t_pbc gmx_unused *pbc,
126                  const struct t_graph gmx_unused *g,
127                  real lambda,
128                  const t_mdatoms gmx_unused *md, t_fcdata gmx_unused *fcd,
129                  int gmx_unused *global_atom_index);
130
131 #ifdef GMX_SIMD_HAVE_REAL
132
133 /* As angles(), but using SIMD to calculate many angles at once.
134  * This routines does not calculate energies and shift forces.
135  */
136 void
137     angles_noener_simd(int nbonds,
138                        const t_iatom forceatoms[], const t_iparams forceparams[],
139                        const rvec x[], rvec f[],
140                        const struct t_pbc *pbc,
141                        const struct t_graph gmx_unused *g,
142                        real gmx_unused lambda,
143                        const t_mdatoms gmx_unused *md, t_fcdata gmx_unused *fcd,
144                        int gmx_unused *global_atom_index);
145
146 /* As pdihs_noener(), but using SIMD to calculate many dihedrals at once. */
147 void
148     pdihs_noener_simd(int nbonds,
149                       const t_iatom forceatoms[], const t_iparams forceparams[],
150                       const rvec x[], rvec f[],
151                       const struct t_pbc *pbc,
152                       const struct t_graph gmx_unused *g,
153                       real gmx_unused lambda,
154                       const t_mdatoms gmx_unused *md, t_fcdata gmx_unused *fcd,
155                       int gmx_unused *global_atom_index);
156
157 /* As rbdihs(), when not needing energy or shift force, using SIMD to calculate many dihedrals at once. */
158 void
159     rbdihs_noener_simd(int nbonds,
160                        const t_iatom forceatoms[], const t_iparams forceparams[],
161                        const rvec x[], rvec f[],
162                        const struct t_pbc *pbc,
163                        const struct t_graph gmx_unused *g,
164                        real gmx_unused lambda,
165                        const t_mdatoms gmx_unused *md, t_fcdata gmx_unused *fcd,
166                        int gmx_unused *global_atom_index);
167
168 #endif
169
170 //! \endcond
171
172 #ifdef __cplusplus
173 }
174 #endif
175
176 #endif