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("文件总数: %d\n子目录总数:%d\n",statdir.GetFileCount(),statdir.GetSubdirCount());  }