44 lines
909 B
C++
44 lines
909 B
C++
/* 程序名：getFiles.c
|
||
功能：返回一个文件夹下的所有文件名
|
||
*/
|
||
#include<io.h>
|
||
#include <stdio.h>
|
||
#include<vector>
|
||
#include<iostream>
|
||
#include <string.h>
|
||
#include <string>
|
||
using namespace std;
|
||
|
||
int searchDir( char *path, vector<string> &dir)
|
||
{
|
||
using namespace std;
|
||
struct _finddata_t fa;//创建找到的结构体
|
||
long handle;
|
||
int flag=0;
|
||
char temp[100]={0};
|
||
string path_temp=path;
|
||
// path_temp=path_temp.substr(0,path_temp.length()-1);
|
||
|
||
if((handle = _findfirst(strcat(path,"*"),&fa)) == -1L)//如果不是目录的话
|
||
return 0;
|
||
|
||
do//是目录，先执行循环
|
||
{
|
||
if( fa.attrib == _A_SUBDIR && ~strcmp(fa.name,".")&& ~strcmp(fa.name,".."))
|
||
{
|
||
strcat( temp, path_temp.c_str());
|
||
strcat( temp, fa.name);
|
||
|
||
if(flag++)
|
||
dir.push_back(temp);
|
||
else;
|
||
|
||
memset(temp,0,100);
|
||
}
|
||
}while(_findnext(handle,&fa) == 0); /* 成功找到时返回0*/
|
||
|
||
_findclose(handle);
|
||
|
||
return 0;
|
||
|
||
} |