C++实现文件查找可指定格式

自己写的一个文件查找的函数,可以指定格式,也可以查找下级目录,废话不多说,代码如下:

void getFilesall(string path, vector& files, vector& 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);
	}
}

你可能感兴趣的:(C/C++)