c++遍历目录下的所有文件
要实现一个遍历指定目录下的小功能,没找到类似于py中类似于os.listdir()的函数。于是从网上找了能实现功能点的代码,感觉以后会用到,怕忘了便记录下来。
#include<opencv.hpp>
#include <iostream>
#include<vector>
#include<io.h>
void getFilesPath(std::string path,std::vector<std::string>& files) {
//没有返回值,为了得到结果,引用传参
//文件句柄
//注意:有些文章代码此处是long类型,实测运行中会报错访问异常
intptr_t hFile = 0;
//文件信息
struct _finddata_t fileinfo;
std::string p;
hFile = _findfirst(p.assign(path).append("/*").c_str(),&fileinfo);//前一个参数是个形如 “/home/username/othoer/*”的字符串*应该是指想要匹配该目录下的所有文件
if (hFile != -1) {//查找文件发现子文件目录,若是返回-1则没有子文件
do {
//如果是目录,递归查找
if ((fileinfo.attrib & _A_SUBDIR)) {
//发现还有文件夹
if ((std::strcmp(fileinfo.name, ".") != 0) && (std::strcmp(fileinfo.name, "..") != 0)) {
//当文件夹不是“.”和“..”时,开始套娃
getFilesPath(p.assign(path).append("/").append(fileinfo.name).c_str(), files);
}
}
else//如果不是,把文件绝对路径存入vector中
{
files.push_back(p.assign(path).append("/").append(fileinfo.name));
}
} while (_findnext(hFile, &fileinfo)==0);
_findclose(hFile);
}
}
这个函数实现了和py中os.listdir类似的功能,但os.listdir不能获取更深层目录下的文件,此方法可以根据根据自己的需求调整。核心函数是:_findfirst()和_findindex()。不知道为啥只有配合while才能拿到数据,看样子这俩玩意是迭代器和生成器。上面函数可以简化一下不用拿多级目录下的东西。
intptr_t hFile = 0;
//文件信息
struct _finddata_t fileinfo;
std::string p;
hFile = _findfirst(p.assign(path).append("/*").c_str(),&fileinfo);
while(_findnext(hFile, &fileinfo)==0){
std::cout<<fileinfo.name<<std::endl;
}
//前两轮可能会先打印出:“.”和“..”看起来像在表示当前路径和上级路径。
// 两轮之后会打印目录里有的、人类能看到的文件
//可以用
//if ((std::strcmp(fileinfo.name, ".") == 0) || (std::strcmp(fileinfo.name, "..") == 0))
// continute;
//把前面提到的两个货过滤掉