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