c/c++文件遍历

//CBrowseDir.h

#pragma once

#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.cpp

#include "CBrowseDir.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);  
				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)  
{	
	//add your own code

	return true;  
}  

void CBrowseDir::ProcessDir(const char   
	*currentdir,const char *parentdir)  
{  
	//add your own code
}  

//CStatDir.h

#pragma once
#include "CBrowseDir.h"

//从CBrowseDir派生出的子类，用来统计目录中的文件及子目录个数  
class CStatDir:public CBrowseDir  
{  
protected:  
	int m_nFileCount;   //保存文件个数  
	int m_nSubdirCount; //保存子目录个数  

public:  
	//缺省构造器  
	CStatDir() : CBrowseDir()
	{  
		//初始化数据成员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);  
	}  
}; 

//main.cpp

#include "CStatDir.h"

int main(void)
{
	
	char buf[256];  
	printf("请输入要统计的目录名:");  
	gets(buf);  

	//构造类对象  
	CStatDir statdir;  

	//设置要遍历的目录  
	if (!statdir.SetInitDir(buf))  
	{  
		puts("目录不存在。");  
		return 1;  
	}  

	//开始遍历  
	statdir.BeginBrowse("*.*");  
	printf("文件总数: %d\n子目录总数:%d\n",statdir.GetFileCount(),statdir.GetSubdirCount());  

	return 0;
}