tree.cpp

our homework big project

// tree.cpp : 实现文件
//

#include "stdafx.h"
#include "sql05.h"
#include "tree.h"
#include "tree.h"


// Ctree

IMPLEMENT_DYNCREATE(Ctree, CTreeView)
Ctree::Ctree()
{
}

Ctree::~Ctree()
{
}

BEGIN_MESSAGE_MAP(Ctree, CTreeView)
	ON_COMMAND(ID_REFLESH, OnReflesh)     //刷新的响应
END_MESSAGE_MAP()


// Ctree 诊断

#ifdef _DEBUG
void Ctree::AssertValid() const
{
	CTreeView::AssertValid();
}

void Ctree::Dump(CDumpContext& dc) const
{
	CTreeView::Dump(dc);
}
#endif //_DEBUG


// Ctree 消息处理程序



void Ctree::OnInitialUpdate()
{
	CTreeView::OnInitialUpdate();
	m_imagelist.Create(IDB_BITMAP1,16,1,RGB(255,0,255));
	this->GetTreeCtrl().SetImageList(&m_imagelist,TVSIL_NORMAL);
	
	HTREEITEM h1 = GetTreeCtrl().InsertItem("主要语法关键词",0,0);		//关键词的tree
	
	GetTreeCtrl().InsertItem("create",1,1,h1);
	GetTreeCtrl().InsertItem("drop",1,1,h1);
	GetTreeCtrl().InsertItem("insert",1,1,h1);
	GetTreeCtrl().InsertItem("delete",1,1,h1);
	GetTreeCtrl().InsertItem("select",1,1,h1);

	CreateTree();	
}

void Ctree::CreateTree()
{
    h2 = GetTreeCtrl().InsertItem("表名",0,0);
	WIN32_FIND_DATA fd;
	HANDLE hFind = ::FindFirstFile("data\\*.recd",&fd);
	
	CString name;

	//遍历当前的DATA文件夹下的*.recd记录，并添加到tree里

	if(hFind!= INVALID_HANDLE_VALUE){
		do{
			if(!(fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)){
				name=fd.cFileName;
				name=name.Left(name.GetLength()-5);
				GetTreeCtrl().InsertItem(name ,1,1,h2);
			}
			
		}while(::FindNextFile(hFind,&fd));
		::FindClose(hFind);
	}

}

//刷新函数

void Ctree::OnReflesh()
{
	this->GetTreeCtrl().DeleteItem(h2);
	CreateTree();

}