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