ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/group/trunk/OOPSE-4/src/applications/staticProps/BondOrderParameter.cpp
(Generate patch)

Comparing trunk/OOPSE-4/src/applications/staticProps/BondOrderParameter.cpp (file contents):
Revision 3005 by chuckv, Mon Sep 18 21:31:23 2006 UTC vs.
Revision 3012 by gezelter, Thu Sep 21 18:04:52 2006 UTC

# Line 45 | Line 45
45   *     Phys Rev B, 28,784,1983
46   *
47   */
48 <
48 >
49   #include "applications/staticProps/BondOrderParameter.hpp"
50   #include "utils/simError.h"
51   #include "io/DumpReader.hpp"
52   #include "primitives/Molecule.hpp"
53   #include "utils/NumericConstant.hpp"
54 < #include "math/RealSphericalHarmonic.hpp"
54 > #include "math/SphericalHarmonic.hpp"
55 >
56   namespace oopse {
57  
58 +  BondOrderParameter::BondOrderParameter(SimInfo* info,
59 +                                         const std::string& filename,
60 +                                         const std::string& sele,
61 +                                         double rCut, int lNumber, int nbins) : StaticAnalyser(info, filename), selectionScript_(sele), evaluator_(info), seleMan_(info){
62 +    
63 +    setOutputName(getPrefix(filename) + ".bo");
64  
65 <  BondOrderParameter::BondOrderParameter(SimInfo* info, const std::string& filename, const std::string& sele1,
66 <                                         const std::string& sele2, double rCut, int lNumber)
67 <    : StaticAnalyser(info, filename),
61 <      selectionScript1_(sele1), evaluator1_(info),
62 <      seleMan1_(info){
63 <
64 <    setOutputName(getPrefix(filename) + ".obo");
65 <
66 <    evaluator1_.loadScriptString(sele1);
67 <    evaluator2_.loadScriptString(sele2);
68 <
69 <    if (!evaluator1_.isDynamic()) {
70 <      seleMan1_.setSelectionSet(evaluator1_.evaluate());
71 <    }else {
72 <      sprintf( painCave.errMsg,
73 <               "--sele1 must be static selection\n");
74 <      painCave.severity = OOPSE_ERROR;
75 <      painCave.isFatal = 1;
76 <      simError();
65 >    evaluator_.loadScriptString(sele);
66 >    if (!evaluator_.isDynamic()) {
67 >      seleMan_.setSelectionSet(evaluator_.evaluate());
68      }
69  
70 <    /* Set up cutoff radius and type of order parameter we are calcuating*/
70 >    // Set up cutoff radius and order of the Legendre Polynomial:
71 >
72      lNumber_ = lNumber;
73      rCut_ = rCut;
74 <    mSize_ = 2*lNumber_+1;
74 >    mSize_ = 2*lNumber_+1;    
75  
76 <    int i;
85 <    int j;
86 <    StuntDouble* sd1;
87 <    StuntDouble* sd2;
88 <    for (sd1 = seleMan1_.beginSelected(i), sd2 = seleMan1_.beginSelected(j);
89 <         sd1 != NULL && sd2 != NULL;
90 <         sd1 = seleMan1_.nextSelected(i), sd2 = seleMan2_.nextSelected(j)) {
91 <      for (sd2 = seleMan1_.beginSelected(j),sd2
92 <             sdPairs_.push_back(std::make_pair(sd1, sd2));
93 <           }
76 >    // Q can take values from 0 to 1
77  
78 +    MinQ_ = 0.0;
79 +    MaxQ_ = 1.0;
80 +    deltaQ_ = (MaxQ_ - MinQ_) / nbins;
81 +    Q_histogram_.resize(nbins);
82  
83 <    }
83 >    // W_6 for icosahedral clusters is 11 / sqrt(4199) = 0.169754, so we'll
84 >    // use values for MinW_ and MaxW_ that are slightly larger than this:
85  
86 <    void BondOrderParameter::process
87 <      () {
88 <      Molecule* mol;
89 <      RigidBody* rb;
102 <      SimInfo::MoleculeIterator mi;
103 <      Molecule::RigidBodyIterator rbIter;
104 <      RealType theta;
105 <      RealType phi;
106 <      RealType r;
107 <      RealType dist;
108 <      RealType* QBar_lm;
109 <      RealType QSq_l;
110 <      int nBonds;
111 <      int m, m_index;
112 <      RealSphericalHarmonic sphericalHarmonic;
86 >    MinW_ = -0.18;
87 >    MaxW_ = 0.18;
88 >    deltaW_ = (MaxW_ - MinW_) / nbins;
89 >    W_histogram_.resize(nbins);
90  
91 +  }
92  
93 <      DumpReader reader(info_, dumpFilename_);
94 <      int nFrames = reader.getNFrames();
93 >  BondOrderParameter::~BondOrderParameter() {
94 >    Q_histogram_.clear();
95 >    W_histogram_.clear();
96 >  }
97  
98 <      /*Set the l for the spherical harmonic, it doesn't change*/
99 <      sphericalHarmonic.setL(lNumber_);
98 >  void BondOrderParameter::initalizeHistogram() {
99 >    std::fill(Q_histogram_.begin(), Q_histogram_.end(), 0);
100 >    std::fill(W_histogram_.begin(), W_histogram_.end(), 0);
101 >  }
102  
103 <      for (int i = 0; i < nFrames; i += step_) {
104 <        reader.readFrame(i);
105 <        currentSnapshot_ = info_->getSnapshotManager()->getCurrentSnapshot();
106 <        nBonds = 0;
103 >  void BondOrderParameter::process() {
104 >    Molecule* mol;
105 >    Atom* atom;
106 >    RigidBody* rb;
107 >    int myIndex;
108 >    SimInfo::MoleculeIterator mi;
109 >    Molecule::RigidBodyIterator rbIter;
110 >    Molecule::AtomIterator ai;
111 >    StuntDouble* sd;
112 >    Vector3d vec;
113 >    RealType costheta;
114 >    RealType phi;
115 >    RealType r;
116 >    RealType dist;
117 >    std::map<int, ComplexType> QBar_lm;
118 >    RealType QSq_l;
119 >    RealType Q_l;
120 >    ComplexType W_l;
121 >    ComplexType W_l_hat;
122 >    int nBonds;
123 >    SphericalHarmonic sphericalHarmonic;
124 >    int i, j;
125 >    // Make arrays for Wigner3jm
126 >    double* THRCOF = new double[mSize_];
127 >    // Variables for Wigner routine
128 >    double l_ = (double)lNumber_;
129 >    double m1Pass, m2Min, m2Max;
130 >    int error, m1, m2, m3;
131  
132 <        for (mol = info_->beginMolecule(mi); mol != NULL; mol = info_->nextMolecule(mi)) {
133 <          //change the positions of atoms which belong to the rigidbodies
128 <          for (rb = mol->beginRigidBody(rbIter); rb != NULL; rb = mol->nextRigidBody(rbIter)) {
129 <            rb->updateAtoms();
130 <          }
132 >    // Set the l for the spherical harmonic, it doesn't change
133 >    sphericalHarmonic.setL(lNumber_);
134  
135 <        }
135 >    DumpReader reader(info_, dumpFilename_);    
136 >    int nFrames = reader.getNFrames();
137 >    frameCounter_ = 0;
138  
139 +    for (int istep = 0; istep < nFrames; istep += step_) {
140 +      reader.readFrame(istep);
141 +      frameCounter_++;
142 +      currentSnapshot_ = info_->getSnapshotManager()->getCurrentSnapshot();
143 +      
144 +      if (evaluator_.isDynamic()) {
145 +        seleMan_.setSelectionSet(evaluator_.evaluate());
146 +      }
147  
148 <        /* Setup QBar */
136 <        QBar_lm = new double[mSize_];
148 >      // update the positions of atoms which belong to the rigidbodies
149  
150 <        /* Calculate "bonds" and build Q_lm(r) where Q_lm = Y_lm(theta(r),phi(r)) */
151 <        for (std::vector<std::pair<StuntDouble*, StuntDouble*> >::iterator j = sdPairs_.begin(); j != sdPairs_.end(); ++j) {
152 <          Vector3d vec = j->first->getPos() - j->second->getPos();
153 <          currentSnapshot_->wrapVector(vec);
154 <          /* The spherical harmonics are wrt any arbitray coordiate sysetm,
155 <           * we choose standard spherical coordinates */
156 <          r = sqrt(pow(vec.x(),2)+pow(vec.y(),2)+pow(vec.z(),2));
150 >      for (mol = info_->beginMolecule(mi); mol != NULL;
151 >           mol = info_->nextMolecule(mi)) {
152 >        for (rb = mol->beginRigidBody(rbIter); rb != NULL;
153 >             rb = mol->nextRigidBody(rbIter)) {
154 >          rb->updateAtoms();
155 >        }        
156 >      }      
157 >      
158 >      // outer loop is over the selected StuntDoubles:
159  
160 <          /* Check to see if neighbor is in bond cuttoff*/
161 <          if (r<rCut_){
148 <            theta = atan(vec.y()/vec.x());
149 <            phi = acos(vec.z()/r);
150 <            for(int m_index = 0; m_index < mSize_; m_index++){
151 <              sphericalHarmonic.setM(m_index-lNumber_);
152 <              QBar_lm(m_index) += sphericalHarmonic.getValueAt(theta,phi);
153 <            }
154 <            nBonds++;
155 <          }
156 <        }
160 >      for (sd = seleMan_.beginSelected(i); sd != NULL;
161 >           sd = seleMan_.nextSelected(i)) {
162  
163 <        /*Normalize Qbar by number of Bonds*/
159 <        for ( int m_index = 0;m_index < mSize_; m_index++){
160 <          QBar_lm(m_index) = QBar_lm(m_index)/nBonds;
161 <        }
163 >        myIndex = sd->getGlobalIndex();
164  
165 +        // For this central atom, zero out nBonds and QBar_lm
166  
167 <      }
167 >        nBonds = 0;
168 >      
169 >        for (int m = -lNumber_; m <= lNumber_; m++) {
170 >          QBar_lm[m] = 0.0;
171 >        }
172 >        
173 >        // inner loop is over all other atoms in the system:
174 >        
175 >        for (mol = info_->beginMolecule(mi); mol != NULL;
176 >             mol = info_->nextMolecule(mi)) {
177 >          for (atom = mol->beginAtom(ai); atom != NULL;
178 >               atom = mol->nextAtom(ai)) {
179  
180 <      /*Normalize by number of frames*/
167 <      for ( int m_index = 0;m_index < mSize_; m_index++){
168 <        QBar_lm(m_index) = QBar_lm(m_index)/nFrames;
169 <      }
180 >            if (atom->getGlobalIndex() != myIndex) {
181  
182 +              vec = sd->getPos() - atom->getPos();      
183 +              currentSnapshot_->wrapVector(vec);
184 +              
185 +              // Calculate "bonds" and build Q_lm(r) where
186 +              //      Q_lm = Y_lm(theta(r),phi(r))                
187 +              // The spherical harmonics are wrt any arbitrary coordinate
188 +              // system, we choose standard spherical coordinates
189 +              
190 +              r = vec.length();
191 +              
192 +              // Check to see if neighbor is in bond cutoff
193 +              
194 +              if (r < rCut_) {
195 +                costheta = vec.z() / r;
196 +                phi = atan2(vec.y(), vec.x());
197 +                
198 +                for(int m = -lNumber_; m <= lNumber_; m++){
199 +                  sphericalHarmonic.setM(m);
200 +                  QBar_lm[m] += sphericalHarmonic.getValueAt(costheta,phi);
201 +                }
202 +                nBonds++;
203 +              }  
204 +            }
205 +          }
206 +        }
207 +        
208 +        // Normalize Qbar2
209 +        for (int m = -lNumber_;m <= lNumber_; m++){
210 +          QBar_lm[m] /= nBonds;
211 +          std::cout << "m = " << m << " QBLM = " << QBar_lm[m] << "\n";
212 +        }
213  
214 +        // Find second order invariant Q_l
215  
216 <      /* Find second order invariant Q_l*/
216 >        QSq_l = 0.0;
217 >        for (int m = -lNumber_; m <= lNumber_; m++){
218 >          QSq_l += norm(QBar_lm[m]);
219 >        }
220 >        std::cout << "qsq_l = " << QSq_l << "\n";
221 >        Q_l = sqrt(QSq_l * 4.0 * NumericConstant::PI /
222 >                   (2.0*(RealType)lNumber_ + 1.0));
223  
224 <      for (int m_index = 0 ;m_index <= sizeM_; m++){
225 <        QSq_l += pow(QBar_lm(m),2);
226 <      }
227 <      Q_l_ = sqrt((4*NumericConstant::PI/lNumber_+1)*QSq_l);
224 >        // Find Third Order Invariant W_l
225 >        
226 >        W_l = 0.0;
227 >        for (int m1 = -lNumber_; m1 <= lNumber_; m1++) {
228 >          // Zero work array
229 >          for (int ii = 0; ii < mSize_; ii++){
230 >            THRCOF[i] = 0.0;
231 >          }
232 >          // Get Wigner coefficients
233 >          m1Pass = (double)m1;
234 >          Wigner3jm(&l_, &l_, &l_, &m1Pass, &m2Min, &m2Max, THRCOF, &mSize_, &error);
235 >          for (int m_index = 1; m_index < (int)(m2Max - m2Min-1.0); m_index++) {
236 >            m2 = floor(m2Min) + m_index - 1;
237 >            m3 = -m1-m2;
238 >            W_l += THRCOF[m_index]*QBar_lm[m1]*QBar_lm[m2]*QBar_lm[m3];
239 >          }
240 >        }
241 >        
242 >        W_l_hat = W_l / pow(QSq_l, 1.5);
243 >        
244 >        // accumulate histogram data for Q_l and W_l_hat:
245  
246 <      /* Find Third Order Invariant W_l*/
247 <
248 <      /* Make arrays for Wigner3jm */
183 <      double* THRCOF = new double[mSize_];
184 <      /* Variables for Wigner routine */
185 <      double l_ = (double)lNumber_;
186 <      double m2Min;
187 <      double m2Max;
188 <      int error;
189 <      int m1;
190 <      int m2;
191 <      int m3;
192 <
193 <      for (int m1 = -lNumber_;m <= lNumber_;m1++){
194 <        /* Zero work array */
195 <        for (i=0; i<mSize_;i++){
196 <          THRCOF[i] = 0.0;      
197 <        }
198 <        /* Get wigner coefficients */
199 <        Wigner3jm(&l_,&l_,&l_,&(double)m1,&m2Min,&m2Max&,THRCOF,&mSize_,&error);
200 <        for (m_index=1; i<(m2Max-M2Min-1.0);m_index++){
201 <          m2 = floor(m2Min) + m_index - 1;
202 <          m3 = -m1-m2;
203 <          W_l_ += THRCOF(m_index)*QBar_lm(m1+lNumber_)*QBar_lm(m2+lNumber_)*QBar_lm(m3+lNumber_);
204 <        }
246 >        std::cout << "Ql = " << Q_l << " Wl = " << W_l_hat << "\n";
247 >        collectHistogram(Q_l, real(W_l_hat));
248 >                
249        }
250 +    }
251 +    
252 +    writeOrderParameter();
253 +    
254 +  }
255  
256  
257 <      writeOrderParameter();
257 >  void BondOrderParameter::collectHistogram(RealType Q_l, RealType W_l_hat) {
258  
259 +    if (Q_l >= MinQ_ && Q_l < MaxQ_) {
260 +      int qbin = (Q_l - MinQ_) / deltaQ_;
261 +      Q_histogram_[qbin] += 1;
262 +      Qcount_++;
263 +      sumQ_ += Q_l;
264 +      sumQ2_ += Q_l * Q_l;
265 +    } else {
266 +      sprintf( painCave.errMsg,
267 +               "Q_l value outside reasonable range\n");
268 +      painCave.severity = OOPSE_ERROR;
269 +      painCave.isFatal = 1;
270 +      simError();  
271      }
272  
273 +    if (W_l_hat >= MinW_ && W_l_hat < MaxW_) {
274 +      int wbin = (W_l_hat - MinW_) / deltaW_;
275 +      W_histogram_[wbin] += 1;
276 +      Wcount_++;
277 +      sumW_  += W_l_hat;
278 +      sumW2_ += W_l_hat*W_l_hat;
279 +    } else {
280 +      sprintf( painCave.errMsg,
281 +               "W_l_hat value outside reasonable range\n");
282 +      painCave.severity = OOPSE_ERROR;
283 +      painCave.isFatal = 1;
284 +      simError();  
285 +    }
286 +  }  
287  
288 <    void BondOrderParameter::writeOrderParameter() {
288 >  void BondOrderParameter::writeOrderParameter() {
289  
290 <      std::ofstream os(getOutputFileName().c_str());
216 <      os << "#radial distribution function\n";
217 <      os<< "#selection1: (" << selectionScript1_ << ")\t";
218 <      os << "selection2: (" << selectionScript2_ << ")\n";
219 <      os << "#p2\tdirector_x\tdirector_y\tdiretor_z\tangle(degree)\n";
290 >    std::ofstream osq((getOutputFileName() + "q").c_str());
291  
292 <      for (std::size_t i = 0; i < orderParams_.size(); ++i) {
222 <        os <<  orderParams_[i].p2 << "\t"
223 <           <<  orderParams_[i].director[0] << "\t"
224 <           <<  orderParams_[i].director[1] << "\t"
225 <           <<  orderParams_[i].director[2] << "\t"
226 <           <<  orderParams_[i].angle << "\n";
292 >    if (osq.is_open()) {
293  
294 +      RealType qAvg = sumQ_ / (RealType) Qcount_;
295 +      RealType qStdDev = sumQ2_ / (RealType) Qcount_ - qAvg*qAvg;
296 +      
297 +      osq << "# Bond Order Parameter Q_" << lNumber_ << "\n";
298 +      osq << "# selection: (" << selectionScript_ << ")\n";
299 +      osq << "# <Q_" << lNumber_ << ">: " << qAvg << "\n";
300 +      osq << "# std. dev.: " << qStdDev << "\n";
301 +      
302 +      // Normalize by number of frames and write it out:
303 +      for (int i = 0; i < Q_histogram_.size(); ++i) {
304 +        RealType Qval = MinQ_ + (i + 0.5) * deltaQ_;
305 +        osq << Qval << "\t" << Q_histogram_[i] / frameCounter_ << "\n";
306        }
307 <
307 >      
308 >      osq.close();
309 >    } else {
310 >      sprintf(painCave.errMsg, "BondOrderParameter: unable to open %s\n",
311 >              (getOutputFileName() + "q").c_str());
312 >      painCave.isFatal = 1;
313 >      simError();  
314      }
315  
316 <  }
316 >    std::ofstream osw((getOutputFileName() + "w").c_str());
317  
318 +    if (osw.is_open()) {
319 +
320 +      RealType wAvg = sumW_ / (RealType) Wcount_;
321 +      RealType wStdDev = sumW2_ / (RealType) Wcount_ - wAvg*wAvg;
322 +      
323 +      osw << "# Bond Order Parameter W_" << lNumber_ << "\n";
324 +      osw << "# selection: (" << selectionScript_ << ")\n";
325 +      osw << "# <W_" << lNumber_ << ">: " << wAvg << "\n";
326 +      osw << "# std. dev.: " << wStdDev << "\n";
327 +      
328 +      // Normalize by number of frames and write it out:
329 +      for (int i = 0; i < W_histogram_.size(); ++i) {
330 +        RealType Wval = MinW_ + (i + 0.5) * deltaW_;
331 +        osw << Wval << "\t" << W_histogram_[i] / frameCounter_ << "\n";
332 +      }
333 +      
334 +      osw.close();
335 +    } else {
336 +      sprintf(painCave.errMsg, "BondOrderParameter: unable to open %s\n",
337 +              (getOutputFileName() + "w").c_str());
338 +      painCave.isFatal = 1;
339 +      simError();  
340 +    }
341 +  }
342 + }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines