您的位置:首页 > 编程语言 > C语言/C++

c++ 读取所有图片

2015-10-24 19:49 477 查看
copyright by Jun Yang, SUN YAT-SEN UNIVERSITY

//FileList.h
////////////////////////////////////////////////////
//获取目录路径下面所有文件的内容
//适用于linux和windows系统
////////////////////////////////////////////////////
#ifndef _FILELIST_H_
#define _FILELIST_H_

#include <string>
#include <vector>
#include <iostream>
using namespace std;
#ifdef _WIN32
#include <Windows.h>
#include <strsafe.h>
#else
#include <dirent.h>
#endif
//////////////////////////////////////////////////////////////////////////
//获取指定目录下所有文件的文件名,不包括文件夹,在GetFileFromDir中使用
//strDir: 输入,目录路径
//FileDirList: 输出,文件路径列表
//返回:空
//////////////////////////////////////////////////////////////////////////
vector<string> GetFileNameFromDir(string strDir)
{
vector<string> vFileDirList;
#ifdef _WIN32
WIN32_FIND_DATAA ffd;
//LARGE_INTEGER filesize;
string szDir;
//size_t length_of_arg;
HANDLE hFind = INVALID_HANDLE_VALUE;
DWORD dwError=0;

szDir=strDir+"\\*";
hFind = FindFirstFileA(szDir.c_str(), &ffd);

if (INVALID_HANDLE_VALUE == hFind)
{
cout<<"get file name error"<<endl;
return;
}
do
{
if (!(ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
{
string filename=ffd.cFileName;//(const char*)
string filedir=strDir+"\\"+filename;
vFileDirList.push_back(filedir);
}
}while (FindNextFileA(hFind, &ffd) != 0);

dwError = GetLastError();
if (dwError != ERROR_NO_MORE_FILES)
{
cout<<"FindFirstFile error"<<endl;
return;
}
FindClose(hFind);
#else
DIR *dir;
struct dirent *ptr;
dir = opendir(strDir.c_str());
while( (ptr = readdir(dir)) != NULL)
{
string filename = string(ptr->d_name);
if (filename == "." || filename == ".."){
continue;
}
string path = strDir + string("/") + filename;
vFileDirList.push_back(path);
}
closedir(dir);
sort(vFileDirList.begin(), vFileDirList.end());
#endif

return vFileDirList;
}
#endif
int main(void)
{
vector<string> files = GetFileNameFromDir("/Volumes/RamDisk/Caches");
for (int i = 0; i < files.size(); ++i){
cout << files[i] << endl;
}
return 0;

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: