RobotView.cpp

// RobotView.cpp : implementation of the CRobotView class
//

#include "stdafx.h"
#include "Robot.h"

#include "RobotDoc.h"
#include "RobotView.h"
#include "MainFrm.h"

#include "SetPara.h"
#include "MyGA.h"
#include "string.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif


/////////////////////////////////////////////////////////////////////////////
// CRobotView

IMPLEMENT_DYNCREATE(CRobotView, CView)

BEGIN_MESSAGE_MAP(CRobotView, CView)
	//{{AFX_MSG_MAP(CRobotView)
	ON_WM_SIZE()
	ON_COMMAND(ID_BTSTART, OnBtstart)
	ON_COMMAND(ID_BTERASE, OnBterase)
	ON_COMMAND(ID_BTADD, OnBtadd)
	ON_COMMAND(ID_BTPARA, OnBtpara)
	ON_COMMAND(ID_MENUADD, OnMenuadd)
	ON_COMMAND(ID_MENUERASE, OnMenuerase)
	ON_COMMAND(ID_MENUPARA, OnMenupara)
	ON_COMMAND(ID_MENUSTART, OnMenustart)
	ON_WM_LBUTTONDOWN()
	ON_WM_CANCELMODE()
	ON_WM_CREATE()
	ON_COMMAND(ID_BTPAUSE, OnBtpause)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CRobotView construction/destruction
extern struct Fitness fit[1000];
CRobotView::CRobotView()
{
	// TODO: add construction code here
	m_origX = m_origY = 20; //画图的起点
	m_bEraseBlock = m_bAddBlock = m_bSetPara = false;
	m_bNeedInit = true;
	m_CurOpID = ID_APP_ABOUT;
	m_preState = &m_bEraseBlock;
	m_pThread = NULL;
}
CRobotView::~CRobotView()
{
}

