C++如何實(shí)現(xiàn)遍歷目錄下文件

這篇文章主要介紹了C++如何實(shí)現(xiàn)遍歷目錄下文件，具有一定借鑒價(jià)值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

C++ 遍歷目錄下文件

function：遍歷目錄下所有文件，返回文件總數(shù)，子文件夾總數(shù)（修改一下可以獲得全部文件名等）。

實(shí)例代碼：

#include "stdlib.h" 
#include "direct.h" 
#include "string.h" 
#include "io.h" 
#include "stdio.h"  
#include "iostream" 
using namespace std; 
 
class CBrowseDir 
{ 
protected: 
  //存放初始目錄的絕對路徑，以'\'結(jié)尾 
  char m_szInitDir[_MAX_PATH]; 
 
public: 
  //缺省構(gòu)造器 
  CBrowseDir(); 
 
  //設(shè)置初始目錄為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); 
 
  //函數(shù)BrowseDir每找到一個(gè)文件,就調(diào)用ProcessFile 
  //并把文件名作為參數(shù)傳遞過去 
  //如果返回false,表示中止遍歷文件 
  //用戶可以覆寫該函數(shù),加入自己的處理代碼 
  virtual bool ProcessFile(const char *filename); 
 
  //函數(shù)BrowseDir每進(jìn)入一個(gè)目錄,就調(diào)用ProcessDir 
  //并把正在處理的目錄名及上一級目錄名作為參數(shù)傳遞過去 
  //如果正在處理的是初始目錄,則parentdir=NULL 
  //用戶可以覆寫該函數(shù),加入自己的處理代碼 
  //比如用戶可以在這里統(tǒng)計(jì)子目錄的個(gè)數(shù) 
  virtual void ProcessDir(const char *currentdir,const char *parentdir); 
}; 
 
CBrowseDir::CBrowseDir() 
{ 
  //用當(dāng)前目錄初始化m_szInitDir 
  getcwd(m_szInitDir,_MAX_PATH); 
 
  //如果目錄的最后一個(gè)字母不是'\',則在最后加上一個(gè)'\' 
  int len=strlen(m_szInitDir); 
  if (m_szInitDir[len-1] != '\\') 
    strcat(m_szInitDir,"\\"); 
} 
 
