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::OnSelected(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()
458 "Hello World Validate"
461 MyEvent event( MyCommandEvent, 0 );
462 wxString bar( wxT("This is a Foo_DoFirstThing event") );
463 event.SetText( bar );
464 wxPostEvent( this, event );
469 //================================================================
472 vtkImageData* WxTreeView::GetSelectedImage(int dim)
475 wxArrayTreeItemIds id;
476 // TO DO : TEST THAT STYLE IS MULTIPLE
477 unsigned int nb = mTreeListCtrl->GetSelections(id);
480 // Collect the brute vector of Image nodes
481 std::vector<DicomNode*> im;
482 for (unsigned int i=0; i<nb; ++i)
485 (TreeItemData *)mTreeListCtrl->GetItemData(id[i]);
486 if ((data) && (data->IsDicomNode()))
488 if (data->GetDicomNode()->GetType()==DicomNode::Image)
490 im.push_back ( data->GetDicomNode() );
493 else if (data->GetDicomNode()->GetType()==DicomNode::Series)
495 DicomNode::ChildrenListType::iterator j;
496 for (j =data->GetDicomNode()->GetChildrenList().begin();
497 j!=data->GetDicomNode()->GetChildrenList().end();
505 // Create the output data
508 // Only one image : give it
509 vtkImageData* out = vtkImageData::New();
510 out->ShallowCopy(mReader.GetImage(im.front()->ImageGetFullFileName()));
513 else if (im.size()>1)
515 vtkImageData* first = mReader.GetImage( im.front()->ImageGetFullFileName() );
516 if (first->GetDataDimension()==2)
519 vtkImageData* out = vtkImageData::New();
520 out->CopyStructure(first);
521 out->SetScalarType(first->GetScalarType());
523 first->GetExtent(ext);
526 // LG : TODO : Z Spacing ?
528 out->AllocateScalars();
530 //first->Print(std::cout);
531 // out->Print(std::cout);
534 first->GetDimensions(dim);
535 unsigned long imsize =
536 ( (unsigned long)first->GetScalarPointer(0,1,0)
537 - (unsigned long)first->GetScalarPointer(0,0,0))
541 std::vector<DicomNode*>::iterator it;
542 for (it=im.begin(); it!=im.end(); ++it)
544 //std::cout << "copying slice "<<slice <<std::endl;
545 vtkImageData* cur = mReader.GetImage( (*it)->ImageGetFullFileName() );
547 void* src = cur->GetScalarPointer(0,0,0);
548 void* dst = out->GetScalarPointer(0,0,slice);
549 // std::cout << "src="<<src<<std::endl;
550 // std::cout << "dst="<<dst<<std::endl;
551 // std::cout << "siz="<<imsize<<std::endl;
552 memcpy(dst,src,imsize);
561 std::vector<DicomNode*>::iterator it;
562 for (it=im.begin(); it!=im.end(); ++it)
564 vtkImageData* out = vtkImageData::New();
565 out->ShallowCopy(mReader.GetImage((*it)->ImageGetFullFileName()));
574 void WxTreeView::GetSelectedImages(std::vector<vtkImageData*>& s, int dim)
578 BEGIN_EVENT_TABLE(WxTreeView, wxPanel)
580 EVT_SIZE(MyFrame::OnSize)
582 EVT_MENU(LIST_QUIT, MyFrame::OnQuit)
583 EVT_MENU(LIST_ABOUT, MyFrame::OnAbout)
584 EVT_MENU(LIST_LIST_VIEW, MyFrame::OnListView)
585 EVT_MENU(LIST_REPORT_VIEW, MyFrame::OnReportView)
586 EVT_MENU(LIST_ICON_VIEW, MyFrame::OnIconView)
587 EVT_MENU(LIST_ICON_TEXT_VIEW, MyFrame::OnIconTextView)
588 EVT_MENU(LIST_SMALL_ICON_VIEW, MyFrame::OnSmallIconView)
589 EVT_MENU(LIST_SMALL_ICON_TEXT_VIEW, MyFrame::OnSmallIconTextView)
590 EVT_MENU(LIST_VIRTUAL_VIEW, MyFrame::OnVirtualView)
591 EVT_MENU(LIST_SMALL_VIRTUAL_VIEW, MyFrame::OnSmallVirtualView)
593 EVT_MENU(LIST_FOCUS_LAST, MyFrame::OnFocusLast)
594 EVT_MENU(LIST_TOGGLE_FIRST, MyFrame::OnToggleFirstSel)
595 EVT_MENU(LIST_DESELECT_ALL, MyFrame::OnDeselectAll)
596 EVT_MENU(LIST_SELECT_ALL, MyFrame::OnSelectAll)
597 EVT_MENU(LIST_DELETE, MyFrame::OnDelete)
598 EVT_MENU(LIST_ADD, MyFrame::OnAdd)
599 EVT_MENU(LIST_EDIT, MyFrame::OnEdit)
600 EVT_MENU(LIST_DELETE_ALL, MyFrame::OnDeleteAll)
601 EVT_MENU(LIST_SORT, MyFrame::OnSort)
602 EVT_MENU(LIST_SET_FG_COL, MyFrame::OnSetFgColour)
603 EVT_MENU(LIST_SET_BG_COL, MyFrame::OnSetBgColour)
604 EVT_MENU(LIST_TOGGLE_MULTI_SEL, MyFrame::OnToggleMultiSel)
605 EVT_MENU(LIST_SHOW_COL_INFO, MyFrame::OnShowColInfo)
606 EVT_MENU(LIST_SHOW_SEL_INFO, MyFrame::OnShowSelInfo)
607 EVT_MENU(LIST_FREEZE, MyFrame::OnFreeze)
608 EVT_MENU(LIST_THAW, MyFrame::OnThaw)
609 EVT_MENU(LIST_TOGGLE_LINES, MyFrame::OnToggleLines)
610 EVT_MENU(LIST_MAC_USE_GENERIC, MyFrame::OnToggleMacUseGeneric)
612 EVT_UPDATE_UI(LIST_SHOW_COL_INFO, MyFrame::OnUpdateShowColInfo)
613 EVT_UPDATE_UI(LIST_TOGGLE_MULTI_SEL, MyFrame::OnUpdateToggleMultiSel)
616 BEGIN_EVENT_TABLE(MyListCtrl, wxListCtrl)
617 EVT_LIST_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnBeginDrag)
618 EVT_LIST_BEGIN_RDRAG(LIST_CTRL, MyListCtrl::OnBeginRDrag)
619 EVT_LIST_BEGIN_LABEL_EDIT(LIST_CTRL, MyListCtrl::OnBeginLabelEdit)
620 EVT_LIST_END_LABEL_EDIT(LIST_CTRL, MyListCtrl::OnEndLabelEdit)
621 EVT_LIST_DELETE_ITEM(LIST_CTRL, MyListCtrl::OnDeleteItem)
622 EVT_LIST_DELETE_ALL_ITEMS(LIST_CTRL, MyListCtrl::OnDeleteAllItems)
623 #if WXWIN_COMPATIBILITY_2_4
624 EVT_LIST_GET_INFO(LIST_CTRL, MyListCtrl::OnGetInfo)
625 EVT_LIST_SET_INFO(LIST_CTRL, MyListCtrl::OnSetInfo)
628 EVT_LIST_ITEM_SELECTED(-1, WxTreeView::OnSelected)
630 EVT_LIST_ITEM_DESELECTED(LIST_CTRL, MyListCtrl::OnDeselected)
631 EVT_LIST_KEY_DOWN(LIST_CTRL, MyListCtrl::OnListKeyDown)
632 EVT_LIST_ITEM_ACTIVATED(LIST_CTRL, MyListCtrl::OnActivated)
633 EVT_LIST_ITEM_FOCUSED(LIST_CTRL, MyListCtrl::OnFocused)
635 EVT_LIST_COL_RIGHT_CLICK(LIST_CTRL, MyListCtrl::OnColRightClick)
637 EVT_LIST_COL_CLICK(-1, WxTreeView::OnColClick)
639 EVT_LIST_COL_BEGIN_DRAG(LIST_CTRL, MyListCtrl::OnColBeginDrag)
640 EVT_LIST_COL_DRAGGING(LIST_CTRL, MyListCtrl::OnColDragging)
641 EVT_LIST_COL_END_DRAG(LIST_CTRL, MyListCtrl::OnColEndDrag)
643 EVT_LIST_CACHE_HINT(LIST_CTRL, MyListCtrl::OnCacheHint)
646 EVT_CONTEXT_MENU(MyListCtrl::OnContextMenu)
648 EVT_CHAR(MyListCtrl::OnChar)
650 EVT_RIGHT_DOWN(MyListCtrl::OnRightClick)
654 } // EO namespace creaImageIO