Feature #1711

[crea.git] / lib / creaDevManagerLib / modelCDMMain.h

diff --git a/lib/creaDevManagerLib/modelCDMMain.h b/lib/creaDevManagerLib/modelCDMMain.h
index 83d86acaf4a3252a83a84d7045c05b989519eaaf..d75355ce27f0a70ac624482ec3a2e718c95f005d 100644 (file)
--- a/lib/creaDevManagerLib/modelCDMMain.h
+++ b/lib/creaDevManagerLib/modelCDMMain.h
@@ -36,7 +36,12 @@
 #define MODELCDMMAIN_H_
 
 #include<iostream>
+#include<map>
 
+#include<creaWx.h>
+#include<wx/treectrl.h>
+
+#include "modelCDMIProjectTreeNode.h"
 #include "modelCDMProject.h"
 
 class modelCDMMain
@@ -46,6 +51,8 @@ public:
   ~modelCDMMain();
 
   modelCDMProject* GetProject() const;
+  std::map< wxTreeItemId, modelCDMIProjectTreeNode* >& GetModelElements();
+
 
   bool CreateProject(
       const std::string& name,
@@ -65,8 +72,10 @@ public:
       std::string*& result
   );
 
+
 private:
   modelCDMProject* project;
+  std::map< wxTreeItemId, modelCDMIProjectTreeNode* > modelElements;
 };