C++文件夹中文件的读取

最新推荐文章于 2022-12-14 16:57:51 发布

VickyD1023

最新推荐文章于 2022-12-14 16:57:51 发布

阅读量642

点赞数

分类专栏： C++ 文章标签：文件读取

本文链接：https://blog.csdn.net/q6324266/article/details/52297101

版权

C++ 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

在做图像处理算法时，我们经常需要读取文件夹下所有的图片。每次这样的操作都要上网查一番，很麻烦。这里mark一种方便的文件读取方法。

文件夹下文件的读取需要包含头文件io.h。代码如下：

#include <iostream>
#include <vector>
#ifdef LINUX
#include <dirent.h>
#endif
#ifdef WIN32
#include <io.h>
#include <direct.h>
#endif

using namespace std;

#ifdef LINUX
void getFiles(char *str, vector<string> &files, vector<string> &files_names)
{
	DIR *dir;
	struct dirent *ptr;
	char base[1000];

	if ((dir = opendir(str)) == NULL)
	{
		perror("Open dir error...");
		return;
	}

	while ((ptr = readdir(dir)) != NULL)
	{
		if (strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0)    ///current dir OR parrent dir
			continue;
		else if (ptr->d_type == 8)    ///file
		{
			char name[256];
			sprintf(name, "%s/%s", str, ptr->d_name);
			//string name = str + "/" + string(ptr->d_name);
			files.push_back(name);
			files_names.push_back(ptr->d_name);
		}

		else if (ptr->d_type == 4)    ///dir
		{
			memset(base, '\0', sizeof(base));
			strcpy(base, str);
			strcat(base, "/");
			strcat(base, ptr->d_name);
			getFiles(base, files, files_names);
		}
	}
	closedir(dir);
	return;
}
#endif // LINUX

#ifdef WIN32
void getFiles(string path, vector<string>& files, vector<string> &files_names)
{
	//文件句柄      
	intptr_t   hFile = 0;
	//文件信息      
	struct _finddata_t fileinfo;
	string p;
	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)
					getFiles(p.assign(path).append("\\").append(fileinfo.name), files, files_names);
			}
			else
			{
				files.push_back(p.assign(path).append("\\").append(fileinfo.name));
				files_names.push_back(fileinfo.name);
			}
		} while (_findnext(hFile, &fileinfo) == 0);
		_findclose(hFile);
	}
}
#endif // WIN32

int main(int argc, char *argv[])
{

	char *filePath = argv[1];

	vector<string> files;
	vector<string> files_name;


	获取该路径下的所有文件      
	getFiles(filePath, files, files_name);

	for (int i = 0; i < files.size(); i++)
	{
		cout << files[i].c_str() << endl;
		cout << files_name[i].c_str() << endl;

	}

	return 0;
}