Feature #1711

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

diff --git a/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.h b/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.h
index 12fbf2e52ced621c1f8ec2d2a4d9520206829c51..82a9698c7011a85b37fec8833648b99a09a5b715 100755 (executable)
--- a/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.h
+++ b/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.h
@@ -2,8 +2,10 @@
 # ---------------------------------------------------------------------
 #
 # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image 
-#                        pour la Santé)
+#                        pour la Sant�)
 # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
+# Previous Authors : Laurent Guigues, Jean-Pierre Roux
+# CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
 #
 #  This software is governed by the CeCILL-B license under French law and 
 #  abiding by the rules of distribution of free software. You can  use, 
@@ -20,14 +22,15 @@
 #
 #  The fact that you are presently reading this means that you have had
 #  knowledge of the CeCILL-B license and that you accept its terms.
-# ------------------------------------------------------------------------ */ 
+# ------------------------------------------------------------------------ 
+ */
 
 
 /*
  * wxCreaDevManagerTreeCtrl.h
  *
  *  Created on: 19/10/2012
- *      Author: daniel
+ *      Author: Daniel Felipe Gonzalez Obando
  */
 
 #ifndef WXCDMPROJECTSTREECTRL_H_
@@ -35,9 +38,11 @@
 
 #include <creaWx.h>
 #include <wx/treectrl.h>
-#include "modelCDMProjectsTree.h"
+#include "modelCDMProject.h"
+#include "modelCDMIProjectTreeNode.h"
 
 #include <vector>
+#include <map>
 
 class wxCDMProjectsTreeCtrl: public wxTreeCtrl
 {
@@ -53,18 +58,33 @@ public:
   );
   ~wxCDMProjectsTreeCtrl();
   bool Create(
-    wxWindow *parent,
-    wxWindowID id=wxID_ANY,
-    const wxPoint &pos=wxDefaultPosition,
-    const wxSize &size=wxDefaultSize,
-    long style=wxTR_DEFAULT_STYLE,
-    const wxValidator &validator=wxDefaultValidator,
-    const wxString &name=_("Projects tree")
+      wxWindow *parent,
+      wxWindowID id=wxID_ANY,
+      const wxPoint &pos=wxDefaultPosition,
+      const wxSize &size=wxDefaultSize,
+      long style=wxTR_DEFAULT_STYLE,
+      const wxValidator &validator=wxDefaultValidator,
+      const wxString &name=_("Projects tree")
   );
 
-  void BuildTree(const modelCDMProjectsTree& tree);
+  void BuildTree(std::map< wxTreeItemId, modelCDMIProjectTreeNode* >& modelElements, modelCDMProject* tree = NULL);
+
 private:
-  void BuildTree(const std::vector<modelCDMProjectsTreeNode>& tree, wxTreeItemId parent);
+
+  int ID_AIcon;
+  int ID_ApIcon;
+  int ID_BBIcon;
+  int ID_Cicon;
+  int ID_CMIcon;
+  int ID_FdIcon;
+  int ID_FlIcon;
+  int ID_LbIcon;
+  int ID_LIcon;
+  int ID_PrIcon;
+  int ID_PkIcon;
+
+  void BuildTree(const std::vector<modelCDMIProjectTreeNode*>& tree, std::map< wxTreeItemId, modelCDMIProjectTreeNode* >& modelElements, wxTreeItemId parent);
+  int GetIconId(modelCDMIProjectTreeNode* node);
 };
 
 #endif /* WXCDMPROJECTSTREECTRL_H_ */