home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / wxos2240.zip / wxWindows-2.4.0 / src / common / dircmn.cpp < prev    next >
C/C++ Source or Header  |  2002-07-17  |  5KB  |  187 lines
  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: dircmn.cpp,v 1.6 2002/07/17 16:58:05 MBN Exp $
  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::HasFiles() and HasSubDirs()
  48. // ----------------------------------------------------------------------------
  49.  
  50. // dumb generic implementation
  51.  
  52. bool wxDir::HasFiles(const wxString& spec)
  53. {
  54.     wxString s;
  55.     return GetFirst(&s, spec, wxDIR_FILES | wxDIR_HIDDEN);
  56. }
  57.  
  58. // we have a (much) faster version for Unix
  59. #if (defined(__CYGWIN__) && defined(__WINDOWS__)) || !defined(__UNIX_LIKE__) || defined(__WXMAC__)
  60.  
  61. bool wxDir::HasSubDirs(const wxString& spec)
  62. {
  63.     wxString s;
  64.     return GetFirst(&s, spec, wxDIR_DIRS | wxDIR_HIDDEN);
  65. }
  66.  
  67. #endif // !Unix
  68.  
  69. // ----------------------------------------------------------------------------
  70. // wxDir::Traverse()
  71. // ----------------------------------------------------------------------------
  72.  
  73. size_t wxDir::Traverse(wxDirTraverser& sink,
  74.                        const wxString& filespec,
  75.                        int flags) const
  76. {
  77.     wxCHECK_MSG( IsOpened(), (size_t)-1,
  78.                  _T("dir must be opened before traversing it") );
  79.  
  80.     // the total number of files found
  81.     size_t nFiles = 0;
  82.  
  83.     // the name of this dir with path delimiter at the end
  84.     wxString prefix = GetName();
  85.     prefix += wxFILE_SEP_PATH;
  86.  
  87.     // first, recurse into subdirs
  88.     if ( flags & wxDIR_DIRS )
  89.     {
  90.         wxString dirname;
  91.         bool cont = GetFirst(&dirname, _T(""), wxDIR_DIRS | wxDIR_HIDDEN);
  92.         while ( cont )
  93.         {
  94.             wxDirTraverseResult res = sink.OnDir(prefix + dirname);
  95.  
  96.             if ( res == wxDIR_STOP )
  97.                 break;
  98.  
  99.             if ( res == wxDIR_CONTINUE )
  100.             {
  101.                 wxDir subdir(prefix + dirname);
  102.                 if ( subdir.IsOpened() )
  103.                 {
  104.                     nFiles += subdir.Traverse(sink, filespec, flags);
  105.                 }
  106.             }
  107.             else
  108.             {
  109.                 wxASSERT_MSG( res == wxDIR_IGNORE,
  110.                               _T("unexpected OnDir() return value") );
  111.             }
  112.  
  113.             cont = GetNext(&dirname);
  114.         }
  115.     }
  116.  
  117.     // now enum our own files
  118.     if ( flags & wxDIR_FILES )
  119.     {
  120.         flags &= ~wxDIR_DIRS;
  121.  
  122.         wxString filename;
  123.         bool cont = GetFirst(&filename, filespec, flags);
  124.         while ( cont )
  125.         {
  126.             wxDirTraverseResult res = sink.OnFile(prefix + filename);
  127.             if ( res == wxDIR_STOP )
  128.                 break;
  129.  
  130.             wxASSERT_MSG( res == wxDIR_CONTINUE,
  131.                           _T("unexpected OnFile() return value") );
  132.  
  133.             nFiles++;
  134.  
  135.             cont = GetNext(&filename);
  136.         }
  137.     }
  138.  
  139.     return nFiles;
  140. }
  141.  
  142. // ----------------------------------------------------------------------------
  143. // wxDir::GetAllFiles()
  144. // ----------------------------------------------------------------------------
  145.  
  146. class wxDirTraverserSimple : public wxDirTraverser
  147. {
  148. public:
  149.     wxDirTraverserSimple(wxArrayString& files) : m_files(files) { }
  150.  
  151.     virtual wxDirTraverseResult OnFile(const wxString& filename)
  152.     {
  153.         m_files.Add(filename);
  154.         return wxDIR_CONTINUE;
  155.     }
  156.  
  157.     virtual wxDirTraverseResult OnDir(const wxString& WXUNUSED(dirname))
  158.     {
  159.         return wxDIR_CONTINUE;
  160.     }
  161.  
  162. private:
  163.     wxArrayString& m_files;
  164. };
  165.  
  166. /* static */
  167. size_t wxDir::GetAllFiles(const wxString& dirname,
  168.                           wxArrayString *files,
  169.                           const wxString& filespec,
  170.                           int flags)
  171. {
  172.     wxCHECK_MSG( files, (size_t)-1, _T("NULL pointer in wxDir::GetAllFiles") );
  173.  
  174.     size_t nFiles = 0;
  175.  
  176.     wxDir dir(dirname);
  177.     if ( dir.IsOpened() )
  178.     {
  179.         wxDirTraverserSimple traverser(*files);
  180.  
  181.         nFiles += dir.Traverse(traverser, filespec, flags);
  182.     }
  183.  
  184.     return nFiles;
  185. }
  186.  
  187.