]> Creatis software - creaImageIO.git/blobdiff - src2/creaImageIOWxTreeView.cpp
Implemented calls from/to the interface to several methods.
[creaImageIO.git] / src2 / creaImageIOWxTreeView.cpp
index a1d9142d5fd7276fb7938a7dc6035066e5587982..1b3d20d7d0ff829db0b92e30b4ae12ab1aca9add 100644 (file)
@@ -1,7 +1,16 @@
 #include <creaImageIOWxTreeView.h>
 #include <creaImageIOSystem.h>
 #include <wx/splitter.h>
+int wxCALLBACK MyCompareFunction(long item1, long item2, long WXUNUSED(sortData))
+{
+    // inverse the order
+    if (item1 < item2)
+        return -1;
+    if (item1 > item2)
+        return 1;
 
+    return 0;
+}
 namespace creaImageIO
 {
   //=====================================================================
@@ -24,7 +33,7 @@ namespace creaImageIO
     // Global sizer
     wxBoxSizer *sizer = new wxBoxSizer(wxHORIZONTAL);
     
-    int ctrl_style = wxLC_REPORT;
+    int ctrl_style = wxLC_REPORT | wxLC_VRULES;
     int col_style = wxLIST_FORMAT_LEFT;
 
     // Creating the ListCtrl for the levels > 0 (not for Root level)
@@ -32,37 +41,71 @@ namespace creaImageIO
         i < handler->GetTree().GetNumberOfLevels();
         ++i)
       {
-       GimmickDebugMessage(5,"Creating ListCtrl for level "<<i
+
+       GimmickDebugMessage(5,"Creating view for level "<<i
                            <<std::endl);
        LevelType level;
-       wxListCtrl* ctrl = new wxListCtrl(this,
+
+       // If the first level : parent = this
+       wxWindow* sparent = this;
+       // else parent = last splitter
+       if (i>1) sparent = mLevelList[i-2].wxSplitter;
+
+       level.wxSplitter = new wxSplitterWindow( sparent , -1);
+       //          level.wxSplitter->SetMinimumPaneSize(100);
+       
+       wxListCtrl* ctrl = new wxListCtrl(level.wxSplitter,
                                          i,
                                          wxDefaultPosition, 
                                          wxDefaultSize,
                                          ctrl_style);
        level.wxCtrl = ctrl;
+       level.wxSplitter->Initialize(ctrl);
 
        // Create the columns : one for each attribute of the level
        int col = 0;
+                       //ctrl->InsertItem(0, "1");
+       
        tree::LevelDescriptor::AttributeDescriptorListType::const_iterator a;
        for (a  = handler->GetTree().GetAttributeDescriptorList(i).begin();
             a != handler->GetTree().GetAttributeDescriptorList(i).end();
             ++a)
          {
-           GimmickDebugMessage(5,"Creating column "<<col<<" : "
+                 if(col==0)
+                 {
+                   wxListItem it;
+                   it.SetTextColour(*wxRED);
+                   it.SetText(_T("#C"));
+                   it.SetColumn(col);
+                   //              ctrl->InsertColumn(col, _("Children"), col_style);
+                   ctrl->InsertColumn(col,it);
+                   //ctrl->InsertItem(0, "1");
+                   //level.key.push_back(_("Children"));
+                   //level.key.push_back(handler->GetTree().GetChildrenList().size());
+                   col++;
+                 }
+                 
+               GimmickDebugMessage(5,"Creating column "<<col<<" : "
                                <<a->GetName()
                                <<std::endl);
-           ctrl->InsertColumn(col, 
-                              crea::std2wx(a->GetName()),
-                              col_style);
-           level.key.push_back(a->GetKey());
-           // ctrl->SetColumnWidth(col, wxLIST_AUTOSIZE );
-           col++;
+               ctrl->InsertColumn(col, 
+                               crea::std2wx(a->GetName()),
+                               col_style);
+               level.key.push_back(a->GetKey());
+               //          ctrl->SetColumnWidth(col, wxLIST_AUTOSIZE );
+               col++;
+               
+               
          }
+         
        mLevelList.push_back(level);
-       sizer->Add( ctrl ,1, wxGROW  ,0);
       }
     
+
+    /// Initialize the first level splitter
+      sizer->Add( mLevelList[0].wxSplitter ,1, wxGROW  ,0);
+
     UpdateLevel(1);
 
     SetSizer( sizer );     
@@ -87,19 +130,14 @@ namespace creaImageIO
   {
     tree::Node* node;
   };
   //=====================================================================
-
-  //=====================================================================
-  /// 
-  void WxTreeView::UpdateLevel( int level )
+   std::vector<tree::Node*> WxTreeView::GetSelected(int level)
   {
-    GimmickDebugMessage(1,
-                       GetTreeHandler()->GetTree().GetLabel()
-                       <<" view : updating level "<<level
-                       <<std::endl);
     int l = level - 1;
     // the selection of upper level
     std::vector<tree::Node*> sel;
+       
     if (level == 1) 
       {
        sel.push_back(GetTreeHandler()->GetTree().GetTree());
@@ -117,7 +155,81 @@ namespace creaImageIO
              }
          }     
       }
+       
+         return sel;
+  }
+
+  //=====================================================================
+  
+  ///Removes selected nodes on given level
+  void WxTreeView::RemoveSelected( int level )
+  {
+         std::vector<tree::Node*> sel=GetSelected(level+1);
+         bool erase=false;
+         if (wxMessageBox(_T("Delete file(s) ?"),
+                        _T("Remove Files"),
+                        wxYES_NO,this ) == wxYES)
+         {
+           erase = true;
+         }
+         if(erase)
+         {
+               std::vector<tree::Node*>::iterator i;
+               for (i=sel.begin(); i!=sel.end(); ++i)
+               {
+                       GimmickDebugMessage(2,
+                                       "deleting '"
+                                       <<(*i)->GetLabel()
+                                       <<"'"<<level
+                                       <<std::endl);
+                               GetTreeHandler()->Remove(*i);
+               }
+
+               UpdateLevel(level);
+         }
+         
+  }
+    
+
+  //=====================================================================
+
+  //=====================================================================
+  /// 
+  void WxTreeView::UpdateLevel( int level )
+  {
+    GimmickDebugMessage(1,
+                       GetTreeHandler()->GetTree().GetLabel()
+                       <<" view : updating level "<<level
+                       <<std::endl);
+    
+    wxBusyCursor busy;
+    RecursiveUpdateLevel(level);
+    int i;
+    for (i=0; i<level-1; i++)
+      {
+       if (!GetSplitter(i)->IsSplit()) 
+         GetSplitter(i)->SplitVertically(  GetCtrl(i), GetSplitter(i+1),
+                                           100 );
+      }
+    if (GetSplitter(i)->IsSplit()) GetSplitter(i)->Unsplit();    
+    
+  }
+  //=====================================================================
+  
+  /// 
+  void WxTreeView::RecursiveUpdateLevel( int level )
+  {
+         GimmickDebugMessage(2,
+                       GetTreeHandler()->GetTree().GetLabel()
+                       <<" view : updating level (recursive)"<<level
+                       <<std::endl);
+
 
+    std::vector<tree::Node*> sel=GetSelected(level);
+
+         int l = level - 1;
     // to speed up inserting we hide the control temporarily
     GetCtrl(l)->Hide();
     GetCtrl(l)->DeleteAllItems();
@@ -131,6 +243,7 @@ namespace creaImageIO
                            <<"'"<<level
                            <<std::endl);
        
+       //Adds columns
        GetTreeHandler()->LoadChildren(*i,1);
        tree::Node::ChildrenListType::reverse_iterator j;
        for (j = (*i)->GetChildrenList().rbegin(); 
@@ -150,20 +263,46 @@ namespace creaImageIO
            data->node = *j;
            item.SetData(data);
            
+         
            long id = GetCtrl(l)->InsertItem(item);
 
-           for (int k=0; k<GetCtrl(l)->GetColumnCount(); k++)
+           std::ostringstream oss;
+           int n= GetTreeHandler()->GetNumberOfChildren(*j);
+           
+           oss << n;
+           std::string s(oss.str());
+           
+           item.SetText( crea::std2wx(s));
+           //      item.SetTextColour(*wxRED);
+           //      item.SetBackgroundColour(*wxBLACK); 
+           item.SetColumn(0);
+           GetCtrl(l)->SetItem(item);
+           //      GetCtrl(l)->SetItem(id,0, crea::std2wx(s));
+           //      GetCtrl(l)->SetColumnWidth(0, wxLIST_AUTOSIZE );
+
+         for (int k=1; k<GetCtrl(l)->GetColumnCount(); k++)
              {
-               GetCtrl(l)->SetItem
-                 (id,k,
-                  crea::std2wx
-                  ( (*j)->GetAttribute(mLevelList[l].key[k]) ));
+
+               std::string val = (*j)->GetAttribute(mLevelList[l].key[k-1]);
+               if (val.size()==0) val = "?";
+               item.SetText( crea::std2wx(val));
+               //              item.SetTextColour(*wxBLACK);
+               //              item.SetBackgroundColour(*wxWHITE); 
+               item.SetColumn(k);
+               GetCtrl(l)->SetItem(item);
+
+               //              GetCtrl(l)->SetItem(id,k, crea::std2wx(val));
+               // GetCtrl(l)->SetColumnWidth(k, wxLIST_AUTOSIZE );
+               
              }
            
          }
       }
     
     GetCtrl(l)->Show();