BOOL CRobotView::PreCreateWindow(CREATESTRUCT& cs)
{
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs
	return CView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CRobotView drawing

void CRobotView::OnDraw(CDC* pDC)
{
	CPoint   point;
	
	CRobotDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	// TODO: add draw code for native data here
	
	DrawChart(pDC,m_origX,m_origY);
	DrawBlocks(pDC,m_origX,m_origY);
	ShowPara(pDC);

	BYTE red;
	BYTE green;
	BYTE blue;
	COLORREF m_PathColor;
	if(pDoc->GARoad->m_bAlreadyRun){

		for(int i=0;i<pDoc->GARoad->popSize;i++){

			red = (20*i)%256;
			green = 0;//(50*i)%256;
			blue = (80*i)%256;
			m_PathColor = RGB(red,green,blue);
			DrawPath(pDC, &(pDoc->GARoad->parent_pop.ind[i].xPath), m_PathColor);
		}
	}

	if(pDoc->GARoad->m_bAlreadyStarted){

		red = 255;
		green = 0;
		blue = 0;
		m_PathColor = RGB(red,green,blue);
		DrawPath(pDC, &(pDoc->GARoad->path1), m_PathColor);//NULL  pDoc->GARoad->curBestGenome->genVec

		red = 0;
		green = 0;
		blue = 255;
		m_PathColor = RGB(red,green,blue);
		DrawPath(pDC, &(pDoc->GARoad->path2), m_PathColor);

		red = 0;
		green = 255;
		blue = 0;
		m_PathColor = RGB(red,green,blue);
		DrawPath(pDC, &(pDoc->GARoad->path3), m_PathColor);
	}
}

/////////////////////////////////////////////////////////////////////////////
// CRobotView diagnostics

#ifdef _DEBUG
void CRobotView::AssertValid() const
{
	CView::AssertValid();
}

void CRobotView::Dump(CDumpContext& dc) const
{
	CView::Dump(dc);
}

CRobotDoc* CRobotView::GetDocument() // non-debug version is inline
{
	ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CRobotDoc)));
	return (CRobotDoc*)m_pDocument;
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CRobotView message handlers
void CRobotView::DrawChart(CDC* pDC,int originX,int originY)
{
	CRobotDoc* pDoc = GetDocument();
	int chartWidth = m_CurPara.width;
	int chartHeight = m_CurPara.height;
	gridWidth = min(m_cxWnd/chartWidth,(int)(m_cyWnd*7/8/chartHeight));
	
	CPen m_ChartPen1(PS_SOLID,1,RGB(100,100,100));//Line color
	CPen* m_OldPen = pDC->SelectObject (&m_ChartPen1);

	//V-line
	for (int i = 0; i <= chartHeight; i++){

		int x0 = originX;
		int x1 = x0 + gridWidth*chartWidth ;
		int y = originY + gridWidth*i;
		pDC->MoveTo(x0,y);
		pDC->LineTo(x1,y);
		
	}

	if(m_CurPara.smooth)
	{
		COLORREF oldColor = pDC->SetTextColor(RGB(0,0,255));
		pDC->MoveTo(chartWidth*gridWidth+250,originY);//Y
		pDC->LineTo(chartWidth*gridWidth+250,originY+chartHeight*gridWidth);
		pDC->MoveTo(chartWidth*gridWidth+250,originY+15);
		pDC->LineTo(chartWidth*gridWidth+245,originY+25);
		pDC->MoveTo(chartWidth*gridWidth+250,originY+15);
		pDC->LineTo(chartWidth*gridWidth+255,originY+25);
		pDC->TextOut(chartWidth*gridWidth+250,originY,"Distance(Optimized)");
		pDC->MoveTo(chartWidth*gridWidth+250,originY+chartHeight*gridWidth);//X
		pDC->LineTo(chartWidth*gridWidth+250+chartHeight*gridWidth,originY+chartHeight*gridWidth);
		pDC->MoveTo(chartWidth*gridWidth+250+chartHeight*gridWidth,originY+chartHeight*gridWidth);
		pDC->LineTo(chartWidth*gridWidth+250+chartHeight*gridWidth-10,originY+chartHeight*gridWidth+5);
		pDC->MoveTo(chartWidth*gridWidth+250+chartHeight*gridWidth,originY+chartHeight*gridWidth);
		pDC->LineTo(chartWidth*gridWidth+250+chartHeight*gridWidth-10,originY+chartHeight*gridWidth-5);
		pDC->TextOut(chartWidth*gridWidth+155+chartHeight*gridWidth,originY+chartHeight*gridWidth+10,"Smooth(Optimized)");
		for(int i=0; i<m_CurPara.pSize;i++)
			pDC->TextOut(chartWidth*gridWidth+250+fit[i].B*250,originY+chartHeight*gridWidth-fit[i].A*250,"*");//Pareto
	}
	else if(m_CurPara.safe)
	{
		COLORREF oldColor = pDC->SetTextColor(RGB(0,0,255));
		pDC->MoveTo(chartWidth*gridWidth+250,originY);// Y
		pDC->LineTo(chartWidth*gridWidth+250,originY+chartHeight*gridWidth);
		pDC->MoveTo(chartWidth*gridWidth+250,originY+15);
		pDC->LineTo(chartWidth*gridWidth+245,originY+25);
		pDC->MoveTo(chartWidth*gridWidth+250,originY+15);
		pDC->LineTo(chartWidth*gridWidth+255,originY+25);
		pDC->TextOut(chartWidth*gridWidth+250,originY,"Distance(Optimized)");
		pDC->MoveTo(chartWidth*gridWidth+250,originY+chartHeight*gridWidth);// X
		pDC->LineTo(chartWidth*gridWidth+250+chartHeight*gridWidth,originY+chartHeight*gridWidth);
		pDC->MoveTo(chartWidth*gridWidth+250+chartHeight*gridWidth,originY+chartHeight*gridWidth);
		pDC->LineTo(chartWidth*gridWidth+250+chartHeight*gridWidth-10,originY+chartHeight*gridWidth+5);
		pDC->MoveTo(chartWidth*gridWidth+250+chartHeight*gridWidth,originY+chartHeight*gridWidth);pDC->LineTo(chartWidth*gridWidth+250+chartHeight*gridWidth-10,originY+chartHeight*gridWidth-5);
		pDC->TextOut(chartWidth*gridWidth+155+chartHeight*gridWidth,originY+chartHeight*gridWidth+10,"Safety(Optimized)");
		for(int i=0; i<m_CurPara.pSize;i++)
			pDC->TextOut(chartWidth*gridWidth+250+(1-fit[i].C)*250,originY+chartHeight*gridWidth-fit[i].A*250,"*");// Pareto
	}
	
	COLORREF oldColor = pDC->SetTextColor(RGB(100,100,100));//Text color
	// H-line
	CPen m_ChartPen(PS_SOLID,1,RGB(100,100,100));//Line color
	m_OldPen = pDC->SelectObject (&m_ChartPen);
	for (int i = 0; i <= chartWidth; i++)
	{
		int y0 = originY;
		int y1 = y0 + gridWidth*chartHeight;
		int x = originX + gridWidth*i;
		pDC->MoveTo(x,y0);
		pDC->LineTo(x,y1);
	}

	//Start and End point
	CFont font;
	LOGFONT lf;
	memset(&lf, 0, sizeof(LOGFONT));
	lf.lfHeight = gridWidth/2;
	strcpy(lf.lfFaceName, "Arial");
	VERIFY(font.CreateFontIndirect(&lf));
	CFont* oldFont = pDC->SelectObject(&font);
	
	int startX = originX+(gridWidth/12);
	int startY = originY+gridWidth/3;
	pDC->TextOut(startX,startY,"Start");
	int endX = originX+(chartWidth-1)*gridWidth+gridWidth/12;
	int endY = originY+(chartHeight-1)*gridWidth+gridWidth/3;
	pDC->TextOut(endX,endY,"End");

	pDC->SelectObject(m_OldPen);
	pDC->SelectObject(oldFont);
}

