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);
279 GetCtrl(l)->SetItem(item);
280 // GetCtrl(l)->SetItem(id,0, crea::std2wx(s));
281 // GetCtrl(l)->SetColumnWidth(0, wxLIST_AUTOSIZE );
283 for (int k=1; k<GetCtrl(l)->GetColumnCount(); k++)
286 std::string val = (*j)->GetAttribute(mLevelList[l].key[k-1]);
287 if (val.size()==0) val = "?";
288 item.SetText( crea::std2wx(val));
289 // item.SetTextColour(*wxBLACK);
290 // item.SetBackgroundColour(*wxWHITE);
292 GetCtrl(l)->SetItem(item);
294 // GetCtrl(l)->SetItem(id,k, crea::std2wx(val));
295 // GetCtrl(l)->SetColumnWidth(k, wxLIST_AUTOSIZE );
305 if (level<mLevelList.size()) UpdateLevel(level+1);
307 //=====================================================================
310 //================================================================
311 void WxTreeView::OnSelectedChanged(wxListEvent& event)
313 GimmickDebugMessage(1,
314 GetTreeHandler()->GetTree().GetLabel()
315 <<" view : item selected "
319 info.m_itemId = event.m_itemIndex;
322 // retrieve the level
323 wxObject* obj = event.GetEventObject();
324 unsigned int level = 0;
325 for (level = 0; level<mLevelList.size(); ++level)
327 if ( GetCtrl(level) == obj ) break;
329 GimmickDebugMessage(1,
332 if (level<mLevelList.size()-1) UpdateLevel( level + 2 );
333 if (level==mLevelList.size()-1) ValidateSelectedImages ();
336 //================================================================
338 //================================================================
339 void WxTreeView::OnColClick(wxListEvent& event)
342 //Obtain the column name and the level that needs to be organized
343 int colNum=event.m_col;
345 " Column " <<event.m_col
349 wxObject* ctrl = event.GetEventObject();
350 unsigned int level = 0;
351 for (level = 0; level<mLevelList.size(); ++level)
353 if ( GetCtrl(level) == ctrl ) break;
357 std::vector<tree::Node*> sel=GetSelected(level+1);
359 tree::Node* nodes[20];
363 // to speed up inserting we hide the control temporarily
364 GetCtrl(level)->Hide();
365 GetCtrl(level)->DeleteAllItems();
367 std::vector<tree::Node*>::iterator i;
369 for (i=sel.begin(); i!=sel.end(); ++i)
373 GetTreeHandler()->LoadChildren(*i,1);
374 tree::Node::ChildrenListType::reverse_iterator j;
376 for (j = (*i)->GetChildrenList().rbegin();
377 j!= (*i)->GetChildrenList().rend();
390 for (k=1; k<num; ++k)
396 <<(nodes[m-1])->GetAttribute(mLevelList[level].key[colNum-1])
397 <<"Comparison Key 2: "<<(index)->GetAttribute(mLevelList[level].key[colNum-1])
399 while ((m > 0) && ((nodes[m-1])->GetAttribute(mLevelList[level].key[colNum-1]) >
400 (index)->GetAttribute(mLevelList[level].key[colNum-1])))
402 nodes[m] = nodes[m-1];
408 for (m=0; m<num; ++m)
410 tree::Node* node=nodes[m];
412 item.SetMask(wxLIST_MASK_STATE |
414 // wxLIST_MASK_IMAGE |
416 // wxLIST_MASK_WIDTH |
420 long id = GetCtrl(level)->InsertItem(item);
422 if ((node)->GetChildrenLoaded())
424 n=(node)->GetChildrenList().size();
428 n= GetTreeHandler()->GetNumberOfChildren(node);
430 std::ostringstream oss;
433 std::string s(oss.str());
435 GetCtrl(level)->SetItem(id,0, crea::std2wx(s));
436 GetCtrl(level)->SetColumnWidth(0, wxLIST_AUTOSIZE );
438 for (int k=1; k<GetCtrl(level)->GetColumnCount(); k++)
440 std::string val = (node)->GetAttribute(mLevelList[level].key[k-1]);
441 if (val.size()==0) val = "?";
442 GetCtrl(level)->SetItem(id,k, crea::std2wx(val));
443 GetCtrl(level)->SetColumnWidth(k, wxLIST_AUTOSIZE );
450 GetCtrl(level)->Show();
453 //================================================================
455 void WxTreeView::ValidateSelectedImages()
457 int level=mLevelList.size();
458 std::vector<tree::Node*> sel=GetSelected(level+1);
461 std::vector<tree::Node*>::iterator i;
467 //Validation between image sizes
468 for (i=sel.begin(); i!=sel.end() && valid; ++i)
472 row=(*i)->GetAttribute(mLevelList[level-1].key[1]);
473 col=(*i)->GetAttribute(mLevelList[level-1].key[2]);
474 plane=(*i)->GetAttribute(mLevelList[level-1].key[3]);
478 if(((*i)->GetAttribute(mLevelList[level-1].key[1]))!=row ||
479 ((*i)->GetAttribute(mLevelList[level-1].key[2]))!=col ||
480 ((*i)->GetAttribute(mLevelList[level-1].key[3]))!=plane)
488 "State check: Planes:"
493 //Dimention validation
494 //Compatibility with maximum
501 std::istringstream t(s);
510 if(plane==""){planes=1;}
514 else if (cols>1) dim=2;
515 else if (rows>1) dim=1;
520 "Unknown image dimension : cannot select !"
528 else if (dim>GetMaxDimension())
530 GimmickMessage(1,"Selecting "<<dim<<"D images is not allowed !"
534 if ( dim == GetMaxDimension() )
536 GimmickMessage(1,"Cannot add this image to selection : would result in a "<<dim+1<<"D image !" << std::endl);
541 //Send an event telling wether the selection is valid or not
542 wxCommandEvent event( 0, GetId() );
543 event.SetEventObject( this );
548 GetEventHandler()->ProcessEvent( event );
553 //================================================================
556 vtkImageData* WxTreeView::GetSelectedImage(int dim)
559 wxArrayTreeItemIds id;
560 // TO DO : TEST THAT STYLE IS MULTIPLE
561 unsigned int nb = mTreeListCtrl->GetSelections(id);
564 // Collect the brute vector of Image nodes
565 std::vector<DicomNode*> im;
566 for (unsigned int i=0; i<nb; ++i)
569 (TreeItemData *)mTreeListCtrl->GetItemData(id[i]);
570 if ((data) && (data->IsDicomNode()))
572 if (data->GetDicomNode()->GetType()==DicomNode::Image)
574 im.push_back ( data->GetDicomNode() );
577 else if (data->GetDicomNode()->GetType()==DicomNode::Series)
579 DicomNode::ChildrenListType::iterator j;
580 for (j =data->GetDicomNode()->GetChildrenList().begin();
581 j!=data->GetDicomNode()->GetChildrenList().end();
589 // Create the output data
592 // Only one image : give it
593 vtkImageData* out = vtkImageData::New();
594 out->ShallowCopy(mReader.GetImage(im.front()->ImageGetFullFileName()));
597 else if (im.size()>1)
599 vtkImageData* first = mReader.GetImage( im.front()->ImageGetFullFileName() );
600 if (first->GetDataDimension()==2)
603 vtkImageData* out = vtkImageData::New();
604 out->CopyStructure(first);
605 out->SetScalarType(first->GetScalarType());
607 first->GetExtent(ext);
610 // LG : TODO : Z Spacing ?
612 out->AllocateScalars();
614 //first->Print(std::cout);
615 // out->Print(std::cout);
618 first->GetDimensions(dim);
619 unsigned long imsize =
620 ( (unsigned long)first->GetScalarPointer(0,1,0)
621 - (unsigned long)first->GetScalarPointer(0,0,0))
625 std::vector<DicomNode*>::iterator it;
626 for (it=im.begin(); it!=im.end(); ++it)
628 //std::cout << "copying slice "<<slice <<std::endl;
629 vtkImageData* cur = mReader.GetImage( (*it)->ImageGetFullFileName() );
631 void* src = cur->GetScalarPointer(0,0,0);
632 void* dst = out->GetScalarPointer(0,0,slice);
633 // std::cout << "src="<<src<<std::endl;
634 // std::cout << "dst="<<dst<<std::endl;
635 // std::cout << "siz="<<imsize<<std::endl;
636 memcpy(dst,src,imsize);
645 std::vector<DicomNode*>::iterator it;
646 for (it=im.begin(); it!=im.end(); ++it)
648 vtkImageData* out = vtkImageData::New();
649 out->ShallowCopy(mReader.GetImage((*it)->ImageGetFullFileName()));
658 void WxTreeView::GetSelectedImages(std::vector<vtkImageData*>& s, int dim)
662 BEGIN_EVENT_TABLE(WxTreeView, wxPanel)
664 EVT_SIZE(MyFrame::OnSize)
666 EVT_MENU(LIST_QUIT, MyFrame::OnQuit)
667 EVT_MENU(LIST_ABOUT, MyFrame::OnAbout)
668 EVT_MENU(LIST_LIST_VIEW, MyFrame::OnListView)
669 EVT_MENU(LIST_REPORT_VIEW, MyFrame::OnReportView)
670 EVT_MENU(LIST_ICON_VIEW, MyFrame::OnIconView)
671 EVT_MENU(LIST_ICON_TEXT_VIEW, MyFrame::OnIconTextView)
672 EVT_MENU(LIST_SMALL_ICON_VIEW, MyFrame::OnSmallIconView)
673 EVT_MENU(LIST_SMALL_ICON_TEXT_VIEW, MyFrame::OnSmallIconTextView)
674 EVT_MENU(LIST_VIRTUAL_VIEW, MyFrame::OnVirtualView)
675 EVT_MENU(LIST_SMALL_VIRTUAL_VIEW, MyFrame::OnSmallVirtualView)
677 EVT_MENU(LIST_FOCUS_LAST, MyFrame::OnFocusLast)
678 EVT_MENU(LIST_TOGGLE_FIRST, MyFrame::OnToggleFirstSel)
679 EVT_MENU(LIST_DESELECT_ALL, MyFrame::OnDeselectAll)
680 EVT_MENU(LIST_SELECT_ALL, MyFrame::OnSelectAll)
681 EVT_MENU(LIST_DELETE, MyFrame::OnDelete)
682 EVT_MENU(LIST_ADD, MyFrame::OnAdd)
683 EVT_MENU(LIST_EDIT, MyFrame::OnEdit)
684 EVT_MENU(LIST_DELETE_ALL, MyFrame::OnDeleteAll)
685 EVT_MENU(LIST_SORT, MyFrame::OnSort)
686 EVT_MENU(LIST_SET_FG_COL, MyFrame::OnSetFgColour)
687 EVT_MENU(LIST_SET_BG_COL, MyFrame::OnSetBgColour)
688 EVT_MENU(LIST_TOGGLE_MULTI_SEL, MyFrame::OnToggleMultiSel)
689 EVT_MENU(LIST_SHOW_COL_INFO, MyFrame::OnShowColInfo)
690 EVT_MENU(LIST_SHOW_SEL_INFO, MyFrame::OnShowSelInfo)
691 EVT_MENU(LIST_FREEZE, MyFrame::OnFreeze)
692 EVT_MENU(LIST_THAW, MyFrame::OnThaw)
693 EVT_MENU(LIST_TOGGLE_LINES, MyFrame::OnToggleLines)
694 EVT_MENU(LIST_MAC_USE_GENERIC, MyFrame::OnToggleMacUseGeneric)
696 EVT_UPDATE_UI(LIST_SHOW_COL_INFO, MyFrame::OnUpdateShowColInfo)
697 EVT_UPDATE_UI(LIST_TOGGLE_MULTI_SEL, MyFrame::OnUpdateToggleMultiSel)
700 BEGIN_EVENT_TABLE(MyListCtrl, wxListCtrl)
701 EVT_LIST_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnBeginDrag)
702 EVT_LIST_BEGIN_RDRAG(LIST_CTRL, MyListCtrl::OnBeginRDrag)
703 EVT_LIST_BEGIN_LABEL_EDIT(LIST_CTRL, MyListCtrl::OnBeginLabelEdit)
704 EVT_LIST_END_LABEL_EDIT(LIST_CTRL, MyListCtrl::OnEndLabelEdit)
705 EVT_LIST_DELETE_ITEM(LIST_CTRL, MyListCtrl::OnDeleteItem)
706 EVT_LIST_DELETE_ALL_ITEMS(LIST_CTRL, MyListCtrl::OnDeleteAllItems)
707 #if WXWIN_COMPATIBILITY_2_4
708 EVT_LIST_GET_INFO(LIST_CTRL, MyListCtrl::OnGetInfo)
709 EVT_LIST_SET_INFO(LIST_CTRL, MyListCtrl::OnSetInfo)
712 EVT_LIST_ITEM_SELECTED(-1, WxTreeView::OnSelectedChanged)
714 EVT_LIST_ITEM_DESELECTED(-1, WxTreeView::OnSelectedChanged)
716 EVT_LIST_KEY_DOWN(LIST_CTRL, MyListCtrl::OnListKeyDown)
717 EVT_LIST_ITEM_ACTIVATED(LIST_CTRL, MyListCtrl::OnActivated)
718 EVT_LIST_ITEM_FOCUSED(LIST_CTRL, MyListCtrl::OnFocused)
720 EVT_LIST_COL_RIGHT_CLICK(LIST_CTRL, MyListCtrl::OnColRightClick)
722 EVT_LIST_COL_CLICK(-1, WxTreeView::OnColClick)
724 EVT_LIST_COL_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnColBeginDrag)
725 EVT_LIST_COL_DRAGGING(LIST_CTRL, MyListCtrl::OnColDragging)
726 EVT_LIST_COL_END_DRAG(LIST_CTRL, MyListCtrl::OnColEndDrag)
728 EVT_LIST_CACHE_HINT(LIST_CTRL, MyListCtrl::OnCacheHint)
731 EVT_CONTEXT_MENU(MyListCtrl::OnContextMenu)
733 EVT_CHAR(MyListCtrl::OnChar)
735 EVT_RIGHT_DOWN(MyListCtrl::OnRightClick)
739 } // EO namespace creaImageIO