+
+    
+    if (level<mLevelList.size()) UpdateLevel(level+1);
  }
   //=====================================================================
 
@@ -176,7 +315,6 @@ namespace creaImageIO
                        <<" view : item selected "
                        <<std::endl);
 
-    
     wxListItem info;
     info.m_itemId = event.m_itemIndex;
     
@@ -192,12 +330,251 @@ namespace creaImageIO
                        " Level "<<level+1
                        <<std::endl);
     if (level<mLevelList.size()-1) UpdateLevel( level + 2 ); 
+       if (level==mLevelList.size()-1) ValidateSelectedImages ();
+
+  }
+  //================================================================
+
+//================================================================
+  void WxTreeView::OnColClick(wxListEvent& event)
+  { 
+         /*
+       //Obtain the column name and the level that needs to be organized
+       int colNum=event.m_col;
+    GimmickMessage(1,
+                       " Column " <<event.m_col
+                       <<std::endl);
+
+       
+    wxObject* ctrl = event.GetEventObject(); 
+       unsigned int level = 0;
+    for (level = 0; level<mLevelList.size(); ++level)
+      {
+       if ( GetCtrl(level) == ctrl ) break;
+      }
+
+         wxBusyCursor busy;
+    std::vector<tree::Node*> sel=GetSelected(level+1);
+
+       tree::Node* nodes[20];
+
+         int l = level - 1;
+    // to speed up inserting we hide the control temporarily
+    GetCtrl(level)->Hide();
+    GetCtrl(level)->DeleteAllItems();
     
+    std::vector<tree::Node*>::iterator i;
+       int num=0;
+    for (i=sel.begin(); i!=sel.end(); ++i)
+      {
+       
+               //Adds columns
+               GetTreeHandler()->LoadChildren(*i,1);
+               tree::Node::ChildrenListType::reverse_iterator j;
+               
+               for (j = (*i)->GetChildrenList().rbegin(); 
+                       j!= (*i)->GetChildrenList().rend(); 
+                       ++j)
+                       {
+                               nodes[num]=*j;
+                               num++;
+                       }
+      }
+       
+         //Sorting elements
+         int k, m; 
+         tree::Node* index;
+         
+       
+       for (k=1; k<num; ++k)
+       {
+               index = nodes[k];
+               m = k;
+               GimmickMessage(1,
+                       "Comparison Key 1: "
+                           <<(nodes[m-1])->GetAttribute(mLevelList[level].key[colNum-1])
+                               <<"Comparison Key 2: "<<(index)->GetAttribute(mLevelList[level].key[colNum-1])
+                           <<std::endl);
+               while ((m > 0) && ((nodes[m-1])->GetAttribute(mLevelList[level].key[colNum-1]) > 
+                                                  (index)->GetAttribute(mLevelList[level].key[colNum-1])))
+               {
+               nodes[m] = nodes[m-1];
+               m = m - 1;
+               }
+               nodes[m] = index;
+       }
+
+       for (m=0; m<num; ++m)
+       {
+               tree::Node* node=nodes[m];
+           wxListItem item;
+           item.SetMask(wxLIST_MASK_STATE | 
+                        wxLIST_MASK_TEXT |
+                        //                      wxLIST_MASK_IMAGE |
+                        wxLIST_MASK_DATA |
+                        //                      wxLIST_MASK_WIDTH |
+                        wxLIST_MASK_FORMAT
+                        );
+
+           long id = GetCtrl(level)->InsertItem(item);
+               int n;
+               if ((node)->GetChildrenLoaded()) 
+               {
+                       n=(node)->GetChildrenList().size();
+               }
+               else
+               {
+                       n= GetTreeHandler()->GetNumberOfChildren(node);
+               }
+               std::ostringstream oss;
+               
+               oss << n;
+               std::string s(oss.str());
+
+               GetCtrl(level)->SetItem(id,0, crea::std2wx(s));
+               GetCtrl(level)->SetColumnWidth(0, wxLIST_AUTOSIZE );
+               
+         for (int k=1; k<GetCtrl(level)->GetColumnCount(); k++)
+             {
+               std::string val = (node)->GetAttribute(mLevelList[level].key[k-1]);
+               if (val.size()==0) val = "?";
+               GetCtrl(level)->SetItem(id,k, crea::std2wx(val));
+               GetCtrl(level)->SetColumnWidth(k, wxLIST_AUTOSIZE );
+               
+             }
+           
+         }
+
+    
+    GetCtrl(level)->Show();
+*/
+  }
+  //================================================================
+
+  void WxTreeView::ValidateSelectedImages()
+  {
+         GimmickMessage(1,
+                       "Hello World Validate"
+                           <<std::endl);
+         /*
+       MyEvent event( MyCommandEvent, 0 );
+       wxString bar( wxT("This is a Foo_DoFirstThing event") );
+       event.SetText( bar );
+       wxPostEvent( this, event );
+       */
+       
   }
