templatetransview.cpp

人民邮电出版社的——Visual C++数字图像处理典型算法及实现（源码） 希望大家能用上

// TemplateTransView.cpp : implementation of the CTemplateTransView class
//

#include "stdafx.h"
#include "TemplateTrans.h"

#include "TemplateTransDoc.h"
#include "TemplateTransView.h"
#include "mainfrm.h"

#include "SmoothDlg.h"
#include "MidFilterDlg.h"
#include "SharpThreDlg.h"
#include "ColorDlg.h"
#include "ColorTable.h"

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

/////////////////////////////////////////////////////////////////////////////
// CTemplateTransView

IMPLEMENT_DYNCREATE(CTemplateTransView, CScrollView)

BEGIN_MESSAGE_MAP(CTemplateTransView, CScrollView)
	//{{AFX_MSG_MAP(CTemplateTransView)
	ON_WM_ERASEBKGND()
	ON_COMMAND(ID_EDIT_COPY, OnEditCopy)
	ON_UPDATE_COMMAND_UI(ID_EDIT_COPY, OnUpdateEditCopy)
	ON_COMMAND(ID_EDIT_PASTE, OnEditPaste)
	ON_UPDATE_COMMAND_UI(ID_EDIT_PASTE, OnUpdateEditPaste)
	ON_COMMAND(ID_ENHA_COLOR, OnEnhaColor)
	ON_COMMAND(ID_ENHA_GRADSHARP, OnEnhaGradsharp)
	ON_COMMAND(ID_ENHA_MidianF, OnENHAMidianF)
	ON_COMMAND(ID_ENHA_SHARP, OnEnhaSharp)
	ON_COMMAND(ID_ENHA_SMOOTH, OnEnhaSmooth)
	ON_COMMAND(ID_FILE_256ToGray, OnFILE256ToGray)
	//}}AFX_MSG_MAP
	// Standard printing commands
	ON_COMMAND(ID_FILE_PRINT, CScrollView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_DIRECT, CScrollView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_PREVIEW, CScrollView::OnFilePrintPreview)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTemplateTransView construction/destruction

CTemplateTransView::CTemplateTransView()
{
	// TODO: add construction code here

}

CTemplateTransView::~CTemplateTransView()
{
}

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

	return CScrollView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CTemplateTransView drawing

void CTemplateTransView::OnDraw(CDC* pDC)
{
	BeginWaitCursor();

	CTemplateTransDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);

	HDIB hDIB = pDoc->GetHDIB();
	if (hDIB != NULL)
	{
		LPSTR lpDIB = (LPSTR) ::GlobalLock((HGLOBAL) hDIB);
		int cxDIB = (int) pDoc->GetDibImage()->DIBWidth(lpDIB);
		int cyDIB = (int) pDoc->GetDibImage()->DIBHeight(lpDIB);

		::GlobalUnlock((HGLOBAL) hDIB);
		
		CRect rcDIB;
		rcDIB.top = rcDIB.left = 0;
		rcDIB.right = cxDIB;
		rcDIB.bottom = cyDIB;
		
		CRect rcDest;
		
		if (pDC->IsPrinting())
		{
			// 是打印，计算输出图像的位置和大小，以便符合页面
			
			// 获取打印页面的水平宽度(象素)
			int cxPage = pDC->GetDeviceCaps(HORZRES);			
			// 获取打印页面的垂直高度(象素)
			int cyPage = pDC->GetDeviceCaps(VERTRES);			
			// 获取打印机每英寸象素数
			int cxInch = pDC->GetDeviceCaps(LOGPIXELSX);
			int cyInch = pDC->GetDeviceCaps(LOGPIXELSY);
			
			// 计算打印图像大小（缩放，根据页面宽度调整图像大小）
			rcDest.top = rcDest.left = 0;
			rcDest.bottom = (int)(((double)cyDIB * cxPage * cyInch)
					/ ((double)cxDIB * cxInch));
			rcDest.right = cxPage;
			
			// 计算打印图像位置（垂直居中）
			int temp = cyPage - (rcDest.bottom - rcDest.top);
			rcDest.bottom += temp/2;
			rcDest.top += temp/2;
		}
		else   
		{
			// 不必缩放图像
			rcDest = rcDIB;
		}
		
		pDoc->GetDibImage()->PaintDIB(pDC->m_hDC, &rcDest, pDoc->GetHDIB(),
			&rcDIB, pDoc->GetDocPalette());
	}
	
	EndWaitCursor();
}

