C++如何实现遍历目录下文件

发布时间:2021-07-10 14:47:23 作者:小新
来源:亿速云 阅读:219

这篇文章主要介绍了C++如何实现遍历目录下文件,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。

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

已在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("文件总数: %d\n",file_vec.size()); 
  system("pause"); 
}

感谢你能够认真阅读完这篇文章,希望小编分享的“C++如何实现遍历目录下文件”这篇文章对大家有帮助,同时也希望大家多多支持亿速云,关注亿速云行业资讯频道,更多相关知识等着你来学习!

推荐阅读:
  1. java怎么遍历目录下的所有文件
  2. python 遍历目录(包括子目录)下所有文件的实例

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

c++

上一篇:如何安装Docker并部署 Spring Boot 项目

下一篇:Python中怎么实现一个单平台均衡策略

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》