Merge branch 'release-2016'
[alexxy/gromacs.git] / src / gromacs / topology / idef.cpp
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,2016, 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 #include "gmxpre.h"
38
39 #include "idef.h"
40
41 #include <cstdio>
42
43 #include "gromacs/topology/ifunc.h"
44 #include "gromacs/utility/fatalerror.h"
45 #include "gromacs/utility/txtdump.h"
46
47 static void pr_harm(FILE *fp, const t_iparams *iparams, const char *r, const char *kr)
48 {
49     fprintf(fp, "%sA=%12.5e, %sA=%12.5e, %sB=%12.5e, %sB=%12.5e\n",
50             r, iparams->harmonic.rA, kr, iparams->harmonic.krA,
51             r, iparams->harmonic.rB, kr, iparams->harmonic.krB);
52 }
53
54 void pr_iparams(FILE *fp, t_functype ftype, const t_iparams *iparams)
55 {
56     switch (ftype)
57     {
58         case F_ANGLES:
59         case F_G96ANGLES:
60             pr_harm(fp, iparams, "th", "ct");
61             break;
62         case F_CROSS_BOND_BONDS:
63             fprintf(fp, "r1e=%15.8e, r2e=%15.8e, krr=%15.8e\n",
64                     iparams->cross_bb.r1e, iparams->cross_bb.r2e,
65                     iparams->cross_bb.krr);
66             break;
67         case F_CROSS_BOND_ANGLES:
68             fprintf(fp, "r1e=%15.8e, r1e=%15.8e, r3e=%15.8e, krt=%15.8e\n",
69                     iparams->cross_ba.r1e, iparams->cross_ba.r2e,
70                     iparams->cross_ba.r3e, iparams->cross_ba.krt);
71             break;
72         case F_LINEAR_ANGLES:
73             fprintf(fp, "klinA=%15.8e, aA=%15.8e, klinB=%15.8e, aB=%15.8e\n",
74                     iparams->linangle.klinA, iparams->linangle.aA,
75                     iparams->linangle.klinB, iparams->linangle.aB);
76             break;
77         case F_UREY_BRADLEY:
78             fprintf(fp, "thetaA=%15.8e, kthetaA=%15.8e, r13A=%15.8e, kUBA=%15.8e, thetaB=%15.8e, kthetaB=%15.8e, r13B=%15.8e, kUBB=%15.8e\n", iparams->u_b.thetaA, iparams->u_b.kthetaA, iparams->u_b.r13A, iparams->u_b.kUBA, iparams->u_b.thetaB, iparams->u_b.kthetaB, iparams->u_b.r13B, iparams->u_b.kUBB);
79             break;
80         case F_QUARTIC_ANGLES:
81             fprintf(fp, "theta=%15.8e", iparams->qangle.theta);
82             for (int i = 0; i < 5; i++)
83             {
84                 fprintf(fp, ", c%c=%15.8e", '0'+i, iparams->qangle.c[i]);
85             }
86             fprintf(fp, "\n");
87             break;
88         case F_BHAM:
89             fprintf(fp, "a=%15.8e, b=%15.8e, c=%15.8e\n",
90                     iparams->bham.a, iparams->bham.b, iparams->bham.c);
91             break;
92         case F_BONDS:
93         case F_G96BONDS:
94         case F_HARMONIC:
95             pr_harm(fp, iparams, "b0", "cb");
96             break;
97         case F_IDIHS:
98             pr_harm(fp, iparams, "xi", "cx");
99             break;
100         case F_MORSE:
101             fprintf(fp, "b0A=%15.8e, cbA=%15.8e, betaA=%15.8e, b0B=%15.8e, cbB=%15.8e, betaB=%15.8e\n",
102                     iparams->morse.b0A, iparams->morse.cbA, iparams->morse.betaA,
103                     iparams->morse.b0B, iparams->morse.cbB, iparams->morse.betaB);
104             break;
105         case F_CUBICBONDS:
106             fprintf(fp, "b0=%15.8e, kb=%15.8e, kcub=%15.8e\n",
107                     iparams->cubic.b0, iparams->cubic.kb, iparams->cubic.kcub);
108             break;
109         case F_CONNBONDS:
110             fprintf(fp, "\n");
111             break;
112         case F_FENEBONDS:
113             fprintf(fp, "bm=%15.8e, kb=%15.8e\n", iparams->fene.bm, iparams->fene.kb);
114             break;
115         case F_RESTRBONDS:
116             fprintf(fp, "lowA=%15.8e, up1A=%15.8e, up2A=%15.8e, kA=%15.8e, lowB=%15.8e, up1B=%15.8e, up2B=%15.8e, kB=%15.8e,\n",
117                     iparams->restraint.lowA, iparams->restraint.up1A,
118                     iparams->restraint.up2A, iparams->restraint.kA,
119                     iparams->restraint.lowB, iparams->restraint.up1B,
120                     iparams->restraint.up2B, iparams->restraint.kB);
121             break;
122         case F_TABBONDS:
123         case F_TABBONDSNC:
124         case F_TABANGLES:
125         case F_TABDIHS:
126             fprintf(fp, "tab=%d, kA=%15.8e, kB=%15.8e\n",
127                     iparams->tab.table, iparams->tab.kA, iparams->tab.kB);
128             break;
129         case F_POLARIZATION:
130             fprintf(fp, "alpha=%15.8e\n", iparams->polarize.alpha);
131             break;
132         case F_ANHARM_POL:
133             fprintf(fp, "alpha=%15.8e drcut=%15.8e khyp=%15.8e\n",
134                     iparams->anharm_polarize.alpha,
135                     iparams->anharm_polarize.drcut,
136                     iparams->anharm_polarize.khyp);
137             break;
138         case F_THOLE_POL:
139             fprintf(fp, "a=%15.8e, alpha1=%15.8e, alpha2=%15.8e, rfac=%15.8e\n",
140                     iparams->thole.a, iparams->thole.alpha1, iparams->thole.alpha2,
141                     iparams->thole.rfac);
142             break;
143         case F_WATER_POL:
144             fprintf(fp, "al_x=%15.8e, al_y=%15.8e, al_z=%15.8e, rOH=%9.6f, rHH=%9.6f, rOD=%9.6f\n",
145                     iparams->wpol.al_x, iparams->wpol.al_y, iparams->wpol.al_z,
146                     iparams->wpol.rOH, iparams->wpol.rHH, iparams->wpol.rOD);
147             break;
148         case F_LJ:
149             fprintf(fp, "c6=%15.8e, c12=%15.8e\n", iparams->lj.c6, iparams->lj.c12);
150             break;
151         case F_LJ14:
152             fprintf(fp, "c6A=%15.8e, c12A=%15.8e, c6B=%15.8e, c12B=%15.8e\n",
153                     iparams->lj14.c6A, iparams->lj14.c12A,
154                     iparams->lj14.c6B, iparams->lj14.c12B);
155             break;
156         case F_LJC14_Q:
157             fprintf(fp, "fqq=%15.8e, qi=%15.8e, qj=%15.8e, c6=%15.8e, c12=%15.8e\n",
158                     iparams->ljc14.fqq,
159                     iparams->ljc14.qi, iparams->ljc14.qj,
160                     iparams->ljc14.c6, iparams->ljc14.c12);
161             break;
162         case F_LJC_PAIRS_NB:
163             fprintf(fp, "qi=%15.8e, qj=%15.8e, c6=%15.8e, c12=%15.8e\n",
164                     iparams->ljcnb.qi, iparams->ljcnb.qj,
165                     iparams->ljcnb.c6, iparams->ljcnb.c12);
166             break;
167         case F_PDIHS:
168         case F_PIDIHS:
169         case F_ANGRES:
170         case F_ANGRESZ:
171             fprintf(fp, "phiA=%15.8e, cpA=%15.8e, phiB=%15.8e, cpB=%15.8e, mult=%d\n",
172                     iparams->pdihs.phiA, iparams->pdihs.cpA,
173                     iparams->pdihs.phiB, iparams->pdihs.cpB,
174                     iparams->pdihs.mult);
175             break;
176         case F_DISRES:
177             fprintf(fp, "label=%4d, type=%1d, low=%15.8e, up1=%15.8e, up2=%15.8e, fac=%15.8e)\n",
178                     iparams->disres.label, iparams->disres.type,
179                     iparams->disres.low, iparams->disres.up1,
180                     iparams->disres.up2, iparams->disres.kfac);
181             break;
182         case F_ORIRES:
183             fprintf(fp, "ex=%4d, label=%d, power=%4d, c=%15.8e, obs=%15.8e, kfac=%15.8e)\n",
184                     iparams->orires.ex, iparams->orires.label, iparams->orires.power,
185                     iparams->orires.c, iparams->orires.obs, iparams->orires.kfac);
186             break;
187         case F_DIHRES:
188             fprintf(fp, "phiA=%15.8e, dphiA=%15.8e, kfacA=%15.8e, phiB=%15.8e, dphiB=%15.8e, kfacB=%15.8e\n",
189                     iparams->dihres.phiA, iparams->dihres.dphiA, iparams->dihres.kfacA,
190                     iparams->dihres.phiB, iparams->dihres.dphiB, iparams->dihres.kfacB);
191             break;
192         case F_POSRES:
193             fprintf(fp, "pos0A=(%15.8e,%15.8e,%15.8e), fcA=(%15.8e,%15.8e,%15.8e), pos0B=(%15.8e,%15.8e,%15.8e), fcB=(%15.8e,%15.8e,%15.8e)\n",
194                     iparams->posres.pos0A[XX], iparams->posres.pos0A[YY],
195                     iparams->posres.pos0A[ZZ], iparams->posres.fcA[XX],
196                     iparams->posres.fcA[YY], iparams->posres.fcA[ZZ],
197                     iparams->posres.pos0B[XX], iparams->posres.pos0B[YY],
198                     iparams->posres.pos0B[ZZ], iparams->posres.fcB[XX],
199                     iparams->posres.fcB[YY], iparams->posres.fcB[ZZ]);
200             break;
201         case F_FBPOSRES:
202             fprintf(fp, "pos0=(%15.8e,%15.8e,%15.8e), geometry=%d, r=%15.8e, k=%15.8e\n",
203                     iparams->fbposres.pos0[XX], iparams->fbposres.pos0[YY],
204                     iparams->fbposres.pos0[ZZ], iparams->fbposres.geom,
205                     iparams->fbposres.r,        iparams->fbposres.k);
206             break;
207         case F_RBDIHS:
208             for (int i = 0; i < NR_RBDIHS; i++)
209             {
210                 fprintf(fp, "%srbcA[%d]=%15.8e", i == 0 ? "" : ", ", i, iparams->rbdihs.rbcA[i]);
211             }
212             fprintf(fp, "\n");
213             for (int i = 0; i < NR_RBDIHS; i++)
214             {
215                 fprintf(fp, "%srbcB[%d]=%15.8e", i == 0 ? "" : ", ", i, iparams->rbdihs.rbcB[i]);
216             }
217             fprintf(fp, "\n");
218             break;
219         case F_FOURDIHS:
220         {
221             /* Use the OPLS -> Ryckaert-Bellemans formula backwards to get
222              * the OPLS potential constants back.
223              */
224             const real *rbcA = iparams->rbdihs.rbcA;
225             const real *rbcB = iparams->rbdihs.rbcB;
226             real        VA[4], VB[4];
227
228             VA[3] = -0.25*rbcA[4];
229             VA[2] = -0.5*rbcA[3];
230             VA[1] = 4.0*VA[3]-rbcA[2];
231             VA[0] = 3.0*VA[2]-2.0*rbcA[1];
232
233             VB[3] = -0.25*rbcB[4];
234             VB[2] = -0.5*rbcB[3];
235             VB[1] = 4.0*VB[3]-rbcB[2];
236             VB[0] = 3.0*VB[2]-2.0*rbcB[1];
237
238             for (int i = 0; i < NR_FOURDIHS; i++)
239             {
240                 fprintf(fp, "%sFourA[%d]=%15.8e", i == 0 ? "" : ", ", i, VA[i]);
241             }
242             fprintf(fp, "\n");
243             for (int i = 0; i < NR_FOURDIHS; i++)
244             {
245                 fprintf(fp, "%sFourB[%d]=%15.8e", i == 0 ? "" : ", ", i, VB[i]);
246             }
247             fprintf(fp, "\n");
248             break;
249         }
250
251         case F_CONSTR:
252         case F_CONSTRNC:
253             fprintf(fp, "dA=%15.8e, dB=%15.8e\n", iparams->constr.dA, iparams->constr.dB);
254             break;
255         case F_SETTLE:
256             fprintf(fp, "doh=%15.8e, dhh=%15.8e\n", iparams->settle.doh,
257                     iparams->settle.dhh);
258             break;
259         case F_VSITE2:
260             fprintf(fp, "a=%15.8e\n", iparams->vsite.a);
261             break;
262         case F_VSITE3:
263         case F_VSITE3FD:
264         case F_VSITE3FAD:
265             fprintf(fp, "a=%15.8e, b=%15.8e\n", iparams->vsite.a, iparams->vsite.b);
266             break;
267         case F_VSITE3OUT:
268         case F_VSITE4FD:
269         case F_VSITE4FDN:
270             fprintf(fp, "a=%15.8e, b=%15.8e, c=%15.8e\n",
271                     iparams->vsite.a, iparams->vsite.b, iparams->vsite.c);
272             break;
273         case F_VSITEN:
274             fprintf(fp, "n=%2d, a=%15.8e\n", iparams->vsiten.n, iparams->vsiten.a);
275             break;
276         case F_GB12:
277         case F_GB13:
278         case F_GB14:
279             fprintf(fp, "sar=%15.8e, st=%15.8e, pi=%15.8e, gbr=%15.8e, bmlt=%15.8e\n", iparams->gb.sar, iparams->gb.st, iparams->gb.pi, iparams->gb.gbr, iparams->gb.bmlt);
280             break;
281         case F_CMAP:
282             fprintf(fp, "cmapA=%1d, cmapB=%1d\n", iparams->cmap.cmapA, iparams->cmap.cmapB);
283             break;
284         case  F_RESTRANGLES:
285             pr_harm(fp, iparams, "ktheta", "costheta0");
286             break;
287         case  F_RESTRDIHS:
288             fprintf(fp, "phiA=%15.8e, cpA=%15.8e",
289                     iparams->pdihs.phiA, iparams->pdihs.cpA);
290             break;
291         case  F_CBTDIHS:
292             fprintf(fp, "kphi=%15.8e", iparams->cbtdihs.cbtcA[0]);
293             for (int i = 1; i < NR_CBTDIHS; i++)
294             {
295                 fprintf(fp, ", cbtcA[%d]=%15.8e", i-1, iparams->cbtdihs.cbtcA[i]);
296             }
297             fprintf(fp, "\n");
298             break;
299         default:
300             gmx_fatal(FARGS, "unknown function type %d (%s) in %s line %d",
301                       ftype, interaction_function[ftype].name, __FILE__, __LINE__);
302     }
303 }
304
305 void pr_ilist(FILE *fp, int indent, const char *title,
306               const t_functype *functype, const t_ilist *ilist,
307               gmx_bool bShowNumbers,
308               gmx_bool bShowParameters, const t_iparams *iparams)
309 {
310     int      i, j, k, type, ftype;
311     t_iatom *iatoms;
312
313     if (available(fp, ilist, indent, title) && ilist->nr > 0)
314     {
315         indent = pr_title(fp, indent, title);
316         pr_indent(fp, indent);
317         fprintf(fp, "nr: %d\n", ilist->nr);
318         if (ilist->nr > 0)
319         {
320             pr_indent(fp, indent);
321             fprintf(fp, "iatoms:\n");
322             iatoms = ilist->iatoms;
323             for (i = j = 0; i < ilist->nr; )
324             {
325                 pr_indent(fp, indent+INDENT);
326                 type  = *(iatoms++);
327                 ftype = functype[type];
328                 if (bShowNumbers)
329                 {
330                     fprintf(fp, "%d type=%d ", j, type);
331                 }
332                 j++;
333                 printf("(%s)", interaction_function[ftype].name);
334                 for (k = 0; k < interaction_function[ftype].nratoms; k++)
335                 {
336                     fprintf(fp, " %3d", *(iatoms++));
337                 }
338                 if (bShowParameters)
339                 {
340                     fprintf(fp, "  ");
341                     pr_iparams(fp, ftype,  &iparams[type]);
342                 }
343                 fprintf(fp, "\n");
344                 i += 1+interaction_function[ftype].nratoms;
345             }
346         }
347     }
348 }
349
350 static void pr_cmap(FILE *fp, int indent, const char *title,
351                     const gmx_cmap_t *cmap_grid, gmx_bool bShowNumbers)
352 {
353     int  i, j, nelem;
354     real dx, idx;
355
356     dx    = 360.0 / cmap_grid->grid_spacing;
357     nelem = cmap_grid->grid_spacing*cmap_grid->grid_spacing;
358
359     if (available(fp, cmap_grid, indent, title))
360     {
361         fprintf(fp, "%s\n", title);
362
363         for (i = 0; i < cmap_grid->ngrid; i++)
364         {
365             idx = -180.0;
366             fprintf(fp, "%8s %8s %8s %8s\n", "V", "dVdx", "dVdy", "d2dV");
367
368             fprintf(fp, "grid[%3d]={\n", bShowNumbers ? i : -1);
369
370             for (j = 0; j < nelem; j++)
371             {
372                 if ( (j%cmap_grid->grid_spacing) == 0)
373                 {
374                     fprintf(fp, "%8.1f\n", idx);
375                     idx += dx;
376                 }
377
378                 fprintf(fp, "%8.3f ", cmap_grid->cmapdata[i].cmap[j*4]);
379                 fprintf(fp, "%8.3f ", cmap_grid->cmapdata[i].cmap[j*4+1]);
380                 fprintf(fp, "%8.3f ", cmap_grid->cmapdata[i].cmap[j*4+2]);
381                 fprintf(fp, "%8.3f\n", cmap_grid->cmapdata[i].cmap[j*4+3]);
382             }
383             fprintf(fp, "\n");
384         }
385     }
386
387 }
388
389 void pr_ffparams(FILE *fp, int indent, const char *title,
390                  const gmx_ffparams_t *ffparams,
391                  gmx_bool bShowNumbers)
392 {
393     int i;
394
395     indent = pr_title(fp, indent, title);
396     pr_indent(fp, indent);
397     fprintf(fp, "atnr=%d\n", ffparams->atnr);
398     pr_indent(fp, indent);
399     fprintf(fp, "ntypes=%d\n", ffparams->ntypes);
400     for (i = 0; i < ffparams->ntypes; i++)
401     {
402         pr_indent(fp, indent+INDENT);
403         fprintf(fp, "functype[%d]=%s, ",
404                 bShowNumbers ? i : -1,
405                 interaction_function[ffparams->functype[i]].name);
406         pr_iparams(fp, ffparams->functype[i], &ffparams->iparams[i]);
407     }
408     pr_double(fp, indent, "reppow", ffparams->reppow);
409     pr_real(fp, indent, "fudgeQQ", ffparams->fudgeQQ);
410     pr_cmap(fp, indent, "cmap", &ffparams->cmap_grid, bShowNumbers);
411 }
412
413 void pr_idef(FILE *fp, int indent, const char *title, const t_idef *idef,
414              gmx_bool bShowNumbers, gmx_bool bShowParameters)
415 {
416     int i, j;
417
418     if (available(fp, idef, indent, title))
419     {
420         indent = pr_title(fp, indent, title);
421         pr_indent(fp, indent);
422         fprintf(fp, "atnr=%d\n", idef->atnr);
423         pr_indent(fp, indent);
424         fprintf(fp, "ntypes=%d\n", idef->ntypes);
425         for (i = 0; i < idef->ntypes; i++)
426         {
427             pr_indent(fp, indent+INDENT);
428             fprintf(fp, "functype[%d]=%s, ",
429                     bShowNumbers ? i : -1,
430                     interaction_function[idef->functype[i]].name);
431             pr_iparams(fp, idef->functype[i], &idef->iparams[i]);
432         }
433         pr_real(fp, indent, "fudgeQQ", idef->fudgeQQ);
434
435         for (j = 0; (j < F_NRE); j++)
436         {
437             pr_ilist(fp, indent, interaction_function[j].longname,
438                      idef->functype, &idef->il[j], bShowNumbers,
439                      bShowParameters, idef->iparams);
440         }
441     }
442 }