采用标准c进行目录文件遍历

图像处理的时候经常需要对一个目录的所有图像进行处理，遍历文件得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);

	}

}