BrowseDir

#pragma once

#include <direct.h>
#include <io.h>
#include <string>
#include <vector>

size_t getFiles(std::string path, std::vector<std::string>& files, const char *filespec = "*.*", const bool& bIncludeSub = true);

size_t getFiles(std::string path, std::vector<std::string>& files, const char *filespec/* = "*.*"*/, const bool& bIncludeSub/* = true*/)
{
    if (path[path.length() - 1] != '\\')
    {
        path.append("\\");
    }

    //文件句柄
    intptr_t  hFile = 0;
    //文件信息  
    _finddata_t fileinfo;
    std::string p;
    if ((hFile = _findfirst(p.assign(path).append(filespec).c_str(), &fileinfo)) != -1)
    {
        do
        {
            //如果不是目录,加入列表  
            if (!(fileinfo.attrib &  _A_SUBDIR))
            {
                files.push_back(p.assign(path).append(fileinfo.name));
            }
        } while (_findnext(hFile, &fileinfo) == 0);
        _findclose(hFile);
    }

    if (!bIncludeSub)
    {
        return files.size();
    }

    //上面只是查找了文件,这里查找目录
    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 && !(fileinfo.attrib & _A_SYSTEM))
                    getFiles(p.assign(path).append(fileinfo.name), files, filespec, true);
            }
        } while (_findnext(hFile, &fileinfo) == 0);
        _findclose(hFile);
    }

    return files.size();
}


posted @ 2017-06-22 14:36  秋月的私语  阅读(170)  评论(0编辑  收藏  举报