C++讀取特定路徑下文件目錄及文件名稱

 

這幾天在弄一個項目，需要讀取給定路徑下的所有文件夾名稱或所有帶后綴的文件名。查了下C++的資料，發現有很多方法，這里總結其中的一種，其他用法類似。主要用到了以下幾個頭文件(類)：io.h，fstream，string。

1、讀取某給定路徑下所有文件夾與文件名稱，並帶完整路徑。代碼如下：

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);
    }
}

該函數有兩個參數，第一個為路徑字符串(string類型，最好為絕對路徑)；第二個參數為文件夾與文件名稱存儲變量(vector類型,引用傳遞)。在主函數中調用格式(並將結果保存在文件"AllFiles.txt"中，第一行為總數)：

char* filePath = "E:\\YunShi";
vector<string> files;
char* distAll = "AllFiles.txt";
getAllFiles(filePath, files);
ofstream ofn(distAll);
int size = files.size();
ofn<<size<<endl;
for (int i = 0; i < size; i++) {
    ofn<<files[i]<<endl;
}
ofn.close();

結果如下圖：

同理，只讀取某給定路徑下的當前文件夾名(以下類似，只給出函數，調用案例同上)：

void getJustCurrentDir(string path, vector<string>& files) {
    //文件句柄
    long hFile = 0;
    //文件信息 
    struct _finddata_t fileinfo;
    string p;
    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(fileinfo.name);
                    //files.push_back(p.assign(path).append("\\").append(fileinfo.name));
                }
            }
        } while (_findnext(hFile, &fileinfo) == 0);
        _findclose(hFile);
    }
}

只讀取某給定路徑下的當前文件名：

void getJustCurrentFile(string path, vector<string>& files) {
    //文件句柄
    long hFile = 0;
    //文件信息
    struct _finddata_t fileinfo;
    string p;
    if ((hFile = _findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -1) {
        do {
            if ((fileinfo.attrib & _A_SUBDIR)) {
                ;
            } else {
                files.push_back(fileinfo.name);
                //files.push_back(p.assign(path).append("\\").append(fileinfo.name));
            }
        } while (_findnext(hFile, &fileinfo) == 0);
        _findclose(hFile);
    }
}

只讀取某給定路徑下的所有文件名(即包含當前目錄及子目錄的文件)：

void getFilesAll(string path, vector<string>& files) {
    //文件句柄
    long hFile = 0;
    //文件信息
    struct _finddata_t fileinfo;
    string p;  
    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));
                    getFilesAll(p.assign(path).append("\\").append(fileinfo.name), files);
                }
            } else {  
                files.push_back(p.assign(path).append("\\").append(fileinfo.name));
            }
        } while (_findnext(hFile, &fileinfo) == 0);
        _findclose(hFile);
    }
}

至此，簡單的讀取功能就實現了，以后只要指定路徑，就能輕松地得到對應的路徑與文件名稱。