前些日子,我在做程序中遇到这样一个问题,要用树型控件快速浏览指定文件夹中的文件,经过一番周折,终于做出来了,想到有些仁兄可能遇到和我相同的困难,所以将我的做法写出来,希望对大家有些帮助!
基本原理是用了函数的递归调用,再加入参数跟踪。我在调试程序的时候,跟踪发现:程序只有在遍历完一个目录下所有子目录后,才返回同级的目录。这样便可以用参数的值来决定目录的层次。
//CalNum-用来记录第几层目录,strFile-路径
并添加以下内容:
基本原理是用了函数的递归调用,再加入参数跟踪。我在调试程序的时候,跟踪发现:程序只有在遍历完一个目录下所有子目录后,才返回同级的目录。这样便可以用参数的值来决定目录的层次。
isdot是用来判断是否./或者../,每个文件夹都有这样两个隐藏的目录,代表当前目录与上级目录,若不排除这两种情况将导致死循环。
(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,"课程表");//遍历"课程表"文件夹内的所有目录 }