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

 

#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;
}