c++ 获取指定目录下的所有文件

获取指定目录下的所有文件

//1、获取指定目录下的所有文件
DWORD GetAllFiles(string strPath,vector<string> &vFiles,string strFileType = "",bool bRec = true/*是否递归子文件夹*/,bool bPorn = true/*true 返回全路径否则返回文件名*/)
{
	//文件句柄
	long hFile = 0;
	//文件信息
	_finddata_t fileinfo;
	string p;

	//文件数量 递归调用只能被初始化一次
	static  DWORD dwFileCount = 0;
	//若果strPath下没有指定类型的文件，则看是否有子文件夹继续搜
	if ((hFile = _findfirst(p.assign(strPath).append("\\*" + strFileType).c_str(),&fileinfo)) != -1)
	{
		do
		{
			dwFileCount++;
			if (bPorn) //保存全路径
			{
				vFiles.push_back(p.assign(strPath).append("\\").append(fileinfo.name));
			}
			else     //保存文件名
			{
				vFiles.push_back(fileinfo.name);
			}

			if (bRec && fileinfo.attrib & _A_SUBDIR)
			{
				if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
				{
					//递归搜索
					GetAllFiles(p.assign(strPath).append("\\").append(fileinfo.name), vFiles, strFileType, bRec, bPorn);
				}
			}
		} while (_findnext(hFile,&fileinfo) == 0);
		_findclose(hFile);
	}
	else
	{
		if ((hFile = _findfirst(p.assign(strPath).append("\\*").c_str(), &fileinfo)) != -1)
		{
			do
			{
				if (bRec && fileinfo.attrib & _A_SUBDIR)
				{
					if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
					{
						//递归搜索
						GetAllFiles(p.assign(strPath).append("\\").append(fileinfo.name), vFiles, strFileType, bRec, bPorn);
					}
				}
			} while (_findnext(hFile, &fileinfo) == 0);
			_findclose(hFile);

		}
	}

	return dwFileCount;
}
int main()
{
	vector<string> files;
	string path = "D:\\C++\\GetAllFileName";
	string filetype = "";
	bool bRec = true/*是否递归子文件夹*/;
	bool bPorn = true/*true 返回全路径否则返回文件名*/;

	DWORD dwFileCount = GetAllFiles(path, files, filetype, bRec, bPorn);
	printf("总共 %d 个文件\n", dwFileCount);
	for (int i = 0; i < files.size(); i++)
	{
		printf("%s\n",files[i].c_str());
	}

	getchar();
    return 0;
}