C++17获取目录下所有的图像文件路径

C++17支持了更多的文件操作，这些更加方便了，今天尝试了一下递归读取一个目录下所有的图像文件的操作。直接上代码：

#include<iostream>
#include<vector>
#include<filesystem>

using namespace std;
using namespace std::filesystem;

// 检查一个路径是否是目录
bool checkIsDir(const string &dir) {
    if (!exists(dir)) {
        cout<<dir<<" not exists. Please check."<<endl;
        return false;
    }
    directory_entry entry(dir);
    if (entry.is_directory())
        return true;
    return false;
}

// 搜索一个目录下所有的图像文件，以 jpg,jpeg,png 结尾的文件
void getAllImageFiles(const string dir, vector<string> &files) {
    // 首先检查目录是否为空，以及是否是目录
    if (!checkIsDir(dir)) return;
    // 递归遍历所有的文件
    directory_iterator iters(dir);
    for(auto &iter: iters) {
        // 路径的拼接，基础知识还是不过关，只能用这个笨办法来了
        string file_path(dir);
        file_path += "/"; file_path += iter.path().filename();
        // 查看是否是目录，如果是目录则循环递归
        if (checkIsDir(file_path)) {
            getAllImageFiles(file_path, files);
        } else { //不是目录则检查后缀是否是图像
            string extension = iter.path().extension(); // 获取文件的后缀名
            // 可以扩展成你自己想要的文件类型来进行筛选, 比如加上.gif .bmp之类的
            if (extension == ".jpg" || extension == ".png" || extension == ".jpeg") {
                files.push_back(file_path);
            }
        }
    }
}

int main(int argc, char** argv) {
    vector<string> imgs;
    getAllImageFiles(argv[1], imgs);
    cout<<"There are "<<imgs.size()<<" images in "<<argv[1]<<endl;
}