www.pudn.com > jtlab.rar > Line.cpp

#include "stdafx.h" 
#include "VCadView.h" 
#include "math.h" 
#include "VCad.h" 
#include "VCadDoc.h" 
#include "MainFrm.h" 
#include "Entity.h" 
#include "CreateCmd.h" 
 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
void iSwap(int* a,int* b); 
IMPLEMENT_SERIAL(CLine, CEntity, 0) 
//CLine class说明于Entity.h中 
CLine::CLine() 
{ 
	Init(); 
} 
 
CLine::CLine(const CLine& line) 
		: CEntity(line) 
{ 
	m_begin = line.m_begin; 
	m_end	 = line.m_end; 
} 
 
CLine::CLine(const Position& begin,const Position& end) 
{ 
	Init(); 
	m_begin = begin ; 
	m_end = end ; 
} 
 
CLine::~CLine() 
{ 
} 
 
CLine& CLine::operator = (const CLine& line) 
{ 
	// 处理特殊情况：line1 = line1 
	if(this == &line) 
		return *this; 
	// 一般情形：line2 = line1 
	CEntity::operator = (line); // 调用基类的重载“=”操作 
	m_begin = line.m_begin; 
	m_end = line.m_end; 
	return *this; 
} 
 
CEntity* CLine::Copy() 
{ 
	CLine*	pEntity = new CLine(m_begin, m_end); 
	return pEntity; 
} 
 
void CLine::Init() 
{ 
	CEntity::Init(); 
	m_type = etLine; 
	m_begin.Init(); 
	m_end.Init(); 
} 
 
int	CLine::GetType() 
{ 
	return etLine; 
} 
 
Position CLine::GetBeginPos() 
{ 
	return m_begin; 
} 
 
Position CLine::GetEndPos() 
{ 
	return m_end; 
} 
 
