用VC++遍历目录

用VC++遍历目录

--- 所谓遍历目录，就是给定一个目录，访问其中的所有文件(包括子目录下的文件)。迭代是比较常用的遍历算法。本文利用C++面向对象的特性，通过一个类CBrowseDir，对目录遍历进行了封装。用户只需掌握该类四个成员函数的用法，就可以在自己的程序中，很方便地实现目录遍历。

---- 类CBrowseDir使用了迭代算法。因为算法不是本文重点，笔者不打算展开进一步讨论，对其感兴趣者可参考相关资料。

一、类成员函数说明：

---- bool SetInitDir(const char *dir);

---- 功能：设置要遍历的目录。

---- 参数：dir 指向要遍历的目录，可以使用相对路径，比如"d:../hawk"；还可以使用网络路径，比如"//wf/d/hawk"(其中wf是主机名，d是共享目录，hawk是目录)。

---- 返回值：返回true，表示设置成功；返回false，说明目录不可用。

---- bool BeginBrowse(const char *filespec);

---- 功能：开始遍历目录中由filespec指定的文件(包括隐藏文件)。

---- 参数：filespec 指定文件类型，可以使用通配符*和?，比如"*.exe"或"a?.*"都是合法参数。注意：filespec中不能包含路径，象"hawk/*.*"是错误的。

---- 返回值：函数返回true，表明已顺利遍历完所有文件；返回false，遍历过程被用户中止。

---- virtual bool ProcessFile(const char *filename);

---- 功能：虚函数。每找到一个文件，程序就会调用ProcessFile，并把文件名作为参数传递给函数。如果函数返回false，则强制遍历中止，并导致类成员函数函数BeginBrowse返回false。 用户应该覆写此函数，以加入自己的处理代码。

---- 参数：filename 指向一个文件名。注意：filename使用绝对路径。

---- 返回值：返回true，继续遍历；否则，中止遍历。

---- virtual void ProcessDir (const char *currentdir,const char *parentdir);

---- 功能：虚函数。在遍历过程中，每进入一个子目录，程序就会调用ProcessDir，并把目录名及其上一级目录名作为参数传递给函数。如果该目录是成员函数SetInitDir指定的初始目录，则parentdir=NULL。用户可以覆写此函数，以加入自己的处理代码。比如可以在这里统计子目录的个数。

---- 参数：currentdir 指向一个子目录。
---- parentdir 指向currentdir的父目录。
---- 注意：currentdir和parentdir均使用绝对路径。

二、使用：

---- 把类CBrowseDir的头文件BrowseDir.h及实现文件BrowseDir.cpp加到项目(Project)中，然后派生自己的类并覆写虚函数ProcessFile和ProcessDir。遍历目录时，先构造一个派生类对象，用成员函数SetInitDir指定目录，然后调用BeginBrowse开始遍历。

---- 本文提供了一个例子 example.cpp，它从CBrowseDir派生出子类CStatDir，通过统计函数ProcessFile及ProcessDir的调用次数，可以得知目录中的文件及子目录个数。程序都有注释，这里就不再罗嗦了。

三、注意事项：

---- 1. 类CBrowseDir会改变当前工作目录。同一个相对路径，使用CBrowseDir前后，可能会有不同的含义。因此用户编程时，要小心使用相对路径。

---- 2. 如果项目(Project)是一个MFC应用程序，直接加入BrowseDir.h及BrowseDir.cpp会导致编译出错。这是因为缺省情况下，MFC项目使用了预编译头(Precompiled Header)，而BrowseDir.h和BrowseDir.cpp是用标准C++语句编写的，没用预编译。一个解决办法是先用类向导生成类CBrowseDir的"架子"，再把相应的代码拷贝过去。

---- 本文代码均在Win95、Visual C++ 5.0环境下调试通过。

附源代码：

/**************************************************
这是CBrowseDir的类定义文件 BrowseDir.h

/**************************************************
#include "stdlib.h"

class CBrowseDir
{
protected:
//存放初始目录的绝对路径，以'/'结尾
char m_szInitDir[_MAX_PATH];

public:
//缺省构造器
CBrowseDir();

//设置初始目录为dir，如果返回false，表示目录不可用
bool SetInitDir(const char *dir);

//开始遍历初始目录及其子目录下由filespec指定类型的文件
//filespec可以使用通配符 * ?，不能包含路径。
//如果返回false，表示遍历过程被用户中止
bool BeginBrowse(const char *filespec);

protected:
//遍历目录dir下由filespec指定的文件
//对于子目录,采用迭代的方法
//如果返回false,表示中止遍历文件
bool BrowseDir(const char *dir,const char *filespec);

//函数BrowseDir每找到一个文件,就调用ProcessFile
//并把文件名作为参数传递过去
//如果返回false,表示中止遍历文件
//用户可以覆写该函数,加入自己的处理代码
virtual bool ProcessFile(const char *filename);

//函数BrowseDir每进入一个目录,就调用ProcessDir
//并把正在处理的目录名及上一级目录名作为参数传递过去
//如果正在处理的是初始目录,则parentdir=NULL
//用户可以覆写该函数,加入自己的处理代码
//比如用户可以在这里统计子目录的个数
virtual void ProcessDir(const char
*currentdir,const char *parentdir);
};

/*********************************************/

