获取目录下所有文件文件名

//获取目录路径下面所有文件的内容
//适用于linux和windows系统


#include <iostream>
#include <fstream>
#include <vector>
#include <string>
#ifdef _WIN32
#include <Windows.h>
#include <strsafe.h>
#else
#include <dirent.h>
#endif


using namespace std;


//
//获取指定目录下所有文件的文件名，不包括文件夹，在GetFileFromDir中使用
//strDir: 输入，目录路径
//vFileDirList： 输出，文件路径列表
//返回：空
//
void GetFileNameFromDir(string strDir,vector<string>& vFileDirList)
{
#ifdef _WIN32
WIN32_FIND_DATAA ffd;
LARGE_INTEGER filesize;
string szDir;
//size_t length_of_arg;
HANDLE hFind = INVALID_HANDLE_VALUE;
DWORD dwError=0;


szDir=strDir+"\\*";
hFind = FindFirstFileA(szDir.c_str(), &ffd);


if (INVALID_HANDLE_VALUE == hFind) 
{
cout<<"get file name error"<<endl;
return;
}
do
{
if (!(ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
{
string filename=ffd.cFileName;//(const char*)
string filedir=strDir+"\\"+filename;
vFileDirList.push_back(filedir);
}
}while (FindNextFileA(hFind, &ffd) != 0);


dwError = GetLastError();
if (dwError != ERROR_NO_MORE_FILES) 
{
cout<<"FindFirstFile error"<<endl;
return;
}
FindClose(hFind);
#else
DIR *dir;
struct dirent *ptr;
dir = opendir(strDir.c_str());
while( (ptr = readdir(dir)) != NULL)
{
string path = strDir + string("/") + string(ptr->d_name);
vFileDirList.push_back(path);
}
closedir(dir);
sort(vFileDirList.begin(), vFileDirList.end());
#endif
}


//
//获取指定目录下所有文件内容
//strDir: 输入，目录路径
//vFileNameList： 输出，文件内容列表
//返回：空
//
void GetFileFromDir(string strDir,vector<string>& vFileList)
{
//获取目录路径下所有文件的文件名
vector<string> vFileDirList;
GetFileNameFromDir(strDir,vFileDirList);
//读取文件
vector<string>::iterator iterFileDir;
for (iterFileDir=vFileDirList.begin();iterFileDir!=vFileDirList.end();iterFileDir++)
{
ifstream infile((*iterFileDir).c_str());
if (!infile)
{
cout<<"read file error"<<endl;
return;
}
string line="";
string file="";
while (getline(infile,line))
{
file+=line;
}
vFileList.push_back(file);
}
}


int main(int argc,char* argv[])
{
if (argc!=2)
{
cout<<"paramater error"<<endl;
return -1;
}
else
{
string dir(argv[1]);
vector<string> files;
GetFileFromDir(dir,files);
vector<string>::iterator iterFile;
for (iterFile=files.begin();iterFile!=files.end();iterFile++)
{
cout<<*iterFile<<endl;
}
cout<<endl;
system("pause");
return 0;
}
 
 
}