改进《遍历文件夹并建成目录树》
改进《遍历文件夹并建成目录树》
作者：miao baoli
刚刚看到VC知识库二十期中的一篇文章《遍历文件夹并建成目录树》。主要是用一个递归函数来扫描目标文件夹，但是这个函数还存在目录层数的限制。
我写了另外一个函数，实现的结果跟上文一模一样，但消除了目录层数的限制, 原代码与改进后的代码如下：
//原来的函数原型
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();//关闭
}
//原来的函数调用
BOOL CFileTreeDlg::OnInitDialog()
{
CDialog::OnInitDialog();
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
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,"成绩表");//遍历"成绩表"文件夹内的所有目录
return TRUE;
}
//改过后的函数原型
void CFileTreeDlg::BrowseDir( CString strDir, HTREEITEM parent )
{
CFileFind ff;
CString szDir = strDir;
HTREEITEM hSubItem;
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();
hSubItem =
m_FileTree.InsertItem( strTitle, 0, 0,parent );
BrowseDir( strPath, hSubItem );
}
else if(!ff.IsDirectory() && !ff.IsDots())
{
CString strTitle = ff.GetFileTitle();
m_FileTree.InsertItem( strTitle, 0, 0, parent );
}
}
ff.Close();
}
//改过后的函数调用
BOOL CFileTreeDlg::OnInitDialog()
{
CDialog::OnInitDialog();
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
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 );
BrowseDir( "成绩表", NULL );
return TRUE;
}(全文完)