ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/trunk/src/applications/staticProps/pAngle.cpp
Revision: 1991
Committed: Wed Apr 23 20:34:17 2014 UTC (11 years ago) by gezelter
File size: 11792 byte(s)
Log Message:
added selection offset 2 for pAngle

File Contents

# User Rev Content
1 gezelter 1413 /*
2     * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved.
3     *
4     * The University of Notre Dame grants you ("Licensee") a
5     * non-exclusive, royalty free, license to use, modify and
6     * redistribute this software in source and binary code form, provided
7     * that the following conditions are met:
8     *
9     * 1. Redistributions of source code must retain the above copyright
10     * notice, this list of conditions and the following disclaimer.
11     *
12     * 2. Redistributions in binary form must reproduce the above copyright
13     * notice, this list of conditions and the following disclaimer in the
14     * documentation and/or other materials provided with the
15     * distribution.
16     *
17     * This software is provided "AS IS," without a warranty of any
18     * kind. All express or implied conditions, representations and
19     * warranties, including any implied warranty of merchantability,
20     * fitness for a particular purpose or non-infringement, are hereby
21     * excluded. The University of Notre Dame and its licensors shall not
22     * be liable for any damages suffered by licensee as a result of
23     * using, modifying or distributing the software or its
24     * derivatives. In no event will the University of Notre Dame or its
25     * licensors be liable for any lost revenue, profit or data, or for
26     * direct, indirect, special, consequential, incidental or punitive
27     * damages, however caused and regardless of the theory of liability,
28     * arising out of the use of or inability to use software, even if the
29     * University of Notre Dame has been advised of the possibility of
30     * such damages.
31     *
32     * SUPPORT OPEN SCIENCE! If you use OpenMD or its source code in your
33     * research, please cite the appropriate papers when you publish your
34     * work. Good starting points are:
35     *
36     * [1] Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).
37     * [2] Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).
38 gezelter 1879 * [3] Sun, Lin & Gezelter, J. Chem. Phys. 128, 234107 (2008).
39 gezelter 1782 * [4] Kuang & Gezelter, J. Chem. Phys. 133, 164101 (2010).
40     * [4] , Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011). *
41 gezelter 1413 */
42    
43     /* Calculates Rho(theta) */
44    
45     #include <algorithm>
46 gezelter 1782 #include <fstream>
47 gezelter 1413 #include "applications/staticProps/pAngle.hpp"
48     #include "utils/simError.h"
49     #include "io/DumpReader.hpp"
50     #include "primitives/Molecule.hpp"
51 gezelter 1782 #include "brains/Thermo.hpp"
52    
53 gezelter 1413 namespace OpenMD {
54    
55     pAngle::pAngle(SimInfo* info, const std::string& filename,
56 gezelter 1979 const std::string& sele1, int nthetabins)
57     : StaticAnalyser(info, filename), selectionScript1_(sele1),
58     evaluator1_(info), evaluator2_(info), seleMan1_(info), seleMan2_(info),
59     nThetaBins_(nthetabins),
60     doVect_(true), doOffset_(false) {
61 gezelter 1413
62 gezelter 1979 setOutputName(getPrefix(filename) + ".pAngle");
63    
64     evaluator1_.loadScriptString(sele1);
65     if (!evaluator1_.isDynamic()) {
66     seleMan1_.setSelectionSet(evaluator1_.evaluate());
67 gezelter 1413 }
68    
69     count_.resize(nThetaBins_);
70 gezelter 1979 histogram_.resize(nThetaBins_);
71     }
72    
73     pAngle::pAngle(SimInfo* info, const std::string& filename,
74     const std::string& sele1, const std::string& sele2,
75     int nthetabins)
76     : StaticAnalyser(info, filename), selectionScript1_(sele1),
77     selectionScript2_(sele2), evaluator1_(info), evaluator2_(info),
78     seleMan1_(info), seleMan2_(info), nThetaBins_(nthetabins),
79     doVect_(false), doOffset_(false) {
80 gezelter 1413
81     setOutputName(getPrefix(filename) + ".pAngle");
82 gezelter 1979
83     evaluator1_.loadScriptString(sele1);
84     if (!evaluator1_.isDynamic()) {
85     seleMan1_.setSelectionSet(evaluator1_.evaluate());
86     }
87    
88     evaluator2_.loadScriptString(sele2);
89     if (!evaluator2_.isDynamic()) {
90     seleMan2_.setSelectionSet(evaluator2_.evaluate());
91     }
92    
93     count_.resize(nThetaBins_);
94     histogram_.resize(nThetaBins_);
95 gezelter 1413 }
96 gezelter 1979
97     pAngle::pAngle(SimInfo* info, const std::string& filename,
98     const std::string& sele1, int seleOffset, int nthetabins)
99     : StaticAnalyser(info, filename), selectionScript1_(sele1),
100     evaluator1_(info), evaluator2_(info), seleMan1_(info), seleMan2_(info),
101 gezelter 1988 nThetaBins_(nthetabins), seleOffset_(seleOffset),
102 gezelter 1991 doVect_(false), doOffset_(true), doOffset2_(false) {
103 gezelter 1979
104     setOutputName(getPrefix(filename) + ".pAngle");
105    
106     evaluator1_.loadScriptString(sele1);
107     if (!evaluator1_.isDynamic()) {
108     seleMan1_.setSelectionSet(evaluator1_.evaluate());
109     }
110    
111     count_.resize(nThetaBins_);
112     histogram_.resize(nThetaBins_);
113     }
114 gezelter 1991
115     pAngle::pAngle(SimInfo* info, const std::string& filename,
116     const std::string& sele1, int seleOffset, int seleOffset2,
117     int nthetabins)
118     : StaticAnalyser(info, filename), selectionScript1_(sele1),
119     evaluator1_(info), evaluator2_(info), seleMan1_(info), seleMan2_(info),
120     nThetaBins_(nthetabins), seleOffset_(seleOffset),
121     seleOffset2_(seleOffset2),
122     doVect_(false), doOffset_(true), doOffset2_(true) {
123    
124     setOutputName(getPrefix(filename) + ".pAngle");
125    
126     evaluator1_.loadScriptString(sele1);
127     if (!evaluator1_.isDynamic()) {
128     seleMan1_.setSelectionSet(evaluator1_.evaluate());
129     }
130    
131     count_.resize(nThetaBins_);
132     histogram_.resize(nThetaBins_);
133     }
134 gezelter 1413
135     void pAngle::process() {
136     Molecule* mol;
137     RigidBody* rb;
138     SimInfo::MoleculeIterator mi;
139     Molecule::RigidBodyIterator rbIter;
140 gezelter 1979 StuntDouble* sd1;
141     StuntDouble* sd2;
142     int ii;
143     int jj;
144 gezelter 1413
145 gezelter 1782 Thermo thermo(info_);
146 gezelter 1413 DumpReader reader(info_, dumpFilename_);
147     int nFrames = reader.getNFrames();
148 gezelter 1979
149 gezelter 1413 nProcessed_ = nFrames/step_;
150    
151     std::fill(histogram_.begin(), histogram_.end(), 0.0);
152     std::fill(count_.begin(), count_.end(), 0);
153    
154     for (int istep = 0; istep < nFrames; istep += step_) {
155     reader.readFrame(istep);
156     currentSnapshot_ = info_->getSnapshotManager()->getCurrentSnapshot();
157    
158     for (mol = info_->beginMolecule(mi); mol != NULL;
159     mol = info_->nextMolecule(mi)) {
160     //change the positions of atoms which belong to the rigidbodies
161     for (rb = mol->beginRigidBody(rbIter); rb != NULL;
162     rb = mol->nextRigidBody(rbIter)) {
163     rb->updateAtoms();
164     }
165     }
166    
167 gezelter 1782 Vector3d CenterOfMass = thermo.getCom();
168 gezelter 1413
169 gezelter 1979 if (evaluator1_.isDynamic()) {
170     seleMan1_.setSelectionSet(evaluator1_.evaluate());
171 gezelter 1413 }
172    
173 gezelter 1979 if (doVect_) {
174 gezelter 1988
175 gezelter 1782
176 gezelter 1979 for (sd1 = seleMan1_.beginSelected(ii); sd1 != NULL;
177     sd1 = seleMan1_.nextSelected(ii)) {
178 gezelter 1413
179 gezelter 1979 Vector3d pos = sd1->getPos();
180    
181     Vector3d r1 = CenterOfMass - pos;
182     // only do this if the stunt double actually has a vector associated
183     // with it
184     if (sd1->isDirectional()) {
185     Vector3d vec = sd1->getA().getColumn(2);
186     RealType distance = r1.length();
187    
188     vec.normalize();
189     r1.normalize();
190     RealType cosangle = dot(r1, vec);
191    
192     int binNo = int(nThetaBins_ * (1.0 + cosangle) / 2.0);
193     count_[binNo]++;
194     }
195 gezelter 1413 }
196 gezelter 1979 } else {
197     if (doOffset_) {
198    
199     for (sd1 = seleMan1_.beginSelected(ii); sd1 != NULL;
200     sd1 = seleMan1_.nextSelected(ii)) {
201    
202     // This will require careful rewriting if StaticProps is
203     // ever parallelized. For an example, see
204     // Thermo::getTaggedAtomPairDistance
205 gezelter 1991 Vector3d r1;
206 gezelter 1979
207 gezelter 1991 if (doOffset2_) {
208     int sd1Aind = sd1->getGlobalIndex() + seleOffset2_;
209     StuntDouble* sd1A = info_->getIOIndexToIntegrableObject(sd1Aind);
210     r1 = CenterOfMass - sd1A->getPos();
211     } else {
212     r1 = CenterOfMass - sd1->getPos();
213     }
214    
215     if (usePeriodicBoundaryConditions_)
216     currentSnapshot_->wrapVector(r1);
217    
218    
219 gezelter 1979 int sd2Index = sd1->getGlobalIndex() + seleOffset_;
220     sd2 = info_->getIOIndexToIntegrableObject(sd2Index);
221 gezelter 1991
222 gezelter 1979 Vector3d r2 = CenterOfMass - sd2->getPos();
223     if (usePeriodicBoundaryConditions_)
224     currentSnapshot_->wrapVector(r1);
225    
226     Vector3d rc = 0.5*(r1 + r2);
227     if (usePeriodicBoundaryConditions_)
228     currentSnapshot_->wrapVector(rc);
229    
230     Vector3d vec = r1-r2;
231     if (usePeriodicBoundaryConditions_)
232     currentSnapshot_->wrapVector(vec);
233    
234     rc.normalize();
235     vec.normalize();
236     RealType cosangle = dot(rc, vec);
237     int binNo = int(nThetaBins_ * (1.0 + cosangle) / 2.0);
238     count_[binNo]++;
239     }
240     } else {
241    
242     if (evaluator2_.isDynamic()) {
243     seleMan2_.setSelectionSet(evaluator2_.evaluate());
244     }
245    
246     if (seleMan1_.getSelectionCount() != seleMan2_.getSelectionCount() ) {
247     sprintf( painCave.errMsg,
248     "In frame %d, the number of selected StuntDoubles are\n"
249     "\tnot the same in --sele1 and sele2\n", istep);
250     painCave.severity = OPENMD_INFO;
251     painCave.isFatal = 0;
252     simError();
253     }
254    
255     for (sd1 = seleMan1_.beginSelected(ii),
256     sd2 = seleMan2_.beginSelected(jj);
257     sd1 != NULL && sd2 != NULL;
258     sd1 = seleMan1_.nextSelected(ii),
259     sd2 = seleMan2_.nextSelected(jj)) {
260    
261     Vector3d r1 = CenterOfMass - sd1->getPos();
262     if (usePeriodicBoundaryConditions_)
263     currentSnapshot_->wrapVector(r1);
264    
265     Vector3d r2 = CenterOfMass - sd2->getPos();
266     if (usePeriodicBoundaryConditions_)
267     currentSnapshot_->wrapVector(r1);
268    
269     Vector3d rc = 0.5*(r1 + r2);
270     if (usePeriodicBoundaryConditions_)
271     currentSnapshot_->wrapVector(rc);
272    
273     Vector3d vec = r1-r2;
274     if (usePeriodicBoundaryConditions_)
275     currentSnapshot_->wrapVector(vec);
276    
277     rc.normalize();
278     vec.normalize();
279     RealType cosangle = dot(rc, vec);
280     int binNo = int(nThetaBins_ * (1.0 + cosangle) / 2.0);
281     count_[binNo]++;
282    
283     }
284     }
285 gezelter 1413 }
286     }
287 gezelter 1979
288 gezelter 1413 processHistogram();
289     writeProbs();
290    
291     }
292    
293     void pAngle::processHistogram() {
294    
295     int atot = 0;
296 gezelter 1782 for(unsigned int i = 0; i < count_.size(); ++i)
297 gezelter 1413 atot += count_[i];
298    
299 gezelter 1782 for(unsigned int i = 0; i < count_.size(); ++i) {
300     histogram_[i] = double(count_[i] / double(atot));
301 gezelter 1413 }
302     }
303    
304    
305     void pAngle::writeProbs() {
306    
307     std::ofstream rdfStream(outputFilename_.c_str());
308     if (rdfStream.is_open()) {
309     rdfStream << "#pAngle\n";
310     rdfStream << "#nFrames:\t" << nProcessed_ << "\n";
311 gezelter 1979 rdfStream << "#selection1: (" << selectionScript1_ << ")";
312     if (!doVect_) {
313     rdfStream << "\tselection2: (" << selectionScript2_ << ")";
314     }
315     rdfStream << "\n";
316 gezelter 1413 rdfStream << "#cos(theta)\tp(cos(theta))\n";
317 kstocke1 1522 RealType dct = 2.0 / histogram_.size();
318 gezelter 1782 for (unsigned int i = 0; i < histogram_.size(); ++i) {
319 kstocke1 1522 RealType ct = -1.0 + (2.0 * i + 1) / (histogram_.size());
320     rdfStream << ct << "\t" << histogram_[i]/dct << "\n";
321 gezelter 1413 }
322    
323     } else {
324    
325 gezelter 1979 sprintf(painCave.errMsg, "pAngle: unable to open %s\n",
326     outputFilename_.c_str());
327 gezelter 1413 painCave.isFatal = 1;
328     simError();
329     }
330    
331     rdfStream.close();
332     }
333    
334     }
335    

Properties

Name Value
svn:keywords Author Id Revision Date