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

admin

[mw_shl_code=cpp,true]#include <string>
#include <io.h>
3 C+ a* [3 l' H1 f- |#include <vector>
#include <iostream>
- {( G! T( A) f$ p#include <fstream>
using namespace std;

         
/************************************************************************/
( n4 X3 G* x8 O$ `" w/*  获取文件夹下所有文件名
    输入：     
        path    :   文件夹路径
        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
( B% {) J; e4 n: A$ z# H                    文件名, exd = ".jt"
    输出：
        files   :   获取的文件名列表 含有路径
*/
9 c# T6 ~3 f$ {2 Z! o, n4 f; ?1 q/************************************************************************/

void CONFIG_getFiles( string path, string exd, vector<string>& files )
8 F/ d( H# c- A" g4 F{
        //文件句柄
    long   hFile   =   0;
, o0 k6 r6 i4 M    //文件信息
    struct _finddata_t fileinfo;
    string pathName, exdName;
' A9 P! |2 D3 v# }' }
     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
" N- K& K' A+ A$ T# ^4 V9 b& ^    {
        do
        {
9 r" z- w+ F1 b# e2 [6 p            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
! A' p; X! i! d, K- _            {
" L" d  i- y/ q9 P+ H3 x/ y                               
; k1 z  V# M# @& e" P+ |              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
2 O7 Q$ S# Z( I1 I  j' U- C. d( s: G                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );

            }
& |; f( |6 G/ V3 E& J            else
            {
                                if (strstr(fileinfo.name,exd.c_str()))
                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
( Y4 O8 u2 A5 M6 `# F4 O                                }
# p6 L4 Q: t& Z% q  y# i- z- Z
7 `" H6 Q, L1 F1 W3 [3 q                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
            }
( P% w3 `& ^+ L; G3 ^( O        }while(_findnext(hFile, &fileinfo)  == 0);
        _findclose(hFile);
' _! K4 _3 B5 e9 n1 _( w    }
7 u2 ~& N- R! b+ g3 N7 S: q8 }* k8 e}
0 Z6 }7 f0 X) O[/mw_shl_code]