C/C++读取文件夹下所有文件(转帖 未测试)_C/C++_编程开发_程序员俱乐部

#include "stdlib.h"
#include "direct.h"
#include "string.h"
#include "io.h"
#include "stdio.h" 
#include "iostream.h"

class CBrowseDir
{
protected:
    //存放初始目录的绝对路径，以'\'结尾
    char m_szInitDir[_MAX_PATH];
    
public:
    //缺省构造器
    CBrowseDir();
    
    //设置初始目录为dir，如果返回false，表示目录不可用
    bool SetInitDir(const char *dir);
    
    //开始遍历初始目录及其子目录下由filespec指定类型的文件
    //filespec可以使用通配符 * ?，不能包含路径。
    //如果返回false，表示遍历过程被用户中止
    bool BeginBrowse(const char *filespec);
    
protected:
    //遍历目录dir下由filespec指定的文件
    //对于子目录,采用迭代的方法
    //如果返回false,表示中止遍历文件
    bool BrowseDir(const char *dir,const char *filespec);
    
    //函数BrowseDir每找到一个文件,就调用ProcessFile
    //并把文件名作为参数传递过去
    //如果返回false,表示中止遍历文件
    //用户可以覆写该函数,加入自己的处理代码
    virtual bool ProcessFile(const char *filename);
    
    //函数BrowseDir每进入一个目录,就调用ProcessDir
    //并把正在处理的目录名及上一级目录名作为参数传递过去
    //如果正在处理的是初始目录,则parentdir=NULL
    //用户可以覆写该函数,加入自己的处理代码
    //比如用户可以在这里统计子目录的个数
    virtual void ProcessDir(const char *currentdir,const char *parentdir);
};

CBrowseDir::CBrowseDir()
{
    //用当前目录初始化m_szInitDir
    getcwd(m_szInitDir,_MAX_PATH);   
    //如果目录的最后一个字母不是'\',则在最后加上一个'\'
    int len=strlen(m_szInitDir);
    if (m_szInitDir[len-1] != '\\')
        strcat(m_szInitDir,"\\");
}

bool CBrowseDir::SetInitDir(const char *dir)
{
    //先把dir转换为绝对路径
    if (_fullpath(m_szInitDir,dir,_MAX_PATH) == NULL)
        return false;   
 
    //判断目录是否存在
    if (_chdir(m_szInitDir) != 0)
        return false;
    
    //如果目录的最后一个字母不是'\',则在最后加上一个'\'
    int len=strlen(m_szInitDir);
    if (m_szInitDir[len-1] != '\\')
        strcat(m_szInitDir,"\\");
    
    return true;
}

bool CBrowseDir::BeginBrowse(const char *filespec)
{
    ProcessDir(m_szInitDir,NULL);
    return BrowseDir(m_szInitDir,filespec);
}

bool CBrowseDir::BrowseDir(const char *dir,const char *filespec)
{
    _chdir(dir);
    
    //首先查找dir中符合要求的文件
    long hFile;
    _finddata_t fileinfo;
    if ((hFile=_findfirst(filespec,&fileinfo)) != -1)
    {
        do
        {
            //检查是不是目录
            //如果不是,则进行处理
            if (!(fileinfo.attrib & _A_SUBDIR))
            {
                char filename[_MAX_PATH];
                strcpy(filename,dir);
                strcat(filename,fileinfo.name);
                cout << filename << endl;
                if (!ProcessFile(filename))
                    return false;
            }
        } while (_findnext(hFile,&fileinfo) == 0);
        _findclose(hFile);
    }
    //查找dir中的子目录
    //因为在处理dir中的文件时，派生类的ProcessFile有可能改变了
    //当前目录，因此还要重新设置当前目录为dir。
    //执行过_findfirst后，可能系统记录下了相关信息，因此改变目录
    //对_findnext没有影响。
    _chdir(dir);
    if ((hFile=_findfirst("*.*",&fileinfo)) != -1)
    {
        do
        {
            //检查是不是目录
            //如果是,再检查是不是 . 或 .. 
            //如果不是,进行迭代
            if ((fileinfo.attrib & _A_SUBDIR))
            {
                if (strcmp(fileinfo.name,".") != 0 && strcmp
                    (fileinfo.name,"..") != 0)
                {
                    char subdir[_MAX_PATH];
                    strcpy(subdir,dir);
                    strcat(subdir,fileinfo.name);
                    strcat(subdir,"\\");
                    ProcessDir(subdir,dir);
                    if (!BrowseDir(subdir,filespec))
                        return false;
                }
            }
        } while (_findnext(hFile,&fileinfo) == 0);
        _findclose(hFile);
    }
    return true;
}