void CLine::Draw(CDC * pDC, int drawMode ) 
{ 
	CPoint pt_begin, pt_end; // 屏幕坐标的起点和终点 
	g_pView->WorldtoScreen(m_begin,pt_begin); // 将世界坐标转化为屏幕坐标 
	g_pView->WorldtoScreen(m_end,pt_end) ; 
	int		n = GetROP2(pDC->GetSafeHdc()); // 得到原来的绘图模式 
	CPen	pen;  
	 
	if( drawMode == dmNormal )  // 如果为正常状态的绘制，根据成员变量创建画笔 
	{ 
	pen.CreatePen(m_lineStyle,m_lineWidth,m_color) ; 
	} 
	else // 非正常状态调用SetDrawEnvir函数设置绘图环境 
		::SetDrawEnvir(pDC, drawMode, &pen); 
	 
	CPen* pOldPen = pDC->SelectObject(&pen); // 得到原来的画笔 
	pDC->SetMapMode(MM_LOENGLISH);	 
   if( drawMode == dmNormal ) 
   {//pDC->MoveTo(pt_begin);修改点 
    // pDC->LineTo(pt_end); 
    Bres_Line(pt_begin.x,pt_begin.y,pt_end.x,pt_end.y,RGB(255,0,122),pDC); 
   } 
   else {pDC->MoveTo(pt_begin); // 根据屏幕坐标绘制图元 
         pDC->LineTo(pt_end);} 
	pDC->SelectObject(pOldPen); // 恢复原来的画笔  
	pDC->SetROP2(n); // 恢复原来的绘图模式 
} 
 
 
void CLine::Serialize(CArchive& ar)  
{ 
	CEntity::Serialize(ar); 
	m_begin.Serialize(ar); 
	m_end.Serialize(ar); 
} 
//以下为Breshenham直线绘制算法 
int CLine::Bres_Line(int x1, int y1, int x2, int y2, COLORREF color,CDC* pDC) 
{ 
 int iTag; 
 int dx,dy,tx,ty,inc1,inc2,d,curx,cury,i;  
 pDC->SetPixel(x1,y1,color); 
 if(x1==x2&&y1==y2) 
 { 
  return 1; 
 } 
 
 iTag=0; 
 dx=abs(x2-x1); 
 dy=abs(y2-y1); 
 if(dx0?1:((x2-x1)==0?0:-1); 
 ty=(y2-y1)>0?1:((y2-y1)==0?0:-1);  
 curx=x1; 
 cury=y1; 
 inc1=2*dy; 
 inc2=2*dx; 
 d=inc1-dx; 
 for (i=1;i<=dx;i++) 
 {pDC->SetPixel(curx,cury,color); 
  if(d>=0) 
  { 
   curx+=tx; 
   cury+=ty; 
   d-=inc2; 
  } 
  else if(iTag) cury+=ty; 
       else curx+=tx; 
  d+=inc1; 
 } 
 return 0; 
} 
 
void iSwap(int* a,int* b) 
{ 
 int tmp=*a; 
 *a=*b; 
 *b=tmp; 
} 
 
//class Createline 说明于CreateCmd.h中 
CCreateLine::CCreateLine() 
	: m_begin(0,0), m_end(0,0) 
{ 
	m_nStep = 0; // 初始化操作步为 0 
} 
 
CCreateLine::~CCreateLine() 
{ 
} 
 
int CCreateLine::GetType() 
{ 
	return ctCreateLine;  
} 
 
int	CCreateLine::OnLButtonDown(UINT nFlags, const Position& pos)  
{ 
	m_nStep ++; // 每次单击鼠标左键时操作步加 1  
	switch(m_nStep) // 根据操作步执行相应的操作 
	{ 
		case 1:  
		{ 
			m_begin = m_end = pos; 
			::Prompt("请输入直线的末端点："); 
			break; 
		} 
		case 2: 
		{ 
			CDC*	pDC = g_pView->GetDC(); // 得到设备环境指针  
 
			// 擦除在拖动状态时显示的最后一条线 
			CLine*	pTempLine = new CLine(m_begin,m_end);  
			pTempLine->Draw(pDC, dmDrag); 
			delete pTempLine; 
 
			// 如果在按鼠标左键的过程中同时按下了Shift键， 
			//	那么根据鼠标单击位置绘制水平线或竖直线 
			if( nFlags & MK_SHIFT ){  
				double dx = pos.x - m_begin.x; 
				double dy = pos.y - m_begin.y; 
				if(fabs(dx) <= fabs(dy)) // 如果鼠标单击位置在X方向靠近起点 
					m_end.Set(m_begin.x, pos.y); // 那么终点的x坐标与起点的相同 
				else 
					m_end.Set(pos.x,m_begin.y); 
			} 
			else 
				m_end = pos; // 如果未按下Shift键, 则终点为鼠标单击位置			 
						 
			CLine*	pNewLine = new CLine(m_begin,m_end);// 根据起点和终点创建直线 
			pNewLine->Draw(pDC,dmNormal); // 绘制直线 
			g_pDoc->m_EntityList.AddTail(pNewLine); // 将直线指针添加到图元链表 
			g_pDoc->SetModifiedFlag(TRUE);// set modified flag ; 
			 
			g_pView->ReleaseDC(pDC); // 释放设备环境指针 
			 
			m_nStep = 0;  // 将操作步重置为 0 
			::Prompt("请输入直线的起点："); 
			// 如果将当前的终点设置为起点，并将操作步设置为1，即可绘制Ployline 
			//	m_pBegin = m_pEnd; 
			//	m_nStep = 0;  
			break; 
		} 
		 
	} 
	return 0; 
} 
 
int	CCreateLine::OnMouseMove(UINT nFlags, const Position& pos) 
{ 
	// 用一静态变量nPreRefresh记录进入OnMouseMove状态时的刷新次数 
	static	int nPreRefresh = g_nRefresh;  
	// 布尔变量bRefresh说明在OnMouseMove过程中视窗是否被刷新 
	BOOL	bRefresh = FALSE;  
	// nCurRefresh用于记录当前的刷新次数 
	int		nCurRefresh = g_nRefresh;  
	// 如果nCurRefresh和nPreRefresh不相等，说明视窗曾被刷新过 
	if(nCurRefresh != nPreRefresh){  
		bRefresh = TRUE; 
		nPreRefresh = nCurRefresh;  
	} 
 
	switch(m_nStep) 
	{ 
		case 0: 
			::Prompt("请输入直线的起点："); 
			break; 
		case 1: 
		{ 
			Position	prePos, curPos; 
			prePos = m_end; // 获得鼠标所在的前一个位置 
			 
			// 如果在按鼠标左键的过程中同时按下了Shift键， 
			//	那么根据鼠标单击位置绘制水平线或竖直线 
			if( nFlags & MK_SHIFT ){ 
				double dx = pos.x - m_begin.x; 
				double dy = pos.y - m_begin.y; 
				if(fabs(dx)>=fabs(dy)) 
					curPos.Set(pos.x,m_begin.y); 
				else 
					curPos.Set(m_begin.x, pos.y); 
			} 
			else 
				curPos = pos; 
 
			CDC*	pDC = g_pView->GetDC(); // 得到设备环境指针 
			 
			// 创建临时对象擦除上一条橡皮线 
			CLine*	pTempLine1 = new CLine(m_begin, prePos); 
			if(!bRefresh) // 当视窗没有被刷新时，重画原来的橡皮线使其被擦除 
				pTempLine1->Draw(pDC, dmDrag); 
			delete pTempLine1; 
			// 创建临时对象，根据当前位置绘制一条橡皮线 
			CLine*	pTempLine2 = new CLine(m_begin, curPos);	 
			pTempLine2->Draw(pDC, dmDrag); 
			delete pTempLine2; 
  
			g_pView->ReleaseDC(pDC); // 释放设备环境指针			 
 
			m_end = curPos; // 将当前位置设置为直线终点，以备下一次鼠标移动时用 
			break; 
		} 
	} 
	return 0; 
} 
// 单击鼠标右键取消当前的操作 
int	CCreateLine::OnRButtonDown(UINT nFlags, const Position& pos)  
{ 
	// 如果当前的操作步为 1 ，那么要在结束本次操作前擦除上次鼠标移动时绘制的橡皮线 
	if(m_nStep == 1){  
		CDC*	pDC = g_pView->GetDC(); // 得到设备环境指针 
		Position	prePos = m_end; // 获得鼠标所在的前一个位置 
		CLine*	pTempLine = new CLine(m_begin, prePos);  
		pTempLine->Draw(pDC, dmDrag); // 擦除上一次绘制的橡皮线 
		delete pTempLine; 
		g_pView->ReleaseDC(pDC); // 释放设备环境指针 
	} 
	m_nStep = 0; // 将操作步重置为 0  
	::Prompt("请输入直线的起点："); 
	return 0; 
} 
// 调用Cancel 函数取消本次操作 
int CCreateLine::Cancel() 
{ 
	// 如果当前的操作步为 1 ，那么要在结束本次操作前擦除上次鼠标移动时绘制的橡皮线 
	if(m_nStep == 1){  
		CDC*	pDC = g_pView->GetDC(); // 得到设备环境指针 
		Position	prePos = m_end; // 获得鼠标所在的前一个位置 
		CLine*	pTempLine = new CLine(m_begin, prePos);  
		pTempLine->Draw(pDC, dmDrag); // 擦除上一次绘制的橡皮线 
		delete pTempLine; 
		g_pView->ReleaseDC(pDC); // 释放设备环境指针 
	} 
	m_nStep = 0; // 将操作步重置为 0  
	::Prompt("就绪"); // 等待提示新类型的命令操作 
	return 0 ; 
}