bool CBrowseDir::SetInitDir(const char *dir) 
{ 
  //先把dir轉(zhuǎn)換為絕對路徑 
  if (_fullpath(m_szInitDir,dir,_MAX_PATH) == NULL) 
    return false; 
 
  //判斷目錄是否存在 
  if (_chdir(m_szInitDir) != 0) 
    return false; 
 
  //如果目錄的最后一個(gè)字母不是'\',則在最后加上一個(gè)'\' 
  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 
    { 
      //檢查是不是目錄 
      //如果不是,則進(jìn)行處理 
      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中的子目錄 
  //因?yàn)樵谔幚韉ir中的文件時(shí)，派生類的ProcessFile有可能改變了 
  //當(dāng)前目錄，因此還要重新設(shè)置當(dāng)前目錄為dir。 
  //執(zhí)行過_findfirst后，可能系統(tǒng)記錄下了相關(guān)信息，因此改變目錄 
  //對_findnext沒有影響。 
  _chdir(dir); 
  if ((hFile=_findfirst("*.*",&fileinfo)) != -1) 
  { 
    do 
    { 
      //檢查是不是目錄 
      //如果是,再檢查是不是 . 或 ..  
      //如果不是,進(jìn)行迭代 
      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派生出的子類，用來統(tǒng)計(jì)目錄中的文件及子目錄個(gè)數(shù) 
class CStatDir:public CBrowseDir 
{ 
protected: 
  int m_nFileCount;  //保存文件個(gè)數(shù) 
  int m_nSubdirCount; //保存子目錄個(gè)數(shù) 
 
public: 
  //缺省構(gòu)造器 
  CStatDir() 
  { 
    //初始化數(shù)據(jù)成員m_nFileCount和m_nSubdirCount 
    m_nFileCount=m_nSubdirCount=0; 
  } 
 
  //返回文件個(gè)數(shù) 
  int GetFileCount() 
  { 
    return m_nFileCount; 
  } 
 
  //返回子目錄個(gè)數(shù) 
  int GetSubdirCount() 
  { 
    //因?yàn)檫M(jìn)入初始目錄時(shí)，也會(huì)調(diào)用函數(shù)ProcessDir， 
    //所以減1后才是真正的子目錄個(gè)數(shù)。 
    return m_nSubdirCount-1; 
  } 
 
protected: 
  //覆寫虛函數(shù)ProcessFile，每調(diào)用一次，文件個(gè)數(shù)加1 
  virtual bool ProcessFile(const char *filename) 
  { 
    m_nFileCount++; 
    return CBrowseDir::ProcessFile(filename); 
  } 
 
  //覆寫虛函數(shù)ProcessDir，每調(diào)用一次，子目錄個(gè)數(shù)加1 
  virtual void ProcessDir 
    (const char *currentdir,const char *parentdir) 
  { 
    m_nSubdirCount++; 
    CBrowseDir::ProcessDir(currentdir,parentdir); 
  } 
}; 
 
void main() 
{ 
  //獲取目錄名 
  char buf[256]; 
  printf("請輸入要統(tǒng)計(jì)的目錄名:"); 
  gets(buf); 
 
  //構(gòu)造類對象 
  CStatDir statdir; 
 
  //設(shè)置要遍歷的目錄 
  if (!statdir.SetInitDir(buf)) 
  { 
    puts("目錄不存在。"); 
    return; 
  } 
 
  //開始遍歷 
  statdir.BeginBrowse("*.*"); 
  printf("文件總數(shù): %d\n子目錄總數(shù):%d\n",statdir.GetFileCount(),statdir.GetSubdirCount()); 
}

已在windows上驗(yàn)證有效。

下面我加了BeginBrowseFilenames函數(shù)，以vector<char*>形式返回目錄中所有文件名。

實(shí)例代碼：

#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: 
  //存放初始目錄的絕對路徑，以'\'結(jié)尾 
  char m_szInitDir[_MAX_PATH]; 
 
public: 
  //缺省構(gòu)造器 
  CBrowseDir(); 
 
  //設(shè)置初始目錄為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); 
  //函數(shù)BrowseDir每找到一個(gè)文件,就調(diào)用ProcessFile 
  //并把文件名作為參數(shù)傳遞過去 
  //如果返回false,表示中止遍歷文件 
  //用戶可以覆寫該函數(shù),加入自己的處理代碼 
  virtual bool ProcessFile(const char *filename); 
 
  //函數(shù)BrowseDir每進(jìn)入一個(gè)目錄,就調(diào)用ProcessDir 
  //并把正在處理的目錄名及上一級目錄名作為參數(shù)傳遞過去 
  //如果正在處理的是初始目錄,則parentdir=NULL 
  //用戶可以覆寫該函數(shù),加入自己的處理代碼 
  //比如用戶可以在這里統(tǒng)計(jì)子目錄的個(gè)數(shù) 
  virtual void ProcessDir(const char *currentdir,const char *parentdir); 
}; 
 
CBrowseDir::CBrowseDir() 
{ 
  //用當(dāng)前目錄初始化m_szInitDir 
  getcwd(m_szInitDir,_MAX_PATH); 
 
  //如果目錄的最后一個(gè)字母不是'\',則在最后加上一個(gè)'\' 
  int len=strlen(m_szInitDir); 
  if (m_szInitDir[len-1] != '\\') 
    strcat(m_szInitDir,"\\"); 
} 
 
bool CBrowseDir::SetInitDir(const char *dir) 
{ 
  //先把dir轉(zhuǎn)換為絕對路徑 
  if (_fullpath(m_szInitDir,dir,_MAX_PATH) == NULL) 
    return false; 
 
  //判斷目錄是否存在 
  if (_chdir(m_szInitDir) != 0) 
    return false; 
 
  //如果目錄的最后一個(gè)字母不是'\',則在最后加上一個(gè)'\' 
  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 
    { 
      //檢查是不是目錄 
      //如果不是,則進(jìn)行處理 
      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中的子目錄 
  //因?yàn)樵谔幚韉ir中的文件時(shí)，派生類的ProcessFile有可能改變了 
  //當(dāng)前目錄，因此還要重新設(shè)置當(dāng)前目錄為dir。 
  //執(zhí)行過_findfirst后，可能系統(tǒng)記錄下了相關(guān)信息，因此改變目錄 
  //對_findnext沒有影響。 
  _chdir(dir); 
  if ((hFile=_findfirst("*.*",&fileinfo)) != -1) 
  { 
    do 
    { 
      //檢查是不是目錄 
      //如果是,再檢查是不是 . 或 ..  
      //如果不是,進(jìn)行迭代 
      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 
    { 
      //檢查是不是目錄 
      //如果不是,則進(jìn)行處理 
      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中的子目錄 
  //因?yàn)樵谔幚韉ir中的文件時(shí)，派生類的ProcessFile有可能改變了 
  //當(dāng)前目錄，因此還要重新設(shè)置當(dāng)前目錄為dir。 
  //執(zhí)行過_findfirst后，可能系統(tǒng)記錄下了相關(guān)信息，因此改變目錄 
  //對_findnext沒有影響。 
  _chdir(dir); 
  if ((hFile=_findfirst("*.*",&fileinfo)) != -1) 
  { 
    do 
    { 
      //檢查是不是目錄 
      //如果是,再檢查是不是 . 或 ..  
      //如果不是,進(jìn)行迭代 
      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派生出的子類，用來統(tǒng)計(jì)目錄中的文件及子目錄個(gè)數(shù) 
class CStatDir:public CBrowseDir 
{ 
protected: 
  int m_nFileCount;  //保存文件個(gè)數(shù) 
  int m_nSubdirCount; //保存子目錄個(gè)數(shù) 
 
public: 
  //缺省構(gòu)造器 
  CStatDir() 
  { 
    //初始化數(shù)據(jù)成員m_nFileCount和m_nSubdirCount 
    m_nFileCount=m_nSubdirCount=0; 
  } 
 
  //返回文件個(gè)數(shù) 
  int GetFileCount() 
  { 
    return m_nFileCount; 
  } 
 
  //返回子目錄個(gè)數(shù) 
  int GetSubdirCount() 
  { 
    //因?yàn)檫M(jìn)入初始目錄時(shí)，也會(huì)調(diào)用函數(shù)ProcessDir， 
    //所以減1后才是真正的子目錄個(gè)數(shù)。 
    return m_nSubdirCount-1; 
  } 
 
protected: 
  //覆寫虛函數(shù)ProcessFile，每調(diào)用一次，文件個(gè)數(shù)加1 
  virtual bool ProcessFile(const char *filename) 
  { 
    m_nFileCount++; 
    return CBrowseDir::ProcessFile(filename); 
  } 
 
  //覆寫虛函數(shù)ProcessDir，每調(diào)用一次，子目錄個(gè)數(shù)加1 
  virtual void ProcessDir 
    (const char *currentdir,const char *parentdir) 
  { 
    m_nSubdirCount++; 
    CBrowseDir::ProcessDir(currentdir,parentdir); 
  } 
}; 
 
void main() 
{ 
  //獲取目錄名 
  char buf[256]; 
  printf("請輸入要統(tǒng)計(jì)的目錄名:"); 
  gets(buf); 
 
  //構(gòu)造類對象 
  CStatDir statdir; 
 
  //設(shè)置要遍歷的目錄 
  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("文件總數(shù): %d\n",file_vec.size()); 
  system("pause"); 
}

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享的“C++如何實(shí)現(xiàn)遍歷目錄下文件”這篇文章對大家有幫助，同時(shí)也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，更多相關(guān)知識(shí)等著你來學(xué)習(xí)!