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

admin

[mw_shl_code=cpp,true]#include <string>
. s4 M, h( K# z7 }#include <io.h>
#include <vector>
#include <iostream>
#include <fstream>
" g; J! b6 W) W; t! vusing namespace std;

         
$ f, U1 E5 g/ ~/************************************************************************/
/*  获取文件夹下所有文件名
    输入：     
! c. r( m. U/ Z! _* A* C        path    :   文件夹路径
* H% K+ N6 T3 C- ?" V/ L$ i        exd     :   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
) O( ^  |) m& [6 D5 ^9 {5 |                    文件名, exd = ".jt"
    输出：
$ u) B& e; v' k1 r        files   :   获取的文件名列表 含有路径
*/
/************************************************************************/
" O+ |5 K( A% Q) d
void CONFIG_getFiles( string path, string exd, vector<string>& files )
{
        //文件句柄
    long   hFile   =   0;
    //文件信息
, `( L( E0 |5 ?+ C    struct _finddata_t fileinfo;
2 l4 m  u; c" z) Q3 Z+ ?    string pathName, exdName;

     string p;
, ?2 d  ^& g/ ~- K2 p; |        if((hFile = _findfirst(pathName.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)
+ K8 j: p3 V  k% t0 \8 K    {
        do
- p& x2 A* t: d. Z5 _8 s5 }        {
            //如果是文件夹中仍有文件夹,迭代之
2 w6 X' B* M4 k" P3 g            //如果不是,加入列表
$ k# n9 g# ?% h# S* q( `0 K            // 不推荐使用，硬要使用的话，需要修改else 里面的语句
           if((fileinfo.attrib &  _A_SUBDIR))
            {
                               
9 J  a' w  H0 l              if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)
! L; F# \2 A6 G. O3 }                           CONFIG_getFiles( pathName.assign(path).append("\\").append(fileinfo.name), exd, files );
: V7 {: j8 m4 [% Z: g
            }
            else
            {
/ Q  K( x( j3 H                                if (strstr(fileinfo.name,exd.c_str()))
# S% A7 }. J' {! k5 [. z                                {
                                         files.push_back(pathName.assign(path).append("\\").append(fileinfo.name)); // 要得到绝对目录使用该语句
                                }
$ c8 M2 Y2 ]7 }  M! b/ s7 ~+ s4 l
                    //如果使用
                  //  files.push_back(fileinfo.name); // 只要得到文件名字使用该语句
( r- T; \0 m6 R2 q            }
# o& S$ j; n2 R7 ^        }while(_findnext(hFile, &fileinfo)  == 0);
& w0 B) A5 f2 x) ^        _findclose(hFile);
& m  M: g0 i8 |( ~    }
" z- Z1 r& n  m: M% Z  G}
[/mw_shl_code]