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::execute()
  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     aParam->data()->string(ModelAPI_ResultParameter::STATE())->setValue("");
  50     return;
  51   }
  52   // Value
  53   std::string outErrorMessage;
  54   double aValue = evaluate(anExpression, outErrorMessage);
  55   AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
  56   aValueAttribute->setValue(aValue);
  57   setResult(aParam);
  58   // Name
  59   std::string aName = string(ParametersPlugin_Parameter::VARIABLE_ID())->value();
  60   std::ostringstream sstream;
  61   sstream << aValue;
  62   std::string aParamValue = sstream.str();
  63   data()->setName(aName + " ("+ aParamValue + ")");
  64   aParam->data()->setName(aName);
  65   // Error
  66   AttributeStringPtr aErrorAttr = aParam->data()->string(ModelAPI_ResultParameter::STATE());
  67   std::string aStateMsg;
  68   if (outErrorMessage.empty()) {
  69     aStateMsg = "Result: " + aParamValue;
  70   } else {
  71     aStateMsg = "Error:\n" + outErrorMessage;
  72   }
  73   aErrorAttr->setValue(aStateMsg);
  74   //if(!outErrorMessage.empty()) {
  75   //  data()->execState(ModelAPI_StateExecFailed);
  76   //}
  77 }
  78
  79 double ParametersPlugin_Parameter::evaluate(const std::string& theExpression, std::string& theError)
  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     std::string aParamName = *it;
  87     ObjectPtr aParamObj = document()->objectByName(ModelAPI_ResultParameter::group(), aParamName);
  88     ResultParameterPtr aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aParamObj);
  89     if(!aParam.get()) continue;
  90     AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
  91     std::ostringstream sstream;
  92     sstream << aValueAttribute->value();
  93     std::string aParamValue = sstream.str();
  94     aContext.push_back(aParamName + "=" + aParamValue);
  95   }
  96   myInterp->extendLocalContext(aContext);
  97   return myInterp->evaluate(theExpression, theError);
  98 }