[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 351 by tim, Tue Feb 15 19:36:07 2005 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"
-+
+namespace oopse {
-+
+SelectionEvaluator::SelectionEvaluator(SimInfo* si) : info(si), nameFinder(info), distanceFinder(info), isLoaded_(false){
-+
+    nStuntDouble = info->getNGlobalAtoms() + info->getNRigidBodies();
-+
+}
-+
-+
+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();
-+
+        std::cerr << "SelectionCompiler Error: " << errorMessage << std::endl;
-+
+        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() {
-+
+    //for (int i = scriptLevelMax; --i >= 0; )
-+
+    //    stack[i].clear();
-+
+    //scriptLevel = 0;
-+
+    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) {
-+
+  std::ifstream 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(BitSet& 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;
-+
+        }
-+
+    }
-+
-+
+}
-+
-+
+  BitSet SelectionEvaluator::expression(const std::vector<Token>& code, int pcStart) {
-+
+    BitSet bs;
-+
+    std::stack<BitSet> 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 = BitSet(nStuntDouble);
-+
+        bs.setAll();
-+
+        stack.push(bs);
-+
+        break;
-+
+      case Token::none:
-+
+        bs = BitSet(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();
-+
+  }
-+
-+
-+
-+
+BitSet SelectionEvaluator::comparatorInstruction(const Token& instruction) {
-+
+    int comparator = instruction.tok;
-+
+    int property = instruction.intValue;
-+
+    float comparisonValue = boost::any_cast<float>(instruction.value);
-+
+    float propertyValue;
-+
+    BitSet 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);
-+
+        }
-+
-+
+        //change the positions of atoms which belong to the rigidbodies
-+
+        for (rb = mol->beginRigidBody(rbIter); rb != NULL; rb = mol->nextRigidBody(rbIter)) {
-+
+            compareProperty(rb, bs, property, comparator, comparisonValue);
-+
+        }
-+
+    }
-+
-+
+    return bs;
-+
+}
-+
-+
+void SelectionEvaluator::compareProperty(StuntDouble* sd, BitSet& bs, int property, int comparator, float comparisonValue) {
-+
+        double propertyValue;
-+
+        switch (property) {
-+
+        case Token::mass:
-+
+            propertyValue = sd->getMass();
-+
+            break;
-+
+        case Token::charge:
-+
+            return;
-+
+            //break;
-+
+        case Token::dipole:
-+
+            return;
-+
+            //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, BitSet& 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(BitSet& bs){
-+
+    bs = expression(statement, 1);
-+
+}
-+
-+
+BitSet SelectionEvaluator::lookupValue(const std::string& variable){
-+
-+
+    BitSet bs(nStuntDouble);
-+
+    std::map<std::string, boost::any>::iterator i = variables.find(variable);
-+
-+
+    if (i != variables.end()) {
-+
+        if (i->second.type() == typeid(BitSet)) {
-+
+            return boost::any_cast<BitSet>(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;
-+
+}
-+
-+
+BitSet 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();
-+
+}
-+
-+
+BitSet SelectionEvaluator::evaluate() {
-+
+    BitSet bs(nStuntDouble);
-+
+    if (isLoaded_) {
-+
+        pc = 0;
-+
+        instructionDispatchLoop(bs);
-+
+    }
-+
-+
+    return bs;
-+
+}
-+
-+
+BitSet SelectionEvaluator::indexInstruction(const boost::any& value) {
-+
+    BitSet bs(nStuntDouble);
-+
-+
+    if (value.type() == typeid(int)) {
-+
+        int index = boost::any_cast<int>(value);
-+
+        if (index < 0 || index >= bs.size()) {
-+
+            invalidIndex(index);
-+
+        } else {
-+
+            bs.setBitOn(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.setRangeOn(indexRange.first, indexRange.second);
-+
+        }
-+
+    }
-+
-+
+    return bs;
-+
+}
-+
-+
+//BitSet SelectionEvaluator::evaluate(int frameNo) {
-+
+//
-+
+//}
-+
-+
+}

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 351 by tim, Tue Feb 15 19:36:07 2005 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 351 by tim, Tue Feb 15 19:36:07 2005 UTC