]> Creatis software - creaImageIO.git/blobdiff - src2/creaImageIOWxTreeView.cpp
Added validation on frame exit
[creaImageIO.git] / src2 / creaImageIOWxTreeView.cpp
index 2a58cd393a9598dfb4af7b88fd5112efe2dd4e79..4aca9f841bb524e9657ded94c9797e43875cb9df 100644 (file)
@@ -1,16 +1,73 @@
 #include <creaImageIOWxTreeView.h>
+#include <creaImageIOGimmickView.h>
 #include <creaImageIOSystem.h>
 #include <wx/splitter.h>
+///Comparing function for ordering algorithm. Takes parameters as strings.
+int wxCALLBACK CompareFunctionStrings(long item1, long item2, long sortData)
+{      
+       std::string s1((char*)((long*)item1)[1]);
+       std::string s2((char*)((long*)item2)[1]);
+       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;
+
+       }
+       
+}
+
+///Comparing function for ordering algorithm. Takes parameters as ints.
+int wxCALLBACK CompareFunctionInts(long item1, long item2, long sortData)
+{      
+       int val1=atoi((char*)((long*)item1)[1]);
+       int val2=atoi((char*)((long*)item2)[1]);
+       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
 {
   //=====================================================================
   // CTor
   WxTreeView::WxTreeView(TreeHandler* handler,
+                        GimmickView* gimmick,
                         wxWindow* parent,
                         const wxWindowID id)
     : wxPanel(parent,id),
-      TreeView(handler)
+      TreeView(handler,gimmick)
   {
     GimmickDebugMessage(1,"WxTreeView::WxTreeView"
                        <<std::endl);
@@ -55,7 +112,6 @@ namespace creaImageIO
 
        // 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();
@@ -67,12 +123,8 @@ namespace creaImageIO
                    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++;
                  }
                  
@@ -96,7 +148,9 @@ namespace creaImageIO
     /// Initialize the first level splitter
  
       sizer->Add( mLevelList[0].wxSplitter ,1, wxGROW  ,0);
-
+       mColumnSelected=1;
+       mLastSelected=0;
+       mDirection=true;
     UpdateLevel(1);
 
     SetSizer( sizer );     
@@ -120,6 +174,7 @@ namespace creaImageIO
   struct ItemData
   {
     tree::Node* node;
+       int id;
   };
  
   //=====================================================================
@@ -142,11 +197,20 @@ namespace creaImageIO
              {
                long adr = GetCtrl(l-1)->GetItemData(i);
                tree::Node* n = ((ItemData*)adr)->node;
-               sel.push_back(n);
+                       if(mLastSelected==i)
+                       {
+                               std::vector<tree::Node*>::iterator it;
+                               it = sel.begin();
+                               it = sel.insert ( it , n );
+                       }
+                       else
+                       {
+                               sel.push_back(n);
+                       }
              }
          }     
       }
-       
+        
          return sel;
   }
 
@@ -186,7 +250,7 @@ namespace creaImageIO
 
  
   //=====================================================================
-  /// 
+  /// Updates a level of the view (adds or removes children, proganizes, etc.)
   void WxTreeView::UpdateLevel( int level )
   {
     GimmickDebugMessage(1,
@@ -208,7 +272,7 @@ namespace creaImageIO
   }
   //=====================================================================
   
-  /// 
+  /// Recursive method called upon by UpdateLevel to refresh all windows
   void WxTreeView::RecursiveUpdateLevel( int level )
   {
          GimmickDebugMessage(2,
@@ -233,7 +297,10 @@ namespace creaImageIO
                            <<(*i)->GetLabel()
                            <<"'"<<level
                            <<std::endl);
-       
+       int _id=0;
+       int r=100;
+       int g=100;
+       int b=100;
        //Adds columns
        GetTreeHandler()->LoadChildren(*i,1);
        tree::Node::ChildrenListType::reverse_iterator j;
@@ -251,29 +318,59 @@ namespace creaImageIO
                         );
 
            ItemData* data = new ItemData;
-           data->node = *j;
-           item.SetData(data);
            
-         
-           long id = GetCtrl(l)->InsertItem(item);
+               data->node = *j;
+               item.SetId(_id);
 
+               data->id = _id;
+           item.SetData(data);
+           
+               _id++;
+           long id=GetCtrl(l)->InsertItem(item);
+               
            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); 
