src/TransferWnd.cpp

   1 //
   2 // This file is part of the aMule Project.
   3 //
   4 // Copyright (c) 2003-2011 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         // Set disabled image for clear complete button
  91         CastChild(ID_BTNCLRCOMPL, wxBitmapButton)->SetBitmapDisabled(amuleDlgImages(34));
  92
  93         // We want to use our own popup
  94         m_dlTab->SetPopupHandler( this );
  95
  96         // Set default category
  97         theApp->glob_prefs->GetCategory(0)->title = GetCatTitle(thePrefs::GetAllcatFilter());
  98         theApp->glob_prefs->GetCategory(0)->path = thePrefs::GetIncomingDir();
  99
 100         // Show default + userdefined categories
 101         for ( uint32 i = 0; i < theApp->glob_prefs->GetCatCount(); i++ ) {
 102                 m_dlTab->AddPage( new wxPanel(m_dlTab), theApp->glob_prefs->GetCategory(i)->title );
 103         }
 104
 105         m_menu = NULL;
 106         m_splitter = 0;
 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         clientlistctrl->SetShowing(show);
 114         // Load the last used splitter position
 115         m_splitter = config->Read( wxT("/GUI/TransferWnd/Splitter"), 463l );
 116 }
 117
 118
 119 CTransferWnd::~CTransferWnd()
 120 {
 121         wxConfigBase *config = wxConfigBase::Get();
 122
 123         if ( !clientlistctrl->GetShowing() ) {
 124                 // Save the splitter position
 125                 config->Write( wxT("/GUI/TransferWnd/Splitter"), m_splitter );
 126
 127                 // Save the visible status of the list
 128                 config->Write( wxT("/GUI/TransferWnd/ShowClientList"), false );
 129         } else {
 130                 wxSplitterWindow* splitter = CastChild( wxT("splitterWnd"), wxSplitterWindow );
 131
 132                 // Save the splitter position
 133                 config->Write( wxT("/GUI/TransferWnd/Splitter"), splitter->GetSashPosition() );
 134
 135                 // Save the visible status of the list
 136                 config->Write( wxT("/GUI/TransferWnd/ShowClientList"), true );
 137         }
 138 }
 139
 140
 141 void CTransferWnd::AddCategory( Category_Struct* category )
 142 {
 143         // Add the initial page
 144         m_dlTab->AddPage( new wxPanel(m_dlTab), category->title );
 145
 146         // Update the title
 147         UpdateCategory( m_dlTab->GetPageCount() - 1 );
 148
 149         theApp->amuledlg->m_searchwnd->UpdateCatChoice();
 150 }
 151
 152 void CTransferWnd::UpdateCategory(int index)
 153 {
 154         uint32 nrCats = theApp->glob_prefs->GetCatCount();
 155         std::vector<uint16> files, downloads;
 156         bool showCatTabInfos = thePrefs::ShowCatTabInfos();
 157         if (showCatTabInfos) {
 158                 files.insert(files.begin(), nrCats, 0);
 159                 downloads.insert(downloads.begin(), nrCats, 0);
 160
 161 #ifdef CLIENT_GUI
 162                 for (CDownQueueRem::const_iterator it = theApp->downloadqueue->begin(); it != theApp->downloadqueue->end(); ++it) {
 163                         CPartFile *cur_file = it->second;
 164 #else
 165                 std::vector<CPartFile*> fileList;
 166                 theApp->downloadqueue->CopyFileList(fileList, true);
 167                 int size = fileList.size();
 168                 for (int i = 0; i < size; ++i ) {
 169                         CPartFile *cur_file = fileList[i];
 170 #endif
 171                         bool downloading = cur_file->GetTransferingSrcCount() > 0;
 172                         int fileCat = cur_file->GetCategory();
 173                         if ((index == -1 || fileCat == index) && cur_file->CheckShowItemInGivenCat(fileCat)) {
 174                                 files[fileCat]++;
 175                                 if (downloading) {
 176                                         downloads[fileCat]++;
 177                                 }
 178                         }
 179                         if (index == -1 && fileCat > 0 && cur_file->CheckShowItemInGivenCat(0)) {
 180                                 files[0]++;
 181                                 if (downloading) {
 182                                         downloads[0]++;
 183                                 }
 184                         }
 185                 }
 186
 187         }
 188         int start, end;
 189         if (index == -1) {
 190                 start = 0;
 191                 end = nrCats - 1;
 192         } else {
 193                 start = index;
 194                 end = index;
 195         }
 196         for (int i = start; i <= end; i++) {
 197                 wxString label = theApp->glob_prefs->GetCategory(i)->title;
 198                 if (showCatTabInfos) {
 199                         label += CFormat(wxT(" (%u/%u)")) % downloads[i] % files[i];
 200                 }
 201                 m_dlTab->SetPageText(i, label);
 202         }
 203 }
 204
 205
 206 void CTransferWnd::OnSetCatStatus( wxCommandEvent& event )
 207 {
 208         if ( event.GetId() == MP_CANCEL ) {
 209                 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) {
 210                         return;
 211                 }
 212         }
 213
 214         CoreNotify_Download_Set_Cat_Status( m_dlTab->GetSelection(), event.GetId() );
 215 }
 216
 217
 218 void CTransferWnd::OnSetCatPriority( wxCommandEvent& event )
 219 {
 220         int priority = 0;
 221
 222         switch ( event.GetId() ) {
 223                 case MP_PRIOLOW:        priority = PR_LOW;      break;
 224                 case MP_PRIONORMAL:     priority = PR_NORMAL;   break;
 225                 case MP_PRIOHIGH:       priority = PR_HIGH;     break;
 226                 case MP_PRIOAUTO:       priority = PR_AUTO;     break;
 227                 default:
 228                         return;
 229         }
 230
 231         CoreNotify_Download_Set_Cat_Prio( m_dlTab->GetSelection(), priority );
 232 }
 233
 234
 235 void CTransferWnd::OnAddCategory(wxCommandEvent& WXUNUSED(event))
 236 {
 237         if (theApp->glob_prefs->GetCatCount() >= 99) {
 238                 wxMessageBox(_("Only 99 categories are supported."), _("Too many categories!"), wxOK | wxICON_EXCLAMATION);
 239                 return;
 240         }
 241         CCatDialog dialog( this,
 242         // Allow browse?
 243 #ifdef CLIENT_GUI
 244         false
 245 #else
 246         true
 247 #endif
 248         );
 249         dialog.ShowModal();
 250 }
 251
 252
 253 void CTransferWnd::OnDelCategory(wxCommandEvent& WXUNUSED(event))
 254 {
 255         RemoveCategory(m_dlTab->GetSelection());
 256 }
 257
 258
 259 void CTransferWnd::RemoveCategory(int index)
 260 {
 261         if ( index > 0 ) {
 262                 theApp->downloadqueue->ResetCatParts(index);
 263                 theApp->glob_prefs->RemoveCat(index);
 264                 RemoveCategoryPage(index);
 265                 if ( theApp->glob_prefs->GetCatCount() == 1 ) {
 266                         thePrefs::SetAllcatFilter( acfAll );
 267                 }
 268                 theApp->glob_prefs->SaveCats();
 269                 theApp->amuledlg->m_searchwnd->UpdateCatChoice();
 270         }
 271 }
 272
 273
 274 void CTransferWnd::RemoveCategoryPage(int index)
 275 {
 276         m_dlTab->RemovePage(index);
 277         m_dlTab->SetSelection(0);
 278         downloadlistctrl->ChangeCategory(0);
 279 }
 280
 281
 282 void CTransferWnd::OnEditCategory( wxCommandEvent& WXUNUSED(event) )
 283 {
 284         CCatDialog dialog( this,
 285         // Allow browse?
 286 #ifdef CLIENT_GUI
 287         false
 288 #else
 289         true
 290 #endif
 291                 , m_dlTab->GetSelection());
 292
 293         dialog.ShowModal();
 294 }
 295
 296
 297 void CTransferWnd::OnSetDefaultCat( wxCommandEvent& event )
 298 {
 299         thePrefs::SetAllcatFilter( static_cast<AllCategoryFilter>(event.GetId() - MP_CAT_SET0) );
 300         theApp->glob_prefs->GetCategory(0)->title = GetCatTitle( thePrefs::GetAllcatFilter() );
 301
 302         UpdateCategory( 0 );
 303
 304         downloadlistctrl->ChangeCategory( 0 );
 305
 306         theApp->glob_prefs->SaveCats();
 307
 308         downloadlistctrl->SortList();
 309 }
 310
 311
 312 void CTransferWnd::OnCategoryChanged(wxNotebookEvent& evt)
 313 {
 314         // First remove currently showing sources (switching cat will deselect all)
 315         CKnownFileVector filesVector;
 316         clientlistctrl->ShowSources(filesVector);
 317         // Then change cat
 318         downloadlistctrl->ChangeCategory(evt.GetSelection());
 319         downloadlistctrl->SortList();
 320 }
 321
 322
 323 void CTransferWnd::OnNMRclickDLtab(wxMouseEvent& evt)
 324 {
 325         // Only handle events from the category-notebook
 326         if ( evt.GetEventObject() != (wxObject*)m_dlTab )
 327                 return;
 328
 329         if ( m_dlTab->GetSelection() == -1 ) {
 330                 return;
 331         }
 332
 333         // Avoid opening another menu when it's already open
 334         if ( m_menu == NULL ) {
 335                 m_menu = new wxMenu( _("Category") );
 336
 337                 if ( m_dlTab->GetSelection() == 0 ) {
 338                         wxMenu* catmenu = new wxMenu();
 339
 340                         catmenu->Append( MP_CAT_SET0,      _("All") );
 341                         catmenu->Append( MP_CAT_SET0 + 1,  _("All others") );
 342
 343                         catmenu->AppendSeparator();
 344
 345                         catmenu->Append( MP_CAT_SET0 + 2,  _("Incomplete") );
 346                         catmenu->Append( MP_CAT_SET0 + 3,  _("Completed") );
 347                         catmenu->Append( MP_CAT_SET0 + 4,  _("Waiting") );
 348                         catmenu->Append( MP_CAT_SET0 + 5,  _("Downloading") );
 349                         catmenu->Append( MP_CAT_SET0 + 6,  _("Erroneous") );
 350                         catmenu->Append( MP_CAT_SET0 + 7,  _("Paused") );
 351                         catmenu->Append( MP_CAT_SET0 + 8,  _("Stopped") );
 352                         catmenu->Append( MP_CAT_SET0 + 15, _("Active") );
 353
 354                         catmenu->AppendSeparator();
 355
 356                         catmenu->Append( MP_CAT_SET0 + 9,  _("Video") );
 357                         catmenu->Append( MP_CAT_SET0 + 10, _("Audio") );
 358                         catmenu->Append( MP_CAT_SET0 + 11, _("Archive") );
 359                         catmenu->Append( MP_CAT_SET0 + 12, _("CD-Images") );
 360                         catmenu->Append( MP_CAT_SET0 + 13, _("Pictures") );
 361                         catmenu->Append( MP_CAT_SET0 + 14, _("Text") );
 362
 363                         m_menu->Append(0, _("Select view filter"), catmenu);
 364                 }
 365
 366                 m_menu->Append( MP_CAT_ADD, _("Add category") );
 367
 368                 if ( m_dlTab->GetSelection() ) {
 369                         m_menu->Append(MP_CAT_EDIT,_("Edit category"));
 370                         m_menu->Append(MP_CAT_REMOVE, _("Remove category"));
 371                 }
 372
 373                 m_menu->AppendSeparator();
 374
 375                 m_menu->Append( MP_CANCEL, _("Cancel"));
 376                 m_menu->Append( MP_STOP,   _("&Stop"));
 377                 m_menu->Append( MP_PAUSE,  _("&Pause"));
 378                 m_menu->Append( MP_RESUME, _("&Resume"));
 379
 380                 PopupMenu(m_menu, evt.GetPosition());
 381
 382                 delete m_menu;
 383                 m_menu = NULL;
 384         }
 385 }
 386
 387
 388 void CTransferWnd::OnBtnClearDownloads( wxCommandEvent& WXUNUSED(evt) )
 389 {
 390     downloadlistctrl->Freeze();
 391     downloadlistctrl->ClearCompleted();
 392     downloadlistctrl->Thaw();
 393 }
 394
 395
 396 void CTransferWnd::Prepare()
 397 {
 398         wxSplitterWindow* splitter = CastChild( wxT("splitterWnd"), wxSplitterWindow );
 399         int height = splitter->GetSize().GetHeight();
 400         int header_height = s_clientlistHeader->GetSize().GetHeight();
 401
 402         if ( m_splitter ) {
 403                 // Some sanity checking
 404                 if ( m_splitter < s_splitterMin ) {
 405                         m_splitter = s_splitterMin;
 406                 } else if ( m_splitter > height - header_height * 2 ) {
 407                         m_splitter = height - header_height * 2;
 408                 }
 409                 splitter->SetSashPosition( m_splitter );
 410                 m_splitter = 0;
 411         }
 412
 413         if ( !clientlistctrl->GetShowing() ) {
 414                 // use a toggle event to close it (calculate size, change button)
 415                 clientlistctrl->SetShowing( true );     // so it will be toggled to false
 416                 wxCommandEvent evt1;
 417                 OnToggleClientList( evt1 );
 418         }
 419 }
 420
 421 void CTransferWnd::OnToggleClientList(wxCommandEvent& WXUNUSED(evt))
 422 {
 423         wxSplitterWindow* splitter = CastChild( wxT("splitterWnd"), wxSplitterWindow );
 424         wxBitmapButton*   button = CastChild( ID_CLIENTTOGGLE, wxBitmapButton );
 425
 426         if ( !clientlistctrl->GetShowing() ) {
 427                 splitter->SetSashPosition( m_splitter );
 428                 m_splitter = 0;
 429
 430                 clientlistctrl->SetShowing( true );
 431
 432                 button->SetBitmapLabel( amuleDlgImages( 10 ) );
 433                 button->SetBitmapFocus( amuleDlgImages( 10 ) );
 434                 button->SetBitmapSelected( amuleDlgImages( 10 ) );
 435                 button->SetBitmapHover( amuleDlgImages( 10 ) );
 436         } else {
 437                 clientlistctrl->SetShowing( false );
 438
 439                 m_splitter = splitter->GetSashPosition();
 440
 441                 // Add the height of the listctrl to the top-window
 442                 int height = clientlistctrl->GetSize().GetHeight()
 443                                          + splitter->GetWindow1()->GetSize().GetHeight();
 444
 445                 splitter->SetSashPosition( height );
 446
 447                 button->SetBitmapLabel( amuleDlgImages( 11 ) );
 448                 button->SetBitmapFocus( amuleDlgImages( 11 ) );
 449                 button->SetBitmapSelected( amuleDlgImages( 11 ) );
 450                 button->SetBitmapHover( amuleDlgImages( 11 ) );
 451         }
 452 }
 453
 454
 455 void CTransferWnd::OnSashPositionChanging(wxSplitterEvent& evt)
 456 {
 457         if ( evt.GetSashPosition() < s_splitterMin ) {
 458                 evt.SetSashPosition( s_splitterMin );
 459         } else {
 460                 wxSplitterWindow* splitter = wxStaticCast( evt.GetEventObject(), wxSplitterWindow);
 461                 wxCHECK_RET(splitter, wxT("ERROR: NULL splitter in CTransferWnd::OnSashPositionChanging"));
 462
 463                 int height = splitter->GetSize().GetHeight();
 464                 int header_height = s_clientlistHeader->GetSize().GetHeight();
 465                 int mousey = wxGetMousePosition().y - splitter->GetScreenRect().GetTop();
 466
 467                 if ( !clientlistctrl->GetShowing() ) {
 468                         // lower window hidden
 469                         if ( height - mousey < header_height * 2 ) {
 470                                 // no moving down if already hidden
 471                                 evt.Veto();
 472                         } else {
 473                                 // show it
 474                                 m_splitter = mousey;    // prevent jumping if it was minimized and is then shown by dragging the sash
 475                                 wxCommandEvent event;
 476                                 OnToggleClientList( event );
 477                         }
 478                 } else {
 479                         // lower window showing
 480                         if ( height - mousey < header_height * 2 ) {
 481                                 // hide it
 482                                 wxCommandEvent event;
 483                                 OnToggleClientList( event );
 484                         } else {
 485                                 // normal resize
 486                                 // If several events queue up, setting the sash to the current mouse position
 487                                 // will speed up things and make sash moving more smoothly.
 488                                 evt.SetSashPosition( mousey );
 489                         }
 490                 }
 491         }
 492 }
 493 // File_checked_for_headers