src/ParametersPlugin/ParametersPlugin_Parameter.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 // File:        ParametersPlugin_Parameter.cpp
   4 // Created:     23 MArch 2015
   5 // Author:      sbh
   6
   7 #include <pyconfig.h>
   8
   9 #include "ParametersPlugin_Parameter.h"
  10 #include <ParametersPlugin_PyInterp.h>
  11
  12 #include <ModelAPI_AttributeString.h>
  13 #include <ModelAPI_ResultParameter.h>
  14 #include <ModelAPI_AttributeDouble.h>
  15 #include <ModelAPI_AttributeRefList.h>
  16 #include <ModelAPI_Tools.h>
  17 #include <ModelAPI_Session.h>
  18 #include <ModelAPI_Validator.h>
  19
  20 #include <string>
  21 #include <sstream>
  22
  23 ParametersPlugin_Parameter::ParametersPlugin_Parameter()
  24 {
  25   myInterp = std::shared_ptr<ParametersPlugin_PyInterp>(new ParametersPlugin_PyInterp());
  26   myInterp->initialize();
  27 }
  28
  29 ParametersPlugin_Parameter::~ParametersPlugin_Parameter()
  30 {
  31 }
  32
  33 void ParametersPlugin_Parameter::initAttributes()
  34 {
  35   data()->addAttribute(VARIABLE_ID(), ModelAPI_AttributeString::typeId());
  36   data()->addAttribute(EXPRESSION_ID(), ModelAPI_AttributeString::typeId());
  37   data()->addAttribute(ARGUMENTS_ID(), ModelAPI_AttributeRefList::typeId());
  38   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), ARGUMENTS_ID());
  39 }
  40
  41 bool ParametersPlugin_Parameter::isInHistory()
  42 {
  43   return false;
  44 }
  45
  46 void ParametersPlugin_Parameter::attributeChanged(const std::string& theID)
  47 {
  48   if (theID == EXPRESSION_ID()) { // recompute only on change of the expression
  49     ResultParameterPtr aParam = document()->createParameter(data());
  50
  51     std::string anExpression = string(EXPRESSION_ID())->value();
  52     if(anExpression.empty()) {
  53       // clear error/result if the expression is empty
  54       setError("", false);
  55       return;
  56     }
  57     std::string outErrorMessage;
  58     double aValue = evaluate(anExpression, outErrorMessage);
  59     // Name
  60     std::string aName = string(VARIABLE_ID())->value();
  61     std::ostringstream sstream;
  62     sstream << aValue;
  63     std::string aParamValue = sstream.str();
  64     data()->setName(aName);
  65     aParam->data()->setName(aName);
  66     // Error
  67     if (!outErrorMessage.empty()) {
  68       std::string aStateMsg("Error: " + outErrorMessage);
  69       data()->execState(ModelAPI_StateExecFailed);
  70       setError(aStateMsg, false);
  71     } else {
  72       static const std::string anEmptyMsg(""); // it is checked in the validator by the empty message
  73       setError(anEmptyMsg, false);
  74       data()->execState(ModelAPI_StateDone);
  75     }
  76     // Value
  77     AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
  78     aValueAttribute->setValue(aValue);
  79     setResult(aParam);
  80   }
  81 }
  82
  83 void ParametersPlugin_Parameter::execute()
  84 {
  85   // just call recompute
  86   attributeChanged(EXPRESSION_ID());
  87 }
  88
  89 double ParametersPlugin_Parameter::evaluate(const std::string& theExpression, std::string& theError)
  90 {
  91
  92   std::list<std::string> anExprParams = myInterp->compile(theExpression);
  93   // find expression's params in the model
  94   std::list<std::string> aContext;
  95   std::list<std::string>::iterator it = anExprParams.begin();
  96   std::list<ResultParameterPtr> aParamsList;
  97   for ( ; it != anExprParams.end(); it++) {
  98     double aValue;
  99     ResultParameterPtr aParamRes;
 100     if (!ModelAPI_Tools::findVariable(*it, aValue, aParamRes)) continue;
 101     aParamsList.push_back(aParamRes);
 102
 103     std::ostringstream sstream;
 104     sstream << aValue;
 105     std::string aParamValue = sstream.str();
 106     aContext.push_back(*it + "=" + aParamValue);
 107   }
 108   // compare the list of parameters to store if changed
 109   AttributeRefListPtr aParams = reflist(ARGUMENTS_ID());
 110   bool aDifferent = aParams->size() != aParamsList.size();
 111   if (!aDifferent) {
 112     std::list<ResultParameterPtr>::iterator aNewIter = aParamsList.begin();
 113     std::list<ObjectPtr> anOldList = aParams->list();
 114     std::list<ObjectPtr>::iterator anOldIter = anOldList.begin();
 115     for(; !aDifferent && aNewIter != aParamsList.end(); aNewIter++, anOldIter++) {
 116       if (*aNewIter != *anOldIter)
 117         aDifferent = true;
 118     }
 119   }
 120   if (aDifferent) {
 121     aParams->clear();
 122     std::list<ResultParameterPtr>::iterator aNewIter = aParamsList.begin();
 123     for(; aNewIter != aParamsList.end(); aNewIter++) {
 124       aParams->append(*aNewIter);
 125     }
 126   }
 127
 128   myInterp->extendLocalContext(aContext);
 129   double result = myInterp->evaluate(theExpression, theError);
 130   myInterp->clearLocalContext();
 131   return result;
 132 }
 133
 134 bool ParametersPlugin_Parameter::isPreviewNeeded() const
 135 {
 136   return false;
 137 }