OptimizerBase.cpp 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. #include "OptimizerBase.h"
  2. #include <iostream>
  3. namespace mdd{
  4. OptimizerBase::opt_state OptimizerBase::updateReader()
  5. {
  6. opt_state ret;
  7. ret.module_state = _module->update();
  8. if (ret.module_state != state::STATE_ERROR)
  9. {
  10. for (size_t i = 0; i < _output_vals.size(); ++i)
  11. {
  12. _output_vals[i] = _outputs[i]->getValue()["value"][0].get<double>();
  13. //std::cout << "get: " << i << ": " << _output_vals[i] << std::endl;
  14. }
  15. ret.opt_value = _func_expr.value();
  16. //std::cout << "get: " << "opt" << ": " << ret.opt_value << std::endl;
  17. }
  18. return ret;
  19. }
  20. bool OptimizerBase::addModifier(std::string input_id, const json& limit)
  21. {
  22. if (_module != nullptr)
  23. {
  24. auto ptr = _module->getInput(input_id);
  25. if (ptr != nullptr)
  26. {
  27. _inputs.push_back(ptr);
  28. _input_limits.push_back(limit);
  29. return true;
  30. }
  31. }
  32. return false;
  33. }
  34. bool OptimizerBase::addReader(std::string output_id)
  35. {
  36. if (_module != nullptr)
  37. {
  38. auto ptr = _module->getOutput(output_id);
  39. if (ptr != nullptr)
  40. {
  41. _outputs.push_back(ptr);
  42. _output_vals.push_back(ptr->getValue()["value"][0].get<double>());
  43. return true;
  44. }
  45. }
  46. return false;
  47. }
  48. bool OptimizerBase::changeModifier(std::string input_id, const json& limit)
  49. {
  50. for (size_t i = 0; i < _inputs.size(); i++)
  51. {
  52. if (_inputs[i]->getID() == input_id)
  53. {
  54. _input_limits[i] = limit;
  55. return true;
  56. }
  57. }
  58. return false;
  59. }
  60. void OptimizerBase::removeModifier(std::string input_id)
  61. {
  62. for (auto it = _inputs.begin(); it != _inputs.end(); ++it) {
  63. if ((*it)->getID() == input_id)
  64. {
  65. _inputs.erase(it);
  66. }
  67. }
  68. }
  69. void OptimizerBase::removeReader(std::string output_id)
  70. {
  71. for (auto it = _outputs.begin(); it != _outputs.end(); ++it) {
  72. if ((*it)->getID() == output_id)
  73. {
  74. _outputs.erase(it);
  75. }
  76. }
  77. }
  78. bool OptimizerBase::setEvaluation(std::string func)
  79. {
  80. exprtk::symbol_table<double> symbol_table;
  81. for (size_t i = 0; i < _output_vals.size(); ++i)
  82. {
  83. symbol_table.add_variable("out" + std::to_string(i), _output_vals[i]);
  84. }
  85. symbol_table.add_constants();
  86. _func_expr.register_symbol_table(symbol_table);
  87. exprtk::parser<double> parser;
  88. return parser.compile(func, _func_expr);
  89. }
  90. }