这是CBrowseDir的类实现文件 BrowseDir.cpp

/***********************************************/
#include "stdlib.h"
#include "direct.h"
#include "string.h"
#include "io.h"

#include "browsedir.h"

CBrowseDir::CBrowseDir()
{
//用当前目录初始化m_szInitDir
getcwd(m_szInitDir,_MAX_PATH);

//如果目录的最后一个字母不是'/',则在最后加上一个'/'
int len=strlen(m_szInitDir);
if (m_szInitDir[len-1] != '//')
strcat(m_szInitDir,"//");
}

bool CBrowseDir::SetInitDir(const char *dir)
{
//先把dir转换为绝对路径
if (_fullpath(m_szInitDir,dir,_MAX_PATH) == NULL)
return false;

//判断目录是否存在
if (_chdir(m_szInitDir) != 0)
return false;

//如果目录的最后一个字母不是'/',则在最后加上一个'/'
int len=strlen(m_szInitDir);
if (m_szInitDir[len-1] != '//')
strcat(m_szInitDir,"//");

return true;
}

bool CBrowseDir::BeginBrowse(const char *filespec)
{
ProcessDir(m_szInitDir,NULL);
return BrowseDir(m_szInitDir,filespec);
}

bool CBrowseDir::BrowseDir
(const char *dir,const char *filespec)
{
_chdir(dir);

//首先查找dir中符合要求的文件
long hFile;
_finddata_t fileinfo;
if ((hFile=_findfirst(filespec,&fileinfo)) != -1)
{
do
{
//检查是不是目录
//如果不是,则进行处理
if (!(fileinfo.attrib & _A_SUBDIR))
{
char filename[_MAX_PATH];
strcpy(filename,dir);
strcat(filename,fileinfo.name);
if (!ProcessFile(filename))
return false;
}
} while (_findnext(hFile,&fileinfo) == 0);
_findclose(hFile);
}

//查找dir中的子目录
//因为在处理dir中的文件时，派生类的ProcessFile有可能改变了
//当前目录，因此还要重新设置当前目录为dir。
//执行过_findfirst后，可能系统记录下了相关信息，因此改变目录
//对_findnext没有影响。
_chdir(dir);
if ((hFile=_findfirst("*.*",&fileinfo)) != -1)
{
do
{
//检查是不是目录
//如果是,再检查是不是 . 或 ..
//如果不是,进行迭代
if ((fileinfo.attrib & _A_SUBDIR))
{
if (strcmp(fileinfo.name,".") != 0 && strcmp
(fileinfo.name,"..") != 0)
{
char subdir[_MAX_PATH];
strcpy(subdir,dir);
strcat(subdir,fileinfo.name);
strcat(subdir,"//");
ProcessDir(subdir,dir);
if (!BrowseDir(subdir,filespec))
return false;
}
}
} while (_findnext(hFile,&fileinfo) == 0);
_findclose(hFile);
}
return true;
}

bool CBrowseDir::ProcessFile(const char *filename)
{
return true;
}

void CBrowseDir::ProcessDir(const char
*currentdir,const char *parentdir)
{
}

