1 files changed, 23 insertions, 0 deletions
diff --git a/Project/ControlElementContainer.cpp b/Project/ControlElementContainer.cpp
index e461529..f8accdc 100644
--- a/Project/ControlElementContainer.cpp
+++ b/Project/ControlElementContainer.cpp
@@ -50,6 +50,29 @@ void ControlElementContainer::ClearContainer()
 void ControlElementContainer::FillContainer(std::vector<ControlElement*> controlElementList,
                                             std::vector<ConnectionLine*> connectionLineList)
 {
+    ClearContainer();
     m_ctrlElementsList = controlElementList;
     m_cLineList = connectionLineList;
+
+    for(auto it = controlElementList.begin(), itEnd = controlElementList.end(); it != itEnd; ++it) {
+        if(Constant* constant = dynamic_cast<Constant*>(*it)) {
+            m_constantList.push_back(constant);
+        } else if(Exponential* exponential = dynamic_cast<Exponential*>(*it)) {
+            m_exponentialList.push_back(exponential);
+        } else if(Gain* gain = dynamic_cast<Gain*>(*it)) {
+            m_gainList.push_back(gain);
+        } else if(IOControl* ioControl = dynamic_cast<IOControl*>(*it)) {
+            m_ioControlList.push_back(ioControl);
+        } else if(Limiter* limiter = dynamic_cast<Limiter*>(*it)) {
+            m_limiterList.push_back(limiter);
+        } else if(Multiplier* multiplier = dynamic_cast<Multiplier*>(*it)) {
+            m_multiplierList.push_back(multiplier);
+        } else if(RateLimiter* rateLimiter = dynamic_cast<RateLimiter*>(*it)) {
+            m_rateLimiterList.push_back(rateLimiter);
+        } else if(Sum* sum = dynamic_cast<Sum*>(*it)) {
+            m_sumList.push_back(sum);
+        } else if(TransferFunction* tf = dynamic_cast<TransferFunction*>(*it)) {
+            m_tfList.push_back(tf);
+        }
+    }
 }