获取文件夹下所有某一类型的文件

1 代码

#include <iostream>
#include <fstream>
#include <sstream>
#include <string>
#include <vector>
#include <math.h>
#include <regex>
#include <cassert>
#include <io.h>
#include <algorithm>
#include <chrono>
using namespace std;
void getAllFiles(std::string path, regex flag, vector<string>& files)
{
    //文件句柄  
    long long  hFile = 0;
    //文件信息  
    struct _finddata_t fileinfo;  //很少用的文件信息读取结构
    string p;  //string类很有意思的一个赋值函数:assign()，有很多重载版本
    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), flag, files);//递归当前文件夹
                                                                                               //getAllFiles(p.assign(path).append("/").append(fileinfo.name), files_landmarks);//递归当前文件夹
                }
            }
            else    //文件处理
            {
                if (regex_search(fileinfo.name, flag))
                {
                    //files.push_back(p.assign(path).append("/").append(fileinfo.name));//文件名	
                    files.push_back(fileinfo.name);
                }
            }
        } while (_findnext(hFile, &fileinfo) == 0);  //寻找下一个，成功返回0，否则-1
        _findclose(hFile);
    }
    sort(files.begin(), files.end());
}

2 Example

int main(int argc, char** argv)
{
	std::string strPath = argv[1];
    vector<string> vecPLYFileName;
    regex regPLY(".ply$");   
    getAllFiles(strPath, regPLY, vecPLYFileName);
}