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

admin

[mw_shl_code=cpp,true]#include <string>
#include <io.h>
#include <vector>
) S, v! d) P" W9 [#include <iostream>
  Q$ D$ @6 [3 T# G#include <fstream>
6 h0 a' E2 }/ b$ ?6 f/ F' G* m9 Yusing namespace std;

         
5 v& O3 B# a  L/************************************************************************/
/*  获取文件夹下所有文件名
    输入：     
: w& U* A; p* f+ i        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
    输出：
& I& \# T/ Z, k* h        files   :   获取的文件名列表 含有路径
* X1 p9 c0 y! ^& `+ j  O7 L*/
/ U/ x6 S! c4 ^6 N' T( e/************************************************************************/

1 z2 ]1 Q4 o5 ]: K& Qvoid CONFIG_getFiles( string path, string exd, vector<string>& files )
{
        //文件句柄
: z  v) e. B. @3 _2 m8 W8 F    long   hFile   =   0;
6 \7 [& H( K0 ]' t    //文件信息
    struct _finddata_t fileinfo;
    string pathName, exdName;

     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
- q! O8 b% R0 l4 o! _6 y; w    {
1 D0 p. ~! G% N2 A/ A* ~* ]8 d        do
& Y+ n- c7 O! c$ b        {
, b/ A4 z# K  ?5 q            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
, @( ]+ g9 o2 o' U& z' |            {
                               
              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
3 }6 v$ l$ ?' b3 S8 O$ A
% ?+ `: Q0 x& L" K5 w            }
            else
            {
3 M" D/ k# h) H                                if (strstr(fileinfo.name,exd.c_str()))
                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
+ A/ B  s+ Q# V                                }

                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
* y& y+ o, p2 I& J- B            }
        }while(_findnext(hFile, &fileinfo)  == 0);
* R) [  a* u, j, L0 j; _4 Z        _findclose(hFile);
" R! T6 U; n5 {1 ~9 H3 C    }
7 X* ~2 F' F  W0 ^5 M- @9 E}
# o' d) W+ }) R" W3 R4 O[/mw_shl_code]
4 v) ?6 v! `8 ]+ C: o4 U