图像处理的时候经常需要对一个目录的所有图像进行处理,遍历文件得c代码:
在windows中需要使用到宽字符。
另外,可以使用opencv封装的目录访问操作,下次给出。
// DirTraverse.cpp : Defines the entry point for the console application.
//
#include "stdafx.h"
#include "windows.h"
#include <string.h>
#include <Strsafe.h>
#include <iostream>
using namespace std;
char *w2c(char *pcstr,const wchar_t *pwstr, size_t len);
void TraverseDirectory(wchar_t Dir[MAX_PATH]);
char * imageDir = "D:\\enroll\\";
int main( int argc, const char** argv )
{
locale loc( "chs" ); //支持中文输出,否则wchar可能无法输出值为中文的变量
wcout.imbue( loc );
TraverseDirectory(L"D:\\images\\all_0407\\");
system("pause");
return 0;
}
//Converting a WChar string to a Ansi string
char *w2c(char *pcstr,const wchar_t *pwstr, size_t len)
{
int nlength=wcslen(pwstr);
//获取转换后的长度
int nbytes = WideCharToMultiByte( 0, 0, pwstr, nlength, NULL,0,NULL, NULL );
if(nbytes>len) nbytes=len;
// 通过以上得到的结果,转换unicode 字符为ascii 字符
WideCharToMultiByte( 0,0, pwstr, nlength, pcstr, nbytes, NULL, NULL );
return pcstr ;
}
//传入要遍历的文件夹路径,并遍历相应文件夹
void TraverseDirectory(wchar_t Dir[MAX_PATH])
{
WIN32_FIND_DATA FindFileData;
HANDLE hFind=INVALID_HANDLE_VALUE;
wchar_t DirSpec[MAX_PATH]; //定义要遍历的文件夹的目录
DWORD dwError;
StringCchCopy(DirSpec,MAX_PATH,Dir);
StringCchCat(DirSpec,MAX_PATH,TEXT("\\*")); //定义要遍历的文件夹的完整路径\*
hFind=FindFirstFile(DirSpec,&FindFileData); //找到文件夹中的第一个文件
if(hFind==INVALID_HANDLE_VALUE) //如果hFind句柄创建失败,输出错误信息
{
FindClose(hFind);
return;
}
else
{
while(FindNextFile(hFind,&FindFileData)!=0) //当文件或者文件夹存在时
{
if((FindFileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)!=0&&wcscmp(FindFileData.cFileName,L".")==0||wcscmp(FindFileData.cFileName,L"..")==0) //判断是文件夹&&表示为"."||表示为"."
{
continue;
}
if((FindFileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)!=0) //判断如果是文件夹
{
wchar_t DirAdd[MAX_PATH];
StringCchCopy(DirAdd,MAX_PATH,Dir);
StringCchCat(DirAdd,MAX_PATH,TEXT("\\"));
StringCchCat(DirAdd,MAX_PATH,FindFileData.cFileName); //拼接得到此文件夹的完整路径
TraverseDirectory(DirAdd); //实现递归调用
}
if((FindFileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)==0) //如果不是文件夹
{
//wcout<<Dir<<"\\"<<FindFileData.cFileName<<endl; //输出完整路径
char * fname = (char*)malloc(sizeof(char)*(2*wcslen(FindFileData.cFileName)+1));
memset(fname,0,2*wcslen(FindFileData.cFileName)+1);
w2c(fname, FindFileData.cFileName, 2*wcslen(FindFileData.cFileName)+1); // 获取文件名
cout<<fname<<endl;
if(strstr(fname, ".jpg"))
{
char * fullName = (char*)malloc(strlen(imageDir)+strlen(fname)+1);
strcpy(fullName, imageDir);
strcat(fullName, fname); // 获取完全路径
// process
// TODO
//
free(fullName);
}
// free
free(fname);
}
}
FindClose(hFind);
}
}