--- trunk/src/applications/staticProps/BondOrderParameter.cpp 2006/06/27 16:36:25 995 +++ trunk/src/applications/staticProps/BondOrderParameter.cpp 2006/09/20 20:13:40 1041 @@ -1,182 +1,333 @@ -/* - * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved. - * - * The University of Notre Dame grants you ("Licensee") a - * non-exclusive, royalty free, license to use, modify and - * redistribute this software in source and binary code form, provided - * that the following conditions are met: - * - * 1. Acknowledgement of the program authors must be made in any - * publication of scientific results based in part on use of the - * program. An acceptable form of acknowledgement is citation of - * the article in which the program was described (Matthew - * A. Meineke, Charles F. Vardeman II, Teng Lin, Christopher - * J. Fennell and J. Daniel Gezelter, "OOPSE: An Object-Oriented - * Parallel Simulation Engine for Molecular Dynamics," - * J. Comput. Chem. 26, pp. 252-271 (2005)) - * - * 2. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * - * 3. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the - * distribution. - * - * This software is provided "AS IS," without a warranty of any - * kind. All express or implied conditions, representations and - * warranties, including any implied warranty of merchantability, - * fitness for a particular purpose or non-infringement, are hereby - * excluded. The University of Notre Dame and its licensors shall not - * be liable for any damages suffered by licensee as a result of - * using, modifying or distributing the software or its - * derivatives. In no event will the University of Notre Dame or its - * licensors be liable for any lost revenue, profit or data, or for - * direct, indirect, special, consequential, incidental or punitive - * damages, however caused and regardless of the theory of liability, - * arising out of the use of or inability to use software, even if the - * University of Notre Dame has been advised of the possibility of - * such damages. - */ - -#include "applications/staticProps/BondOrderParameter.hpp" -#include "utils/simError.h" -#include "io/DumpReader.hpp" -#include "primitives/Molecule.hpp" -#include "utils/NumericConstant.hpp" -namespace oopse { - - -BondOrderParameter::BondOrderParameter(SimInfo* info, const std::string& filename, const std::string& sele1, - const std::string& sele2, double rCut, int lNumber) - : StaticAnalyser(info, filename), - selectionScript1_(sele1), evaluator1_(info), - seleMan1_(info){ - - setOutputName(getPrefix(filename) + ".p2"); - - evaluator1_.loadScriptString(sele1); - evaluator2_.loadScriptString(sele2); - - if (!evaluator1_.isDynamic()) { - seleMan1_.setSelectionSet(evaluator1_.evaluate()); - }else { - sprintf( painCave.errMsg, - "--sele1 must be static selection\n"); - painCave.severity = OOPSE_ERROR; - painCave.isFatal = 1; - simError(); - } - - - int i; - int j; - StuntDouble* sd1; - StuntDouble* sd2; - for (sd1 = seleMan1_.beginSelected(i), sd2 = seleMan1_.beginSelected(j); - sd1 != NULL && sd2 != NULL; - sd1 = seleMan1_.nextSelected(i), sd2 = seleMan2_.nextSelected(j)) { - - sdPairs_.push_back(std::make_pair(sd1, sd2)); - } - - - } - -void BondOrderParameter::process() { - Molecule* mol; - RigidBody* rb; - SimInfo::MoleculeIterator mi; - Molecule::RigidBodyIterator rbIter; - - DumpReader reader(info_, dumpFilename_); - int nFrames = reader.getNFrames(); - - for (int i = 0; i < nFrames; i += step_) { - reader.readFrame(i); - currentSnapshot_ = info_->getSnapshotManager()->getCurrentSnapshot(); - - - for (mol = info_->beginMolecule(mi); mol != NULL; mol = info_->nextMolecule(mi)) { - //change the positions of atoms which belong to the rigidbodies - for (rb = mol->beginRigidBody(rbIter); rb != NULL; rb = mol->nextRigidBody(rbIter)) { - rb->updateAtoms(); - } - - } - - Mat3x3d orderTensor(0.0); - for (std::vector >::iterator j = sdPairs_.begin(); j != sdPairs_.end(); ++j) { - Vector3d vec = j->first->getPos() - j->second->getPos(); - currentSnapshot_->wrapVector(vec); - vec.normalize(); - orderTensor +=outProduct(vec, vec); - } - - orderTensor /= sdPairs_.size(); - orderTensor -= (RealType)(1.0/3.0) * Mat3x3d::identity(); - - Vector3d eigenvalues; - Mat3x3d eigenvectors; - Mat3x3d::diagonalize(orderTensor, eigenvalues, eigenvectors); - - int which; - RealType maxEval = 0.0; - for(int k = 0; k< 3; k++){ - if(fabs(eigenvalues[k]) > maxEval){ - which = k; - maxEval = fabs(eigenvalues[k]); - } - } - RealType p2 = 1.5 * maxEval; - - //the eigen vector is already normalized in SquareMatrix3::diagonalize - Vector3d director = eigenvectors.getColumn(which); - if (director[0] < 0) { - director.negate(); - } - - RealType angle = 0.0; - for (std::vector >::iterator j = sdPairs_.begin(); j != sdPairs_.end(); ++j) { - Vector3d vec = j->first->getPos() - j->second->getPos(); - currentSnapshot_->wrapVector(vec); - vec.normalize(); - - angle += acos(dot(vec, director)) ; - } - angle = angle / (sdPairs_.size() * NumericConstant::PI) * 180.0; - - OrderParam param; - param.p2 = p2; - param.director = director; - param.angle = angle; - - orderParams_.push_back(param); - - } - - writeP2(); - -} - -void BondOrderParameter::writeOrderParameter() { - - std::ofstream os(getOutputFileName().c_str()); - os << "#radial distribution function\n"; - os<< "#selection1: (" << selectionScript1_ << ")\t"; - os << "selection2: (" << selectionScript2_ << ")\n"; - os << "#p2\tdirector_x\tdirector_y\tdiretor_z\tangle(degree)\n"; - - for (std::size_t i = 0; i < orderParams_.size(); ++i) { - os << orderParams_[i].p2 << "\t" - << orderParams_[i].director[0] << "\t" - << orderParams_[i].director[1] << "\t" - << orderParams_[i].director[2] << "\t" - << orderParams_[i].angle << "\n"; - - } - -} - -} - +/* + * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved. + * + * The University of Notre Dame grants you ("Licensee") a + * non-exclusive, royalty free, license to use, modify and + * redistribute this software in source and binary code form, provided + * that the following conditions are met: + * + * 1. Acknowledgement of the program authors must be made in any + * publication of scientific results based in part on use of the + * program. An acceptable form of acknowledgement is citation of + * the article in which the program was described (Matthew + * A. Meineke, Charles F. Vardeman II, Teng Lin, Christopher + * J. Fennell and J. Daniel Gezelter, "OOPSE: An Object-Oriented + * Parallel Simulation Engine for Molecular Dynamics," + * J. Comput. Chem. 26, pp. 252-271 (2005)) + * + * 2. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * 3. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the + * distribution. + * + * This software is provided "AS IS," without a warranty of any + * kind. All express or implied conditions, representations and + * warranties, including any implied warranty of merchantability, + * fitness for a particular purpose or non-infringement, are hereby + * excluded. The University of Notre Dame and its licensors shall not + * be liable for any damages suffered by licensee as a result of + * using, modifying or distributing the software or its + * derivatives. In no event will the University of Notre Dame or its + * licensors be liable for any lost revenue, profit or data, or for + * direct, indirect, special, consequential, incidental or punitive + * damages, however caused and regardless of the theory of liability, + * arising out of the use of or inability to use software, even if the + * University of Notre Dame has been advised of the possibility of + * such damages. + */ + + +/* Creates orientational bond order parameters as outlined by + * Bond-orientaional order in liquids and glasses, Steinhart,Nelson,Ronchetti + * Phys Rev B, 28,784,1983 + * + */ + +#include "applications/staticProps/BondOrderParameter.hpp" +#include "utils/simError.h" +#include "io/DumpReader.hpp" +#include "primitives/Molecule.hpp" +#include "utils/NumericConstant.hpp" +#include "math/RealSphericalHarmonic.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){ + + setOutputName(getPrefix(filename) + ".bo"); + + evaluator_.loadScriptString(sele); + if (!evaluator_.isDynamic()) { + seleMan_.setSelectionSet(evaluator_.evaluate()); + } + + // Set up cutoff radius and order of the Legendre Polynomial: + + lNumber_ = lNumber; + rCut_ = rCut; + mSize_ = 2*lNumber_+1; + + // Q can take values from 0 to 1 + + MinQ_ = 0.0; + MaxQ_ = 1.0; + 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; + deltaW_ = (MaxW_ - MinW_) / nbins; + W_histogram_.resize(nbins); + + } + + BondOrderParameter::~BondOrderParameter() { + Q_histogram_.clear(); + W_histogram_.clear(); + } + + void BondOrderParameter::initalizeHistogram() { + std::fill(Q_histogram_.begin(), Q_histogram_.end(), 0); + std::fill(W_histogram_.begin(), W_histogram_.end(), 0); + } + + void BondOrderParameter::process() { + Molecule* mol; + Atom* atom; + RigidBody* rb; + SimInfo::MoleculeIterator mi; + Molecule::RigidBodyIterator rbIter; + Molecule::AtomIterator ai; + StuntDouble* sd; + RealType theta; + RealType phi; + RealType r; + RealType dist; + std::map QBar_lm; + RealType QSq_l; + RealType Q_l; + int nBonds; + RealSphericalHarmonic sphericalHarmonic; + int i, j; + + // Set the l for the spherical harmonic, it doesn't change + sphericalHarmonic.setL(lNumber_); + + + DumpReader reader(info_, dumpFilename_); + int nFrames = reader.getNFrames(); + frameCounter_ = 0; + + for (int istep = 0; istep < nFrames; istep += step_) { + reader.readFrame(istep); + frameCounter_++; + currentSnapshot_ = info_->getSnapshotManager()->getCurrentSnapshot(); + + if (evaluator_.isDynamic()) { + seleMan_.setSelectionSet(evaluator_.evaluate()); + } + + // update the positions of atoms which belong to the rigidbodies + + for (mol = info_->beginMolecule(mi); mol != NULL; + mol = info_->nextMolecule(mi)) { + for (rb = mol->beginRigidBody(rbIter); rb != NULL; + rb = mol->nextRigidBody(rbIter)) { + rb->updateAtoms(); + } + } + + // outer loop is over the selected StuntDoubles: + + for (sd = seleMan_.beginSelected(i); sd != NULL; + sd = seleMan_.nextSelected(i)) { + + // For this central atom, zero out nBonds and QBar_lm + + nBonds = 0; + + for (int m = -lNumber_; m <= lNumber_; m++) { + QBar_lm[m] = 0.0; + } + + // inner loop is over all other atoms in the system: + + for (mol = info_->beginMolecule(mi); mol != NULL; + mol = info_->nextMolecule(mi)) { + for (atom = mol->beginAtom(ai); atom != NULL; + atom = mol->nextAtom(ai)) { + + + Vector3d vec = sd->getPos() - atom->getPos(); + currentSnapshot_->wrapVector(vec); + + // Calculate "bonds" and build Q_lm(r) where + // Q_lm = Y_lm(theta(r),phi(r)) + // The spherical harmonics are wrt any arbitrary coordinate + // system, we choose standard spherical coordinates + + r = sqrt(pow(vec.x(),2)+pow(vec.y(),2)+pow(vec.z(),2)); + + // Check to see if neighbor is in bond cutoff + + if (r < rCut_) { + theta = atan2(vec.y(), vec.x()); + phi = acos(vec.z()/r); + for(int m = -lNumber_; m <= lNumber_; m++){ + sphericalHarmonic.setM(m); + QBar_lm[m] += sphericalHarmonic.getValueAt(theta,phi); + } + nBonds++; + } + } + } + + // Normalize Qbar + for (int m = -lNumber_;m <= lNumber_; m++){ + QBar_lm[m] /= nBonds; + } + + // Find second order invariant Q_l + + QSq_l = 0.0; + for (int m = -lNumber_; m <= lNumber_; m++){ + QSq_l += pow(QBar_lm[m], 2); + } + Q_l = sqrt(QSq_l*(4.0 * NumericConstant::PI / (2.0*(RealType)lNumber_ + 1))); + + // Find Third Order Invariant W_l + + // Make arrays for Wigner3jm + double* THRCOF = new double[mSize_]; + // Variables for Wigner routine + double l_ = (double)lNumber_; + double m1Pass, m2Min, m2Max; + int error, m1, m2, m3; + + RealType W_l; + RealType W_l_hat; + 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; + } + // Get Wigner coefficients + m1Pass = (double)m1; + Wigner3jm(&l_, &l_, &l_, &m1Pass, &m2Min, &m2Max, THRCOF, &mSize_, &error); + for (int m_index = 1; i < (int)(m2Max - m2Min-1.0); m_index++) { + m2 = floor(m2Min) + m_index - 1; + m3 = -m1-m2; + W_l += THRCOF[m_index]*QBar_lm[m1+lNumber_]*QBar_lm[m2+lNumber_]*QBar_lm[m3+lNumber_]; + } + } + + W_l_hat = W_l / pow(QSq_l, 1.5); + + // accumulate histogram data for Q_l and W_l_hat: + + collectHistogram(Q_l, W_l_hat); + + } + } + + writeOrderParameter(); + + } + + + 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(); + } + + 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(); + } + } + + void BondOrderParameter::writeOrderParameter() { + + 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 << "# selection: (" << selectionScript_ << ")\n"; + osq << "# : " << qAvg << "\n"; + osq << "# std. dev.: " << qStdDev << "\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"; + } + + osq.close(); + } else { + sprintf(painCave.errMsg, "BondOrderParameter: unable to open %s\n", + (getOutputFileName() + "q").c_str()); + painCave.isFatal = 1; + simError(); + } + + 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 << "# selection: (" << selectionScript_ << ")\n"; + osw << "# : " << wAvg << "\n"; + osw << "# std. dev.: " << wStdDev << "\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"; + } + + osw.close(); + } else { + sprintf(painCave.errMsg, "BondOrderParameter: unable to open %s\n", + (getOutputFileName() + "w").c_str()); + painCave.isFatal = 1; + simError(); + } + } +}