遍历文件夹并建成目录树

遍历文件夹并建成目录树
作者：

前些日子，我在做程序中遇到这样一个问题，要用树型控件快速浏览指定文件夹中的文件，经过一番周折，终于做出来了，想到有些仁兄可能遇到和我相同的困难，所以将我的做法写出来，希望对大家有些帮助！


(本文程序运行效果图)

基本原理是用了函数的递归调用，再加入参数跟踪。我在调试程序的时候，跟踪发现：程序只有在遍历完一个目录下所有子目录后，才返回同级的目录。这样便可以用参数的值来决定目录的层次。

(1)在对话框上添加树型控件，选择Style中Has Buttons、Has Lines、Lines At Root。
(2)定义变量：

HTREEITEM strRoot; 	//根目录
HTREEITEM strHTFir;	//第一层
HTREEITEM strHTSec;	//第二层
HTREEITEM strHtThi;	//第三层
HTREEITEM strHtFor;	//第四层
HTREEITEM strHtFif;	//第五层
// 没有找到更好的方法，所以只能这样来显示，估计一下大约能用到几层，如果找到好方法，请告诉我，非常感谢！
HTREEITEM strHtEnd;	//文件目录

(3)添加遍历文件夹的函数BrowseFile(int CallNum, CString strFile)
//CalNum－用来记录第几层目录，strFile－路径
并添加以下内容：

void CFileTreeDlg::BrowseFile(int CallNum, CString strFile)
{
	CallNum++;
	CFileFind ff;
	CString szDir = strFile;
	
	if(szDir.Right(1) != "\\")
		szDir += "\\";
	
	szDir += "*.*";
	
	BOOL res = ff.FindFile(szDir);
	while(res)
	{
		res = ff.FindNextFile();
		if(ff.IsDirectory() && !ff.IsDots())//目录是文件夹
		{
			//如果是一个子目录，用递归继续往深一层找

			CString strPath = ff.GetFilePath(); //得到路径，做为递归调用的开始
			CString strTitle = ff.GetFileTitle();//得到目录名，做为树控的结点
			int i =0;
			switch(CallNum)
			{
				case 1:
					strHTFir = m_FileTree.InsertItem(strTitle,0,0,NULL);		
					break;
				case 2:
					strHTSec = m_FileTree.InsertItem(strTitle,0,0,strHTFir);	
					break;
				case 3:
					strHtThi = m_FileTree.InsertItem(strTitle,0,0,strHTSec);
					break;
				case 4:
					strHtFor = m_FileTree.InsertItem(strTitle,0,0,strHtThi);		
					break;
				default:
					strHtFif = m_FileTree.InsertItem(strTitle,0,0,strHtFor);
					break;					
			}
			BrowseFile(CallNum,strPath);//递归调用
		}
		else if(!ff.IsDirectory() && !ff.IsDots())//到达最低层的文件
		{
			//显示当前访问的文件
			CString strPath;
			CString strTitle;
			strPath = ff.GetFilePath();
			strTitle = ff.GetFileTitle();
			switch(CallNum)
			{
			case 1:
				strRoot = m_FileTree.InsertItem(strTitle,0,0,NULL);
				break;
			case 2:
				strHtEnd = m_FileTree.InsertItem(strTitle,0,0,strHTFir);
				break;
			case 3:
				strHtEnd = m_FileTree.InsertItem(strTitle,0,0,strHTSec);		
				break;
			case 4:
				strHtEnd = m_FileTree.InsertItem(strTitle,0,0,strHtThi);
				break;
			case 5:
				strHtEnd = m_FileTree.InsertItem(strTitle,0,0,strHtFor);
				break;
			default:
				strHtEnd = m_FileTree.InsertItem(strTitle,0,0,strHtFif);
				break;
			}
		}
	}
	ff.Close();//关闭
}

(4)在OnInitDialog()函数中设置树控件的图标并调用自定义函数BrowseFile()

BOOL CFileTreeDlg::OnInitDialog()
{
	m_iImageList.Create(24, 24, TRUE,1, 0);	//创建图标链
	HICON hIcon = NULL;
	hIcon = (HICON)::LoadImage(::AfxGetInstanceHandle(), 
		MAKEINTRESOURCE(IDI_KEBIAO), IMAGE_ICON, 24, 24, 0);
	m_iImageList.Add(hIcon);
	m_FileTree.SetImageList ( &m_iImageList,TVSIL_NORMAL );
	BrowseFile(0,"课程表");//遍历"课程表"文件夹内的所有目录
}

(全文完)
--------------------next---------------------

thank you very much~~~~~~~~ ( benny_pan1980 发表于 2005-5-22 11:32:00)
 
居然不用shell？ ( VC70新手 发表于 2004-1-11 4:11:00)
 
BrowseFile(HTREEITEM hItem, CString strFile)
{
  HTREEITEM hTempItem;
  ....
  //文件夹
  hTempItem=m_FileTree.InsertItem(strTitle,0,0,hItem);
  BrowseFile(hTempItem,strPath);
  //文件
  m_FileTree.InserItem(strTitle,0,0,hItem);
  ....
} ( zhonghaishan 发表于 2003-7-21 17:42:00)
 
谢谢楼上各位的帮忙！大二了，我们学校要考四级了，鄙人最近不得不忍痛割爱，去忙于准备四级，谢谢大家给我的补充！！！！！！ ( lsjpyx 发表于 2003-4-9 12:51:00)
 
HTREEITEM suu[10];
if(CallNum==1)
suu[CallNum] = m_FileTree.InsertItem(strTitle,0,0,NULL);
else{
suu[CallNum] = m_FileTree.InsertItem(strTitle,0,0,suu[CallNum-1]);
}
BrowseFile(CallNum,strPath);
}
else if(!ff.IsDirectory() && !ff.IsDots())
{
//显示当前访问的文件
CString strPath;
CString strTitle;
strPath = ff.GetFilePath();
strTitle = ff.GetFileTitle();
if(CallNum==1)
suu[CallNum] = m_FileTree.InsertItem(strTitle,0,0,NULL);
else{
suu[CallNum] = m_FileTree.InsertItem(strTitle,0,0,suu[CallNum-1]);
}
}不知道这样改一下是不是能提高效率!!!!!!! ( byb 发表于 2003-4-8 17:20:00)
 
maybe use HTREEITEM as the parameter in BrowseFile
function.

good luck! ( proceedings 发表于 2003-4-2 20:35:00)
 
.......................................................

--------------------next---------------------