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

admin

  _' l+ z# H- V$ T- f  Q' K[mw_shl_code=cpp,true]#include <string>
% \% P- E8 }9 U: M2 m3 I. M#include <io.h>
1 ?$ b# Z3 t2 a( g4 ~#include <vector>
4 j1 U. K5 k+ g( }5 a8 O#include <iostream>
4 m# S6 g0 b5 T4 j- v. H- |0 p, L#include <fstream>
using namespace std;

         
6 ~+ Y  c. l' U: c' ]. u/************************************************************************/
/*  获取文件夹下所有文件名
  R, l) A! m( i8 B! _    输入：     
; m" e% G- i' P" p4 \        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
& @) k4 z' O: M; u1 |9 v1 a                    文件名, exd = ".jt"
    输出：
        files   :   获取的文件名列表 含有路径
*/
( J# I+ U' w/ o/ p4 @. I/************************************************************************/
7 p; u; |, q. M1 S+ z3 S
void CONFIG_getFiles( string path, string exd, vector<string>& files )
{
9 k& g, J/ `7 @        //文件句柄
1 O# x: Z* }2 T# o6 ^- I+ |    long   hFile   =   0;
    //文件信息
    struct _finddata_t fileinfo;
* }4 l: r4 x6 M5 p! A& O    string pathName, exdName;

     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
* i; z% ]( V2 [    {
        do
        {
            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
: _: j4 P* P" u8 g# I/ e5 }/ \( f% o            {
                               
              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
6 y3 ?' y# w2 F# Y9 N
" o- T' y/ F! G8 E6 i/ }            }
0 W: G& ?  Y. l0 _3 v2 z6 j            else
            {
                                if (strstr(fileinfo.name,exd.c_str()))
                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
% l2 B2 ^0 t5 M: T3 @6 S                                }
  B8 E( ]7 U, ^- r5 t  e
6 x& L* B6 r* B- M& ]' M! Q                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
3 E+ `0 U5 {& W# h: Z) u% @9 e            }
1 i8 x7 C& I6 y5 l        }while(_findnext(hFile, &fileinfo)  == 0);
        _findclose(hFile);
. p* o5 f/ f$ K" d    }
}
[/mw_shl_code]