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

admin

[mw_shl_code=cpp,true]#include <string>
#include <io.h>
4 r% Q! M! A& Z8 r#include <vector>
#include <iostream>
0 {6 Q2 [" e; M8 }#include <fstream>
; ?  h6 P$ d8 q' v9 o/ Rusing namespace std;
+ M- }+ B5 ^6 g: o- a
         
/************************************************************************/
' i+ f  W( Y% ]) f& F  G" \/*  获取文件夹下所有文件名
    输入：     
9 ~" i& P: N; R' r$ e; ?" t: D        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
    输出：
        files   :   获取的文件名列表 含有路径
*/
/************************************************************************/
0 a6 p" i# ?; e' R* I. o4 w
void CONFIG_getFiles( string path, string exd, vector<string>& files )
{
        //文件句柄
    long   hFile   =   0;
, `* }% G$ k4 R8 t1 x1 }6 a1 Q    //文件信息
    struct _finddata_t fileinfo;
3 g; y' D7 p. Y% s) B    string pathName, exdName;
' u. k* ?; c- T0 D
8 V( L7 q  a! R" x9 ?8 A* L     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
& W7 |- {7 N  m. n" A" d0 h8 w# r    {
        do
0 T4 w3 ^( @& }* T3 s7 W, @& D        {
            //如果是文件夹中仍有文件夹,迭代之
) q! m2 x5 Z# }0 F3 @            //如果不是,加入列表
/ N: d4 _* l/ R) _/ G) V  N# z) a            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
8 Y9 g9 q  P2 z7 X( J5 \           if((fileinfo.attrib &  _A_SUBDIR))
            {
                               
  t3 \6 t! |$ K$ C              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
1 ?) C( f) ]& j% ^8 J1 i                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
0 s, u. Y6 p" b7 i
            }
8 W1 P& C  |- ~9 }8 |            else
            {
                                if (strstr(fileinfo.name,exd.c_str()))
                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
                                }
& W. C- f: L7 P% H0 ?! d0 ?
) m4 |3 }8 D1 @, ]' g* H2 B+ b                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
5 ~! P9 ]3 o5 G8 R7 }            }
5 W$ M- a7 G; r: I" \6 `: [3 k        }while(_findnext(hFile, &fileinfo)  == 0);
& {4 O- @; C/ K        _findclose(hFile);
. P5 }4 f3 R* V    }
- m. z; U7 D/ f. w/ O& u5 U}
[/mw_shl_code]
- s' n5 Y/ z4 }. J7 a7 J3 s