用C++遍历目录

/**************************************************
   这是CBrowseDir的类定义文件 BrowseDir.h 
**************************************************/
#ifndef BROWSEDIR_H
#define BROWSEDIR_H
#include "stdlib.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);
};

#endif

/****************************/

/*********************************************

这是CBrowseDir的类实现文件 BrowseDir.cpp

***********************************************/
#include "stdlib.h"
#include "direct.h"
#include "string.h"
#include "io.h"

#include "browsedir.h"

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);
				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)
{

}


/***************************/
/*************************************************
这是例子example.cpp 
  
*************************************************/
#include "stdio.h"
#include "stdlib.h"

#include "BrowseDir.h"

//从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++;
            printf("file name -> %s\n", filename);
			return CBrowseDir::ProcessFile(filename);
		}

		//覆写虚函数ProcessDir，每调用一次，子目录个数加1
		virtual void ProcessDir(const char *currentdir,const char *parentdir)
		{
			m_nSubdirCount++;
			printf("dir name -> %s%s\n", parentdir, currentdir);
			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());
    system("pause");
}