C++遍历文件夹下所有文件

来源:互联网 发布:罗布泊病毒说 知乎 编辑:程序博客网 时间:2024/05/08 17:24

      为数不多的好用的代码,遍历文件夹获取所有子文件名,"filespec"可用通配符“*?”。注意如果用相对路径的话,获取所有文件名后应再调用SetInitDir将初始目录改为当前目录,否则中间生成的文件都会放在之前的“InitDir”内。

      C/C++遍历文件夹感觉真是很不好用,建议还是使用C/C++做单任务处理,然后通过脚本语言实现遍历比较合理。


CBrowseDir.h

#include <io.h>#include <stdlib.h>#include <direct.h>#include <iostream>#include <string>#include <vector>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.cpp

#include "CBrowseDir.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;//判断目录是否存在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){}


0 0