X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=src2%2FcreaImageIOWxTreeView.cpp;h=1f3ccfeb1ff109fa35ba57ee90497ccb41cabd92;hb=384558d116ebb2ca264900f837c4a921d695515a;hp=a1d9142d5fd7276fb7938a7dc6035066e5587982;hpb=0489e5c255bda4eff3b6e8df2c04ab12b8efd293;p=creaImageIO.git

diff --git a/src2/creaImageIOWxTreeView.cpp b/src2/creaImageIOWxTreeView.cpp
index a1d9142..1f3ccfe 100644
--- a/src2/creaImageIOWxTreeView.cpp
+++ b/src2/creaImageIOWxTreeView.cpp
@@ -1,6 +1,71 @@
 #include <creaImageIOWxTreeView.h>
 #include <creaImageIOSystem.h>
 #include <wx/splitter.h>
+int wxCALLBACK CompareFunctionStrings(long item1, long item2, long sortData)
+{	
+	std::string s1((char*)((long*)item1)[1]);
+	std::string s2((char*)((long*)item2)[1]);
+	 GimmickMessage(1,
+		" Check: STRING 1: " 
+			<<s1
+			<<" STRING 2: "
+			<<s2
+			<<std::endl);
+	if(sortData==1)
+	{
+		// inverse the order
+		if (s1 < s2)
+			return 1;
+		if (s1 > s2)
+			return -1;
+
+		return 0;
+	}
+	else
+	{
+		if (s1 < s2)
+			return -1;
+		if (s1 > s2)
+			return 1;
+
+		return 0;
+
+	}
+	
+}
+
+int wxCALLBACK CompareFunctionInts(long item1, long item2, long sortData)
+{	
+	int val1=atoi((char*)((long*)item1)[1]);
+	int val2=atoi((char*)((long*)item2)[1]);
+	 GimmickMessage(1,
+		" Check: STRING 1: " 
+			<<val1
+			<<" STRING 2: "
+			<<val2
+			<<std::endl);
+	if(sortData==1)
+	{
+		// inverse the order
+		if (val1 < val2)
+			return 1;
+		if (val1 > val2)
+			return -1;
+
+		return 0;
+	}
+	else
+	{
+		if (val1 < val2)
+			return -1;
+		if (val1 > val2)
+			return 1;
+
+		return 0;
+
+	}
+	
+}
 
 namespace creaImageIO
 {
@@ -24,7 +89,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 +97,67 @@ 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;
+	
 	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"));
+		
+		    ctrl->InsertColumn(col,it);
+		    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);
+	mColumnSelected=1;
+	mDirection=true;
     UpdateLevel(1);
 
     SetSizer( sizer );     
@@ -86,20 +181,16 @@ namespace creaImageIO
   struct ItemData
   {
     tree::Node* node;
+	int id;
   };
+ 
   //=====================================================================
-
-  //=====================================================================
-  /// 
-  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 +208,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();
@@ -130,7 +295,8 @@ namespace creaImageIO
 			    <<(*i)->GetLabel()
 			    <<"'"<<level
 			    <<std::endl);
-	
+	int _id=0;
+	//Adds columns
 	GetTreeHandler()->LoadChildren(*i,1);
 	tree::Node::ChildrenListType::reverse_iterator j;
 	for (j = (*i)->GetChildrenList().rbegin(); 
@@ -147,36 +313,78 @@ namespace creaImageIO
 			 );
 
 	    ItemData* data = new ItemData;
-	    data->node = *j;
+	    
+		data->node = *j;
+		item.SetId(_id);
+		/*
+		std::string a=(*j)->GetAttribute(mLevelList[l].key[mColumnSelected-1]);
+		GimmickMessage(1,
+			"State Check: Att VAL"
+			    <<a
+			    <<std::endl);
+	*/
+		data->id = _id;
 	    item.SetData(data);
 	    
+		_id++;
+	  
 	    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);
+
+		//Sets the last level as selecte....How to make it select only the first time?
+		//if (level==mLevelList.size()) item.SetState(wxLIST_STATE_SELECTED);
+
+	    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);
  }
   //=====================================================================
 
 
   //================================================================
