文件夹遍历

  /**************************************************   
        这是CBrowseDir的类定义文件   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);   
  };   
    
    
  /*********************************************/   
    
  //这是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   "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++;   
  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());   
  }   







 #include <iostream>   
  #include <iomanip>   
  #include <fstream>   
  #include <cstdio>   
  #include <stack>   
    
  #include <windows.h>   
    
  using namespace std;   
    
  /*   
  * 遍历lpszPath下所有文件及文件夹,并按顺序显示其中的内容.   
  */   
  /*   
  * 如果扫描到文件夹,则将其存入 Dirs 队列中,并显示名称,   
  * 如果扫描到文件,则显示其名称;   
  * 当前文件夹处理完毕后就处理其下一级文件夹,下一级文件夹从队   
  * 列中得到.   
  */   
    
  void function( LPCTSTR lpszPath,ostream & out)   
  {   
  //开始查找;   
  stack<TCHAR*> Dirs;   
  stack<int> DirDepth;   
    
  TCHAR *tmp=new TCHAR[lstrlen(lpszPath)+1];   
  lstrcpy(tmp,lpszPath);   
    
  if(tmp[lstrlen(tmp)-1]=='\\')   
  tmp[lstrlen(tmp)-1]='\0';   
    
  TCHAR szFind[MAX_PATH*2];   
  TCHAR szFile[MAX_PATH*2];   
  TCHAR *curdir;   
  int curdepth=1; //当前文件夹的深度   
    
  Dirs.push(tmp);   
  DirDepth.push(curdepth);   
    
  for(;!Dirs.empty();)   
  {   
  curdir=Dirs.top();   
  curdepth=DirDepth.top();   
  Dirs.pop();   
  DirDepth.pop();   
    
  lstrcpy(szFind,curdir);   
  lstrcat(szFind, "\\*.*"); // 找所有文件   
  WIN32_FIND_DATA wfd;   
    
  HANDLE hFind = FindFirstFile(szFind, &wfd);   
    
  if (hFind != INVALID_HANDLE_VALUE) // 如果找到   
  {   
  if (curdepth >1)   
  out<<" ";   
  for(int i=1;i<curdepth-1;++i)   
  out<<'|'<<" ";   
  out<<'+'<<curdir<<endl;   
    
  do   
  {   
    
  if (wfd.cFileName[0] == '.')   
  continue; // 过滤"." ".." 这两个目录   
    
  if (wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)   
  {   
  wsprintf(szFile, "%s\\%s", curdir, wfd.cFileName);   
    
  //function(szFile); // 如果找到的是目录，则进入此目录进行递归   
  TCHAR* tmp=new TCHAR[lstrlen(szFile)+2];   
  lstrcpy(tmp,szFile);   
  Dirs.push(tmp);   
  DirDepth.push(curdepth+1);   
  }   
  else   
  {   
  //输出文件名   
  out<<" ";   
  for(int i=1;i<curdepth;++i)   
  out<<'|'<<" ";   
  out<<wfd.cFileName<<endl;   
  }   
  } while (FindNextFile(hFind, &wfd));   
    
  }// if   
    
  delete [] curdir;   
    
  FindClose(hFind); // 关闭查找句柄   
    
  }// for()   
  }   
    
  int main(int argc,char *argv[])   
  {   
  ofstream fout("遍历结果.txt");   
  if(argc<=1)   
  {   
  cerr<<endl<<"文件夹遍历,请输入路径:";   
  TCHAR path[MAX_PATH];   
  cin>>path;   
  function(path,fout);   
  }   
  else   
  {   
  function(argv[1],fout);   
  }   
    
  return 0;   
  }