分享一下我老师大神的人工智能教程！零基础，通俗易懂！http://blog.csdn.net/jiangjunshow
也欢迎大家转载本篇文章。分享知识，造福人民，实现我们中华民族伟大复兴！
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");} 给我老师的人工智能教程打call！http://blog.csdn.net/jiangjunshow
285
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
