imgrotateview.cpp

visual c++数字图像与图形处理中的光盘内容

// ImgRotateView.cpp : implementation of the CImgRotateView class
//

#include "stdafx.h"
#include "ImgRotate.h"

#include "ImgRotateDoc.h"
#include "ImgRotateView.h"

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

#include "math.h"
#include "ImageGeometry.h"
#include "ImageRotateDialog.h"

/////////////////////////////////////////////////////////////////////////////
// CImgRotateView

IMPLEMENT_DYNCREATE(CImgRotateView,  CView)

BEGIN_MESSAGE_MAP(CImgRotateView,  CView)
	//{{AFX_MSG_MAP(CImgRotateView)
	ON_COMMAND(IDM_IMAGE_ROTATE,  OnImageRotate)
	ON_UPDATE_COMMAND_UI(IDM_IMAGE_ROTATE,  OnUpdateImageRotate)
	ON_COMMAND(IDM_IMAGE_RESTORE,  OnImageRestore)
	ON_UPDATE_COMMAND_UI(IDM_IMAGE_RESTORE,  OnUpdateImageRestore)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CImgRotateView construction/destruction

CImgRotateView::CImgRotateView()
{
	// TODO: add construction code here
	m_nWidth = 800;
	m_nHeight = 600;
	m_dwOperation = 0L;
	
	m_nQuality = 0;
	m_fTheta = 0.0f;
}

CImgRotateView::~CImgRotateView()
{
}

