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

admin

( }$ A& D. ?, I! I; }- Q8 m/ q- G[mw_shl_code=cpp,true]#include <string>
$ j0 D. p! R+ ^" Z#include <io.h>
# z. y. B& k5 r. s#include <vector>
#include <iostream>
#include <fstream>
3 U  K  r- h# Q3 B5 V, wusing namespace std;
2 {* ?4 n1 g' @9 G% v
$ J& o) W, ?% r8 T' K% W         
/************************************************************************/
/*  获取文件夹下所有文件名
    输入：     
3 ?2 y3 h1 \4 l( U5 R, u% ^        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
8 S: g3 P% P9 P% U! B/ \: q6 S6 Q    输出：
% S. n/ f% v9 D' [! o/ d        files   :   获取的文件名列表 含有路径
*/
. ?1 F/ f3 i: N% u0 E) x1 ?/************************************************************************/

void CONFIG_getFiles( string path, string exd, vector<string>& files )
{
+ x4 R* y% _0 A        //文件句柄
- a2 e& \4 r9 N( u/ W% e) ~    long   hFile   =   0;
4 H* U5 Y5 x& i% m/ b- L    //文件信息
% t% a0 a+ X6 ?, h; q9 |    struct _finddata_t fileinfo;
  w  m/ A0 X; M7 X5 l2 B' b5 E    string pathName, exdName;
7 z9 f; t: m; L; Z+ o" o  H+ U
     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
5 {9 I6 ~* ~1 Q    {
! f0 a, k+ ^. S- o        do
        {
  J! L" P3 w$ Q; r$ W2 Z/ d            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
! G/ x. f4 e9 V7 t            {
                               
" j. U' g( r4 `5 w  d2 e$ {  d              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );

% p) a7 ]  G6 ^" M! {            }
            else
            {
                                if (strstr(fileinfo.name,exd.c_str()))
( Y( ^& D' T' [7 m7 e( b' D) p                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
0 z$ G+ f0 `- d. _                                }

                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
( e! g9 H. T8 N& [7 j$ W; {, N            }
' p4 w* q/ K8 s* ~- \) J        }while(_findnext(hFile, &fileinfo)  == 0);
        _findclose(hFile);
    }
$ A1 z% ^- E0 i6 j( c}
/ f$ A2 a3 ~  k8 j) N$ {( }[/mw_shl_code]