From 31033e01b4736ce235410776c48251b84e00f21c Mon Sep 17 00:00:00 2001
From: Martin Hanzik <martin@hanzik.com>
Date: Tue, 24 Apr 2018 23:42:28 +0200
Subject: [PATCH] GraphicsBox getters and type fixes

---
 agui2/src/Graphics/GraphicsBox.cpp | 8 ++++++++
 agui2/src/Graphics/GraphicsBox.hpp | 9 +++++++--
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/agui2/src/Graphics/GraphicsBox.cpp b/agui2/src/Graphics/GraphicsBox.cpp
index 4ebc170fa9..75895c9162 100644
--- a/agui2/src/Graphics/GraphicsBox.cpp
+++ b/agui2/src/Graphics/GraphicsBox.cpp
@@ -49,3 +49,11 @@ void GraphicsBox::paint(QPainter* painter, const QStyleOptionGraphicsItem* optio
     painter->fillRect(this->boundRect, this->color);
     painter->drawText(this->boundRect, Qt::AlignCenter, this->text);
 }
+
+const std::vector<InputConnectionBox*>& GraphicsBox::getInputConnectionBoxes() const {
+    return inputConnectionBoxes;
+}
+
+OutputConnectionBox* GraphicsBox::getOutputConnectionBox() const {
+    return outputConnectionBox;
+}
\ No newline at end of file
diff --git a/agui2/src/Graphics/GraphicsBox.hpp b/agui2/src/Graphics/GraphicsBox.hpp
index 8baa6e23ef..9e761320be 100644
--- a/agui2/src/Graphics/GraphicsBox.hpp
+++ b/agui2/src/Graphics/GraphicsBox.hpp
@@ -6,6 +6,8 @@
 #include <Models/ModelBox.hpp>
 
 class ConnectionBox;
+class InputConnectionBox;
+class OutputConnectionBox;
 
 class GraphicsBox : public QGraphicsObject {
     Q_OBJECT
@@ -17,6 +19,9 @@ public:
 
     ModelBox* getModelBox() const { return this->modelBox.get(); }
 
+    const std::vector<InputConnectionBox*>& getInputConnectionBoxes() const;
+    OutputConnectionBox* getOutputConnectionBox() const;
+
 protected:
     QColor color;
 
@@ -27,8 +32,8 @@ private:
 
     std::unique_ptr<ModelBox> modelBox;
 
-    std::vector<ConnectionBox*> inputConnectionBoxes;
-    ConnectionBox* outputConnectionBox;
+    std::vector<InputConnectionBox*> inputConnectionBoxes;
+    OutputConnectionBox* outputConnectionBox;
 };
 
 
-- 
GitLab