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),
  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 = 20;
 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         view->SetMaxDimension(mSelectionMaxDimension);
 190         view->SetMinDimension(mSelectionMinDimension);
 191
 192     // TO DO : TEST THAT A VIEW WITH SAME NAME IS NOT
 193     // ALREADY IN THE MAP
 194     GetTreeViewMap()[name] = view;
 195
 196     // Add Notebook page
 197     mNotebook->AddPage( view, crea::std2wx(name) );
 198
 199   }
 200   //======================================================================
 201   /// Returns the selected Image so that it complies with the given parameter(<4D)
 202   vtkImageData* WxGimmickView::GetSelectedImage(int dim)
 203   {
 204           return 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           GetTreeViewMap()["Local database"]->GetSelectedImages(s,dim);
 212   }
 213
 214   //=================================================
 215   void WxGimmickView::CreateIconList()
 216   {
 217     // Size of the icons;
 218     int size = 24;
 219
 220     wxIcon icons[20];
 221     // should correspond to Icon_xxx enum
 222     icons[Icon_accept] = wxIcon(accept_xpm);
 223     icons[Icon_add] = wxIcon(add_xpm);
 224     icons[Icon_folder_down] = wxIcon(folder_down_xpm);
 225     icons[Icon_page_down] = wxIcon(page_down_xpm);
 226     icons[Icon_remove] = wxIcon(remove_xpm);
 227     icons[Icon_database_add] = wxIcon(database_add_xpm);
 228     icons[Icon_help] = wxIcon(help_xpm);
 229
 230     //   unsigned int NbIcons = 8;
 231     // Make an image list containing small icons
 232     mIcon = new wxImageList(size,size,true);
 233
 234     // Make all icons the same size = size of the first one
 235     int sizeOrig = icons[0].GetWidth();
 236     for ( size_t i = 0; i < icon_number; i++ )
 237       {
 238         if ( size == sizeOrig )
 239           {
 240             mIcon->Add(icons[i]);
 241           }
 242         else
 243           {
 244             mIcon->Add(wxBitmap(wxBitmap(icons[i]).ConvertToImage().Rescale(size, size)));
 245           }
 246       }
 247   }
 248   //=================================================
 249
 250
 251   //=================================================
 252   void WxGimmickView::OnAddFiles(wxCommandEvent& event)
 253   {
 254    long style = wxOPEN | wxFILE_MUST_EXIST | wxFD_MULTIPLE;
 255     std::string wc("*.*");
 256     wxFileDialog* FD = new wxFileDialog( 0,
 257                                          _T("Select file"),
 258                                          _T(""),
 259                                          _T(""),
 260                                          crea::std2wx(wc),
 261                                          style,
 262                                          wxDefaultPosition);
 263
 264     if (FD->ShowModal()==wxID_OK)
 265       {
 266         wxBusyCursor busy;
 267
 268         wxArrayString files;
 269         FD->GetPaths(files);
 270         unsigned int i;
 271         std::vector<std::string> filenames;
 272         for (i=0;i<files.GetCount();++i)
 273         {
 274           filenames.push_back(wx2std(files[i]));
 275           GimmickMessage(2,"Adding File "<<files[i]<<"."<<std::endl);
 276         }
 277
 278         mProgressDialog =
 279           new wxProgressDialog(_T("Adding file(s)"),
 280                                _T(""),
 281                                1000,
 282                                this,
 283                                wxPD_ELAPSED_TIME |
 284                                //                              wxPD_ESTIMATED_TIME |
 285                                //                              wxPD_REMAINING_TIME |
 286                                wxPD_CAN_ABORT );
 287
 288         // TO DO : select the current tree handler
 289         mGimmick->AddFiles("Local database",filenames);
 290
 291         mProgressDialog->Pulse(_T("Updating view..."));
 292
 293         UpdateTreeViewLevel("Local database",1);
 294         delete mProgressDialog;
 295         DisplayAddSummary();
 296       }
 297
 298   }
 299   //=================================================
 300
 301   //=================================================
 302   void WxGimmickView::OnAddDir(wxCommandEvent& event)
 303   {
 304     long style = wxDD_DEFAULT_STYLE | wxDD_DIR_MUST_EXIST;
 305     wxDirDialog* FD =
 306       new wxDirDialog( 0,
 307                        _T("Select directory"),
 308                        mCurrentDirectory,
 309                        style);
 310
 311     if (FD->ShowModal()==wxID_OK)
 312       {
 313
 314         bool recurse = false;
 315         if (wxMessageBox(_T("Recurse into sub-directories ?"),
 316                          _T("Scan directory"),
 317                          wxYES_NO,this ) == wxYES)
 318           {
 319             recurse = true;
 320           }
 321
 322         wxBusyCursor busy;
 323         wxString title(_T("Adding directory"));
 324         if (recurse)
 325           title = _T("Adding directory (recursive)");
 326         mProgressDialog =
 327           new wxProgressDialog(_T("Adding directory"),
 328                                _T(""),
 329                                1000,
 330                                this,
 331                                wxPD_ELAPSED_TIME |
 332                                //                              wxPD_ESTIMATED_TIME |
 333                                //                              wxPD_REMAINING_TIME |
 334                                wxPD_CAN_ABORT );
 335         std::string dirname = wx2std (FD->GetPath()) ;
 336         mCurrentDirectory = FD->GetPath();
 337
 338         // TO DO : select the current tree handler
 339         mGimmick->AddDir("Local database",dirname,recurse);
 340
 341         mProgressDialog->Pulse(_T("Updating view..."));
 342
 343         UpdateTreeViewLevel("Local database",1);
 344         delete mProgressDialog;
 345         DisplayAddSummary();
 346       }
 347   }
 348   //=================================================
 349
 350   //=================================================
 351   void WxGimmickView::SetMessage(const wxString& mess)
 352   {
 353     wxBusyCursor busy;
 354
 355     mText->SetLabel(_T("Status: ")+mess);
 356    }
 357   //=================================================
 358
 359    //=================================================
 360   void WxGimmickView::OnRemove(wxCommandEvent& event)
 361   {
 362         //TODO Select current tree handler
 363     wxBusyCursor busy;
 364     GetTreeViewMap()["Local database"]->RemoveSelected(1);
 365   }
 366   //=================================================
 367
 368   //=================================================
 369   /// AddProgress Gimmick callback
 370   void WxGimmickView::OnAddProgress( Gimmick::AddProgress& p)
 371   {
 372
 373     char mess[200];
 374     sprintf(mess,"%i dirs - %i files - %i handled - %i added",
 375            p.GetNumberScannedDirs(),
 376            p.GetNumberScannedFiles(),
 377            p.GetNumberHandledFiles(),
 378            p.GetNumberAddedFiles());
 379     //    std::cout << "OnAddProgress "<<mess<<std::endl;
 380     wxString s(wxString::From8BitData(mess));
 381     //  std::cout << "Pulse"<<std::endl;
 382     if (!mProgressDialog->Pulse(s))
 383       {
 384         p.SetStop();
 385       }
 386     //  std::cout << "OnAddProgress ok"<<std::endl;
 387   }
 388   //=================================================
 389
 390   //=================================================
 391   void WxGimmickView::DisplayAddSummary()
 392   {
 393     const Gimmick::AddProgress& p = mGimmick->GetAddProgress();
 394     std::stringstream mess;
 395     mess << "Dirs \tscanned\t: " << p.GetNumberScannedDirs()  << "\n";
 396     mess << "Files\tscanned\t: " << p.GetNumberScannedFiles() << "\n";
 397     mess << "Files\thandled\t: " << p.GetNumberHandledFiles() << "\n\n";
 398     mess << "Files\tadded  \t: " << p.GetNumberAddedFiles()   << "\n\n";
 399
 400     /*    char times[500];
 401     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",
 402             summary.parse_time,
 403             (int)( summary.parse_time*100./summary.total_time),
 404             summary.file_scan_time,
 405             (int)(summary.file_scan_time*100./summary.total_time),
 406             summary.update_structs_time,
 407             (int)(summary.update_structs_time*100./summary.total_time),
 408             summary.update_database_time,
 409             (int)(summary.update_database_time*100./summary.total_time),
 410             summary.total_time );
 411
 412     mess << times;
 413     */
 414     wxMessageBox(std2wx(mess.str()),_T("Addition result"),wxOK,this);
 415   }
 416    //=================================================
 417
 418    //=================================================
 419   BEGIN_EVENT_TABLE(WxGimmickView, wxPanel)
 420     EVT_TOOL(TOOL_ADDFILES_ID, WxGimmickView::OnAddFiles)
 421     EVT_TOOL(TOOL_ADDDIR_ID, WxGimmickView::OnAddDir)
 422         EVT_TOOL(TOOL_REMOVE_ID, WxGimmickView::OnRemove)
 423   END_EVENT_TABLE()
 424   //=================================================
 425
 426 } // EO namespace creaImageIO
 427
 428