ctrldownloadtabbar.cpp

著名的下载软件核心Shareaza

//
// CtrlDownloadTabBar.cpp
//
// Copyright (c) Shareaza Development Team, 2002-2004.
// This file is part of SHAREAZA (www.shareaza.com)
//
// Shareaza is free software; you can redistribute it
// and/or modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2 of
// the License, or (at your option) any later version.
//
// Shareaza is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Shareaza; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//

#include "StdAfx.h"
#include "Shareaza.h"
#include "Settings.h"
#include "CoolInterface.h"
#include "CoolMenu.h"
#include "Transfers.h"
#include "Downloads.h"
#include "Download.h"
#include "DownloadGroup.h"
#include "DownloadGroups.h"
#include "CtrlDownloadTabBar.h"
#include "DlgDownloadGroup.h"
#include "ShellIcons.h"
#include "Skin.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

BEGIN_MESSAGE_MAP(CDownloadTabBar, CControlBar)
	ON_WM_LBUTTONDOWN()
	ON_WM_RBUTTONUP()
	ON_WM_LBUTTONDBLCLK()
	ON_WM_MOUSEMOVE()
	ON_WM_TIMER()
	ON_WM_CREATE()
	ON_WM_MEASUREITEM()
	ON_WM_DRAWITEM()
	ON_COMMAND(ID_DOWNLOAD_GROUP_NEW, OnDownloadGroupNew)
	ON_UPDATE_COMMAND_UI(ID_DOWNLOAD_GROUP_REMOVE, OnUpdateDownloadGroupRemove)
	ON_COMMAND(ID_DOWNLOAD_GROUP_REMOVE, OnDownloadGroupRemove)
	ON_UPDATE_COMMAND_UI(ID_DOWNLOAD_GROUP_PROPERTIES, OnUpdateDownloadGroupProperties)
	ON_COMMAND(ID_DOWNLOAD_GROUP_PROPERTIES, OnDownloadGroupProperties)
	ON_UPDATE_COMMAND_UI(ID_DOWNLOAD_GROUP_RESUME, OnUpdateDownloadGroupResume)
	ON_COMMAND(ID_DOWNLOAD_GROUP_RESUME, OnDownloadGroupResume)
	ON_UPDATE_COMMAND_UI(ID_DOWNLOAD_GROUP_PAUSE, OnUpdateDownloadGroupPause)
	ON_COMMAND(ID_DOWNLOAD_GROUP_PAUSE, OnDownloadGroupPause)
	ON_UPDATE_COMMAND_UI(ID_DOWNLOAD_GROUP_CLEAR, OnUpdateDownloadGroupClear)
	ON_COMMAND(ID_DOWNLOAD_GROUP_CLEAR, OnDownloadGroupClear)
END_MESSAGE_MAP()


/////////////////////////////////////////////////////////////////////////////
// CDownloadTabBar construction

CDownloadTabBar::CDownloadTabBar()
{
	m_bTimer		= FALSE;
	m_bMenuGray		= FALSE;
	m_nCookie		= 0;
	m_nMaximumWidth	= 140;
	m_nMessage		= 0;
	m_pHot			= NULL;
}

CDownloadTabBar::~CDownloadTabBar()
{
	for ( POSITION pos = m_pItems.GetHeadPosition() ; pos ; )
	{
		delete (TabItem*)m_pItems.GetNext( pos );
	}
	
	m_pItems.RemoveAll();
}

/////////////////////////////////////////////////////////////////////////////
// CDownloadTabBar operations

BOOL CDownloadTabBar::Create(CWnd* pParentWnd, DWORD dwStyle, UINT nID)
{
	CRect rc;
	dwStyle |= WS_CHILD;
	return CWnd::Create( NULL, NULL, dwStyle, rc, pParentWnd, nID, NULL );
}

void CDownloadTabBar::SetWatermark(HBITMAP hBitmap)
{
	if ( m_bmImage.m_hObject != NULL ) m_bmImage.DeleteObject();
	if ( hBitmap != NULL ) m_bmImage.Attach( hBitmap );
}

/////////////////////////////////////////////////////////////////////////////
// CDownloadTabBar message handlers

int CDownloadTabBar::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if ( CControlBar::OnCreate( lpCreateStruct ) == -1 ) return -1;
	m_dwStyle |= CBRS_BORDER_3D;
	return 0;
}

CSize CDownloadTabBar::CalcFixedLayout(BOOL bStretch, BOOL bHorz)
{
	CSize size( 32767, 26 );
	
	if ( CWnd* pParent = AfxGetMainWnd() )
	{
		CRect rc;
		pParent->GetWindowRect( &rc );
		if ( rc.Width() > 32 ) size.cx = rc.Width() + 2;
	}
	
	return size;
}

void CDownloadTabBar::Update(int nCookie)
{
	CSingleLock pLock( &DownloadGroups.m_pSection, TRUE );
	
	if ( m_nCookie != DownloadGroups.GetGroupCookie() )
	{
		UpdateGroups( nCookie );
	}
	else
	{
		UpdateStates( nCookie );
	}
}

void CDownloadTabBar::UpdateGroups(int nCookie)
{
	for ( POSITION pos = m_pItems.GetHeadPosition() ; pos ; )
	{
		delete (TabItem*)m_pItems.GetNext( pos );
	}
	
	m_pItems.RemoveAll();
	
	BOOL bFoundHot = ( m_pHot == NULL );
	
	for ( pos = DownloadGroups.GetIterator() ; pos ; )
	{
		CDownloadGroup* pGroup = DownloadGroups.GetNext( pos );
		m_pItems.AddTail( new TabItem( pGroup, nCookie ) );
		
		if ( ! bFoundHot && pGroup == m_pHot->m_pGroup ) bFoundHot = TRUE;
	}
	
	if ( ! bFoundHot ) m_pHot = NULL;
	
	m_nCookie = DownloadGroups.GetGroupCookie();
	Invalidate();
}