+
+               //Setting the color according to the parent
+               /*
+               if(l==0)
+               {
+               item.SetBackgroundColour(wxColour(r,g,b)); 
+                       mColorMap.insert(NodeColorPair(*j,wxColour(r,g,b)));
+               }
+               else if(l!=mLevelList.size()-1)
+               {
+                       item.SetBackgroundColour(mColorMap[*i]); 
+                       mColorMap.insert(NodeColorPair(*j,mColorMap[*i]));
+               }
+               else
+               {
+                       item.SetBackgroundColour(mColorMap[*i]); 
+               }
+                       r+=20;
+                       g+=20;
+                       b+=20;*/
+
            item.SetColumn(0);
+
+               //Sets the last level as selected....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++)
              {
-
+  
                std::string val = (*j)->GetAttribute(mLevelList[l].key[k-1]);
                if (val.size()==0) val = "?";
                item.SetText( crea::std2wx(val));
@@ -299,7 +396,7 @@ namespace creaImageIO
 
 
   //================================================================
-  void WxTreeView::OnSelected(wxListEvent& event)
+  void WxTreeView::OnSelectedChanged(wxListEvent& event)
   { 
     GimmickDebugMessage(1,
                        GetTreeHandler()->GetTree().GetLabel()
@@ -308,8 +405,8 @@ namespace creaImageIO
 
     wxListItem info;
     info.m_itemId = event.m_itemIndex;
-    
-
+    mLastSelected=event.m_itemIndex;
+       
     // retrieve the level
     wxObject* obj = event.GetEventObject();   
     unsigned int level = 0;
@@ -320,13 +417,203 @@ namespace creaImageIO
     GimmickDebugMessage(1,
                        " Level "<<level+1
                        <<std::endl);
+
+       
+       mSelected=GetSelected(level+2);
+       
     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;
+    GimmickDebugMessage(1,
+               " Column selected: " <<event.m_col
+                       <<std::endl);
+
+       if(mColumnSelected!=0)
+       {
+               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);
+               }       
+                 
+               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
+               std::vector<tree::Node*>::iterator selection;
+               std::vector<long> change;
+               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]);
+                       tree::Node* n= ((ItemData*)((long*)item)[0])->node;
+                       for(selection=mSelected.begin();selection!=mSelected.end();++selection)
+                       {
+                               if((*selection)->GetAttribute("ID").compare(n->GetAttribute("ID"))==0)
+                               {
+                                       change.push_back(it);   
+                               }
+                       }
+                       
+               }
+               std::vector<long>::iterator selectedIts;
+               for(selectedIts=change.begin();selectedIts!=change.end();++selectedIts)
+               {
+                       GetCtrl(level)->SetItemState(*selectedIts,wxLIST_STATE_SELECTED, wxLIST_MASK_STATE 
+               | wxLIST_MASK_TEXT |wxLIST_MASK_IMAGE | wxLIST_MASK_DATA | wxLIST_MASK_WIDTH | wxLIST_MASK_FORMAT);
+                       
+               }
+       }
+       
+   }
+  //================================================================
+
+
+  //================================================================
+  void WxTreeView::ValidateSelectedImages()
+  {
+    GimmickDebugMessage(7,
+                       "WxTreeView::ValidateSelectedImages" 
+                       <<std::endl);
+    std::vector<tree::Node*> sel(GetSelected(mLevelList.size()+1));
+    GetGimmickView()->OnSelectionChange(sel);
+    /*
+    //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::GetNodes(std::vector<tree::Node*>& nodes, bool direction)
+  {
+       long item = mLastSelected;
+       int level=mLevelList.size()-1;
+       //Gets current item data
+       long adr = GetCtrl(level)->GetItemData(item);
+       //Extracts the node
+       tree::Node* nod = ((ItemData*)adr)->node;
+    for ( ;; )
+    {
+               if(direction)
+               {
+                       item = GetCtrl(level)->GetNextItem(item,
+                                     wxLIST_NEXT_ABOVE);
+               }
+               else
+               {
+                       item = GetCtrl(level)->GetNextItem(item,
+                                     wxLIST_NEXT_BELOW);
+               }
+        if ( item == -1 )
+               {
+            break;
+               }
+               if(GetCtrl(level)->GetItemState(item, wxLIST_STATE_SELECTED)==0)
+               {
+                       adr = GetCtrl(level)->GetItemData(item);
+                       nod = ((ItemData*)adr)->node;
+                       nodes.push_back(nod);
+               }
+    }
+
+  }
 
+  //================================================================
+  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)
@@ -377,15 +664,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)
@@ -403,4 +693,3 @@ END_EVENT_TABLE()
   
 } // EO namespace creaImageIO
 
-