/*************************************************
这是例子example.cpp

/*************************************************
#include "stdio.h"

#include "BrowseDir.h"

//从CBrowseDir派生出的子类，用来统计目录中的文件及子目录个数
class CStatDir:public CBrowseDir
{
protected:
int m_nFileCount; //保存文件个数
int m_nSubdirCount; //保存子目录个数

public:
//缺省构造器
CStatDir()
{
//初始化数据成员m_nFileCount和m_nSubdirCount
m_nFileCount=m_nSubdirCount=0;
}

//返回文件个数
int GetFileCount()
{
return m_nFileCount;
}

//返回子目录个数
int GetSubdirCount()
{
//因为进入初始目录时，也会调用函数ProcessDir，
//所以减1后才是真正的子目录个数。
return m_nSubdirCount-1;
}

protected:
//覆写虚函数ProcessFile，每调用一次，文件个数加1
virtual bool ProcessFile(const char *filename)
{
m_nFileCount++;
return CBrowseDir::ProcessFile(filename);
}

//覆写虚函数ProcessDir，每调用一次，子目录个数加1
virtual void ProcessDir
(const char *currentdir,const char *parentdir)
{
m_nSubdirCount++;
CBrowseDir::ProcessDir(currentdir,parentdir);
}
};

void main()
{
//获取目录名
char buf[256];
printf("请输入要统计的目录名:");
gets(buf);

//构造类对象
CStatDir statdir;

//设置要遍历的目录
if (!statdir.SetInitDir(buf))
{
puts("目录不存在。");
return;
}

//开始遍历
statdir.BeginBrowse("*.*");

//统计结果中，子目录个数不含 . 及 ..
printf("文件总数: %d/n子目录总数:
%d/n",statdir.GetFileCount(),
statdir.GetSubdirCount());
} 
 

回复: 用VC++遍历目录
作 者: alanlong    短消息  
回复主题: 用VC++遍历目录  2006-9-18 上午11:23  

回复 1 of 1  回复 
 
 
　在应用程序的开发过程中，经常会遇到如何查找某一文件以确定此文件路径的问题。利用CFileFind类可以比较方便地在当前目录下进行文件查找，但却不能对其子目录中的文件进行搜寻。而实际应用中往往需要对某一整个目录树，甚至是整个C盘或D盘驱动器进行文件搜寻。通过实践，我们在Visual C＋＋ 6.0中编程实现了如何遍历任意目录树，以查找某一特定的文件。 　　在下面的具体陈述中可以看到，在确定要查找的文件名和要进行搜索的目录的名称后，将调用函数Search_Directory进行文件的查找。首先依次查找当前目录下的每一个实体（文件或是子目录），如果是某一子目录，则进入该子目录并递归调用函数Search_Dirctory进行查找，查找完毕之后, 再返回上一级目录；如果不是子目录而是某一文件，则判断其是否就是我们要查找的文件，如果是则输出其完整的文件路径。这样，通过Search_Directory函数的反复递归调用，就可以实现对整个目录，包括子目录的遍历搜索。下面将举例详细讲述如何在VC＋＋中编程实现在整个目录树中的文件查找。

　　1．在Visual C＋＋ 6.0（VC＋＋ 5.0与之类似）中用默认方式创建了一基于对话框的应用程序Search。在主窗口对话框上放置一命令按钮，其Caption为“Search File”，ID为ID—BUTTON—SEARCH。单击此按钮将完成文件的查找工作。

　　2．利用ClassWizard为“Search File”按钮的BN_CLICKED 事件添加处理函数OnButtonSearch，代码如下：

　　＃include 〈direct.h〉

　　＃include 〈io.h〉

　　......

　　void CSearchDlg::OnButtonSearch()

　　{

// TODO: Add your control notification handler code here

char szFilename[80];

　　// 字符串 szFilename 表示要查找的文件名

strcpy(szFilename,″Mytext.txt″);

_chdir(″d://″); // 进入要查找的路径（也可为某一具体的目录）

　　// 查找文件, 如果查到则显示文件的路径全名

Search_Directory(szFilename);

　　// 为CSearchDlg类的一成员函数

MessageBox(″查找文件完毕!″);

　　// 显示查找完毕的信息

　　}

　　3．在CSearchDlg类中增加成员函数Search_Directory，它将完成具体的文件查找工作，代码如下：

　　void CSearchDlg::Search_Directory(char＊ szFilename)

　　{

long handle;

struct _finddata_t filestruct;　　

　　//表示文件(或目录)的信息

char path_search[_MAX_PATH];

　　//表示查找到的路径结果

　　// 开始查找工作, 找到当前目录下的第一个实体(文件或子目录)，

　　// ″＊″表示查找任何的文件或子目录, filestruct为查找结果

handle = _findfirst(″＊″, ＆filestruct);

　　// 如果handle为－1, 表示当前目录为空, 则结束查找而返回

if((handle == －1)) return;

　　// 检查找到的第一个实体是否是一个目录(filestruct.name为其名称)

if( ::GetFileAttributes(filestruct.name) ＆ FILE—ATTRIBUTE—DIRECTORY )

{

　　// 如果是目录, 则进入该目录并递归调用函数Search_Dirctory进行查找，

　　// 注意: 如果目录名的首字符为′.′(即为″.″或″..″), 则不用进行查找

if( filestruct.name[0] != ′.′ )

{

—chdir(filestruct.name);

Search_Directory(szFilename);

　　// 查找完毕之后, 返回上一级目录

—chdir(″..″);

}

}

else // 如果第一个实体不是目录, 则检查是否是要查找的文件

{

　　// stricmp对两字符串进行小写形式的对比, 返回为0表示完全一致

if( !stricmp(filestruct.name, szFilename) )

{

　　// 先获得当前工作目录的全路径

　　　　—getcwd(path_search,—MAX—PATH);

　　// 再获得文件的完整的路径名(包含文件的名称)

strcat(path_search,″//″);

strcat(path—search,filestruct.name);

MessageBox(path_search); //输出显示

}

}

　　// 继续对当前目录中的下一个子目录或文件进行与上面同样的查找

while(!(—findnext(handle,＆filestruct)))

{

　　if( ::GetFileAttributes(filestruct.name) ＆ FILE—ATTRIBUTE—DIRECTORY )

{

if(＊filestruct.name != ′.′)

{

—chdir(filestruct.name);

Search_Directory(szFilename);

—chdir(″..″);

}

}

else

{

if(!stricmp(filestruct.name,szFilename))

{

—getcwd(path—search,—MAX—PATH);

strcat(path_search,″//″);

strcat(path_search,filestruct.name);

MessageBox(path_search);

}

}

}

—findclose(handle);

　　// 最后结束整个查找工作

　　}

　　这样我们就可以对整个目录进行遍历搜索，查找某一特定的文件，并输出显示其完整的文件路径。以上的程序在Visual C＋＋ 6.0中已调试通过