--- trunk/src/applications/staticProps/BondOrderParameter.cpp 2006/09/21 18:04:52 1044 +++ trunk/src/applications/staticProps/BondOrderParameter.cpp 2006/09/26 01:30:32 1052 @@ -51,14 +51,13 @@ #include "io/DumpReader.hpp" #include "primitives/Molecule.hpp" #include "utils/NumericConstant.hpp" -#include "math/SphericalHarmonic.hpp" namespace oopse { BondOrderParameter::BondOrderParameter(SimInfo* info, const std::string& filename, const std::string& sele, - double rCut, int lNumber, int nbins) : StaticAnalyser(info, filename), selectionScript_(sele), evaluator_(info), seleMan_(info){ + double rCut, int nbins) : StaticAnalyser(info, filename), selectionScript_(sele), evaluator_(info), seleMan_(info){ setOutputName(getPrefix(filename) + ".bo"); @@ -69,25 +68,23 @@ namespace oopse { // Set up cutoff radius and order of the Legendre Polynomial: - lNumber_ = lNumber; rCut_ = rCut; - mSize_ = 2*lNumber_+1; + nBins_ = nbins; + Qcount_.resize(lMax_+1); + Wcount_.resize(lMax_+1); // Q can take values from 0 to 1 MinQ_ = 0.0; - MaxQ_ = 1.0; + MaxQ_ = 1.1; deltaQ_ = (MaxQ_ - MinQ_) / nbins; - Q_histogram_.resize(nbins); // W_6 for icosahedral clusters is 11 / sqrt(4199) = 0.169754, so we'll // use values for MinW_ and MaxW_ that are slightly larger than this: - MinW_ = -0.18; - MaxW_ = 0.18; + MinW_ = -0.25; + MaxW_ = 0.25; deltaW_ = (MaxW_ - MinW_) / nbins; - W_histogram_.resize(nbins); - } BondOrderParameter::~BondOrderParameter() { @@ -96,8 +93,12 @@ namespace oopse { } void BondOrderParameter::initalizeHistogram() { - std::fill(Q_histogram_.begin(), Q_histogram_.end(), 0); - std::fill(W_histogram_.begin(), W_histogram_.end(), 0); + for (int bin = 0; bin < nBins_; bin++) { + for (int l = 0; l <= lMax_; l++) { + Q_histogram_[std::make_pair(bin,l)] = 0; + W_histogram_[std::make_pair(bin,l)] = 0; + } + } } void BondOrderParameter::process() { @@ -114,28 +115,40 @@ namespace oopse { RealType phi; RealType r; RealType dist; - std::map QBar_lm; - RealType QSq_l; - RealType Q_l; - ComplexType W_l; - ComplexType W_l_hat; - int nBonds; + std::map,ComplexType> q; + std::vector q_l; + std::vector q2; + std::vector w; + std::vector w_hat; + std::map,ComplexType> QBar; + std::vector Q2; + std::vector Q; + std::vector W; + std::vector W_hat; + int nBonds, Nbonds; SphericalHarmonic sphericalHarmonic; int i, j; // Make arrays for Wigner3jm - double* THRCOF = new double[mSize_]; + double* THRCOF = new double[2*lMax_+1]; // Variables for Wigner routine - double l_ = (double)lNumber_; - double m1Pass, m2Min, m2Max; - int error, m1, m2, m3; + double lPass, m1Pass, m2Min, m2Max; + int error, m1, m2, m3, mSize; + mSize = 2*lMax_+1; - // Set the l for the spherical harmonic, it doesn't change - sphericalHarmonic.setL(lNumber_); - DumpReader reader(info_, dumpFilename_); int nFrames = reader.getNFrames(); frameCounter_ = 0; + q_l.resize(lMax_+1); + q2.resize(lMax_+1); + w.resize(lMax_+1); + w_hat.resize(lMax_+1); + + Q2.resize(lMax_+1); + Q.resize(lMax_+1); + W.resize(lMax_+1); + W_hat.resize(lMax_+1); + for (int istep = 0; istep < nFrames; istep += step_) { reader.readFrame(istep); frameCounter_++; @@ -153,21 +166,20 @@ namespace oopse { rb = mol->nextRigidBody(rbIter)) { rb->updateAtoms(); } - } - + } + // outer loop is over the selected StuntDoubles: for (sd = seleMan_.beginSelected(i); sd != NULL; sd = seleMan_.nextSelected(i)) { myIndex = sd->getGlobalIndex(); - - // For this central atom, zero out nBonds and QBar_lm - nBonds = 0; - - for (int m = -lNumber_; m <= lNumber_; m++) { - QBar_lm[m] = 0.0; + + for (int l = 0; l <= lMax_; l++) { + for (int m = -l; m <= l; m++) { + q[std::make_pair(l,m)] = 0.0; + } } // inner loop is over all other atoms in the system: @@ -194,10 +206,13 @@ namespace oopse { if (r < rCut_) { costheta = vec.z() / r; phi = atan2(vec.y(), vec.x()); - - for(int m = -lNumber_; m <= lNumber_; m++){ - sphericalHarmonic.setM(m); - QBar_lm[m] += sphericalHarmonic.getValueAt(costheta,phi); + + for (int l = 0; l <= lMax_; l++) { + sphericalHarmonic.setL(l); + for(int m = -l; m <= l; m++){ + sphericalHarmonic.setM(m); + q[std::make_pair(l,m)] += sphericalHarmonic.getValueAt(costheta, phi); + } } nBonds++; } @@ -205,107 +220,177 @@ namespace oopse { } } - // Normalize Qbar2 - for (int m = -lNumber_;m <= lNumber_; m++){ - QBar_lm[m] /= nBonds; - std::cout << "m = " << m << " QBLM = " << QBar_lm[m] << "\n"; + + for (int l = 0; l <= lMax_; l++) { + q_l[l] = 0.0; + for(int m = -l; m <= l; m++) { + q_l[l] += norm(q[std::make_pair(l,m)]); + } + q_l[l] *= 4.0*NumericConstant::PI/(RealType)(2*l + 1); + q_l[l] = sqrt(q_l[l])/(RealType)nBonds; } // Find second order invariant Q_l - - QSq_l = 0.0; - for (int m = -lNumber_; m <= lNumber_; m++){ - QSq_l += norm(QBar_lm[m]); + + for (int l = 0; l <= lMax_; l++) { + q2[l] = 0.0; + for (int m = -l; m <= l; m++){ + q2[l] += norm(q[std::make_pair(l,m)]); + } + q_l[l] = sqrt(q2[l] * 4.0 * NumericConstant::PI / + (RealType)(2*l + 1))/(RealType)nBonds; } - std::cout << "qsq_l = " << QSq_l << "\n"; - Q_l = sqrt(QSq_l * 4.0 * NumericConstant::PI / - (2.0*(RealType)lNumber_ + 1.0)); // Find Third Order Invariant W_l - - W_l = 0.0; - for (int m1 = -lNumber_; m1 <= lNumber_; m1++) { - // Zero work array - for (int ii = 0; ii < mSize_; ii++){ - THRCOF[i] = 0.0; + + for (int l = 0; l <= lMax_; l++) { + w[l] = 0.0; + lPass = (double)l; + for (int m1 = -l; m1 <= l; m1++) { + // Zero work array + for (int ii = 0; ii < 2*l + 1; ii++){ + THRCOF[ii] = 0.0; + } + // Get Wigner coefficients + m1Pass = (double)m1; + + Wigner3jm(&lPass, &lPass, &lPass, + &m1Pass, &m2Min, &m2Max, + THRCOF, &mSize, &error); + + for (int mmm = 0; mmm < (int)(m2Max - m2Min); mmm++) { + m2 = (int)floor(m2Min) + mmm; + m3 = -m1-m2; + w[l] += THRCOF[mmm] * + q[std::make_pair(l,m1)] * + q[std::make_pair(l,m2)] * + q[std::make_pair(l,m3)]; + } } - // Get Wigner coefficients - m1Pass = (double)m1; - Wigner3jm(&l_, &l_, &l_, &m1Pass, &m2Min, &m2Max, THRCOF, &mSize_, &error); - for (int m_index = 1; m_index < (int)(m2Max - m2Min-1.0); m_index++) { - m2 = floor(m2Min) + m_index - 1; - m3 = -m1-m2; - W_l += THRCOF[m_index]*QBar_lm[m1]*QBar_lm[m2]*QBar_lm[m3]; - } + + w_hat[l] = w[l] / pow(q2[l], 1.5); } - - W_l_hat = W_l / pow(QSq_l, 1.5); - - // accumulate histogram data for Q_l and W_l_hat: - std::cout << "Ql = " << Q_l << " Wl = " << W_l_hat << "\n"; - collectHistogram(Q_l, real(W_l_hat)); - + collectHistogram(q_l, w_hat); + + Nbonds += nBonds; + for (int l = 0; l <= lMax_; l++) { + for (int m = -l; m <= l; m++) { + QBar[std::make_pair(l,m)] += q[std::make_pair(l,m)]; + } + } } } - - writeOrderParameter(); + + // Normalize Qbar2 + for (int l = 0; l <= lMax_; l++) { + for (int m = -l; m <= l; m++){ + QBar[std::make_pair(l,m)] /= Nbonds; + } + } + // Find second order invariant Q_l + + for (int l = 0; l <= lMax_; l++) { + Q2[l] = 0.0; + for (int m = -l; m <= l; m++){ + Q2[l] += norm(QBar[std::make_pair(l,m)]); + } + Q[l] = sqrt(Q2[l] * 4.0 * NumericConstant::PI / (RealType)(2*l + 1)); + } + + // Find Third Order Invariant W_l + + for (int l = 0; l <= lMax_; l++) { + W[l] = 0.0; + lPass = (double)l; + for (int m1 = -l; m1 <= l; m1++) { + // Zero work array + for (int ii = 0; ii < 2*l + 1; ii++){ + THRCOF[ii] = 0.0; + } + // Get Wigner coefficients + m1Pass = (double)m1; + + Wigner3jm(&lPass, &lPass, &lPass, + &m1Pass, &m2Min, &m2Max, + THRCOF, &mSize, &error); + + for (int mmm = 0; mmm < (int)(m2Max - m2Min); mmm++) { + m2 = (int)floor(m2Min) + mmm; + m3 = -m1-m2; + W[l] += THRCOF[mmm] * + QBar[std::make_pair(l,m1)] * + QBar[std::make_pair(l,m2)] * + QBar[std::make_pair(l,m3)]; + } + } + + W_hat[l] = W[l] / pow(Q2[l], 1.5); + } + + writeOrderParameter(Q, W_hat); } + void BondOrderParameter::collectHistogram(std::vector q, + std::vector what) { - void BondOrderParameter::collectHistogram(RealType Q_l, RealType W_l_hat) { - - if (Q_l >= MinQ_ && Q_l < MaxQ_) { - int qbin = (Q_l - MinQ_) / deltaQ_; - Q_histogram_[qbin] += 1; - Qcount_++; - sumQ_ += Q_l; - sumQ2_ += Q_l * Q_l; - } else { - sprintf( painCave.errMsg, - "Q_l value outside reasonable range\n"); - painCave.severity = OOPSE_ERROR; - painCave.isFatal = 1; - simError(); + for (int l = 0; l <= lMax_; l++) { + if (q[l] >= MinQ_ && q[l] < MaxQ_) { + int qbin = (q[l] - MinQ_) / deltaQ_; + Q_histogram_[std::make_pair(qbin,l)] += 1; + Qcount_[l]++; + } else { + sprintf( painCave.errMsg, + "q_l value outside reasonable range\n"); + painCave.severity = OOPSE_ERROR; + painCave.isFatal = 1; + simError(); + } } - if (W_l_hat >= MinW_ && W_l_hat < MaxW_) { - int wbin = (W_l_hat - MinW_) / deltaW_; - W_histogram_[wbin] += 1; - Wcount_++; - sumW_ += W_l_hat; - sumW2_ += W_l_hat*W_l_hat; - } else { - sprintf( painCave.errMsg, - "W_l_hat value outside reasonable range\n"); - painCave.severity = OOPSE_ERROR; - painCave.isFatal = 1; - simError(); + for (int l = 0; l <= lMax_; l++) { + if (real(what[l]) >= MinW_ && real(what[l]) < MaxW_) { + int wbin = (real(what[l]) - MinW_) / deltaW_; + W_histogram_[std::make_pair(wbin,l)] += 1; + Wcount_[l]++; + } else { + sprintf( painCave.errMsg, + "Re[w_hat] value outside reasonable range\n"); + painCave.severity = OOPSE_ERROR; + painCave.isFatal = 1; + simError(); + } } + } - void BondOrderParameter::writeOrderParameter() { + void BondOrderParameter::writeOrderParameter(std::vector Q, + std::vector What) { + std::ofstream osq((getOutputFileName() + "q").c_str()); if (osq.is_open()) { - - RealType qAvg = sumQ_ / (RealType) Qcount_; - RealType qStdDev = sumQ2_ / (RealType) Qcount_ - qAvg*qAvg; - osq << "# Bond Order Parameter Q_" << lNumber_ << "\n"; + osq << "# Bond Order Parameters\n"; osq << "# selection: (" << selectionScript_ << ")\n"; - osq << "# : " << qAvg << "\n"; - osq << "# std. dev.: " << qStdDev << "\n"; - + osq << "# \n"; + for (int l = 0; l <= lMax_; l++) { + osq << "# : " << Q[l] << "\n"; + } // Normalize by number of frames and write it out: - for (int i = 0; i < Q_histogram_.size(); ++i) { - RealType Qval = MinQ_ + (i + 0.5) * deltaQ_; - osq << Qval << "\t" << Q_histogram_[i] / frameCounter_ << "\n"; + for (int i = 0; i < nBins_; ++i) { + RealType Qval = MinQ_ + (i + 0.5) * deltaQ_; + osq << Qval; + for (int l = 0; l <= lMax_; l++) { + osq << "\t" << (RealType)Q_histogram_[std::make_pair(i,l)] / (RealType)Qcount_[l]; + } + osq << "\n"; } - + osq.close(); + } else { sprintf(painCave.errMsg, "BondOrderParameter: unable to open %s\n", (getOutputFileName() + "q").c_str()); @@ -316,21 +401,22 @@ namespace oopse { std::ofstream osw((getOutputFileName() + "w").c_str()); if (osw.is_open()) { - - RealType wAvg = sumW_ / (RealType) Wcount_; - RealType wStdDev = sumW2_ / (RealType) Wcount_ - wAvg*wAvg; - - osw << "# Bond Order Parameter W_" << lNumber_ << "\n"; + osw << "# Bond Order Parameters\n"; osw << "# selection: (" << selectionScript_ << ")\n"; - osw << "# : " << wAvg << "\n"; - osw << "# std. dev.: " << wStdDev << "\n"; - + osw << "# \n"; + for (int l = 0; l <= lMax_; l++) { + osw << "# : " << real(What[l]) << "\n"; + } // Normalize by number of frames and write it out: - for (int i = 0; i < W_histogram_.size(); ++i) { - RealType Wval = MinW_ + (i + 0.5) * deltaW_; - osw << Wval << "\t" << W_histogram_[i] / frameCounter_ << "\n"; + for (int i = 0; i < nBins_; ++i) { + RealType Wval = MinW_ + (i + 0.5) * deltaW_; + osw << Wval; + for (int l = 0; l <= lMax_; l++) { + osw << "\t" << (RealType)W_histogram_[std::make_pair(i,l)] / (RealType)Wcount_[l]; + } + osw << "\n"; } - + osw.close(); } else { sprintf(painCave.errMsg, "BondOrderParameter: unable to open %s\n", @@ -338,5 +424,6 @@ namespace oopse { painCave.isFatal = 1; simError(); } + } }