mainfrm.cpp

B+树的演示程序

// MainFrm.cpp : CMainFrame 类的实现
//

#include "stdafx.h"
#include "BPlusTree.h"

#include "MainFrm.h"
#include ".\mainfrm.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CMainFrame

IMPLEMENT_DYNAMIC(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	ON_WM_CREATE()
	ON_WM_SETFOCUS()
END_MESSAGE_MAP()

static UINT indicators[] =
{
	ID_SEPARATOR,           // 状态行指示器
	ID_INDICATOR_CAPS,
	ID_INDICATOR_NUM,
	ID_INDICATOR_SCRL,
};


// CMainFrame 构造/析构

CMainFrame::CMainFrame()
{
	// TODO: 在此添加成员初始化代码
	m_hItemNode = NULL;
	m_isDlgShow = true;
	m_isTreeShow = true;
	m_BView = TRUE;
}

CMainFrame::~CMainFrame()
{
	// m_wndDlg->DestroyWindow();
	// delete m_wndDlg;
}


int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;
	// 创建一个视图以占用框架的工作区
	if (!m_wndView.Create(NULL, NULL, AFX_WS_DEFAULT_VIEW,
		CRect(0, 0, 0, 0), this, AFX_IDW_PANE_FIRST, NULL))
	{
		TRACE0("未能创建视图窗口\n");
		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("未能创建工具栏\n");
		return -1;      // 未能创建
	}

	/*
	if (!m_wndDlgBar.Create(IDR_MAINFRAME,this))
	//	CBRS_ALIGN_TOP, AFX_IDW_DIALOGBAR))
	{
		TRACE0("未能创建对话栏\n");
		return -1;		// 未能创建
	}

	if (!m_wndReBar.Create(this) ||
		!m_wndReBar.AddBar(&m_wndToolBar) ||
		!m_wndReBar.AddBar(&m_wndDlgBar))
	{
		TRACE0("未能创建 Rebar\n");
		return -1;      // 未能创建
	}	
	*/

	/*
	m_wndDlg = new MyTreeDlg(this);	
	if(!m_wndDlg)
	{
		TRACE0("no dlg\n");
		return -1;
	}
	if (!m_wndDlg->Create(IDD_DLG_TREE,this))
	{
		TRACE0("未能创建 Rebar\n");
		return -1;      // 未能创建
	}
	m_wndDlg->SetWindowPos(&CWnd::wndTop,400,120,150,400,SWP_NOZORDER|SWP_SHOWWINDOW);
	m_wndDlg->ShowWindow(SW_SHOW);
	

	if (!m_wndTree.Create(WS_CHILD|WS_VISIBLE|TVS_HASLINES|TVS_HASBUTTONS|TVS_LINESATROOT,CRect(0, 0, 0, 0), m_wndDlg, 111))
	{
		TRACE0("Failed to create instant bar child\n");
		return -1;
	}*/

	if (!m_coolbarDlg.Create(_T("coolbar dialog"),this,CSize(600,60),TRUE,111))
	{
		TRACE0("未能创建coolbar dialog\n");
		return -1;      // 未能创建
	}
	if (!m_wndDlg.Create(IDR_MAINFRAME,&m_coolbarDlg))
	{
		TRACE0("未能创建dialog\n");
		return -1;      // 未能创建
	}
	m_wndDlg.ShowWindow(SW_SHOW);

	if (!m_coolbarTree.Create(_T("coolbar tree"),this,CSize(150,500),TRUE,112))
	{
		TRACE0("Failed to create mybar\n");
		return -1;
	}
	if (!m_wndTree.Create(WS_CHILD|WS_VISIBLE|TVS_HASLINES|TVS_HASBUTTONS|TVS_LINESATROOT,CRect(0, 0, 0, 0), &m_coolbarTree, 113))
	{
		TRACE0("Failed to create instant bar child\n");
		return -1;
	}
	m_wndTree.ModifyStyleEx(0, WS_EX_CLIENTEDGE);


	if (!m_wndStatusBar.Create(this) ||
		!m_wndStatusBar.SetIndicators(indicators,
		  sizeof(indicators)/sizeof(UINT)))
	{
		TRACE0("未能创建状态栏\n");
		return -1;      // 未能创建
	}
	// TODO: 如果不需要工具提示，则将此移除
	m_wndToolBar.SetBarStyle(m_wndToolBar.GetBarStyle() | CBRS_TOOLTIPS | CBRS_FLYBY);

	m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
	EnableDocking(CBRS_ALIGN_ANY);
	DockControlBar(&m_wndToolBar,AFX_IDW_DOCKBAR_TOP);

	m_coolbarDlg.SetBarStyle(m_coolbarDlg.GetBarStyle() | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC); 
	m_coolbarTree.SetBarStyle(m_coolbarTree.GetBarStyle() | CBRS_TOOLTIPS |	CBRS_FLYBY | CBRS_SIZE_DYNAMIC);
	m_coolbarDlg.EnableDocking(CBRS_ALIGN_ANY);
	m_coolbarTree.EnableDocking(CBRS_ALIGN_ANY);
	DockControlBar(&m_coolbarDlg, AFX_IDW_DOCKBAR_TOP);
	DockControlBar(&m_coolbarTree, AFX_IDW_DOCKBAR_RIGHT);

	m_control.SetParentWnd(this);

	return 0;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: 在此处通过修改 CREATESTRUCT cs 来修改窗口类或
	// 样式

	cs.style = WS_OVERLAPPED | WS_CAPTION | FWS_ADDTOTITLE
		 | WS_THICKFRAME | WS_MINIMIZEBOX | WS_MAXIMIZEBOX | WS_MAXIMIZE | WS_SYSMENU;

	cs.dwExStyle &= ~WS_EX_CLIENTEDGE;
	cs.lpszClass = AfxRegisterWndClass(0);
	return TRUE;
}