void CRobotView::OnSize(UINT nType, int cx, int cy) //调整图格大小以适合不同窗口大小
{
	CView::OnSize(nType, cx, cy);
	
	// TODO: Add your message handler code here
	m_cxWnd = cx;
	m_cyWnd = cy;
}

void CRobotView::DrawBlocks(CDC* pDC,int originX,int originY)
{
	CRobotDoc* pDoc = GetDocument();
	int width = m_CurPara.width;
	int height = m_CurPara.height;
	int upleftX,upleftY,lowrightX,lowrightY;
	CRect blockRect;
	CBrush m_BlockBrush;
	m_BlockBrush.CreateSolidBrush(RGB(0,0,0));
	CBrush* oldBrush = pDC->SelectObject(&m_BlockBrush);

	for (int i = 0; i < height; i++)
		for (int j = 0; j < width; j++)
		{
			if ( IsBlock(i,j) )
			{
				upleftX = originX+j*gridWidth;
				upleftY = originY+i*gridWidth;
				lowrightX = upleftX+gridWidth+1;
				lowrightY = upleftY+gridWidth+1;
				blockRect.SetRect(upleftX,upleftY,lowrightX,lowrightY);
				pDC->Rectangle(blockRect);
			}
		}

	pDC->SelectObject(oldBrush);
}

void CRobotView::DrawPath(CDC* pDC,vector<int>* path,COLORREF m_PathColor)
{
	if (path)//如果路径不为空则画路径
	{
		CRobotDoc* pDoc = GetDocument();
		int chartH,chartW;
		GetChartSize(chartH,chartW);
		
		CPen m_PathPen(PS_DASH,2,m_PathColor);
		CPen* m_OldPen = pDC->SelectObject(&m_PathPen);

		if (path->size()!=0 && chartH!=0 && chartW!=0)
		{
			int len = path->size();
			int curY,curX;
			curY = (*path)[0]/chartW;
			curX = (*path)[0]%chartW;
			curY = m_origY + curY*gridWidth + (gridWidth+1)/2;
			curX = m_origX + curX*gridWidth + (gridWidth+1)/2;
			pDC->MoveTo(curX,curY);

			for (int i = 1; i < len; i++)
			{
				curY = (*path)[i]/chartW;
				curX = (*path)[i]%chartW;
				curY = m_origY + curY*gridWidth + (gridWidth+1)/2;
				curX = m_origX + curX*gridWidth + (gridWidth+1)/2;
				pDC->LineTo(curX,curY);
				pDC->MoveTo(curX,curY);
			}
		}
		pDC->SelectObject(m_OldPen);
	}
}

