From 6c0e98a2727d07e1fbb38b78c27d68e98ad09465 Mon Sep 17 00:00:00 2001
From: Thales Lima Oliveira <thaleslima.ufu@gmail.com>
Date: Sat, 13 Jun 2020 00:01:53 -0300
Subject: Device context implemented

---
 Project/ControlElement.cpp | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'Project/ControlElement.cpp')

diff --git a/Project/ControlElement.cpp b/Project/ControlElement.cpp
index 2af9a6f..c2bb9d2 100644
--- a/Project/ControlElement.cpp
+++ b/Project/ControlElement.cpp
@@ -112,6 +112,14 @@ void ControlElement::DrawNodes() const
     }
 }
 
+void ControlElement::DrawDCNodes(wxGraphicsContext* gc) const
+{
+    for (auto node : m_nodeList) {
+        DrawDCCircle(node->GetPosition(), node->GetRadius(), 10, gc);
+        if (node->GetNodeType() == Node::NodeType::NODE_IN) { DrawDCTriangle(node->GetInTrianglePts(), gc); }
+    }
+}
+
 void ControlElement::StartMove(wxPoint2DDouble position)
 {
     m_moveStartPt = position;
@@ -147,7 +155,7 @@ ControlElement* ControlElement::GetControlElementFromID(std::vector<ControlEleme
         ControlElement* element = *it;
         if(element->GetID() == id) return element;
     }
-    return NULL;
+    return nullptr;
 }
 
 void ControlElement::SaveControlNodes(rapidxml::xml_document<>& doc, rapidxml::xml_node<>* elementNode)
-- 
cgit