Apply clang-format to source tree
[alexxy/gromacs.git] / src / gromacs / math / veccompare.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,2019, 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 "veccompare.h"
40
41 #include <cmath>
42 #include <cstdio>
43
44 #include "gromacs/utility/compare.h"
45
46 void cmp_rvec(FILE* fp, const char* s, int index, const rvec i1, const rvec i2, real ftol, real abstol)
47 {
48     if (!equal_real(i1[XX], i2[XX], ftol, abstol) || !equal_real(i1[YY], i2[YY], ftol, abstol)
49         || !equal_real(i1[ZZ], i2[ZZ], ftol, abstol))
50     {
51         if (index != -1)
52         {
53             fprintf(fp, "%s[%5d] (%12.5e %12.5e %12.5e) - (%12.5e %12.5e %12.5e)\n", s, index,
54                     i1[XX], i1[YY], i1[ZZ], i2[XX], i2[YY], i2[ZZ]);
55         }
56         else
57         {
58             fprintf(fp, "%s (%12.5e %12.5e %12.5e) - (%12.5e %12.5e %12.5e)\n", s, i1[XX], i1[YY],
59                     i1[ZZ], i2[XX], i2[YY], i2[ZZ]);
60         }
61     }
62 }
63
64 void cmp_ivec(FILE* fp, const char* s, int index, const ivec i1, const ivec i2)
65 {
66     if ((i1[XX] != i2[XX]) || (i1[YY] != i2[YY]) || (i1[ZZ] != i2[ZZ]))
67     {
68         if (index != -1)
69         {
70             fprintf(fp, "%s[%5d] (%8d,%8d,%8d - %8d,%8d,%8d)\n", s, index, i1[XX], i1[YY], i1[ZZ],
71                     i2[XX], i2[YY], i2[ZZ]);
72         }
73         else
74         {
75             fprintf(fp, "%s (%8d,%8d,%8d - %8d,%8d,%8d)\n", s, i1[XX], i1[YY], i1[ZZ], i2[XX],
76                     i2[YY], i2[ZZ]);
77         }
78     }
79 }
80
81 static void cmp_rvecs_rmstol(FILE* fp, const char* title, int n, const rvec x1[], const rvec x2[], real ftol, real abstol)
82 {
83     int    i, m;
84     double rms;
85
86     /* For a vector you are usally not interested in a relative difference
87      * on a component that is very small compared to the other components.
88      * Therefore we do the relative comparision relative to the RMS component.
89      */
90     rms = 0.0;
91     for (i = 0; (i < n); i++)
92     {
93         for (m = 0; m < DIM; m++)
94         {
95             rms += x1[i][m] * x1[i][m] + x2[i][m] * x2[i][m];
96         }
97     }
98     rms = sqrt(rms / (2 * n * DIM));
99
100     /* Convert the relative tolerance into an absolute tolerance */
101     if (ftol * rms < abstol)
102     {
103         abstol = ftol * rms;
104     }
105
106     /* And now do the actual comparision */
107     for (i = 0; (i < n); i++)
108     {
109         cmp_rvec(fp, title, i, x1[i], x2[i], 0.0, abstol);
110     }
111 }
112
113 void cmp_rvecs(FILE* fp, const char* title, int n, const rvec x1[], const rvec x2[], gmx_bool bRMSD, real ftol, real abstol)
114 {
115     int    i, m;
116     double d, ssd;
117
118     if (bRMSD)
119     {
120         ssd = 0;
121         for (i = 0; (i < n); i++)
122         {
123             for (m = 0; m < DIM; m++)
124             {
125                 d = x1[i][m] - x2[i][m];
126                 ssd += d * d;
127             }
128         }
129         fprintf(fp, "%s RMSD %g\n", title, std::sqrt(ssd / n));
130     }
131     else
132     {
133         cmp_rvecs_rmstol(fp, title, n, x1, x2, ftol, abstol);
134     }
135 }