1 files changed, 125 insertions, 0 deletions
diff --git a/Project/ControlEditor.h b/Project/ControlEditor.h
new file mode 100644
index 0000000..eb1bc4f
--- /dev/null
+++ b/Project/ControlEditor.h
@@ -0,0 +1,125 @@
+#ifndef CONTROLEDITOR_H
+#define CONTROLEDITOR_H
+
+#include "ControlEditorBase.h"
+#include <wx/wrapsizer.h>
+#include <wx/dcclient.h>
+#include <wx/dcscreen.h>
+#include <wx/graphics.h>
+#include <GL/gl.h>
+#include <GL/glu.h>
+
+#include "IOControl.h"
+
+class FileHanding;
+class Camera;
+class Element;
+class ControlElement;
+class TransferFunction;
+class ConnectionLine;
+class Sum;
+class Multiplier;
+class Limiter;
+class RateLimiter;
+class Exponential;
+class Constant;
+class Gain;
+
+enum ControlElementButtonID { ID_IO = 0, ID_TF, ID_SUM, ID_CONST, ID_LIMITER, ID_GAIN, ID_MULT, ID_EXP, ID_RATELIM };
+
+class ControlElementButton : public wxWindow
+{
+   public:
+    ControlElementButton(wxWindow* parent, wxString label, wxImage image, wxWindowID id = wxID_ANY);
+    ~ControlElementButton();
+
+   protected:
+    virtual void OnPaint(wxPaintEvent& event);
+    virtual void OnMouseEnter(wxMouseEvent& event);
+    virtual void OnMouseLeave(wxMouseEvent& event);
+    virtual void OnLeftClickDown(wxMouseEvent& event);
+    virtual void OnLeftClickUp(wxMouseEvent& event);
+
+    wxString m_label;
+    wxFont m_font;
+    wxPoint m_labelPosition;
+
+    wxImage m_image;
+    wxSize m_imageSize;
+    wxPoint m_imagePosition;
+
+    int m_borderSize = 2;
+    bool m_mouseAbove = false;
+    bool m_selected = false;
+
+    wxSize m_buttonSize;
+};
+
+class ControlEditor : public ControlEditorBase
+{
+   public:
+    enum ControlEditorMode {
+        MODE_EDIT = 0,
+        MODE_MOVE_ELEMENT,
+        MODE_MOVE_LINE,
+        MODE_DRAG,
+        MODE_DRAG_INSERT,
+        MODE_INSERT,
+        MODE_INSERT_LINE,
+        MODE_SELECTION_RECT,
+        MODE_PASTE,
+        MODE_DRAG_PASTE
+    };
+
+    ControlEditor(wxWindow* parent,
+                  int ioflags = IOControl::IN_TERMINAL_VOLTAGE | IOControl::IN_VELOCITY | IOControl::OUT_FIELD_VOLTAGE |
+                                IOControl::OUT_MEC_POWER);
+    virtual ~ControlEditor();
+
+    virtual void AddElement(ControlElementButtonID id);
+    virtual void Redraw() { m_glCanvas->Refresh(); }
+    virtual void RotateSelectedElements(bool clockwise);
+    virtual void DeleteSelectedElements();
+    virtual void CheckConnections();
+    virtual std::vector<ConnectionLine*> GetConnectionLineList() const { return m_connectionList; }
+    virtual std::vector<ControlElement*> GetControlElementList() const { return m_elementList; }
+    virtual void SetElementsList(std::vector<ControlElement*> elementList) { m_elementList = elementList; }
+    virtual void SetConnectionsList(std::vector<ConnectionLine*> connectionList) { m_connectionList = connectionList; }
+
+   protected:
+    virtual void OnImportClick(wxCommandEvent& event);
+    virtual void OnExportClick(wxCommandEvent& event);
+    virtual void OnKeyDown(wxKeyEvent& event);
+    virtual void OnIdle(wxIdleEvent& event);
+    virtual void OnScroll(wxMouseEvent& event);
+    virtual void OnDoubleClick(wxMouseEvent& event);
+    virtual void OnLeftClickDown(wxMouseEvent& event);
+    virtual void OnLeftClickUp(wxMouseEvent& event);
+    virtual void OnMiddleDown(wxMouseEvent& event);
+    virtual void OnMiddleUp(wxMouseEvent& event);
+    virtual void OnMouseMotion(wxMouseEvent& event);
+    virtual void OnPaint(wxPaintEvent& event);
+    virtual void LeftClickDown(wxMouseEvent& event);
+
+    void BuildControlElementPanel();
+    void SetViewport();
+
+    std::vector<ConnectionLine*>::iterator DeleteLineFromList(std::vector<ConnectionLine*>::iterator& it);
+
+    wxGLContext* m_glContext = NULL;
+    Camera* m_camera = NULL;
+
+    ControlEditorMode m_mode = MODE_EDIT;
+
+    wxRect2DDouble m_selectionRect;
+    wxPoint2DDouble m_startSelRect;
+
+    std::vector<ControlElement*> m_elementList;
+    std::vector<ConnectionLine*> m_connectionList;
+
+    bool m_firstDraw = true;
+    int m_ioFlags;
+    
+    int m_lastElementID = 0;
+};
+#endif  // CONTROLEDITOR_H