Re-organize BlueGene toolchain files

[gromacs.git] / src / tools / gmx_potential.c

/*
 * This file is part of the GROMACS molecular simulation package.
 *
 * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
 * Copyright (c) 2001-2004, The GROMACS development team,
 * check out http://www.gromacs.org for more information.
 * Copyright (c) 2012, by the GROMACS development team, led by
 * David van der Spoel, Berk Hess, Erik Lindahl, and including many
 * others, as listed in the AUTHORS file in the top-level source
 * directory and at http://www.gromacs.org.
 *
 * GROMACS is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public License
 * as published by the Free Software Foundation; either version 2.1
 * of the License, or (at your option) any later version.
 *
 * GROMACS is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with GROMACS; if not, see
 * http://www.gnu.org/licenses, or write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
 *
 * If you want to redistribute modifications to GROMACS, please
 * consider that scientific software is very special. Version
 * control is crucial - bugs must be traceable. We will be happy to
 * consider code for inclusion in the official distribution, but
 * derived work must not be called official GROMACS. Details are found
 * in the README & COPYING files - if they are missing, get the
 * official version at http://www.gromacs.org.
 *
 * To help us fund GROMACS development, we humbly ask that you cite
 * the research papers on the package. Check out http://www.gromacs.org.
 */
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include "gmx_header_config.h"

#include <math.h>
#include <ctype.h>

#include "sysstuff.h"
#include <string.h>
#include "typedefs.h"
#include "smalloc.h"
#include "macros.h"
#include "princ.h"
#include "rmpbc.h"
#include "vec.h"
#include "xvgr.h"
#include "pbc.h"
#include "copyrite.h"
#include "futil.h"
#include "statutil.h"
#include "index.h"
#include "gmx_ana.h"

/* Suppress Cygwin compiler warnings from using newlib version of
 * ctype.h */
#ifdef GMX_CYGWIN
#undef toupper
#endif

#define EPS0 8.85419E-12
#define ELC 1.60219E-19

/****************************************************************************/
/* This program calculates the electrostatic potential across the box by    */
/* determining the charge density in slices of the box and integrating these*/
/* twice.                                                                   */
/* Peter Tieleman, April 1995                                               */
/* It now also calculates electrostatic potential in spherical micelles,    */
/* using \frac{1}{r}\frac{d^2r\Psi}{r^2} = - \frac{\rho}{\epsilon_0}        */
/* This probably sucks but it seems to work.                                */
/****************************************************************************/

static int ce=0, cb=0;

/* this routine integrates the array data and returns the resulting array */
/* routine uses simple trapezoid rule                                     */
void p_integrate(double *result, double data[], int ndata, double slWidth)
{
  int i, slice;
  double sum;
  
  if (ndata <= 2) 
    fprintf(stderr,"Warning: nr of slices very small. This will result"
            "in nonsense.\n");

  fprintf(stderr,"Integrating from slice %d to slice %d\n",cb, ndata-ce);

  for (slice = cb; slice < (ndata-ce); slice ++)
  {
    sum = 0;
    for (i = cb; i < slice; i++)
      sum += slWidth * (data[i] + 0.5 * (data[i+1] - data[i]));
    result[slice] = sum;
  }
  return;
}