UINT ThreadFunc(LPVOID pParam)
{
	CRobotView* m_CurView = (CRobotView*)pParam;
	CRobotDoc* pDoc = m_CurView->GetDocument();

	MyGA* m_Robot = pDoc->GARoad;
	if (m_Robot)
	{
		MyGA* m_Robot = pDoc->GARoad;
		m_Robot->gamain(m_CurView);
		m_Robot->Init(m_CurPara);
	}

	return 0;
}

void CRobotView::OnBtstart() 
{
	// TODO: Add your command handler code here
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,FALSE);
	*m_preState = false;

	m_CurOpID = ID_BTSTART;
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,TRUE);

	//if (pDoc->GARoad->m_bAlreadyStarted)
	{
		m_pThread = AfxBeginThread(ThreadFunc,this);
	}	
}

void CRobotView::OnBtpause() 
{
	// TODO: Add your command handler code here
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,FALSE);
	*m_preState = false;

	m_CurOpID = ID_BTPAUSE;
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,TRUE);
	if(m_pThread)
	{
		m_pThread->SuspendThread();

		//Update one last time when pause, in order to make it continous
		CRobotDoc* pDoc = GetDocument();
		MyGA* m_Robot = pDoc->GARoad;
		Invalidate();
		UpdateWindow();
	}	
}

void CRobotView::OnBterase() 
{
	// TODO: Add your command handler code here
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,FALSE);
	*m_preState = false;
	m_bEraseBlock = !m_bEraseBlock;
	m_preState = &m_bEraseBlock;
	m_CurOpID = ID_BTERASE;
	if (m_bEraseBlock)
		((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,TRUE);
	else
		((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,FALSE);
}

void CRobotView::OnBtadd() 
{
	// TODO: Add your command handler code here
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,FALSE);
	*m_preState = false;
	m_bAddBlock = !m_bAddBlock;
	m_preState = &m_bAddBlock;
	m_CurOpID = ID_BTADD;
	if (m_bAddBlock)
		((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,TRUE);
	else
		((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,FALSE);	
}

void CRobotView::OnBtpara() //Parameter settings
{
	// TODO: Add your command handler code here
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,FALSE);
	*m_preState = false;
	m_bSetPara = !m_bSetPara;
	m_preState = &m_bSetPara;
	m_CurOpID = ID_BTPARA;

	SetPara m_ChangePara;
	if (m_ChangePara.DoModal() == IDOK)
	{
		//Adjust size of the chart
		CRobotDoc* pDoc = GetDocument();
		AdjustChart(m_ChangePara.m_Height,m_ChangePara.m_Width);

		m_CurPara.height = m_ChangePara.m_Height;
		m_CurPara.width = m_ChangePara.m_Width;
		m_CurPara.propC = m_ChangePara.m_Cross;
		m_CurPara.propM = m_ChangePara.m_Mutation;
		m_CurPara.pSize = m_ChangePara.m_PopSize;
		m_CurPara.T = m_ChangePara.m_GenNum;
		m_CurPara.length = m_ChangePara.m_Length;
		m_CurPara.smooth = m_ChangePara.m_Smooth;
		m_CurPara.safe = m_ChangePara.m_Safe;
		
		Invalidate();
		UpdateWindow();
	}
}

void CRobotView::OnBtresult()
{
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,FALSE);
	*m_preState = false;
	m_CurOpID = ID_BTPARA;
	((CMainFrame*)::AfxGetMainWnd())->checkButton(m_CurOpID,TRUE);
	Invalidate();
	UpdateWindow();
}
void CRobotView::OnMenuadd() 
{
	// TODO: Add your command handler code here
	OnBtadd();	
}

