mainfrm.cpp

用K均值算法实现聚类分析的Vc源码

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

#include "stdafx.h"
#include "Clustering.h"

#include "MainFrm.h"
#include "InputDlg.h"
#include "ParameterDlg.h" 
#include "math.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(ID_PATTERN_INPUT, OnPatternInput)
	ON_COMMAND(ID_PATTERN_SORT, OnPatternSort)
	ON_COMMAND(ID_SHOW_RESULT, OnShowResult)
	//}}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
	this->m_center=NULL;
	this->m_pattern=NULL;
	this->m_Point=NULL;
}

CMainFrame::~CMainFrame()
{
	delete []m_pattern;
	delete []m_Point;
}

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::OnPatternInput() 
{
	// TODO: Add your command handler code here
	CInputDlg inputDlg;
	int ID=inputDlg.DoModal();
	if(ID==IDOK)
		m_strPoint=inputDlg.m_strPoint;

}

void CMainFrame::OnPatternSort() 
{
	// TODO: Add your command handler code here
	CParameterDlg parameterDlg;
	int ID=parameterDlg.DoModal();
	if(ID==IDOK)
	{
		this->m_CenterNum=parameterDlg.m_CenterNum;
		this->m_IterativeNum=parameterDlg.m_IterativeNum;
	}
	else
		return;
	KSort();
//	int m=1;
}

void CMainFrame::OnShowResult() 
{
	// TODO: Add your command handler code here
	CView *pview=this->GetActiveView();
	pview->SendMessage(WM_SHOWRESULT,0,0);
}

void CMainFrame::KSort()
{
	int	i,j;
	int MAX=1000;
	BOOL change=true;//退出标志，false时表示样品所属类别不再变化，中止计算
	int counter=0;//记录当前已经循环的次数
	double distance;//到各中心的距离
	distance=MAX;
	patternnum=m_strPoint.GetLength()/4;
	
	m_center=new Center[m_CenterNum];
	m_pattern=new Pattern[patternnum];
	m_Point=new CPoint[patternnum];
	CString temp;
	for ( i=0;i<patternnum;i++)
	{
		temp=m_strPoint.Mid(i*4,3);
		m_Point[i].x=atoi(temp.Left(1));
		m_Point[i].y=atoi(temp.Right(1));
		m_pattern[i].distance=MAX;
		m_pattern[i].point=m_Point[i];
		m_pattern[i].category=i+1;
	}
	for ( i=0;i<m_CenterNum;i++)//初始化，前centernum个模版各自分为一类
	{
		m_pattern[i].category=i+1;
		m_pattern[i].distance=0;
		for ( j=0;j<m_CenterNum;j++)
			m_center[i].point=m_pattern[i].point;
		m_center[i].index=i+1;
	}

	while (change && counter<m_IterativeNum)
	{
		counter++;
		change=false;
		for( i=0;i<patternnum;i++)//对所有样品重新归类
		{
			//计算第i个模式到各个聚类中心的最小距离，
			int index=1;
			distance=MAX;
            
			for (int j=0;j<m_CenterNum;j++)
			{
				if (distance>GetDistance(m_pattern[i],m_center[j]))
				{
					distance=GetDistance(m_pattern[i],m_center[j]);
					index=j;//找到最小距离,是到第index个聚类中心的距离
				}
			}
			//比较原中心号与新中心号
			//相同：更新距离。
			//不同：1，新距离小，则归入新中心，更新距离，重新计算前后两个聚类中心模式
			//2，新距离大于原距离，不处理；
			//counter＋＋
			if (m_pattern[i].category==m_center[index].index)//属于原类
			{
				m_pattern[i].distance=distance;
			}
			else//不属于原类
			{
				int tpcenter=m_pattern[i].category;//记录原类号
				m_pattern[i].category=m_center[index].index;//归入新类
				m_pattern[i].distance=distance;
				if(tpcenter!=0)
				{
					for (int k=0;k<m_CenterNum;k++)
						if (m_center[k].index==tpcenter)
							CalCenter(&m_center[k]);//计算原属类中心
				}
				CalCenter(&m_center[index]);//计算新属类中心
				change=true;
			}
		}		
	}//end of while 
	delete []m_center;
}

double CMainFrame::GetDistance(Pattern pattern,Center center)
{
	CPoint point1,point2;
	double distance;
	point1=pattern.point;
	point2=center.point;
	distance=sqrt((point1.x-point2.x)*(point1.x-point2.x)+(point1.y-point2.y)*(point1.y-point2.y));
	return distance;
}

void CMainFrame::CalCenter(Center *pcenter)
{
	CPoint point;//临时存储中心的特征值
	int a=0;//记录该类中元素个数
	point.x=0;
	point.y=0;
	for (int i=0;i<patternnum;i++)
		if (m_pattern[i].category==pcenter->index)//累加中心所有样品
		{
			a++;			
			point.x+=m_pattern[i].point.x;
			point.y+=m_pattern[i].point.y;
		}
	pcenter->patternnum=a;	
	if(a!=0)
	{
		pcenter->point.x=point.x/a;
		pcenter->point.y=point.y/a;
	}
	else
	{
		pcenter->point=point;
	}
}