Click here to Skip to main content
15,860,859 members
Articles / Desktop Programming / MFC
Article

Win32 file name iteration STL way

Rate me:
Please Sign up or sign in to vote.
4.96/5 (20 votes)
21 Nov 20041 min read 130.2K   668   43   33
This simple class shows how to iterate file names by using STL iterator interface.

Introduction

Several times I worked with file names, I usually used Win32 API such as ::FindFirstFile.. But it turns out that it's so boring work. Finally, I realized I can use STL's great feature, iterator, to handle file name iteration. That's why I made a simple STL iterator class for file name iteration.

Usage

win32_file_iterator itBegin("c:\\*.*"), itEnd;
std::copy(itBegin, itEnd, ostream_iterator<std::string>(cout, "\n"));

The code above shows the simplest way to use the class. Actually, you can use almost all of STL algorithm, I think..

win32_file_iterator itBegin("c:\\*.*"), itEnd;
std::vector<std::string> vec(itBegin, itEnd);

You also can fill the STL container by using the constructor that takes begin iterator and end iterator.

Actually, win32_file_iterator class' constructor takes three parameters. The first one is the filter string that is for calling ::FindFirstFile function. Second one is the flag that specifies whether dereferenced path is full path or not. For example, if it's true, the returned path string is c:\test\aa.txt, otherwise it'll be aa.txt only. The last parameter is the other flags which specify file attribute. For simplicity, I used Win32 API's FILE_ATTRIBUTE_XXX flags..

If you want to get only directory names, and which is full path, the code will look like this:

win32_file_iterator itBegin("c:\\*", true, FILE_ATTRIBUTE_DIRECTORY);

So easy, huh?

Source

#include <windows.h>
#include <iterator>
#include <string>

class win32_file_iterator : 
 public std::iterator<std::input_iterator_tag, std::string>
{
private:

 class internal_handle_data{
 public:
  internal_handle_data():_h(NULL), _ref(0){}
  void setHandle(HANDLE handle){ _h = handle; }
  HANDLE getHandle(){ return _h; }
  void incRef(){ _ref++; }
  unsigned decRef(){ return --_ref; }
  operator HANDLE(){ return _h; }

 private:
  HANDLE _h;
  unsigned _ref;
 };


public:

 win32_file_iterator(std::string strfilter, bool bFullPath = false, 
   int flag = FILE_ATTRIBUTE_ARCHIVE|FILE_ATTRIBUTE_NORMAL|FILE_ATTRIBUTE_DIRECTORY):
   _bEnd(false), _bFullPath(bFullPath), _flag(flag){
        HANDLE h = ::FindFirstFile(strfilter.c_str(), &_wfd);
  _handle.setHandle(h);
  if(h == INVALID_HANDLE_VALUE){
   _bEnd = true;
  }else{
   _handle.incRef();
   std::string::size_type n1 = strfilter.find_last_of("\\");
   _strroot = strfilter.substr(0,n1+1);
   _chkvalid(_wfd);
  }
 }

 win32_file_iterator():_bEnd(true){}

 win32_file_iterator(win32_file_iterator& rhs){
  
  _handle = rhs._handle;
  _handle.incRef();
  _flag = rhs._flag;
  _bFullPath = rhs._bFullPath;
  _bEnd = rhs._bEnd;
  _wfd = rhs._wfd;
  _strfname = rhs._strfname;
  _strroot = rhs._strroot;  
 }

 ~win32_file_iterator(){
  if(_handle.decRef() == 0 && _handle.getHandle() != NULL ){
   FindClose(_handle);
  }
  
 }

 reference operator*(){
  return _strfname;
 }

 bool operator==(const win32_file_iterator& rhs) const{
  return (_bEnd == rhs._bEnd);
 }

 bool operator!=(const win32_file_iterator& rhs) const{
  return (_bEnd != rhs._bEnd);
 }


 win32_file_iterator& operator++(){
  _findnext();
  return *this;
 }

 win32_file_iterator& operator++(int){
  _findnext();
  return *this;
 }

private:

 void _findnext(){
  BOOL b = ::FindNextFile(_handle, &_wfd);
  if(b){
   _chkvalid(_wfd);
  }else{
   _bEnd = true;
  }
 }

 void _chkvalid(WIN32_FIND_DATA& _wfd){
  if(_wfd.dwFileAttributes & _flag){
   _getval(_wfd);
  }
  else{
   _findnext();
  }
 }

 void _getval(WIN32_FIND_DATA& wfd){
  if(_bFullPath)
   _strfname = _strroot+ wfd.cFileName;
  else
   _strfname = wfd.cFileName;
 }


private:
 int _flag;
 bool _bFullPath;
 bool _bEnd;
 internal_handle_data _handle;
 WIN32_FIND_DATA _wfd;
 std::string _strroot;
 std::string _strfname;
};

