wndtraffic.cpp

著名的下载软件核心Shareaza

//
// WndTraffic.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 "WndTraffic.h"
#include "WndMain.h"
#include "DlgGraphList.h"
#include "GraphLine.h"

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

IMPLEMENT_SERIAL(CTrafficWnd, CChildWnd, 0)

BEGIN_MESSAGE_MAP(CTrafficWnd, CChildWnd)
	//{{AFX_MSG_MAP(CTrafficWnd)
	ON_WM_CREATE()
	ON_WM_DESTROY()
	ON_WM_PAINT()
	ON_WM_TIMER()
	ON_WM_CONTEXTMENU()
	ON_UPDATE_COMMAND_UI(ID_TRAFFIC_GRID, OnUpdateTrafficGrid)
	ON_COMMAND(ID_TRAFFIC_GRID, OnTrafficGrid)
	ON_UPDATE_COMMAND_UI(ID_TRAFFIC_AXIS, OnUpdateTrafficAxis)
	ON_COMMAND(ID_TRAFFIC_AXIS, OnTrafficAxis)
	ON_UPDATE_COMMAND_UI(ID_TRAFFIC_LEGEND, OnUpdateTrafficLegend)
	ON_COMMAND(ID_TRAFFIC_LEGEND, OnTrafficLegend)
	ON_COMMAND(ID_TRAFFIC_SETUP, OnTrafficSetup)
	ON_COMMAND(ID_TRAFFIC_CLEAR, OnTrafficClear)
	ON_WM_LBUTTONDBLCLK()
	ON_COMMAND(ID_TRAFFIC_WINDOW, OnTrafficWindow)
	ON_WM_SIZE()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()


/////////////////////////////////////////////////////////////////////////////
// CTrafficWnd construction

CTrafficWnd::CTrafficWnd(DWORD nUnique)
{
	m_nUnique		= nUnique;
	m_pGraph		= new CLineGraph();
	
	Create( IDR_TRAFFICFRAME );
}

CTrafficWnd::~CTrafficWnd()
{
	if ( m_pGraph ) delete m_pGraph;
}

/////////////////////////////////////////////////////////////////////////////
// CTrafficWnd message handlers

int CTrafficWnd::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if ( CChildWnd::OnCreate( lpCreateStruct ) == -1 ) return -1;
	
	if ( m_nUnique == 0 ) FindFreeUnique();
	
	if ( ! Serialize( FALSE ) && m_nUnique == 1 ) m_pGraph->CreateDefaults();
	
	UpdateCaption();
	
	return 0;
}

void CTrafficWnd::OnDestroy() 
{
	KillTimer( 2 );
	Serialize( TRUE );

	CChildWnd::OnDestroy();
}

void CTrafficWnd::FindFreeUnique()
{
	CWindowManager*	pWindows = GetManager();

	for ( m_nUnique = 1 ; m_nUnique < 1000 ; m_nUnique++ )
	{
		CTrafficWnd* pChild = NULL;
		BOOL bAvailable		= TRUE;

		while ( pChild = (CTrafficWnd*)pWindows->Find( RUNTIME_CLASS(CTrafficWnd), pChild ) )
		{
			if ( pChild != this && pChild->m_nUnique == m_nUnique )
			{
				bAvailable = FALSE;
				break;
			}
		}

		if ( bAvailable ) break;
	}
}

void CTrafficWnd::SetUpdateRate()
{
	KillTimer( 2 );
	
	if ( TRUE )
	{
		CLineGraph* pGraph = (CLineGraph*)m_pGraph;
		SetTimer( 2, pGraph->m_nSpeed, NULL );
	}
}

void CTrafficWnd::OnSize(UINT nType, int cx, int cy) 
{
	CChildWnd::OnSize( nType, cx, cy );
	Invalidate();	
}

void CTrafficWnd::OnPaint() 
{
	CPaintDC dc( this );
	CRect rc;

	GetClientRect( &rc );
	m_pGraph->BufferedPaint( &dc, &rc );
}

void CTrafficWnd::OnTimer(UINT nIDEvent) 
{
	DWORD nNow = GetTickCount();
	
	if ( nIDEvent == 2 && m_pGraph->Update() ) Invalidate();
}

void CTrafficWnd::OnContextMenu(CWnd* pWnd, CPoint point) 
{
	TrackPopupMenu( _T("CTrafficWnd"), point, ID_TRAFFIC_SETUP );
}

