Gain.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 "Gain.h"
#include "../../forms/GainForm.h"
 
Gain::Gain(int id) : ControlElement(id)
{
    m_triPts.resize(3);
    SetValue(m_value);
    Node* nodeIn = new Node(m_position + wxPoint2DDouble(-m_width / 2, 0), Node::NodeType::NODE_IN, m_borderSize);
    nodeIn->StartMove(m_position);
    Node* nodeOut = new Node(m_position + wxPoint2DDouble(m_width / 2, 0), Node::NodeType::NODE_OUT, m_borderSize);
    nodeOut->SetAngle(180.0);
    nodeOut->StartMove(m_position);
    m_nodeList.push_back(nodeIn);
    m_nodeList.push_back(nodeOut);
}
 
Gain::~Gain()
{
    //if(m_glText) delete m_glText;
    if (m_gcText) delete m_gcText;
    for (auto& node : m_nodeList) if (node) delete node;
    m_nodeList.clear();
}
 
void Gain::DrawDC(GUIColour* guiColour, wxPoint2DDouble translation, double scale, wxGraphicsContext* gc) const
{
    if (m_selected) {
        gc->SetPen(*wxTRANSPARENT_PEN);
        gc->SetBrush(wxBrush(guiColour->selection));
        double borderSize = (m_borderSize * 2.0 + 1.0) / scale;
        std::vector<wxPoint2DDouble> m_triSelectedPts;
        if (m_angle == 0.0) {
            m_triSelectedPts.push_back(m_triPts[0] - wxPoint2DDouble(borderSize / 2, borderSize / 1.5));
            m_triSelectedPts.push_back(m_triPts[1] - wxPoint2DDouble(borderSize / 2, -borderSize / 1.5));
            m_triSelectedPts.push_back(m_triPts[2] - wxPoint2DDouble(-borderSize, 0));
        }
        else if (m_angle == 90.0) {
            m_triSelectedPts.push_back(m_triPts[0] - wxPoint2DDouble(borderSize / 1.5, borderSize / 2));
            m_triSelectedPts.push_back(m_triPts[1] - wxPoint2DDouble(-borderSize / 1.5, borderSize / 2));
            m_triSelectedPts.push_back(m_triPts[2] - wxPoint2DDouble(0, -borderSize));
        }
        else if (m_angle == 180.0) {
            m_triSelectedPts.push_back(m_triPts[0] - wxPoint2DDouble(borderSize, 0));
            m_triSelectedPts.push_back(m_triPts[1] - wxPoint2DDouble(-borderSize / 2, borderSize / 1.5));
            m_triSelectedPts.push_back(m_triPts[2] - wxPoint2DDouble(-borderSize / 2, -borderSize / 1.5));
        }
        else if (m_angle == 270.0) {
            m_triSelectedPts.push_back(m_triPts[0] - wxPoint2DDouble(0, borderSize));
            m_triSelectedPts.push_back(m_triPts[1] - wxPoint2DDouble(-borderSize / 1.5, -borderSize / 2));
            m_triSelectedPts.push_back(m_triPts[2] - wxPoint2DDouble(borderSize / 1.5, -borderSize / 2));
        }
        DrawDCTriangle(m_triSelectedPts, gc);
    }
 
    gc->SetPen(wxPen(guiColour->enabled, 1));
    gc->SetBrush(guiColour->background);
    DrawDCTriangle(m_triPts, gc);
 
    // Plot number.
    if (m_angle == 0.0)
        m_gcText->Draw(m_position + wxPoint2DDouble(-m_width / 2 + 2 + m_borderSize, -m_gcText->GetHeight() / 2), gc, 0, guiColour->text);
    else if (m_angle == 90.0)
        m_gcText->Draw(m_position + wxPoint2DDouble(-m_gcText->GetWidth() / 2, -m_height / 2 + 2 + m_borderSize), gc, 0, guiColour->text);
    else if (m_angle == 180.0)
        m_gcText->Draw(m_position + wxPoint2DDouble(m_width / 2 - m_gcText->GetWidth() - 2 - m_borderSize, -m_gcText->GetHeight() / 2), gc, 0, guiColour->text);
    else if (m_angle == 270.0)
        m_gcText->Draw(m_position + wxPoint2DDouble(-m_gcText->GetWidth() / 2, m_height / 2 - m_gcText->GetHeight() - 2 - m_borderSize), gc, 0, guiColour->text);
 
    gc->SetPen(*wxTRANSPARENT_PEN);
    gc->SetBrush(guiColour->enabled);
    DrawDCNodes(gc);
}
 
bool Gain::ShowForm(wxWindow* parent, Element* element, wxWindow* workspace)
{
    GainForm form(parent, this);
    form.CenterOnParent();
    if (form.ShowModal() == wxID_OK) {
        return true;
    }
    return false;
}
 
