在做图像处理算法过程中,有时需要将算法应用于某个文件夹下的所有图片,然后对比查看处理效果。有时文件夹内还有子文件夹,这时就需要递归遍历当前文件夹,以将算法应用于所有图片。遍历文件夹程序网上可以找到一些,有些风格很乱,读起来让人发懵;还有些程序是基于mfc库的,用起来虽简单,但显得比较重。下面这个使用win32 api遍历文件夹程序摘自网络,经过自己的改写,理解起来比较容易,用起来也比较方便。
对于指定的某个文件夹,遍历过程以文件名为序,决定先遍历哪个文件或哪个子文件夹,
比如b开头的文件或文件夹会先于m开头的文件或文件夹被遍历。
#include "stdafx.h"
#include <cv.h>
#include <cxcore.h>
#include <highgui.h>
using namespace cv;
// 遍历文件
void TraverseFiles(const char *lpPath = NULL)
{
char szFileName[MAX_PATH];
char szFilePath[MAX_PATH];
WIN32_FIND_DATA findFileData;
strcpy_s(szFilePath, lpPath);
strcat_s(szFilePath, "\\*.*");
HANDLE hFind = ::FindFirstFile(szFilePath, &findFileData);
if (INVALID_HANDLE_VALUE == hFind)
{
return;
}
do
{
if (findFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
if (findFileData.cFileName[0] != '.')
{
strcpy_s(szFilePath, lpPath);
strcat_s(szFilePath, "\\");
strcat_s(szFilePath, findFileData.cFileName);
// 递归遍历子文件夹
TraverseFiles(szFilePath);
// 遍历结束后返回上一级文件夹
strncpy_s(szFilePath, szFilePath, strlen(szFilePath) - strlen(findFileData.cFileName) - 1);
strcat_s(szFilePath, "\\*.*");
}
}
else
{
// 遍历当前文件夹
memset(szFileName, 0, sizeof(szFileName));
strncpy_s(szFileName, szFilePath, strlen(szFilePath) - 3);
strcat_s(szFileName, findFileData.cFileName);
// 测试>>读图并显示
printf("-----> %s \n", szFileName);
Mat img = imread(szFileName);
if (img.data != NULL)
{
imshow("image", img);
}
waitKey(200);
}
} while (FindNextFile(hFind, &findFileData));
FindClose(hFind);
}
int main(int argc, _TCHAR* argv[])
{
TraverseFiles("E:\\image");
waitKey();
return 0;
}