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

admin

# V* m" f. |% q) P4 W9 \/ R$ n[mw_shl_code=cpp,true]#include <string>
5 P! s* `8 S4 }- k; Q: ?#include <io.h>
  V1 w! u. |0 X#include <vector>
#include <iostream>
; Z% E; a7 j2 J7 [0 m6 R#include <fstream>
using namespace std;

/ Y- h& T' I5 J4 u5 r         
/************************************************************************/
. n9 Z4 Y4 U' G% Z4 F1 |7 `+ k; }. q/*  获取文件夹下所有文件名
- t3 Q* F8 |0 t6 Y! u    输入：     
        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
6 \3 I9 G: h9 E- _0 z                    文件名, exd = ".jt"
    输出：
        files   :   获取的文件名列表 含有路径
*/
0 k! X) S) ]+ U% P/************************************************************************/
1 `9 Z. g; K+ X5 P" C  _# J
void CONFIG_getFiles( string path, string exd, vector<string>& files )
3 d$ Q$ {2 U: L0 k+ \. E{
        //文件句柄
  j/ T! t) h2 Q    long   hFile   =   0;
2 P& f- }& @% _    //文件信息
    struct _finddata_t fileinfo;
    string pathName, exdName;

     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
    {
5 u, |+ @% V6 |( d9 S, m: W2 F        do
0 r8 n" L) }1 Y        {
" O3 ^. M$ E! ?) s; S8 V            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
. p. ~! i5 F" B0 ^            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
            {
                               
1 y5 A/ q3 U. T2 p! i/ j; e              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
! b+ Z1 ]' Q6 N: Z
2 f" _  K  D( ^1 K% t3 v$ C! z& \            }
            else
            {
                                if (strstr(fileinfo.name,exd.c_str()))
$ b8 Y3 u* R) ?# r                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
                                }

6 ~. `0 K' _! W- v/ ^                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
+ s* K2 A; S  Y3 H            }
% d4 k* Q) i# o* T        }while(_findnext(hFile, &fileinfo)  == 0);
( ]5 f4 l1 J3 I4 ?5 G: ]4 e/ D        _findclose(hFile);
    }
( \( u) G+ V4 v: n" p, ?4 F}
[/mw_shl_code]