custom_function.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <string>
#include <vector>
#include <QWidget>
#include <QString>
#include <QDomDocument>
#include <QString>
#include "PlotJuggler/plotdata.h"
#include "PlotJuggler/transform_function.h"
 
using namespace PJ;
 
class CustomFunction;
 
typedef std::shared_ptr<CustomFunction> CustomPlotPtr;
 
struct SnippetData
{
  QString alias_name;
  QString global_vars;
  QString function;
  QString linked_source;
  QStringList additional_sources;
};
 
typedef std::map<QString, SnippetData> SnippetsMap;
 
SnippetData GetSnippetFromXML(const QDomElement& snippets_element);
 
SnippetsMap GetSnippetsFromXML(const QString& xml_text);
 
SnippetsMap GetSnippetsFromXML(const QDomElement& snippets_element);
 
QDomElement ExportSnippetToXML(const SnippetData& snippet, QDomDocument& destination_doc);
 
QDomElement ExportSnippets(const SnippetsMap& snippets, QDomDocument& destination_doc);
 
class CustomFunction : public PJ::TransformFunction
{
public:
  CustomFunction(SnippetData snippet = {});
 
  void setSnippet(const SnippetData& snippet);
 
  void reset() override;
 
  int numInputs() const override
  {
    return -1;
  }
 
  int numOutputs() const override
  {
    return 1;
  }
 
  QString aliasName() const
  {
    return _snippet.alias_name;
  }
 
  void calculate() override;
 
  bool xmlSaveState(QDomDocument& doc, QDomElement& parent_element) const override;
 
  bool xmlLoadState(const QDomElement& parent_element) override;
 
  const SnippetData& snippet() const;
 
  virtual QString language() const = 0;
 
  virtual void initEngine() = 0;
 
  void calculateAndAdd(PlotDataMapRef& src_data);
 
  virtual void calculatePoints(const std::vector<const PlotData*>& src_data,
                               size_t point_index,
                               std::vector<PlotData::Point>& new_points) = 0;
 
protected:
  SnippetData _snippet;
  std::string _linked_plot_name;
  std::string _plot_name;
 
  std::vector<std::string> _used_channels;
};