mainfrm.cpp

一个含有朴素贝耶斯分类方法和决策树分类方法的源代码

// MainFrm.cpp : implementation of the CMainFrame class
//

#include "stdafx.h"
#include "HeartDM.h"

#include "MainFrm.h"

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

/////////////////////////////////////////////////////////////////////////////
// CMainFrame

IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	//{{AFX_MSG_MAP(CMainFrame)
	ON_WM_CREATE()
	ON_COMMAND(IDD_USER_SEARCH, OnUserSearch)
	ON_COMMAND(IDD_USER_RESULT, OnUserResult)
	ON_COMMAND(IDD_USER_PRODUCER, OnUserProducer)
	ON_COMMAND(IDD_PREPARE_DEL, OnPrepareDel)
	ON_COMMAND(IDD_USERDATACLASS, OnUserdataclass)
	ON_COMMAND(IDD_USERINPUTDATA, OnUserinputdata)
	ON_COMMAND(IDD_EXECUTE, OnExecute)
	ON_COMMAND(IDD_USERDELETE, OnUserdelete)
	ON_COMMAND(IDD_DECISION, OnDecision)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

static UINT indicators[] =
{
	ID_SEPARATOR,           // status line indicator
	ID_INDICATOR_CAPS,
	ID_INDICATOR_NUM,
	ID_INDICATOR_SCRL,
};

/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction

CMainFrame::CMainFrame()
{
	// TODO: add member initialization code here
	
}

CMainFrame::~CMainFrame()
{
}

int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
		| CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
		!m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
	{
		TRACE0("Failed to create toolbar\n");
		return -1;      // fail to create
	}

	if (!m_wndStatusBar.Create(this) ||
		!m_wndStatusBar.SetIndicators(indicators,
		  sizeof(indicators)/sizeof(UINT)))
	{
		TRACE0("Failed to create status bar\n");
		return -1;      // fail to create
	}

	// TODO: Delete these three lines if you don't want the toolbar to
	//  be dockable
	m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
	EnableDocking(CBRS_ALIGN_ANY);
	DockControlBar(&m_wndToolBar);

	return 0;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs

	return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// CMainFrame diagnostics

#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
	CFrameWnd::AssertValid();
}

void CMainFrame::Dump(CDumpContext& dc) const
{
	CFrameWnd::Dump(dc);
}

#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CMainFrame message handlers


void CMainFrame::OnUserSearch() 
{
	// TODO: Add your command handler code here
	m_ListView->ListUser();	
}

void CMainFrame::OnUserResult() 
{
	// TODO: Add your command handler code here
	m_ListView->ListPrepareRecord();	
}

void CMainFrame::OnUserProducer() 
{
	// TODO: Add your command handler code here
	m_ListView->PrepareData();	
}

void CMainFrame::OnPrepareDel() 
{
	// TODO: Add your command handler code here
	m_ListView->PrepareDataDel();	
}

void CMainFrame::OnUserdataclass() 
{
	// TODO: Add your command handler code here
	m_ListView->BayesianClass();	
}

void CMainFrame::OnUserinputdata() 
{
	// TODO: Add your command handler code here
	m_ListView->InputData();	
}

void CMainFrame::OnExecute() 
{
	// TODO: Add your command handler code here
	m_ListView->InDatabase();	
}

void CMainFrame::OnUserdelete() 
{
	// TODO: Add your command handler code here
	m_ListView->UserDelete();	
}

void CMainFrame::OnDecision() 
{
	// TODO: Add your command handler code here

	int res=m_ListView->C5tree.tree();
	CString s;
	s.Format("%d",res);
	CString str="结果是：";
	str+=s;
	AfxMessageBox(str);

}