X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcreaDevManagerLib%2FwxCDMProjectsTreeCtrl.cxx;h=d0bbafb67534d17273a5a9bced6ca492631de45e;hb=bae167768b5d3ccae6ec9b04c844d80e0368a52d;hp=61bd7e7752de661ebbd72431474d1fe361b90946;hpb=62dfbfc86db5ed98a2272d0b6620af1711791c27;p=crea.git

diff --git a/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx b/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx
index 61bd7e7..d0bbafb 100755
--- a/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx
+++ b/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx
@@ -34,8 +34,27 @@
  */
 
 #include "wxCDMProjectsTreeCtrl.h"
-#include "creaDevManagerIds.h"
+
 #include <vector>
+#include <sstream>
+
+#include <wx/imaglist.h>
+
+#include "creaDevManagerIds.h"
+#include "images/AIcon20.xpm"
+#include "images/ApIcon20.xpm"
+#include "images/BBIcon20.xpm"
+#include "images/BBSIcon20.xpm"
+#include "images/BBGIcon20.xpm"
+#include "images/CIcon20.xpm"
+#include "images/CFIcon20.xpm"
+#include "images/CMIcon20.xpm"
+#include "images/FdIcon20.xpm"
+#include "images/FlIcon20.xpm"
+#include "images/LbIcon20.xpm"
+#include "images/LIcon20.xpm"
+#include "images/PrIcon20.xpm"
+#include "images/PkIcon20.xpm"
 
 wxCDMProjectsTreeCtrl::wxCDMProjectsTreeCtrl(
     wxWindow *parent,
@@ -65,51 +84,185 @@ bool wxCDMProjectsTreeCtrl::Create(
 )
 {
   wxTreeCtrl::Create (parent, id, pos, size, style, validator, name);
-  wxTreeItemId rootIndex = this-> AddRoot(_("No Open Project"));
+
+  this->DeleteAllItems();
+
+  wxImageList* images = new wxImageList(20, 20, true);
+  this->ID_AIcon = images->Add(wxIcon(AIcon20));
+  this->ID_ApIcon = images->Add(wxIcon(ApIcon20));
+  this->ID_BBIcon = images->Add(wxIcon(BBIcon20));
+  this->ID_BBGIcon = images->Add(wxIcon(BBGIcon20));
+  this->ID_BBSIcon = images->Add(wxIcon(BBSIcon20));
+  this->ID_Cicon = images->Add(wxIcon(CIcon20));
+  this->ID_CMIcon = images->Add(wxIcon(CMIcon20));
+  this->ID_FdIcon = images->Add(wxIcon(FdIcon20));
+  this->ID_FlIcon = images->Add(wxIcon(FlIcon20));
+  this->ID_CFIcon = images->Add(wxIcon(CFIcon20));
+  this->ID_LbIcon = images->Add(wxIcon(LbIcon20));
+  this->ID_LIcon = images->Add(wxIcon(LIcon20));
+  this->ID_PrIcon = images->Add(wxIcon(PrIcon20));
+  this->ID_PkIcon = images->Add(wxIcon(PkIcon20));
+  this->AssignImageList(images);
+
+  wxTreeItemId rootIndex = this->AddRoot(wxT("No Open Project"), this->ID_Cicon, this->ID_Cicon);
+
   this->Update();
   return TRUE;
 }
 
-void wxCDMProjectsTreeCtrl::BuildTree(modelCDMProject* projectTree)
+void wxCDMProjectsTreeCtrl::BuildTree(std::map< wxCDMTreeItemId, modelCDMIProjectTreeNode* >& modelElements, modelCDMProject* projectTree)
 {
+  this->Unselect();
+  std::cout << "building tree" << std::endl;
   this->DeleteAllItems();
+  modelElements.clear();
   if(projectTree != NULL)
     {
-      wxTreeItemId rootIndex;
-      rootIndex= this-> AddRoot(crea::std2wx(projectTree->GetName()));
-      projectTree->SetId(rootIndex);
+      projectTree->SetId(this->AddRoot(crea::std2wx(projectTree->GetName()), this->ID_PrIcon));
+	  
+	  modelElements[projectTree->GetId()] = projectTree;
 
       std::cout << "Building TreeCtrl for " << projectTree->GetName() << std::endl;
-      this->BuildTree(projectTree->GetChildren(), rootIndex);
+      this->BuildTree(projectTree->GetChildren(), modelElements, projectTree->GetId());
 
-      this->Expand(rootIndex);
+      this->Expand(projectTree->GetId().GetWxId());
 
       this->Update();
     }
   else
     {
-      wxTreeItemId rootIndex = this-> AddRoot(_("No Open Project"));
+      wxTreeItemId rootIndex = this-> AddRoot(_("No Open Project"), this->ID_Cicon);
     }
 }
 
-void wxCDMProjectsTreeCtrl::BuildTree(const std::vector<modelCDMIProjectTreeNode*>& treeNodes, wxTreeItemId parent)
+void wxCDMProjectsTreeCtrl::BuildTree(const std::vector<modelCDMIProjectTreeNode*>& treeNodes, std::map< wxCDMTreeItemId, modelCDMIProjectTreeNode* >& modelElements, const wxCDMTreeItemId& parent)
 {
-  for (int i = 0; i < treeNodes.size(); i++)
+  for (int i = 0; i < (int)(treeNodes.size()); i++)
     {
       //cout << projectsTree[i].GetName() << endl;
-      wxTreeItemId parentNodeIndex;
       if(treeNodes[i] != NULL)
         {
+          int idIcon = GetIconId(treeNodes[i]);
           wxString nodeName((treeNodes[i]->GetName()).c_str(), wxConvUTF8);
-          parentNodeIndex = this->AppendItem(parent, nodeName);
-          treeNodes[i]->SetId(parentNodeIndex);
+          treeNodes[i]->SetId(wxCDMTreeItemId(this->AppendItem(parent.GetWxId(), nodeName, idIcon)));
+	  
+		  modelElements[treeNodes[i]->GetId()] = treeNodes[i];
 
           std::vector<modelCDMIProjectTreeNode*> innerChildren = treeNodes[i]->GetChildren();
           if(innerChildren.size() > 0)
             {
-              this->BuildTree(innerChildren, parentNodeIndex);
+              this->BuildTree(innerChildren, modelElements, treeNodes[i]->GetId());
             }
         }
 
     }
 }
+
+int wxCDMProjectsTreeCtrl::GetIconId(modelCDMIProjectTreeNode* node)
+{
+  modelCDMIProjectTreeNode* element = dynamic_cast<modelCDMProject*>(node);
+  if(element != NULL)
+    {
+      return this->ID_PrIcon;
+    }
+  else
+    {
+      element = dynamic_cast<modelCDMAppli*>(node);
+      if(element != NULL)
+        {
+          return this->ID_ApIcon;
+        }
+      else
+        {
+          element = dynamic_cast<modelCDMApplication*>(node);
+          if(element != NULL)
+            {
+              return this->ID_AIcon;
+            }
+          else
+            {
+              element = dynamic_cast<modelCDMLib*>(node);
+              if(element != NULL)
+                {
+                  return this->ID_LbIcon;
+                }
+              else
+                {
+                  element = dynamic_cast<modelCDMLibrary*>(node);
+                  if(element != NULL)
+                    {
+                      return this->ID_LIcon;
+                    }
+                  else
+                    {
+                      element = dynamic_cast<modelCDMPackage*>(node);
+                      if(element != NULL)
+                        {
+                          return this->ID_PkIcon;
+                        }
+                      else
+                        {
+                          element = dynamic_cast<modelCDMBlackBox*>(node);
+                          if(element != NULL)
+                            {
+                              return this->ID_BBIcon;
+                            }
+                          else
+                            {
+                              element = dynamic_cast<modelCDMCMakeListsFile*>(node);
+                              if(element != NULL)
+                                {
+                                  return this->ID_CMIcon;
+                                }
+                              else
+                                {
+                                  element = dynamic_cast<modelCDMCodeFile*>(node);
+                                  if(element != NULL)
+                                    {
+                                      return this->ID_CFIcon;
+                                    }
+                                  else
+                                    {
+                                      element = dynamic_cast<modelCDMBBSFile*>(node);
+                                      if(element != NULL)
+                                        {
+                                          return this->ID_BBSIcon;
+                                        }
+                                      else
+                                        {
+                                          element = dynamic_cast<modelCDMBBGFile*>(node);
+                                          if(element != NULL)
+                                            {
+                                              return this->ID_BBGIcon;
+                                            }
+                                          else
+                                            {
+                                              element = dynamic_cast<modelCDMFolder*>(node);
+                                              if(element != NULL)
+                                                {
+                                                  return this->ID_FdIcon;
+                                                }
+                                              else
+                                                {
+                                                  element = dynamic_cast<modelCDMFile*>(node);
+                                                  if(element != NULL)
+                                                    {
+                                                      return this->ID_FlIcon;
+                                                    }
+                                                  else
+                                                    {
+                                                      return this->ID_Cicon;
+                                                    }
+                                                }
+                                            }
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
+}