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 "ClientListCtrl.h"             // Needed for CClientListCtrl
  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_SPLATTER, CTransferWnd::OnSashPositionChanging)
  60
  61         EVT_BUTTON(ID_BTNCLRCOMPL,              CTransferWnd::OnBtnClearDownloads)
  62         EVT_BUTTON(ID_BTNSWITCHUP,              CTransferWnd::SwitchUploadList)
  63         EVT_BUTTON(ID_CLIENTTOGGLE,             CTransferWnd::OnToggleClientList)
  64
  65         EVT_MENU_RANGE(MP_CAT_SET0, MP_CAT_SET0 + 15, CTransferWnd::OnSetDefaultCat)
  66         EVT_MENU(MP_CAT_ADD,                    CTransferWnd::OnAddCategory)
  67         EVT_MENU(MP_CAT_EDIT,                   CTransferWnd::OnEditCategory)
  68         EVT_MENU(MP_CAT_REMOVE,                 CTransferWnd::OnDelCategory)
  69         EVT_MENU(MP_PRIOLOW,                    CTransferWnd::OnSetCatPriority)
  70         EVT_MENU(MP_PRIONORMAL,                 CTransferWnd::OnSetCatPriority)
  71         EVT_MENU(MP_PRIOHIGH,                   CTransferWnd::OnSetCatPriority)
  72         EVT_MENU(MP_PRIOAUTO,                   CTransferWnd::OnSetCatPriority)
  73         EVT_MENU(MP_PAUSE,                      CTransferWnd::OnSetCatStatus)
  74         EVT_MENU(MP_STOP,                       CTransferWnd::OnSetCatStatus)
  75         EVT_MENU(MP_CANCEL,                     CTransferWnd::OnSetCatStatus)
  76         EVT_MENU(MP_RESUME,                     CTransferWnd::OnSetCatStatus)
  77 END_EVENT_TABLE()
  78
  79
  80
  81 CTransferWnd::CTransferWnd( wxWindow* pParent )
  82         : wxPanel( pParent, -1 )
  83 {
  84         wxSizer* content = transferDlg(this, true);
  85         content->Show(this, true);
  86
  87         downloadlistctrl = CastChild( wxT("downloadList"), CDownloadListCtrl );
  88         clientlistctrl   = CastChild( ID_CLIENTLIST, CClientListCtrl );
  89         m_dlTab          = CastChild( ID_CATEGORIES, CMuleNotebook );
  90
  91         CMuleNotebook* nb = CastChild( ID_CATEGORIES, CMuleNotebook );
  92         // We want to use our own popup
  93         nb->SetPopupHandler( this );
  94
  95         // Set default category
  96         theApp->glob_prefs->GetCategory(0)->title = GetCatTitle(thePrefs::GetAllcatType());
  97         theApp->glob_prefs->GetCategory(0)->path = thePrefs::GetIncomingDir();
  98
  99         // Show default + userdefined categories
 100         for ( uint32 i = 0; i < theApp->glob_prefs->GetCatCount(); i++ ) {
 101                 m_dlTab->AddPage( new wxPanel(m_dlTab), theApp->glob_prefs->GetCategory(i)->title );
 102         }
 103
 104         m_menu = NULL;
 105         m_splitter = 0;
 106
 107
 108         wxConfigBase *config = wxConfigBase::Get();
 109
 110         // Check if the clientlist is hidden
 111         bool show = true;
 112         config->Read( wxT("/GUI/TransferWnd/ShowClientList"), &show, true );
 113
 114         if ( !show ) {
 115                 // Disable the client-list
 116                 wxCommandEvent event;
 117                 OnToggleClientList( event );
 118         }
 119
 120         // Load the last used splitter position
 121         m_splitter = config->Read( wxT("/GUI/TransferWnd/Splitter"), 463l );
 122 }
 123
 124
 125 CTransferWnd::~CTransferWnd()
 126 {
 127         wxConfigBase *config = wxConfigBase::Get();
 128
 129         if ( clientlistctrl->GetListView() == vtNone ) {
 130                 // Save the splitter position
 131                 config->Write( wxT("/GUI/TransferWnd/Splitter"), m_splitter );
 132
 133                 // Save the visible status of the list
 134                 config->Write( wxT("/GUI/TransferWnd/ShowClientList"), false );
 135         } else {
 136                 wxSplitterWindow* splitter = CastChild( wxT("splitterWnd"), wxSplitterWindow );
 137
 138                 // Save the splitter position
 139                 config->Write( wxT("/GUI/TransferWnd/Splitter"), splitter->GetSashPosition() );
 140
 141                 // Save the visible status of the list
 142                 config->Write( wxT("/GUI/TransferWnd/ShowClientList"), true );
 143         }
 144 }
 145
 146
 147 void CTransferWnd::AddCategory( Category_Struct* category )
 148 {
 149         // Add the initial page
 150         m_dlTab->AddPage( new wxPanel(m_dlTab), category->title );
 151
 152         // Update the title
 153         UpdateCategory( m_dlTab->GetPageCount() - 1 );
 154
 155         theApp->amuledlg->m_searchwnd->UpdateCatChoice();
 156 }
 157
 158 void CTransferWnd::UpdateCategory( int index, bool titleChanged )
 159 {
 160         wxString label = theApp->glob_prefs->GetCategory( index )->title;
 161
 162         if ( thePrefs::ShowCatTabInfos() ) {
 163                 uint16 files = 0;
 164                 uint16 download = 0;
 165
 166                 for ( unsigned int i = 0; i < theApp->downloadqueue->GetFileCount(); ++i ) {
 167                         CPartFile *cur_file = theApp->downloadqueue->GetFileByIndex(i);
 168
 169                         if ( cur_file && cur_file->CheckShowItemInGivenCat(index) ) {
 170                                 files++;
 171
 172                                 if ( cur_file->GetTransferingSrcCount() ) {
 173                                         download++;
 174                                 }
 175                         }
 176                 }
 177
 178                 label += wxString::Format(wxT(" (%u/%u)"), download, files );
 179         }
 180
 181         m_dlTab->SetPageText( index, label );
 182
 183
 184         if ( titleChanged ) {
 185                 theApp->amuledlg->m_searchwnd->UpdateCatChoice();
 186         }
 187
 188 }
 189
 190
 191 void CTransferWnd::OnSetCatStatus( wxCommandEvent& event )
 192 {
 193         if ( event.GetId() == MP_CANCEL ) {
 194                 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) {
 195                         return;
 196                 }
 197         }
 198
 199         CoreNotify_Download_Set_Cat_Status( m_dlTab->GetSelection(), event.GetId() );
 200 }
 201
 202
 203 void CTransferWnd::OnSetCatPriority( wxCommandEvent& event )
 204 {
 205         int priority = 0;
 206
 207         switch ( event.GetId() ) {
 208                 case MP_PRIOLOW:        priority = PR_LOW;      break;
 209                 case MP_PRIONORMAL:     priority = PR_NORMAL;   break;
 210                 case MP_PRIOHIGH:       priority = PR_HIGH;     break;
 211                 case MP_PRIOAUTO:       priority = PR_AUTO;     break;
 212                 default:
 213                         return;
 214         }
 215
 216         CoreNotify_Download_Set_Cat_Prio( m_dlTab->GetSelection(), priority );
 217 }
 218
 219
 220 void CTransferWnd::OnAddCategory(wxCommandEvent& WXUNUSED(event))
 221 {
 222         CCatDialog dialog( this,
 223         // Allow browse?
 224 #ifdef CLIENT_GUI
 225         false
 226 #else
 227         true
 228 #endif
 229         );
 230         if (dialog.ShowModal() == wxOK) {
 231                 // Add the files on this folder.
 232                 Category_Struct* newcat =
 233                         theApp->glob_prefs->GetCategory(
 234                                 theApp->glob_prefs->GetCatCount()-1);
 235                 theApp->sharedfiles->AddFilesFromDirectory(newcat->path);
 236                 theApp->sharedfiles->Reload();
 237         }
 238 }
 239
 240
 241 void CTransferWnd::OnDelCategory(wxCommandEvent& WXUNUSED(event))
 242 {
 243         RemoveCategory(m_dlTab->GetSelection());
 244 }
 245
 246
 247 void CTransferWnd::RemoveCategory(int index)
 248 {
 249         if ( index > 0 ) {
 250                 theApp->downloadqueue->ResetCatParts(index);
 251                 theApp->glob_prefs->RemoveCat(index);
 252                 RemoveCategoryPage(index);
 253                 if ( theApp->glob_prefs->GetCatCount() == 1 ) {
 254                         thePrefs::SetAllcatType(0);
 255                 }
 256                 theApp->glob_prefs->SaveCats();
 257                 theApp->amuledlg->m_searchwnd->UpdateCatChoice();
 258         }
 259 }
 260
 261
 262 void CTransferWnd::RemoveCategoryPage(int index)
 263 {
 264         m_dlTab->RemovePage(index);
 265         m_dlTab->SetSelection(0);
 266         downloadlistctrl->ChangeCategory(0);
 267 }
 268
 269
 270 void CTransferWnd::OnEditCategory( wxCommandEvent& WXUNUSED(event) )
 271 {
 272         Category_Struct* cat = theApp->glob_prefs->GetCategory(m_dlTab->GetSelection());
 273         CPath oldpath = cat->path;
 274         CCatDialog dialog( this,
 275         // Allow browse?
 276 #ifdef CLIENT_GUI
 277         false
 278 #else
 279         true
 280 #endif
 281                 , m_dlTab->GetSelection());
 282
 283         if (dialog.ShowModal() == wxOK) {
 284                 if (!oldpath.IsSameDir(cat->path)) {
 285                         theApp->sharedfiles->AddFilesFromDirectory(cat->path);
 286                         theApp->sharedfiles->Reload();
 287                 }
 288         }
 289 }
 290
 291
 292 void CTransferWnd::OnSetDefaultCat( wxCommandEvent& event )
 293 {
 294         thePrefs::SetAllcatType( event.GetId() - MP_CAT_SET0 );
 295         theApp->glob_prefs->GetCategory(0)->title = GetCatTitle( thePrefs::GetAllcatType() );
 296
 297         UpdateCategory( 0 );
 298
 299         downloadlistctrl->ChangeCategory( 0 );
 300
 301         theApp->glob_prefs->SaveCats();
 302
 303         downloadlistctrl->SortList();
 304 }
 305
 306
 307 void CTransferWnd::ShowQueueCount(uint32 number)
 308 {
 309         wxString str = wxString::Format( wxT("%u (%u %s)"), number, theStats::GetBannedCount(), _("Banned") );
 310         wxStaticText* label = CastChild( ID_CLIENTCOUNT, wxStaticText );
 311
 312         label->SetLabel( str );
 313         label->GetParent()->Layout();
 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->GetListView() == vtNone ) {
 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
 435 void CTransferWnd::SwitchUploadList(wxCommandEvent& WXUNUSED(evt))
 436 {
 437         clientlistctrl->ToggleView();
 438         wxStaticText* label = CastChild( wxT("uploadTitle"), wxStaticText );
 439
 440         switch ( clientlistctrl->GetListView() ) {
 441                 case vtNone:
 442                         return;
 443
 444                 case vtUploading:
 445                         label->SetLabel( _("Uploads") );
 446                         break;
 447
 448                 case vtQueued:
 449                         label->SetLabel( _("On Queue") );
 450                         break;
 451
 452                 case vtClients:
 453                         label->SetLabel( _("Clients") );
 454                         break;
 455         }
 456
 457         label->GetParent()->Layout();
 458 }
 459
 460
 461 void CTransferWnd::OnToggleClientList(wxCommandEvent& WXUNUSED(evt))
 462 {
 463         wxSplitterWindow* splitter = CastChild( wxT("splitterWnd"), wxSplitterWindow );
 464         wxBitmapButton*   button = CastChild( ID_CLIENTTOGGLE, wxBitmapButton );
 465
 466         // Keeps track of what was last selected.
 467         static ViewType lastView = vtNone;
 468
 469
 470         if ( clientlistctrl->GetListView() == vtNone ) {
 471                 splitter->SetSashPosition( m_splitter );
 472
 473                 clientlistctrl->SetListView( lastView );
 474
 475                 button->SetBitmapLabel( amuleDlgImages( 10 ) );
 476                 button->SetBitmapFocus( amuleDlgImages( 10 ) );
 477                 button->SetBitmapSelected( amuleDlgImages( 10 ) );
 478
 479                 m_splitter = 0;
 480         } else {
 481                 lastView = clientlistctrl->GetListView();
 482                 clientlistctrl->SetListView( vtNone );
 483
 484                 int pos = splitter->GetSashPosition();
 485
 486                 // Add the height of the listctrl to the top-window
 487                 int height  = clientlistctrl->GetSize().GetHeight();
 488                 height += splitter->GetWindow1()->GetSize().GetHeight();
 489
 490                 splitter->SetSashPosition( height );
 491
 492                 m_splitter = pos;
 493
 494                 button->SetBitmapLabel( amuleDlgImages( 11 ) );
 495                 button->SetBitmapFocus( amuleDlgImages( 11 ) );
 496                 button->SetBitmapSelected( amuleDlgImages( 11 ) );
 497         }
 498
 499         FindWindow(ID_BTNSWITCHUP)->Enable( clientlistctrl->GetListView() != vtNone );
 500 }
 501
 502
 503 void CTransferWnd::OnSashPositionChanging(wxSplitterEvent& evt)
 504 {
 505         int header_height = s_clientlistHeader->GetSize().GetHeight();
 506
 507         if ( evt.GetSashPosition() < 90 ) {
 508                 evt.SetSashPosition( 90 );
 509         } else {
 510                 wxSplitterWindow* splitter = wxStaticCast( evt.GetEventObject(), wxSplitterWindow);
 511
 512                 int height = splitter->GetWindow1()->GetSize().GetHeight();
 513                     height += splitter->GetWindow2()->GetSize().GetHeight();
 514
 515                 if ( clientlistctrl->GetListView() == vtNone ) {
 516                         if ( height - evt.GetSashPosition() < header_height * 2 ) {
 517                                 evt.Veto();
 518                         } else {
 519                                 wxCommandEvent event;
 520                                 OnToggleClientList( event );
 521                         }
 522                 } else {
 523                         if ( height - evt.GetSashPosition() < header_height * 2 ) {
 524                                 evt.SetSashPosition( height - header_height );
 525
 526                                 wxCommandEvent event;
 527                                 OnToggleClientList( event );
 528                         }
 529                 }
 530         }
 531 }
 532 // File_checked_for_headers