void calc_potential(const char *fn, atom_id **index, int gnx[], 
                    double ***slPotential, double ***slCharge, 
                    double ***slField, int *nslices, 
                    t_topology *top, int ePBC,
                    int axis, int nr_grps, double *slWidth,
                    double fudge_z, gmx_bool bSpherical, gmx_bool bCorrect,
                    const output_env_t oenv)
{
  rvec *x0;              /* coordinates without pbc */
  matrix box;            /* box (3x3) */
  int natoms;            /* nr. atoms in trj */
  t_trxstatus *status;
  int **slCount,         /* nr. of atoms in one slice for a group */
      i,j,n,             /* loop indices */
      teller = 0,      
      ax1=0, ax2=0,
      nr_frames = 0,     /* number of frames */
      slice;             /* current slice */
  double slVolume;         /* volume of slice for spherical averaging */
  double qsum,nn;
  real t;
  double z;
  rvec xcm;
  gmx_rmpbc_t  gpbc=NULL;

  switch(axis)
  {
  case 0:
    ax1 = 1; ax2 = 2;
    break;
  case 1:
    ax1 = 0; ax2 = 2;
    break;
  case 2:
    ax1 = 0; ax2 = 1;
    break;
  default:
    gmx_fatal(FARGS,"Invalid axes. Terminating\n");
  }

  if ((natoms = read_first_x(oenv,&status,fn,&t,&x0,box)) == 0)
    gmx_fatal(FARGS,"Could not read coordinates from statusfile\n");

  if (! *nslices)
    *nslices = (int)(box[axis][axis] * 10); /* default value */

  fprintf(stderr,"\nDividing the box in %d slices\n",*nslices);

  snew(*slField, nr_grps);
  snew(*slCharge, nr_grps);
  snew(*slPotential, nr_grps);

  for (i = 0; i < nr_grps; i++)
  {
    snew((*slField)[i], *nslices);
    snew((*slCharge)[i], *nslices);
    snew((*slPotential)[i], *nslices);
  }


  gpbc = gmx_rmpbc_init(&top->idef,ePBC,natoms,box);
  
  /*********** Start processing trajectory ***********/
  do 
  {
    *slWidth = box[axis][axis]/(*nslices);
    teller++;
    gmx_rmpbc(gpbc,natoms,box,x0);

    /* calculate position of center of mass based on group 1 */
    calc_xcm(x0, gnx[0], index[0], top->atoms.atom, xcm, FALSE);
    svmul(-1,xcm,xcm);
          
    for (n = 0; n < nr_grps; n++)
    {      
        /* Check whether we actually have all positions of the requested index
         * group in the trajectory file */
        if (gnx[n] > natoms)
        {
            gmx_fatal(FARGS, "You selected a group with %d atoms, but only %d atoms\n"
                             "were found in the trajectory.\n", gnx[n], natoms);
        }
      for (i = 0; i < gnx[n]; i++)   /* loop over all atoms in index file */
      {
        if (bSpherical)
        {
          rvec_add(x0[index[n][i]], xcm, x0[index[n][i]]);
          /* only distance from origin counts, not sign */
          slice = norm(x0[index[n][i]])/(*slWidth);
          
          /* this is a nice check for spherical groups but not for
             all water in a cubic box since a lot will fall outside
             the sphere
            if (slice > (*nslices)) 
            {
             fprintf(stderr,"Warning: slice = %d\n",slice);
            }
          */
          (*slCharge)[n][slice] += top->atoms.atom[index[n][i]].q;
        }
        else
        {
          z = x0[index[n][i]][axis];
          z = z + fudge_z;
          if (z < 0) 
            z += box[axis][axis];
          if (z > box[axis][axis])
            z -= box[axis][axis];
          /* determine which slice atom is in */
          slice = (z / (*slWidth)); 
          (*slCharge)[n][slice] += top->atoms.atom[index[n][i]].q;
        }
      }
    }
    nr_frames++;
  } while (read_next_x(oenv,status,&t,natoms,x0,box));

  gmx_rmpbc_done(gpbc);

  /*********** done with status file **********/
  close_trj(status);
  
  /* slCharge now contains the total charge per slice, summed over all
     frames. Now divide by nr_frames and integrate twice 
   */

  
  if (bSpherical)
    fprintf(stderr,"\n\nRead %d frames from trajectory. Calculating potential"
            "in spherical coordinates\n", nr_frames);
  else
    fprintf(stderr,"\n\nRead %d frames from trajectory. Calculating potential\n",
            nr_frames);

  for (n =0; n < nr_grps; n++)
  {
      for (i = 0; i < *nslices; i++)
      {
          if (bSpherical)
          {
              /* charge per volume is now the summed charge, divided by the nr
              of frames and by the volume of the slice it's in, 4pi r^2 dr
              */
              slVolume = 4*M_PI * sqr(i) * sqr(*slWidth) * *slWidth;
              if (slVolume == 0)
              {
                  (*slCharge)[n][i] = 0;
              }
              else
              {
                  (*slCharge)[n][i] = (*slCharge)[n][i] / (nr_frames * slVolume);
              }
          }
          else
          {
              /* get charge per volume */
              (*slCharge)[n][i] = (*slCharge)[n][i] * (*nslices) /
              (nr_frames * box[axis][axis] * box[ax1][ax1] * box[ax2][ax2]);              
          }
      }
      /* Now we have charge densities */
  }
  
  if(bCorrect && !bSpherical)
  {
      for(n =0; n < nr_grps; n++)
      {
          nn = 0;
          qsum = 0;
          for (i = 0; i < *nslices; i++)
          {
              if( fabs((*slCharge)[n][i]) >= GMX_DOUBLE_MIN )
              {
                  nn++;
                  qsum += (*slCharge)[n][i];
              }
          }          
          qsum /= nn;
          for (i = 0; i < *nslices; i++)
          {
              if( fabs((*slCharge)[n][i]) >= GMX_DOUBLE_MIN )
              {
                  (*slCharge)[n][i] -= qsum;
              }
          }
      } 
  }
  
  for(n =0; n < nr_grps; n++)
  {
      /* integrate twice to get field and potential */
      p_integrate((*slField)[n], (*slCharge)[n], *nslices, *slWidth);
  }
  
  
  if(bCorrect && !bSpherical)
  {
      for(n =0; n < nr_grps; n++)
      {
          nn = 0;
          qsum = 0;
          for (i = 0; i < *nslices; i++)
          {
              if( fabs((*slCharge)[n][i]) >= GMX_DOUBLE_MIN )
              {
                  nn++;
                  qsum += (*slField)[n][i];
              }
          }          
          qsum /= nn;
          for (i = 0; i < *nslices; i++)
          {
              if( fabs((*slCharge)[n][i]) >= GMX_DOUBLE_MIN )
              {
                  (*slField)[n][i] -= qsum;
              }
          }
      } 
  }
  
  for(n =0; n < nr_grps; n++)
  {      
      p_integrate((*slPotential)[n],(*slField)[n], *nslices, *slWidth);
  }
  
  /* Now correct for eps0 and in spherical case for r*/
  for (n = 0; n < nr_grps; n++)
    for (i = 0; i < *nslices; i++)
    {
      if (bSpherical)
      {
        (*slPotential)[n][i] = ELC * (*slPotential)[n][i] * -1.0E9 / 
          (EPS0 * i * (*slWidth));
        (*slField)[n][i] = ELC * (*slField)[n][i] * 1E18 / 
          (EPS0 * i * (*slWidth));
      }
      else 
      {
        (*slPotential)[n][i] = ELC * (*slPotential)[n][i] * -1.0E9 / EPS0  ;
        (*slField)[n][i] = ELC * (*slField)[n][i] * 1E18 / EPS0;
      }
    }
  
  sfree(x0);  /* free memory used by coordinate array */
}