bool CBrowseDir::ProcessFile(const char *filename)
{
    return true;
}

void CBrowseDir::ProcessDir(const char 
                            *currentdir,const char *parentdir)
{
}

//从CBrowseDir派生出的子类，用来统计目录中的文件及子目录个数
class CStatDir:public CBrowseDir
{
protected:
    int m_nFileCount;   //保存文件个数
    int m_nSubdirCount; //保存子目录个数
    
public:
    //缺省构造器
    CStatDir()
    {
        //初始化数据成员m_nFileCount和m_nSubdirCount
        m_nFileCount=m_nSubdirCount=0;
    }
    
    //返回文件个数
    int GetFileCount()
    {
        return m_nFileCount;
    }
    
    //返回子目录个数
    int GetSubdirCount()
    {
        //因为进入初始目录时，也会调用函数ProcessDir，
        //所以减1后才是真正的子目录个数。
        return m_nSubdirCount-1;
    }
    
protected:
    //覆写虚函数ProcessFile，每调用一次，文件个数加1
    virtual bool ProcessFile(const char *filename)
    {
        m_nFileCount++;
        return CBrowseDir::ProcessFile(filename);
    }
    
    //覆写虚函数ProcessDir，每调用一次，子目录个数加1
    virtual void ProcessDir
        (const char *currentdir,const char *parentdir)
    {
        m_nSubdirCount++;
        CBrowseDir::ProcessDir(currentdir,parentdir);
    }
};

void main()
{
    //获取目录名
    char buf[256];
    printf("请输入要统计的目录名:");
    gets(buf);
    
    //构造类对象
    CStatDir statdir;
    
    //设置要遍历的目录
    if (!statdir.SetInitDir(buf))
    {
        puts("目录不存在。");
        return;
    }
    
    //开始遍历
    statdir.BeginBrowse("*.*");
    printf("文件总数: %d\n子目录总数:%d\n",statdir.GetFileCount(),statdir.GetSubdirCount());
} 

//这是我原来做的，是搜索目录的文件的一部分，你可以大致看一下Linux下如何使用目录属性
//代码不全，只是告诉你应该用到Linux的文件目录属性
//其他还需要你自己去做，多动手
static int searchFileTree(char* folder,int type)
{
    struct dirent* dir_ent;
    DIR*   dir;
    struct stat file_type;
    char   fullpath[PATH_MAX+1];

    memset(fullpath,0,PATH_MAX+1);
    if((dir = opendir(folder)) == NULL)//打开目录
    {
        printf("opendir failed!\n");
        return PATHERR;
    }
    while((dir_ent = readdir(dir)) != NULL){//查找目录内容
        if((dir_ent->d_name[0] == '.') ||(strcmp(dir_ent->d_name,"..")==0))
            continue;
        strcpy(fullpath,folder);
        strcat(fullpath,"/");
        strcat(fullpath,dir_ent->d_name);

        if((lstat(fullpath,&file_type)) < 0){
            continue;
        }

        if(S_ISDIR(file_type.st_mode)){
            if(searchFileTree(fullpath,type) == PATHERR){
                return PATHERR;
            }
        }else if(S_ISREG(file_type.st_mode)){
            if(isMedia(dir_ent->d_name,type) == IM_MEDIA)
                add2List(fullpath,dir_ent->d_name);
        }
    }
    closedir(dir);
    return PATHOK;
}