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

admin

[mw_shl_code=cpp,true]#include <string>
#include <io.h>
1 ?0 _' U) i* o) B: O- n* t- A#include <vector>
% Z! k( U( R; M0 A4 K+ _#include <iostream>
#include <fstream>
using namespace std;
# w# n: r2 b/ D% f" x) n
         
" H, U) O! v7 ?3 Q/************************************************************************/
/*  获取文件夹下所有文件名
    输入：     
        path    :   文件夹路径
% f+ O/ l4 A/ E! l7 X        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
( c4 d+ y; y, X% u                    文件名, exd = ".jt"
    输出：
; z# F* A# Z7 p4 E/ |* a        files   :   获取的文件名列表 含有路径
*/
) E6 ]+ Y) Q3 @2 i3 N. ?/************************************************************************/
" T& s9 g2 C2 r* w8 M* F& u; }8 ?3 I
void CONFIG_getFiles( string path, string exd, vector<string>& files )
0 y( X2 X/ N# _/ L: K{
        //文件句柄
) M# N( f5 g/ D- F. w  Z0 ^1 I    long   hFile   =   0;
    //文件信息
    struct _finddata_t fileinfo;
" K6 c  }6 E* J; n    string pathName, exdName;
1 S$ z" u. \, {0 @! T
     string p;
2 |% k. m0 |5 u0 K' l        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
    {
. a) ^4 J$ Y5 |: x/ i+ u        do
        {
            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
* g0 z9 \: A0 L            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
# x9 m2 V4 [( C! O           if((fileinfo.attrib &  _A_SUBDIR))
6 {2 R  d7 a" J& o9 ?- l; V' m            {
% S, J  ]8 q7 g- ^6 ^! E                               
              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );

7 _: B$ |1 U& e            }
            else
" N+ Y) d; q4 W/ ?+ `            {
7 w/ \5 G6 q( X4 V                                if (strstr(fileinfo.name,exd.c_str()))
                                {
% O/ J) {, l( ^* s                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
                                }
/ v9 N5 c3 W! U0 x  a" B
9 o3 j; m& F0 d) i$ k2 Q                    //如果使用
4 E  S* {/ q1 B: A! E6 \  ^                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
8 g0 I2 }' a4 e2 j            }
' L- A$ K  t; d        }while(_findnext(hFile, &fileinfo)  == 0);
        _findclose(hFile);
7 l( P3 n/ T5 I" I    }
}
[/mw_shl_code]
% D; N$ a  q0 p. K- G3 z) w