BOOL CImgRotateView::PreCreateWindow(CREATESTRUCT& cs)
{
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs

	return CView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CImgRotateView drawing

void CImgRotateView::OnDraw(CDC* pDC)
{
	CImgRotateDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	// TODO: add draw code for native data here
	CDib* pDib = pDoc->m_pDib;
	if(pDib)
	{
		BeginWaitCursor();
		m_nWidth = (int)pDib->GetWidth();
		m_nHeight = (int)pDib->GetHeight(); 

		CDC memDC;
		memDC.CreateCompatibleDC(pDC);
		CBitmap ddb;
		ddb.CreateCompatibleBitmap(pDC, m_nWidth, m_nHeight);
		CBitmap* pOldBitmap = memDC.SelectObject(&ddb);

		pDib->Draw(memDC.m_hDC, 0, 0, m_nWidth, m_nHeight, 
					0, 0, m_nWidth, m_nHeight, DIB_RGB_COLORS, SRCCOPY);
		
		if(m_dwOperation == 0L) 
			pDC->BitBlt(0, 0, m_nWidth, m_nHeight, &memDC, 0, 0, SRCCOPY);
		else
			Rotate(pDC, pDib, 0, 0, m_nWidth, m_nHeight);
			//Rotate(pDC, pDib, 0, 0, 200, 200);
		
		memDC.SelectObject(pOldBitmap);
		ddb.DeleteObject();
		EndWaitCursor();
	}
}



/////////////////////////////////////////////////////////////////////////////
// CImgRotateView diagnostics

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

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

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

/////////////////////////////////////////////////////////////////////////////
// CImgRotateView message handlers

void CImgRotateView::OnImageRotate() 
{
	// TODO: Add your command handler code here
	CImageRotateDialog dlg;
	
	dlg.m_fTheta = m_fTheta;
	dlg.m_nQuality = m_nQuality;


	int responeDlg = dlg.DoModal();
	if(responeDlg == IDOK)
	{
		m_fTheta = dlg.m_fTheta;
		m_nQuality = dlg.m_nQuality;
		m_dwOperation = IMAGE_GEOMETRY_ROTATE;
		Invalidate();
	}
}

void CImgRotateView::OnUpdateImageRotate(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	if(m_dwOperation == IMAGE_GEOMETRY_ROTATE)
		pCmdUI ->SetCheck(TRUE);
	else
		pCmdUI ->SetCheck(FALSE);
}

void CImgRotateView::OnImageRestore() 
{
	// TODO: Add your command handler code here
	m_dwOperation = 0L;
	Invalidate();
}

void CImgRotateView::OnUpdateImageRestore(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	if(m_dwOperation == 0L)
		pCmdUI ->SetCheck(TRUE);
	else
		pCmdUI ->SetCheck(FALSE);	

}

void CImgRotateView::Rotate(CDC *pDC,  CDib *pDib,  int x,  int y,  int nWidth,  int nHeight)
{
	ASSERT(pDib);

	//限制处理的区域----防止任意填写数据而使用访问无效
	if((x > (m_nWidth - 1)) || (y > (m_nHeight - 1))) return ;
	//实际处理的宽度和高度
	if(nWidth >(m_nWidth - x))
	{
		MessageBox("区域宽度越界");
		return;
	}
	if(nHeight > (m_nHeight - y))
	{
		MessageBox("区域高度越界");
		return;
	}
	
	// 32位颜色数据:输入源
	DWORD dwSizeSrc = m_nWidth * m_nHeight * 4;

	//分配全局内存 32位源数据
	BYTE* pbyBitsSrc32 =  new BYTE[dwSizeSrc];
	if(pbyBitsSrc32 == NULL) return;
	memset(pbyBitsSrc32, 0, dwSizeSrc);
	


	//CDib类只提供了一个全部数据的函数
	pDib->GetDdbData32(pbyBitsSrc32);


	//将角度转化成弧度
	float fRadTheta = (float)(m_fTheta * PIE / 180.0);

	//计算其三角函数值
	float fSinTheta = (float)sin(fRadTheta);
	float fCosTheta = (float)cos(fRadTheta);

	//右下角点(基于右手系)旋转后的x值
	float fxrb = (nWidth * fCosTheta - nHeight * fSinTheta);
	//右上角点(基于右手系)旋转后的x值
	float fxtb = (nWidth * fCosTheta + nHeight * fSinTheta);

	//右下角点(基于右手系)旋转后的x值
	float fyrt = ((-nWidth) * fSinTheta + nHeight * fCosTheta);
	//左上角点(基于右手系)旋转后的x值
	float fylt = (nWidth * fSinTheta + nHeight * fCosTheta);

	int nWidthDst = (int)(max(ABS(fxrb), ABS(fxtb))  + 0.5f);
	int nHeightDst = (int)(max(ABS(fyrt), ABS(fylt)) + 0.5f);


	//分配目标缓冲区
	DWORD dwSizeDst = ((DWORD)(nWidthDst * nHeightDst)) * 4;
	
	BYTE* pbyBitsDst32 =  new BYTE[ dwSizeDst ];
	if(pbyBitsDst32 == NULL) return;
	memset(pbyBitsDst32, 255,  dwSizeDst);
	
	CImageGeometry* ig = new CImageGeometry();


	//设置插值质量
	if(m_nQuality == 0)
		ig->SetQuality( IMAGE_GEOMETRY_NEAREST_NEIGHBOR_INTERPOLATE );
	else if(m_nQuality == 1)
		ig->SetQuality( IMAGE_GEOMETRY_BILINEAR_INTERPOLATE );
	else if(m_nQuality == 2)
		ig->SetQuality( IMAGE_GEOMETRY_THREE_ORDER_INTERPOLATE );

	ig->Rotate(pbyBitsSrc32, x, y, nWidth, nHeight, m_nWidth, m_nHeight, pbyBitsDst32, m_fTheta, nWidthDst, nHeightDst);

	CClientDC dc(this);
	CDC memDC;
	memDC.CreateCompatibleDC(&dc);
	HBITMAP hBitmap = ig->CreateDdb(pDC->m_hDC, nWidthDst, nHeightDst, pbyBitsDst32);
	HBITMAP hOldBitmap = (HBITMAP)memDC.SelectObject(hBitmap);
	pDC->BitBlt(0, 0, nWidthDst, nHeightDst, &memDC, 0, 0, SRCCOPY);
	::DeleteObject(hBitmap);
	memDC.SelectObject(hOldBitmap);

	delete ig;
	delete[] pbyBitsSrc32;
	delete[] pbyBitsDst32;
	ReleaseDC(&dc);
}