C++讀取某個資料夾下面的子資料夾及其所有檔案

老司機的詩和遠方發表於2020-04-06

下面是輸出當前目錄下的所有資料夾以及檔案的絕對路徑(當然也可以是相對路徑,由輸入的路徑決定),下面的函式介面可以改裝為單輸出檔案或者資料夾的介面,這是一個大方面的總介面。

#include <fstream>
#include <iostream>
#include <string>
#include <sstream>
#include <vector>
using namespace std;
void getAllFiles(string path, vector<string>& files)
{
    //檔案控制程式碼  
    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), files);//遞迴當前資料夾
                }
            }
            else    //檔案處理
            {
                files.push_back(p.assign(path).append("/").append(fileinfo.name));//檔名
            }
        } while (_findnext(hFile, &fileinfo) == 0);  //尋找下一個,成功返回0,否則-1
        _findclose(hFile);
    }
}
//測試
void main()
{
    string DATA_DIR = "D:/CoderMaker/data_sets/lfw";
    vector<string> files;
    //測試
    char * DistAll = "AllFiles.txt";
    getAllFiles(DATA_DIR, files);//所有檔案與資料夾的路徑都輸出
    ofstream ofn(DistAll);  //輸出檔案流
    int size = files.size();
    int  FaiNum = 0;
    ofn << size << endl;
    for (int i = 0; i<size; i++)
    {
        ofn << files[i] << endl;
    }
    ofn.close();
    return 0;
}

測試結果
這裡寫圖片描述

相關文章