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

admin

' }" M6 r+ i  m: N. U; H7 O+ |[mw_shl_code=cpp,true]#include <string>
#include <io.h>
! ?+ M; n  U1 W; N#include <vector>
#include <iostream>
  m1 _: T! ^; U9 u" X) s- U! n#include <fstream>
using namespace std;

/ ~* B! @8 m( ~4 z9 v* D         
/************************************************************************/
/*  获取文件夹下所有文件名
8 `% Y& A; J7 ]" h6 Y9 a    输入：     
7 _6 ^. G& Y3 e1 N6 i        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
    输出：
' P: h* a, B4 y3 {        files   :   获取的文件名列表 含有路径
$ m7 _) ?  a; B1 ~) ]*/
' |0 T. O8 x5 c6 h) l/************************************************************************/

' M8 U$ g4 ]! T- k. W3 |void CONFIG_getFiles( string path, string exd, vector<string>& files )
; F9 }( o# {' _9 a9 ?7 t{
        //文件句柄
4 }. V( J- Z9 c    long   hFile   =   0;
    //文件信息
    struct _finddata_t fileinfo;
/ B& U; V" B) z( ~' ^4 z% [    string pathName, exdName;
1 _, S# w6 G8 ?6 d
$ o; M+ y6 r: O2 e3 ?  c     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
( x: |$ M, }6 u7 \- X$ \* e    {
        do
) v9 \  D4 c8 d8 U1 x( C        {
- X7 s( c' u$ T- G$ j+ v            //如果是文件夹中仍有文件夹,迭代之
7 S* K3 @, N0 z. C, g7 s" D            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
5 M' `  k0 [6 U  ~* G           if((fileinfo.attrib &  _A_SUBDIR))
# t% K0 F, L( Q* n( J  ^            {
- i8 ]5 d5 F  A                               
9 W. M! K/ Y2 [% F              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );

: W4 G1 V8 \$ u6 f9 S            }
            else
6 C* \- G9 h# p  A8 z            {
                                if (strstr(fileinfo.name,exd.c_str()))
' }3 Y* Q/ E/ M                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
                                }
1 l6 i) J1 i& i$ Y5 a
- z8 d% [+ d* f' h6 T& a+ w                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
: X! i. K" }) c' D' r0 T# v6 {            }
2 P( M5 k$ \$ S( M+ Z        }while(_findnext(hFile, &fileinfo)  == 0);
9 M/ f0 D8 f" |1 l. `& A        _findclose(hFile);
    }
}
+ T% w3 e3 Q5 F$ P' H. |3 g8 P[/mw_shl_code]
/ R" M$ D* Q& j6 l/ A% }) k* A