From b43f89144d6122b3100efb3971c8fa32a3001888 Mon Sep 17 00:00:00 2001
From: Thales Lima Oliveira <thaleslima.ufu@gmail.com>
Date: Tue, 18 Apr 2017 19:05:35 -0300
Subject: Some bugfixes and generalization

---
 Project/ControlElementContainer.h | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'Project/ControlElementContainer.h')

diff --git a/Project/ControlElementContainer.h b/Project/ControlElementContainer.h
index 5ff218f..119ba82 100644
--- a/Project/ControlElementContainer.h
+++ b/Project/ControlElementContainer.h
@@ -24,8 +24,10 @@ class ControlElementContainer
     ~ControlElementContainer();
 
     virtual void FillContainer(ControlEditor* editor);
+    virtual void FillContainer(std::vector<ControlElement*> controlElementList, std::vector<ConnectionLine*> connectionLineList);
     virtual void ClearContainer();
-
+    
+    std::vector<ControlElement*> GetControlElementsList() const { return m_ctrlElementsList; }
     std::vector<ConnectionLine*> GetConnectionLineList() const { return m_cLineList; }
     std::vector<Constant*> GetConstantList() const { return m_constantList; }
     std::vector<Exponential*> GetExponentialList() const { return m_exponentialList; }
@@ -37,8 +39,10 @@ class ControlElementContainer
     std::vector<Sum*> GetSumList() const { return m_sumList; }
     std::vector<TransferFunction*> GetTFList() const { return m_tfList; }
    protected:
-    std::vector<ConnectionLine*> m_cLineList;
+    std::vector<ControlElement*> m_ctrlElementsList;
     std::vector<Constant*> m_constantList;
+
+    std::vector<ConnectionLine*> m_cLineList;
     std::vector<Exponential*> m_exponentialList;
     std::vector<Gain*> m_gainList;
     std::vector<IOControl*> m_ioControlList;
-- 
cgit