1 files changed, 56 insertions, 0 deletions
diff --git a/Project/ControlElementContainer.h b/Project/ControlElementContainer.h
new file mode 100644
index 0000000..119ba82
--- /dev/null
+++ b/Project/ControlElementContainer.h
@@ -0,0 +1,56 @@
+#ifndef CONTROLELEMENTCONTAINER_H
+#define CONTROLELEMENTCONTAINER_H
+
+#include <vector>
+
+class ControlEditor;
+class ControlElement;
+
+#include "ConnectionLine.h"
+#include "Constant.h"
+#include "Exponential.h"
+#include "Gain.h"
+#include "IOControl.h"
+#include "Limiter.h"
+#include "Multiplier.h"
+#include "RateLimiter.h"
+#include "Sum.h"
+#include "TransferFunction.h"
+
+class ControlElementContainer
+{
+   public:
+    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; }
+    std::vector<Gain*> GetGainList() const { return m_gainList; }
+    std::vector<IOControl*> GetIOControlList() const { return m_ioControlList; }
+    std::vector<Limiter*> GetLimiterList() const { return m_limiterList; }
+    std::vector<Multiplier*> GetMultiplierList() const { return m_multiplierList; }
+    std::vector<RateLimiter*> GetRateLimiterList() const { return m_rateLimiterList; }
+    std::vector<Sum*> GetSumList() const { return m_sumList; }
+    std::vector<TransferFunction*> GetTFList() const { return m_tfList; }
+   protected:
+    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;
+    std::vector<Limiter*> m_limiterList;
+    std::vector<Multiplier*> m_multiplierList;
+    std::vector<RateLimiter*> m_rateLimiterList;
+    std::vector<Sum*> m_sumList;
+    std::vector<TransferFunction*> m_tfList;
+};
+
+#endif  // CONTROLELEMENTCONTAINER_H