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
  14 #include <string>
  15 #include <sstream>
  16
  17 ParametersPlugin_Parameter::ParametersPlugin_Parameter()
  18 {
  19   myInterp = new ParametersPlugin_PyInterp();
  20   myInterp->initialize();
  21 }
  22
  23 ParametersPlugin_Parameter::~ParametersPlugin_Parameter()
  24 {
  25   myInterp->destroy();
  26   delete myInterp;
  27 }
  28
  29 void ParametersPlugin_Parameter::initAttributes()
  30 {
  31   data()->addAttribute(ParametersPlugin_Parameter::VARIABLE_ID(),
  32                        ModelAPI_AttributeString::typeId());
  33   data()->addAttribute(ParametersPlugin_Parameter::EXPRESSION_ID(),
  34                        ModelAPI_AttributeString::typeId());
  35 }
  36
  37 bool ParametersPlugin_Parameter::isInHistory()
  38 {
  39   return false;
  40 }
  41
  42 void ParametersPlugin_Parameter::attributeChanged(const std::string&)
  43 {
  44   ResultParameterPtr aParam = document()->createParameter(data());
  45
  46   std::string anExpression = string(ParametersPlugin_Parameter::EXPRESSION_ID())->value();
  47   if(anExpression.empty()) {
  48     // clear error/result if the expression is empty
  49     setError("", false);
  50     return;
  51   }
  52   std::string outErrorMessage;
  53   double aValue = evaluate(anExpression, outErrorMessage);
  54   // Name
  55   std::string aName = string(ParametersPlugin_Parameter::VARIABLE_ID())->value();
  56   std::ostringstream sstream;
  57   sstream << aValue;
  58   std::string aParamValue = sstream.str();
  59   data()->setName(aName + " ("+ aParamValue + ")");
  60   aParam->data()->setName(aName);
  61   // Error
  62   std::string aStateMsg;
  63   if (!outErrorMessage.empty()) {
  64     aStateMsg = "Error: " + outErrorMessage;
  65     data()->execState(ModelAPI_StateExecFailed);
  66   }
  67   setError(aStateMsg, false);
  68   // Value
  69   AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
  70   aValueAttribute->setValue(aValue);
  71   setResult(aParam);
  72 }
  73
  74 void ParametersPlugin_Parameter::execute()
  75 {
  76 }
  77
  78 double ParametersPlugin_Parameter::evaluate(const std::string& theExpression, std::string& theError)
  79 {
  80   std::list<std::string> anExprParams = myInterp->compile(theExpression);
  81   // find expression's params in the model
  82   std::list<std::string> aContext;
  83   std::list<std::string>::iterator it = anExprParams.begin();
  84   for ( ; it != anExprParams.end(); it++) {
  85     std::string aParamName = *it;
  86     ObjectPtr aParamObj = document()->objectByName(ModelAPI_ResultParameter::group(), aParamName);
  87     ResultParameterPtr aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aParamObj);
  88     if(!aParam.get()) continue;
  89     AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
  90     std::ostringstream sstream;
  91     sstream << aValueAttribute->value();
  92     std::string aParamValue = sstream.str();
  93     aContext.push_back(aParamName + "=" + aParamValue);
  94   }
  95   myInterp->extendLocalContext(aContext);
  96   return myInterp->evaluate(theExpression, theError);
  97 }