lib/creaDevManagerLib/wxCDMProjectsTreeCtrl.cxx

   1 /*
   2 # ---------------------------------------------------------------------
   3 #
   4 # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
   5 #                        pour la Sant�)
   6 # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
   7 # Previous Authors : Laurent Guigues, Jean-Pierre Roux
   8 # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
   9 #
  10 #  This software is governed by the CeCILL-B license under French law and
  11 #  abiding by the rules of distribution of free software. You can  use,
  12 #  modify and/ or redistribute the software under the terms of the CeCILL-B
  13 #  license as circulated by CEA, CNRS and INRIA at the following URL
  14 #  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  15 #  or in the file LICENSE.txt.
  16 #
  17 #  As a counterpart to the access to the source code and  rights to copy,
  18 #  modify and redistribute granted by the license, users are provided only
  19 #  with a limited warranty  and the software's author,  the holder of the
  20 #  economic rights,  and the successive licensors  have only  limited
  21 #  liability.
  22 #
  23 #  The fact that you are presently reading this means that you have had
  24 #  knowledge of the CeCILL-B license and that you accept its terms.
  25 # ------------------------------------------------------------------------
  26  */
  27
  28
  29 /*
  30  * wxCreaDevManagerTreeCtrl.cpp
  31  *
  32  *  Created on: 19/10/2012
  33  *      Author: Daniel Felipe Gonzalez Obando
  34  */
  35
  36 #include "wxCDMProjectsTreeCtrl.h"
  37
  38 #include <vector>
  39
  40 #include <wx/imaglist.h>
  41
  42 #include "creaDevManagerIds.h"
  43 #include "images/AIcon20.xpm"
  44 #include "images/ApIcon20.xpm"
  45 #include "images/BBIcon20.xpm"
  46 #include "images/CIcon20.xpm"
  47 #include "images/CMIcon20.xpm"
  48 #include "images/FdIcon20.xpm"
  49 #include "images/FlIcon20.xpm"
  50 #include "images/LbIcon20.xpm"
  51 #include "images/LIcon20.xpm"
  52 #include "images/PrIcon20.xpm"
  53 #include "images/PkIcon20.xpm"
  54
  55 wxCDMProjectsTreeCtrl::wxCDMProjectsTreeCtrl(
  56     wxWindow *parent,
  57     wxWindowID id,
  58     const wxPoint &pos,
  59     const wxSize &size,
  60     long style,
  61     const wxValidator &validator,
  62     const wxString &name
  63 )
  64 {
  65   wxCDMProjectsTreeCtrl::Create(parent, id, pos, size, style, validator, name);
  66 }
  67
  68 wxCDMProjectsTreeCtrl::~wxCDMProjectsTreeCtrl()
  69 {
  70 }
  71
  72 bool wxCDMProjectsTreeCtrl::Create(
  73     wxWindow* parent,
  74     wxWindowID id,
  75     const wxPoint &pos,
  76     const wxSize &size,
  77     long style,
  78     const wxValidator &validator,
  79     const wxString &name
  80 )
  81 {
  82   wxTreeCtrl::Create (parent, id, pos, size, style, validator, name);
  83
  84   wxImageList* images = new wxImageList(20, 20, true);
  85   this->ID_AIcon = images->Add(wxIcon(AIcon20));
  86   this->ID_ApIcon = images->Add(wxIcon(ApIcon20));
  87   this->ID_BBIcon = images->Add(wxIcon(BBIcon20));
  88   this->ID_Cicon = images->Add(wxIcon(CIcon20));
  89   this->ID_CMIcon = images->Add(wxIcon(CMIcon20));
  90   this->ID_FdIcon = images->Add(wxIcon(FdIcon20));
  91   this->ID_FlIcon = images->Add(wxIcon(FlIcon20));
  92   this->ID_LbIcon = images->Add(wxIcon(LbIcon20));
  93   this->ID_LIcon = images->Add(wxIcon(LIcon20));
  94   this->ID_PrIcon = images->Add(wxIcon(PrIcon20));
  95   this->ID_PkIcon = images->Add(wxIcon(PkIcon20));
  96   this->AssignImageList(images);
  97
  98   wxTreeItemId rootIndex = this-> AddRoot(_("No Open Project"), this->ID_Cicon, this->ID_Cicon);
  99   this->Update();
 100   return TRUE;
 101 }
 102
 103 void wxCDMProjectsTreeCtrl::BuildTree(std::map< wxTreeItemId, modelCDMIProjectTreeNode* >& modelElements, modelCDMProject* projectTree)
 104 {
 105   std::cout << "building tree" << std::endl;
 106   this->DeleteAllItems();
 107   modelElements.clear();
 108   if(projectTree != NULL)
 109     {
 110       wxTreeItemId rootIndex;
 111       rootIndex= this-> AddRoot(crea::std2wx(projectTree->GetName()), this->ID_PrIcon, this->ID_PrIcon);
 112       projectTree->SetId(rootIndex);
 113       modelElements[rootIndex] = projectTree;
 114
 115       std::cout << "Building TreeCtrl for " << projectTree->GetName() << std::endl;
 116       this->BuildTree(projectTree->GetChildren(), modelElements, rootIndex);
 117
 118       this->Expand(rootIndex);
 119
 120       this->Update();
 121     }
 122   else
 123     {
 124       wxTreeItemId rootIndex = this-> AddRoot(_("No Open Project"));
 125     }
 126 }
 127
 128 void wxCDMProjectsTreeCtrl::BuildTree(const std::vector<modelCDMIProjectTreeNode*>& treeNodes, std::map< wxTreeItemId, modelCDMIProjectTreeNode* >& modelElements, wxTreeItemId parent)
 129 {
 130   for (int i = 0; i < treeNodes.size(); i++)
 131     {
 132       //cout << projectsTree[i].GetName() << endl;
 133       wxTreeItemId parentNodeIndex;
 134       if(treeNodes[i] != NULL)
 135         {
 136           int idIcon = GetIconId(treeNodes[i]);
 137           wxString nodeName((treeNodes[i]->GetName()).c_str(), wxConvUTF8);
 138           parentNodeIndex = this->AppendItem(parent, nodeName, idIcon, idIcon);
 139           treeNodes[i]->SetId(parentNodeIndex);
 140           modelElements[parentNodeIndex] = treeNodes[i];
 141
 142           std::vector<modelCDMIProjectTreeNode*> innerChildren = treeNodes[i]->GetChildren();
 143           if(innerChildren.size() > 0)
 144             {
 145               this->BuildTree(innerChildren, modelElements, parentNodeIndex);
 146             }
 147         }
 148
 149     }
 150 }
 151
 152 int wxCDMProjectsTreeCtrl::GetIconId(modelCDMIProjectTreeNode* node)
 153 {
 154   modelCDMIProjectTreeNode* element = dynamic_cast<modelCDMProject*>(node);
 155   if(element != NULL)
 156     {
 157       return this->ID_PrIcon;
 158     }
 159   else
 160     {
 161       element = dynamic_cast<modelCDMAppli*>(node);
 162       if(element != NULL)
 163         {
 164           return this->ID_ApIcon;
 165         }
 166       else
 167         {
 168           element = dynamic_cast<modelCDMApplication*>(node);
 169           if(element != NULL)
 170             {
 171               return this->ID_AIcon;
 172             }
 173           else
 174             {
 175               element = dynamic_cast<modelCDMLib*>(node);
 176               if(element != NULL)
 177                 {
 178                   return this->ID_LbIcon;
 179                 }
 180               else
 181                 {
 182                   element = dynamic_cast<modelCDMLibrary*>(node);
 183                   if(element != NULL)
 184                     {
 185                       return this->ID_LIcon;
 186                     }
 187                   else
 188                     {
 189                       element = dynamic_cast<modelCDMPackage*>(node);
 190                       if(element != NULL)
 191                         {
 192                           return this->ID_PkIcon;
 193                         }
 194                       else
 195                         {
 196                           element = dynamic_cast<modelCDMBlackBox*>(node);
 197                           if(element != NULL)
 198                             {
 199                               return this->ID_BBIcon;
 200                             }
 201                           else
 202                             {
 203                               element = dynamic_cast<modelCDMCMakeListsFile*>(node);
 204                               if(element != NULL)
 205                                 {
 206                                   return this->ID_CMIcon;
 207                                 }
 208                               else
 209                                 {
 210                                   element = dynamic_cast<modelCDMFolder*>(node);
 211                                   if(element != NULL)
 212                                     {
 213                                       return this->ID_FdIcon;
 214                                     }
 215                                   else
 216                                     {
 217                                       element = dynamic_cast<modelCDMFile*>(node);
 218                                       if(element != NULL)
 219                                         {
 220                                           return this->ID_FlIcon;
 221                                         }
 222                                       else
 223                                         {
 224                                           return this->ID_Cicon;
 225                                         }
 226                                     }
 227                                 }
 228                             }
 229                         }
 230                     }
 231                 }
 232             }
 233         }
 234     }
 235 }