WndTraffic.cpp

Go to the documentation of this file.

//
// WndTraffic.cpp
//
// Copyright (c) Shareaza Development Team, 2002-2005.
// 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)
{
        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 );
}

---