+
   //================================================================
 
 
+  vtkImageData* WxTreeView::GetSelectedImage(int dim)
+  {
+         /*
+               wxArrayTreeItemIds id;
+    // TO DO : TEST THAT STYLE IS MULTIPLE 
+    unsigned int nb = mTreeListCtrl->GetSelections(id);
+    f.clear();
+
+    // Collect the brute vector of Image nodes
+    std::vector<DicomNode*> im;
+    for (unsigned int i=0; i<nb; ++i)
+      {
+       TreeItemData *data = 
+         (TreeItemData *)mTreeListCtrl->GetItemData(id[i]);
+       if ((data) && (data->IsDicomNode()))
+         {
+           if (data->GetDicomNode()->GetType()==DicomNode::Image)
+             {
+               im.push_back ( data->GetDicomNode() );
 
+             }
+           else if (data->GetDicomNode()->GetType()==DicomNode::Series)
+             {
+               DicomNode::ChildrenListType::iterator j;
+               for (j =data->GetDicomNode()->GetChildrenList().begin();
+                    j!=data->GetDicomNode()->GetChildrenList().end();
+                    j++) 
+                 {
+                   im.push_back ( *j );
+                 }
+             }
+         }
+      }
+    // Create the output data
+    if (im.size()==1) 
+      {
+       // Only one image : give it
+       vtkImageData* out = vtkImageData::New();
+       out->ShallowCopy(mReader.GetImage(im.front()->ImageGetFullFileName()));
+       f.push_back( out );
+      }
+    else if (im.size()>1)
+      {
+       vtkImageData* first = mReader.GetImage( im.front()->ImageGetFullFileName() );
+       if (first->GetDataDimension()==2) 
+         {     
+           // n2D to 3D
+           vtkImageData* out = vtkImageData::New();
+           out->CopyStructure(first);  
+           out->SetScalarType(first->GetScalarType());
+           int ext[6];
+           first->GetExtent(ext);
+           ext[5] = im.size();
+           out->SetExtent(ext);
+           // LG : TODO : Z Spacing  ?
+           
+           out->AllocateScalars();
+           
+           //first->Print(std::cout);
+           //      out->Print(std::cout);
+           
+           int dim[3];
+           first->GetDimensions(dim);
+           unsigned long imsize = 
+             ( (unsigned long)first->GetScalarPointer(0,1,0)
+               - (unsigned long)first->GetScalarPointer(0,0,0))
+             *dim[1];
+
+           int slice = 0;
+           std::vector<DicomNode*>::iterator it;
+           for (it=im.begin(); it!=im.end(); ++it) 
+             {
+               //std::cout << "copying slice "<<slice <<std::endl;
+               vtkImageData* cur = mReader.GetImage( (*it)->ImageGetFullFileName() );
+               
+               void* src = cur->GetScalarPointer(0,0,0);
+               void* dst = out->GetScalarPointer(0,0,slice);
+               //              std::cout << "src="<<src<<std::endl;
+               //              std::cout << "dst="<<dst<<std::endl;
+               //              std::cout << "siz="<<imsize<<std::endl;
+               memcpy(dst,src,imsize);
+
+               slice++;
+             }
+           f.push_back(out);
+         }
+       else 
+         {
+           // n3D
+           std::vector<DicomNode*>::iterator it;
+           for (it=im.begin(); it!=im.end(); ++it) 
+             {
+               vtkImageData* out = vtkImageData::New();
+               out->ShallowCopy(mReader.GetImage((*it)->ImageGetFullFileName()));
+               f.push_back(out);
+             }
+         }
+      }
+         */
+       return NULL;
+  }
+
+  void WxTreeView::GetSelectedImages(std::vector<vtkImageData*>& s, int dim)
+  {
+
+  }
   BEGIN_EVENT_TABLE(WxTreeView, wxPanel)
   /*
     EVT_SIZE(MyFrame::OnSize)
@@ -255,8 +632,10 @@ BEGIN_EVENT_TABLE(MyListCtrl, wxListCtrl)
     EVT_LIST_ITEM_ACTIVATED(LIST_CTRL, MyListCtrl::OnActivated)
     EVT_LIST_ITEM_FOCUSED(LIST_CTRL, MyListCtrl::OnFocused)
 
-    EVT_LIST_COL_CLICK(LIST_CTRL, MyListCtrl::OnColClick)
     EVT_LIST_COL_RIGHT_CLICK(LIST_CTRL, MyListCtrl::OnColRightClick)
+       */
+    EVT_LIST_COL_CLICK(-1, WxTreeView::OnColClick)
+       /*
     EVT_LIST_COL_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnColBeginDrag)
     EVT_LIST_COL_DRAGGING(LIST_CTRL, MyListCtrl::OnColDragging)
     EVT_LIST_COL_END_DRAG(LIST_CTRL, MyListCtrl::OnColEndDrag)