// 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();
	
}