src/TransferWnd.cpp

   1 //
   2 // This file is part of the aMule Project.
   3 //
   4 // Copyright (c) 2003-2008 aMule Team ( admin@amule.org / http://www.amule.org )
   5 // Copyright (c) 2002 Merkur ( devs@emule-project.net / http://www.emule-project.net )
   6 //
   7 // Any parts of this program derived from the xMule, lMule or eMule project,
   8 // or contributed by third-party developers are copyrighted by their
   9 // respective authors.
  10 //
  11 // This program is free software; you can redistribute it and/or modify
  12 // it under the terms of the GNU General Public License as published by
  13 // the Free Software Foundation; either version 2 of the License, or
  14 // (at your option) any later version.
  15 //
  16 // This program is distributed in the hope that it will be useful,
  17 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 // GNU General Public License for more details.
  20 //
  21 // You should have received a copy of the GNU General Public License
  22 // along with this program; if not, write to the Free Software
  23 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301, USA
  24 //
  25
  26 #include "TransferWnd.h"        // Interface declarations
  27
  28 #include <common/MenuIDs.h>
  29
  30 #include <wx/config.h>
  31
  32
  33 // This include must be before amuleDlg.h
  34 #ifdef __WXMSW__
  35     #include <wx/msw/winundef.h>        // Needed for windows compilation
  36 #endif
  37
  38
  39 #include "amuleDlg.h"                   // Needed for CamuleDlg
  40 #include "PartFile.h"                   // Needed for PR_LOW
  41 #include "DownloadQueue.h"              // Needed for CDownloadQueue
  42 #include "CatDialog.h"                  // Needed for CCatDialog
  43 #include "DownloadListCtrl.h"           // Needed for CDownloadListCtrl
  44 #include "SourceListCtrl.h"             // Needed for CSourceListCtrl
  45 #include "amule.h"                      // Needed for theApp
  46 #include "muuli_wdr.h"                  // Needed for ID_CATEGORIES
  47 #include "SearchDlg.h"                  // Needed for CSearchDlg->UpdateCatChoice()
  48 #include "MuleNotebook.h"
  49 #include "Preferences.h"
  50 #include "Statistics.h"                 // Needed for theStats
  51 #include "SharedFileList.h"             // Needed for CSharedFileList
  52 #include "GuiEvents.h"                  // Needed for CoreNotify_*
  53
  54
  55 BEGIN_EVENT_TABLE(CTransferWnd, wxPanel)
  56         EVT_RIGHT_DOWN(CTransferWnd::OnNMRclickDLtab)
  57         EVT_NOTEBOOK_PAGE_CHANGED(ID_CATEGORIES,        CTransferWnd::OnCategoryChanged)
  58
  59         EVT_SPLITTER_SASH_POS_CHANGING(ID_DOWNLOADSSPLATTER, CTransferWnd::OnSashPositionChanging)
  60
  61         EVT_BUTTON(ID_BTNCLRCOMPL,              CTransferWnd::OnBtnClearDownloads)
  62         EVT_BUTTON(ID_CLIENTTOGGLE,             CTransferWnd::OnToggleClientList)
  63
  64         EVT_MENU_RANGE(MP_CAT_SET0, MP_CAT_SET0 + 15, CTransferWnd::OnSetDefaultCat)
  65         EVT_MENU(MP_CAT_ADD,                    CTransferWnd::OnAddCategory)
  66         EVT_MENU(MP_CAT_EDIT,                   CTransferWnd::OnEditCategory)
  67         EVT_MENU(MP_CAT_REMOVE,                 CTransferWnd::OnDelCategory)
  68         EVT_MENU(MP_PRIOLOW,                    CTransferWnd::OnSetCatPriority)
  69         EVT_MENU(MP_PRIONORMAL,                 CTransferWnd::OnSetCatPriority)
  70         EVT_MENU(MP_PRIOHIGH,                   CTransferWnd::OnSetCatPriority)
  71         EVT_MENU(MP_PRIOAUTO,                   CTransferWnd::OnSetCatPriority)
  72         EVT_MENU(MP_PAUSE,                      CTransferWnd::OnSetCatStatus)
  73         EVT_MENU(MP_STOP,                       CTransferWnd::OnSetCatStatus)
  74         EVT_MENU(MP_CANCEL,                     CTransferWnd::OnSetCatStatus)
  75         EVT_MENU(MP_RESUME,                     CTransferWnd::OnSetCatStatus)
  76 END_EVENT_TABLE()
  77
  78
  79
  80 CTransferWnd::CTransferWnd( wxWindow* pParent )
  81         : wxPanel( pParent, -1 )
  82 {
  83         wxSizer* content = transferDlg(this, true);
  84         content->Show(this, true);
  85
  86         downloadlistctrl = CastChild( wxT("downloadList"), CDownloadListCtrl );
  87         clientlistctrl   = CastChild( ID_CLIENTLIST, CSourceListCtrl );
  88         m_dlTab          = CastChild( ID_CATEGORIES, CMuleNotebook );
  89
  90         CMuleNotebook* nb = CastChild( ID_CATEGORIES, CMuleNotebook );
  91         // We want to use our own popup
  92         nb->SetPopupHandler( this );
  93
  94         // Set default category
  95         theApp->glob_prefs->GetCategory(0)->title = GetCatTitle(thePrefs::GetAllcatType());
  96         theApp->glob_prefs->GetCategory(0)->path = thePrefs::GetIncomingDir();
  97
  98         // Show default + userdefined categories
  99         for ( uint32 i = 0; i < theApp->glob_prefs->GetCatCount(); i++ ) {
 100                 m_dlTab->AddPage( new wxPanel(m_dlTab), theApp->glob_prefs->GetCategory(i)->title );
 101         }
 102
 103         m_menu = NULL;
 104         m_splitter = 0;
 105
 106         wxConfigBase *config = wxConfigBase::Get();
 107
 108         // Check if the clientlist is hidden
 109         bool show = true;
 110         config->Read( wxT("/GUI/TransferWnd/ShowClientList"), &show, true );
 111
 112         if ( !show ) {
 113                 // Disable the client-list
 114                 wxCommandEvent event;
 115                 OnToggleClientList( event );
 116         }
 117
 118         // Load the last used splitter position
 119         m_splitter = config->Read( wxT("/GUI/TransferWnd/Splitter"), 463l );
 120 }
 121
 122
 123 CTransferWnd::~CTransferWnd()
 124 {
 125         wxConfigBase *config = wxConfigBase::Get();
 126
 127         if ( !clientlistctrl->GetShowing() ) {
 128                 // Save the splitter position
 129                 config->Write( wxT("/GUI/TransferWnd/Splitter"), m_splitter );
 130
 131                 // Save the visible status of the list
 132                 config->Write( wxT("/GUI/TransferWnd/ShowClientList"), false );
 133         } else {
 134                 wxSplitterWindow* splitter = CastChild( wxT("splitterWnd"), wxSplitterWindow );
 135
 136                 // Save the splitter position
 137                 config->Write( wxT("/GUI/TransferWnd/Splitter"), splitter->GetSashPosition() );
 138
 139                 // Save the visible status of the list
 140                 config->Write( wxT("/GUI/TransferWnd/ShowClientList"), true );
 141         }
 142 }
 143
 144
 145 void CTransferWnd::AddCategory( Category_Struct* category )
 146 {
 147         // Add the initial page
 148         m_dlTab->AddPage( new wxPanel(m_dlTab), category->title );
 149
 150         // Update the title
 151         UpdateCategory( m_dlTab->GetPageCount() - 1 );
 152
 153         theApp->amuledlg->m_searchwnd->UpdateCatChoice();
 154 }
 155
 156 void CTransferWnd::UpdateCategory( int index, bool titleChanged )
 157 {
 158         wxString label = theApp->glob_prefs->GetCategory( index )->title;
 159
 160         if ( thePrefs::ShowCatTabInfos() ) {
 161                 uint16 files = 0;
 162                 uint16 download = 0;
 163
 164                 for ( unsigned int i = 0; i < theApp->downloadqueue->GetFileCount(); ++i ) {
 165                         CPartFile *cur_file = theApp->downloadqueue->GetFileByIndex(i);
 166
 167                         if ( cur_file && cur_file->CheckShowItemInGivenCat(index) ) {
 168                                 files++;
 169
 170                                 if ( cur_file->GetTransferingSrcCount() ) {
 171                                         download++;
 172                                 }
 173                         }
 174                 }
 175
 176                 label += wxString::Format(wxT(" (%u/%u)"), download, files );
 177         }
 178
 179         m_dlTab->SetPageText( index, label );
 180
 181
 182         if ( titleChanged ) {
 183                 theApp->amuledlg->m_searchwnd->UpdateCatChoice();
 184         }
 185
 186 }
 187
 188
 189 void CTransferWnd::OnSetCatStatus( wxCommandEvent& event )
 190 {
 191         if ( event.GetId() == MP_CANCEL ) {
 192                 if ( wxMessageBox(_("Are you sure you wish to cancel and delete all files in this category?"),_("Confirmation Required"), wxYES_NO|wxCENTRE|wxICON_EXCLAMATION, this) == wxNO) {
 193                         return;
 194                 }
 195         }
 196
 197         CoreNotify_Download_Set_Cat_Status( m_dlTab->GetSelection(), event.GetId() );
 198 }
 199
 200
 201 void CTransferWnd::OnSetCatPriority( wxCommandEvent& event )
 202 {
 203         int priority = 0;
 204
 205         switch ( event.GetId() ) {
 206                 case MP_PRIOLOW:        priority = PR_LOW;      break;
 207                 case MP_PRIONORMAL:     priority = PR_NORMAL;   break;
 208                 case MP_PRIOHIGH:       priority = PR_HIGH;     break;
 209                 case MP_PRIOAUTO:       priority = PR_AUTO;     break;
 210                 default:
 211                         return;
 212         }
 213
 214         CoreNotify_Download_Set_Cat_Prio( m_dlTab->GetSelection(), priority );
 215 }
 216
 217
 218 void CTransferWnd::OnAddCategory(wxCommandEvent& WXUNUSED(event))
 219 {
 220         CCatDialog dialog( this,
 221         // Allow browse?
 222 #ifdef CLIENT_GUI
 223         false
 224 #else
 225         true
 226 #endif
 227         );
 228         if (dialog.ShowModal() == wxOK) {
 229                 // Add the files on this folder.
 230                 Category_Struct* newcat =
 231                         theApp->glob_prefs->GetCategory(
 232                                 theApp->glob_prefs->GetCatCount()-1);
 233                 theApp->sharedfiles->AddFilesFromDirectory(newcat->path);
 234                 theApp->sharedfiles->Reload();
 235         }
 236 }
 237
 238
 239 void CTransferWnd::OnDelCategory(wxCommandEvent& WXUNUSED(event))
 240 {
 241         RemoveCategory(m_dlTab->GetSelection());
 242 }
 243
 244
 245 void CTransferWnd::RemoveCategory(int index)
 246 {
 247         if ( index > 0 ) {
 248                 theApp->downloadqueue->ResetCatParts(index);
 249                 theApp->glob_prefs->RemoveCat(index);
 250                 RemoveCategoryPage(index);
 251                 if ( theApp->glob_prefs->GetCatCount() == 1 ) {
 252                         thePrefs::SetAllcatType(0);
 253                 }
 254                 theApp->glob_prefs->SaveCats();
 255                 theApp->amuledlg->m_searchwnd->UpdateCatChoice();
 256         }
 257 }
 258
 259
 260 void CTransferWnd::RemoveCategoryPage(int index)
 261 {
 262         m_dlTab->RemovePage(index);
 263         m_dlTab->SetSelection(0);
 264         downloadlistctrl->ChangeCategory(0);
 265 }
 266
 267
 268 void CTransferWnd::OnEditCategory( wxCommandEvent& WXUNUSED(event) )
 269 {
 270         Category_Struct* cat = theApp->glob_prefs->GetCategory(m_dlTab->GetSelection());
 271         CPath oldpath = cat->path;
 272         CCatDialog dialog( this,
 273         // Allow browse?
 274 #ifdef CLIENT_GUI
 275         false
 276 #else
 277         true
 278 #endif
 279                 , m_dlTab->GetSelection());
 280
 281         if (dialog.ShowModal() == wxOK) {
 282                 if (!oldpath.IsSameDir(cat->path)) {
 283                         theApp->sharedfiles->AddFilesFromDirectory(cat->path);
 284                         theApp->sharedfiles->Reload();
 285                 }
 286         }
 287 }
 288
 289
 290 void CTransferWnd::OnSetDefaultCat( wxCommandEvent& event )
 291 {
 292         thePrefs::SetAllcatType( event.GetId() - MP_CAT_SET0 );
 293         theApp->glob_prefs->GetCategory(0)->title = GetCatTitle( thePrefs::GetAllcatType() );
 294
 295         UpdateCategory( 0 );
 296
 297         downloadlistctrl->ChangeCategory( 0 );
 298
 299         theApp->glob_prefs->SaveCats();
 300
 301         downloadlistctrl->SortList();
 302 }
 303
 304
 305 void CTransferWnd::ShowQueueCount(uint32 number)
 306 {
 307 #if 0
 308         wxString str = wxString::Format( wxT("%u (%u %s)"), number, theStats::GetBannedCount(), _("Banned") );
 309         wxStaticText* label = CastChild( ID_CLIENTCOUNT, wxStaticText );
 310
 311         label->SetLabel( str );
 312         label->GetParent()->Layout();
 313 #endif
 314 }
 315
 316
 317 void CTransferWnd::OnCategoryChanged(wxNotebookEvent& evt)
 318 {
 319   downloadlistctrl->ChangeCategory(evt.GetSelection());
 320   downloadlistctrl->SortList();
 321 }
 322
 323
 324 void CTransferWnd::OnNMRclickDLtab(wxMouseEvent& evt)
 325 {
 326         // Only handle events from the category-notebook
 327         if ( evt.GetEventObject() != (wxObject*)m_dlTab )
 328                 return;
 329
 330         if ( m_dlTab->GetSelection() == -1 ) {
 331                 return;
 332         }
 333
 334         // Avoid opening another menu when it's already open
 335         if ( m_menu == NULL ) {
 336                 m_menu = new wxMenu( _("Category") );
 337
 338                 if ( m_dlTab->GetSelection() == 0 ) {
 339                         wxMenu* catmenu = new wxMenu();
 340
 341                         catmenu->Append( MP_CAT_SET0,      _("All") );
 342                         catmenu->Append( MP_CAT_SET0 + 1,  _("All others") );
 343
 344                         catmenu->AppendSeparator();
 345
 346                         catmenu->Append( MP_CAT_SET0 + 2,  _("Incomplete") );
 347                         catmenu->Append( MP_CAT_SET0 + 3,  _("Completed") );
 348                         catmenu->Append( MP_CAT_SET0 + 4,  _("Waiting") );
 349                         catmenu->Append( MP_CAT_SET0 + 5,  _("Downloading") );
 350                         catmenu->Append( MP_CAT_SET0 + 6,  _("Erroneous") );
 351                         catmenu->Append( MP_CAT_SET0 + 7,  _("Paused") );
 352                         catmenu->Append( MP_CAT_SET0 + 8,  _("Stopped") );
 353                         catmenu->Append( MP_CAT_SET0 + 15, _("Active") );
 354
 355                         catmenu->AppendSeparator();
 356
 357                         catmenu->Append( MP_CAT_SET0 + 9,  _("Video") );
 358                         catmenu->Append( MP_CAT_SET0 + 10, _("Audio") );
 359                         catmenu->Append( MP_CAT_SET0 + 11, _("Archive") );
 360                         catmenu->Append( MP_CAT_SET0 + 12, _("CD-Images") );
 361                         catmenu->Append( MP_CAT_SET0 + 13, _("Pictures") );
 362                         catmenu->Append( MP_CAT_SET0 + 14, _("Text") );
 363
 364                         m_menu->Append(0, _("Select view filter"), catmenu);
 365                 }
 366
 367                 m_menu->Append( MP_CAT_ADD, _("Add category") );
 368
 369                 if ( m_dlTab->GetSelection() ) {
 370                         m_menu->Append(MP_CAT_EDIT,_("Edit category"));
 371                         m_menu->Append(MP_CAT_REMOVE, _("Remove category"));
 372                 }
 373
 374                 m_menu->AppendSeparator();
 375
 376                 m_menu->Append( MP_CANCEL, _("Cancel"));
 377                 m_menu->Append( MP_STOP,   _("&Stop"));
 378                 m_menu->Append( MP_PAUSE,  _("&Pause"));
 379                 m_menu->Append( MP_RESUME, _("&Resume"));
 380
 381                 PopupMenu(m_menu, evt.GetPosition());
 382
 383                 delete m_menu;
 384                 m_menu = NULL;
 385         }
 386 }
 387
 388
 389 void CTransferWnd::OnBtnClearDownloads( wxCommandEvent& WXUNUSED(evt) )
 390 {
 391     downloadlistctrl->Freeze();
 392     downloadlistctrl->ClearCompleted();
 393     downloadlistctrl->Thaw();
 394 }
 395
 396
 397 void CTransferWnd::UpdateCatTabTitles()
 398 {
 399         for ( uint8 i = 0; i < m_dlTab->GetPageCount(); i++ ) {
 400                 UpdateCategory( i, false );
 401         }
 402 }
 403
 404
 405 void CTransferWnd::Prepare()
 406 {
 407         int header_height = s_clientlistHeader->GetSize().GetHeight();
 408         wxSplitterWindow* splitter = CastChild( wxT("splitterWnd"), wxSplitterWindow );
 409
 410         if ( !clientlistctrl->GetShowing() ) {
 411                 int height = splitter->GetWindow1()->GetSize().GetHeight();
 412                 height += splitter->GetWindow2()->GetSize().GetHeight();
 413
 414                 splitter->SetSashPosition( height - header_height );
 415         } else if ( m_splitter ) {
 416                 // Some sainity checking
 417                 if ( m_splitter < 90 ) {
 418                         m_splitter = 90;
 419                 } else {
 420                         int height = splitter->GetWindow1()->GetSize().GetHeight();
 421                         height += splitter->GetWindow2()->GetSize().GetHeight();
 422
 423                         if ( height - header_height * 2 < m_splitter ) {
 424                                 m_splitter = height - header_height * 2;
 425                         }
 426                 }
 427
 428                 splitter->SetSashPosition( m_splitter );
 429
 430                 m_splitter = 0;
 431         }
 432 }
 433
 434 void CTransferWnd::OnToggleClientList(wxCommandEvent& WXUNUSED(evt))
 435 {
 436         wxSplitterWindow* splitter = CastChild( wxT("splitterWnd"), wxSplitterWindow );
 437         wxBitmapButton*   button = CastChild( ID_CLIENTTOGGLE, wxBitmapButton );
 438
 439         if ( !clientlistctrl->GetShowing() ) {
 440                 splitter->SetSashPosition( m_splitter );
 441
 442                 clientlistctrl->SetShowing( true );
 443
 444                 button->SetBitmapLabel( amuleDlgImages( 10 ) );
 445                 button->SetBitmapFocus( amuleDlgImages( 10 ) );
 446                 button->SetBitmapSelected( amuleDlgImages( 10 ) );
 447
 448                 m_splitter = 0;
 449         } else {
 450                 clientlistctrl->SetShowing( false );
 451
 452                 int pos = splitter->GetSashPosition();
 453
 454                 // Add the height of the listctrl to the top-window
 455                 int height  = clientlistctrl->GetSize().GetHeight();
 456                 height += splitter->GetWindow1()->GetSize().GetHeight();
 457
 458                 splitter->SetSashPosition( height );
 459
 460                 m_splitter = pos;
 461
 462                 button->SetBitmapLabel( amuleDlgImages( 11 ) );
 463                 button->SetBitmapFocus( amuleDlgImages( 11 ) );
 464                 button->SetBitmapSelected( amuleDlgImages( 11 ) );
 465         }
 466 }
 467
 468
 469 void CTransferWnd::OnSashPositionChanging(wxSplitterEvent& evt)
 470 {
 471         int header_height = s_clientlistHeader->GetSize().GetHeight();
 472
 473         if ( evt.GetSashPosition() < 90 ) {
 474                 evt.SetSashPosition( 90 );
 475         } else {
 476                 wxSplitterWindow* splitter = wxStaticCast( evt.GetEventObject(), wxSplitterWindow);
 477
 478                 int height = splitter->GetWindow1()->GetSize().GetHeight();
 479                     height += splitter->GetWindow2()->GetSize().GetHeight();
 480
 481                 if ( !clientlistctrl->GetShowing() ) {
 482                         if ( height - evt.GetSashPosition() < header_height * 2 ) {
 483                                 evt.Veto();
 484                         } else {
 485                                 wxCommandEvent event;
 486                                 OnToggleClientList( event );
 487                         }
 488                 } else {
 489                         if ( height - evt.GetSashPosition() < header_height * 2 ) {
 490                                 evt.SetSashPosition( height - header_height );
 491
 492                                 wxCommandEvent event;
 493                                 OnToggleClientList( event );
 494                         }
 495                 }
 496         }
 497 }
 498 // File_checked_for_headers