-  void WxTreeView::OnSelected(wxListEvent& event)
+  void WxTreeView::OnSelectedChanged(wxListEvent& event)
   { 
     GimmickDebugMessage(1,
 			GetTreeHandler()->GetTree().GetLabel()
 			<<" view : item selected "
 			<<std::endl);
 
-    
     wxListItem info;
     info.m_itemId = event.m_itemIndex;
     
@@ -192,12 +400,130 @@ 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
+	mColumnSelected=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;
+      }
+	  UpdateLevel(level+1);
+	  wxBusyCursor busy;
+ 
+	  int l = level - 1;
+
+	  //Sets the data for the items to be sorted
+	std::string att;
+	unsigned int ty=0;
+	int n = GetCtrl(level)->GetItemCount();
+	for (int i = 0; i < n; i++)
+	  {
+		  
+	    //Gets current item data
+		long adr = GetCtrl(level)->GetItemData(i);
+		//Extracts the node and the type of attribute
+		tree::Node* nod = ((ItemData*)adr)->node;
+		if(i==0)
+		  {
+			(*nod).GetAttributeDescriptor(mLevelList[level].key[mColumnSelected-1]).DecodeType(ty);
+		  }
+		//Obtains the organizing attribute
+		att=(*nod).GetAttribute(mLevelList[level].key[mColumnSelected-1]);
+		
+		char* d= new char [att.size()+1];
+		strcpy (d, att.c_str());
+
+		//Creates array
+		long* lp= new long[2];
+		lp[0]=adr;
+		lp[1]=(long)d;
+		 
+		//Sets it as the data
+		GetCtrl(level)->SetItemData(i,(long)lp);
+	  }	
+	  
+	  //int ty=GetAttributeDescriptor(mLevelList[level].key[mColumnSelected-1]).GetType();
+	  if(mDirection)
+	  {
+		  if(ty==1)
+		  {
+			  GetCtrl(level)->SortItems(CompareFunctionInts, 0);
+		  }
+		  else
+		  {
+			  GetCtrl(level)->SortItems(CompareFunctionStrings, 0);
+		  }
+		
+		mDirection=false;
+	  }
+	  else
+	  {
+		 if(ty==1)
+		  {
+			  GetCtrl(level)->SortItems(CompareFunctionInts, 1);
+		  }
+		  else
+		  {
+			  GetCtrl(level)->SortItems(CompareFunctionStrings, 1);
+		  }
+		  mDirection=true;
+	  }
 
+	//Resets original data
+		
+	long it = -1;
+	for ( ;; )
+    {
+        it = GetCtrl(level)->GetNextItem(it,
+                                     wxLIST_NEXT_ALL);
+        if ( it == -1 )
+            break;
+		  //Gets current item data, extracts the node and resets it
+		long item = GetCtrl(level)->GetItemData(it);
+		GetCtrl(level)->SetItemData(it,((long*)item)[0]);
+		
+	  }
+	
+   }
+  //================================================================
 
+  void WxTreeView::ValidateSelectedImages()
+  {
+		//Send an event telling wether the selection is valid or not
+		wxCommandEvent event( 0, GetId() );
+		event.SetEventObject( this );
+		std::vector<tree::Node*> sel=GetSelected((mLevelList.size()+1));
+		event.SetClientData(&sel);
+		GetEventHandler()->ProcessEvent( event );
+  }
+  //================================================================
+  void WxTreeView::GetSelectedAsString(std::vector<std::string>&s)
+  {
+	  int level=mLevelList.size();
+	std::vector<tree::Node*> sel=GetSelected(level+1);
+	std::vector<tree::Node*>::iterator i;
+	
+    for (i=sel.begin(); i!=sel.end(); ++i)
+      {
+		  std::string filename=(*i)->GetAttribute("FullFileName");
+		  s.push_back(filename);
+	  }
+  }
+  //================================================================
   BEGIN_EVENT_TABLE(WxTreeView, wxPanel)
   /*
     EVT_SIZE(MyFrame::OnSize)
@@ -248,15 +574,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)
@@ -274,4 +603,3 @@ END_EVENT_TABLE()
   
 } // EO namespace creaImageIO
 
-