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

admin

) S0 g8 b2 n/ ]8 J8 G) ^8 e+ `[mw_shl_code=cpp,true]#include <string>
#include <io.h>
% q! i. }, V0 k% l#include <vector>
#include <iostream>
#include <fstream>
using namespace std;

         
/************************************************************************/
2 C" u4 T" ?: J8 r. K( C/*  获取文件夹下所有文件名
% \+ r: I6 B8 \% J3 ?2 X  g    输入：     
8 e2 E6 M9 l! r7 \5 N( @8 c        path    :   文件夹路径
# A0 M9 ^/ o; C+ B1 Y        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
                    文件名, exd = ".jt"
9 X5 M- e* x6 @; H    输出：
' q* ?: ^1 U) {* G9 W/ J        files   :   获取的文件名列表 含有路径
) ^$ k% u% h% H* m2 o/ U. g. _*/
/************************************************************************/

- h- y6 i$ S8 Hvoid CONFIG_getFiles( string path, string exd, vector<string>& files )
" i# p0 m$ j6 R) U* a. a: R{
+ [+ i7 k* A" G  ]        //文件句柄
" z9 S4 Q1 K* @    long   hFile   =   0;
    //文件信息
    struct _finddata_t fileinfo;
    string pathName, exdName;

     string p;
2 ~% V* j8 A( u: \5 w' c  h        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
    {
        do
        {
% B, }% D  i2 t3 f            //如果是文件夹中仍有文件夹,迭代之
            //如果不是,加入列表
            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
* u- A% o. I! ]' u" n3 r  d            {
) i2 ?3 _7 G% Z' ]7 t                               
! P0 A" h! x, v# X1 g              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
$ f( e% ?9 U; Y7 P/ Y
5 C8 f, z: m3 a2 Z5 ?# W/ k            }
            else
% U9 r, w) P5 w            {
3 c# s' z$ q) _! g# `                                if (strstr(fileinfo.name,exd.c_str()))
/ G" }4 }* P, }5 N( x8 |: W. b) d                                {
- ^* X* Y  U3 L8 ]2 M1 ?; W                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
+ S0 z7 ?  o& S- y; C& n                                }
& B8 z0 h2 Y+ E; P9 Z. n
                    //如果使用
! x. V! A1 y/ @7 ^) I                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
5 N  O, p: |' Q, w& T            }
. }6 ?5 l. X9 l( E9 J; ]1 c) r' n+ n        }while(_findnext(hFile, &fileinfo)  == 0);
        _findclose(hFile);
    }
) w: i# M- N$ e) w. L& x- T}
[/mw_shl_code]