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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines