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