src/gromacs/gmxana/hxprops.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,2017,2018, 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 "hxprops.h"
  40
  41 #include <cmath>
  42 #include <cstring>
  43
  44 #include <algorithm>
  45
  46 #include "gromacs/listed-forces/bonded.h"
  47 #include "gromacs/math/functions.h"
  48 #include "gromacs/math/units.h"
  49 #include "gromacs/math/vec.h"
  50 #include "gromacs/topology/index.h"
  51 #include "gromacs/topology/topology.h"
  52 #include "gromacs/utility/arraysize.h"
  53 #include "gromacs/utility/fatalerror.h"
  54 #include "gromacs/utility/smalloc.h"
  55
  56 real ellipticity(int nres, t_bb bb[])
  57 {
  58     typedef struct {
  59         real phi, psi, w;
  60     } t_ppwstr;
  61     // Avoid warnings about narrowing conversions from double to real
  62 #ifdef _MSC_VER
  63 #pragma warning(disable: 4838)
  64 #endif
  65     static const t_ppwstr ppw[] = {
  66         {  -67,  -44,  0.31 },
  67         {  -66,  -41,  0.31 },
  68         {  -59,  -44,  0.44 },
  69         {  -57,  -47,  0.56 },
  70         {  -53,  -52,  0.78 },
  71         {  -48,  -57,  1.00 },
  72         {  -70.5, -35.8, 0.15 },
  73         {  -57,  -79,  0.23 },
  74         {  -38,  -78,  1.20 },
  75         {  -60,  -30,  0.24 },
  76         {  -54,  -28,  0.46 },
  77         {  -44,  -33,  0.68 }
  78     };
  79 #ifdef _MSC_VER
  80 #pragma warning(default: 4838)
  81 #endif
  82 #define NPPW asize(ppw)
  83
  84     int        i, j;
  85     real       ell, pp2, phi, psi;
  86
  87     ell = 0;
  88     for (i = 0; (i < nres); i++)
  89     {
  90         phi = bb[i].phi;
  91         psi = bb[i].psi;
  92         for (j = 0; (j < NPPW); j++)
  93         {
  94             pp2 = gmx::square(phi-ppw[j].phi)+gmx::square(psi-ppw[j].psi);
  95             if (pp2 < 64)
  96             {
  97                 bb[i].nhx++;
  98                 ell += ppw[j].w;
  99                 break;
 100             }
 101         }
 102     }
 103     return ell;
 104 }
 105
 106 real ahx_len(int gnx, const int index[], rvec x[])
 107 /* Assume we have a list of Calpha atoms only! */
 108 {
 109     rvec dx;
 110
 111     rvec_sub(x[index[0]], x[index[gnx-1]], dx);
 112
 113     return norm(dx);
 114 }
 115
 116 real radius(FILE *fp, int nca, const int ca_index[], rvec x[])
 117 /* Assume we have all the backbone */
 118 {
 119     real dl2, dlt;
 120     int  i, ai;
 121
 122     dlt = 0;
 123     for (i = 0; (i < nca); i++)
 124     {
 125         ai  = ca_index[i];
 126         dl2 = gmx::square(x[ai][XX])+gmx::square(x[ai][YY]);
 127
 128         if (fp)
 129         {
 130             fprintf(fp, "  %10g", dl2);
 131         }
 132
 133         dlt += dl2;
 134     }
 135     if (fp)
 136     {
 137         fprintf(fp, "\n");
 138     }
 139
 140     return std::sqrt(dlt/nca);
 141 }
 142
 143 static real rot(rvec x1, const rvec x2)
 144 {
 145     real phi1, dphi, cp, sp;
 146     real xx, yy;
 147
 148     phi1 = std::atan2(x1[YY], x1[XX]);
 149     cp   = std::cos(phi1);
 150     sp   = std::sin(phi1);
 151     xx   = cp*x2[XX]+sp*x2[YY];
 152     yy   = -sp*x2[XX]+cp*x2[YY];
 153
 154     dphi = RAD2DEG*std::atan2(yy, xx);
 155
 156     return dphi;
 157 }
 158
 159 real twist(int nca, const int caindex[], rvec x[])
 160 {
 161     real pt, dphi;
 162     int  i, a0, a1;
 163
 164     pt = 0;
 165     a0 = caindex[0];
 166     for (i = 1; (i < nca); i++)
 167     {
 168         a1 = caindex[i];
 169
 170         dphi = rot(x[a0], x[a1]);
 171         if (dphi < -90)
 172         {
 173             dphi += 360;
 174         }
 175         pt += dphi;
 176         a0  = a1;
 177     }
 178
 179     return (pt/(nca-1));
 180 }
 181
 182 real ca_phi(int gnx, const int index[], rvec x[])
 183 /* Assume we have a list of Calpha atoms only! */
 184 {
 185     real phi, phitot;
 186     int  i, ai, aj, ak, al, t1, t2, t3;
 187     rvec r_ij, r_kj, r_kl, m, n;
 188
 189     if (gnx <= 4)
 190     {
 191         return 0;
 192     }
 193
 194     phitot = 0;
 195     for (i = 0; (i < gnx-4); i++)
 196     {
 197         ai  = index[i+0];
 198         aj  = index[i+1];
 199         ak  = index[i+2];
 200         al  = index[i+3];
 201         phi = RAD2DEG*
 202             dih_angle(x[ai], x[aj], x[ak], x[al], nullptr,
 203                       r_ij, r_kj, r_kl, m, n,
 204                       &t1, &t2, &t3);
 205         phitot += phi;
 206     }
 207
 208     return (phitot/(gnx-4.0));
 209 }
 210
 211 real dip(int nbb, int const bbind[], const rvec x[], const t_atom atom[])
 212 {
 213     int  i, m, ai;
 214     rvec dipje;
 215     real q;
 216
 217     clear_rvec(dipje);
 218     for (i = 0; (i < nbb); i++)
 219     {
 220         ai = bbind[i];
 221         q  = atom[ai].q;
 222         for (m = 0; (m < DIM); m++)
 223         {
 224             dipje[m] += x[ai][m]*q;
 225         }
 226     }
 227     return norm(dipje);
 228 }
 229
 230 real rise(int gnx, const int index[], rvec x[])
 231 /* Assume we have a list of Calpha atoms only! */
 232 {
 233     real z, z0, ztot;
 234     int  i, ai;
 235
 236     ai   = index[0];
 237     z0   = x[ai][ZZ];
 238     ztot = 0;
 239     for (i = 1; (i < gnx); i++)
 240     {
 241         ai    = index[i];
 242         z     = x[ai][ZZ];
 243         ztot += (z-z0);
 244         z0    = z;
 245     }
 246
 247     return (ztot/(gnx-1.0));
 248 }
 249
 250 void av_hblen(FILE *fp3, FILE *fp3a,
 251               FILE *fp4, FILE *fp4a,
 252               FILE *fp5, FILE *fp5a,
 253               real t, int nres, t_bb bb[])
 254 {
 255     int  i, n3 = 0, n4 = 0, n5 = 0;
 256     real d3 = 0, d4 = 0, d5 = 0;
 257
 258     for (i = 0; (i < nres-3); i++)
 259     {
 260         if (bb[i].bHelix)
 261         {
 262             fprintf(fp3a,  "%10g", bb[i].d3);
 263             n3++;
 264             d3 += bb[i].d3;
 265             if (i < nres-4)
 266             {
 267                 fprintf(fp4a, "%10g", bb[i].d4);
 268                 n4++;
 269                 d4 += bb[i].d4;
 270             }
 271             if (i < nres-5)
 272             {
 273                 fprintf(fp5a, "%10g", bb[i].d5);
 274                 n5++;
 275                 d5 += bb[i].d5;
 276             }
 277         }
 278     }
 279     fprintf(fp3, "%10g  %10g\n", t, d3/n3);
 280     fprintf(fp4, "%10g  %10g\n", t, d4/n4);
 281     fprintf(fp5, "%10g  %10g\n", t, d5/n5);
 282     fprintf(fp3a, "\n");
 283     fprintf(fp4a, "\n");
 284     fprintf(fp5a, "\n");
 285
 286 }
 287
 288 void av_phipsi(FILE *fphi, FILE *fpsi, FILE *fphi2, FILE *fpsi2,
 289                real t, int nres, t_bb bb[])
 290 {
 291     int  i, n = 0;
 292     real phi = 0, psi = 0;
 293
 294     fprintf(fphi2, "%10g", t);
 295     fprintf(fpsi2, "%10g", t);
 296     for (i = 0; (i < nres); i++)
 297     {
 298         if (bb[i].bHelix)
 299         {
 300             phi += bb[i].phi;
 301             psi += bb[i].psi;
 302             fprintf(fphi2, "  %10g", bb[i].phi);
 303             fprintf(fpsi2, "  %10g", bb[i].psi);
 304             n++;
 305         }
 306     }
 307     fprintf(fphi, "%10g  %10g\n", t, (phi/n));
 308     fprintf(fpsi, "%10g  %10g\n", t, (psi/n));
 309     fprintf(fphi2, "\n");
 310     fprintf(fpsi2, "\n");
 311 }
 312
 313 static void set_ahcity(int nbb, t_bb bb[])
 314 {
 315     real pp2;
 316     int  n;
 317
 318     for (n = 0; (n < nbb); n++)
 319     {
 320         pp2 = gmx::square(bb[n].phi-PHI_AHX)+gmx::square(bb[n].psi-PSI_AHX);
 321
 322         bb[n].bHelix = FALSE;
 323         if (pp2 < 2500)
 324         {
 325             if ((bb[n].d4 < 0.36) || ((n > 0) && bb[n-1].bHelix))
 326             {
 327                 bb[n].bHelix = TRUE;
 328             }
 329         }
 330     }
 331 }
 332
 333 t_bb *mkbbind(const char *fn, int *nres, int *nbb, int res0,
 334               int *nall, int **index,
 335               char ***atomname, t_atom atom[],
 336               t_resinfo *resinfo)
 337 {
 338     static const char * bb_nm[] = { "N", "H", "CA", "C", "O", "HN" };
 339 #define NBB asize(bb_nm)
 340     t_bb               *bb;
 341     char               *grpname;
 342     int                 ai, i, i0, i1, j, k, ri, rnr, gnx, r0, r1;
 343
 344     fprintf(stderr, "Please select a group containing the entire backbone\n");
 345     rd_index(fn, 1, &gnx, index, &grpname);
 346     *nall = gnx;
 347     fprintf(stderr, "Checking group %s\n", grpname);
 348     r0 = r1 = atom[(*index)[0]].resind;
 349     for (i = 1; (i < gnx); i++)
 350     {
 351         r0 = std::min(r0, atom[(*index)[i]].resind);
 352         r1 = std::max(r1, atom[(*index)[i]].resind);
 353     }
 354     rnr = r1-r0+1;
 355     fprintf(stderr, "There are %d residues\n", rnr);
 356     snew(bb, rnr);
 357     for (i = 0; (i < rnr); i++)
 358     {
 359         bb[i].N = bb[i].H = bb[i].CA = bb[i].C = bb[i].O = -1, bb[i].resno = res0+i;
 360     }
 361
 362     for (i = j = 0; (i < gnx); i++)
 363     {
 364         ai = (*index)[i];
 365         ri = atom[ai].resind-r0;
 366         if (std::strcmp(*(resinfo[ri].name), "PRO") == 0)
 367         {
 368             if (std::strcmp(*(atomname[ai]), "CD") == 0)
 369             {
 370                 bb[ri].H = ai;
 371             }
 372         }
 373         for (k = 0; (k < NBB); k++)
 374         {
 375             if (std::strcmp(bb_nm[k], *(atomname[ai])) == 0)
 376             {
 377                 j++;
 378                 break;
 379             }
 380         }
 381         switch (k)
 382         {
 383             case 0:
 384                 bb[ri].N = ai;
 385                 break;
 386             case 1:
 387             case 5:
 388                 /* No attempt to address the case where some weird input has both H and HN atoms in the group */
 389                 bb[ri].H = ai;
 390                 break;
 391             case 2:
 392                 bb[ri].CA = ai;
 393                 break;
 394             case 3:
 395                 bb[ri].C = ai;
 396                 break;
 397             case 4:
 398                 bb[ri].O = ai;
 399                 break;
 400             default:
 401                 break;
 402         }
 403     }
 404
 405     for (i0 = 0; (i0 < rnr); i0++)
 406     {
 407         if ((bb[i0].N != -1) && (bb[i0].H != -1) &&
 408             (bb[i0].CA != -1) &&
 409             (bb[i0].C != -1) && (bb[i0].O != -1))
 410         {
 411             break;
 412         }
 413     }
 414     for (i1 = rnr-1; (i1 >= 0); i1--)
 415     {
 416         if ((bb[i1].N != -1) && (bb[i1].H != -1) &&
 417             (bb[i1].CA != -1) &&
 418             (bb[i1].C != -1) && (bb[i1].O != -1))
 419         {
 420             break;
 421         }
 422     }
 423     if (i0 == 0)
 424     {
 425         i0++;
 426     }
 427     if (i1 == rnr-1)
 428     {
 429         i1--;
 430     }
 431
 432     for (i = i0; (i < i1); i++)
 433     {
 434         bb[i].Cprev = bb[i-1].C;
 435         bb[i].Nnext = bb[i+1].N;
 436     }
 437     rnr = std::max(0, i1-i0+1);
 438     fprintf(stderr, "There are %d complete backbone residues (from %d to %d)\n",
 439             rnr, bb[i0].resno, bb[i1].resno);
 440     if (rnr == 0)
 441     {
 442         gmx_fatal(FARGS, "Zero complete backbone residues were found, cannot proceed");
 443     }
 444     for (i = 0; (i < rnr); i++, i0++)
 445     {
 446         bb[i] = bb[i0];
 447     }
 448
 449     /* Set the labels */
 450     for (i = 0; (i < rnr); i++)
 451     {
 452         ri = atom[bb[i].CA].resind;
 453         sprintf(bb[i].label, "%s%d", *(resinfo[ri].name), ri+res0);
 454     }
 455
 456     *nres = rnr;
 457     *nbb  = rnr*asize(bb_nm);
 458
 459     return bb;
 460 }
 461
 462 real pprms(FILE *fp, int nbb, t_bb bb[])
 463 {
 464     int  i, n;
 465     real rms, rmst, rms2;
 466
 467     rmst = rms2 = 0;
 468     for (i = n = 0; (i < nbb); i++)
 469     {
 470         if (bb[i].bHelix)
 471         {
 472             rms   = std::sqrt(bb[i].pprms2);
 473             rmst += rms;
 474             rms2 += bb[i].pprms2;
 475             fprintf(fp, "%10g  ", rms);
 476             n++;
 477         }
 478     }
 479     fprintf(fp, "\n");
 480     rms = std::sqrt(rms2/n-gmx::square(rmst/n));
 481
 482     return rms;
 483 }
 484
 485 void calc_hxprops(int nres, t_bb bb[], const rvec x[])
 486 {
 487     int  i, ao, an, t1, t2, t3;
 488     rvec dx, r_ij, r_kj, r_kl, m, n;
 489
 490     for (i = 0; (i < nres); i++)
 491     {
 492         ao       = bb[i].O;
 493         bb[i].d4 = bb[i].d3 = bb[i].d5 = 0;
 494         if (i < nres-3)
 495         {
 496             an = bb[i+3].N;
 497             rvec_sub(x[ao], x[an], dx);
 498             bb[i].d3 = norm(dx);
 499         }
 500         if (i < nres-4)
 501         {
 502             an = bb[i+4].N;
 503             rvec_sub(x[ao], x[an], dx);
 504             bb[i].d4 = norm(dx);
 505         }
 506         if (i < nres-5)
 507         {
 508             an = bb[i+5].N;
 509             rvec_sub(x[ao], x[an], dx);
 510             bb[i].d5 = norm(dx);
 511         }
 512
 513         bb[i].phi = RAD2DEG*
 514             dih_angle(x[bb[i].Cprev], x[bb[i].N], x[bb[i].CA], x[bb[i].C], nullptr,
 515                       r_ij, r_kj, r_kl, m, n,
 516                       &t1, &t2, &t3);
 517         bb[i].psi = RAD2DEG*
 518             dih_angle(x[bb[i].N], x[bb[i].CA], x[bb[i].C], x[bb[i].Nnext], nullptr,
 519                       r_ij, r_kj, r_kl, m, n,
 520                       &t1, &t2, &t3);
 521         bb[i].pprms2 = gmx::square(bb[i].phi-PHI_AHX)+gmx::square(bb[i].psi-PSI_AHX);
 522
 523         bb[i].jcaha +=
 524             1.4*std::sin((bb[i].psi+138.0)*DEG2RAD) -
 525             4.1*std::cos(2.0*DEG2RAD*(bb[i].psi+138.0)) +
 526             2.0*std::cos(2.0*DEG2RAD*(bb[i].phi+30.0));
 527     }
 528 }
 529
 530 static void check_ahx(int nres, t_bb bb[],
 531                       int *hstart, int *hend)
 532 {
 533     int  h0, h1, h0sav, h1sav;
 534
 535     set_ahcity(nres, bb);
 536     h0 = h0sav = h1sav = 0;
 537     do
 538     {
 539         for (; (!bb[h0].bHelix) && (h0 < nres-4); h0++)
 540         {
 541             ;
 542         }
 543         for (h1 = h0; bb[h1+1].bHelix && (h1 < nres-1); h1++)
 544         {
 545             ;
 546         }
 547         if (h1 > h0)
 548         {
 549             /*fprintf(stderr,"Helix from %d to %d\n",h0,h1);*/
 550             if (h1-h0 > h1sav-h0sav)
 551             {
 552                 h0sav = h0;
 553                 h1sav = h1;
 554             }
 555         }
 556         h0 = h1+1;
 557     }
 558     while (h1 < nres-1);
 559     *hstart = h0sav;
 560     *hend   = h1sav;
 561 }
 562
 563 void do_start_end(int nres, t_bb bb[], int *nbb, int bbindex[],
 564                   int *nca, int caindex[],
 565                   gmx_bool bRange, int rStart, int rEnd)
 566 {
 567     int    i, j, hstart = 0, hend = 0;
 568
 569     if (bRange)
 570     {
 571         for (i = 0; (i < nres); i++)
 572         {
 573             if ((bb[i].resno >= rStart) && (bb[i].resno <= rEnd))
 574             {
 575                 bb[i].bHelix = TRUE;
 576             }
 577             if (bb[i].resno == rStart)
 578             {
 579                 hstart = i;
 580             }
 581             if (bb[i].resno == rEnd)
 582             {
 583                 hend = i;
 584             }
 585         }
 586     }
 587     else
 588     {
 589         /* Find start and end of longest helix fragment */
 590         check_ahx(nres, bb, &hstart, &hend);
 591     }
 592     fprintf(stderr, "helix from: %d through %d\n",
 593             bb[hstart].resno, bb[hend].resno);
 594
 595     for (j = 0, i = hstart; (i <= hend); i++)
 596     {
 597         bbindex[j++]      = bb[i].N;
 598         bbindex[j++]      = bb[i].H;
 599         bbindex[j++]      = bb[i].CA;
 600         bbindex[j++]      = bb[i].C;
 601         bbindex[j++]      = bb[i].O;
 602         caindex[i-hstart] = bb[i].CA;
 603     }
 604     *nbb = j;
 605     *nca = (hend-hstart+1);
 606 }
 607
 608 void pr_bb(FILE *fp, int nres, t_bb bb[])
 609 {
 610     int i;
 611
 612     fprintf(fp, "\n");
 613     fprintf(fp, "%3s %3s %3s %3s %3s %7s %7s %7s %7s %7s %3s\n",
 614             "AA", "N", "Ca", "C", "O", "Phi", "Psi", "D3", "D4", "D5", "Hx?");
 615     for (i = 0; (i < nres); i++)
 616     {
 617         fprintf(fp, "%3d %3d %3d %3d %3d %7.2f %7.2f %7.3f %7.3f %7.3f %3s\n",
 618                 bb[i].resno, bb[i].N, bb[i].CA, bb[i].C, bb[i].O,
 619                 bb[i].phi, bb[i].psi, bb[i].d3, bb[i].d4, bb[i].d5,
 620                 bb[i].bHelix ? "Yes" : "No");
 621     }
 622     fprintf(fp, "\n");
 623 }