X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=src2%2FcreaImageIOWxTreeView.cpp;h=1b3d20d7d0ff829db0b92e30b4ae12ab1aca9add;hb=633701590bade8ba6d0f8c7b37c033596b4b2c2a;hp=a258ce88c0b082165b888018ebbf76189b8493ea;hpb=befba590823a2a6d3800ac9d026ec0f3888ba0aa;p=creaImageIO.git

diff --git a/src2/creaImageIOWxTreeView.cpp b/src2/creaImageIOWxTreeView.cpp
index a258ce8..1b3d20d 100644
--- a/src2/creaImageIOWxTreeView.cpp
+++ b/src2/creaImageIOWxTreeView.cpp
@@ -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,14 +263,37 @@ 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 );
+		
 	      }
 	    
 	  }
@@ -179,7 +315,6 @@ namespace creaImageIO
 			<<" view : item selected "
 			<<std::endl);
 
-    
     wxListItem info;
     info.m_itemId = event.m_itemIndex;
     
@@ -195,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)
@@ -258,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)