Comment

The code might have many terrible bugs. But what I want was to show the way we can use STL like iteration to find filenames. I wish it'll help you. You can use this code in whatever ways you want, comments are welcome..

And also check out boost::filesystem library.. it's well-written but a little bit heavy. It needs an additional DLL, I suppose.

License

This article has no explicit license attached to it but may contain usage terms in the article text or the download files themselves. If in doubt please contact the author via the discussion board below.

A list of licenses authors might use can be found here


Written By
United States United States
study, study, That's all I can say Smile | :)

Comments and Discussions

 
GeneralMy vote of 3 Pin
buyong8-Sep-11 21:25
buyong8-Sep-11 21:25 
Questiontype &quot;reference&quot; ? Pin
Gerald E.24-Aug-05 23:39
Gerald E.24-Aug-05 23:39 
AnswerRe: type &quot;reference&quot; ? Pin
mvalle9-Sep-05 3:23
mvalle9-Sep-05 3:23 
AnswerRe: type "reference"? Pin
Hans-Martin Jensen13-May-06 1:37
Hans-Martin Jensen13-May-06 1:37 
The "reference" type is not declared for iterator in VC6. Use the equivalent "value_type&", i.e.
<br />
value_type& operator*(){<br />
  return _strfname;<br />
}<br />

GeneralSeveral issues with the published code Pin
.:floyd:.2-Dec-04 4:33
.:floyd:.2-Dec-04 4:33 
GeneralBrilliant Pin
Robert Bielik24-Nov-04 21:03
Robert Bielik24-Nov-04 21:03 
GeneralSuggestion Pin
Robert Bielik24-Nov-04 21:07
Robert Bielik24-Nov-04 21:07 
GeneralRe: Suggestion Pin
bektek25-Nov-04 8:15
bektek25-Nov-04 8:15 
GeneralOne very minor comment Pin
sdoyle24-Nov-04 6:47
sdoyle24-Nov-04 6:47 
GeneralRe: One very minor comment Pin
bektek24-Nov-04 10:44
bektek24-Nov-04 10:44 
GeneralStlSoft and WinSTL Pin
Neville Franks23-Nov-04 23:18
Neville Franks23-Nov-04 23:18 
GeneralRe: StlSoft and WinSTL Pin
bektek23-Nov-04 23:20
bektek23-Nov-04 23:20 
GeneralRe: StlSoft and WinSTL Pin
Robert Bielik25-Nov-04 9:14
Robert Bielik25-Nov-04 9:14 
GeneralRe: StlSoft and WinSTL Pin
Neville Franks25-Nov-04 9:23
Neville Franks25-Nov-04 9:23 
GeneralRe: StlSoft and WinSTL Pin
.:floyd:.2-Dec-04 5:01
.:floyd:.2-Dec-04 5:01 
GeneralRe: StlSoft and WinSTL Pin
Robert Bielik2-Dec-04 19:59
Robert Bielik2-Dec-04 19:59 
GeneralLooks nice. Pin
rtw170123-Nov-04 17:16
rtw170123-Nov-04 17:16 
GeneralTake a look at boost::filesystem Pin
Phan Manh Dan22-Nov-04 22:33
Phan Manh Dan22-Nov-04 22:33 
GeneralRe: Take a look at boost::filesystem Pin
bektek23-Nov-04 9:16
bektek23-Nov-04 9:16 
GeneralRe: Take a look at boost::filesystem Pin
occam26-Nov-04 5:37
occam26-Nov-04 5:37 
GeneralRe: Take a look at boost::filesystem Pin
Anonymous26-Nov-04 22:09
Anonymous26-Nov-04 22:09 
GeneralRe: Take a look at boost::filesystem Pin
occam27-Nov-04 1:07
occam27-Nov-04 1:07 
GeneralRe: Take a look at boost::filesystem Pin
nastanet28-Nov-04 22:24
nastanet28-Nov-04 22:24 
GeneralRe: Take a look at boost::filesystem Pin
bektek29-Nov-04 0:17
bektek29-Nov-04 0:17 
GeneralRe: Take a look at boost::filesystem Pin
nastanet29-Nov-04 0:36
nastanet29-Nov-04 0:36 

General General    News News    Suggestion Suggestion    Question Question    Bug Bug    Answer Answer    Joke Joke    Praise Praise    Rant Rant    Admin Admin   

Use Ctrl+Left/Right to switch messages, Ctrl+Up/Down to switch threads, Ctrl+Shift+Left/Right to switch pages.