1 files changed, 97 insertions, 0 deletions
diff --git a/Project/RateLimiter.cpp b/Project/RateLimiter.cpp
new file mode 100644
index 0000000..a003e56
--- /dev/null
+++ b/Project/RateLimiter.cpp
@@ -0,0 +1,97 @@
+#include "RateLimiter.h"
+#include "RateLimiterForm.h"
+
+RateLimiter::RateLimiter() : ControlElement()
+{
+    m_width = m_height = 36.0;
+    Node* nodeIn = new Node(m_position + wxPoint2DDouble(-18, 0), Node::NODE_IN, m_borderSize);
+    nodeIn->StartMove(m_position);
+    Node* nodeOut = new Node(m_position + wxPoint2DDouble(18, 0), Node::NODE_OUT, m_borderSize);
+    nodeOut->SetAngle(180.0);
+    nodeOut->StartMove(m_position);
+    m_nodeList.push_back(nodeIn);
+    m_nodeList.push_back(nodeOut);
+}
+
+RateLimiter::~RateLimiter()
+{
+}
+
+void RateLimiter::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);
+
+    // Plot symbol.
+    std::vector<wxPoint2DDouble> axis;
+    axis.push_back(m_position + wxPoint2DDouble(-10, 0));
+    axis.push_back(m_position + wxPoint2DDouble(10, 0));
+    axis.push_back(m_position + wxPoint2DDouble(0, -10));
+    axis.push_back(m_position + wxPoint2DDouble(0, 10));
+    DrawLine(axis, GL_LINES);
+    
+    glLineWidth(2.0);
+    std::vector<wxPoint2DDouble> limSymbol;
+    limSymbol.push_back(m_position + wxPoint2DDouble(7, -7));
+    limSymbol.push_back(m_position + wxPoint2DDouble(-7, 7));
+    glColor4d(0.0, 0.3, 1.0, 1.0);
+    DrawLine(limSymbol);
+
+    glColor4d(0.0, 0.0, 0.0, 1.0);
+    DrawNodes();
+}
+
+bool RateLimiter::ShowForm(wxWindow* parent, Element* element)
+{
+    RateLimiterForm* form = new RateLimiterForm(parent, this);
+    if(form->ShowModal() == wxID_OK) {
+        form->Destroy();
+        return true;
+    }
+    form->Destroy();
+    return false;
+}
+
+void RateLimiter::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 RateLimiter::UpdatePoints()
+{
+    if(m_angle == 0.0) {
+        m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(-18, 0));
+        m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(18, 0));
+    } else if(m_angle == 90.0) {
+        m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(0, -18));
+        m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(0, 18));
+    } else if(m_angle == 180.0) {
+        m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(18, 0));
+        m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(-18, 0));
+    } else if(m_angle == 270.0) {
+        m_nodeList[0]->SetPosition(m_position + wxPoint2DDouble(0, 18));
+        m_nodeList[1]->SetPosition(m_position + wxPoint2DDouble(0, -18));
+    }
+}