X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=src2%2FcreaImageIOWxTreeView.cpp;h=bd640783e8a210ffec1bdbec9dfa76f3317fd5fe;hb=2f0e50f69eb904d629cf8db62eb9a4094db93097;hp=2a58cd393a9598dfb4af7b88fd5112efe2dd4e79;hpb=dd070f00561f17bc9ac754a7116516da5d41fbbd;p=creaImageIO.git

diff --git a/src2/creaImageIOWxTreeView.cpp b/src2/creaImageIOWxTreeView.cpp
index 2a58cd3..bd64078 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
 {
   //=====================================================================
@@ -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)