ctrlmatch.cpp

著名的下载软件核心Shareaza

//
// CtrlMatch.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 "CtrlMatch.h"
#include "MatchObjects.h"
#include "QueryHit.h"
#include "ShellIcons.h"
#include "CoolInterface.h"
#include "VendorCache.h"
#include "Network.h"
#include "Schema.h"
#include "Skin.h"
#include "WndBaseMatch.h"

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

BEGIN_MESSAGE_MAP(CMatchCtrl, CWnd)
	ON_WM_CREATE()
	ON_WM_DESTROY()
	ON_WM_SIZE()
	ON_WM_ERASEBKGND()
	ON_WM_PAINT()
	ON_WM_VSCROLL()
	ON_WM_LBUTTONDOWN()
	ON_WM_MOUSEMOVE()
	ON_WM_LBUTTONUP()
	ON_WM_LBUTTONDBLCLK()
	ON_WM_KEYDOWN()
	ON_WM_RBUTTONDOWN()
	ON_WM_RBUTTONUP()
	ON_WM_SETCURSOR()
	ON_WM_MOUSEWHEEL()
	ON_WM_HSCROLL()
	ON_WM_TIMER()
	ON_NOTIFY(HDN_ITEMCHANGEDW, IDC_MATCH_HEADER, OnChangeHeader)
	ON_NOTIFY(HDN_ITEMCHANGEDA, IDC_MATCH_HEADER, OnChangeHeader)
	ON_NOTIFY(HDN_ENDDRAG, IDC_MATCH_HEADER, OnChangeHeader)
	ON_NOTIFY(HDN_ITEMCLICKW, IDC_MATCH_HEADER, OnClickHeader)
	ON_NOTIFY(HDN_ITEMCLICKA, IDC_MATCH_HEADER, OnClickHeader)
	ON_WM_SETFOCUS()
	ON_WM_KILLFOCUS()
END_MESSAGE_MAP()

#define HEADER_HEIGHT	20
#define ITEM_HEIGHT		17


/////////////////////////////////////////////////////////////////////////////
// CMatchCtrl construction

CMatchCtrl::CMatchCtrl()
{
	m_pMatches		= NULL;
	m_sType			= _T("Search");
	
	m_pSchema		= NULL;
	m_nTopIndex		= 0;
	m_nHitIndex		= 0;
	m_nBottomIndex	= 0xFFFFFFFF;
	m_nFocus		= 0xFFFFFFFF;
	m_nPageCount	= 1;
	m_nCurrentWidth	= 0;
	m_nMessage		= 0;
	m_bSearchLink	= FALSE;
	m_bTips			= TRUE;
}

CMatchCtrl::~CMatchCtrl()
{
}

/////////////////////////////////////////////////////////////////////////////
// CMatchCtrl system

BOOL CMatchCtrl::Create(CMatchList* pMatches, CWnd* pParentWnd)
{
	CRect rect( 0, 0, 0, 0 );
	m_pMatches = pMatches;
	DWORD dwStyle = WS_CHILD|WS_VSCROLL|WS_TABSTOP|WS_VISIBLE;
	return CWnd::Create( NULL, NULL, dwStyle, rect, pParentWnd, IDC_MATCHES, NULL );
}

int CMatchCtrl::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if ( CWnd::OnCreate( lpCreateStruct ) == -1 ) return -1;
	
	CRect rc;
	
	if ( ! m_wndHeader.Create( WS_CHILD|WS_VISIBLE|HDS_BUTTONS|HDS_DRAGDROP|HDS_HOTTRACK|HDS_FULLDRAG,
		rc, this, IDC_MATCH_HEADER ) ) return -1;
	m_wndHeader.SetFont( &theApp.m_gdiFont );
	
	if ( ! m_wndTip.Create( this ) ) return -1;
	
	EnableToolTips( TRUE );
	
	InsertColumn( MATCH_COL_NAME, _T("File"), HDF_LEFT, 200 );
	InsertColumn( MATCH_COL_TYPE, _T("Type"), HDF_CENTER, 40 );
	InsertColumn( MATCH_COL_SIZE, _T("Size"), HDF_CENTER, 60 );
	InsertColumn( MATCH_COL_RATING, _T("Rating"), HDF_CENTER, 12*5 );
	InsertColumn( MATCH_COL_STATUS, _T("Status"), HDF_CENTER, 16*3 );
	InsertColumn( MATCH_COL_COUNT, _T("Host/Count"), HDF_CENTER, 120 );
	InsertColumn( MATCH_COL_SPEED, _T("Speed"), HDF_CENTER, 60 );
	InsertColumn( MATCH_COL_CLIENT, _T("Client"), HDF_CENTER, 80 );
	
	Skin.Translate( _T("CMatchCtrl"), &m_wndHeader );
	
	CBitmap bmStar;
	bmStar.LoadBitmap( IDB_SMALL_STAR );
	if ( ! m_pStars.Create( 12, 12, ILC_COLOR16|ILC_MASK, 7, 0 ) )
		m_pStars.Create( 12, 12, ILC_COLOR24|ILC_MASK, 7, 0 );
	m_pStars.Add( &bmStar, RGB( 0, 255, 0 ) );
	
	LoadColumnState();
	
	UpdateScroll();
	
	return 0;
}

