Some bugfixes and generalization

1 files changed, 7 insertions, 0 deletions

diff --git a/Project/ControlElementContainer.cpp b/Project/ControlElementContainer.cpp
index 0e0e9d4..f5372a8 100644
--- a/Project/ControlElementContainer.cpp
+++ b/Project/ControlElementContainer.cpp
@@ -7,6 +7,7 @@ ControlElementContainer::~ControlElementContainer() {}
 void ControlElementContainer::FillContainer(ControlEditor* editor)
 {
     ClearContainer();
+    m_ctrlElementsList = editor->GetControlElementList();
     m_cLineList = editor->GetConnectionLineList();
     auto cElementList = editor->GetControlElementList();
     for(auto it = cElementList.begin(), itEnd = cElementList.end(); it != itEnd; ++it) {
@@ -45,3 +46,9 @@ void ControlElementContainer::ClearContainer()
     m_sumList.clear();
     m_tfList.clear();
 }
+
+void ControlElementContainer::FillContainer(std::vector<ControlElement*> controlElementList, std::vector<ConnectionLine*> connectionLineList)
+{
+    m_ctrlElementsList = controlElementList;
+    m_cLineList = connectionLineList;
+}