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_Tools.h>
  14
  15 #include <string>
  16 #include <sstream>
  17
  18 ParametersPlugin_Parameter::ParametersPlugin_Parameter()
  19 {
  20   myInterp = new ParametersPlugin_PyInterp();
  21   myInterp->initialize();
  22 }
  23
  24 ParametersPlugin_Parameter::~ParametersPlugin_Parameter()
  25 {
  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);
  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
  81   std::list<std::string> anExprParams = myInterp->compile(theExpression);
  82   // find expression's params in the model
  83   std::list<std::string> aContext;
  84   std::list<std::string>::iterator it = anExprParams.begin();
  85   for ( ; it != anExprParams.end(); it++) {
  86     double aValue;
  87     if (!ModelAPI_Tools::findVariable(*it, aValue)) continue;
  88
  89     std::ostringstream sstream;
  90     sstream << aValue;
  91     std::string aParamValue = sstream.str();
  92     aContext.push_back(*it + "=" + aParamValue);
  93   }
  94   myInterp->extendLocalContext(aContext);
  95   double result = myInterp->evaluate(theExpression, theError);
  96   myInterp->clearLocalContext();
  97   return result;
  98 }