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

admin

[mw_shl_code=cpp,true]#include <string>
$ U& ]" M& L$ x( H3 r#include <io.h>
#include <vector>
7 f) E0 g1 O: F$ b$ _2 p$ x0 F. m$ Q" y) V#include <iostream>
#include <fstream>
using namespace std;
1 r/ u0 j4 V% k8 ]
* g+ B8 w+ V. b4 L- h/ s         
* o5 f- c3 X( y/************************************************************************/
2 y. Q0 X9 Y: y/ \* d/*  获取文件夹下所有文件名
    输入：     
        path    :   文件夹路径
: c: x0 `& ^# U. |& p8 \        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
    输出：
) N$ N6 z1 n: m9 I) }2 g- j        files   :   获取的文件名列表 含有路径
*/
% S* s" W5 g- e' i  e3 @# q/************************************************************************/
7 v. ]$ N) A' B7 Q% U3 B, c2 e
void CONFIG_getFiles( string path, string exd, vector<string>& files )
{
        //文件句柄
( \. T% m# k& V$ A    long   hFile   =   0;
. v: T0 U! ]8 \: n* ^    //文件信息
    struct _finddata_t fileinfo;
& u$ A/ ^( u6 B- c, C    string pathName, exdName;

0 a1 s& s/ I, n* ~- H9 S     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
    {
0 s- |1 s8 I# S( p        do
+ |: |( E0 i0 t4 z4 [# \& I7 L) |        {
            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
! V: L2 f1 o0 M; f2 s' S            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
0 `8 L3 D! d4 E% Q( b           if((fileinfo.attrib &  _A_SUBDIR))
            {
                               
              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
" t0 k. w: q8 ?4 X! S6 B# E                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );

3 K9 h  n( T/ U. z% T            }
2 w! g4 ?5 I* G8 Z) N            else
            {
                                if (strstr(fileinfo.name,exd.c_str()))
                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
' i7 Y2 F, w5 {2 A+ l( s                                }

                    //如果使用
$ A  Q0 N7 q5 ]- H4 o2 w! x                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
9 G; Q7 Q% _. e: K/ f2 ]+ u) l            }
2 \3 s# u- j4 o' L( s9 }        }while(_findnext(hFile, &fileinfo)  == 0);
" F$ \' D: X% b1 P" i        _findclose(hFile);
" c1 |6 p. W7 J9 B; ^/ }4 f) G    }
0 k. J$ Y+ N, \, \( y" _, D) z+ N}
[/mw_shl_code]
/ j1 q/ ?! n& ^6 z! ]