src/common/dircmn.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/common/dircmn.cpp
   3 // Purpose:     wxDir methods common to all implementations
   4 // Author:      Vadim Zeitlin
   5 // Modified by:
   6 // Created:     19.05.01
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 2001 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
   9 // License:     wxWindows license
  10 ///////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 /* this is done in platform-specific files
  21 #ifdef __GNUG__
  22     #pragma implementation "dir.h"
  23 #endif
  24 */
  25
  26 // For compilers that support precompilation, includes "wx.h".
  27 #include "wx/wxprec.h"
  28
  29 #ifdef __BORLANDC__
  30     #pragma hdrstop
  31 #endif
  32
  33 #ifndef WX_PRECOMP
  34     #include "wx/string.h"
  35     #include "wx/log.h"
  36     #include "wx/intl.h"
  37     #include "wx/filefn.h"
  38 #endif //WX_PRECOMP
  39
  40 #include "wx/dir.h"
  41
  42 // ============================================================================
  43 // implementation
  44 // ============================================================================
  45
  46 // ----------------------------------------------------------------------------
  47 // wxDir::Traverse()
  48 // ----------------------------------------------------------------------------
  49
  50 size_t wxDir::Traverse(wxDirTraverser& sink,
  51                        const wxString& filespec,
  52                        int flags) const
  53 {
  54     wxCHECK_MSG( IsOpened(), (size_t)-1,
  55                  _T("dir must be opened before traversing it") );
  56
  57     // the total number of files found
  58     size_t nFiles = 0;
  59
  60     // the name of this dir with path delimiter at the end
  61     wxString prefix = GetName();
  62     prefix += wxFILE_SEP_PATH;
  63
  64     // first, recurse into subdirs
  65     if ( flags & wxDIR_DIRS )
  66     {
  67         wxString dirname;
  68         bool cont = GetFirst(&dirname, _T(""), wxDIR_DIRS | wxDIR_HIDDEN);
  69         while ( cont )
  70         {
  71             wxDirTraverseResult res = sink.OnDir(prefix + dirname);
  72
  73             if ( res == wxDIR_STOP )
  74                 break;
  75
  76             if ( res == wxDIR_CONTINUE )
  77             {
  78                 wxDir subdir(prefix + dirname);
  79                 if ( subdir.IsOpened() )
  80                 {
  81                     nFiles += subdir.Traverse(sink, filespec, flags);
  82                 }
  83             }
  84             else
  85             {
  86                 wxASSERT_MSG( res == wxDIR_IGNORE,
  87                               _T("unexpected OnDir() return value") );
  88             }
  89
  90             cont = GetNext(&dirname);
  91         }
  92     }
  93
  94     // now enum our own files
  95     if ( flags & wxDIR_FILES )
  96     {
  97         flags &= ~wxDIR_DIRS;
  98
  99         wxString filename;
 100         bool cont = GetFirst(&filename, filespec, flags);
 101         while ( cont )
 102         {
 103             wxDirTraverseResult res = sink.OnFile(prefix + filename);
 104             if ( res == wxDIR_STOP )
 105                 break;
 106
 107             wxASSERT_MSG( res == wxDIR_CONTINUE,
 108                           _T("unexpected OnFile() return value") );
 109
 110             nFiles++;
 111
 112             cont = GetNext(&filename);
 113         }
 114     }
 115
 116     return nFiles;
 117 }
 118
 119 // ----------------------------------------------------------------------------
 120 // wxDir::GetAllFiles()
 121 // ----------------------------------------------------------------------------
 122
 123 class wxDirTraverserSimple : public wxDirTraverser
 124 {
 125 public:
 126     wxDirTraverserSimple(wxArrayString& files) : m_files(files) { }
 127
 128     virtual wxDirTraverseResult OnFile(const wxString& filename)
 129     {
 130         m_files.Add(filename);
 131         return wxDIR_CONTINUE;
 132     }
 133
 134     virtual wxDirTraverseResult OnDir(const wxString& WXUNUSED(dirname))
 135     {
 136         return wxDIR_CONTINUE;
 137     }
 138
 139 private:
 140     wxArrayString& m_files;
 141 };
 142
 143 /* static */
 144 size_t wxDir::GetAllFiles(const wxString& dirname,
 145                           wxArrayString *files,
 146                           const wxString& filespec,
 147                           int flags)
 148 {
 149     wxCHECK_MSG( files, (size_t)-1, _T("NULL pointer in wxDir::GetAllFiles") );
 150
 151     size_t nFiles = 0;
 152
 153     wxDir dir(dirname);
 154     if ( dir.IsOpened() )
 155     {
 156         wxDirTraverserSimple traverser(*files);
 157
 158         nFiles += dir.Traverse(traverser, filespec, flags);
 159     }
 160
 161     return nFiles;
 162 }
 163