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