]> Creatis software - creaImageIO.git/blobdiff - src2/creaImageIOWxTreeView.cpp
Added validation function (visible in ok button being enabled or not)
[creaImageIO.git] / src2 / creaImageIOWxTreeView.cpp
index 2a58cd393a9598dfb4af7b88fd5112efe2dd4e79..bd640783e8a210ffec1bdbec9dfa76f3317fd5fe 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
 {
   //=====================================================================
@@ -299,7 +308,7 @@ namespace creaImageIO
 
 
   //================================================================
-  void WxTreeView::OnSelected(wxListEvent& event)
+  void WxTreeView::OnSelectedChanged(wxListEvent& event)
   { 
     GimmickDebugMessage(1,
                        GetTreeHandler()->GetTree().GetLabel()
@@ -321,12 +330,335 @@ 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()
+  {
+       int level=mLevelList.size();
+       std::vector<tree::Node*> sel=GetSelected(level+1);
+       if(sel.size()>0)
+       {
+               std::vector<tree::Node*>::iterator i;
+               bool valid=true;
+               std::string row;
+               std::string col;
+               std::string plane;
+               
+               //Validation between image sizes
+               for (i=sel.begin(); i!=sel.end() && valid; ++i)
+               {
+                       if(i==sel.begin())
+                       {
+                               row=(*i)->GetAttribute(mLevelList[level-1].key[1]);
+                               col=(*i)->GetAttribute(mLevelList[level-1].key[2]);
+                               plane=(*i)->GetAttribute(mLevelList[level-1].key[3]);
+                       }
+                       else
+                       {
+                               if(((*i)->GetAttribute(mLevelList[level-1].key[1]))!=row ||
+                                       ((*i)->GetAttribute(mLevelList[level-1].key[2]))!=col ||
+                                       ((*i)->GetAttribute(mLevelList[level-1].key[3]))!=plane)
+                                       {
+                                               valid=false;
+                                       }
+                       }
+               }
+
+               GimmickMessage(1,
+                       "State check: Planes:" 
+                       <<plane<<" Cols: "
+                       <<col<<" Rows:"
+                               <<row<<" "
+                               << std::endl);
+               //Dimention validation
+               //Compatibility with maximum 
+               if(valid)
+               {       
+                       int rows;
+                       int cols;
+                       int planes;
+                       std::string s;
+                       std::istringstream t(s);
+                       s=row;
+                       t >> rows;
+                       if(row==""){rows=1;}
+                       s=col;
+                       t >> cols;
+                       if(col==""){cols=1;}
+                       s=plane;
+                       t >> planes;
+                       if(plane==""){planes=1;}
+
+                       int dim = 0;
+                       if (planes>1) dim=3;
+                       else if (cols>1) dim=2;
+                       else if (rows>1) dim=1;
+                   
+                       if (dim == 0) 
+                       {
+                       GimmickMessage(1,
+                               "Unknown image dimension : cannot select !" 
+                               <<dim<<" "
+                               <<plane<<" "
+                               <<col<<" "
+                               <<row<<" "
+                               << std::endl);
+                               valid= false;
+                       }
+                       else if (dim>GetMaxDimension())
+                       {
+                               GimmickMessage(1,"Selecting "<<dim<<"D images is not allowed !" 
+                               << std::endl);
+                               valid= false;
+                       }
+                       if ( dim == GetMaxDimension() )
+                       {
+                               GimmickMessage(1,"Cannot add this image to selection : would result in a "<<dim+1<<"D image !" << std::endl);
+                               valid= false;
+                       }
+               }
+                 
+               //Send an event telling wether the selection is valid or not
+               wxCommandEvent event( 0, GetId() );
+               event.SetEventObject( this );
+               if(valid)
+               {event.SetInt(0);}
+               else
+               {event.SetInt(1);}
+               GetEventHandler()->ProcessEvent( 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)
@@ -377,15 +709,18 @@ BEGIN_EVENT_TABLE(MyListCtrl, wxListCtrl)
     EVT_LIST_SET_INFO(LIST_CTRL, MyListCtrl::OnSetInfo)
 #endif
   */
-    EVT_LIST_ITEM_SELECTED(-1, WxTreeView::OnSelected)
-  /*
-    EVT_LIST_ITEM_DESELECTED(LIST_CTRL, MyListCtrl::OnDeselected)
+    EVT_LIST_ITEM_SELECTED(-1, WxTreeView::OnSelectedChanged)
+  
+    EVT_LIST_ITEM_DESELECTED(-1, WxTreeView::OnSelectedChanged)
+       /*
     EVT_LIST_KEY_DOWN(LIST_CTRL, MyListCtrl::OnListKeyDown)
     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)