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

admin

[mw_shl_code=cpp,true]#include <string>
#include <io.h>
#include <vector>
) G+ W0 c0 j, w, f8 k. F' I#include <iostream>
8 ~; k, L4 Z$ a7 G4 C6 d) ^#include <fstream>
using namespace std;

         
/************************************************************************/
/*  获取文件夹下所有文件名
    输入：     
; P- V$ |4 C2 t. o( u9 o        path    :   文件夹路径
( q5 e- i0 Q( V  E9 Y4 _        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
8 O; U! C) v+ B  M/ y  j    输出：
  o( H* A3 X/ u: S6 T5 e0 `5 h        files   :   获取的文件名列表 含有路径
*/
/************************************************************************/
( L1 d* f  Y1 [9 `3 C
void CONFIG_getFiles( string path, string exd, vector<string>& files )
! ^0 F5 t& N* }) o+ {1 r{
  ]) e2 |  l9 g: P& D        //文件句柄
    long   hFile   =   0;
    //文件信息
8 M+ e5 j( L7 N; q$ X    struct _finddata_t fileinfo;
) s/ J; b6 t3 C& [* \; {9 s    string pathName, exdName;
. [& k% x: W* x2 s
     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
    {
        do
        {
+ v/ S, q8 A, R; `            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
7 Z* T! y: T- i! r6 R            {
* P1 L4 w& [2 K( A6 J/ _* T" o                               
/ \1 k: Q$ M* u+ B0 U( T7 l              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
" A/ B( V0 Q5 @0 U: a6 _  u: R/ O                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );

9 G$ @7 b4 `0 z. y- \  f5 V            }
            else
            {
                                if (strstr(fileinfo.name,exd.c_str()))
                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
" s9 [$ \" I" j                                }
) Y. I9 H# K4 l6 \% J, H, a! k0 T3 ?
                    //如果使用
' I% ^% G1 J: I: m& Z# m3 t! n                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
2 t- r; B( h: E1 h3 @            }
# `3 L1 y5 j) {/ [        }while(_findnext(hFile, &fileinfo)  == 0);
        _findclose(hFile);
+ f* I, Z; @4 A6 L    }
}
3 }& ]' d) i9 q/ c1 C[/mw_shl_code]