shannondlg.cpp

香农编码器

// ShannonDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Shannon.h"
#include "ShannonDlg.h"
#include <stdlib.h>
#include <math.h>
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CShannonDlg dialog

CShannonDlg::CShannonDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CShannonDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CShannonDlg)
	m_char = _T("");
	m_gailv = 0.0;
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CShannonDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CShannonDlg)
	DDX_Control(pDX, IDC_LIST4, m_coder);
	DDX_Control(pDX, IDC_LIST3, m_length);
	DDX_Control(pDX, IDC_LIST2, m_leijia);
	DDX_Control(pDX, IDC_LIST1, m_fuhao);
	DDX_Text(pDX, IDC_EDIT2, m_char);
	DDX_Text(pDX, IDC_EDIT3, m_gailv);
	DDV_MinMaxDouble(pDX, m_gailv, 0., 1.);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CShannonDlg, CDialog)
	//{{AFX_MSG_MAP(CShannonDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_ADD_BUTTON, OnAddButton)
	ON_BN_CLICKED(IDC_DEL_BUTTON, OnDelButton)
	ON_BN_CLICKED(IDC_CODER_BUTTON, OnCoderButton)
	ON_BN_CLICKED(IDC_OK_BUTTON, OnOkButton)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CShannonDlg message handlers

BOOL CShannonDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// TODO: Add extra initialization here
	n = 0 ;



	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CShannonDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CShannonDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CShannonDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CShannonDlg::OnAddButton() 
{
	// TODO: Add your control notification handler code here
	UpdateData(TRUE) ;
	CString str , str1 ;

	coder[n].ch = m_char ;
	coder[n].gl = m_gailv ;
	str1.Format("%.3lf",m_gailv) ;
	str = coder[n].ch + "----" + str1 ;
	m_fuhao.AddString(str) ;
	n++ ;
	m_char="" ;
	m_gailv=0 ;
	UpdateData(FALSE) ;
}

/*
int CMP( const void *a , const void *b ){
	CODE *A = (CODE *)a ;
 	CODE *B = (CODE *)b ; 
	if ( A->gl-B->gl>0 )	return -1 ;
 	else	return 1 ;     
}

*/


void CShannonDlg::OnDelButton() 
{
	// TODO: Add your control notification handler code here
	int nIndex = m_fuhao.GetCurSel() ;
	//m_fuhao.GetSel()
	m_fuhao.DeleteString(nIndex) ;
	for ( int i = nIndex ; i < n-1 ; i ++){
		coder[i] = coder[i+1] ;
	}
	n -- ;
}

void CShannonDlg::OnCoderButton() 
{
	// TODO: Add your control notification handler code here
//	qsort ( coder, n , sizeof(CODE), CMP ) ;
	double sum = 0 , temper;
	int i , j ;
	CString str , str1 ; 
	for ( i=0; i<n;i++){
		sum += coder[i].gl ;
  		coder[i].len=ceil((-1)*log(coder[i].gl)/log(2)) ;
	}
	if (sum!=1)	
		MessageBox("大哥，不满足完备性呀!!!");

	else {
	
		CODE temp ;
		
//		for ( i = 0 ; i < n ; i ++ )
//		m_fuhao.DeleteString( i ) ;
		m_fuhao.ResetContent();
		m_leijia.ResetContent();
		m_length.ResetContent();
		m_coder.ResetContent();
		
 		for ( i = 0 ; i < n - 1 ; i ++ )
			for ( j = i + 1 ; j < n ; j ++ )
				if (coder[i].gl<coder[j].gl){
					temp = coder[i] ; 
					coder[i] = coder[j] ;
					coder[j] = temp ;
				}

			
		sum = 0 ;
		for ( i = 0 ; i < n ; i ++ ){

//			str1.Format("%.3lf",m_gailv) ;
//			str = coder[n].ch + "----" + str1 ;
			str1.Format("%.3lf",coder[i].gl) ;
			str = coder[i].ch +	"----" + str1 ;
			m_fuhao.AddString(str) ;
			str1.Format("%d",coder[i].len) ;
			m_length.AddString(str1) ;
			str1.Format("%.3lf",sum) ;
			m_leijia.AddString(str1) ;
		
			temper = sum ;
			for ( j=0;j<coder[i].len;j++){
			  		temper *= 2;
          			if (temper>=1){
						coder[i].co[j]='1';
                		temper-=1 ;
					} 
					else	coder[i].co[j]='0' ;
			} 
			coder[i].co[j]='\0' ;
        	//code[i].co = (int)(temp << code[i].len ) ;
			sum += coder[i].gl ; 
			str1.Format("%s",coder[i].co) ;
			m_coder.AddString(str1) ;
		}

	}
}

void CShannonDlg::OnOkButton() 
{
	// TODO: Add your control notification handler code here
	m_fuhao.ResetContent();
	m_length.ResetContent();
	m_leijia.ResetContent();
	m_coder.ResetContent();
}