void CDownloadTabBar::UpdateStates(int nCookie)
{
	BOOL bChanged = FALSE;
	
	for ( POSITION posNext = m_pItems.GetHeadPosition() ; posNext ; )
	{
		POSITION posThis	= posNext;
		TabItem* pItem		= (TabItem*)m_pItems.GetNext( posNext );
		
		if ( DownloadGroups.Check( pItem->m_pGroup ) )
		{
			bChanged |= pItem->Update( nCookie );
		}
		else
		{
			m_pItems.RemoveAt( posThis );
			delete pItem;
		}
	}
	
	if ( bChanged ) Invalidate();
}

CDownloadTabBar::TabItem* CDownloadTabBar::HitTest(const CPoint& point, CRect* pItemRect) const
{
	if ( m_pItems.IsEmpty() ) return NULL;
	
	CRect rc;
	GetClientRect( &rc );
	CalcInsideRect( rc, FALSE );
	
	rc.left -= m_cyTopBorder;
	rc.top -= m_cxLeftBorder;
	rc.right += m_cyBottomBorder;
	rc.bottom += m_cxRightBorder;
	
	CRect rcItem( rc.left + 3, rc.top + 1, 0, rc.bottom - 1 );
	rcItem.right = ( rc.Width() - 3 * m_pItems.GetCount() ) / m_pItems.GetCount() + 3;
	rcItem.right = min( rcItem.right, m_nMaximumWidth );
	
	for ( POSITION pos = m_pItems.GetHeadPosition() ; pos ; )
	{
		TabItem* pItem = (TabItem*)m_pItems.GetNext( pos );
		
		if ( rcItem.PtInRect( point ) )
		{
			if ( pItemRect ) *pItemRect = rcItem;
			return pItem;
		}
		
		rcItem.OffsetRect( rcItem.Width() + 3, 0 );
	}
	
	return NULL;
}

int CDownloadTabBar::OnToolHitTest(CPoint point, TOOLINFO* pTI) const
{
	CRect rcItem;
	TabItem* pItem = HitTest( point, &rcItem );
	
	if ( pItem == NULL ) return -1;
	if ( pTI == NULL ) return 1;
	
	pTI->uFlags		= 0;
	pTI->hwnd		= GetSafeHwnd();
	pTI->uId		= (UINT)pItem->m_pGroup;
	pTI->rect		= rcItem;
	pTI->lpszText	= _tcsdup( pItem->m_sCaption );
	
	return pTI->uId;
}

void CDownloadTabBar::DoPaint(CDC* pDC)
{
	ASSERT_VALID(this);
	ASSERT_VALID(pDC);
	
	CDC* pOutDC = pDC;
	CRect rc;
	
	GetClientRect( &rc );
	
	if ( m_bmImage.m_hObject != NULL )
	{
		pDC = CoolInterface.GetBuffer( *pDC, rc.Size() );
		CoolInterface.DrawWatermark( pDC, &rc, &m_bmImage );
	}
	
	DrawBorders( pDC, rc );
	
	CFont* pOldFont = (CFont*)pDC->SelectObject( &CoolInterface.m_fntNormal );
	
	if ( m_pItems.GetCount() )
	{
		CRect rcItem( rc.left + 3, rc.top + 1, 0, rc.bottom - 1 );
		rcItem.right = ( rc.Width() - 3 * m_pItems.GetCount() ) / m_pItems.GetCount() + 3;
		rcItem.right = min( rcItem.right, m_nMaximumWidth );
		
		for ( POSITION pos = m_pItems.GetHeadPosition() ; pos ; )
		{
			TabItem* pItem = (TabItem*)m_pItems.GetNext( pos );
			
			pItem->Paint( this, pDC, &rcItem, m_pHot == pItem, pDC != pOutDC );
			pDC->ExcludeClipRect( &rcItem );
			
			rcItem.OffsetRect( rcItem.Width() + 3, 0 );
		}
		
		if ( pDC == pOutDC ) pDC->FillSolidRect( &rc, CoolInterface.m_crMidtone );
	}
	else
	{
		CSize  sz = pDC->GetTextExtent( m_sMessage );
		CPoint pt = rc.CenterPoint();
		pt.x -= sz.cx / 2; pt.y -= sz.cy / 2 + 1;
		
		pDC->SetBkColor( CoolInterface.m_crMidtone );
		pDC->SetTextColor( CoolInterface.m_crDisabled );
		
		if ( pDC == pOutDC )
		{
			pDC->SetBkMode( OPAQUE );
			pDC->ExtTextOut( pt.x, pt.y, ETO_CLIPPED|ETO_OPAQUE, &rc, m_sMessage, NULL );
		}
		else
		{
			pDC->SetBkMode( TRANSPARENT );
			pDC->ExtTextOut( pt.x, pt.y, ETO_CLIPPED, &rc, m_sMessage, NULL );
		}
	}
	
	pDC->SelectObject( pOldFont );
	
	if ( pDC != pOutDC )
	{
		GetClientRect( &rc );
		pOutDC->BitBlt( 0, 0, rc.Width(), rc.Height(), pDC, 0, 0, SRCCOPY );
		pDC->SelectClipRgn( NULL );
	}
}

void CDownloadTabBar::OnMouseMove(UINT nFlags, CPoint point) 
{
	TabItem* pItem = HitTest( point );
	
	if ( pItem != m_pHot )
	{
		m_pHot = pItem;
		Invalidate();
	}
	
	if ( ! m_bTimer && m_pHot )
	{
		SetTimer( 1, 100, NULL );
		m_bTimer = TRUE;
	}