linux dirent.h 实现getFiles

    1.   void getFiles(std::string root_dir, std::string ext, std::vector<std::string>& files) {
    2.         DIR *dir;
    3.         struct dirent *ptr;
    5.         if ((dir = opendir(root_dir.c_str())) == NULL) {
    6.             gLogInfo << "Open dir error..." << std::endl;
    7.             return;
    8.         }
    10.         while ((ptr = readdir(dir)) != NULL) {
    11.             if (strcmp(ptr->d_name,".") == 0 || strcmp(ptr->d_name,"..") == 0) {
    12.                 continue;
    13.             } else if(ptr->d_type == 8)  {// file
    14.                 char* dot = strchr(ptr->d_name, '.');
    15.                 if (dot && !strcasecmp(dot, ext.c_str())) {
    16.                     std::string filename(root_dir);
    17.                     filename.append("/").append(ptr->d_name);
    18.                     files.push_back(filename);
    19.                 }
    20.             } else if(ptr->d_type == 10) { // link file  
    21.                 continue;
    22.             } else if(ptr->d_type == 4)  {// dir
    23.                 std::string dir_path(root_dir);
    24.                 dir_path.append("/").append(ptr->d_name);
    25.                 getFiles(dir_path.c_str(), ext, files);
    26.             }  
    27.         }
    29.         closedir(dir);  
    30.     }
    std::vector<std::string> filenames;
    filenames.reserve(20480);
    for (size_t i = 0; i < onnx_args.dataDirs.size(); i++) {
        getFiles(onnx_args.dataDirs[i], ".jpg", filenames);
    }

    参考