b6af91893c313ad3ce5b6daba67e6d3df068092d
[alexxy/gromacs.git] / src / gromacs / gmxana / gmx_helix.c
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, 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 "config.h"
40
41 #include <math.h>
42 #include <string.h>
43
44 #include "gromacs/fileio/confio.h"
45 #include "fitahx.h"
46 #include "gromacs/utility/futil.h"
47 #include "gstat.h"
48 #include "hxprops.h"
49 #include "gromacs/legacyheaders/macros.h"
50 #include "gromacs/math/utilities.h"
51 #include "gromacs/fileio/tpxio.h"
52 #include "gromacs/fileio/trxio.h"
53 #include "gromacs/utility/smalloc.h"
54 #include "gromacs/legacyheaders/txtdump.h"
55 #include "gromacs/legacyheaders/typedefs.h"
56 #include "gromacs/math/vec.h"
57 #include "gromacs/legacyheaders/viewit.h"
58 #include "gmx_ana.h"
59
60 #include "gromacs/commandline/pargs.h"
61 #include "gromacs/fileio/xvgr.h"
62 #include "gromacs/pbcutil/rmpbc.h"
63 #include "gromacs/utility/fatalerror.h"
64
65 int gmx_helix(int argc, char *argv[])
66 {
67     const char        *desc[] = {
68         "[THISMODULE] computes all kinds of helix properties. First, the peptide",
69         "is checked to find the longest helical part, as determined by",
70         "hydrogen bonds and [GRK]phi[grk]/[GRK]psi[grk] angles.",
71         "That bit is fitted",
72         "to an ideal helix around the [IT]z[it]-axis and centered around the origin.",
73         "Then the following properties are computed:[PAR]",
74         "[BB]1.[bb] Helix radius (file [TT]radius.xvg[tt]). This is merely the",
75         "RMS deviation in two dimensions for all C[GRK]alpha[grk] atoms.",
76         "it is calculated as [SQRT]([SUM][sum][SUB]i[sub] (x^2(i)+y^2(i)))/N[sqrt] where N is the number",
77         "of backbone atoms. For an ideal helix the radius is 0.23 nm[BR]",
78         "[BB]2.[bb] Twist (file [TT]twist.xvg[tt]). The average helical angle per",
79         "residue is calculated. For an [GRK]alpha[grk]-helix it is 100 degrees,",
80         "for 3-10 helices it will be smaller, and ",
81         "for 5-helices it will be larger.[BR]",
82         "[BB]3.[bb] Rise per residue (file [TT]rise.xvg[tt]). The helical rise per",
83         "residue is plotted as the difference in [IT]z[it]-coordinate between C[GRK]alpha[grk]",
84         "atoms. For an ideal helix, this is 0.15 nm[BR]",
85         "[BB]4.[bb] Total helix length (file [TT]len-ahx.xvg[tt]). The total length",
86         "of the",
87         "helix in nm. This is simply the average rise (see above) times the",
88         "number of helical residues (see below).[BR]",
89         "[BB]5.[bb] Helix dipole, backbone only (file [TT]dip-ahx.xvg[tt]).[BR]",
90         "[BB]6.[bb] RMS deviation from ideal helix, calculated for the C[GRK]alpha[grk]",
91         "atoms only (file [TT]rms-ahx.xvg[tt]).[BR]",
92         "[BB]7.[bb] Average C[GRK]alpha[grk] - C[GRK]alpha[grk] dihedral angle (file [TT]phi-ahx.xvg[tt]).[BR]",
93         "[BB]8.[bb] Average [GRK]phi[grk] and [GRK]psi[grk] angles (file [TT]phipsi.xvg[tt]).[BR]",
94         "[BB]9.[bb] Ellipticity at 222 nm according to Hirst and Brooks.",
95         "[PAR]"
96     };
97     static gmx_bool    bCheck = FALSE, bFit = TRUE, bDBG = FALSE, bEV = FALSE;
98     static int         rStart = 0, rEnd = 0, r0 = 1;
99     t_pargs            pa []  = {
100         { "-r0", FALSE, etINT, {&r0},
101           "The first residue number in the sequence" },
102         { "-q",  FALSE, etBOOL, {&bCheck},
103           "Check at every step which part of the sequence is helical" },
104         { "-F",  FALSE, etBOOL, {&bFit},
105           "Toggle fit to a perfect helix" },
106         { "-db", FALSE, etBOOL, {&bDBG},
107           "Print debug info" },
108         { "-ev", FALSE, etBOOL, {&bEV},
109           "Write a new 'trajectory' file for ED" },
110         { "-ahxstart", FALSE, etINT, {&rStart},
111           "First residue in helix" },
112         { "-ahxend", FALSE, etINT, {&rEnd},
113           "Last residue in helix" }
114     };
115
116     typedef struct {
117         FILE       *fp, *fp2;
118         gmx_bool    bfp2;
119         const char *filenm;
120         const char *title;
121         const char *xaxis;
122         const char *yaxis;
123         real        val;
124     } t_xvgrfile;
125
126     t_xvgrfile     xf[efhNR] = {
127         { NULL, NULL, TRUE,  "radius",  "Helix radius",               NULL, "r (nm)", 0.0 },
128         { NULL, NULL, TRUE,  "twist",   "Twist per residue",          NULL, "Angle (deg)", 0.0 },
129         { NULL, NULL, TRUE,  "rise",    "Rise per residue",           NULL, "Rise (nm)", 0.0 },
130         { NULL, NULL, FALSE, "len-ahx", "Length of the Helix",        NULL, "Length (nm)", 0.0 },
131         { NULL, NULL, FALSE, "dip-ahx", "Helix Backbone Dipole",      NULL, "rq (nm e)", 0.0 },
132         { NULL, NULL, TRUE,  "rms-ahx", "RMS Deviation from Ideal Helix", NULL, "RMS (nm)", 0.0 },
133         { NULL, NULL, FALSE, "rmsa-ahx", "Average RMSD per Residue",   "Residue", "RMS (nm)", 0.0 },
134         { NULL, NULL, FALSE,  "cd222",   "Ellipticity at 222 nm", NULL, "nm", 0.0 },
135         { NULL, NULL, TRUE,  "pprms",   "RMS Distance from \\8a\\4-helix", NULL, "deg", 0.0 },
136         { NULL, NULL, TRUE,  "caphi",   "Average Ca-Ca Dihedral",     NULL, "\\8F\\4(deg)", 0.0 },
137         { NULL, NULL, TRUE,  "phi",     "Average \\8F\\4 angles", NULL, "deg", 0.0 },
138         { NULL, NULL, TRUE,  "psi",     "Average \\8Y\\4 angles", NULL, "deg", 0.0 },
139         { NULL, NULL, TRUE,  "hb3",     "Average n-n+3 hbond length", NULL, "nm", 0.0 },
140         { NULL, NULL, TRUE,  "hb4",     "Average n-n+4 hbond length", NULL, "nm", 0.0 },
141         { NULL, NULL, TRUE,  "hb5",     "Average n-n+5 hbond length", NULL, "nm", 0.0 },
142         { NULL, NULL, FALSE,  "JCaHa",   "J-Coupling Values",        "Residue", "Hz", 0.0 },
143         { NULL, NULL, FALSE,  "helicity", "Helicity per Residue",     "Residue", "% of time", 0.0 }
144     };
145
146     output_env_t   oenv;
147     char           buf[54];
148     t_trxstatus   *status;
149     int            natoms, nre, nres;
150     t_bb          *bb;
151     int            i, j, m, nall, nbb, nca, teller, nSel = 0;
152     atom_id       *bbindex, *caindex, *allindex;
153     t_topology    *top;
154     int            ePBC;
155     rvec          *x, *xref;
156     real           t;
157     real           rms;
158     matrix         box;
159     gmx_rmpbc_t    gpbc = NULL;
160     gmx_bool       bRange;
161     t_filenm       fnm[] = {
162         { efTPX, NULL,  NULL,   ffREAD  },
163         { efNDX, NULL,  NULL,   ffREAD  },
164         { efTRX, "-f",  NULL,   ffREAD  },
165         { efSTO, "-cz", "zconf", ffWRITE },
166     };
167 #define NFILE asize(fnm)
168
169     if (!parse_common_args(&argc, argv, PCA_CAN_VIEW | PCA_CAN_TIME,
170                            NFILE, fnm, asize(pa), pa, asize(desc), desc, 0, NULL, &oenv))
171     {
172         return 0;
173     }
174
175     bRange = (opt2parg_bSet("-ahxstart", asize(pa), pa) &&
176               opt2parg_bSet("-ahxend", asize(pa), pa));
177
178     top = read_top(ftp2fn(efTPX, NFILE, fnm), &ePBC);
179
180     natoms = read_first_x(oenv, &status, opt2fn("-f", NFILE, fnm), &t, &x, box);
181
182     if (natoms != top->atoms.nr)
183     {
184         gmx_fatal(FARGS, "Sorry can only run when the number of atoms in the run input file (%d) is equal to the number in the trajectory (%d)",
185                   top->atoms.nr, natoms);
186     }
187
188     bb = mkbbind(ftp2fn(efNDX, NFILE, fnm), &nres, &nbb, r0, &nall, &allindex,
189                  top->atoms.atomname, top->atoms.atom, top->atoms.resinfo);
190     snew(bbindex, natoms);
191     snew(caindex, nres);
192
193     fprintf(stderr, "nall=%d\n", nall);
194
195     /* Open output files, default x-axis is time */
196     for (i = 0; (i < efhNR); i++)
197     {
198         sprintf(buf, "%s.xvg", xf[i].filenm);
199         remove(buf);
200         xf[i].fp = xvgropen(buf, xf[i].title,
201                             xf[i].xaxis ? xf[i].xaxis : "Time (ps)",
202                             xf[i].yaxis, oenv);
203         if (xf[i].bfp2)
204         {
205             sprintf(buf, "%s.out", xf[i].filenm);
206             remove(buf);
207             xf[i].fp2 = gmx_ffopen(buf, "w");
208         }
209     }
210
211     /* Read reference frame from tpx file to compute helix length */
212     snew(xref, top->atoms.nr);
213     read_tpx(ftp2fn(efTPX, NFILE, fnm),
214              NULL, NULL, &natoms, xref, NULL, NULL, NULL);
215     calc_hxprops(nres, bb, xref);
216     do_start_end(nres, bb, &nbb, bbindex, &nca, caindex, bRange, rStart, rEnd);
217     sfree(xref);
218     if (bDBG)
219     {
220         fprintf(stderr, "nca=%d, nbb=%d\n", nca, nbb);
221         pr_bb(stdout, nres, bb);
222     }
223
224     gpbc = gmx_rmpbc_init(&top->idef, ePBC, natoms);
225
226     teller = 0;
227     do
228     {
229         if ((teller++ % 10) == 0)
230         {
231             fprintf(stderr, "\rt=%.2f", t);
232         }
233         gmx_rmpbc(gpbc, natoms, box, x);
234
235
236         calc_hxprops(nres, bb, x);
237         if (bCheck)
238         {
239             do_start_end(nres, bb, &nbb, bbindex, &nca, caindex, FALSE, 0, 0);
240         }
241
242         if (nca >= 5)
243         {
244             rms = fit_ahx(nres, bb, natoms, nall, allindex, x, nca, caindex, bFit);
245
246             if (teller == 1)
247             {
248                 write_sto_conf(opt2fn("-cz", NFILE, fnm), "Helix fitted to Z-Axis",
249                                &(top->atoms), x, NULL, ePBC, box);
250             }
251
252             xf[efhRAD].val   = radius(xf[efhRAD].fp2, nca, caindex, x);
253             xf[efhTWIST].val = twist(nca, caindex, x);
254             xf[efhRISE].val  = rise(nca, caindex, x);
255             xf[efhLEN].val   = ahx_len(nca, caindex, x);
256             xf[efhCD222].val = ellipticity(nres, bb);
257             xf[efhDIP].val   = dip(nbb, bbindex, x, top->atoms.atom);
258             xf[efhRMS].val   = rms;
259             xf[efhCPHI].val  = ca_phi(nca, caindex, x);
260             xf[efhPPRMS].val = pprms(xf[efhPPRMS].fp2, nres, bb);
261
262             for (j = 0; (j <= efhCPHI); j++)
263             {
264                 fprintf(xf[j].fp,   "%10g  %10g\n", t, xf[j].val);
265             }
266
267             av_phipsi(xf[efhPHI].fp, xf[efhPSI].fp, xf[efhPHI].fp2, xf[efhPSI].fp2,
268                       t, nres, bb);
269             av_hblen(xf[efhHB3].fp, xf[efhHB3].fp2,
270                      xf[efhHB4].fp, xf[efhHB4].fp2,
271                      xf[efhHB5].fp, xf[efhHB5].fp2,
272                      t, nres, bb);
273         }
274     }
275     while (read_next_x(oenv, status, &t, x, box));
276     fprintf(stderr, "\n");
277
278     gmx_rmpbc_done(gpbc);
279
280     close_trj(status);
281
282     for (i = 0; (i < nres); i++)
283     {
284         if (bb[i].nrms > 0)
285         {
286             fprintf(xf[efhRMSA].fp, "%10d  %10g\n", r0+i, bb[i].rmsa/bb[i].nrms);
287         }
288         fprintf(xf[efhAHX].fp, "%10d  %10g\n", r0+i, (bb[i].nhx*100.0)/(real )teller);
289         fprintf(xf[efhJCA].fp, "%10d  %10g\n",
290                 r0+i, 140.3+(bb[i].jcaha/(double)teller));
291     }
292
293     for (i = 0; (i < efhNR); i++)
294     {
295         gmx_ffclose(xf[i].fp);
296         if (xf[i].bfp2)
297         {
298             gmx_ffclose(xf[i].fp2);
299         }
300         do_view(oenv, xf[i].filenm, "-nxy");
301     }
302
303     return 0;
304 }