遍历文件夹并建成目录树

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

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

首先,在对话框上添加树型控件,选择Style中Has Buttons、Has Lines、Lines At Root。定义变量:

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

添加遍历文件夹的函数

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();//关闭
}

在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,"课程表");//遍历"课程表"文件夹内的所有目录
}

转载于:https://www.cnblogs.com/rogee/archive/2011/03/31/2000482.html

你可能感兴趣的:(遍历文件夹并建成目录树)