matchobjects.h

p2p软件

//
// MatchObjects.h
//
// 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
//

#if !defined(AFX_MATCHOBJECTS_H__3D4FE6DE_027F_44B9_A70E_A229D970D7A8__INCLUDED_)
#define AFX_MATCHOBJECTS_H__3D4FE6DE_027F_44B9_A70E_A229D970D7A8__INCLUDED_

#pragma once

class CSchema;
class CSchemaMember;
class CQuerySearch;
class CQueryHit;
class CMatchFile;


class CMatchList
{
// Construction
public:
	CMatchList();
	virtual ~CMatchList();
	
// Attributes
public:
	CMutex		m_pSection;
public:
	CString		m_sFilter;
	BOOL		m_bFilterBusy;
	BOOL		m_bFilterPush;
	BOOL		m_bFilterUnstable;
	BOOL		m_bFilterReject;
	BOOL		m_bFilterLocal;
	BOOL		m_bFilterBogus;
	QWORD		m_nFilterMinSize;
	QWORD		m_nFilterMaxSize;
	DWORD		m_nFilterSources;
	int			m_nSortColumn;
	BOOL		m_bSortDir;
	CSchema*	m_pSchema;
	BOOL		m_bNew;
public:
	CMatchFile**	m_pFiles;
	DWORD			m_nFiles;
	DWORD			m_nItems;
	DWORD			m_nFilteredFiles;
	DWORD			m_nFilteredHits;
	BOOL			m_bUpdated;
	DWORD			m_nUpdateMin;
	DWORD			m_nUpdateMax;
	CPtrList		m_pSelectedFiles;
	CPtrList		m_pSelectedHits;
protected:
	DWORD			m_nBuffer;
	CMatchFile**	m_pSizeMap;
	CMatchFile**	m_pMapSHA1;
	CMatchFile**	m_pMapTiger;
	CMatchFile**	m_pMapED2K;
	LPTSTR			m_pszFilter;
	CSchemaMember**	m_pColumns;
	int				m_nColumns;
	
// Operations
public:
	void		AddHits(CQueryHit* pHits, CQuerySearch* pFilter = NULL, BOOL bRequire = FALSE);
	DWORD		FileToItem(CMatchFile* pFile);
	void		Clear();
	BOOL		Select(CMatchFile* pFile, CQueryHit* pHit, BOOL bSelected = TRUE);
	CMatchFile*	GetSelectedFile(BOOL bFromHit = FALSE) const;
	CQueryHit*	GetSelectedHit() const;
	int			GetSelectedCount() const;
	BOOL		ClearSelection();
	void		Filter();
	void		SelectSchema(CSchema* pSchema, CPtrList* pColumns);
	void		SetSortColumn(int nColumn = -1, BOOL bDirection = FALSE);
	void		UpdateRange(DWORD nMin = 0, DWORD nMax = 0xFFFFFFFF);
	void		ClearUpdated();
	void		ClearNew();
	void		Serialize(CArchive& ar);
protected:
	void		InsertSorted(CMatchFile* pFile);
	BOOL		FilterHit(CQueryHit* pHit);
	
	friend class CMatchFile;

};


class CMatchFile
{
// Construction
public:
	CMatchFile(CMatchList* pList, CQueryHit* pHit = NULL);
	virtual ~CMatchFile();
	
// Attributes
public:
	CMatchList*	m_pList;
	CQueryHit*	m_pHits;
	CQueryHit*	m_pBest;
	DWORD		m_nTotal;
	DWORD		m_nFiltered;
	DWORD		m_nSources;
	CMatchFile*	m_pNextSize;
	CMatchFile*	m_pNextSHA1;
	CMatchFile*	m_pNextTiger;
	CMatchFile*	m_pNextED2K;
public:
	BOOL		m_bSHA1;
	SHA1		m_pSHA1;
	BOOL		m_bTiger;
	TIGEROOT	m_pTiger;
	BOOL		m_bED2K;
	MD4			m_pED2K;
	QWORD		m_nSize;
	CString		m_sSize;
public:
	TRISTATE	m_bBusy;
	TRISTATE	m_bPush;
	TRISTATE	m_bStable;
	BOOL		m_bPreview;
	DWORD		m_nSpeed;
	CString		m_sSpeed;
	int			m_nRating;
	int			m_nRated;
	BOOL		m_bDRM;
	BOOL		m_bCollection;
public:
	BOOL		m_bExpanded;
	BOOL		m_bSelected;
	BOOL		m_bExisting;
	BOOL		m_bDownload;
	BOOL		m_bNew;
	BOOL		m_bOneValid;
	int			m_nShellIndex;
public:
	CString*	m_pColumns;
	int			m_nColumns;
	BYTE*		m_pPreview;
	DWORD		m_nPreview;
	
// Operations
public:
	BOOL		Add(CQueryHit* pHit, BOOL bForce = FALSE);
	BOOL		Check(CQueryHit* pHit) const;
	BOOL		Expand(BOOL bExpand = TRUE);
	inline int	Compare(CMatchFile* pFile) const;
	CString		GetURN() const;
	void		Serialize(CArchive& ar, int nVersion);
protected:
	inline DWORD	Filter();
	inline void		Added(CQueryHit* pHit);
	inline void		ClearNew();
public:
	
	inline DWORD GetFilteredCount() const
	{
		if ( m_pList->m_bFilterLocal && m_bExisting ) return 0;
		return m_nSources >= m_pList->m_nFilterSources ? m_nFiltered : 0;
	}
	
	inline DWORD GetItemCount() const
	{
		if ( m_pList->m_bFilterLocal && m_bExisting ) return 0;
		if ( m_nSources < m_pList->m_nFilterSources ) return 0;

		if ( m_nFiltered == 1 || ! m_bExpanded )
			return 1;
		else
			return m_nFiltered + 1;
	}
	
	inline int GetRating() const
	{
		int nRating = 0;
		
		if ( m_bPush != TS_TRUE ) nRating += 4;
		if ( m_bBusy != TS_TRUE ) nRating += 2;
		if ( m_bStable == TS_TRUE ) nRating ++;

		return nRating;
	}
	
	friend class CMatchList;
};


#endif // !defined(AFX_MATCHOBJECTS_H__3D4FE6DE_027F_44B9_A70E_A229D970D7A8__INCLUDED_)