C++遍歷目錄下文件

轉載自：http://blog.csdn.net/abcjennifer/article/details/18147551

function：遍歷目錄下所有文件，返回文件總數，子文件夾總數（修改一下可以獲得全部文件名等）。

#include "stdlib.h"
#include "direct.h"
#include "string.h"
#include "io.h"
#include "stdio.h"
#include "iostream"
using namespace std;

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());
}

已在windows上驗證有效。

轉載自這裏。

下面我加了BeginBrowseFilenames函數，以vector<char*>形式返回目錄中所有文件名。

#include "stdlib.h"
#include "direct.h"
#include "string.h"
#include "io.h"
#include "stdio.h"
#include <vector>
#include "iostream"
using namespace std;

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);
    vector<char*> BeginBrowseFilenames(const char *filespec);

protected:
    //遍歷目錄dir下由filespec指定的文件
    //對於子目錄,採用迭代的方法
    //如果返回false,表示中止遍歷文件
    bool BrowseDir(const char *dir,const char *filespec);
    vector<char*> GetDirFilenames(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;
}

vector<char*> CBrowseDir::BeginBrowseFilenames(const char *filespec)
{
    ProcessDir(m_szInitDir,NULL);
    return GetDirFilenames(m_szInitDir,filespec);
}

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;
}

vector<char*> CBrowseDir::GetDirFilenames(const char *dir,const char *filespec)
{
    _chdir(dir);
    vector<char*>filename_vector;
    filename_vector.clear();

    //首先查找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);
                filename_vector.push_back(filename);
            }
        } 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);
                    return GetDirFilenames(subdir,filespec);
                }
            }
        } while (_findnext(hFile,&fileinfo) == 0);
        _findclose(hFile);
    }
    return filename_vector;
}

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;
    }

    //開始遍歷

    vector<char*>file_vec = statdir.BeginBrowseFilenames("*.*");
    printf("文件總數: %d\n",file_vec.size());
}