void CRobotView::OnMenuerase() 
{
	// TODO: Add your command handler code here
	OnBterase();
}

void CRobotView::OnMenupara() 
{
	// TODO: Add your command handler code here
	OnBtpara();	
}

void CRobotView::OnMenustart() 
{
	// TODO: Add your command handler code here
	OnBtstart();
}

void CRobotView::OnLButtonDown(UINT nFlags, CPoint point) 
{
	// TODO: Add your message handler code here and/or call default
	
	CView::OnLButtonDown(nFlags, point);
	int curBlockULX = (point.x-m_origX)/gridWidth;
	int curBlockULY = (point.y-m_origY)/gridWidth;

	CRobotDoc* pDoc = GetDocument();

	if (m_bEraseBlock)//erase block state
	{
		ClearBlock(curBlockULY,curBlockULX);
		Invalidate();
		UpdateWindow();
	}
	else if (m_bAddBlock)//erase block state
	{
		SetBlock(curBlockULY,curBlockULX);
		Invalidate();
		UpdateWindow();
	}
}

void CRobotView::ShowPara(CDC* pDC)
{
	CRobotDoc* pDoc = GetDocument();

	int curBound = m_origX+m_CurPara.width*gridWidth+1;
	int startX = curBound+20;
	int startY = 20;

	CFont font;
	LOGFONT lf;
	memset(&lf, 0, sizeof(LOGFONT));
	lf.lfHeight = 16;
	strcpy(lf.lfFaceName, "Arial");
	VERIFY(font.CreateFontIndirect(&lf));
	CFont* oldFont = pDC->SelectObject(&font);
	COLORREF oldColor = pDC->SetTextColor(RGB(100,100,100));
	pDC->TextOut(startX,startY,"Population Size:");
	pDC->TextOut(startX,startY+30,"Current Generations:");
	pDC->TextOut(startX,startY+60,"Distance:");
	pDC->TextOut(startX,startY+90,"Smooth:");
	pDC->TextOut(startX,startY+120,"Safety:");
	pDC->TextOut(startX,startY+170,"Distance:");
	pDC->TextOut(startX,startY+200,"Smooth:");
	pDC->TextOut(startX,startY+230,"Safety:");
	pDC->SelectObject(oldFont);
	
	 oldColor = pDC->SetTextColor(RGB(255,0,255));
	
	CString gen;
	gen.Format(_T("%d"),m_CurPara.pSize);
	pDC->TextOut(startX+120,startY,gen);
	gen.Format(_T("%d"),pDoc->GARoad->curGenNum);
	pDC->TextOut(startX+120,startY+30,gen);
	gen.Format(_T("%.3f"),fit[0].A);
	pDC->TextOut(startX+120,startY+60,gen);
	gen.Format(_T("%.3f"),fit[0].B);
	pDC->TextOut(startX+120,startY+90,gen);
	gen.Format(_T("%.3f"),fit[0].C);
	pDC->TextOut(startX+120,startY+120,gen);
	CPen m_ChartPen1(PS_SOLID,1,RGB(0,0,255));//Line Color
	pDC->SelectObject (&m_ChartPen1);
	pDC->MoveTo(startX+120,startY+170);
	pDC->LineTo(startX+170,startY+170);
	CPen m_ChartPen2(PS_SOLID,1,RGB(0,255,0));//Line Color
	pDC->SelectObject (&m_ChartPen2);
	pDC->MoveTo(startX+120,startY+200);
	pDC->LineTo(startX+170,startY+200);
	CPen m_ChartPen3(PS_SOLID,1,RGB(255,0,0));//Line Color
	pDC->SelectObject (&m_ChartPen3);
	pDC->MoveTo(startX+120,startY+230);
	pDC->LineTo(startX+170,startY+230);

	pDC->SetTextColor(oldColor);
}