Merge release-4-6 into master
[alexxy/gromacs.git] / src / gromacs / trajectoryanalysis / modules / angle.h
1 /*
2  *
3  *                This source code is part of
4  *
5  *                 G   R   O   M   A   C   S
6  *
7  *          GROningen MAchine for Chemical Simulations
8  *
9  * Written by David van der Spoel, Erik Lindahl, Berk Hess, and others.
10  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
11  * Copyright (c) 2001-2009, The GROMACS development team,
12  * check out http://www.gromacs.org for more information.
13
14  * This program is free software; you can redistribute it and/or
15  * modify it under the terms of the GNU General Public License
16  * as published by the Free Software Foundation; either version 2
17  * of the License, or (at your option) any later version.
18  *
19  * If you want to redistribute modifications, please consider that
20  * scientific software is very special. Version control is crucial -
21  * bugs must be traceable. We will be happy to consider code for
22  * inclusion in the official distribution, but derived work must not
23  * be called official GROMACS. Details are found in the README & COPYING
24  * files - if they are missing, get the official version at www.gromacs.org.
25  *
26  * To help us fund GROMACS development, we humbly ask that you cite
27  * the papers on the package - you can find them in the top README file.
28  *
29  * For more info, check our website at http://www.gromacs.org
30  */
31 /*! \internal \file
32  * \brief
33  * Declares trajectory analysis module for angle calculations.
34  *
35  * \author Teemu Murtola <teemu.murtola@cbr.su.se>
36  * \ingroup module_trajectoryanalysis
37  */
38 #ifndef GMX_TRAJECTORYANALYSIS_MODULES_ANGLE_H
39 #define GMX_TRAJECTORYANALYSIS_MODULES_ANGLE_H
40
41 #include <string>
42
43 #include "../analysismodule.h"
44 #include "gromacs/analysisdata/analysisdata.h"
45 #include "gromacs/selection/selection.h"
46
47 namespace gmx
48 {
49
50 class SelectionOptionInfo;
51
52 namespace analysismodules
53 {
54
55 class Angle : public TrajectoryAnalysisModule
56 {
57     public:
58         static const char name[];
59         static const char shortDescription[];
60
61         Angle();
62         virtual ~Angle();
63
64         virtual void initOptions(Options *options,
65                                  TrajectoryAnalysisSettings *settings);
66         virtual void optionsFinished(Options *options,
67                                      TrajectoryAnalysisSettings *settings);
68         virtual void initAnalysis(const TrajectoryAnalysisSettings &settings,
69                                   const TopologyInformation &top);
70
71         virtual void analyzeFrame(int frnr, const t_trxframe &fr, t_pbc *pbc,
72                                   TrajectoryAnalysisModuleData *pdata);
73
74         virtual void finishAnalysis(int nframes);
75         virtual void writeOutput();
76
77     private:
78         void checkSelections(const SelectionList &sel1,
79                              const SelectionList &sel2) const;
80
81         SelectionList           sel1_;
82         SelectionList           sel2_;
83         SelectionOptionInfo    *sel1info_;
84         SelectionOptionInfo    *sel2info_;
85         std::string             fnAngle_;
86         std::string             fnDump_;
87
88         std::string             g1type_;
89         std::string             g2type_;
90         bool                    bSplit1_;
91         bool                    bSplit2_;
92         bool                    bMulti_;
93         bool                    bAll_;
94         bool                    bDumpDist_;
95
96         AnalysisData            data_;
97         int                     natoms1_;
98         int                     natoms2_;
99         rvec                   *vt0_;
100
101         // Copy and assign disallowed by base.
102 };
103
104 } // namespace analysismodules
105
106 } // namespace gmxana
107
108 #endif