C++获取指定文件夹下的所有文件名（或路径）

来源：互联网发布：数据库中的模式是编辑：程序博客网时间：2024/04/29 21:40
#include <io.h>  #include <fstream>  #include <string>  #include <vector>  #include <iostream>  using namespace std;      //其实两个字符串连在一起比如string可惜写成：str1+str2+“xxx”。。。//获取所有的文件名  void GetAllFiles( string path, vector<string>& files)    {          long   hFile   =   0;        //文件信息        struct _finddata_t fileinfo;//用来存储文件信息的结构体        string p;        if((hFile = _findfirst(p.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)  //第一次查找      {            do            {                 if((fileinfo.attrib &  _A_SUBDIR))  //如果查找到的是文件夹              {                    if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)  //进入文件夹查找                  {                      files.push_back(p.assign(path).append("\\").append(fileinfo.name) );                      GetAllFiles( p.assign(path).append("\\").append(fileinfo.name), files );                   }              }                else //如果查找到的不是是文件夹               {                    files.push_back(p.assign(fileinfo.name) );  //将文件路径保存，也可以只保存文件名:    p.assign(path).append("\\").append(fileinfo.name)              }             }while(_findnext(hFile, &fileinfo)  == 0);              _findclose(hFile); //结束查找      }     }      //获取特定格式的文件名  void GetAllFormatFiles( string path, vector<string>& files,string format)    {        //文件句柄        long   hFile   =   0;        //文件信息        struct _finddata_t fileinfo;        string p;        if((hFile = _findfirst(p.assign(path).append("\\*" + format).c_str(),&fileinfo)) !=  -1)        {            do            {                  if((fileinfo.attrib &  _A_SUBDIR))                {                    if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)                    {                      //files.push_back(p.assign(path).append("\\").append(fileinfo.name) );                      GetAllFormatFiles( p.assign(path).append("\\").append(fileinfo.name), files,format);                   }              }                else                {                    files.push_back( p.assign(fileinfo.name));  //将文件路径保存，也可以只保存文件名:    p.assign(path).append("\\").append(fileinfo.name)              }            }while(_findnext(hFile, &fileinfo)  == 0);              _findclose(hFile);       }   }     // 该函数有两个参数，第一个为路径字符串(string类型，最好为绝对路径)；  // 第二个参数为文件夹与文件名称存储变量(vector类型,引用传递)。  // 在主函数中调用格式(并将结果保存在文件"AllFiles.txt"中，第一行为总数)：    int main()  {      string filePath = "D:\\文档\\课程\\数据结构\\MOOC练习\\testPCL\\x64\\Debug";        vector<string> files;        char * distAll = "AllFiles.txt";        //读取所有的文件，包括子文件的文件      //GetAllFiles(filePath, files);        //读取所有格式为jpg的文件      string format = ".dll";      GetAllFormatFiles(filePath, files,format);      ofstream ofn(distAll);      int size = files.size();       ofn<<size<<endl;      for (int i = 0;i<size;i++)        {            ofn<<files[i]<<endl; // 写入文件          cout<< files[i] << endl;//输出到屏幕      }      ofn.close();      return 0;  }
0 0