代码驿站移动版

当前位置：主页 > 软件编程 > C代码 >

C++ 遍历目录下文件简单实现实例

时间：2021-02-20 09:06:11 | 栏目：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"); 
}

感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！

您可能感兴趣的文章:

相关文章