ControlEditor.cpp

/*
 *  Copyright (C) 2017  Thales Lima Oliveira <thales@ufu.br>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#include "ControlEditor.h"
 
#include "../elements/controlElement/ConnectionLine.h"
#include "../elements/controlElement/Constant.h"
#include "../elements/controlElement/ControlElement.h"
#include "../elements/controlElement/Divider.h"
#include "../elements/controlElement/Exponential.h"
#include "../elements/controlElement/Gain.h"
#include "../elements/controlElement/Limiter.h"
#include "../elements/controlElement/MathExpression.h"
#include "../elements/controlElement/MathOperation.h"
#include "../elements/controlElement/Multiplier.h"
#include "../elements/controlElement/RateLimiter.h"
#include "../elements/controlElement/Sum.h"
#include "../elements/controlElement/TransferFunction.h"
#include "../elements/controlElement/ControlElementContainer.h"
#include "../elements/controlElement/ControlElementSolver.h"
 
#ifdef USING_WX_3_0_X
#include "utils/DegreesAndRadians.h"
#endif
#include "../utils/Camera.h"
#include "../utils/FileHanding.h"
#include "../utils/Path.h"
 
#include "ChartView.h"
#include "../utils/ElementPlotData.h"
 
ControlElementButton::ControlElementButton(wxWindow* parent, wxString label, wxImage image, wxWindowID id)
    : wxWindow(parent, id)
{
    SetBackgroundColour(*wxWHITE);
    m_font = wxSystemSettings::GetFont(wxSYS_DEFAULT_GUI_FONT);
    m_label = label;
    m_image = image;
    m_imageSize = wxSize(image.GetWidth(), image.GetHeight());
 
    // Calculate label size.
    wxScreenDC dc;
    dc.SetFont(m_font);
    wxSize textSize = dc.GetTextExtent(label);
 
    int buttonWidth = 0;
    if (textSize.GetWidth() > m_imageSize.GetWidth()) {
        buttonWidth = textSize.GetWidth();
        m_imagePosition = wxPoint((buttonWidth - m_imageSize.GetWidth()) / 2 + m_borderSize, m_borderSize);
        m_labelPosition = wxPoint(m_borderSize, m_imageSize.GetHeight() + m_borderSize);
    }
    else {
        buttonWidth = m_imageSize.GetWidth();
        m_imagePosition = wxPoint(m_borderSize, m_borderSize);
        m_labelPosition =
            wxPoint((buttonWidth - textSize.GetWidth()) / 2 + m_borderSize, m_imageSize.GetHeight() + m_borderSize);
    }
    m_buttonSize =
        wxSize(buttonWidth + 2 * m_borderSize, textSize.GetHeight() + m_imageSize.GetHeight() + 2 * m_borderSize);
    SetMinSize(m_buttonSize + wxSize(m_borderSize, m_borderSize));
 
    // Events.
    Bind(wxEVT_PAINT, &ControlElementButton::OnPaint, this);
    Bind(wxEVT_ENTER_WINDOW, &ControlElementButton::OnMouseEnter, this);
    Bind(wxEVT_LEAVE_WINDOW, &ControlElementButton::OnMouseLeave, this);
    Bind(wxEVT_LEFT_DOWN, &ControlElementButton::OnLeftClickDown, this);
    Bind(wxEVT_LEFT_UP, &ControlElementButton::OnLeftClickUp, this);
}
 
ControlElementButton::~ControlElementButton() {}
void ControlElementButton::OnPaint(wxPaintEvent& event)
{
    wxPaintDC dc(this);
    wxGraphicsContext* gc = wxGraphicsContext::Create(dc);
    if (gc) {
        if (m_mouseAbove) {
            if (m_selected) {
                gc->SetPen(wxPen(wxColour(0, 125, 255, 255), m_borderSize - 1));
                gc->SetBrush(wxBrush(wxColour(0, 125, 255, 100)));
            }
            else {
                gc->SetPen(*wxTRANSPARENT_PEN);
                gc->SetBrush(wxBrush(wxColour(0, 125, 255, 70)));
            }
            gc->DrawRectangle(m_borderSize / 2, m_borderSize / 2, m_buttonSize.GetWidth(), m_buttonSize.GetHeight());
        }
        gc->DrawBitmap(gc->CreateBitmapFromImage(m_image), m_imagePosition.x, m_imagePosition.y, m_imageSize.GetWidth(),
            m_imageSize.GetHeight());
        gc->SetFont(m_font, *wxBLACK);
        gc->DrawText(m_label, m_labelPosition.x, m_labelPosition.y);
        delete gc;
    }
}
 
void ControlElementButton::OnMouseEnter(wxMouseEvent& event)
{
    m_mouseAbove = true;
    Refresh();
    event.Skip();
}
 
void ControlElementButton::OnMouseLeave(wxMouseEvent& event)
{
    m_mouseAbove = false;
    Refresh();
    event.Skip();
}
 
void ControlElementButton::OnLeftClickDown(wxMouseEvent& event)
{
    m_selected = true;
    Refresh();
    event.Skip();
}
 
void ControlElementButton::OnLeftClickUp(wxMouseEvent& event)
{
    m_selected = false;
    Refresh();
    event.Skip();
}
 
ControlEditor::ControlEditor(wxWindow* parent, int ioflags) : ControlEditorBase(parent)
{
    BuildControlElementPanel();
    //m_glContext = new wxGLContext(m_glCanvas, sharedGLContext);
    //m_glContext->SetCurrent(*m_glCanvas);
    m_camera = new Camera();
    m_selectionRect = wxRect2DDouble(0, 0, 0, 0);
    m_cePanel->SetBackgroundColour(wxColour(255, 255, 255));
    m_cePanel->SetBackgroundStyle(wxBG_STYLE_PAINT);  // To allow wxBufferedPaintDC works properly.
    // m_camera->SetScale(1.2);
    m_ioFlags = ioflags;
 
    BuildColourList();
}
ControlEditor::~ControlEditor()
{
    // m_tfButton->Disconnect(wxEVT_LEFT_DOWN, wxMouseEventHandler(ControlEditor::LeftClickDown), m_tfButton, this);
}
 
void ControlEditor::BuildControlElementPanel()
{
    m_panelControlElements->SetDoubleBuffered(true);
    wxWrapSizer* wrapSizer = new wxWrapSizer();
    m_panelControlElements->SetSizer(wrapSizer);
 
    wxFileName exeFileName(wxStandardPaths::Get().GetExecutablePath());
    //wxString exePath = exeFileName.GetPath();
 
    ControlElementButton* ioButton = new ControlElementButton(
        m_panelControlElements, _("In/Out"),
        wxImage(Paths::GetDataPath() + "/images/control/io.png"),
        static_cast<int>(ControlElementButtonID::ID_IO));
    wrapSizer->Add(ioButton, 0, wxALL, 5);
    ioButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* tfButton = new ControlElementButton(
        m_panelControlElements, _("Transfer fcn"),
        wxImage(Paths::GetDataPath() + "/images/control/transferFunc.png"),
        static_cast<int>(ControlElementButtonID::ID_TF));
    wrapSizer->Add(tfButton, 0, wxALL, 5);
    tfButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* sumButton = new ControlElementButton(
        m_panelControlElements, _("Sum"),
        wxImage(Paths::GetDataPath() + "/images/control/sum.png"),
        static_cast<int>(ControlElementButtonID::ID_SUM));
    wrapSizer->Add(sumButton, 0, wxALL, 5);
    sumButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* constButton = new ControlElementButton(
        m_panelControlElements, _("Constant"),
        wxImage(Paths::GetDataPath() + "/images/control/value.png"),
        static_cast<int>(ControlElementButtonID::ID_CONST));
    wrapSizer->Add(constButton, 0, wxALL, 5);
    constButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* gainButton = new ControlElementButton(
        m_panelControlElements, _("Gain"),
        wxImage(Paths::GetDataPath() + "/images/control/gain.png"),
        static_cast<int>(ControlElementButtonID::ID_GAIN));
    wrapSizer->Add(gainButton, 0, wxALL, 5);
    gainButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* limButton = new ControlElementButton(
        m_panelControlElements, _("Limiter"),
        wxImage(Paths::GetDataPath() + "/images/control/limiter.png"),
        static_cast<int>(ControlElementButtonID::ID_LIMITER));
    wrapSizer->Add(limButton, 0, wxALL, 5);
    limButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* rateLimButton = new ControlElementButton(
        m_panelControlElements, _("Rate limiter"),
        wxImage(Paths::GetDataPath() + "/images/control/rateLimiter.png"),
        static_cast<int>(ControlElementButtonID::ID_RATELIM));
    wrapSizer->Add(rateLimButton, 0, wxALL, 5);
    rateLimButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* multButton = new ControlElementButton(
        m_panelControlElements, _("Multiplier"),
        wxImage(Paths::GetDataPath() + "/images/control/mult.png"),
        static_cast<int>(ControlElementButtonID::ID_MULT));
    wrapSizer->Add(multButton, 0, wxALL, 5);
    multButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* divButton = new ControlElementButton(
        m_panelControlElements, _("Divider"),
        wxImage(Paths::GetDataPath() + "/images/control/div.png"),
        static_cast<int>(ControlElementButtonID::ID_MATH_DIV));
    wrapSizer->Add(divButton, 0, wxALL, 5);
    divButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* mathExprButton = new ControlElementButton(
        m_panelControlElements, _("Math Expression"),
        wxImage(Paths::GetDataPath() + "/images/control/mathExpr.png"),
        static_cast<int>(ControlElementButtonID::ID_MATH_EXPR));
    wrapSizer->Add(mathExprButton, 0, wxALL, 5);
    mathExprButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
 
    ControlElementButton* satButton = new ControlElementButton(
        m_panelControlElements, _("Exponential"),
        wxImage(Paths::GetDataPath() + "/images/control/sat.png"),
        static_cast<int>(ControlElementButtonID::ID_EXP));
    wrapSizer->Add(satButton, 0, wxALL, 5);
    satButton->Bind(wxEVT_LEFT_DOWN, &ControlEditor::LeftClickDown, this);
}
 
void ControlEditor::LeftClickDown(wxMouseEvent& event)
{
    AddElement(static_cast<ControlElementButtonID>(event.GetId()));
    event.Skip();
}
 
//void ControlEditor::SetViewport()
//{
//    glClearColor(1.0, 1.0, 1.0, 1.0);  // White background.
//    glClear(GL_COLOR_BUFFER_BIT);
//    glDisable(GL_DEPTH_TEST);
//    glDisable(GL_TEXTURE_2D);
//    glEnable(GL_COLOR_MATERIAL);
//    glEnable(GL_BLEND);
//    glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
//    glEnable(GL_LINE_SMOOTH);
//
//    double width = static_cast<double>(m_glCanvas->GetSize().x) - 1;
//    double height = static_cast<double>(m_glCanvas->GetSize().y) - 1;
//
//    // Viewport fit the screen.
//    glViewport(0, 0, width, height);
//
//    glMatrixMode(GL_PROJECTION);
//    glLoadIdentity();
//    gluOrtho2D(0.0, width, height, 0.0);
//
//    glMatrixMode(GL_MODELVIEW);
//    glLoadIdentity();
//}
 
void ControlEditor::AddElement(ControlElementButtonID id)
{
    switch (id) {
    case ControlElementButtonID::ID_IO: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto io = std::make_shared<IOControl>(m_ioFlags, GetNextID());
        m_elementList.push_back(io);
    } break;
    case ControlElementButtonID::ID_TF: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto tf = std::make_shared<TransferFunction>(GetNextID());
        m_elementList.push_back(tf);
    } break;
    case ControlElementButtonID::ID_SUM: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto sum = std::make_shared<Sum>(GetNextID());
        m_elementList.push_back(sum);
    } break;
    case ControlElementButtonID::ID_CONST: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto constant = std::make_shared<Constant>(GetNextID());
        m_elementList.push_back(constant);
    } break;
    case ControlElementButtonID::ID_LIMITER: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto limiter = std::make_shared<Limiter>(GetNextID());
        m_elementList.push_back(limiter);
    } break;
    case ControlElementButtonID::ID_GAIN: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto gain = std::make_shared<Gain>(GetNextID());
        m_elementList.push_back(gain);
    } break;
    case ControlElementButtonID::ID_MULT: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto mult = std::make_shared<Multiplier>(GetNextID());
        m_elementList.push_back(mult);
    } break;
    case ControlElementButtonID::ID_EXP: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto exp = std::make_shared<Exponential>(GetNextID());
        m_elementList.push_back(exp);
    } break;
    case ControlElementButtonID::ID_RATELIM: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto rateLim = std::make_shared<RateLimiter>(GetNextID());
        m_elementList.push_back(rateLim);
    } break;
    case ControlElementButtonID::ID_MATH_DIV: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto divider = std::make_shared<Divider>(GetNextID());
        m_elementList.push_back(divider);
    } break;
    case ControlElementButtonID::ID_MATH_EXPR: {
        m_mode = ControlEditorMode::MODE_INSERT;
        auto mathExpr = std::make_shared<MathExpression>(GetNextID());
        m_elementList.push_back(mathExpr);
    } break;
    }
}
 
void ControlEditor::OnPaint(wxPaintEvent& event)
{
    //wxPaintDC dc(m_glCanvas);
    //SetViewport();
    //
    //glScaled(m_camera->GetScale(), m_camera->GetScale(), 0.0);                          // Scale
    //glTranslated(m_camera->GetTranslation().m_x, m_camera->GetTranslation().m_y, 0.0);  // Translation
    //
    //for(auto it = m_connectionList.begin(), itEnd = m_connectionList.end(); it != itEnd; ++it) {
    //    ConnectionLine* line = *it;
    //    line->Draw(m_camera->GetTranslation(), m_camera->GetScale());
    //}
    //
    //for(auto it = m_elementList.begin(), itEnd = m_elementList.end(); it != itEnd; ++it) {
    //    Element* element = *it;
    //    element->Draw(m_camera->GetTranslation(), m_camera->GetScale());
    //}
    //
    //glLineWidth(1.0);
    //glColor4d(0.0, 0.5, 1.0, 1.0);
    //glBegin(GL_LINE_LOOP);
    //glVertex2d(m_selectionRect.m_x, m_selectionRect.m_y);
    //glVertex2d(m_selectionRect.m_x, m_selectionRect.m_y + m_selectionRect.m_height);
    //glVertex2d(m_selectionRect.m_x + m_selectionRect.m_width, m_selectionRect.m_y + m_selectionRect.m_height);
    //glVertex2d(m_selectionRect.m_x + m_selectionRect.m_width, m_selectionRect.m_y);
    //glEnd();
    //glColor4d(0.0, 0.5, 1.0, 0.3);
    //glBegin(GL_QUADS);
    //glVertex2d(m_selectionRect.m_x, m_selectionRect.m_y);
    //glVertex2d(m_selectionRect.m_x, m_selectionRect.m_y + m_selectionRect.m_height);
    //glVertex2d(m_selectionRect.m_x + m_selectionRect.m_width, m_selectionRect.m_y + m_selectionRect.m_height);
    //glVertex2d(m_selectionRect.m_x + m_selectionRect.m_width, m_selectionRect.m_y);
    //glEnd();
    //
    //glFlush();  // Sends all pending information directly to the GPU.
    //m_glCanvas->SwapBuffers();
    //event.Skip();
    wxBufferedPaintDC dc(m_cePanel);
    dc.Clear();
    wxGraphicsContext* gc = wxGraphicsContext::Create(dc);
 
    // Draw
    if (gc) {
 
        gc->Scale(m_camera->GetScale(), m_camera->GetScale());
        gc->Translate(m_camera->GetTranslation().m_x, m_camera->GetTranslation().m_y);
 
        for (auto line : m_connectionList) {
            //ConnectionLine* line = *it;
            line->DrawDC(m_camera->GetTranslation(), m_camera->GetScale(), gc);
        }
 
        for (auto element : m_elementList) {
            element->DrawDC(m_camera->GetTranslation(), m_camera->GetScale(), gc);
        }
 
        // Selection rectangle
        gc->SetPen(wxPen(wxColour(0, 125, 255, 255)));
        gc->SetBrush(wxBrush(wxColour(0, 125, 255, 125)));
        gc->DrawRectangle(m_selectionRect.m_x, m_selectionRect.m_y, m_selectionRect.m_width, m_selectionRect.m_height);
 
        delete gc;
    }
    event.Skip();
}
 
void ControlEditor::OnDoubleClick(wxMouseEvent& event)
{
    wxPoint2DDouble clickPoint = event.GetPosition();
    bool redraw = false;
 
    if (m_mode == ControlEditor::ControlEditorMode::MODE_EDIT) {
        for (auto& element : m_elementList) {
            if (element->Contains(m_camera->ScreenToWorld(clickPoint))) {
                element->ShowForm(this, element.get());
                CheckConnections();
                auto childList = element->GetChildList();
                for (auto itC = childList.begin(), itEndC = childList.end(); itC != itEndC; ++itC) {
                    ConnectionLine* line = static_cast<ConnectionLine*>(*itC);
                    line->UpdatePoints();
                }
                redraw = true;
            }
        }
    }
 
    if (redraw) Redraw();
}
 
void ControlEditor::OnLeftClickDown(wxMouseEvent& event)
{
    wxPoint2DDouble clickPoint = event.GetPosition();
    bool foundElement = false;
 
    if (m_mode == ControlEditorMode::MODE_INSERT) {
        m_mode = ControlEditor::ControlEditorMode::MODE_EDIT;
    }
    else {
        for (auto& element : m_elementList) {
            bool foundNode = false;
            auto nodeList = element->GetNodeList();
            for (auto itN = nodeList.begin(), itNEnd = nodeList.end(); itN != itNEnd; ++itN) {
                Node* node = *itN;
                if (node->Contains(m_camera->ScreenToWorld(clickPoint))) {
                    m_mode = ControlEditorMode::MODE_INSERT_LINE;
                    auto line = std::make_shared<ConnectionLine>(node, GetNextID());
                    m_connectionList.push_back(line);
                    element->AddChild(line.get());
                    line->AddParent(element.get());
                    foundElement = true;
                    foundNode = true;
                }
            }
 
            if (!foundNode) {
                // Set movement initial position (not necessarily will be moved).
                element->StartMove(m_camera->ScreenToWorld(clickPoint));
 
                // Click in an element.
                if (element->Contains(m_camera->ScreenToWorld(clickPoint))) {
                    if (!foundElement) {
                        element->SetSelected();
                        foundElement = true;
                    }
                    m_mode = ControlEditor::ControlEditorMode::MODE_MOVE_ELEMENT;
                }
            }
        }
        if (m_mode != ControlEditorMode::MODE_INSERT_LINE) {
            for (auto& line : m_connectionList) {
                line->StartMove(m_camera->ScreenToWorld(clickPoint));
                if (line->Contains(m_camera->ScreenToWorld(clickPoint))) {
                    line->SetSelected();
                    foundElement = true;
                    m_mode = ControlEditorMode::MODE_MOVE_LINE;
                }
            }
        }
    }
 
    if (!foundElement) {
        m_mode = ControlEditorMode::MODE_SELECTION_RECT;
        m_startSelRect = m_camera->ScreenToWorld(clickPoint);
    }
 
    Redraw();
    event.Skip();
}
 
void ControlEditor::OnLeftClickUp(wxMouseEvent& event)
{
    bool foundNode = false;
    for (auto& element : m_elementList) {
        if (m_mode == ControlEditorMode::MODE_INSERT_LINE) {
            auto nodeList = element->GetNodeList();
            for (auto node : nodeList) {
                if (node->Contains(m_camera->ScreenToWorld(event.GetPosition()))) {
                    //ConnectionLine* line = *(m_connectionList.end() - 1);
                    auto line = m_connectionList.back();
                    if (line->AppendNode(node, element.get())) {
                        line->AddParent(element.get());
                        element->AddChild(line.get());
                        line->UpdatePoints();
                        m_mode = ControlEditor::ControlEditorMode::MODE_EDIT;
                        foundNode = true;
                    }
                }
            }
        }
        else if (m_mode == ControlEditorMode::MODE_SELECTION_RECT) {
            if (element->Intersects(m_selectionRect)) {
                element->SetSelected();
            }
            else if (!event.ControlDown()) {
                element->SetSelected(false);
            }
        }
        else if (!event.ControlDown()) {
            if (!element->Contains(m_camera->ScreenToWorld(event.GetPosition()))) { element->SetSelected(false); }
        }
    }
 
    auto& lastLine = m_connectionList.back();
    for (auto& cLine : m_connectionList) {
        if (m_mode == ControlEditorMode::MODE_INSERT_LINE && !foundNode && cLine.get() != lastLine.get()) {
            if (cLine->Contains(m_camera->ScreenToWorld(event.GetPosition()))) {
                //ConnectionLine* iLine = *(m_connectionList.end() - 1);
                auto iLine = m_connectionList.back();
                if (iLine->SetParentLine(cLine.get())) {
                    cLine->AddChild(iLine.get());
                    iLine->UpdatePoints();
                    m_mode = ControlEditor::ControlEditorMode::MODE_EDIT;
                    foundNode = true;
                }
            }
        }
        else if (m_mode == ControlEditorMode::MODE_SELECTION_RECT) {
            if (cLine->Intersects(m_selectionRect)) {
                cLine->SetSelected();
            }
            else if (!event.ControlDown()) {
                cLine->SetSelected(false);
            }
        }
        else if (!event.ControlDown()) {
            if (!cLine->Contains(m_camera->ScreenToWorld(event.GetPosition()))) { cLine->SetSelected(false); }
        }
    }
 
    m_selectionRect = wxRect2DDouble(0, 0, 0, 0);
 
    if (m_mode == ControlEditorMode::MODE_INSERT_LINE && !foundNode) {
        //ConnectionLine* cLine = *(m_connectionList.end() - 1);
        auto cLine = m_connectionList.back();
        // Free nodes
        auto nodeList = cLine->GetNodeList();
        for (auto itN = nodeList.begin(), itEndN = nodeList.end(); itN != itEndN; ++itN) {
            Node* node = *itN;
            node->SetConnected(false);
        }
        // Remove the associated child from parents.
        auto parentList = cLine->GetParentList();
        for (auto it = parentList.begin(), itEnd = parentList.end(); it != itEnd; ++it) {
            Element* element = *it;
            element->RemoveChild(cLine.get());
        }
        m_connectionList.pop_back();
        //if (cLine) delete cLine;
        m_mode = ControlEditor::ControlEditorMode::MODE_EDIT;
    }
    else if (m_mode != ControlEditorMode::MODE_INSERT) {
        m_mode = ControlEditor::ControlEditorMode::MODE_EDIT;
    }
 
    Redraw();
    event.Skip();
}
 
void ControlEditor::OnMiddleDown(wxMouseEvent& event)
{
    // Set to drag mode.
    switch (m_mode) {
    case ControlEditorMode::MODE_INSERT: {
        m_mode = ControlEditorMode::MODE_DRAG_INSERT;
    } break;
    case ControlEditorMode::MODE_PASTE: {
        m_mode = ControlEditorMode::MODE_DRAG_PASTE;
    } break;
    default: {
        m_mode = ControlEditorMode::MODE_DRAG;
    } break;
    }
    m_camera->StartTranslation(m_camera->ScreenToWorld(event.GetPosition()));
}
 
void ControlEditor::OnMiddleUp(wxMouseEvent& event)
{
    switch (m_mode) {
    case ControlEditorMode::MODE_DRAG_INSERT: {
        m_mode = ControlEditorMode::MODE_INSERT;
    } break;
    case ControlEditorMode::MODE_DRAG_PASTE: {
        m_mode = ControlEditorMode::MODE_PASTE;
    } break;
    case ControlEditorMode::MODE_INSERT:
    case ControlEditorMode::MODE_PASTE: {
        // Does nothing.
    } break;
    default: {
        m_mode = ControlEditor::ControlEditorMode::MODE_EDIT;
    } break;
    }
}
 
void ControlEditor::OnMouseMotion(wxMouseEvent& event)
{
    wxPoint2DDouble clickPoint = event.GetPosition();
    bool redraw = false;
 
    switch (m_mode) {
    case ControlEditorMode::MODE_INSERT: {
        //Element* newElement = *(m_elementList.end() - 1);  // Get the last element in the list.
        auto newElement = m_elementList.back();
        newElement->Move(m_camera->ScreenToWorld(clickPoint));
        redraw = true;
    } break;
    case ControlEditorMode::MODE_INSERT_LINE: {
        //ConnectionLine* line = *(m_connectionList.end() - 1);
        auto line = m_connectionList.back();
        line->SetTemporarySecondPoint(m_camera->ScreenToWorld(clickPoint));
        line->UpdatePoints();
        redraw = true;
    } break;
    case ControlEditorMode::MODE_DRAG:
    case ControlEditorMode::MODE_DRAG_INSERT:
    case ControlEditorMode::MODE_DRAG_PASTE: {
        m_camera->SetTranslation(clickPoint);
        redraw = true;
    } break;
    case ControlEditor::ControlEditorMode::MODE_MOVE_ELEMENT: {
        for (auto& element : m_elementList) {
            if (element->IsSelected()) {
                element->Move(m_camera->ScreenToWorld(clickPoint));
                auto childList = element->GetChildList();
                for (auto itC = childList.begin(), itEndC = childList.end(); itC != itEndC; itC++) {
                    ConnectionLine* line = static_cast<ConnectionLine*>(*itC);
                    line->UpdatePoints();
                }
                redraw = true;
            }
        }
    } break;
    case ControlEditorMode::MODE_MOVE_LINE: {
        for (auto& line : m_connectionList) {
            if (line->IsSelected()) {
                line->Move(m_camera->ScreenToWorld(clickPoint));
                redraw = true;
            }
        }
    } break;
    case ControlEditorMode::MODE_SELECTION_RECT: {
        wxPoint2DDouble currentPos = m_camera->ScreenToWorld(clickPoint);
        double x, y, w, h;
        if (currentPos.m_x < m_startSelRect.m_x) {
            x = currentPos.m_x;
            w = m_startSelRect.m_x - currentPos.m_x;
        }
        else {
            x = m_startSelRect.m_x;
            w = currentPos.m_x - m_startSelRect.m_x;
        }
        if (currentPos.m_y < m_startSelRect.m_y) {
            y = currentPos.m_y;
            h = m_startSelRect.m_y - currentPos.m_y;
        }
        else {
            y = m_startSelRect.m_y;
            h = currentPos.m_y - m_startSelRect.m_y;
        }
 
        m_selectionRect = wxRect2DDouble(x, y, w, h);
        redraw = true;
    } break;
    default:
        break;
    }
 
    if (redraw) Redraw();
    event.Skip();
}
 
void ControlEditor::OnScroll(wxMouseEvent& event)
{
    if (event.GetWheelRotation() > 0)
        m_camera->SetScale(event.GetPosition(), +0.05);
    else
        m_camera->SetScale(event.GetPosition(), -0.05);
 
    Redraw();
}
 
void ControlEditor::OnIdle(wxIdleEvent& event)
{
    //if(m_justOpened) {
    //    this->Raise();
    //
    //    // Update all text elements
    //    m_justOpened = false;
    //    for(auto it = m_elementList.begin(), itEnd = m_elementList.end(); it != itEnd; ++it) {
    //        ControlElement* element = *it;
    //        if(!element->UpdateText()) m_justOpened = true;
    //    }
    //    Redraw();
    //}
}
void ControlEditor::OnKeyDown(wxKeyEvent& event)
{
    char key = event.GetUnicodeKey();
    if (key != WXK_NONE) {
        switch (key) {
        case WXK_DELETE:  // Delete selected elements.
        {
            DeleteSelectedElements();
        } break;
        case 'R':  // Rotate the selected elements.
        {
            RotateSelectedElements(event.GetModifiers() != wxMOD_SHIFT);
        } break;
        case 'L': {
            // tests
        } break;
        }
    }
}
 
void ControlEditor::RotateSelectedElements(bool clockwise)
{
    for (auto& element : m_elementList) {
        if (element->IsSelected()) {
            element->Rotate(clockwise);
            auto childList = element->GetChildList();
            for (auto itC = childList.begin(), itEndC = childList.end(); itC != itEndC; itC++) {
                ConnectionLine* line = static_cast<ConnectionLine*>(*itC);
                line->UpdatePoints();
            }
        }
    }
    Redraw();
}
 
void ControlEditor::DeleteSelectedElements()
{
    for (auto it = m_elementList.begin(); it != m_elementList.end();) {
        Element* element = it->get();
        if (element->IsSelected()) {
            // Remove child/parent.
            auto childList = element->GetChildList();
            for (auto child : childList) {
                // The child is always a connection line, but check it.
                //ConnectionLine* child = static_cast<ConnectionLine*>(*itC);
                if (auto childLine = dynamic_cast<ConnectionLine*>(child)) {
                    // Delete the connection line.
                    for (auto itCo = m_connectionList.begin(); itCo != m_connectionList.end(); ) {
                        if (itCo->get() == childLine)
                            itCo = DeleteLineFromList(itCo);
                        else
                            ++itCo;
                    }
                }
 
            }
            m_elementList.erase(it);
            //if (element) delete element;
        }
        else {
            ++it;
        }
    }
 
    for (auto it = m_connectionList.begin(); it != m_connectionList.end(); ) {
        ConnectionLine* line = it->get();
        if (line->IsSelected())
            it = DeleteLineFromList(it);
        else
            ++it;
    }
    Redraw();
}
 
std::vector< std::shared_ptr<ConnectionLine> >::iterator ControlEditor::DeleteLineFromList(std::vector< std::shared_ptr<ConnectionLine> >::iterator& it)
{
    ConnectionLine* cLine = it->get();
 
    // Delete children recursively
    auto childList = cLine->GetLineChildList();
    for (auto child : childList) {
        for (auto itL = m_connectionList.begin(); itL != m_connectionList.end(); ) {
            if (itL->get() == child)
                itL = DeleteLineFromList(itL);
            else
                ++itL;
        }
    }
    // Remove parents
    auto parentList = cLine->GetParentList();
    for (auto parent : parentList) {
        if (parent) parent->RemoveChild(cLine);
    }
    if (cLine->GetParentLine()) cLine->GetParentLine()->RemoveChild(cLine);
 
    // Free nodes
    auto nodeList = cLine->GetNodeList();
    for (auto node : nodeList)
        node->SetConnected(false);
 
    return m_connectionList.erase(it);
}
 
void ControlEditor::CheckConnections()
{
    for (auto it = m_connectionList.begin(); it != m_connectionList.end(); ++it) {
        ConnectionLine* cLine = it->get();
        if (cLine->GetType() == ConnectionLine::ConnectionLineType::ELEMENT_ELEMENT) {
            if (cLine->GetParentList().size() < 2) { it = DeleteLineFromList(it); }
        }
        else if (cLine->GetParentList().size() < 1) {
            it = DeleteLineFromList(it);
        }
    }
}
 
void ControlEditor::OnExportClick(wxCommandEvent& event)
{
    FileHanding fileHandling(this);
 
    wxFileDialog saveFileDialog(this, _("Save CTL file"), "", "", "CTL files (*.ctl)|*.ctl",
        wxFD_SAVE | wxFD_OVERWRITE_PROMPT);
    if (saveFileDialog.ShowModal() == wxID_CANCEL) return;
 
    fileHandling.SaveControl(saveFileDialog.GetPath());
    wxFileName fileName(saveFileDialog.GetPath());
    event.Skip();
}
 
void ControlEditor::OnImportClick(wxCommandEvent& event)
{
    wxFileDialog openFileDialog(this, _("Open CTL file"), "", "", "CTL files (*.ctl)|*.ctl",
        wxFD_OPEN | wxFD_FILE_MUST_EXIST);
    if (openFileDialog.ShowModal() == wxID_CANCEL) return;
 
    wxFileName fileName(openFileDialog.GetPath());
 
    FileHanding fileHandling(this);
    if (!fileHandling.OpenControl(fileName, m_elementList, m_connectionList)) {
        wxMessageDialog msgDialog(this, _("It was not possible to open the selected file."), _("Error"),
            wxOK | wxCENTRE | wxICON_ERROR);
        msgDialog.ShowModal();
    }
    Redraw();
    event.Skip();
}
 
void ControlEditor::OnTestClick(wxCommandEvent& event)
{
    // Reset colour list position for the test plot.
    m_itColourList = --m_colourList.end();
 
    std::vector<IOControl*> ioList;
 
    for (auto& element : m_elementList) {
        if (auto io = dynamic_cast<IOControl*>(element.get())) {
            ioList.push_back(io);
        }
    }
 
    ControlSystemTest csTest(this, ioList, &m_inputType, &m_startTime, &m_slope, &m_timeStep, &m_simTime);
    if (csTest.ShowModal() == wxID_OK) {
        double printStep = 1e-3;
        double pdbStep = 1e-1;
 
        struct InputData {
            wxString name;
            std::vector<double> values;
        };
 
        std::vector<InputData> inputList;
 
        // Store real flags for restoring after the test and set initial value.
        std::vector<IOControl::IOFlags> realFlagValue;
        for (auto* io : ioList) {
            realFlagValue.push_back(io->GetValue());
            if (io->GetType() == Node::NodeType::NODE_OUT) {
                SimTestData testData = io->GetSimTestData();
                io->SetValue(IOControl::IN_TEST);
                io->SetTestValue(testData.initialValue);
                inputList.push_back({ io->GetName(), {} });
            }
        }
 
        wxProgressDialog pbd(_("Test"), _("Initializing..."), 100, this,
            wxPD_APP_MODAL | wxPD_AUTO_HIDE | wxPD_CAN_ABORT | wxPD_SMOOTH);
        ControlElementSolver solver(this, m_timeStep, 1e-5);
        solver.InitializeValues(false);
        if (solver.IsOK()) {
            bool simStopped = false;
            double currentTime = 0.0;
            double printTime = 0.0;
            double pdbTime = 0.0;
            std::vector<double> time;
            std::vector<double> solution;
 
            while (currentTime <= m_simTime) {
                for (auto* io : ioList) {
                    SimTestData testData = io->GetSimTestData();
                    if (currentTime >= testData.startTime) {
                        switch (testData.type) {
                        case 0: {
                            io->SetTestValue(testData.slope);
                        } break;
                        case 1: {
                            io->SetTestValue(testData.slope * (currentTime - testData.startTime));
                        } break;
                        case 2: {
                            io->SetTestValue(testData.slope * std::pow(currentTime - testData.startTime, 2));
                        } break;
                        default: {
                            io->SetTestValue(0.0);
                            break;
                        }
                        }
                    }
                }
 
                solver.SetCurrentTime(currentTime);
                solver.SolveNextStep();
                if (!solver.IsOK()) {
                    wxString msg = wxString::Format(_("Failed to solve the control system.\n%s"), solver.GetErrorMessage());
                    wxMessageDialog msgDialog(this, msg, _("Error"), wxOK | wxCENTRE | wxICON_ERROR);
                    msgDialog.ShowModal();
                    simStopped = true;
                    currentTime = m_simTime;
                }
 
                if (printTime >= printStep) {
                    time.push_back(currentTime);
                    solution.push_back(solver.GetLastSolution());
                    //inputV.push_back(input);
                    int i = 0;
                    for (auto* io : ioList) {
                        if (io->GetType() == Node::NodeType::NODE_OUT) {
                            inputList[i].values.push_back(io->GetTestValue());
                            i++;
                        }
                    }
                    printTime = 0.0;
                }
 
                if (pdbTime > pdbStep) {
                    if (!pbd.Update((currentTime / m_simTime) * 100, wxString::Format("Time = %.2fs", currentTime))) {
                        pbd.Update(100);
                        simStopped = true;
                        currentTime = m_simTime;
                    }
                    pdbTime = 0.0;
                }
 
                printTime += m_timeStep;
                currentTime += m_timeStep;
                pdbTime += m_timeStep;
            }
 
            if (!simStopped) {
                std::vector<ElementPlotData> epdList;
                ElementPlotData curveData(_("I/O"), ElementPlotData::CurveType::CT_TEST);
                int i = 0;
                for (const auto& input : inputList) {
                    curveData.AddData(input.values, input.name);
                    curveData.SetPlot(i);
                    curveData.SetColour(i, GetNextColour());
                    i++;
                }
                //curveData.AddData(inputV, _("Input"));
                curveData.AddData(solution, _("Output"));
                curveData.SetPlot(i);
                curveData.SetColour(i, GetNextColour());
 
                //curveData.SetPlot(0);
                //curveData.SetColour(0, *wxRED);
                //curveData.SetPlot(1);
                //curveData.SetColour(1, *wxBLUE);
 
                epdList.push_back(curveData);
 
                ChartView* cView = new ChartView(this, epdList, time, static_cast<PlotLib>(m_plotLib));
                cView->Show();
                cView->UpdatePlot();
            }
        }
        else {
            wxMessageDialog msgDialog(
                this,
                wxString::Format(_("Failed to solve the control system.\n%s"), solver.GetErrorMessage()),
                _("Error"),
                wxOK | wxCENTRE | wxICON_ERROR
            );
 
            msgDialog.ShowModal();
        }
 
        // Restore real flags.
        for (size_t i = 0; i < ioList.size(); i++) {
            if (ioList[i]->GetType() == Node::NodeType::NODE_OUT)
                ioList[i]->SetValue(realFlagValue[i]);
        }
    }
}
 
void ControlEditor::OnClose(wxCloseEvent& event)
{
    if (m_ctrlContainer) { m_ctrlContainer->FillContainer(this); }
    event.Skip();
}
 
int ControlEditor::GetNextID()
{
    int id = 0;
    for (auto& element : m_elementList) {
        if (element->GetID() > id) id = element->GetID();
    }
    for (auto& line : m_connectionList) {
        if (line->GetID() > id) id = line->GetID();
    }
    id++;
    return id;
}
 
void ControlEditor::BuildColourList()
{
    m_colourList.push_back(wxColour(255, 30, 0));
    m_colourList.push_back(wxColour(0, 30, 255));
    m_colourList.push_back(wxColour(0, 128, 0));
    m_colourList.push_back(wxColour(100, 100, 100));
    m_colourList.push_back(wxColour(255, 128, 0));
    m_colourList.push_back(wxColour(128, 0, 255));
    m_colourList.push_back(wxColour(0, 255, 128));
    m_colourList.push_back(wxColour(255, 255, 0));
    m_colourList.push_back(wxColour(255, 0, 255));
    m_colourList.push_back(wxColour(0, 255, 255));
    m_colourList.push_back(wxColour(128, 255, 0));
    m_colourList.push_back(wxColour(255, 0, 128));
    m_colourList.push_back(wxColour(0, 128, 255));
    m_colourList.push_back(wxColour(128, 128, 128));
    m_colourList.push_back(*wxBLACK);
    m_itColourList = --m_colourList.end();
}
 
wxColour ControlEditor::GetNextColour()
{
    if (*m_itColourList == *wxBLACK)
        m_itColourList = m_colourList.begin();
    else
        ++m_itColourList;
 
    return *m_itColourList;
}