Improve selection interface (remove some pointers).
[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/options/options.h"
46 #include "gromacs/selection/selection.h"
47
48 namespace gmx
49 {
50
51 class SelectionOptionInfo;
52
53 namespace analysismodules
54 {
55
56 class Angle : public TrajectoryAnalysisModule
57 {
58     public:
59         Angle();
60         virtual ~Angle();
61
62         static TrajectoryAnalysisModulePointer create();
63
64         virtual Options &initOptions(TrajectoryAnalysisSettings *settings);
65         virtual void initOptionsDone(TrajectoryAnalysisSettings *settings);
66         virtual void initAnalysis(const TrajectoryAnalysisSettings &settings,
67                                   const TopologyInformation &top);
68
69         virtual void analyzeFrame(int frnr, const t_trxframe &fr, t_pbc *pbc,
70                                   TrajectoryAnalysisModuleData *pdata);
71
72         virtual void finishAnalysis(int nframes);
73         virtual void writeOutput();
74
75     private:
76         void checkSelections(const SelectionList &sel1,
77                              const SelectionList &sel2) const;
78
79         Options                 _options;
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