获取一个文件夹下所有指定类型的文件名

admin

[mw_shl_code=cpp,true]#include <string>
#include <io.h>
8 @9 v; t4 y2 Y  G" Q! {# `. r#include <vector>
#include <iostream>
#include <fstream>
using namespace std;

- s6 f5 f  g! y- ]( J: Y2 K         
/************************************************************************/
/*  获取文件夹下所有文件名
6 ^$ G$ r6 A; N5 A  o( u    输入：     
5 j9 k! B# e, S        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
    输出：
        files   :   获取的文件名列表 含有路径
5 B- Y; I2 U" e- q; X( _2 q*/
/************************************************************************/
7 `7 {. K! f8 j
void CONFIG_getFiles( string path, string exd, vector<string>& files )
{
3 d$ P, t, t% R# F        //文件句柄
    long   hFile   =   0;
* T! B4 U5 |7 k- J3 O/ u; r    //文件信息
( z7 [5 D8 u* J% f7 w    struct _finddata_t fileinfo;
    string pathName, exdName;

( u* {5 F( Y$ C8 c: N( u6 ?     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
    {
        do
        {
: z) d: M& K. F  V            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
6 f$ Q9 w  A! o0 b( y* q           if((fileinfo.attrib &  _A_SUBDIR))
            {
                               
6 N1 K; b1 n' N& R& _              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
3 S1 H7 M8 w! C, L  @                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
: n. @4 }5 p2 X; o+ a/ W+ }! g: H, [$ T! n
            }
7 N3 K1 d2 z: e8 O' E" p            else
* D/ i1 r! E# c            {
                                if (strstr(fileinfo.name,exd.c_str()))
4 q4 o% Q2 m* B6 s8 l) ?                                {
$ j" {% L9 o- O. }$ C! P/ H                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
; V/ A1 M7 N  c& z: d4 s                                }
) I9 U+ Y# q) m
2 L+ X9 N- e% x2 W1 D1 ]                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
            }
        }while(_findnext(hFile, &fileinfo)  == 0);
1 S9 u; @# |, I0 o  H; ~, E! r        _findclose(hFile);
    }
: E9 a# m  B$ K1 v0 [) A9 d}
! g" u# y; ^1 o" D! }+ C[/mw_shl_code]