Windows下使用C++获取目录及子目录下所有文件

转载自：

http://read.pudn.com/downloads192/sourcecode/math/903237/贝叶斯邮件过滤源程序(VC%2B%2B实现的源程序)/ReadDir/BrowseDir.h__.htm

http://blog.csdn.net/abcjennifer/article/details/18147551

BrowseDir.h

#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);};

BrowseDir.cpp

#include "BrowseDir.h"CBrowseDir::CBrowseDir(){//用当前目录初始化m_szInitDirgetcwd(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;//转换目录到m_szInitDir下，判断目录是否存在，若目录不存在返回0if (_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){}

StatDir.h

#include "BrowseDir.h"CBrowseDir::CBrowseDir(){//用当前目录初始化m_szInitDirgetcwd(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;//转换目录到m_szInitDir下，判断目录是否存在，若目录不存在返回0if (_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){}

StatDir.cpp

#include "StatDir.h"//缺省构造器CStatDir::CStatDir(){//初始化数据成员m_nFileCount和m_nSubdirCountm_nFileCount=m_nSubdirCount=0;}//返回文件个数int CStatDir::GetFileCount(){return m_nFileCount;}//返回子目录个数int CStatDir::GetSubdirCount(){//因为进入初始目录时，也会调用函数ProcessDir，//所以减1后才是真正的子目录个数。return m_nSubdirCount-1;}//覆写虚函数ProcessFile，每调用一次，文件个数加1bool CStatDir::ProcessFile(const char *filename){m_nFileCount++;return CBrowseDir::ProcessFile(filename);}//覆写虚函数ProcessDir，每调用一次，子目录个数加1void CStatDir::ProcessDir(const char *currentdir,const char *parentdir){m_nSubdirCount++;CBrowseDir::ProcessDir(currentdir,parentdir);}

main.cpp

#include <stdlib.h>#include <direct.h>#include <string.h>#include <string>#include <io.h>#include <stdio.h>#include <vector>#include <iostream>#include "StatDir.h"using namespace std;int main(){//获取目录名char buf[256];printf("请输入要统计的目录名:");gets(buf);//构造类对象CStatDir statdir;//设置要遍历的目录if (!statdir.SetInitDir(buf)){puts("目录不存在。");return 0;}//开始遍历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("文件总数: %d\n",file_vec.size());printf("子目录数：%d\n", statdir.GetSubdirCount());system("pause");return 0;}