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;

  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