C++獲取指定文件夾下的所有文件名（或路徑）

今天在安裝PCL的時候寫了個博客，但是寫到要添加的dll的時候，有好幾十個dll文件，開始時逐個點擊文件名、複製、粘貼，感覺特別費事，所以在網上找個代碼。這個代碼能實現查找文件夾下的所有文件以及指定類型的文件。自己修改了點然後添加上註釋就直接貼出來了。

#include <io.h>
#include <fstream>
#include <string>
#include <vector>
#include <iostream>
using namespace std;


//獲取所有的文件名
void GetAllFiles( 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) );
					GetAllFiles( p.assign(path).append("\\").append(fileinfo.name), files ); 
				}
			}  
			else //如果查找到的不是是文件夾 
			{  
				files.push_back(p.assign(fileinfo.name) );  //將文件路徑保存，也可以只保存文件名:    p.assign(path).append("\\").append(fileinfo.name)
			} 

		}while(_findnext(hFile, &fileinfo)  == 0);  

		_findclose(hFile); //結束查找
	} 

}  

//獲取特定格式的文件名
void GetAllFormatFiles( string path, vector<string>& files,string format)  
{  
	//文件句柄  
	long   hFile   =   0;  
	//文件信息  
	struct _finddata_t fileinfo;  
	string p;  
	if((hFile = _findfirst(p.assign(path).append("\\*" + format).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) );
					GetAllFormatFiles( p.assign(path).append("\\").append(fileinfo.name), files,format); 
				}
			}  
			else  
			{  
				files.push_back( p.assign(fileinfo.name));  //將文件路徑保存，也可以只保存文件名:    p.assign(path).append("\\").append(fileinfo.name)
			}  
		}while(_findnext(hFile, &fileinfo)  == 0);  

		_findclose(hFile); 
	} 
} 

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

int main()
{
	string filePath = "D:\\文檔\\課程\\數據結構\\MOOC練習\\testPCL\\x64\\Debug";  
	vector<string> files;  
	char * distAll = "AllFiles.txt";

	//讀取所有的文件，包括子文件的文件
	//GetAllFiles(filePath, files);

	//讀取所有格式爲jpg的文件
	string format = ".dll";
	GetAllFormatFiles(filePath, files,format);
	ofstream ofn(distAll);
	int size = files.size(); 
	ofn<<size<<endl;
	for (int i = 0;i<size;i++)  
	{  
		ofn<<files[i]<<endl; // 寫入文件
		cout<< files[i] << endl;//輸出到屏幕
	}
	ofn.close();
	return 0;
}

參考資料：

C++讀取文件夾中所有的文件或者是特定後綴的文件

_finddata_t用法

ofstream和ifstream詳細用法