1 #include <creaImageIOWxTreeView.h>
2 #include <creaImageIOSystem.h>
3 #include <wx/splitter.h>
4 int wxCALLBACK MyCompareFunction(long item1, long item2, long WXUNUSED(sortData))
16 //=====================================================================
18 WxTreeView::WxTreeView(TreeHandler* handler,
24 GimmickDebugMessage(1,"WxTreeView::WxTreeView"
28 // Split part below toolbar into notebook for views and panel
29 // for preview, messages...
31 // mSplitter = new wxSplitterWindow( this , -1);
34 wxBoxSizer *sizer = new wxBoxSizer(wxHORIZONTAL);
36 int ctrl_style = wxLC_REPORT | wxLC_VRULES;
37 int col_style = wxLIST_FORMAT_LEFT;
39 // Creating the ListCtrl for the levels > 0 (not for Root level)
41 i < handler->GetTree().GetNumberOfLevels();
45 GimmickDebugMessage(5,"Creating view for level "<<i
49 // If the first level : parent = this
50 wxWindow* sparent = this;
51 // else parent = last splitter
52 if (i>1) sparent = mLevelList[i-2].wxSplitter;
54 level.wxSplitter = new wxSplitterWindow( sparent , -1);
55 // level.wxSplitter->SetMinimumPaneSize(100);
57 wxListCtrl* ctrl = new wxListCtrl(level.wxSplitter,
63 level.wxSplitter->Initialize(ctrl);
65 // Create the columns : one for each attribute of the level
67 //ctrl->InsertItem(0, "1");
69 tree::LevelDescriptor::AttributeDescriptorListType::const_iterator a;
70 for (a = handler->GetTree().GetAttributeDescriptorList(i).begin();
71 a != handler->GetTree().GetAttributeDescriptorList(i).end();
77 it.SetTextColour(*wxRED);
80 // ctrl->InsertColumn(col, _("Children"), col_style);
81 ctrl->InsertColumn(col,it);
82 //ctrl->InsertItem(0, "1");
83 //level.key.push_back(_("Children"));
84 //level.key.push_back(handler->GetTree().GetChildrenList().size());
88 GimmickDebugMessage(5,"Creating column "<<col<<" : "
91 ctrl->InsertColumn(col,
92 crea::std2wx(a->GetName()),
94 level.key.push_back(a->GetKey());
95 // ctrl->SetColumnWidth(col, wxLIST_AUTOSIZE );
101 mLevelList.push_back(level);
105 /// Initialize the first level splitter
107 sizer->Add( mLevelList[0].wxSplitter ,1, wxGROW ,0);
116 //=====================================================================
118 //=====================================================================
120 WxTreeView::~WxTreeView()
122 GimmickDebugMessage(1,"WxTreeView::~WxTreeView"
125 //=====================================================================
128 //=====================================================================
134 //=====================================================================
135 std::vector<tree::Node*> WxTreeView::GetSelected(int level)
138 // the selection of upper level
139 std::vector<tree::Node*> sel;
143 sel.push_back(GetTreeHandler()->GetTree().GetTree());
147 int n = GetCtrl(l-1)->GetItemCount();
148 for (int i = 0; i < n; i++)
150 if ( GetCtrl(l-1)->GetItemState(i,wxLIST_STATE_SELECTED))
152 long adr = GetCtrl(l-1)->GetItemData(i);
153 tree::Node* n = ((ItemData*)adr)->node;
162 //=====================================================================
164 ///Removes selected nodes on given level
165 void WxTreeView::RemoveSelected( int level )
167 std::vector<tree::Node*> sel=GetSelected(level+1);
169 if (wxMessageBox(_T("Delete file(s) ?"),
171 wxYES_NO,this ) == wxYES)
177 std::vector<tree::Node*>::iterator i;
178 for (i=sel.begin(); i!=sel.end(); ++i)
180 GimmickDebugMessage(2,
185 GetTreeHandler()->Remove(*i);
194 //=====================================================================
197 //=====================================================================
199 void WxTreeView::UpdateLevel( int level )
201 GimmickDebugMessage(1,
202 GetTreeHandler()->GetTree().GetLabel()
203 <<" view : updating level "<<level
207 RecursiveUpdateLevel(level);
209 for (i=0; i<level-1; i++)
211 if (!GetSplitter(i)->IsSplit())
212 GetSplitter(i)->SplitVertically( GetCtrl(i), GetSplitter(i+1),
215 if (GetSplitter(i)->IsSplit()) GetSplitter(i)->Unsplit();
218 //=====================================================================
221 void WxTreeView::RecursiveUpdateLevel( int level )
223 GimmickDebugMessage(2,
224 GetTreeHandler()->GetTree().GetLabel()
225 <<" view : updating level (recursive)"<<level
229 std::vector<tree::Node*> sel=GetSelected(level);
233 // to speed up inserting we hide the control temporarily
235 GetCtrl(l)->DeleteAllItems();
237 std::vector<tree::Node*>::iterator i;
238 for (i=sel.begin(); i!=sel.end(); ++i)
240 GimmickDebugMessage(2,
241 "adding children of '"
247 GetTreeHandler()->LoadChildren(*i,1);
248 tree::Node::ChildrenListType::reverse_iterator j;
249 for (j = (*i)->GetChildrenList().rbegin();
250 j!= (*i)->GetChildrenList().rend();
254 item.SetMask(wxLIST_MASK_STATE |
256 // wxLIST_MASK_IMAGE |
258 // wxLIST_MASK_WIDTH |
262 ItemData* data = new ItemData;
267 long id = GetCtrl(l)->InsertItem(item);
269 std::ostringstream oss;
270 int n= GetTreeHandler()->GetNumberOfChildren(*j);
273 std::string s(oss.str());
275 item.SetText( crea::std2wx(s));
276 // item.SetTextColour(*wxRED);
277 // item.SetBackgroundColour(*wxBLACK);
280 //Sets the last level as selecte....How to make it select only the first time?
281 //if (level==mLevelList.size()) item.SetState(wxLIST_STATE_SELECTED);
283 GetCtrl(l)->SetItem(item);
285 // GetCtrl(l)->SetItem(id,0, crea::std2wx(s));
286 // GetCtrl(l)->SetColumnWidth(0, wxLIST_AUTOSIZE );
288 for (int k=1; k<GetCtrl(l)->GetColumnCount(); k++)
291 std::string val = (*j)->GetAttribute(mLevelList[l].key[k-1]);
292 if (val.size()==0) val = "?";
293 item.SetText( crea::std2wx(val));
294 // item.SetTextColour(*wxBLACK);
295 // item.SetBackgroundColour(*wxWHITE);
297 GetCtrl(l)->SetItem(item);
299 // GetCtrl(l)->SetItem(id,k, crea::std2wx(val));
300 // GetCtrl(l)->SetColumnWidth(k, wxLIST_AUTOSIZE );
310 if (level<mLevelList.size()) UpdateLevel(level+1);
312 //=====================================================================
315 //================================================================
316 void WxTreeView::OnSelectedChanged(wxListEvent& event)
318 GimmickDebugMessage(1,
319 GetTreeHandler()->GetTree().GetLabel()
320 <<" view : item selected "
324 info.m_itemId = event.m_itemIndex;
327 // retrieve the level
328 wxObject* obj = event.GetEventObject();
329 unsigned int level = 0;
330 for (level = 0; level<mLevelList.size(); ++level)
332 if ( GetCtrl(level) == obj ) break;
334 GimmickDebugMessage(1,
337 if (level<mLevelList.size()-1) UpdateLevel( level + 2 );
338 if (level==mLevelList.size()-1) ValidateSelectedImages ();
341 //================================================================
343 //================================================================
344 void WxTreeView::OnColClick(wxListEvent& event)
347 //Obtain the column name and the level that needs to be organized
348 int colNum=event.m_col;
350 " Column " <<event.m_col
354 wxObject* ctrl = event.GetEventObject();
355 unsigned int level = 0;
356 for (level = 0; level<mLevelList.size(); ++level)
358 if ( GetCtrl(level) == ctrl ) break;
362 std::vector<tree::Node*> sel=GetSelected(level+1);
364 tree::Node* nodes[20];
368 // to speed up inserting we hide the control temporarily
369 GetCtrl(level)->Hide();
370 GetCtrl(level)->DeleteAllItems();
372 std::vector<tree::Node*>::iterator i;
374 for (i=sel.begin(); i!=sel.end(); ++i)
378 GetTreeHandler()->LoadChildren(*i,1);
379 tree::Node::ChildrenListType::reverse_iterator j;
381 for (j = (*i)->GetChildrenList().rbegin();
382 j!= (*i)->GetChildrenList().rend();
395 for (k=1; k<num; ++k)
401 <<(nodes[m-1])->GetAttribute(mLevelList[level].key[colNum-1])
402 <<"Comparison Key 2: "<<(index)->GetAttribute(mLevelList[level].key[colNum-1])
404 while ((m > 0) && ((nodes[m-1])->GetAttribute(mLevelList[level].key[colNum-1]) >
405 (index)->GetAttribute(mLevelList[level].key[colNum-1])))
407 nodes[m] = nodes[m-1];
413 for (m=0; m<num; ++m)
415 tree::Node* node=nodes[m];
417 item.SetMask(wxLIST_MASK_STATE |
419 // wxLIST_MASK_IMAGE |
421 // wxLIST_MASK_WIDTH |
425 long id = GetCtrl(level)->InsertItem(item);
427 if ((node)->GetChildrenLoaded())
429 n=(node)->GetChildrenList().size();
433 n= GetTreeHandler()->GetNumberOfChildren(node);
435 std::ostringstream oss;
438 std::string s(oss.str());
440 GetCtrl(level)->SetItem(id,0, crea::std2wx(s));
441 GetCtrl(level)->SetColumnWidth(0, wxLIST_AUTOSIZE );
443 for (int k=1; k<GetCtrl(level)->GetColumnCount(); k++)
445 std::string val = (node)->GetAttribute(mLevelList[level].key[k-1]);
446 if (val.size()==0) val = "?";
447 GetCtrl(level)->SetItem(id,k, crea::std2wx(val));
448 GetCtrl(level)->SetColumnWidth(k, wxLIST_AUTOSIZE );
455 GetCtrl(level)->Show();
458 //================================================================
460 void WxTreeView::ValidateSelectedImages()
462 int level=mLevelList.size();
463 std::vector<tree::Node*> sel=GetSelected(level+1);
468 std::vector<tree::Node*>::iterator i;
474 //Validation between image sizes
475 for (i=sel.begin(); i!=sel.end() && valid; ++i)
479 row=(*i)->GetAttribute(mLevelList[level-1].key[1]);
480 col=(*i)->GetAttribute(mLevelList[level-1].key[2]);
481 plane=(*i)->GetAttribute(mLevelList[level-1].key[3]);
485 if(((*i)->GetAttribute(mLevelList[level-1].key[1]))!=row ||
486 ((*i)->GetAttribute(mLevelList[level-1].key[2]))!=col ||
487 ((*i)->GetAttribute(mLevelList[level-1].key[3]))!=plane)
489 mess="The selected images are not compatible.";
495 //Dimention validation
496 //Compatibility with maximum
504 std::istringstream t(s);
513 if(plane==""){planes=1;}
517 else if (cols>1) dim=2;
518 else if (rows>1) dim=1;
522 mess="Unknown image dimension : cannot select !";
525 else if (dim>GetMaxDimension())
529 mess+="D images is not allowed !";
532 if ( dim == GetMaxDimension() )
534 mess="Cannot add this image to selection : would result in a ";
540 if ( dim < GetMinDimension() && sel.size()<2 )
542 mess="Cannot build the selection as it would result in a ";
544 mess+="D image, and the minimum is ";
545 mess+=GetMinDimension();
553 mess="Cannot have 0 images selected";
557 //Send an event telling wether the selection is valid or not
558 wxCommandEvent event( 0, GetId() );
559 event.SetEventObject( this );
562 mess="Selection OK !";
567 event.SetString(crea::std2wx(mess));
568 GetEventHandler()->ProcessEvent( event );
573 //================================================================
574 vtkImageData* WxTreeView::GetSelectedImage(int dim)
580 void WxTreeView::GetSelectedImages(std::vector<vtkImageData*>& f, int dim)
583 std::vector<tree::Node*> im=GetSelected(level+1);
585 // Create the output data
588 // Only one image : give it
589 vtkImageData* out = vtkImageData::New();
590 out->ShallowCopy(mReader.GetImage(im.front()->ImageGetFullFileName()));
593 else if (im.size()>1)
595 vtkImageData* first = mReader.GetImage( im.front()->ImageGetFullFileName() );
596 if (first->GetDataDimension()==2)
599 vtkImageData* out = vtkImageData::New();
600 out->CopyStructure(first);
601 out->SetScalarType(first->GetScalarType());
603 first->GetExtent(ext);
606 // LG : TODO : Z Spacing ?
608 out->AllocateScalars();
610 //first->Print(std::cout);
611 // out->Print(std::cout);
614 first->GetDimensions(dim);
615 unsigned long imsize =
616 ( (unsigned long)first->GetScalarPointer(0,1,0)
617 - (unsigned long)first->GetScalarPointer(0,0,0))
621 std::vector<DicomNode*>::iterator it;
622 for (it=im.begin(); it!=im.end(); ++it)
624 //std::cout << "copying slice "<<slice <<std::endl;
625 vtkImageData* cur = mReader.GetImage( (*it)->ImageGetFullFileName() );
627 void* src = cur->GetScalarPointer(0,0,0);
628 void* dst = out->GetScalarPointer(0,0,slice);
629 // std::cout << "src="<<src<<std::endl;
630 // std::cout << "dst="<<dst<<std::endl;
631 // std::cout << "siz="<<imsize<<std::endl;
632 memcpy(dst,src,imsize);
641 std::vector<DicomNode*>::iterator it;
642 for (it=im.begin(); it!=im.end(); ++it)
644 vtkImageData* out = vtkImageData::New();
645 out->ShallowCopy(mReader.GetImage((*it)->ImageGetFullFileName()));
652 BEGIN_EVENT_TABLE(WxTreeView, wxPanel)
654 EVT_SIZE(MyFrame::OnSize)
656 EVT_MENU(LIST_QUIT, MyFrame::OnQuit)
657 EVT_MENU(LIST_ABOUT, MyFrame::OnAbout)
658 EVT_MENU(LIST_LIST_VIEW, MyFrame::OnListView)
659 EVT_MENU(LIST_REPORT_VIEW, MyFrame::OnReportView)
660 EVT_MENU(LIST_ICON_VIEW, MyFrame::OnIconView)
661 EVT_MENU(LIST_ICON_TEXT_VIEW, MyFrame::OnIconTextView)
662 EVT_MENU(LIST_SMALL_ICON_VIEW, MyFrame::OnSmallIconView)
663 EVT_MENU(LIST_SMALL_ICON_TEXT_VIEW, MyFrame::OnSmallIconTextView)
664 EVT_MENU(LIST_VIRTUAL_VIEW, MyFrame::OnVirtualView)
665 EVT_MENU(LIST_SMALL_VIRTUAL_VIEW, MyFrame::OnSmallVirtualView)
667 EVT_MENU(LIST_FOCUS_LAST, MyFrame::OnFocusLast)
668 EVT_MENU(LIST_TOGGLE_FIRST, MyFrame::OnToggleFirstSel)
669 EVT_MENU(LIST_DESELECT_ALL, MyFrame::OnDeselectAll)
670 EVT_MENU(LIST_SELECT_ALL, MyFrame::OnSelectAll)
671 EVT_MENU(LIST_DELETE, MyFrame::OnDelete)
672 EVT_MENU(LIST_ADD, MyFrame::OnAdd)
673 EVT_MENU(LIST_EDIT, MyFrame::OnEdit)
674 EVT_MENU(LIST_DELETE_ALL, MyFrame::OnDeleteAll)
675 EVT_MENU(LIST_SORT, MyFrame::OnSort)
676 EVT_MENU(LIST_SET_FG_COL, MyFrame::OnSetFgColour)
677 EVT_MENU(LIST_SET_BG_COL, MyFrame::OnSetBgColour)
678 EVT_MENU(LIST_TOGGLE_MULTI_SEL, MyFrame::OnToggleMultiSel)
679 EVT_MENU(LIST_SHOW_COL_INFO, MyFrame::OnShowColInfo)
680 EVT_MENU(LIST_SHOW_SEL_INFO, MyFrame::OnShowSelInfo)
681 EVT_MENU(LIST_FREEZE, MyFrame::OnFreeze)
682 EVT_MENU(LIST_THAW, MyFrame::OnThaw)
683 EVT_MENU(LIST_TOGGLE_LINES, MyFrame::OnToggleLines)
684 EVT_MENU(LIST_MAC_USE_GENERIC, MyFrame::OnToggleMacUseGeneric)
686 EVT_UPDATE_UI(LIST_SHOW_COL_INFO, MyFrame::OnUpdateShowColInfo)
687 EVT_UPDATE_UI(LIST_TOGGLE_MULTI_SEL, MyFrame::OnUpdateToggleMultiSel)
690 BEGIN_EVENT_TABLE(MyListCtrl, wxListCtrl)
691 EVT_LIST_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnBeginDrag)
692 EVT_LIST_BEGIN_RDRAG(LIST_CTRL, MyListCtrl::OnBeginRDrag)
693 EVT_LIST_BEGIN_LABEL_EDIT(LIST_CTRL, MyListCtrl::OnBeginLabelEdit)
694 EVT_LIST_END_LABEL_EDIT(LIST_CTRL, MyListCtrl::OnEndLabelEdit)
695 EVT_LIST_DELETE_ITEM(LIST_CTRL, MyListCtrl::OnDeleteItem)
696 EVT_LIST_DELETE_ALL_ITEMS(LIST_CTRL, MyListCtrl::OnDeleteAllItems)
697 #if WXWIN_COMPATIBILITY_2_4
698 EVT_LIST_GET_INFO(LIST_CTRL, MyListCtrl::OnGetInfo)
699 EVT_LIST_SET_INFO(LIST_CTRL, MyListCtrl::OnSetInfo)
702 EVT_LIST_ITEM_SELECTED(-1, WxTreeView::OnSelectedChanged)
704 EVT_LIST_ITEM_DESELECTED(-1, WxTreeView::OnSelectedChanged)
706 EVT_LIST_KEY_DOWN(LIST_CTRL, MyListCtrl::OnListKeyDown)
707 EVT_LIST_ITEM_ACTIVATED(LIST_CTRL, MyListCtrl::OnActivated)
708 EVT_LIST_ITEM_FOCUSED(LIST_CTRL, MyListCtrl::OnFocused)
710 EVT_LIST_COL_RIGHT_CLICK(LIST_CTRL, MyListCtrl::OnColRightClick)
712 EVT_LIST_COL_CLICK(-1, WxTreeView::OnColClick)
714 EVT_LIST_COL_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnColBeginDrag)
715 EVT_LIST_COL_DRAGGING(LIST_CTRL, MyListCtrl::OnColDragging)
716 EVT_LIST_COL_END_DRAG(LIST_CTRL, MyListCtrl::OnColEndDrag)
718 EVT_LIST_CACHE_HINT(LIST_CTRL, MyListCtrl::OnCacheHint)
721 EVT_CONTEXT_MENU(MyListCtrl::OnContextMenu)
723 EVT_CHAR(MyListCtrl::OnChar)
725 EVT_RIGHT_DOWN(MyListCtrl::OnRightClick)
729 } // EO namespace creaImageIO