1 #include <creaImageIOWxGimmickView.h>
2 #include <creaImageIOWxTreeView.h>
3 #include <creaImageIOSystem.h>
6 #include "icons/accept.xpm"
7 #include "icons/add.xpm"
8 #include "icons/folder-down.xpm"
9 #include "icons/page-down.xpm"
10 #include "icons/remove.xpm"
11 #include "icons/database-add.xpm"
12 #include "icons/help.xpm"
14 #include <wx/imaglist.h>
15 #include<boost/filesystem/operations.hpp>
20 //======================================================================
21 // The ids of the different tools
27 TOOL_ADDDATABASE_ID = 4,
30 //======================================================================
32 //================================================================
34 const int icon_number = 7;
47 //================================================================
49 //================================================================
51 const icon_id Icon[5] = { Icon_Database,
57 //================================================================
60 //======================================================================
62 WxGimmickView::WxGimmickView(Gimmick* gimmick,
65 const wxPoint& pos, const wxSize& size,
68 int number_of_threads)
69 : wxPanel(parent,id,pos,size),
70 GimmickView(gimmick, number_of_threads),
74 GimmickDebugMessage(1,"WxGimmickView::WxGimmickView"
76 // Sets the current directory to the home dir
77 mCurrentDirectory = std2wx(gimmick->GetHomeDirectory());
79 // Connect the AddProgress callback
80 gimmick->ConnectAddProgressObserver
81 ( boost::bind( &WxGimmickView::OnAddProgress , this, _1 ) );
83 // Create the list of icons (mIcon)
87 wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
89 // Create the tool bar
91 sizer->Add( mToolBar ,0, wxGROW ,0);
93 // Split part below toolbar into notebook for views and panel
94 // for preview, messages...
95 mSplitter = new wxSplitterWindow( this , -1);
99 mNotebook = new wxNotebook(mSplitter,
100 -1,wxDefaultPosition, wxDefaultSize, 0);
106 mSelectionMaxDimension= max_dim;
107 mSelectionMinDimension= min_dim;
113 mBottomPanel = new wxPanel(mSplitter,-1);
115 wxBoxSizer *bottom_sizer = new wxBoxSizer(wxVERTICAL); //HORIZONTAL);
119 mViewer = new WxViewer(mBottomPanel, wxID_ANY, wxT("Gimmick! Viewer"),wxDefaultPosition, wxDefaultSize );
120 pointers.push_back(new ImagePointerHolder(GetDefaultImage()));
121 mViewer->SetImageVector(pointers);
122 mViewer->StartPlayer();
125 bottom_sizer->Add(mViewer,1,wxGROW,1);
128 mText = new wxStaticText(mBottomPanel, wxID_ANY, wxT("Welcome to Gimmick!"));
129 bottom_sizer->Add(mText,0,wxGROW,0);
133 mBottomPanel->SetSizer(bottom_sizer);
136 int hsize = size.GetHeight();
138 int top_minsize = 450;
139 int bottom_minsize = 50;
141 mSplitter->SetMinimumPaneSize( bottom_minsize );
142 mSplitter->SplitHorizontally( mNotebook, mBottomPanel,
145 sizer->Add( mSplitter,1,wxGROW ,0);
154 //======================================================================
156 //======================================================================
158 WxGimmickView::~WxGimmickView()
160 // stop the viewer before application exit.
161 mViewer->StopPlayer();
162 GimmickDebugMessage(1,"WxGimmickView::~WxGimmickView"
165 //======================================================================
167 //======================================================================
168 /// Creates the tool bar
169 void WxGimmickView::CreateToolBar()
171 long style = wxTB_HORIZONTAL | wxNO_BORDER | wxTB_TEXT;
172 mToolBar = new wxToolBar(this,-1,wxDefaultPosition,wxDefaultSize,
175 mToolAddFile = mToolBar->AddTool( TOOL_ADDFILES_ID,
177 mIcon->GetBitmap(Icon_page_down),
178 _T("Add one or more file to database")
180 mToolAddDir = mToolBar->AddTool( TOOL_ADDDIR_ID,
182 mIcon->GetBitmap(Icon_folder_down),
183 _T("Add the content of a folder to database")
185 mToolRemove = mToolBar->AddTool( TOOL_REMOVE_ID,
187 mIcon->GetBitmap(Icon_remove),
188 _T("Remove selected items")
190 mToolAddDatabase = mToolBar->AddTool( TOOL_ADDDATABASE_ID,
192 mIcon->GetBitmap(Icon_database_add),
193 _T("Open a local or distant database")
195 mToolHelp = mToolBar->AddTool( TOOL_HELP_ID,
197 mIcon->GetBitmap(Icon_help),
198 _T("Open help window")
200 //const wxBitmap& bitmap1, const wxString& shortHelpString = "", wxItemKind kind = wxITEM_NORMAL)
204 //======================================================================
207 //======================================================================
208 /// Create the tree view for TreeHandler provided
209 void WxGimmickView::CreateTreeView( TreeHandler* h, TimestampDatabaseHandler* tdh)
211 std::string name(h->GetTree().GetAttribute("Name"));
212 GimmickMessage(2,"Creating the tree view for '"<<
213 name<<"'"<<std::endl);
214 // Create the WxTreeView
215 WxTreeView* view = new WxTreeView(h,tdh,this,mNotebook,-1);
217 // TO DO : TEST THAT A VIEW WITH SAME NAME IS NOT
218 // ALREADY IN THE MAP
219 GetTreeViewMap()[name] = view;
222 mNotebook->AddPage( view, crea::std2wx(name) );
226 //======================================================================
227 void WxGimmickView::GetSelectedImages(std::vector<vtkImageData*>& s, int dim)
229 std::vector<std::string> files;
230 GetTreeViewMap()["Local database"]->GetSelectedAsString(files);
231 ReadImagesNotThreaded(s,files,dim);
233 //======================================================================
235 //======================================================================
236 void WxGimmickView::GetSelectedFiles(std::vector<std::string>& s)
238 GetTreeViewMap()["Local database"]->GetSelectedAsString(s);
240 //======================================================================
242 //======================================================================
243 void WxGimmickView::GetImages(int dim,
244 const std::vector<std::string>& files,
245 std::vector<vtkImageData*>& s)
247 ReadImagesNotThreaded(s,files,dim);
249 //======================================================================
252 //=================================================
253 void WxGimmickView::CreateIconList()
255 // Size of the icons;
259 // should correspond to Icon_xxx enum
260 icons[Icon_accept] = wxIcon(accept_xpm);
261 icons[Icon_add] = wxIcon(add_xpm);
262 icons[Icon_folder_down] = wxIcon(folder_down_xpm);
263 icons[Icon_page_down] = wxIcon(page_down_xpm);
264 icons[Icon_remove] = wxIcon(remove_xpm);
265 icons[Icon_database_add] = wxIcon(database_add_xpm);
266 icons[Icon_help] = wxIcon(help_xpm);
268 // unsigned int NbIcons = 8;
269 // Make an image list containing small icons
270 mIcon = new wxImageList(size,size,true);
272 // Make all icons the same size = size of the first one
273 int sizeOrig = icons[0].GetWidth();
274 for ( size_t i = 0; i < icon_number; i++ )
276 if ( size == sizeOrig )
278 mIcon->Add(icons[i]);
282 mIcon->Add(wxBitmap(wxBitmap(icons[i]).ConvertToImage().Rescale(size, size)));
286 //=================================================
289 //=================================================
290 void WxGimmickView::OnAddFiles(wxCommandEvent& event)
292 long style = wxOPEN | wxFILE_MUST_EXIST | wxFD_MULTIPLE;
293 std::string wc("*.*");
294 wxFileDialog* FD = new wxFileDialog( 0,
302 if (FD->ShowModal()==wxID_OK)
309 std::vector<std::string> filenames;
310 for (i=0;i<files.GetCount();++i)
312 filenames.push_back(wx2std(files[i]));
313 GimmickMessage(2,"Adding File "<<files[i]<<"."<<std::endl);
317 new wxProgressDialog(_T("Adding file(s)"),
322 // wxPD_ESTIMATED_TIME |
323 // wxPD_REMAINING_TIME |
326 // TO DO : select the current tree handler
327 mGimmick->AddFiles("Local database",filenames);
329 mProgressDialog->Pulse(_T("Updating view..."));
331 UpdateTreeViewLevel("Local database",1);
332 delete mProgressDialog;
337 //=================================================
339 //=================================================
340 void WxGimmickView::OnAddDir(wxCommandEvent& event)
342 long style = wxDD_DEFAULT_STYLE | wxDD_DIR_MUST_EXIST;
345 _T("Select directory"),
349 if (FD->ShowModal()==wxID_OK)
352 std::string dirname = wx2std (FD->GetPath());
353 bool recurse = isNeedRecursive(dirname);
356 recurse = wxMessageBox(_T("Recurse into sub-directories ?"), _T("Scan directory"), wxYES_NO,this ) == wxYES ? true : false;
360 wxString title(_T("Adding directory"));
362 title = _T("Adding directory (recursive)");
364 new wxProgressDialog(_T("Adding directory"),
369 // wxPD_ESTIMATED_TIME |
370 // wxPD_REMAINING_TIME |
373 mCurrentDirectory = FD->GetPath();
375 // TO DO : select the current tree handler
376 mGimmick->AddDir("Local database",dirname,recurse);
378 mProgressDialog->Pulse(_T("Updating view..."));
380 UpdateTreeViewLevel("Local database",1);
381 delete mProgressDialog;
386 //=================================================
387 // Test a directory to know if contains sub-directory to analyze
388 bool WxGimmickView::isNeedRecursive(std::string i_name)
390 boost::filesystem::directory_iterator iter(i_name), end_iter;
391 bool bfindir = false;
392 for(; iter != end_iter; ++iter)
394 if(boost::filesystem::is_directory(*iter))
401 //=================================================
403 //=================================================
404 void WxGimmickView::OnSelectionChange(const std::vector<tree::Node*>& sel, bool isSelection, int selection, bool needProcess)
406 GimmickDebugMessage(5,
407 "WxGimmickView::OnSelectionChange"
415 valid= ValidateSelected(NULL,
416 mSelectionMinDimension,
417 mSelectionMaxDimension );
422 std::vector<tree::Node*>::const_iterator i;
423 for(i=sel.begin();i!=sel.end()&&valid;++i)
425 valid= ValidateSelected((*i),
426 mSelectionMinDimension,
427 mSelectionMaxDimension );
432 valid= ValidateSelected(sel.front(),
433 mSelectionMinDimension,
434 mSelectionMaxDimension );
439 std::vector<tree::Node*>::const_iterator i;
440 for(i=sel.begin();i!=sel.end()&&valid;++i)
442 valid= ValidateSelected((*i),
443 mSelectionMinDimension,
444 mSelectionMaxDimension );
448 mText->SetLabel(crea::std2wx(GetMessage()));
451 ReadImageThreaded(sel);
457 ReadImageThreaded(sel);
462 //==================================================
464 //==================================================
465 ///Reads Images (Threaded)
466 void WxGimmickView::ReadImageThreaded(const std::vector<tree::Node*>& sel)
468 GimmickDebugMessage(5,
471 int maxprio = GetMaximalPriority();
472 int prio = maxprio + 2000;
476 //First load the selected images
477 mCurImageItemToShow = sel.front();
480 std::vector<tree::Node*>::const_iterator selected;
481 for(selected=sel.begin();selected!=sel.end();++selected)
483 GimmickDebugMessage(5,
484 "Requesting image from selected "
485 <<(*selected)->GetAttribute("FullFileName")
487 ImagePointerHolder* ph=new ImagePointerHolder(GetDefaultImage());
488 pointers.push_back(ph);
489 RequestReading(*selected,prio,index,ph);
490 // AddEntryToMap(*selected);
494 mViewer->SetImageVector(pointers);
497 std::vector<tree::Node*> up;
498 GetTreeViewMap()["Local database"]->GetNodes(up,true);
499 std::vector<tree::Node*>::iterator iterUp;
500 for(iterUp=up.begin();iterUp!=up.end();++iterUp)
502 GimmickDebugMessage(5,
503 "Requesting image from neighbors up "
504 <<(*iterUp)->GetAttribute("FullFileName")
506 ImagePointerHolder* ph=new ImagePointerHolder(GetDefaultImage());
507 RequestReading(*iterUp,prio,-1,ph);
508 // AddEntryToMap(*iterUp);
510 if (prio == maxprio) break;
515 std::vector<tree::Node*> down;
516 GetTreeViewMap()["Local database"]->GetNodes(down,false);
517 std::vector<tree::Node*>::iterator iterDown;
518 for(iterDown=down.begin();iterDown!=down.end();++iterDown)
520 GimmickDebugMessage(5,
521 "Requesting image from neighbors down "
522 <<(*iterDown)->GetAttribute("FullFileName")
524 ImagePointerHolder* ph=new ImagePointerHolder(GetDefaultImage());
525 RequestReading(*iterDown,prio,-1,ph);
526 // AddEntryToMap(*iterDown);
528 if (prio == maxprio) break;
534 ImagePointerHolder* ph=new ImagePointerHolder(GetDefaultImage());
535 pointers.push_back(ph);
536 mViewer->SetImageVector(pointers);
540 //==================================================
542 //==================================================
544 //==================================================
545 void WxGimmickView::OnInternalIdle()
547 if (!mConstructed) return;
548 static bool first_time = true;
553 // GimmickMessage(1,"WxGimmickView : Refresh viewer"<<std::endl);
554 // mViewer->StartPlayer();
557 mViewer->RefreshIfNecessary();
561 //==================================================
563 //==================================================
564 void WxGimmickView::ClearSelection()
567 pointers.push_back(new ImagePointerHolder(GetDefaultImage()));
568 mViewer->SetImageVector(pointers);
569 mViewer->RefreshIfNecessary();
573 //=================================================
575 //=================================================
576 void WxGimmickView::OnRemove(wxCommandEvent& event)
578 //TODO Select current tree handler
580 GetTreeViewMap()["Local database"]->RemoveSelected();
584 //=================================================
586 //=================================================
587 /// AddProgress Gimmick callback
588 void WxGimmickView::OnAddProgress( Gimmick::AddProgress& p)
593 sprintf(mess,"%i dirs : %i files :\n %i handled - %i added",
594 p.GetNumberScannedDirs(),
595 p.GetNumberScannedFiles(),
596 p.GetNumberHandledFiles(),
597 p.GetNumberAddedFiles());
598 // std::cout << "OnAddProgress "<<mess<<std::endl;
599 wxString s(wxString::From8BitData(mess));
600 // std::cout << "Pulse"<<std::endl;
601 if (!mProgressDialog->Pulse(s))
605 // std::cout << "OnAddProgress ok"<<std::endl;
607 //=================================================
609 //=================================================
610 void WxGimmickView::DisplayAddSummary()
612 const Gimmick::AddProgress& p = mGimmick->GetAddProgress();
613 std::stringstream mess;
614 mess << "Dirs \tscanned\t: " << p.GetNumberScannedDirs() << "\n";
615 mess << "Files\tscanned\t: " << p.GetNumberScannedFiles() << "\n";
616 mess << "Files\thandled\t: " << p.GetNumberHandledFiles() << "\n\n";
617 mess << "Files\tadded \t: " << p.GetNumberAddedFiles() << "\n\n";
620 sprintf(times,"Time to parse dir \t\t: %ld ms \t%d°/o\nTime to read files info \t: %ld ms \t%d°/o\nTime to update structs \t: %ld ms \t%d°/o\nTime to update database \t: %ld ms \t%d°/o\nTotal time \t\t\t: %ld ms",
622 (int)( summary.parse_time*100./summary.total_time),
623 summary.file_scan_time,
624 (int)(summary.file_scan_time*100./summary.total_time),
625 summary.update_structs_time,
626 (int)(summary.update_structs_time*100./summary.total_time),
627 summary.update_database_time,
628 (int)(summary.update_database_time*100./summary.total_time),
629 summary.total_time );
633 wxMessageBox(std2wx(mess.str()),_T("Addition result"),wxOK,this);
636 //=================================================
638 //=================================================
639 BEGIN_EVENT_TABLE(WxGimmickView, wxPanel)
640 EVT_TOOL(TOOL_ADDFILES_ID, WxGimmickView::OnAddFiles)
641 EVT_TOOL(TOOL_ADDDIR_ID, WxGimmickView::OnAddDir)
642 EVT_TOOL(TOOL_REMOVE_ID, WxGimmickView::OnRemove)
644 //=================================================
646 } // EO namespace creaImageIO