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
 102         mSelectionMaxDimension= max_dim;
 103         mSelectionMinDimension= min_dim;
 104
 105     // Create the views
 106     CreateTreeViews();
 107
 108     // Bottom panel
 109     mBottomPanel = new wxPanel(mSplitter,-1);
 110         mText = new wxStaticText(mBottomPanel, wxID_ANY, wxT("Status: Welcome to Gimmick!"));
 111
 112     // Splitting
 113     int hsize = size.GetHeight();
 114     int bottom_minsize = 15;
 115
 116     mSplitter->SetMinimumPaneSize( bottom_minsize );
 117     mSplitter->SplitHorizontally( mNotebook, mBottomPanel,
 118                                    hsize - bottom_minsize);
 119
 120     sizer->Add( mSplitter,1,wxGROW  ,0);
 121
 122
 123     SetSizer( sizer );
 124     SetAutoLayout(true);
 125     Layout();
 126
 127   }
 128   //======================================================================
 129
 130   //======================================================================
 131   /// Destructor
 132   WxGimmickView::~WxGimmickView()
 133   {
 134     GimmickDebugMessage(1,"WxGimmickView::~WxGimmickView"
 135                         <<std::endl);
 136   }
 137   //======================================================================
 138
 139   //======================================================================
 140   /// Creates the tool bar
 141   void WxGimmickView::CreateToolBar()
 142   {
 143     long style = wxTB_HORIZONTAL | wxNO_BORDER | wxTB_TEXT;
 144     mToolBar = new wxToolBar(this,-1,wxDefaultPosition,wxDefaultSize,
 145                              style);
 146
 147     mToolAddFile = mToolBar->AddTool( TOOL_ADDFILES_ID,
 148                                       _T("Add file(s)"),
 149                                       mIcon->GetBitmap(Icon_page_down),
 150                                       _T("Add one or more file to database")
 151                                       );
 152     mToolAddDir = mToolBar->AddTool( TOOL_ADDDIR_ID,
 153                                       _T("Add folder"),
 154                                       mIcon->GetBitmap(Icon_folder_down),
 155                                       _T("Add the content of a folder to database")
 156                                       );
 157     mToolRemove = mToolBar->AddTool( TOOL_REMOVE_ID,
 158                                       _T("Remove"),
 159                                       mIcon->GetBitmap(Icon_remove),
 160                                       _T("Remove selected items")
 161                                       );
 162     mToolAddDatabase = mToolBar->AddTool( TOOL_ADDDATABASE_ID,
 163                                       _T("Open database"),
 164                                       mIcon->GetBitmap(Icon_database_add),
 165                                       _T("Open a local or distant database")
 166                                       );
 167     mToolHelp = mToolBar->AddTool( TOOL_HELP_ID,
 168                                       _T("Help"),
 169                                       mIcon->GetBitmap(Icon_help),
 170                                       _T("Open help window")
 171                                       );
 172     //const wxBitmap& bitmap1, const wxString& shortHelpString = "", wxItemKind kind = wxITEM_NORMAL)
 173
 174     mToolBar->Realize();
 175   }
 176   //======================================================================
 177
 178
 179   //======================================================================
 180   /// Create the tree view for TreeHandler provided
 181   void WxGimmickView::CreateTreeView( TreeHandler* h)
 182   {
 183     std::string name(h->GetTree().GetAttribute("Name"));
 184     GimmickMessage(2,"Creating the tree view for '"<<
 185                    name<<"'"<<std::endl);
 186     // Create the WxTreeView
 187     WxTreeView* view = new WxTreeView(h,mNotebook,-1);
 188
 189     // TO DO : TEST THAT A VIEW WITH SAME NAME IS NOT
 190     // ALREADY IN THE MAP
 191     GetTreeViewMap()[name] = view;
 192
 193     // Add Notebook page
 194     mNotebook->AddPage( view, crea::std2wx(name) );
 195
 196   }
 197   //======================================================================
 198   /// Returns the selected Image so that it complies with the given parameter(<4D)
 199   vtkImageData* WxGimmickView::GetSelectedImage(int dim)
 200   {
 201           return NULL;
 202                   //GetTreeViewMap()["Local database"]->GetSelectedImage(dim);
 203   }
 204
 205   //======================================================================
 206   /// Returns the selected Images so that they comply with the given parameter(4D)
 207   void WxGimmickView::GetSelectedImages(std::vector<vtkImageData*>& s, int dim)
 208   {
 209         int level=GetTreeViewMap()["Local database"]->GetNumberOfLevels();
 210         std::vector<tree::Node*> im=GetTreeViewMap()["Local database"]->GetSelected(level+1);
 211         ReadImagesNotThreaded(s,im,dim);
 212   }
 213   //======================================================================
 214   /// Returns the selected Images so that they comply with the given parameter(4D)
 215   void WxGimmickView::GetSelectedFiles(std::vector<std::string>& s)
 216   {
 217         GetTreeViewMap()["Local database"]->GetSelectedAsString(s);
 218   }
 219   //=================================================
 220   void WxGimmickView::CreateIconList()
 221   {
 222     // Size of the icons;
 223     int size = 24;
 224
 225     wxIcon icons[20];
 226     // should correspond to Icon_xxx enum
 227     icons[Icon_accept] = wxIcon(accept_xpm);
 228     icons[Icon_add] = wxIcon(add_xpm);
 229     icons[Icon_folder_down] = wxIcon(folder_down_xpm);
 230     icons[Icon_page_down] = wxIcon(page_down_xpm);
 231     icons[Icon_remove] = wxIcon(remove_xpm);
 232     icons[Icon_database_add] = wxIcon(database_add_xpm);
 233     icons[Icon_help] = wxIcon(help_xpm);
 234
 235     //   unsigned int NbIcons = 8;
 236     // Make an image list containing small icons
 237     mIcon = new wxImageList(size,size,true);
 238
 239     // Make all icons the same size = size of the first one
 240     int sizeOrig = icons[0].GetWidth();
 241     for ( size_t i = 0; i < icon_number; i++ )
 242       {
 243         if ( size == sizeOrig )
 244           {
 245             mIcon->Add(icons[i]);
 246           }
 247         else
 248           {
 249             mIcon->Add(wxBitmap(wxBitmap(icons[i]).ConvertToImage().Rescale(size, size)));
 250           }
 251       }
 252   }
 253   //=================================================
 254
 255
 256   //=================================================
 257   void WxGimmickView::OnAddFiles(wxCommandEvent& event)
 258   {
 259    long style = wxOPEN | wxFILE_MUST_EXIST | wxFD_MULTIPLE;
 260     std::string wc("*.*");
 261     wxFileDialog* FD = new wxFileDialog( 0,
 262                                          _T("Select file"),
 263                                          _T(""),
 264                                          _T(""),
 265                                          crea::std2wx(wc),
 266                                          style,
 267                                          wxDefaultPosition);
 268
 269     if (FD->ShowModal()==wxID_OK)
 270       {
 271         wxBusyCursor busy;
 272
 273         wxArrayString files;
 274         FD->GetPaths(files);
 275         unsigned int i;
 276         std::vector<std::string> filenames;
 277         for (i=0;i<files.GetCount();++i)
 278         {
 279           filenames.push_back(wx2std(files[i]));
 280           GimmickMessage(2,"Adding File "<<files[i]<<"."<<std::endl);
 281         }
 282
 283         mProgressDialog =
 284           new wxProgressDialog(_T("Adding file(s)"),
 285                                _T(""),
 286                                1000,
 287                                this,
 288                                wxPD_ELAPSED_TIME |
 289                                //                              wxPD_ESTIMATED_TIME |
 290                                //                              wxPD_REMAINING_TIME |
 291                                wxPD_CAN_ABORT );
 292
 293         // TO DO : select the current tree handler
 294         mGimmick->AddFiles("Local database",filenames);
 295
 296         mProgressDialog->Pulse(_T("Updating view..."));
 297
 298         UpdateTreeViewLevel("Local database",1);
 299         delete mProgressDialog;
 300         DisplayAddSummary();
 301       }
 302
 303   }
 304   //=================================================
 305
 306   //=================================================
 307   void WxGimmickView::OnAddDir(wxCommandEvent& event)
 308   {
 309     long style = wxDD_DEFAULT_STYLE | wxDD_DIR_MUST_EXIST;
 310     wxDirDialog* FD =
 311       new wxDirDialog( 0,
 312                        _T("Select directory"),
 313                        mCurrentDirectory,
 314                        style);
 315
 316     if (FD->ShowModal()==wxID_OK)
 317       {
 318
 319         bool recurse = false;
 320         if (wxMessageBox(_T("Recurse into sub-directories ?"),
 321                          _T("Scan directory"),
 322                          wxYES_NO,this ) == wxYES)
 323           {
 324             recurse = true;
 325           }
 326
 327         wxBusyCursor busy;
 328         wxString title(_T("Adding directory"));
 329         if (recurse)
 330           title = _T("Adding directory (recursive)");
 331         mProgressDialog =
 332           new wxProgressDialog(_T("Adding directory"),
 333                                _T(""),
 334                                1000,
 335                                this,
 336                                wxPD_ELAPSED_TIME |
 337                                //                              wxPD_ESTIMATED_TIME |
 338                                //                              wxPD_REMAINING_TIME |
 339                                wxPD_CAN_ABORT );
 340         std::string dirname = wx2std (FD->GetPath()) ;
 341         mCurrentDirectory = FD->GetPath();
 342
 343         // TO DO : select the current tree handler
 344         mGimmick->AddDir("Local database",dirname,recurse);
 345
 346         mProgressDialog->Pulse(_T("Updating view..."));
 347
 348         UpdateTreeViewLevel("Local database",1);
 349         delete mProgressDialog;
 350         DisplayAddSummary();
 351       }
 352   }
 353   //=================================================
 354
 355   //=================================================
 356   bool WxGimmickView::ValidateSelection(void * s)
 357   {
 358     wxBusyCursor busy;
 359         std::vector<tree::Node*> sel=* (std::vector<tree::Node*> *) s;
 360         bool valid=ValidateSelected(sel,mSelectionMinDimension,mSelectionMaxDimension );
 361         mText->SetLabel(_T("Status: ")+GetMessage());
 362         return valid;
 363
 364    }
 365   //=================================================
 366
 367    //=================================================
 368   void WxGimmickView::OnRemove(wxCommandEvent& event)
 369   {
 370         //TODO Select current tree handler
 371     wxBusyCursor busy;
 372     GetTreeViewMap()["Local database"]->RemoveSelected(1);
 373   }
 374   //=================================================
 375
 376   //=================================================
 377   /// AddProgress Gimmick callback
 378   void WxGimmickView::OnAddProgress( Gimmick::AddProgress& p)
 379   {
 380
 381     char mess[200];
 382     sprintf(mess,"%i dirs - %i files - %i handled - %i added",
 383            p.GetNumberScannedDirs(),
 384            p.GetNumberScannedFiles(),
 385            p.GetNumberHandledFiles(),
 386            p.GetNumberAddedFiles());
 387     //    std::cout << "OnAddProgress "<<mess<<std::endl;
 388     wxString s(wxString::From8BitData(mess));
 389     //  std::cout << "Pulse"<<std::endl;
 390     if (!mProgressDialog->Pulse(s))
 391       {
 392         p.SetStop();
 393       }
 394     //  std::cout << "OnAddProgress ok"<<std::endl;
 395   }
 396   //=================================================
 397
 398   //=================================================
 399   void WxGimmickView::DisplayAddSummary()
 400   {
 401     const Gimmick::AddProgress& p = mGimmick->GetAddProgress();
 402     std::stringstream mess;
 403     mess << "Dirs \tscanned\t: " << p.GetNumberScannedDirs()  << "\n";
 404     mess << "Files\tscanned\t: " << p.GetNumberScannedFiles() << "\n";
 405     mess << "Files\thandled\t: " << p.GetNumberHandledFiles() << "\n\n";
 406     mess << "Files\tadded  \t: " << p.GetNumberAddedFiles()   << "\n\n";
 407
 408     /*    char times[500];
 409     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",
 410             summary.parse_time,
 411             (int)( summary.parse_time*100./summary.total_time),
 412             summary.file_scan_time,
 413             (int)(summary.file_scan_time*100./summary.total_time),
 414             summary.update_structs_time,
 415             (int)(summary.update_structs_time*100./summary.total_time),
 416             summary.update_database_time,
 417             (int)(summary.update_database_time*100./summary.total_time),
 418             summary.total_time );
 419
 420     mess << times;
 421     */
 422     wxMessageBox(std2wx(mess.str()),_T("Addition result"),wxOK,this);
 423   }
 424
 425    //=================================================
 426
 427    //=================================================
 428   BEGIN_EVENT_TABLE(WxGimmickView, wxPanel)
 429     EVT_TOOL(TOOL_ADDFILES_ID, WxGimmickView::OnAddFiles)
 430     EVT_TOOL(TOOL_ADDDIR_ID, WxGimmickView::OnAddDir)
 431         EVT_TOOL(TOOL_REMOVE_ID, WxGimmickView::OnRemove)
 432   END_EVENT_TABLE()
 433   //=================================================
 434
 435 } // EO namespace creaImageIO
 436
 437