[转]C++ 获取文件夹下的所有文件名
转自http://www.cnblogs.com/fnlingnzb-learner/p/6424563.html
头文件:#include<io.h>
char * filePath = "D:\\sample";
vector<string> files;
getFiles(filePath, files);
for (string f : files){
cout << f.c_str() << endl;
}
1 void getFiles( string path, vector<string>& files )
2 {
3 //文件句柄
4 intptr_t hFile = 0;
5 //文件信息
6 struct _finddata_t fileinfo;
7 string p;
8 if((hFile = _findfirst(p.assign(path).append("\\*").c_str(),&fileinfo)) != -1)
9 {
10 do
11 {
12 //如果是目录,迭代之
13 //如果不是,加入列表
14 if((fileinfo.attrib & _A_SUBDIR))
15 {
16 if(strcmp(fileinfo.name,".") != 0 && strcmp(fileinfo.name,"..") != 0)
17 getFiles( p.assign(path).append("\\").append(fileinfo.name), files );
18 }
19 else
20 {
21 files.push_back(p.assign(path).append("\\").append(fileinfo.name) );
22 }
23 }while(_findnext(hFile, &fileinfo) == 0);
24 _findclose(hFile);
25 }
26 }
_finddata_t 的使用
那么到底如何查找文件呢?我们需要一个结构体和几个大家可能不太熟悉的函数。这些函数和结构体在<io.h>的头文件中,结构体为struct _finddata_t ,函数为_findfirst、_findnext和_fineclose。具体如何使用,我会慢慢讲来~
首先讲这个结构体吧~ struct _finddata_t ,这个结构体是用来存储文件各种信息的。说实话,这个结构体的具体定义代码,我没有找到,不过还好,文档里面在_find里有比较详细的成员变量介绍。我基本上就把文档翻译过来讲吧:
unsigned atrrib:文件属性的存储位置。它存储一个unsigned单元,用于表示文件的属性。文件属性是用位表示的,主要有以下一些:_A_ARCH(存档)、_A_HIDDEN(隐藏)、_A_NORMAL(正常)、_A_RDONLY(只读)、_A_SUBDIR(文件夹)、_A_SYSTEM(系统)。
这些都是在<io.h>中定义的宏,可以直接使用,而本身的意义其实是一个无符号整型(只不过这个整型应该是2的几次幂,从而保证只有一位为1,而其他位为0)。既然是位表示,那么当一个文件有多个属性时,它往往是通过位或的方式,来得到几个属性的综合。例如只读+隐藏+系统属性,应该为:_A_HIDDEN
| _A_RDONLY | _A_SYSTEM 。
time_t time_create:这里的time_t是一个变量类型(长整型?相当于long int?),用来存储时间的,我们暂时不用理它,只要知道,这个time_create变量是用来存储文件创建时间的就可以了。
time_t time_access:文件最后一次被访问的时间。
time_t time_write:文件最后一次被修改的时间。
_fsize_t size:文件的大小。这里的_fsize_t应该可以相当于unsigned整型,表示文件的字节数。
char name[_MAX_FNAME]:文件的文件名。这里的_MAX_FNAME是一个常量宏,它在<stdlib.h>头文件中被定义,表示的是文件名的最大长度。
以此,我们可以推测出,struct _finddata_t ,大概的定义如下:
1 struct _finddata_t
2 {
3 unsigned attrib;
4 time_t time_create;
5 time_t time_access;
6 time_t time_write;
7 _fsize_t size;
8 char name[_MAX_FNAME];
9 };
前面也说了,这个结构体是用来存储文件信息的,那么如何把一个硬盘文件的文件信息“存到”这个结构体所表示的内存空间里去呢?这就要靠_findfirst、_findnext和_fineclose三个函数的搭配使用了。
首先还是对这三个函数一一介绍一番吧……
long _findfirst( char *filespec, struct _finddata_t *fileinfo );
返回值:如果查找成功的话,将返回一个long型的唯一的查找用的句柄(就是一个唯一编号)。这个句柄将在_findnext函数中被使用。若失败,则返回-1。
参数:
filespec:标明文件的字符串,可支持通配符。比如:*.c,则表示当前文件夹下的所有后缀为C的文件。
fileinfo :这里就是用来存放文件信息的结构体的指针。这个结构体必须在调用此函数前声明,不过不用初始化,只要分配了内存空间就可以了。函数成功后,函数会把找到的文件的信息放入这个结构体中。
int _findnext( long handle, struct _finddata_t *fileinfo );
返回值:若成功返回0,否则返回-1。
参数:
handle:即由_findfirst函数返回回来的句柄。
fileinfo:文件信息结构体的指针。找到文件后,函数将该文件信息放入此结构体中。
int _findclose( long handle );
返回值:成功返回0,失败返回-1。
参数:
handle :_findfirst函数返回回来的句柄。
大家看到这里,估计都能猜到个大概了吧?先用_findfirst查找第一个文件,若成功则用返回的句柄调用_findnext函数查找其他的文件,当查找完毕后用,用_findclose函数结束查找。恩,对,这就是正确思路。下面我们就按照这样的思路来编写一个查找C:\WINDOWS文件夹下的所有exe可执行文件的程序。
1 #include <stdio.h>
2 #include <io.h>
3
4
5 const char *to_search="C:\\WINDOWS\\*.exe"; //欲查找的文件,支持通配符
6
7
8 int main()
9 {
10 long handle; //用于查找的句柄
11 struct _finddata_t fileinfo; //文件信息的结构体
12 handle=_findfirst(to_search,&fileinfo); //第一次查找
13 if(-1==handle)return -1;
14 printf("%s\n",fileinfo.name); //打印出找到的文件的文件名
15 while(!_findnext(handle,&fileinfo)) //循环查找其他符合的文件,知道找不到其他的为止
16 {
17 printf("%s\n",fileinfo.name);
18 }
19 _findclose(handle); //别忘了关闭句柄
20 system("pause");
21 return 0;
22 }
当然,这个文件的查找是在指定的路径中进行,如何遍历硬盘,在整个硬盘中查找文件呢?大家可以在网络上搜索文件递归遍历等方法,这里不再做进一步介绍。
细心的朋友可能会注意到我在程序的末尾用了一个system函数。这个与程序本身并没有影响,和以前介绍给大家的使用getchar()函数的作用相同,只是为了暂停一下,让我们能看到命令提示符上输出的结果而已。不过system函数本身是一个非常强大的函数。大家可以查查MSDN看看~
简单来说,它是一个C语言与操作系统的相互平台,可以在程序里通过这个函数,向操作系统传递command命令
更多例子:
主要用到了以下几个头文件(类):io.h, fstream, string。
首先,读取某给定路径下所有文件夹与文件名称,并带完整路径。实现代码如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
void getAllFiles( string path, vector<string>& files) { //文件句柄 long 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) { files.push_back(p.assign(path).append( "\\" ).append(fileinfo.name) ); getFilesall( p.assign(path).append( "\\" ).append(fileinfo.name), files ); } } else { files.push_back(p.assign(path).append( "\\" ).append(fileinfo.name) ); } } while (_findnext(hFile, &fileinfo) == 0); _findclose(hFile); } } |
该函数有两个参数,第一个为路径字符串(string类型,最好为绝对路径);第二个参数为文件夹与文件名称存储变量(vector类型,引用传递)。
在主函数中调用格式(并将结果保存在文件"AllFiles.txt"中,第一行为总数):
1
2
3
4
5
6
7
8
9
10
11
12
|
char * filePath = "E:\\YunShi" ; vector<string> files; char * distAll = "AllFiles.txt" ; getFilesall(filePath, files); ofstream ofn(distAll); int size = files.size(); ofn<<size<<endl; for ( int i = 0;i<size;i++) { ofn<<files[i]<<endl; } ofn.close(); |
同理,只读取某给定路径下的当前文件夹名(以下类似,只给出函数,调用案例同上):
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
void getJustCurrentDir( string path, vector<string>& files) { //文件句柄 long 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) { files.push_back(fileinfo.name); //files.push_back(p.assign(path).append("\\").append(fileinfo.name) ); } } } while (_findnext(hFile, &fileinfo) == 0); _findclose(hFile); } } |
只读取某给定路径下的当前文件名:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
void getJustCurrentFile( string path, vector<string>& files) { //文件句柄 long 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)) { ; } else { files.push_back(fileinfo.name); //files.push_back(p.assign(path).append("\\").append(fileinfo.name) ); } } while (_findnext(hFile, &fileinfo) == 0); _findclose(hFile); } } |
只读取某给定路径下的所有文件名(即包含当前目录及子目录的文件):
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
void getFilesAll( string path, vector<string>& files) { //文件句柄 long 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) { //files.push_back(p.assign(path).append("\\").append(fileinfo.name) ); getFilesA( p.assign(path).append( "\\" ).append(fileinfo.name), files ); } } else { files.push_back(p.assign(path).append( "\\" ).append(fileinfo.name) ); } } while (_findnext(hFile, &fileinfo) == 0); _findclose(hFile); } } |