void CMatchCtrl::OnDestroy() 
{
	m_wndTip.DestroyWindow();

	SaveColumnState();
	CWnd::OnDestroy();
}

void CMatchCtrl::OnSize(UINT nType, int cx, int cy) 
{
	CWnd::OnSize( nType, cx, cy );
	
	m_nCurrentWidth = cx;

	m_nPageCount = ( cy - HEADER_HEIGHT ) / ITEM_HEIGHT;
	if ( m_nPageCount < 1 ) m_nPageCount = 1;
	m_nBottomIndex = 0xFFFFFFFF;

	UpdateScroll();
}

/////////////////////////////////////////////////////////////////////////////
// CMatchCtrl update

void CMatchCtrl::Update()
{
	CSingleLock pLock( &m_pMatches->m_pSection, TRUE );

	if ( ! m_pMatches->m_bUpdated ) return;

	m_nCacheItems = m_pMatches->m_nItems;
	ScrollTo( GetScrollPos( SB_VERT ) );

	if ( m_pMatches->m_nUpdateMax >= m_nTopIndex &&
		 m_pMatches->m_nUpdateMin <= m_nBottomIndex )
	{
		Invalidate();
	}
	
	m_pMatches->ClearUpdated();
}

void CMatchCtrl::DestructiveUpdate()
{
	m_wndTip.Hide();
}

void CMatchCtrl::SelectSchema(CSchema* pSchema, CPtrList* pColumns)
{
	SaveColumnState();
	
	m_pSchema = pSchema;
	m_pColumns.RemoveAll();
	
	while ( m_wndHeader.DeleteItem( MATCH_COL_MAX ) );
	int nColumn = MATCH_COL_MAX;
	
	if ( pSchema && pColumns )
	{
		m_pColumns.AddTail( pColumns );
		
		for ( POSITION pos = m_pColumns.GetHeadPosition() ; pos ; nColumn++ )
		{
			CSchemaMember* pMember = (CSchemaMember*)m_pColumns.GetNext( pos );
			InsertColumn( nColumn, pMember->m_sTitle, pMember->m_nColumnAlign, pMember->m_nColumnWidth );
		}
	}
	
	m_pMatches->SelectSchema( pSchema, pColumns );
	
	LoadColumnState();
	
	Update();
}

void CMatchCtrl::SetBrowseMode()
{
	SaveColumnState();
	m_sType = _T("Browse");
	HDITEM pZero = { HDI_WIDTH, 0 };
	m_wndHeader.SetItem( MATCH_COL_STATUS, &pZero );
	m_wndHeader.SetItem( MATCH_COL_COUNT, &pZero );
	m_wndHeader.SetItem( MATCH_COL_SPEED, &pZero );
	m_wndHeader.SetItem( MATCH_COL_CLIENT, &pZero );
	LoadColumnState();
}

BOOL CMatchCtrl::HitTestHeader(const CPoint& point)
{
	CRect rc;
	m_wndHeader.GetWindowRect( &rc );
	return rc.PtInRect( point );
}

void CMatchCtrl::SetSortColumn(int nColumn, BOOL bDirection)
{
	CSingleLock pLock( &m_pMatches->m_pSection, TRUE );
	CWaitCursor pCursor;
	
	m_pMatches->SetSortColumn( nColumn, bDirection );
	Update();
	
	if ( m_bmSortAsc.m_hObject == NULL )
	{
		m_bmSortAsc.LoadMappedBitmap( IDB_SORT_ASC );
		m_bmSortDesc.LoadMappedBitmap( IDB_SORT_DESC );
	}
	
	HDITEM pColumn = { HDI_BITMAP|HDI_FORMAT };
	
	for ( int nCol = 0 ; m_wndHeader.GetItem( nCol, &pColumn ) ; nCol++ )
	{
		if ( nCol == nColumn )
		{
			pColumn.fmt |= HDF_BITMAP|HDF_BITMAP_ON_RIGHT;
			pColumn.hbm = (HBITMAP)( bDirection ? m_bmSortAsc.GetSafeHandle() : m_bmSortDesc.GetSafeHandle() );
		}
		else
		{
			pColumn.fmt &= ~HDF_BITMAP;
			pColumn.hbm = NULL;
		}
		
		m_wndHeader.SetItem( nCol, &pColumn );
	}
}

void CMatchCtrl::SetMessage(UINT nMessageID, BOOL bLink)
{
	if ( nMessageID == m_nMessage && m_bSearchLink == bLink ) return;
	
	m_bSearchLink = bLink;
	
	if ( m_nMessage = nMessageID )
		Skin.LoadString( m_sMessage, m_nMessage );
	else
		m_sMessage.Empty();
	
	if ( m_nCacheItems == 0 ) Invalidate();
}

void CMatchCtrl::SetMessage(LPCTSTR pszMessage, BOOL bLink)
{
	if ( m_sMessage == pszMessage && m_bSearchLink == bLink ) return;
	
	m_bSearchLink = bLink;
	m_nMessage = 0;
	m_sMessage = pszMessage;
	
	if ( m_nCacheItems == 0 ) Invalidate();
}

void CMatchCtrl::EnableTips(BOOL bTips)
{
	m_bTips = bTips;
}