c++ 获取文件夹文件并排序

小野617

已于 2023-02-09 14:58:50 修改

阅读量3.6k

点赞数 2

分类专栏： c++ 文章标签：大数据

于 2022-06-29 20:13:19 首次发布

本文链接：https://blog.csdn.net/qq_32149483/article/details/125528674

版权

c++ 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了C++中获取文件名的两种方法，一是按字符串或数值大小排序，二是自适应排序，适用于不同类型的文件。着重讲解了如何根据不同后缀进行文件筛选和排序，包括递归处理子目录。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

下面介绍获取文件的两种方式：

（1）方法一：可能需要根据字符串或数值的大小进行切换

void GetFileNames(string path, vector<string> &filenames, string form)
{
    DIR *pDir;
    struct dirent *ptr;
    string filename;
    if (!(pDir = opendir(path.c_str())))
        return;

    while ((ptr = readdir(pDir)) != 0)
    {
        // 跳过.和..文件
        if (strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0)
            continue;
        filename = ptr->d_name;

        if (filename.substr(filename.find("."), filename.length()) == form)
        {
            // cout<<filename<<endl;
            filenames.push_back(filename);
        }
    }
    closedir(pDir);
    //按字符串的方式进行  
    // sort(filenames.begin(), filenames.end());
    
    //按数值大小的方式  避免 1.txt 10.txt 11.txt的方式进行排序 
    sort(filenames.begin(), filenames.end(), [](string a, string b){
        return stoi(a) < stoi(b);
    });
    //
}

（2）能自适应进行排序，不管数值或者字符串大小（注意需要设置排序的文件后缀）


#include <string>
#include <dirent.h>
#include <vector>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>

using namespace std;

void GetFiles(std::string path, std::vector<std::string>& files)
{
    DIR *dir;
    struct dirent *ptr;
    std::string p;
 
    if((dir= opendir(path.c_str()) )==NULL)
    {
        std::cout << "cannot open:" << path << std::endl;
        return ;
    }
 
    while ((ptr = readdir(dir)) != NULL)
    {
        if (ptr->d_type == 4)
        {
            if (strcmp(ptr->d_name, ".") != 0 && strcmp(ptr->d_name, "..") != 0)
            {
                GetFiles(p.assign(path).append("/").append(ptr->d_name), files);
            }
 
        }
 
        else
        {
            std::string fname(ptr->d_name);
            std::string imgType = fname.substr(fname.rfind("."), fname.length());
            if (imgType == ".jpg" || imgType == ".jpeg" || imgType == ".JPG" || imgType == ".JPEG" || imgType == ".bmp" || imgType == ".BMP" || imgType == ".png" || imgType == ".PNG")
            // if (imgType == ".txt" )
            {
                // files.push_back(p.assign(path).append("/").append(ptr->d_name));  //文件名和路径一起包装好
                files.push_back(p.assign(ptr->d_name));  //只放入文件名
            }
        }
    }
    closedir(dir);
}
 
bool GreaterEqSort(std::string filePath1, std::string filePath2)
{
    int len1 = filePath1.length();
    int len2 = filePath2.length();
    if(len1 < len2)
    {
        return false;
    }
    else if(len1 > len2)
    {
        return true;
    }
    else
    {
        int iter = 0;
        while(iter < len1)
        {
            if(filePath1.at(iter) < filePath2.at(iter))
            {
                return false;
            }
            else if(filePath1.at(iter) > filePath2.at(iter))
            {
                return true;
            }
            ++iter;
        }
    }
    return true;
}
 
bool LessSort(std::string filePath1, std::string filePath2)
{
    return (!GreaterEqSort(filePath1, filePath2));
}
 
 
void pathSort(std::vector<std::string> &paths, int sortMode)
{
 
    if(sortMode == 1)
    {
        std::sort(paths.begin(), paths.end(), LessSort);
    }
}

int main(){
    std::vector<std::string> ImgFiles;
    std::string imgPath="/home/welling/myProject/radar_camera/RTK_calibration/common/kalman/data/pic/";
    GetFiles(imgPath,ImgFiles);
    pathSort(ImgFiles,1);
}