huffmanexpview.cpp

vc编程实现的哈夫曼编码

// HuffmanExpView.cpp : implementation of the CHuffmanExpView class
//

#include "stdafx.h"
#include "HuffmanExp.h"
#include "MainFrm.h"
#include "HuffmanExpDoc.h"
#include "HuffmanExpView.h"
#include "Dib.h"
#include "dlghuffman.h"

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

/////////////////////////////////////////////////////////////////////////////
// CHuffmanExpView

IMPLEMENT_DYNCREATE(CHuffmanExpView, CScrollView)

BEGIN_MESSAGE_MAP(CHuffmanExpView, CScrollView)
	//{{AFX_MSG_MAP(CHuffmanExpView)
	ON_COMMAND(ID_HUFFMAN, OnHuffman)
	//}}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()

/////////////////////////////////////////////////////////////////////////////
// CHuffmanExpView construction/destruction

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

}

CHuffmanExpView::~CHuffmanExpView()
{
}

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

	return CScrollView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CHuffmanExpView drawing

void CHuffmanExpView::OnDraw(CDC* pDC)
{
	CHuffmanExpDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	// TODO: add draw code for native data here
		if (! pDoc->m_pDib->IsEmpty())
		pDoc->m_pDib->Display(pDC, 0, 0);
}

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

	CHuffmanExpDoc* pDoc = GetDocument();

	CSize sizeTotal(pDoc->m_pDib->GetWidth(), pDoc->m_pDib->GetHeight());
	SetScrollSizes(MM_TEXT, sizeTotal);
	
	CMainFrame* pAppFrame = (CMainFrame*) AfxGetApp()->m_pMainWnd;
	ASSERT_KINDOF(CMainFrame, pAppFrame);
	CRect rc;
	pAppFrame->GetClientRect(&rc);
	if (rc.Width() >= sizeTotal.cx && rc.Height() >= sizeTotal.cy &&
		(sizeTotal.cx>0 || sizeTotal.cy>0))
		ResizeParentToFit(FALSE);
	
}

/////////////////////////////////////////////////////////////////////////////
// CHuffmanExpView printing

BOOL CHuffmanExpView::OnPreparePrinting(CPrintInfo* pInfo)
{
	// default preparation
	return DoPreparePrinting(pInfo);
}

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

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

/////////////////////////////////////////////////////////////////////////////
// CHuffmanExpView diagnostics

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

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

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

/////////////////////////////////////////////////////////////////////////////
// CHuffmanExpView message handlers

void CHuffmanExpView::OnHuffman() 
{
	// TODO: Add your command handler code here
	// 查看哈夫曼编码表
	
	// 获取文档
	CHuffmanExpDoc* pDoc = GetDocument();
	
	// 指向源图像象素的指针
	unsigned char *	lpSrc;
	
	// 指向DIB的指针

	LPBYTE	lpDIB;
	
	// 指向DIB象素指针

	LPBYTE   lpDIBBits;
	
	// DIB的高度
	LONG	lHeight;
	
	// DIB的宽度
	LONG	lWidth;
	
	// 图像每行的字节数
	LONG	lLineBytes;
	
	// 图像象素总数
	LONG	lCountSum;
	
	// 循环变量
	LONG	i;
	LONG	j;
	
	// 保存各个灰度值频率的数组指针
	FLOAT * fFreq;
	
	// 获取当前DIB颜色数目
	int		iColorNum;
	
	// 锁定DIB
	lpDIB = (LPBYTE) ::GlobalLock((HGLOBAL) pDoc->m_pDib->m_hDib);

	
	// 找到DIB图像象素起始位置
	 lpDIBBits = ::FindDIBBits(lpDIB);

	
	// 获取当前DIB颜色数目
	iColorNum = ::DIBNumColors(lpDIB);
	
	// 判断是否是8-bpp位图（这里为了方便，只处理8-bpp位图，其它的可以类推）
	if (iColorNum != 256)
	{
		// 提示用户
		MessageBox("目前只支持256色位图哈夫曼编码！", "系统提示" , 
			MB_ICONINFORMATION | MB_OK);
		
		// 解除锁定
		::GlobalUnlock((HGLOBAL) pDoc->m_pDib->m_hDib);
		
		// 返回
		return;
	}
	
	// 更改光标形状
	BeginWaitCursor();
	
	
	
	// 分配内存
	fFreq = new FLOAT[iColorNum];
	
	// 计算DIB宽度
	lWidth = ::DIBWidth(lpDIB);
	
	// 计算DIB高度
	lHeight = ::DIBHeight(lpDIB);
	
	// 计算图像每行的字节数
	lLineBytes = WIDTHBYTES(lWidth * 8);
	
	// 重置计数为0
	for (i = 0; i < iColorNum; i ++)
	{
		// 清零
		fFreq[i] = 0.0;
	}
	
	// 计算各个灰度值的计数（对于非256色位图，此处给数组fFreq赋值方法将不同）
	for (i = 0; i < lHeight; i ++)
	{
		for (j = 0; j < lWidth; j ++)
		{
			// 指向图像指针
			lpSrc = (unsigned char *)lpDIBBits + lLineBytes * i + j;
			
			// 计数加1
			fFreq[*(lpSrc)] += 1;
		}
	}
	
	// 计算图像象素总数
	lCountSum = lHeight * lWidth;
	
	// 计算各个灰度值出现的概率
	for (i = 0; i < iColorNum; i ++)
	{
		// 计算概率
		fFreq[i] /= (FLOAT)lCountSum;
	}
	
	// 计算各个灰度级出现的频率结束
	/*****************************************************************************/
	
	// 创建对话框
	CDlgHuffman dlgPara;
	
	// 初始化变量值

	dlgPara.m_fFreq = fFreq;
	dlgPara.m_iColorNum = iColorNum;
	
	// 显示对话框
	dlgPara.DoModal();
	
	// 解除锁定
	::GlobalUnlock((HGLOBAL) pDoc->m_pDib->m_hDib);

	// 恢复光标
	EndWaitCursor();
	
}