src2/creaImageIOWxGimmickView.cpp

   1 #include <creaImageIOWxGimmickView.h>
   2 #include <creaImageIOWxTreeView.h>
   3 #include <creaImageIOSystem.h>
   4 using namespace crea;
   5 // Icons
   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"
  13
  14 #include <wx/imaglist.h>
  15
  16 namespace creaImageIO
  17 {
  18   //======================================================================
  19   // The ids of the different tools
  20   enum
  21     {
  22       TOOL_ADDFILES_ID = 1,
  23       TOOL_ADDDIR_ID = 2,
  24       TOOL_REMOVE_ID = 3,
  25       TOOL_ADDDATABASE_ID = 4,
  26       TOOL_HELP_ID = 5
  27     };
  28   //======================================================================
  29
  30   //================================================================
  31   //
  32   const int icon_number = 7;
  33   // Icon ids
  34   typedef enum
  35     {
  36       Icon_accept,
  37       Icon_add,
  38       Icon_folder_down,
  39       Icon_page_down,
  40       Icon_remove,
  41       Icon_database_add,
  42       Icon_help
  43     }
  44     icon_id;
  45   //================================================================
  46
  47   //================================================================
  48   /*
  49   const icon_id Icon[5] = { Icon_Database,
  50                             Icon_Patient,
  51                             Icon_Study,
  52                             Icon_Series,
  53                             Icon_Image };
  54   */
  55   //================================================================
  56
  57
  58   //======================================================================
  59   // CTor
  60   WxGimmickView::WxGimmickView(Gimmick* gimmick,
  61                                wxWindow *parent,
  62                                const wxWindowID id,
  63                                const wxPoint& pos, const wxSize& size,
  64                                int min_dim,
  65                                    int max_dim,
  66                                int number_of_threads)
  67     : wxPanel(parent,id,pos,size),
  68       GimmickView(gimmick, number_of_threads),
  69       mProgressDialog(0)
  70   {
  71     GimmickDebugMessage(1,"WxGimmickView::WxGimmickView"
  72                         <<std::endl);
  73     // Sets the current directory to the home dir
  74     mCurrentDirectory =  std2wx(gimmick->GetHomeDirectory());
  75
  76      // Connect the AddProgress callback
  77     gimmick->ConnectAddProgressObserver
  78       ( boost::bind( &WxGimmickView::OnAddProgress , this, _1 ) );
  79
  80     // Create the list of icons (mIcon)
  81     CreateIconList();
  82
  83     // Global sizer
  84     wxBoxSizer  *sizer = new wxBoxSizer(wxVERTICAL);
  85
  86     // Create the tool bar
  87     CreateToolBar();
  88     sizer->Add( mToolBar ,0, wxGROW  ,0);
  89
  90     // Split part below toolbar into notebook for views and panel
  91     // for preview, messages...
  92     mSplitter = new wxSplitterWindow( this , -1);
  93
  94
  95     // Notebook
  96     mNotebook = new wxNotebook(mSplitter,
  97                                -1,wxDefaultPosition, wxDefaultSize, 0);
  98
  99         //Gimmick
 100         mGimmick=gimmick;
 101         mViewer=new WxViewer(this, wxID_ANY, wxT("Gimmick! Viewer"),wxDefaultPosition, wxDefaultSize );
 102
 103
 104         mSelectionMaxDimension= max_dim;
 105         mSelectionMinDimension= min_dim;
 106
 107     // Create the views
 108     CreateTreeViews();
 109
 110     // Bottom panel
 111     mBottomPanel = new wxPanel(mSplitter,-1);
 112         mText = new wxStaticText(mBottomPanel, wxID_ANY, wxT("Status: Welcome to Gimmick!"));
 113
 114     // Splitting
 115     int hsize = size.GetHeight();
 116     int bottom_minsize = 15;
 117
 118     mSplitter->SetMinimumPaneSize( bottom_minsize );
 119     mSplitter->SplitHorizontally( mNotebook, mBottomPanel,
 120                                    hsize - bottom_minsize);
 121
 122     sizer->Add( mSplitter,1,wxGROW  ,0);
 123
 124
 125     SetSizer( sizer );
 126     SetAutoLayout(true);
 127     Layout();
 128
 129   }
 130   //======================================================================
 131
 132   //======================================================================
 133   /// Destructor
 134   WxGimmickView::~WxGimmickView()
 135   {
 136     GimmickDebugMessage(1,"WxGimmickView::~WxGimmickView"
 137                         <<std::endl);
 138   }
 139   //======================================================================
 140
 141   //======================================================================
 142   /// Creates the tool bar
 143   void WxGimmickView::CreateToolBar()
 144   {
 145     long style = wxTB_HORIZONTAL | wxNO_BORDER | wxTB_TEXT;
 146     mToolBar = new wxToolBar(this,-1,wxDefaultPosition,wxDefaultSize,
 147                              style);
 148
 149     mToolAddFile = mToolBar->AddTool( TOOL_ADDFILES_ID,
 150                                       _T("Add file(s)"),
 151                                       mIcon->GetBitmap(Icon_page_down),
 152                                       _T("Add one or more file to database")
 153                                       );
 154     mToolAddDir = mToolBar->AddTool( TOOL_ADDDIR_ID,
 155                                       _T("Add folder"),
 156                                       mIcon->GetBitmap(Icon_folder_down),
 157                                       _T("Add the content of a folder to database")
 158                                       );
 159     mToolRemove = mToolBar->AddTool( TOOL_REMOVE_ID,
 160                                       _T("Remove"),
 161                                       mIcon->GetBitmap(Icon_remove),
 162                                       _T("Remove selected items")
 163                                       );
 164     mToolAddDatabase = mToolBar->AddTool( TOOL_ADDDATABASE_ID,
 165                                       _T("Open database"),
 166                                       mIcon->GetBitmap(Icon_database_add),
 167                                       _T("Open a local or distant database")
 168                                       );
 169     mToolHelp = mToolBar->AddTool( TOOL_HELP_ID,
 170                                       _T("Help"),
 171                                       mIcon->GetBitmap(Icon_help),
 172                                       _T("Open help window")
 173                                       );
 174     //const wxBitmap& bitmap1, const wxString& shortHelpString = "", wxItemKind kind = wxITEM_NORMAL)
 175
 176     mToolBar->Realize();
 177   }
 178   //======================================================================
 179
 180
 181   //======================================================================
 182   /// Create the tree view for TreeHandler provided
 183   void WxGimmickView::CreateTreeView( TreeHandler* h)
 184   {
 185     std::string name(h->GetTree().GetAttribute("Name"));
 186     GimmickMessage(2,"Creating the tree view for '"<<
 187                    name<<"'"<<std::endl);
 188     // Create the WxTreeView
 189     WxTreeView* view = new WxTreeView(h,mNotebook,-1);
 190
 191     // TO DO : TEST THAT A VIEW WITH SAME NAME IS NOT
 192     // ALREADY IN THE MAP
 193     GetTreeViewMap()[name] = view;
 194
 195     // Add Notebook page
 196     mNotebook->AddPage( view, crea::std2wx(name) );
 197
 198   }
 199   //======================================================================
 200   /// Returns the selected Image so that it complies with the given parameter(<4D)
 201   vtkImageData* WxGimmickView::GetSelectedImage(int dim)
 202   {
 203           return NULL;
 204                   //GetTreeViewMap()["Local database"]->GetSelectedImage(dim);
 205   }
 206
 207   //======================================================================
 208   /// Returns the selected Images so that they comply with the given parameter(4D)
 209   void WxGimmickView::GetSelectedImages(std::vector<vtkImageData*>& s, int dim)
 210   {
 211         int level=GetTreeViewMap()["Local database"]->GetNumberOfLevels();
 212         std::vector<tree::Node*> im=GetTreeViewMap()["Local database"]->GetSelected(level+1);
 213         ReadImagesNotThreaded(s,im,dim);
 214   }
 215   //======================================================================
 216   /// Returns the selected Images so that they comply with the given parameter(4D)
 217   void WxGimmickView::GetSelectedFiles(std::vector<std::string>& s)
 218   {
 219         GetTreeViewMap()["Local database"]->GetSelectedAsString(s);
 220   }
 221   //=================================================
 222   void WxGimmickView::CreateIconList()
 223   {
 224     // Size of the icons;
 225     int size = 24;
 226
 227     wxIcon icons[20];
 228     // should correspond to Icon_xxx enum
 229     icons[Icon_accept] = wxIcon(accept_xpm);
 230     icons[Icon_add] = wxIcon(add_xpm);
 231     icons[Icon_folder_down] = wxIcon(folder_down_xpm);
 232     icons[Icon_page_down] = wxIcon(page_down_xpm);
 233     icons[Icon_remove] = wxIcon(remove_xpm);
 234     icons[Icon_database_add] = wxIcon(database_add_xpm);
 235     icons[Icon_help] = wxIcon(help_xpm);
 236
 237     //   unsigned int NbIcons = 8;
 238     // Make an image list containing small icons
 239     mIcon = new wxImageList(size,size,true);
 240
 241     // Make all icons the same size = size of the first one
 242     int sizeOrig = icons[0].GetWidth();
 243     for ( size_t i = 0; i < icon_number; i++ )
 244       {
 245         if ( size == sizeOrig )
 246           {
 247             mIcon->Add(icons[i]);
 248           }
 249         else
 250           {
 251             mIcon->Add(wxBitmap(wxBitmap(icons[i]).ConvertToImage().Rescale(size, size)));
 252           }
 253       }
 254   }
 255   //=================================================
 256
 257
 258   //=================================================
 259   void WxGimmickView::OnAddFiles(wxCommandEvent& event)
 260   {
 261    long style = wxOPEN | wxFILE_MUST_EXIST | wxFD_MULTIPLE;
 262     std::string wc("*.*");
 263     wxFileDialog* FD = new wxFileDialog( 0,
 264                                          _T("Select file"),
 265                                          _T(""),
 266                                          _T(""),
 267                                          crea::std2wx(wc),
 268                                          style,
 269                                          wxDefaultPosition);
 270
 271     if (FD->ShowModal()==wxID_OK)
 272       {
 273         wxBusyCursor busy;
 274
 275         wxArrayString files;
 276         FD->GetPaths(files);
 277         unsigned int i;
 278         std::vector<std::string> filenames;
 279         for (i=0;i<files.GetCount();++i)
 280         {
 281           filenames.push_back(wx2std(files[i]));
 282           GimmickMessage(2,"Adding File "<<files[i]<<"."<<std::endl);
 283         }
 284
 285         mProgressDialog =
 286           new wxProgressDialog(_T("Adding file(s)"),
 287                                _T(""),
 288                                1000,
 289                                this,
 290                                wxPD_ELAPSED_TIME |
 291                                //                              wxPD_ESTIMATED_TIME |
 292                                //                              wxPD_REMAINING_TIME |
 293                                wxPD_CAN_ABORT );
 294
 295         // TO DO : select the current tree handler
 296         mGimmick->AddFiles("Local database",filenames);
 297
 298         mProgressDialog->Pulse(_T("Updating view..."));
 299
 300         UpdateTreeViewLevel("Local database",1);
 301         delete mProgressDialog;
 302         DisplayAddSummary();
 303       }
 304
 305   }
 306   //=================================================
 307
 308   //=================================================
 309   void WxGimmickView::OnAddDir(wxCommandEvent& event)
 310   {
 311     long style = wxDD_DEFAULT_STYLE | wxDD_DIR_MUST_EXIST;
 312     wxDirDialog* FD =
 313       new wxDirDialog( 0,
 314                        _T("Select directory"),
 315                        mCurrentDirectory,
 316                        style);
 317
 318     if (FD->ShowModal()==wxID_OK)
 319       {
 320
 321         bool recurse = false;
 322         if (wxMessageBox(_T("Recurse into sub-directories ?"),
 323                          _T("Scan directory"),
 324                          wxYES_NO,this ) == wxYES)
 325           {
 326             recurse = true;
 327           }
 328
 329         wxBusyCursor busy;
 330         wxString title(_T("Adding directory"));
 331         if (recurse)
 332           title = _T("Adding directory (recursive)");
 333         mProgressDialog =
 334           new wxProgressDialog(_T("Adding directory"),
 335                                _T(""),
 336                                1000,
 337                                this,
 338                                wxPD_ELAPSED_TIME |
 339                                //                              wxPD_ESTIMATED_TIME |
 340                                //                              wxPD_REMAINING_TIME |
 341                                wxPD_CAN_ABORT );
 342         std::string dirname = wx2std (FD->GetPath()) ;
 343         mCurrentDirectory = FD->GetPath();
 344
 345         // TO DO : select the current tree handler
 346         mGimmick->AddDir("Local database",dirname,recurse);
 347
 348         mProgressDialog->Pulse(_T("Updating view..."));
 349
 350         UpdateTreeViewLevel("Local database",1);
 351         delete mProgressDialog;
 352         DisplayAddSummary();
 353       }
 354   }
 355   //=================================================
 356
 357   //=================================================
 358   bool WxGimmickView::ValidateSelection(void * s)
 359   {
 360     wxBusyCursor busy;
 361         std::vector<tree::Node*> sel=* (std::vector<tree::Node*> *) s;
 362         bool valid=ValidateSelected(sel,mSelectionMinDimension,mSelectionMaxDimension );
 363         mText->SetLabel(_T("Status: ")+GetMessage());
 364         if(valid)
 365         {
 366                 ReadImageThreaded(sel);
 367         }
 368         else
 369         {
 370                 mViewer->Hide();
 371         }
 372         return valid;
 373
 374    }
 375
 376   //==================================================
 377
 378   //==================================================
 379   ///Reads Images (Threaded)
 380   void WxGimmickView::ReadImageThreaded(std::vector<tree::Node*> sel)
 381   {
 382         int maxprio = GetMaximalPriority();
 383         int prio = maxprio + 2000;
 384
 385         //First load the selected images
 386         mCurImageItemToShow = sel.front();
 387         std::vector<tree::Node*>::iterator selected;
 388         for(selected=sel.begin();selected!=sel.end();++selected)
 389         {
 390                 GimmickDebugMessage(5,
 391                                 "Requesting image from selected "
 392                                 <<(*selected)->GetAttribute("FullFileName")
 393                                 <<std::endl);
 394                 RequestReading(*selected,prio);
 395                 AddEntryToMap(*selected);
 396                 prio--;
 397         }
 398
 399         //Going up
 400         prio = maxprio + 1000;
 401         std::vector<tree::Node*> up;
 402         GetTreeViewMap()["Local database"]->GetNodes(up,true);
 403         std::vector<tree::Node*>::iterator iterUp;
 404         for(iterUp=up.begin();iterUp!=up.end();++iterUp)
 405         {
 406                 GimmickDebugMessage(5,
 407                                 "Requesting image from neighbors up "
 408                                 <<(*iterUp)->GetAttribute("FullFileName")
 409                                 <<std::endl);
 410                 RequestReading(*iterUp,prio);
 411                 AddEntryToMap(*iterUp);
 412                 prio--;
 413         }
 414
 415         //Going down
 416         prio = maxprio + 999;
 417         std::vector<tree::Node*> down;
 418         GetTreeViewMap()["Local database"]->GetNodes(down,false);
 419         std::vector<tree::Node*>::iterator iterDown;
 420         for(iterDown=down.begin();iterDown!=down.end();++iterDown)
 421         {
 422                 GimmickDebugMessage(5,
 423                                 "Requesting image from neighbors down "
 424                                 <<(*iterDown)->GetAttribute("FullFileName")
 425                                 <<std::endl);
 426                 RequestReading(*iterDown,prio);
 427                 AddEntryToMap(*iterDown);
 428                 prio--;
 429         }
 430   }
 431
 432   //==================================================
 433
 434   //==================================================
 435   /// Processes the queue of image events
 436   void WxGimmickView::ProcessImageEvents()
 437   {
 438         int level=GetTreeViewMap()["Local database"]->GetNumberOfLevels();
 439         std::vector<tree::Node*> sel=GetTreeViewMap()["Local database"]->GetSelected(level+1);
 440           GimmickDebugMessage(5,
 441                                 "Processing Images. Lock Started"
 442                                 <<std::endl);
 443     MultiThreadImageReaderEventLock();
 444
 445         mViewer->ClearImages();
 446     while (!IsQueueEmpty())
 447       {
 448                 vtkImageData* image=GetNextImageQueued();
 449                 tree::Node* node=GetNextNodeQueued();
 450                 if( image!=0 )
 451                 {
 452                         bool found=false;
 453                         std::vector<tree::Node*>::iterator i;
 454                         for(i=sel.begin();i!=sel.end()&&!found;++i)
 455                         {
 456                                 if((*i)==node)
 457                                 {
 458                                         mViewer->AddImage(image);
 459                                         found=true;
 460                                 }
 461                         }
 462                         UnqueueNext();
 463                 }
 464           }
 465           if(!(mViewer->ImagesEmpty()))
 466           {
 467                 mViewer->ShowImages();
 468                 mViewer->Show();
 469           }
 470     ClearQueue();
 471
 472
 473     MultiThreadImageReaderEventUnlock();
 474         GimmickDebugMessage(5,
 475                                 "Processing Images. Lock Ended"
 476                                 <<std::endl);
 477   }
 478
 479   //==================================================
 480
 481   //==================================================
 482    void  WxGimmickView::OnInternalIdle()
 483   {
 484     ProcessImageEvents();
 485   }
 486
 487   //=================================================
 488
 489   //=================================================
 490   void WxGimmickView::OnRemove(wxCommandEvent& event)
 491   {
 492         //TODO Select current tree handler
 493     wxBusyCursor busy;
 494     GetTreeViewMap()["Local database"]->RemoveSelected(1);
 495   }
 496   //=================================================
 497
 498   //=================================================
 499   /// AddProgress Gimmick callback
 500   void WxGimmickView::OnAddProgress( Gimmick::AddProgress& p)
 501   {
 502
 503     char mess[200];
 504     sprintf(mess,"%i dirs - %i files - %i handled - %i added",
 505            p.GetNumberScannedDirs(),
 506            p.GetNumberScannedFiles(),
 507            p.GetNumberHandledFiles(),
 508            p.GetNumberAddedFiles());
 509     //    std::cout << "OnAddProgress "<<mess<<std::endl;
 510     wxString s(wxString::From8BitData(mess));
 511     //  std::cout << "Pulse"<<std::endl;
 512     if (!mProgressDialog->Pulse(s))
 513       {
 514         p.SetStop();
 515       }
 516     //  std::cout << "OnAddProgress ok"<<std::endl;
 517   }
 518   //=================================================
 519
 520   //=================================================
 521   void WxGimmickView::DisplayAddSummary()
 522   {
 523     const Gimmick::AddProgress& p = mGimmick->GetAddProgress();
 524     std::stringstream mess;
 525     mess << "Dirs \tscanned\t: " << p.GetNumberScannedDirs()  << "\n";
 526     mess << "Files\tscanned\t: " << p.GetNumberScannedFiles() << "\n";
 527     mess << "Files\thandled\t: " << p.GetNumberHandledFiles() << "\n\n";
 528     mess << "Files\tadded  \t: " << p.GetNumberAddedFiles()   << "\n\n";
 529
 530     /*    char times[500];
 531     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",
 532             summary.parse_time,
 533             (int)( summary.parse_time*100./summary.total_time),
 534             summary.file_scan_time,
 535             (int)(summary.file_scan_time*100./summary.total_time),
 536             summary.update_structs_time,
 537             (int)(summary.update_structs_time*100./summary.total_time),
 538             summary.update_database_time,
 539             (int)(summary.update_database_time*100./summary.total_time),
 540             summary.total_time );
 541
 542     mess << times;
 543     */
 544     wxMessageBox(std2wx(mess.str()),_T("Addition result"),wxOK,this);
 545   }
 546
 547    //=================================================
 548
 549    //=================================================
 550   BEGIN_EVENT_TABLE(WxGimmickView, wxPanel)
 551     EVT_TOOL(TOOL_ADDFILES_ID, WxGimmickView::OnAddFiles)
 552     EVT_TOOL(TOOL_ADDDIR_ID, WxGimmickView::OnAddDir)
 553         EVT_TOOL(TOOL_REMOVE_ID, WxGimmickView::OnRemove)
 554   END_EVENT_TABLE()
 555   //=================================================
 556
 557 } // EO namespace creaImageIO
 558
 559