C++ 遍历目录下文件
function:遍历目录下所有文件,返回文件总数,子文件夹总数(修改一下可以获得全部文件名等)。
实例代码:
#include "stdlib.h" #include "direct.h" #include "string.h" #include "io.h" #include "stdio.h" #include "iostream" using namespace std;
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::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);
cout << filename << endl;
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) {
}
//从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("文件总数: %dn子目录总数:%dn",statdir.GetFileCount(),statdir.GetSubdirCount());
}
已在windows上验证有效。
下面我加了BeginBrowseFilenames函数,以vector<char*>形式返回目录中所有文件名。
实例代码:
#include "stdlib.h" #include "direct.h" #include "string.h" #include "string" #include "io.h" #include "stdio.h" #include <vector> #include "iostream" using namespace std;
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);
vector<string> BeginBrowseFilenames(const char *filespec);
protected: //遍历目录dir下由filespec指定的文件 //对于子目录,采用迭代的方法 //如果返回false,表示中止遍历文件 bool BrowseDir(const char *dir,const char *filespec);
vector<string> GetDirFilenames(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::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;
}
vector<string> CBrowseDir::BeginBrowseFilenames(const char *filespec) {
ProcessDir(m_szInitDir,NULL);
return GetDirFilenames(m_szInitDir,filespec);
}
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);
cout << filename << endl;
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;
}
vector<string> CBrowseDir::GetDirFilenames(const char *dir,const char *filespec) {
_chdir(dir);
vector<string>filename_vector;
filename_vector.clear();
//首先查找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);
filename_vector.push_back(filename);
}
}
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);
vector<string>tmp= GetDirFilenames(subdir,filespec);
for (vector<string>::iterator it=tmp.begin();
it<tmp.end();
it++) {
filename_vector.push_back(*it);
}
}
}
}
while (_findnext(hFile,&fileinfo) == 0);
_findclose(hFile);
}
return filename_vector;
}
bool CBrowseDir::ProcessFile(const char *filename) {
return true;
}
void CBrowseDir::ProcessDir(const char *currentdir,const char *parentdir) {
}
//从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;
}
//开始遍历 vector<string>file_vec = statdir.BeginBrowseFilenames("*.*");
for(vector<string>::const_iterator it = file_vec.begin();
it < file_vec.end();
++it) std::cout<<*it<<std::endl;
printf("文件总数: %dn",file_vec.size());
system("pause");
}
感谢阅读,希望能帮助到大家,谢谢大家对本站的支持!