Feature #1711

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

diff --git a/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx b/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx
index 8337e48c35da386e7f1e4989a2936d166fbcf3b4..3b471199e0270ef5c25d798b4b8217dceeff025f 100755 (executable)
--- a/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx
+++ b/lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx
@@ -34,9 +34,24 @@
  */
 
 #include "wxCDMProjectsTreeCtrl.h"
-#include "creaDevManagerIds.h"
+
 #include <vector>
 
+#include <wx/imaglist.h>
+
+#include "creaDevManagerIds.h"
+#include "images/AIcon20.xpm"
+#include "images/ApIcon20.xpm"
+#include "images/BBIcon20.xpm"
+#include "images/CIcon20.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,
     wxWindowID id,
@@ -65,30 +80,54 @@ bool wxCDMProjectsTreeCtrl::Create(
 )
 {
   wxTreeCtrl::Create (parent, id, pos, size, style, validator, name);
-  wxTreeItemId rootIndex = this-> AddRoot(_("Open Projects"));
+
+  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_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_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< wxTreeItemId, modelCDMIProjectTreeNode* >& modelElements, modelCDMProject* projectTree)
 {
+  std::cout << "building tree" << std::endl;
+  this->DeleteAllItems();
+  modelElements.clear();
   if(projectTree != NULL)
     {
-      this->DeleteAllItems();
       wxTreeItemId rootIndex;
-      rootIndex= this-> AddRoot(crea::std2wx(projectTree->GetName()));
+      rootIndex= this-> AddRoot(crea::std2wx(projectTree->GetName()), this->ID_PrIcon, this->ID_PrIcon);
       projectTree->SetId(rootIndex);
+      modelElements[rootIndex] = projectTree;
 
       std::cout << "Building TreeCtrl for " << projectTree->GetName() << std::endl;
-      this->BuildTree(projectTree->GetChildren(), rootIndex);
+      this->BuildTree(projectTree->GetChildren(), modelElements, rootIndex);
 
       this->Expand(rootIndex);
 
       this->Update();
     }
+  else
+    {
+      wxTreeItemId rootIndex = this-> AddRoot(_("No Open Project"), this->ID_Cicon, this->ID_Cicon);
+    }
 }
 
-void wxCDMProjectsTreeCtrl::BuildTree(const std::vector<modelCDMIProjectTreeNode*>& treeNodes, wxTreeItemId parent)
+void wxCDMProjectsTreeCtrl::BuildTree(const std::vector<modelCDMIProjectTreeNode*>& treeNodes, std::map< wxTreeItemId, modelCDMIProjectTreeNode* >& modelElements, wxTreeItemId parent)
 {
   for (int i = 0; i < treeNodes.size(); i++)
     {
@@ -96,16 +135,103 @@ void wxCDMProjectsTreeCtrl::BuildTree(const std::vector<modelCDMIProjectTreeNode
       wxTreeItemId parentNodeIndex;
       if(treeNodes[i] != NULL)
         {
+          int idIcon = GetIconId(treeNodes[i]);
           wxString nodeName((treeNodes[i]->GetName()).c_str(), wxConvUTF8);
-          parentNodeIndex = this->AppendItem(parent, nodeName);
+          parentNodeIndex = this->AppendItem(parent, nodeName, idIcon, idIcon);
           treeNodes[i]->SetId(parentNodeIndex);
+          modelElements[parentNodeIndex] = treeNodes[i];
 
           std::vector<modelCDMIProjectTreeNode*> innerChildren = treeNodes[i]->GetChildren();
           if(innerChildren.size() > 0)
             {
-              this->BuildTree(innerChildren, parentNodeIndex);
+              this->BuildTree(innerChildren, modelElements, parentNodeIndex);
             }
         }
 
     }
 }
+
+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<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;
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
+}