自己写的一个文件查找的函数,可以指定格式,也可以查找下级目录,废话不多说,代码如下:
void getFilesall(string path, vector<string>& files, vector<string>& filesname, string format)
{
//文件句柄
long hFile = 0;
//文件信息
struct _finddata_t fileinfo; //很少用的文件信息读取结构
string p;
string name;
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, filesname, format);
}
}
else
{
string filename = fileinfo.name;
size_t filename_size = filename.length();
string filetype = format;
if (filename.compare((filename_size - 4), 4, filetype) == 0)//比较文件格式
{
files.push_back(p.assign(path).append("\\").append(fileinfo.name));//保存文件路径
filesname.push_back(name.assign(fileinfo.name)); //保存文件名
}
}
} while (_findnext(hFile, &fileinfo) == 0); //寻找下一个,成功返回0,否则-1
_findclose(hFile);
}
}