void Gain::Rotate(bool clockwise)
{
    if (clockwise)
        m_angle += 90.0;
    else
        m_angle -= 90.0;
    if (m_angle >= 360.0)
        m_angle = 0.0;
    else if (m_angle < 0)
        m_angle = 270.0;
 
    UpdatePoints();
 
    for (auto it = m_nodeList.begin(), itEnd = m_nodeList.end(); it != itEnd; ++it) {
        Node* node = *it;
        node->Rotate(clockwise);
    }
}
 
void Gain::SetValue(double value)
{
    m_value = value;
    wxString text = "";
    if (std::abs(m_value) > 1e3 || std::abs(m_value) < 1e-3)
        text = wxString::Format("%g", m_value);
    else
        text = StringFromDouble(m_value);
 
    if (m_gcText)
        m_gcText->SetText(text);
    else
        m_gcText = new GCText(text);
 
    m_width = m_gcText->GetWidth() + 18 + 2 * m_borderSize;
    m_height = m_gcText->GetHeight() + 18 + 2 * m_borderSize;
 
    if (m_width > m_height)
        m_height = m_width;
    else
        m_width = m_height;
 
    SetPosition(m_position);  // Update rectangle.
 
    UpdatePoints();
}
 
void Gain::UpdatePoints()
{
    if (m_nodeList.size() != 0) {
        if (m_angle == 0.0) {
            m_triPts[0] = m_rect.GetLeftTop() + wxPoint2DDouble(m_borderSize, m_borderSize);
            m_triPts[1] = m_rect.GetLeftBottom() + wxPoint2DDouble(m_borderSize, -m_borderSize);
            m_triPts[2] = m_position + wxPoint2DDouble(m_width / 2 - m_borderSize, 0);
            m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(-m_width / 2, 0));
            m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(m_width / 2 - 2, 0));
        }
        else if (m_angle == 90.0) {
            m_triPts[0] = m_rect.GetLeftTop() + wxPoint2DDouble(m_borderSize, m_borderSize);
            m_triPts[1] = m_rect.GetRightTop() + wxPoint2DDouble(-m_borderSize, m_borderSize);
            m_triPts[2] = m_position + wxPoint2DDouble(0, m_height / 2 - m_borderSize);
            m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(0, -m_height / 2));
            m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(0, m_height / 2 - 2));
        }
        else if (m_angle == 180.0) {
            m_triPts[0] = m_position + wxPoint2DDouble(-m_width / 2 + m_borderSize, 0);
            m_triPts[1] = m_rect.GetRightTop() + wxPoint2DDouble(-m_borderSize, m_borderSize);
            m_triPts[2] = m_rect.GetRightBottom() + wxPoint2DDouble(-m_borderSize, -m_borderSize);
            m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(m_width / 2, 0));
            m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(-m_width / 2 + 2, 0));
        }
        else if (m_angle == 270.0) {
            m_triPts[0] = m_position + wxPoint2DDouble(0, -m_height / 2 + m_borderSize);
            m_triPts[1] = m_rect.GetRightBottom() + wxPoint2DDouble(-m_borderSize, -m_borderSize);
            m_triPts[2] = m_rect.GetLeftBottom() + wxPoint2DDouble(m_borderSize, -m_borderSize);
            m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(0, m_height / 2));
            m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(0, -m_height / 2 + 2));
        }
    }
}
 
void Gain::Move(wxPoint2DDouble position)
{
    SetPosition(m_movePos + position - m_moveStartPt);
    UpdatePoints();
}
 
void Gain::SetFont(wxFont& font)
{
    if (m_gcText) m_gcText->SetFont(font);
    UpdateText();
}
 
bool Gain::Solve(double* input, double timeStep)
{
    if (!input) {
        m_output = 0.0;
        return true;
    }
    m_output = input[0] * m_value;
    return true;
}
 
Element* Gain::GetCopy()
{
    Gain* copy = new Gain(*this);
    copy->m_gcText = m_gcText ? m_gcText->GetCopy() : nullptr;
    return copy;
}
 
bool Gain::UpdateText()
{
    SetValue(m_value);
    //if (!m_glText->IsTextureOK()) return false;
    return true;
}
 
rapidxml::xml_node<>* Gain::SaveElement(rapidxml::xml_document<>& doc, rapidxml::xml_node<>* elementListNode)
{
    auto elementNode = XMLParser::AppendNode(doc, elementListNode, "Gain");
    XMLParser::SetNodeAttribute(doc, elementNode, "ID", m_elementID);
 
    SaveCADProperties(doc, elementNode);
    SaveControlNodes(doc, elementNode);
 
    // Element properties
    auto value = XMLParser::AppendNode(doc, elementNode, "Value");
    XMLParser::SetNodeValue(doc, value, m_value);
 
    return elementNode;
}
 
bool Gain::OpenElement(rapidxml::xml_node<>* elementNode)
{
    if (!OpenCADProperties(elementNode)) return false;
    if (!OpenControlNodes(elementNode)) return false;
 
    // Element properties
    double value = XMLParser::GetNodeValueDouble(elementNode, "Value");
    SetValue(value);
    return true;
}