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

admin

! B! B) I! u- [) s) o5 x; X* \[mw_shl_code=cpp,true]#include <string>
#include <io.h>
2 f% l" W, P4 ]0 ^: `3 X3 U#include <vector>
#include <iostream>
  w6 _  O1 `$ }( b' X0 S#include <fstream>
- Z- X- o9 [- Q7 h2 n$ ousing namespace std;

         
  K; f8 P0 I: J' a1 P7 f/************************************************************************/
' r1 ]8 r/ B( I: H4 B/*  获取文件夹下所有文件名
/ }7 a6 G$ c- `% a5 g6 E0 ^9 a    输入：     
! M0 y4 o$ k: S/ B' f) ]        path    :   文件夹路径
0 b- l, D' j% q        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
% [% J4 P- P( v" ~( z8 {& f    输出：
/ M* W2 w/ V& F  _# \: u* }, X/ n        files   :   获取的文件名列表 含有路径
*/
$ k; X7 W2 t# H/ B" `/ H% M( U  n/************************************************************************/
7 x' p8 }' d' L- b; c
8 _8 r' G* l) T7 Ovoid CONFIG_getFiles( string path, string exd, vector<string>& files )
{
6 u8 a; [) T% L# N7 D% `        //文件句柄
3 k: M! B) }# t    long   hFile   =   0;
    //文件信息
6 N( W& n4 _9 O! a0 _# u2 h/ i, M    struct _finddata_t fileinfo;
    string pathName, exdName;

* Q& }& F8 W4 W& T     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
    {
; K9 E. `) I% W( Z5 k        do
1 b+ I- W2 `# c        {
            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
7 p, w8 s9 j7 W) E/ @5 {% L            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
0 W5 \  S7 W. N. S           if((fileinfo.attrib &  _A_SUBDIR))
  L2 r! v7 O9 p" \6 n3 }, K3 ?+ x            {
3 z4 B' ]; C; \* H# C( s. o" ~                               
              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
8 @$ c4 C  U; U
! K: W  t, U4 U; D4 H7 t            }
0 [6 P8 Z7 E, @9 f& z& M: c            else
5 X) }* E7 K) o6 _6 Y            {
                                if (strstr(fileinfo.name,exd.c_str()))
1 N# V* N9 j  R6 f# Z                                {
0 Q  V2 X) ]7 }- M                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
7 E; c0 f! d, `                                }
  V" o7 Z: s2 |8 \
* @- k0 K/ S; k# ]. M* q  T6 z/ j                    //如果使用
$ U9 `) |  k3 F4 H0 z! S                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
            }
        }while(_findnext(hFile, &fileinfo)  == 0);
+ S" {) H8 c" O3 z9 t        _findclose(hFile);
$ G) W; _4 \' ?, j6 Y, C    }
}
1 q1 g4 N+ b# ~" }2 \[/mw_shl_code]