void CTrafficWnd::OnUpdateTrafficGrid(CCmdUI* pCmdUI) 
{
	CLineGraph* pLineGraph = (CLineGraph*)m_pGraph;
	pCmdUI->SetCheck( pLineGraph->m_bShowGrid );
}

void CTrafficWnd::OnTrafficGrid() 
{
	CLineGraph* pLineGraph = (CLineGraph*)m_pGraph;
	pLineGraph->m_bShowGrid = ! pLineGraph->m_bShowGrid;
	Invalidate();
}

void CTrafficWnd::OnUpdateTrafficAxis(CCmdUI* pCmdUI) 
{
	CLineGraph* pLineGraph = (CLineGraph*)m_pGraph;
	pCmdUI->SetCheck( pLineGraph->m_bShowAxis );
}

void CTrafficWnd::OnTrafficAxis() 
{
	CLineGraph* pLineGraph = (CLineGraph*)m_pGraph;
	pLineGraph->m_bShowAxis = ! pLineGraph->m_bShowAxis;
	Invalidate();
}

void CTrafficWnd::OnUpdateTrafficLegend(CCmdUI* pCmdUI) 
{
	CLineGraph* pLineGraph = (CLineGraph*)m_pGraph;
	pCmdUI->SetCheck( pLineGraph->m_bShowLegend );
}

void CTrafficWnd::OnTrafficLegend() 
{
	CLineGraph* pLineGraph = (CLineGraph*)m_pGraph;
	pLineGraph->m_bShowLegend = ! pLineGraph->m_bShowLegend;
	Invalidate();
}

void CTrafficWnd::OnTrafficSetup() 
{
	CGraphListDlg dlg( this, (CLineGraph*)m_pGraph );	
	
	dlg.m_sName = m_sName;
	
	if ( dlg.DoModal() == IDOK )
	{
		m_sName = dlg.m_sName;
		
		SetUpdateRate();
		UpdateCaption();
		Invalidate();
	}
}

void CTrafficWnd::OnTrafficClear() 
{
	m_pGraph->Clear();
}

void CTrafficWnd::OnLButtonDblClk(UINT nFlags, CPoint point) 
{
	PostMessage( WM_COMMAND, ID_TRAFFIC_SETUP );
}

void CTrafficWnd::OnTrafficWindow() 
{
	new CTrafficWnd();
}

/////////////////////////////////////////////////////////////////////////////
// CTrafficWnd serialize

BOOL CTrafficWnd::Serialize(BOOL bSave)
{
	WINDOWPLACEMENT pPos = { sizeof(WINDOWPLACEMENT) };
	CString strFile;
	CFile pFile;
	
	strFile.Format( _T("%s\\Data\\Graph%.4i.dat"), (LPCTSTR)Settings.General.Path, m_nUnique );
	
	if ( ! pFile.Open( strFile, bSave ? ( CFile::modeWrite | CFile::modeCreate ) : CFile::modeRead ) )
		return FALSE;
	
	CArchive ar( &pFile, bSave ? CArchive::store : CArchive::load );
	int nVersion = 0;
	
	if ( ar.IsStoring() )
	{
		nVersion = 0xFFFFFFFF;
		ar << nVersion;
		nVersion = 1;
		ar << nVersion;
		
		ar << m_nUnique;
		ar << m_sName;
		
		GetWindowPlacement( &pPos );
		ar.Write( &pPos, sizeof(pPos) );
	}
	else
	{
		ar >> m_nUnique;
		
		if ( m_nUnique == 0xFFFFFFFF )
		{
			ar >> nVersion;
			ar >> m_nUnique;
		}
		
		if ( nVersion >= 1 ) ar >> m_sName;
		
		ar.Read( &pPos, sizeof(pPos) );
		if ( pPos.showCmd == SW_SHOWNORMAL )
			SetWindowPlacement( &pPos );
	}
	
	m_pGraph->Serialize( ar );
	
	ar.Close();
	pFile.Close();
	
	if ( ! bSave ) SetUpdateRate();
	
	return TRUE;
}

void CTrafficWnd::UpdateCaption()
{
	CString strCaption, strName;
	
	if ( m_sName.GetLength() )
	{
		strName = _T(" : ") + m_sName;
	}
	else
	{
		if ( m_nUnique <= 26 )
			strName.Format( _T(" (%c)"), 'A' + m_nUnique - 1 );
		else
			strName.Format( _T(" (%lu)"), m_nUnique );
	}
	
	LoadString( strCaption, IDR_TRAFFICFRAME );
	strCaption += strName;
	SetWindowText( strCaption );
}