Merge pull request #28 from Thales1330/wip/generic-controllers

Wip generic controllers. Chart view implementation required, creating new branch....

1 files changed, 250 insertions, 0 deletions

diff --git a/Project/TransferFunction.cpp b/Project/TransferFunction.cpp
new file mode 100644
index 0000000..9c8ffea
--- /dev/null
+++ b/Project/TransferFunction.cpp
@@ -0,0 +1,250 @@
+#include "TransferFunction.h"
+#include "TransferFunctionForm.h"
+
+TransferFunction::TransferFunction(int id) : ControlElement(id)
+{
+    // Superscript unicode numbers
+    m_supNumber[0] = L'\u2070';
+    m_supNumber[1] = L'\u00B9';
+    m_supNumber[2] = L'\u00B2';
+    m_supNumber[3] = L'\u00B3';
+    m_supNumber[4] = L'\u2074';
+    m_supNumber[5] = L'\u2075';
+    m_supNumber[6] = L'\u2076';
+    m_supNumber[7] = L'\u2077';
+    m_supNumber[8] = L'\u2078';
+    m_supNumber[9] = L'\u2079';
+
+    m_numerator.clear();
+    m_numerator.push_back(1);
+    m_denominator.clear();
+    m_denominator.push_back(1);
+    m_denominator.push_back(1);
+    UpdateTFText();
+
+    Node* node1 = new Node(m_position + wxPoint2DDouble(-m_width / 2, 0), Node::NODE_IN, m_borderSize);
+    node1->StartMove(m_position);
+    Node* node2 = new Node(m_position + wxPoint2DDouble(m_width / 2, 0), Node::NODE_OUT, m_borderSize);
+    node2->SetAngle(180.0);
+    node2->StartMove(m_position);
+    m_nodeList.push_back(node1);
+    m_nodeList.push_back(node2);
+}
+
+TransferFunction::~TransferFunction() {}
+
+void TransferFunction::Draw(wxPoint2DDouble translation, double scale) const
+{
+    glLineWidth(1.0);
+    if(m_selected) {
+        glColor4dv(m_selectionColour.GetRGBA());
+        double borderSize = (m_borderSize * 2.0 + 1.0) / scale;
+        DrawRectangle(m_position, m_width + borderSize, m_height + borderSize);
+    }
+    glColor4d(1.0, 1.0, 1.0, 1.0);
+    DrawRectangle(m_position, m_width, m_height);
+    glColor4d(0.0, 0.0, 0.0, 1.0);
+    DrawRectangle(m_position, m_width, m_height, GL_LINE_LOOP);
+
+    std::vector<wxPoint2DDouble> linePts;
+    linePts.push_back(wxPoint2DDouble(m_position.m_x - m_width / 2 + 5 + m_borderSize, m_position.m_y));
+    linePts.push_back(wxPoint2DDouble(m_position.m_x + m_width / 2 - 5 - m_borderSize, m_position.m_y));
+    DrawLine(linePts);
+
+    DrawNodes();
+
+    glEnable(GL_TEXTURE_2D);
+    glColor4d(0.0, 0.0, 0.0, 1.0);
+    m_glStringNum->bind();
+    m_glStringNum->render(m_position.m_x, m_position.m_y - m_height / 4);
+    m_glStringDen->bind();
+    m_glStringDen->render(m_position.m_x, m_position.m_y + m_height / 4);
+    glDisable(GL_TEXTURE_2D);
+}
+
+void TransferFunction::SetText(wxString numerator, wxString denominator)
+{
+    wxFont font(m_fontSize, wxFONTFAMILY_DEFAULT, wxFONTSTYLE_NORMAL, wxFONTWEIGHT_NORMAL);
+    wxScreenDC dc;
+
+    if(m_glStringNum) {
+        delete m_glStringNum;
+        m_glStringNum = NULL;
+    }
+    m_glStringNum = new wxGLString(numerator);
+    m_glStringNum->setFont(font);
+    m_glStringNum->consolidate(&dc);
+
+    if(m_glStringDen) {
+        delete m_glStringDen;
+        m_glStringDen = NULL;
+    }
+    m_glStringDen = new wxGLString(denominator);
+    m_glStringDen->setFont(font);
+    m_glStringDen->consolidate(&dc);
+
+    double nWidth = m_glStringNum->getWidth() + 5 + m_borderSize;
+    double dWidth = m_glStringDen->getWidth() + 5 + m_borderSize;
+
+    m_width = nWidth > dWidth ? nWidth : dWidth;
+    m_height = m_glStringNum->getheight() + m_glStringDen->getheight() + 2 * m_borderSize;
+    SetPosition(m_position); // Update rect properly.
+}
+
+wxString TransferFunction::GetSuperscriptNumber(int number)
+{
+    wxString strNumber = wxString::Format("%d", number);
+    wxString superscriptStr = "";
+    for(int i = 0; i < (int)strNumber.length(); ++i) {
+        wxString digitStr = strNumber[i];
+        long digit = 0;
+        digitStr.ToLong(&digit);
+        superscriptStr += wxString(m_supNumber[digit]);
+    }
+    return superscriptStr;
+}
+
+void TransferFunction::GetTFString(wxString& numerator, wxString& denominator)
+{
+    numerator = "";
+    denominator = "";
+    int index = static_cast<int>(m_numerator.size()) - 1;
+    for(auto it = m_numerator.begin(), itEnd = m_numerator.end(); it != itEnd; ++it) {
+        double value = *it;
+        if(value != 0.0) {
+            wxString signal;
+            if(index == static_cast<int>(m_numerator.size()) - 1) {
+                if(value >= 0.0)
+                    signal += "";
+                else
+                    signal += "-";
+            } else {
+                if(value >= 0.0)
+                    signal += "+ ";
+                else
+                    signal += "- ";
+            }
+
+            if(index == 0) {
+                numerator += signal + StringFromDouble(std::abs(value), 0);
+                break;
+            } else if(index == 1) {
+                if(value == 1.0) {
+                    numerator += signal + "s";
+                } else {
+                    numerator += signal + StringFromDouble(std::abs(value), 0) + "s";
+                }
+            } else {
+                if(value == 1.0) {
+                    numerator += signal + "s" + GetSuperscriptNumber(index);
+                } else {
+                    numerator += signal + StringFromDouble(std::abs(value), 0) + "s" + GetSuperscriptNumber(index);
+                }
+            }
+            numerator += " ";
+        }
+        --index;
+    }
+
+    index = static_cast<int>(m_denominator.size()) - 1;
+    for(auto it = m_denominator.begin(), itEnd = m_denominator.end(); it != itEnd; ++it) {
+        double value = *it;
+        if(value != 0.0) {
+            wxString signal;
+            if(index == static_cast<int>(m_denominator.size()) - 1) {
+                if(value >= 0.0)
+                    signal += "";
+                else
+                    signal += "-";
+            } else {
+                if(value >= 0.0)
+                    signal += "+ ";
+                else
+                    signal += "- ";
+            }
+
+            if(index == 0) {
+                denominator += signal + StringFromDouble(std::abs(value), 0);
+                break;
+            } else if(index == 1) {
+                if(value == 1.0) {
+                    denominator += signal + "s";
+                } else {
+                    denominator += signal + StringFromDouble(std::abs(value), 0) + "s";
+                }
+            } else {
+                if(value == 1.0) {
+                    denominator += signal + "s" + GetSuperscriptNumber(index);
+                } else {
+                    denominator += signal + StringFromDouble(std::abs(value), 0) + "s" + GetSuperscriptNumber(index);
+                }
+            }
+            denominator += " ";
+        }
+        --index;
+    }
+}
+
+void TransferFunction::UpdateTFText()
+{
+    wxString num, den;
+    GetTFString(num, den);
+    SetText(num, den);
+    if(m_nodeList.size() == 2) {
+        if(m_angle == 0.0) {
+            m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(-m_width / 2, 0));
+            m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(m_width / 2, 0));
+        } else if(m_angle == 90.0) {
+            m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(0, -m_height / 2));
+            m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(0, m_height / 2));
+        } else if(m_angle == 180.0) {
+            m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(m_width / 2, 0));
+            m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(-m_width / 2, 0));
+        } else if(m_angle == 270.0) {
+            m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(0, m_height / 2));
+            m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(0, -m_height / 2));
+        }
+    }
+}
+
+bool TransferFunction::ShowForm(wxWindow* parent, Element* element)
+{
+    TransferFunctionForm* tfForm = new TransferFunctionForm(parent, this);
+    if(tfForm->ShowModal() == wxID_OK) {
+        tfForm->Destroy();
+        return true;
+    }
+    tfForm->Destroy();
+    return false;
+}
+
+void TransferFunction::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;
+
+    if(m_angle == 0.0) {
+        m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(-m_width / 2, 0));
+        m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(m_width / 2, 0));
+    } else if(m_angle == 90.0) {
+        m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(0, -m_height / 2));
+        m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(0, m_height / 2));
+    } else if(m_angle == 180.0) {
+        m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(m_width / 2, 0));
+        m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(-m_width / 2, 0));
+    } else if(m_angle == 270.0) {
+        m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(0, m_height / 2));
+        m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(0, -m_height / 2));
+    }
+
+    for(auto it = m_nodeList.begin(), itEnd = m_nodeList.end(); it != itEnd; ++it) {
+        Node* node = *it;
+        node->Rotate(clockwise);
+    }
+}