// CMainFrame 诊断

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

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

#endif //_DEBUG


// CMainFrame 消息处理程序

void CMainFrame::OnSetFocus(CWnd* /*pOldWnd*/)
{
	// 将焦点前移到视图窗口
	m_wndView.SetFocus();
}

BOOL CMainFrame::OnCmdMsg(UINT nID, int nCode, void* pExtra, AFX_CMDHANDLERINFO* pHandlerInfo)
{
	// 让视图第一次尝试该命令
	if (m_wndView.OnCmdMsg(nID, nCode, pExtra, pHandlerInfo))
		return TRUE;

	// 否则，执行默认处理
	return CFrameWnd::OnCmdMsg(nID, nCode, pExtra, pHandlerInfo);
}

MyBPlusTree* CMainFrame::GetBPlusTree(void)
{
	return &m_control.m_tree;
}

void CMainFrame::DisplayTree(void)
{
	m_wndView.m_drawKey0 = m_control.m_drawKey0;
	m_wndView.m_drawLevel = m_control.m_drawLevel;
	m_wndView.m_iType = m_control.m_iType;
	m_wndView.m_inserPoint = m_control.m_insertPoint;
	m_wndView.m_iNumber = m_control.m_iNumber;

	m_wndView.Invalidate();
	m_wndView.Display();
	if(m_control.m_bSleep)
	{
		if(m_BView)
			Sleep(1000);
		m_control.m_bSleep = false;
	}
	ListTree();
	CString str;
	str.Format("Now There are %d levels, %d nodes and %d keys in the B+ tree!",m_control.m_tree.m_levels,m_control.m_tree.m_nodes,m_control.m_tree.m_keys);
	m_wndStatusBar.SetPaneText(0,str);	
}

void CMainFrame::ListTree(void)
{
	m_wndTree.DeleteAllItems();

	if(!m_control.m_tree.m_root)	return;

	HTREEITEM hItem;
	hItem = ListNode(m_control.m_tree.m_root,TVI_ROOT);
	if(m_control.m_tree.m_levels > 1)
		ListSubNode(m_control.m_tree.m_root,hItem);
	m_wndTree.SelectSetFirstVisible(m_hItemNode);
}

HTREEITEM CMainFrame::ListNode(MyNode* pnode, HTREEITEM hItem)
{
	HTREEITEM hItemTree;
	CString str,temp;
	str = "[";

	for(int i=0;i<pnode->m_n-1;i++)
	{
		temp.Format("%d,",pnode->k[i]);
		str += temp;
	}
	temp.Format("%d]",pnode->k[i]);
	str += temp;

	hItemTree = m_wndTree.InsertItem(str,hItem);
	if(m_control.m_drawKey0 == pnode->k[0])
		m_hItemNode = hItemTree;
	return hItemTree;
}

void CMainFrame::ListSubNode(MyNode* pnode, HTREEITEM hItem)
{
	HTREEITEM hSubItem;
	
	for(int i=0;i<pnode->m_n+1;i++)
	{
		hSubItem = ListNode((MyNode*)pnode->p[i],hItem);
		if(pnode->m_level > 2)
			ListSubNode((MyNode*)pnode->p[i],hSubItem);
		m_wndTree.Expand(hItem,TVE_EXPAND);
	}
}