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

COPY FROM：http://blog.csdn.net/watkinsong/article/details/9223007

图像处理的时候经常需要对一个目录的所有图像进行处理，遍历文件得c代码：

在windows中需要使用到宽字符。

另外，可以使用opencv封装的目录访问操作，下次给出。

[cpp]  view plain copy

1. // DirTraverse.cpp : Defines the entry point for the console application.  
2. //  
3.   
4. #include "stdafx.h"  
5. #include "windows.h"  
6. #include <string.h>  
7. #include <Strsafe.h>  
8. #include <iostream>  
9.   
10.   
11. using namespace std;  
12.   
13. char *w2c(char *pcstr,const wchar_t *pwstr, size_t len);  
14. void TraverseDirectory(wchar_t Dir[MAX_PATH]);  
15. char * imageDir = "D:\\enroll\\";  
16.   
17.   
18. int main( int argc, const char** argv )  
19. {  
20.     locale loc( "chs" );                //支持中文输出，否则wchar可能无法输出值为中文的变量  
21.     wcout.imbue( loc );  
22.   
23.     TraverseDirectory(L"D:\\images\\all_0407\\");  
24.   
25.     system("pause");  
26.     return 0;  
27. }  
28.   
29.   
30.   
31. //Converting a WChar string to a Ansi string    
32. char *w2c(char *pcstr,const wchar_t *pwstr, size_t len)    
33. {    
34.     int nlength=wcslen(pwstr);    
35.     //获取转换后的长度    
36.     int nbytes = WideCharToMultiByte( 0, 0, pwstr, nlength, NULL,0,NULL, NULL );     
37.     if(nbytes>len)   nbytes=len;    
38.     // 通过以上得到的结果，转换unicode 字符为ascii 字符    
39.     WideCharToMultiByte( 0,0, pwstr, nlength,   pcstr, nbytes, NULL,   NULL );    
40.     return pcstr ;    
41. }    
42.   
43. //传入要遍历的文件夹路径，并遍历相应文件夹  
44. void TraverseDirectory(wchar_t Dir[MAX_PATH])      
45. {  
46.     WIN32_FIND_DATA FindFileData;  
47.     HANDLE hFind=INVALID_HANDLE_VALUE;  
48.     wchar_t DirSpec[MAX_PATH];                  //定义要遍历的文件夹的目录  
49.     DWORD dwError;  
50.     StringCchCopy(DirSpec,MAX_PATH,Dir);  
51.         StringCchCat(DirSpec,MAX_PATH,TEXT("\\*"));   //定义要遍历的文件夹的完整路径\*  
52.   
53.     hFind=FindFirstFile(DirSpec,&FindFileData);          //找到文件夹中的第一个文件  
54.   
55.     if(hFind==INVALID_HANDLE_VALUE)                               //如果hFind句柄创建失败，输出错误信息  
56.     {  
57.         FindClose(hFind);   
58.         return;    
59.     }  
60.     else   
61.     {  
62.         while(FindNextFile(hFind,&FindFileData)!=0)                            //当文件或者文件夹存在时  
63.         {  
64.             if((FindFileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)!=0&&wcscmp(FindFileData.cFileName,L".")==0||wcscmp(FindFileData.cFileName,L"..")==0)        //判断是文件夹&&表示为"."||表示为"."  
65.             {  
66.                  continue;  
67.             }  
68.             if((FindFileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)!=0)      //判断如果是文件夹  
69.             {  
70.                 wchar_t DirAdd[MAX_PATH];  
71.                 StringCchCopy(DirAdd,MAX_PATH,Dir);  
72.                 StringCchCat(DirAdd,MAX_PATH,TEXT("\\"));  
73.                 StringCchCat(DirAdd,MAX_PATH,FindFileData.cFileName);       //拼接得到此文件夹的完整路径  
74.                 TraverseDirectory(DirAdd);                                  //实现递归调用  
75.             }  
76.             if((FindFileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)==0)    //如果不是文件夹  
77.             {  
78.                 //wcout<<Dir<<"\\"<<FindFileData.cFileName<<endl;            //输出完整路径  
79.                 char * fname = (char*)malloc(sizeof(char)*(2*wcslen(FindFileData.cFileName)+1));  
80.                 memset(fname,0,2*wcslen(FindFileData.cFileName)+1);  
81.                 w2c(fname, FindFileData.cFileName, 2*wcslen(FindFileData.cFileName)+1); // 获取文件名  
82.                 cout<<fname<<endl;  
83.   
84.                 if(strstr(fname, ".jpg"))  
85.                 {  
86.                     char * fullName = (char*)malloc(strlen(imageDir)+strlen(fname)+1);  
87.                     strcpy(fullName, imageDir);  
88.                     strcat(fullName, fname); // 获取完全路径  
89.                     // process  
90.                     // TODO  
91.                     //  
92.                     free(fullName);  
93.                 }  
94.                 // free  
95.                 free(fname);  
96.             }  
97.         }  
98.         FindClose(hFind);  
99.     }  
100. }