[ViewVC] Diff of: OpenMD/trunk/src/selection/SelectionEvaluator.cpp

Comparing trunk/src/selection/SelectionEvaluator.cpp (file contents):
Revision 277 by tim, Tue Feb 1 06:55:00 2005 UTC vs.
Revision 1364 by cli2, Wed Oct 7 20:49:50 2009 UTC

+ * such damages.
+ */
-+
+#include <stack>
+#include "selection/SelectionEvaluator.hpp"
-+
+#include "primitives/Atom.hpp"
-+
+#include "primitives/DirectionalAtom.hpp"
-+
+#include "primitives/RigidBody.hpp"
-+
+#include "primitives/Molecule.hpp"
-+
+#include "io/basic_ifstrstream.hpp"
-+
+namespace oopse {
-+
-+
-+
+  SelectionEvaluator::SelectionEvaluator(SimInfo* si)
-+
+    : info(si), nameFinder(info), distanceFinder(info), indexFinder(info),
-+
+      isLoaded_(false){
-+
+      nStuntDouble = info->getNGlobalAtoms() + info->getNGlobalRigidBodies();
-+
+    }
-+
-+
+  bool SelectionEvaluator::loadScript(const std::string& filename,
-+
+                                      const std::string& script) {
-+
+    clearDefinitionsAndLoadPredefined();
-+
+    this->filename = filename;
-+
+    this->script = script;
-+
+    if (! compiler.compile(filename, script)) {
-+
+      error = true;
-+
+      errorMessage = compiler.getErrorMessage();
-+
-+
+      sprintf( painCave.errMsg,
-+
+               "SelectionCompiler Error: %s\n", errorMessage.c_str());
-+
+      painCave.severity = OOPSE_ERROR;
-+
+      painCave.isFatal = 1;
-+
+      simError();
-+
+      return false;
-+
+    }
-+
-+
+    pc = 0;
-+
+    aatoken = compiler.getAatokenCompiled();
-+
+    linenumbers = compiler.getLineNumbers();
-+
+    lineIndices = compiler.getLineIndices();
-+
-+
+    std::vector<std::vector<Token> >::const_iterator i;
-+
-+
+    isDynamic_ = false;
-+
+    for (i = aatoken.begin(); i != aatoken.end(); ++i) {
-+
+      if (containDynamicToken(*i)) {
-+
+        isDynamic_ = true;
-+
+        break;
-+
+      }
-+
+    }
-+
-+
+    isLoaded_ = true;
-+
+    return true;
-+
+  }
-+
-+
+  void SelectionEvaluator::clearState() {
-+
+    error = false;
-+
+    errorMessage = "";
-+
+  }
-+
-+
+  bool SelectionEvaluator::loadScriptString(const std::string& script) {
-+
+    clearState();
-+
+    return loadScript("", script);
-+
+  }
-+
-+
+  bool SelectionEvaluator::loadScriptFile(const std::string& filename) {
-+
+    clearState();
-+
+    return loadScriptFileInternal(filename);
-+
+  }
-+
-+
+  bool SelectionEvaluator::loadScriptFileInternal(const std::string & filename) {
-+
+    ifstrstream ifs(filename.c_str());
-+
+    if (!ifs.is_open()) {
-+
+      return false;
-+
+    }
-+
-+
+    const int bufferSize = 65535;
-+
+    char buffer[bufferSize];
-+
+    std::string script;
-+
+    while(ifs.getline(buffer, bufferSize)) {
-+
+      script += buffer;
-+
+    }
-+
+    return loadScript(filename, script);
-+
+  }
-+
-+
+  void SelectionEvaluator::instructionDispatchLoop(OOPSEBitSet& bs){
-+
-+
+    while ( pc < aatoken.size()) {
-+
+      statement = aatoken[pc++];
-+
+      statementLength = statement.size();
-+
+      Token token = statement[0];
-+
+      switch (token.tok) {
-+
+      case Token::define:
-+
+        define();
-+
+        break;
-+
+      case Token::select:
-+
+        select(bs);
-+
+        break;
-+
+      default:
-+
+        unrecognizedCommand(token);
-+
+        return;
-+
+      }
-+
+    }
-+
-+
+  }
-+
-+
+  OOPSEBitSet SelectionEvaluator::expression(const std::vector<Token>& code,
-+
+                                             int pcStart) {
-+
+    OOPSEBitSet bs;
-+
+    std::stack<OOPSEBitSet> stack;
-+
-+
+    for (int pc = pcStart; pc < code.size(); ++pc) {
-+
+      Token instruction = code[pc];
-+
-+
+      switch (instruction.tok) {
-+
+      case Token::expressionBegin:
-+
+        break;
-+
+      case Token::expressionEnd:
-+
+        break;
-+
+      case Token::all:
-+
+        bs = OOPSEBitSet(nStuntDouble);
-+
+        bs.setAll();
-+
+        stack.push(bs);
-+
+        break;
-+
+      case Token::none:
-+
+        bs = OOPSEBitSet(nStuntDouble);
-+
+        stack.push(bs);
-+
+        break;
-+
+      case Token::opOr:
-+
+        bs = stack.top();
-+
+        stack.pop();
-+
+        stack.top() |= bs;
-+
+        break;
-+
+      case Token::opAnd:
-+
+        bs = stack.top();
-+
+        stack.pop();
-+
+        stack.top() &= bs;
-+
+        break;
-+
+      case Token::opNot:
-+
+        stack.top().flip();
-+
+        break;
-+
+      case Token::within:
-+
+        withinInstruction(instruction, stack.top());
-+
+        break;
-+
+        //case Token::selected:
-+
+        //  stack.push(getSelectionSet());
-+
+        //  break;
-+
+      case Token::name:
-+
+        stack.push(nameInstruction(boost::any_cast<std::string>(instruction.value)));
-+
+        break;
-+
+      case Token::index:
-+
+        stack.push(indexInstruction(instruction.value));
-+
+        break;
-+
+      case Token::identifier:
-+
+        stack.push(lookupValue(boost::any_cast<std::string>(instruction.value)));
-+
+        break;
-+
+      case Token::opLT:
-+
+      case Token::opLE:
-+
+      case Token::opGE:
-+
+      case Token::opGT:
-+
+      case Token::opEQ:
-+
+      case Token::opNE:
-+
+        stack.push(comparatorInstruction(instruction));
-+
+        break;
-+
+      default:
-+
+        unrecognizedExpression();
-+
+      }
-+
+    }
-+
+    if (stack.size() != 1)
-+
+      evalError("atom expression compiler error - stack over/underflow");
-+
-+
+    return stack.top();
-+
+  }
-+
-+
-+
-+
+  OOPSEBitSet SelectionEvaluator::comparatorInstruction(const Token& instruction) {
-+
+    int comparator = instruction.tok;
-+
+    int property = instruction.intValue;
-+
+    float comparisonValue = boost::any_cast<float>(instruction.value);
-+
+    float propertyValue;
-+
+    OOPSEBitSet bs(nStuntDouble);
-+
+    bs.clearAll();
-+
-+
+    SimInfo::MoleculeIterator mi;
-+
+    Molecule* mol;
-+
+    Molecule::AtomIterator ai;
-+
+    Atom* atom;
-+
+    Molecule::RigidBodyIterator rbIter;
-+
+    RigidBody* rb;
-+
-+
+    for (mol = info->beginMolecule(mi); mol != NULL;
-+
+         mol = info->nextMolecule(mi)) {
-+
-+
+      for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) {
-+
+        compareProperty(atom, bs, property, comparator, comparisonValue);
-+
+      }
-+
-+
+      for (rb = mol->beginRigidBody(rbIter); rb != NULL;
-+
+           rb = mol->nextRigidBody(rbIter)) {
-+
+        compareProperty(rb, bs, property, comparator, comparisonValue);
-+
+      }
-+
+    }
-+
-+
+    return bs;
-+
+  }
-+
-+
+  void SelectionEvaluator::compareProperty(StuntDouble* sd, OOPSEBitSet& bs,
-+
+                                           int property, int comparator,
-+
+                                           float comparisonValue) {
-+
+    RealType propertyValue = 0.0;
-+
+    switch (property) {
-+
+    case Token::mass:
-+
+      propertyValue = sd->getMass();
-+
+      break;
-+
+    case Token::charge:
-+
+      if (sd->isAtom()){
-+
+        Atom* atom = static_cast<Atom*>(sd);
-+
+        propertyValue = getCharge(atom);
-+
+      } else if (sd->isRigidBody()) {
-+
+        RigidBody* rb = static_cast<RigidBody*>(sd);
-+
+        RigidBody::AtomIterator ai;
-+
+        Atom* atom;
-+
+        for (atom = rb->beginAtom(ai); atom != NULL; atom = rb->nextAtom(ai)) {
-+
+          propertyValue+=  getCharge(atom);
-+
+        }
-+
+      }
-+
+      break;
-+
+    case Token::x:
-+
+      propertyValue = sd->getPos().x();
-+
+      break;
-+
+    case Token::y:
-+
+      propertyValue = sd->getPos().y();
-+
+      break;
-+
+    case Token::z:
-+
+      propertyValue = sd->getPos().z();
-+
+      break;
-+
+    default:
-+
+      unrecognizedAtomProperty(property);
-+
+    }
-+
-+
+    bool match = false;
-+
+    switch (comparator) {
-+
+    case Token::opLT:
-+
+      match = propertyValue < comparisonValue;
-+
+      break;
-+
+    case Token::opLE:
-+
+      match = propertyValue <= comparisonValue;
-+
+      break;
-+
+    case Token::opGE:
-+
+      match = propertyValue >= comparisonValue;
-+
+      break;
-+
+    case Token::opGT:
-+
+      match = propertyValue > comparisonValue;
-+
+      break;
-+
+    case Token::opEQ:
-+
+      match = propertyValue == comparisonValue;
-+
+      break;
-+
+    case Token::opNE:
-+
+      match = propertyValue != comparisonValue;
-+
+      break;
-+
+    }
-+
+    if (match)
-+
+      bs.setBitOn(sd->getGlobalIndex());
-+
-+
+  }
-+
+  void SelectionEvaluator::withinInstruction(const Token& instruction,
-+
+                                             OOPSEBitSet& bs){
-+
-+
+    boost::any withinSpec = instruction.value;
-+
+    float distance;
-+
+    if (withinSpec.type() == typeid(float)){
-+
+      distance = boost::any_cast<float>(withinSpec);
-+
+    } else if (withinSpec.type() == typeid(int)) {
-+
+      distance = boost::any_cast<int>(withinSpec);
-+
+    } else {
-+
+      evalError("casting error in withinInstruction");
-+
+      bs.clearAll();
-+
+    }
-+
-+
+    bs = distanceFinder.find(bs, distance);
-+
+  }
-+
-+
+  void SelectionEvaluator::define() {
-+
+    assert(statement.size() >= 3);
-+
-+
+    std::string variable = boost::any_cast<std::string>(statement[1].value);
-+
-+
+    variables.insert(VariablesType::value_type(variable,
-+
+                                               expression(statement, 2)));
-+
+  }
-+
-+
+  /** @todo */
-+
+  void SelectionEvaluator::predefine(const std::string& script) {
-+
-+
+    if (compiler.compile("#predefine", script)) {
-+
+      std::vector<std::vector<Token> > aatoken = compiler.getAatokenCompiled();
-+
+      if (aatoken.size() != 1) {
-+
+        evalError("predefinition does not have exactly 1 command:"
-+
+                  + script);
-+
+        return;
-+
+      }
-+
+      std::vector<Token> statement = aatoken[0];
-+
+      if (statement.size() > 2) {
-+
+        int tok = statement[1].tok;
-+
+        if (tok == Token::identifier ||
-+
+            (tok & Token::predefinedset) == Token::predefinedset) {
-+
+          std::string variable = boost::any_cast<std::string>(statement[1].value);
-+
+          variables.insert(VariablesType::value_type(variable, statement));
-+
-+
+        } else {
-+
+          evalError("invalid variable name:" + script);
-+
+        }
-+
+      }else {
-+
+        evalError("bad predefinition length:" + script);
-+
+      }
-+
-+
+    } else {
-+
+      evalError("predefined set compile error:" + script +
-+
+                "\ncompile error:" + compiler.getErrorMessage());
-+
+    }
-+
+  }
-+
-+
+  void SelectionEvaluator::select(OOPSEBitSet& bs){
-+
+    bs = expression(statement, 1);
-+
+  }
-+
-+
+  OOPSEBitSet SelectionEvaluator::lookupValue(const std::string& variable){
-+
-+
+    OOPSEBitSet bs(nStuntDouble);
-+
+    std::map<std::string, boost::any>::iterator i = variables.find(variable);
-+
-+
+    if (i != variables.end()) {
-+
+      if (i->second.type() == typeid(OOPSEBitSet)) {
-+
+        return boost::any_cast<OOPSEBitSet>(i->second);
-+
+      } else if (i->second.type() ==  typeid(std::vector<Token>)){
-+
+        bs = expression(boost::any_cast<std::vector<Token> >(i->second), 2);
-+
+        i->second =  bs; /**@todo fixme */
-+
+        return bs;
-+
+      }
-+
+    } else {
-+
+      unrecognizedIdentifier(variable);
-+
+    }
-+
-+
+    return bs;
-+
+  }
-+
-+
+  OOPSEBitSet SelectionEvaluator::nameInstruction(const std::string& name){
-+
+    return nameFinder.match(name);
-+
+  }
-+
-+
+  bool SelectionEvaluator::containDynamicToken(const std::vector<Token>& tokens){
-+
+    std::vector<Token>::const_iterator i;
-+
+    for (i = tokens.begin(); i != tokens.end(); ++i) {
-+
+      if (i->tok & Token::dynamic) {
-+
+        return true;
-+
+      }
-+
+    }
-+
-+
+    return false;
-+
+  }
-+
-+
+  void SelectionEvaluator::clearDefinitionsAndLoadPredefined() {
-+
+    variables.clear();
-+
+    //load predefine
-+
+    //predefine();
-+
+  }
-+
-+
+  OOPSEBitSet SelectionEvaluator::evaluate() {
-+
+    OOPSEBitSet bs(nStuntDouble);
-+
+    if (isLoaded_) {
-+
+      pc = 0;
-+
+      instructionDispatchLoop(bs);
-+
+    }
-+
-+
+    return bs;
-+
+  }
-+
-+
+  OOPSEBitSet SelectionEvaluator::indexInstruction(const boost::any& value) {
-+
+    OOPSEBitSet bs(nStuntDouble);
-+
-+
+    if (value.type() == typeid(int)) {
-+
+      int index = boost::any_cast<int>(value);
-+
+      if (index < 0 || index >= bs.size()) {
-+
+        invalidIndex(index);
-+
+      } else {
-+
+        bs = indexFinder.find(index);
-+
+      }
-+
+    } else if (value.type() == typeid(std::pair<int, int>)) {
-+
+      std::pair<int, int> indexRange= boost::any_cast<std::pair<int, int> >(value);
-+
+      assert(indexRange.first <= indexRange.second);
-+
+      if (indexRange.first < 0 || indexRange.second >= bs.size()) {
-+
+        invalidIndexRange(indexRange);
-+
+      }else {
-+
+        bs = indexFinder.find(indexRange.first, indexRange.second);
-+
+      }
-+
+    }
-+
-+
+    return bs;
-+
+  }
-+
-+
-+
+  RealType SelectionEvaluator::getCharge(Atom* atom) {
-+
+    RealType charge =0.0;
-+
+    AtomType* atomType = atom->getAtomType();
-+
+    if (atomType->isCharge()) {
-+
+      GenericData* data = atomType->getPropertyByName("Charge");
-+
+      if (data != NULL) {
-+
+        DoubleGenericData* doubleData= dynamic_cast<DoubleGenericData*>(data);
-+
-+
+        if (doubleData != NULL) {
-+
+          charge = doubleData->getData();
-+
-+
+        } else {
-+
+          sprintf( painCave.errMsg,
-+
+                   "Can not cast GenericData to DoubleGenericData\n");
-+
+          painCave.severity = OOPSE_ERROR;
-+
+          painCave.isFatal = 1;
-+
+          simError();
-+
+        }
-+
+      }
-+
+    }
-+
-+
+    return charge;
-+
+  }
-+

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing trunk/src/selection/SelectionEvaluator.cpp (file contents): Revision 277 by tim, Tue Feb 1 06:55:00 2005 UTC vs. Revision 1364 by cli2, Wed Oct 7 20:49:50 2009 UTC

Diff Legend

Comparing trunk/src/selection/SelectionEvaluator.cpp (file contents):
Revision 277 by tim, Tue Feb 1 06:55:00 2005 UTC vs.
Revision 1364 by cli2, Wed Oct 7 20:49:50 2009 UTC