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