void getAllFiles(string path, vector& files, string fileType)
{
#ifdef OS_WIN
// 文件句柄
long hFile = 0;
// 文件信息
_finddata_t fileinfo;
string p;
hFile = _findfirst(p.assign(path).append("\\*" + fileType).c_str(), &fileinfo);
if (hFile != -1)
{
do
{
// 保存文件的全路径
files.push_back(p.assign(path).append("\\").append(fileinfo.name));
} while (_findnext(hFile, &fileinfo) == 0); //寻找下一个,成功返回0,否则-1
_findclose(hFile);
}
#endif
#ifdef OS_LINUX
DIR* dir = opendir(path.c_str());//打开指定目录
dirent* pFile = NULL;//定义遍历指针
string p;
if (dir != NULL)
{
while ((pFile = readdir(dir)) != NULL)//开始逐个遍历
{
//这里需要注意,linux平台下一个目录中有"."和".."隐藏文件,需要过滤掉
if (pFile->d_name[0] != '.')//d_name是一个char数组,存放当前遍历到的文件名
{
files.push_back(p.assign(path).append("/").append(pFile->d_name));
}
}
}
closedir(dir);//关闭指定目录
#endif
}