void CTemplateTransView::OnInitialUpdate()
{
	CScrollView::OnInitialUpdate();

	// TODO: calculate the total size of this view

}

/////////////////////////////////////////////////////////////////////////////
// CTemplateTransView printing

BOOL CTemplateTransView::OnPreparePrinting(CPrintInfo* pInfo)
{
	// 设置总页数为一。
	pInfo->SetMaxPage(1);

	return DoPreparePrinting(pInfo);
}

void CTemplateTransView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add extra initialization before printing
}

void CTemplateTransView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add cleanup after printing
}

/////////////////////////////////////////////////////////////////////////////
// CTemplateTransView diagnostics

#ifdef _DEBUG
void CTemplateTransView::AssertValid() const
{
	CScrollView::AssertValid();
}

void CTemplateTransView::Dump(CDumpContext& dc) const
{
	CScrollView::Dump(dc);
}

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

/////////////////////////////////////////////////////////////////////////////
// CTemplateTransView message handlers

BOOL CTemplateTransView::OnEraseBkgnd(CDC* pDC) 
{
	// 主要是为了设置子窗体默认的背景色
	// 背景色由文档成员变量m_refColorBKG指定

	CTemplateTransDoc* pDoc = GetDocument();
	CBrush brush(pDoc->m_refColorBKG);                                              
	CBrush* pOldBrush = pDC->SelectObject(&brush);
		
	CRect rectClip;
	pDC->GetClipBox(&rectClip);			// 获取重绘区域	
	pDC->PatBlt(rectClip.left, rectClip.top, rectClip.Width(), 
		rectClip.Height(), PATCOPY);

	pDC->SelectObject(pOldBrush);                                                  

	return TRUE;
}

void CTemplateTransView::OnEditCopy() 
{
	// 复制当前图像

	CTemplateTransDoc* pDoc = GetDocument();
	
	if (OpenClipboard())
	{
		BeginWaitCursor();
		EmptyClipboard();
		
		// 复制当前图像到剪贴板
		SetClipboardData (CF_DIB, pDoc->GetDibImage()->
			CopyHandle((HANDLE) pDoc->GetHDIB()) );

		CloseClipboard();
		EndWaitCursor();
	}
}

void CTemplateTransView::OnUpdateEditCopy(CCmdUI* pCmdUI) 
{
	// 如果当前DIB对象不空，复制菜单项有效
	pCmdUI->Enable(GetDocument()->GetHDIB() != NULL);
}

void CTemplateTransView::OnEditPaste() 
{
	// 粘贴图像
	
	HDIB hNewDIB = NULL;
	CTemplateTransDoc* pDoc = GetDocument();
	
	if (OpenClipboard())
	{
		BeginWaitCursor();

		// 读取剪贴板中的图像
		hNewDIB = (HDIB)pDoc->GetDibImage()->CopyHandle(::GetClipboardData(CF_DIB));

		CloseClipboard();
		
		if (hNewDIB != NULL)
		{
			pDoc->ReplaceHDIB(hNewDIB);		// 替换DIB，同时释放旧DIB对象		
			pDoc->InitDIBData();			// 更新DIB大小和调色板		
			pDoc->SetModifiedFlag(TRUE);	// 设置脏标记	
			SetScrollSizes(MM_TEXT, pDoc->GetDocSize());	// 重新设置滚动视图大小		
			OnDoRealize((WPARAM)m_hWnd,0);	// 实现新的调色板		
			pDoc->UpdateAllViews(NULL);		// 更新视图
		}

		EndWaitCursor();
	}
}

void CTemplateTransView::OnUpdateEditPaste(CCmdUI* pCmdUI) 
{
	// 如果当前剪贴板中有DIB对象，粘贴菜单项有效
	pCmdUI->Enable(::IsClipboardFormatAvailable(CF_DIB));
}