void plot_potential(double *potential[], double *charge[], double *field[], 
                    const char *afile, const char *bfile, const char *cfile, 
                    int nslices, int nr_grps, const char *grpname[], double slWidth,
                    const output_env_t oenv)
{
  FILE       *pot,     /* xvgr file with potential */
             *cha,     /* xvgr file with charges   */
             *fie;     /* xvgr files with fields   */
  char       buf[256]; /* for xvgr title */
  int        slice, n;

  sprintf(buf,"Electrostatic Potential");
  pot = xvgropen(afile, buf, "Box (nm)","Potential (V)",oenv);
  xvgr_legend(pot,nr_grps,grpname,oenv);

  sprintf(buf,"Charge Distribution");
  cha = xvgropen(bfile, buf, "Box (nm)", "Charge density (q/nm\\S3\\N)",oenv);
  xvgr_legend(cha,nr_grps,grpname,oenv);

  sprintf(buf, "Electric Field");
  fie = xvgropen(cfile, buf, "Box (nm)", "Field (V/nm)",oenv);
  xvgr_legend(fie,nr_grps,grpname,oenv);

  for (slice = cb; slice < (nslices - ce); slice++)
  { 
    fprintf(pot,"%20.16g  ", slice * slWidth);
    fprintf(cha,"%20.16g  ", slice * slWidth);
    fprintf(fie,"%20.16g  ", slice * slWidth);
    for (n = 0; n < nr_grps; n++)
    {
      fprintf(pot,"   %20.16g", potential[n][slice]);
      fprintf(fie,"   %20.16g", field[n][slice]/1e9);  /* convert to V/nm */
      fprintf(cha,"   %20.16g", charge[n][slice]);
    }
    fprintf(pot,"\n");
    fprintf(cha,"\n");
    fprintf(fie,"\n");
  }

  ffclose(pot);
  ffclose(cha);
  ffclose(fie);
}

