reactive_function.h

Go to the documentation of this file.

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */
 
#ifndef REACTIVE_FUNCTION_H
#define REACTIVE_FUNCTION_H
 
#include "PlotJuggler/transform_function.h"
#include <sol/sol.hpp>
 
class TimeseriesRef;
class CreatedSeriesBase;
class CreatedSeriesTime;
class CreatedSeriesXY;
 
namespace PJ
{
struct TimeseriesRef
{
  TimeseriesRef(PlotData* data);
 
  std::pair<double, double> at(unsigned i) const;
 
  void set(unsigned index, double x, double y);
 
  double atTime(double t) const;
 
  unsigned size() const;
 
  void clear() const;
 
  PJ::PlotData* _plot_data = nullptr;
};
 
//-----------------------
 
struct CreatedSeriesBase
{
  CreatedSeriesBase(PlotDataMapRef* data_map, const std::string& name, bool timeseries);
 
  std::pair<double, double> at(unsigned i) const;
 
  void clear();
 
  void push_back(double x, double y);
 
  unsigned size() const;
 
  PJ::PlotDataXY* _plot_data = nullptr;
};
 
struct CreatedSeriesTime : public CreatedSeriesBase
{
  CreatedSeriesTime(PlotDataMapRef* data_map, const std::string& name);
};
 
struct CreatedSeriesXY : public CreatedSeriesBase
{
  CreatedSeriesXY(PlotDataMapRef* data_map, const std::string& name);
};
 
//-----------------------
 
class ReactiveLuaFunction : public PJ::TransformFunction
{
public:
  ReactiveLuaFunction(PlotDataMapRef* data_map, QString lua_global, QString lua_function,
                      QString lua_library);
 
  const char* name() const override
  {
    return "ReactiveLuaFunction";
  }
 
  int numInputs() const override
  {
    return -1;
  }
 
  int numOutputs() const override
  {
    return -1;
  }
 
  void reset() override;
 
  void setTimeTracker(double time_tracker_value);
 
  void calculate() override;
 
  const std::vector<std::string>& createdCurves() const
  {
    return _created_curves;
  }
 
  bool xmlSaveState(QDomDocument& doc, QDomElement& parent_element) const override;
 
  bool xmlLoadState(const QDomElement& parent_element) override;
 
  QString getGlobalCode()
  {
    return QString::fromStdString(_global_code);
  }
 
  QString getFunctionCode()
  {
    return QString::fromStdString(_function_code);
  }
 
protected:
  void prepareLua();
 
  double _tracker_value = 0;
  std::string _global_code;
  std::string _function_code;
  std::string _library_code;
 
  std::vector<std::string> _created_curves;
 
  sol::state _lua_engine;
  sol::protected_function _lua_function;
 
  sol::usertype<TimeseriesRef> _timeseries_ref;
  sol::usertype<CreatedSeriesTime> _created_timeseries;
  sol::usertype<CreatedSeriesXY> _created_scatter;
 
private:
  void init();
};
 
}  // namespace PJ
 
#endif  // REACTIVE_FUNCTION_H