LRESULT CTemplateTransView::OnDoRealize(WPARAM wParam, LPARAM)
{
	ASSERT(wParam != NULL);

	CTemplateTransDoc* pDoc = GetDocument();	
	if (pDoc->GetHDIB() == NULL)
	{
		return 0L;
	}
	
	CPalette* pPal = pDoc->GetDocPalette();
	if (pPal != NULL)
	{
		CMainFrame* pAppFrame = (CMainFrame*) AfxGetApp()->m_pMainWnd;
		ASSERT_KINDOF(CMainFrame, pAppFrame);
		
		CClientDC appDC(pAppFrame);

		// All views but one should be a background palette.
		// wParam contains a handle to the active view, so the SelectPalette
		// bForceBackground flag is FALSE only if wParam == m_hWnd (this view)
		CPalette* oldPalette = appDC.SelectPalette(pPal,((HWND)wParam)!=m_hWnd);	
		if (oldPalette != NULL)
		{
			UINT nColorsChanged = appDC.RealizePalette();
			if (nColorsChanged > 0)
			{
				pDoc->UpdateAllViews(NULL);
			}
			appDC.SelectPalette(oldPalette, TRUE);
		}
		else
		{
			TRACE0("\tCCh1_1View::OnPaletteChanged中调用SelectPalette()失败！\n");
		}
	}
	
	return 0L;
}

void CTemplateTransView::OnActivateView(BOOL bActivate, CView* pActivateView, CView* pDeactiveView) 
{
	CScrollView::OnActivateView(bActivate, pActivateView, pDeactiveView);

	if (bActivate)
	{
		ASSERT(pActivateView == this);
		OnDoRealize((WPARAM)m_hWnd, 0);   // same as SendMessage(WM_DOREALIZE);
	}
}

void CTemplateTransView::CalcWindowRect(LPRECT lpClientRect, UINT nAdjustType) 
{
	CScrollView::OnInitialUpdate();
	ASSERT(GetDocument() != NULL);
	
	SetScrollSizes(MM_TEXT, GetDocument()->GetDocSize());
}

void CTemplateTransView::OnEnhaColor() 
{
	// 伪彩色编码

	CTemplateTransDoc* pDoc = GetDocument();
	
	int		nColor;		// 保存用户选择的伪彩色编码表索引
	LPSTR	lpDIB;		// 指向DIB的指针
	lpDIB = (LPSTR) ::GlobalLock((HGLOBAL) pDoc->GetHDIB());
	
	// 判断是否是8-bpp位图（只处理256色位图的伪彩色变换，其它的可以类推）
	if (pDoc->GetDibImage()->DIBNumColors(lpDIB) != 256)
	{
		MessageBox("目前只支持256色位图的伪彩色变换！", "系统提示" , 
			MB_ICONINFORMATION | MB_OK);
		::GlobalUnlock((HGLOBAL) pDoc->GetHDIB());

		return;
	}
	
	CColorDlg dlgPara;
	if (pDoc->m_nColorIndex >= 0)
	{		
		dlgPara.m_nColor = pDoc->m_nColorIndex;		// 初始选中当前的伪彩色
	}
	else
	{		
		dlgPara.m_nColor = 0;						// 初始选中灰度伪彩色编码表
	}	
	dlgPara.m_lpColorName = (LPSTR) ColorScaleName;	// 指向名称数组的指针	
	dlgPara.m_nColorCount = COLOR_SCALE_COUNT;		// 伪彩色编码数目	
	// 名称字符串长度
	dlgPara.m_nNameLen = sizeof(ColorScaleName) / COLOR_SCALE_COUNT;	
	// 显示对话框，提示用户设定平移量
	if (dlgPara.DoModal() != IDOK)
	{
		return;
	}
	
	nColor = dlgPara.m_nColor;
	delete dlgPara;	
	
	BeginWaitCursor();
	
	// 判断伪彩色编码是否改动
	if (pDoc->m_nColorIndex != nColor)
	{
		pDoc->GetDibImage()->ReplaceColorPal(lpDIB, (BYTE*) ColorsTable[nColor]);		
		pDoc->GetDocPalette()->SetPaletteEntries(0, 256, 
			(LPPALETTEENTRY) ColorsTable[nColor]);	// 替换当前文档调色板		
		pDoc->m_nColorIndex = nColor;				// 更新类成员变量		
		pDoc->SetModifiedFlag(TRUE);				// 设置脏标记		
		OnDoRealize((WPARAM)m_hWnd,0);				// 实现新的调色板		
		pDoc->UpdateAllViews(NULL);					// 更新视图
	}

	::GlobalUnlock((HGLOBAL) pDoc->GetHDIB());