int gmx_potential(int argc,char *argv[])
{
  const char *desc[] = {
    "[TT]g_potential[tt] computes the electrostatical potential across the box. The potential is",
    "calculated by first summing the charges per slice and then integrating",
    "twice of this charge distribution. Periodic boundaries are not taken",
    "into account. Reference of potential is taken to be the left side of",
    "the box. It is also possible to calculate the potential in spherical",
    "coordinates as function of r by calculating a charge distribution in",
    "spherical slices and twice integrating them. epsilon_r is taken as 1,",
    "but 2 is more appropriate in many cases."
  };
  output_env_t oenv;
  static int  axis = 2;                      /* normal to memb. default z  */
  static const char *axtitle="Z"; 
  static int  nslices = 10;                  /* nr of slices defined       */
  static int  ngrps   = 1;
  static gmx_bool bSpherical = FALSE;            /* default is bilayer types   */
  static real fudge_z = 0;                    /* translate coordinates      */
  static gmx_bool bCorrect = 0;
  t_pargs pa [] = {
    { "-d",   FALSE, etSTR, {&axtitle}, 
      "Take the normal on the membrane in direction X, Y or Z." },
    { "-sl",  FALSE, etINT, {&nslices},
      "Calculate potential as function of boxlength, dividing the box"
      " in this number of slices." } ,
    { "-cb",  FALSE, etINT, {&cb},
      "Discard this number of  first slices of box for integration" },
    { "-ce",  FALSE, etINT, {&ce},
      "Discard this number of last slices of box for integration" },
    { "-tz",  FALSE, etREAL, {&fudge_z},
      "Translate all coordinates by this distance in the direction of the box" },
    { "-spherical", FALSE, etBOOL, {&bSpherical},
      "Calculate spherical thingie" },
    { "-ng",       FALSE, etINT, {&ngrps},
      "Number of groups to consider" },
    { "-correct",  FALSE, etBOOL, {&bCorrect},
      "Assume net zero charge of groups to improve accuracy" }
  };
  const char *bugs[] = {
    "Discarding slices for integration should not be necessary."
  };

  double    **potential,                    /* potential per slice        */
            **charge,                       /* total charge per slice     */
            **field,                        /* field per slice            */
            slWidth;                        /* width of one slice         */
  char      **grpname;                      /* groupnames                 */
  int       *ngx;                           /* sizes of groups            */
  t_topology *top;                          /* topology                   */ 
  int       ePBC;
  atom_id   **index;                        /* indices for all groups     */
  t_filenm  fnm[] = {                       /* files for g_order          */
    { efTRX, "-f", NULL,  ffREAD },         /* trajectory file            */
    { efNDX, NULL, NULL,  ffREAD },         /* index file                 */
    { efTPX, NULL, NULL,  ffREAD },         /* topology file              */
    { efXVG,"-o","potential", ffWRITE },    /* xvgr output file           */
    { efXVG,"-oc","charge", ffWRITE },      /* xvgr output file           */
    { efXVG,"-of","field", ffWRITE },       /* xvgr output file           */
  };

#define NFILE asize(fnm)

  CopyRight(stderr,argv[0]);
  parse_common_args(&argc, argv, PCA_CAN_VIEW | PCA_CAN_TIME | PCA_BE_NICE,
                    NFILE,fnm,asize(pa),pa,asize(desc),desc,asize(bugs),bugs,
                    &oenv);

  /* Calculate axis */
  axis = toupper(axtitle[0]) - 'X';
  
  top = read_top(ftp2fn(efTPX,NFILE,fnm),&ePBC);     /* read topology file */

  snew(grpname,ngrps);
  snew(index,ngrps);
  snew(ngx,ngrps);
 
  rd_index(ftp2fn(efNDX,NFILE,fnm),ngrps,ngx,index,grpname); 

  
  calc_potential(ftp2fn(efTRX,NFILE,fnm), index, ngx, 
                 &potential, &charge, &field,
                 &nslices, top, ePBC, axis, ngrps, &slWidth, fudge_z,
                 bSpherical, bCorrect,oenv); 

  plot_potential(potential, charge, field, opt2fn("-o",NFILE,fnm),
                 opt2fn("-oc",NFILE,fnm), opt2fn("-of",NFILE,fnm),
                 nslices, ngrps, (const char**)grpname, slWidth,oenv);

  do_view(oenv,opt2fn("-o",NFILE,fnm), NULL);       /* view xvgr file */
  do_view(oenv,opt2fn("-oc",NFILE,fnm), NULL);      /* view xvgr file */  
  do_view(oenv,opt2fn("-of",NFILE,fnm), NULL);      /* view xvgr file */

  thanx(stderr);
  return 0;
}