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

admin

; [8 T- ^3 g9 j[mw_shl_code=cpp,true]#include <string>
9 o! i7 o# i3 |  U( g#include <io.h>
#include <vector>
#include <iostream>
#include <fstream>
- c$ h$ X2 f' \using namespace std;
/ G& p3 o$ U" _9 L1 Z
         
/************************************************************************/
/*  获取文件夹下所有文件名
* u0 H/ q# ?& E) @$ {- m& z    输入：     
        path    :   文件夹路径
* |& |. J( C( q, I+ X        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
4 t( h  D# O3 \4 S5 B" f% \    输出：
        files   :   获取的文件名列表 含有路径
, _- i8 T7 U% C( z*/
* a& T6 g( j5 D" @% E* ?$ G4 [/************************************************************************/

; f1 D  x5 H/ }% i0 Wvoid CONFIG_getFiles( string path, string exd, vector<string>& files )
. y" ]" I$ w# j{
* t1 ]; y: b" [        //文件句柄
+ [; u( ]: s$ c% O    long   hFile   =   0;
, b' X0 Y$ A, P0 l    //文件信息
    struct _finddata_t fileinfo;
, A) _% p; M( d" I" }    string pathName, exdName;
9 g; v% q2 j; r6 j7 F
6 r9 b8 ~9 R2 }4 G& F     string p;
        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
% w5 J4 h, p, ]5 f    {
        do
        {
! n! d( z3 q  c( X  D+ v" \            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
            {
. c- z- O  n4 V" \                               
              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
) I. ^+ v9 A7 Y$ I# l                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
- Y) `( x1 ^  L7 O" Y- N; r
. F+ }% ?$ n7 k9 N            }
            else
            {
0 p' n" W# K; W1 x+ r* b! v                                if (strstr(fileinfo.name,exd.c_str()))
                                {
: \1 ]( A; V8 G5 y7 M) C% e% b$ D, ]0 o                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
                                }
0 m% I, ?; W$ t2 [6 r/ d6 [, n5 ~
. Y8 d' `! x# C$ j8 R6 E& X                    //如果使用
1 A/ r; @5 v* U* a6 l, E                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
            }
, B3 E7 k+ U" z        }while(_findnext(hFile, &fileinfo)  == 0);
        _findclose(hFile);
    }
" ~3 o8 L8 @7 X$ n+ B}
[/mw_shl_code]
: Q6 q+ P0 M+ U* U1 L* ~4 s& B# v