C++从某个目录中找出某一类型的所有文件

因为需要除了某个路径下的所有图片,因此通过网络搜索等方法完成了这段代码。

#include   
#include   
#include   
#include   
#include   
using namespace std;  
  
  
//获取所有的文件名  
void GetAllFiles( string path, vector& files, string format )    
{    
  
    long   hFile   =   0;    
    //文件信息    
    struct _finddata_t fileinfo;//用来存储文件信息的结构体   
	int len = format.length();
    string p,temp;    
    if((hFile = _findfirst(p.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)  //第一次查找  
    {
        do    
        {     
            if((fileinfo.attrib &  _A_SUBDIR))  //如果查找到的是文件夹  
            {    
                if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)  //进入文件夹查找  
                {  
                    //files.push_back(p.assign(path).append("\\").append(fileinfo.name) );  
                    GetAllFiles( p.assign(path).append("\\").append(fileinfo.name), files,format );   
                }  
            }    
            else //如果查找到的不是是文件夹   
            {    
                //files.push_back(p.assign(fileinfo.name) );  //将文件路径保存,也可以只保存文件名:    p.assign(path).append("\\").append(fileinfo.name)  
				temp = fileinfo.name;
				//判断字符串是否以format格式结尾
				if(temp.length()>len && temp.compare(temp.length()-len,len,format)==0)
				files.push_back(p.assign(path).append("\\").append(fileinfo.name) ); 
            }   
  
        }while(_findnext(hFile, &fileinfo)  == 0);    
  
        _findclose(hFile); //结束查找  
    }   
  
}    
  // 该函数有两个参数,第一个为路径字符串(string类型,最好为绝对路径);  
// 第二个参数为文件夹与文件名称存储变量(vector类型,引用传递)。  
// 在主函数中调用格式(并将结果保存在文件"AllFiles.txt"中,第一行为总数):  
  
int main()  
{  
   string filePath = "D:\\images";
    vector files;
    char * distAll = "AllFiles.txt";  
  
    //读取所有的文件,包括子文件的文件  
    //GetAllFiles(filePath, files);  
  
    //读取所有格式为jpg的文件  
    string format = ".jpg";  
    //GetAllFormatFiles(filePath, files,format);  
	GetAllFiles(filePath, files,format);
    ofstream ofn(distAll);  
    int size = files.size();   
    ofn<
由于之前很少用C++,在修改代码的过程中遇到了一些问题,特别是判断某个文件是不是某个类型时,用到了字符串的操作,如果有更好的,希望在评论区能留下,谢谢。



参考的文章:http://blog.csdn.net/u013094443/article/details/48896703

你可能感兴趣的:(C++学习)