#include "stdafx.h"
#include <math.h>
#include "LineChartCtrl.h"
#include "MemDC.h"
CLine::CLine(int iStyle, COLORREF color, int iThick)
{
m_bDraw = TRUE;
m_iStyle = iStyle;
m_color = color;
m_iThick = iThick;
InitializeCriticalSection ( & g_cs ) ;
}
CLine::CLine(const CLine &line)
{
m_bDraw = line.m_bDraw;
m_iStyle = line.m_iStyle;
m_color = line.m_color;
m_iThick = line.m_iThick;
m_dXValue.clear();
m_dXValue=line.m_dXValue;
m_dYValue.clear();
m_dYValue=line.m_dYValue;
InitializeCriticalSection ( & g_cs ) ;
}
CLine& CLine::operator=(const CLine& line)
{
m_bDraw = line.m_bDraw;
m_iStyle = line.m_iStyle;
m_color = line.m_color;
m_iThick = line.m_iThick;
m_dXValue.clear();
m_dXValue=line.m_dXValue;
m_dYValue.clear();
m_dYValue=line.m_dYValue;
return *this;
}
CLine::~CLine()
{
DeleteCriticalSection ( & g_cs ) ;
}
void CLine::AddPoint(double x, double y)
{
EnterCriticalSection ( & g_cs ) ;
m_dXValue.push_back(x);
m_dYValue.push_back(y);
LeaveCriticalSection ( & g_cs ) ;
}
double CLine::GetPointX(int nIndex)
{
return m_dXValue[nIndex];
}
double CLine::GetPointY(int nIndex)
{
return m_dYValue[nIndex];
}
int CLine::GetPointCount()
{
return static_cast<int>(m_dXValue.size());
}
void CLine::RemoveAllPoints()
{
EnterCriticalSection ( & g_cs ) ;
m_dXValue.clear();
m_dYValue.clear();
LeaveCriticalSection ( & g_cs ) ;
}
void CLine::RemoveUselessPoints()
{
EnterCriticalSection ( & g_cs ) ;
for(int i=0;i<MAX_POINTS;i++)
{
vector<double>::iterator itx = m_dXValue.begin();
m_dXValue.erase(itx);
vector<double>::iterator ity = m_dYValue.begin();
m_dYValue.erase(ity);
}
LeaveCriticalSection ( & g_cs ) ;
}
IMPLEMENT_DYNAMIC(CLineChartCtrl, CStatic)
CLineChartCtrl::CLineChartCtrl()
{
m_clrBkColor = RGB(0,0,0);
m_nRate = 500;
m_iMarginLeft = 5;
m_iMarginRight = 5;
m_iMarginTop = 15;
m_iMarginBottom = 15;
}
CLineChartCtrl::~CLineChartCtrl()
{
}
BEGIN_MESSAGE_MAP(CLineChartCtrl, CStatic)
ON_WM_TIMER()
ON_WM_PAINT()
ON_WM_ERASEBKGND()
ON_WM_SIZE()
END_MESSAGE_MAP()
void CLineChartCtrl::OnTimer(UINT_PTR nIDEvent)
{
Invalidate(TRUE);
CStatic::OnTimer(nIDEvent);
}
void CLineChartCtrl::OnPaint()
{
CPaintDC dc(this);
CRect rectCtrl;
GetClientRect (&rectCtrl);
CMyMemDC memDC(&dc, &rectCtrl);
DrawAll(&memDC);
}
BOOL CLineChartCtrl::OnEraseBkgnd(CDC* pDC)
{
return FALSE;
}
void CLineChartCtrl::DrawAll(CDC* pDC)
{
DrawBackground(pDC);
DrawAxises(pDC);
DrawLines(pDC);
}
void CLineChartCtrl::DrawBackground(CDC* pDC)
{
pDC->FillSolidRect ( & m_rectCtrl , m_clrBkColor ) ;
}
void CLineChartCtrl::DrawAxises(CDC* pDC)
{
if(m_axisY.m_bDraw)
{
CPen pen(m_axisY.m_iStyle, m_axisY.m_iThick, m_axisY.m_color);
CPen *oldPen = pDC->SelectObject(&pen);
pDC->MoveTo(m_rectPlot.left, m_rectPlot.top);
pDC->LineTo(m_rectPlot.right, m_rectPlot.top);
pDC->LineTo(m_rectPlot.right, m_rectPlot.bottom);
pDC->LineTo(m_rectPlot.left, m_rectPlot.bottom);
pDC->LineTo(m_rectPlot.left, m_rectPlot.top);
int iGridCount = 5;
int iGrid =(int)floor((m_axisY.m_dMaxValue-m_axisY.m_dMinValue)/iGridCount);
if(iGrid>0)
{
CString str;
int textHeight =16;
pDC->SetTextColor(m_axisY.m_color);
int y0=(int)m_axisY.m_dMinValue;
for(int i=0;i<iGridCount+1;i++)
{
int y = (int)(m_rectPlot.bottom - (y0+i*iGrid-m_axisY.m_dMinValue)/m_axisY.m_dValuePerPixel);
str.Format(_T("%d"), y0+i*iGrid);
pDC->DrawText(str, CRect(m_rectAxisY.left, y-textHeight/2,m_rectAxisY.right-5,y+textHeight/2), DT_RIGHT|DT_BOTTOM);
if(m_axisY.m_bGrid)
{
pDC->MoveTo(m_rectPlot.left,y);
pDC->LineTo(m_rectPlot.right,y);
}
}
}
pDC->SelectObject(oldPen);
}
if(m_axisX.m_bDraw)
{
if(m_axisX.m_bGrid)
{
}
}
}
void CLineChartCtrl::DrawLines(CDC *pDC)
{
for(int i=0;i<GetLineCount();i++)
{
CLine& line = GetLine(i);
if(!line.m_bDraw)continue;
int iPointCount = line.GetPointCount();
if(iPointCount<=0)return;
CPen pen(line.m_iStyle, line.m_iThick, line.m_color);
CPen *oldPen = pDC->SelectObject(&pen);
int dataPointStart=0;
if(iPointCount>m_rectPlot.Width())
{
dataPointStart = iPointCount-m_rectPlot.Width();
}
if(iPointCount>MAX_POINTS)
{
dataPointStart = iPointCount-MAX_POINTS;
}
for(int k=dataPointStart;k<iPointCount;k++)
{
int x = m_rectPlot.right-(iPointCount-k);
int y = (int)(m_rectPlot.bottom - ((line.GetPointY(k)-m_axisY.m_dMinValue)/m_axisY.m_dValuePerPixel));
m_ptInLine[k-dataPointStart].x = x;
m_ptInLine[k-dataPointStart].y = y;
}
pDC->Polyline(m_ptInLine, iPointCount-dataPointStart);
pDC->SelectObject(oldPen);
TRACE("point count =%d\n",iPointCount);
if(iPointCount>2*MAX_POINTS)
{
line.RemoveUselessPoints();
TRACE("Too many points, removing...\n");
}
}
}
void CLineChartCtrl::Start()
{
if(m_nRate > 0)
{
SetTimer(1,m_nRate,NULL);
}
}
void CLineChartCtrl::Stop()
{
KillTimer(1);
}
void CLineChartCtrl::CalcLayout()
{
GetClientRect(&m_rectCtrl);
int rectAxisYWidth = 50;
m_rectAxisY.left = m_rectCtrl.left + m_iMarginLeft;
m_rectAxisY.right = m_rectAxisY.left + rectAxisYWidth;
m_rectAxisY.top = m_rectCtrl.top + m_iMarginTop;
m_rectAxisY.bottom = m_rectCtrl.bottom - m_iMarginBottom;
m_rectAxisX.left = m_rectAxisY.right;
m_rectAxisX.right = m_rectCtrl.right-m_iMarginRight;
m_rectAxisX.bottom = m_rectCtrl.bottom - m_iMarginBottom;
m_rectAxisX.top = m_rectAxisX.bottom -20;
m_rectPlot.left = m_rectAxisY.right;
m_rectPlot.right = m_rectCtrl.right-m_iMarginRight;
m_rectPlot.top = m_rectCtrl.top + m_iMarginTop;
m_rectPlot.bottom = m_rectAxisX.top;
m_axisY.m_dValuePerPixel = (m_axisY.m_dMaxValue- m_axisY.m_dMinValue) / m_rectPlot.Height();
TRACE("m_axisY.m_dValuePerPixel=%.1f\n",m_axisY.m_dValuePerPixel);
}
void CLineChartCtrl::AddLine(int iStyle, COLORREF color, int iThick)
{
m_lines.push_back(CLine(iStyle,color,iThick));
}
CLine& CLineChartCtrl::GetLine(int nIndex)
{
return m_lines[nIndex];
}
int CLineChartCtrl::GetLineCount()
{
return static_cast<int>(m_lines.size());
}
void CLineChartCtrl::SetRate(int nRate)
{
m_nRate = nRate;
}
void CLineChartCtrl::SetBkColor(COLORREF clrBkColor)
{
m_clrBkColor = clrBkColor;
}
CAxis& CLineChartCtrl::GetAxisY()
{
return m_axisY;
}
CAxis& CLineChartCtrl::GetAxisX()
{
return m_axisX;
}
void CLineChartCtrl::OnSize(UINT nType, int cx, int cy)
{
CStatic::OnSize(nType, cx, cy);
CalcLayout();
Invalidate();
}
