Sort all includes in src/gromacs
[alexxy/gromacs.git] / src / gromacs / mdlib / nlistheuristics.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) 2012,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 "gromacs/legacyheaders/typedefs.h"
40 #include "gromacs/legacyheaders/types/nlistheuristics.h"
41 #include "gromacs/math/vec.h"
42 #include "gromacs/utility/cstringutil.h"
43 #include "gromacs/utility/fatalerror.h"
44
45 void reset_nlistheuristics(gmx_nlheur_t *nlh, gmx_int64_t step)
46 {
47     nlh->lt_runav     = 0;
48     nlh->lt_runav2    = 0;
49     nlh->step_nscheck = step;
50 }
51
52 void init_nlistheuristics(gmx_nlheur_t *nlh,
53                           gmx_bool bGStatEveryStep, gmx_int64_t step)
54 {
55     nlh->bGStatEveryStep = bGStatEveryStep;
56     nlh->nns             = 0;
57     nlh->nabnsb          = 0;
58     nlh->s1              = 0;
59     nlh->s2              = 0;
60     nlh->ab              = 0;
61
62     reset_nlistheuristics(nlh, step);
63 }
64
65 void update_nliststatistics(gmx_nlheur_t *nlh, gmx_int64_t step)
66 {
67     gmx_int64_t     nl_lt;
68     char            sbuf[STEPSTRSIZE], sbuf2[STEPSTRSIZE];
69
70     /* Determine the neighbor list life time */
71     nl_lt = step - nlh->step_ns;
72     if (debug)
73     {
74         fprintf(debug, "%d atoms beyond ns buffer, updating neighbor list after %s steps\n", nlh->nabnsb, gmx_step_str(nl_lt, sbuf));
75     }
76     nlh->nns++;
77     nlh->s1 += nl_lt;
78     nlh->s2 += nl_lt*nl_lt;
79     nlh->ab += nlh->nabnsb;
80     if (nlh->lt_runav == 0)
81     {
82         nlh->lt_runav  = nl_lt;
83         /* Initialize the fluctuation average
84          * such that at startup we check after 0 steps.
85          */
86         nlh->lt_runav2 = sqr(nl_lt/2.0);
87     }
88     /* Running average with 0.9 gives an exp. history of 9.5 */
89     nlh->lt_runav2 = 0.9*nlh->lt_runav2 + 0.1*sqr(nlh->lt_runav - nl_lt);
90     nlh->lt_runav  = 0.9*nlh->lt_runav  + 0.1*nl_lt;
91     if (nlh->bGStatEveryStep)
92     {
93         /* Always check the nlist validity */
94         nlh->step_nscheck = step;
95     }
96     else
97     {
98         /* We check after:  <life time> - 2*sigma
99          * The factor 2 is quite conservative,
100          * but we assume that with nstlist=-1 the user
101          * prefers exact integration over performance.
102          */
103         nlh->step_nscheck = step
104             + (int)(nlh->lt_runav - 2.0*sqrt(nlh->lt_runav2)) - 1;
105     }
106     if (debug)
107     {
108         fprintf(debug, "nlist life time %s run av. %4.1f sig %3.1f check %s check with -gcom %d\n",
109                 gmx_step_str(nl_lt, sbuf), nlh->lt_runav, sqrt(nlh->lt_runav2),
110                 gmx_step_str(nlh->step_nscheck-step+1, sbuf2),
111                 (int)(nlh->lt_runav - 2.0*sqrt(nlh->lt_runav2)));
112     }
113 }
114
115 void set_nlistheuristics(gmx_nlheur_t *nlh, gmx_bool bReset, gmx_int64_t step)
116 {
117     int d;
118
119     if (bReset)
120     {
121         reset_nlistheuristics(nlh, step);
122     }
123     else
124     {
125         update_nliststatistics(nlh, step);
126     }
127
128     nlh->step_ns = step;
129     /* Initialize the cumulative coordinate scaling matrix */
130     clear_mat(nlh->scale_tot);
131     for (d = 0; d < DIM; d++)
132     {
133         nlh->scale_tot[d][d] = 1.0;
134     }
135 }