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

admin

[mw_shl_code=cpp,true]#include <string>
5 V) E. o0 Q" M+ z+ q: h$ j& R# F" }#include <io.h>
#include <vector>
#include <iostream>
#include <fstream>
using namespace std;
7 x( H4 k5 N  s9 B0 |) R* T( K
         
) r$ P* L0 O' N1 C+ F" [/************************************************************************/
. K& E  n; V! ~6 A# ]/*  获取文件夹下所有文件名
    输入：     
  @8 z% P* O3 c1 t1 ]        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
9 w8 Z  q  Z$ h* u% n, z3 F) v                    文件名, exd = ".jt"
; w" M' i7 V' |* T' e& s    输出：
        files   :   获取的文件名列表 含有路径
; E2 l5 y0 q+ j0 m. L*/
  |' ^0 h8 Q% Q. [5 G& {% f/************************************************************************/

; r' S; E" w9 rvoid CONFIG_getFiles( string path, string exd, vector<string>& files )
{
2 i" N; I2 r) L( ^# f7 _. p        //文件句柄
7 d, q# A( y! n* h% n* |    long   hFile   =   0;
7 F+ i- I7 Y6 F3 I2 l, m, J    //文件信息
    struct _finddata_t fileinfo;
    string pathName, exdName;
2 ]$ t# G# [0 y5 S3 A$ v
; b  ^; ^( N+ P$ H7 b     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
    {
        do
        {
            //如果是文件夹中仍有文件夹,迭代之
1 v" Y  B( p$ s, a3 S$ R+ T            //如果不是,加入列表
$ w( [2 B4 w6 u. l0 W4 e            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
! R" o6 G- W9 o1 ~4 N8 q# ~6 m& u            {
                               
2 `0 B! H. ]5 \+ M3 r" ]  _: {              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
; p4 d2 t; B% f( Y
            }
# z  }, F7 D7 ]9 [5 @; P# P4 _' [/ H            else
8 Q7 u4 \& j; u% V/ O& Y            {
                                if (strstr(fileinfo.name,exd.c_str()))
6 ?0 p7 f* x  i+ E8 B0 H- D                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
! h) U* S: ?. Y7 P5 u0 m' j                                }
  A! X7 |  H9 e+ P
                    //如果使用
* }1 u" d: x9 v4 Y                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
0 @. g" t0 t  g            }
, @% Q0 j( T5 l% F: ~        }while(_findnext(hFile, &fileinfo)  == 0);
        _findclose(hFile);
" u* E: h, ]2 T) \; T    }
}
[/mw_shl_code]