// Created: 19.05.01
// RCS-ID: $Id$
// Copyright: (c) 2001 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
-// License: wxWindows license
+// License: wxWindows licence
///////////////////////////////////////////////////////////////////////////////
// ============================================================================
// headers
// ----------------------------------------------------------------------------
-/* this is done in platform-specific files
-#ifdef __GNUG__
- #pragma implementation "dir.h"
-#endif
-*/
-
// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"
#include "wx/log.h"
#include "wx/intl.h"
#include "wx/filefn.h"
+ #include "wx/arrstr.h"
#endif //WX_PRECOMP
#include "wx/dir.h"
+#include "wx/filename.h"
// ============================================================================
// implementation
// ============================================================================
+// ----------------------------------------------------------------------------
+// wxDirTraverser
+// ----------------------------------------------------------------------------
+
+wxDirTraverseResult
+wxDirTraverser::OnOpenError(const wxString& WXUNUSED(dirname))
+{
+ return wxDIR_IGNORE;
+}
+
// ----------------------------------------------------------------------------
// wxDir::HasFiles() and HasSubDirs()
// ----------------------------------------------------------------------------
// dumb generic implementation
-bool wxDir::HasFiles(const wxString& spec)
+bool wxDir::HasFiles(const wxString& spec) const
{
wxString s;
return GetFirst(&s, spec, wxDIR_FILES | wxDIR_HIDDEN);
}
// we have a (much) faster version for Unix
-#ifndef __UNIX_LIKE__
+#if (defined(__CYGWIN__) && defined(__WINDOWS__)) || !defined(__UNIX_LIKE__) || defined(__EMX__) || defined(__WINE__)
-bool wxDir::HasSubDirs(const wxString& spec)
+bool wxDir::HasSubDirs(const wxString& spec) const
{
wxString s;
return GetFirst(&s, spec, wxDIR_DIRS | wxDIR_HIDDEN);
int flags) const
{
wxCHECK_MSG( IsOpened(), (size_t)-1,
- _T("dir must be opened before traversing it") );
+ wxT("dir must be opened before traversing it") );
// the total number of files found
size_t nFiles = 0;
if ( flags & wxDIR_DIRS )
{
wxString dirname;
- bool cont = GetFirst(&dirname, _T(""), wxDIR_DIRS | wxDIR_HIDDEN);
- while ( cont )
+ for ( bool cont = GetFirst(&dirname, wxEmptyString, wxDIR_DIRS | (flags & wxDIR_HIDDEN) );
+ cont;
+ cont = cont && GetNext(&dirname) )
{
- wxDirTraverseResult res = sink.OnDir(prefix + dirname);
+ const wxString fulldirname = prefix + dirname;
- if ( res == wxDIR_STOP )
- break;
-
- if ( res == wxDIR_CONTINUE )
- {
- wxDir subdir(prefix + dirname);
- if ( subdir.IsOpened() )
- {
- nFiles += subdir.Traverse(sink, filespec, flags);
- }
- }
- else
+ switch ( sink.OnDir(fulldirname) )
{
- wxASSERT_MSG( res == wxDIR_IGNORE,
- _T("unexpected OnDir() return value") );
+ default:
+ wxFAIL_MSG(wxT("unexpected OnDir() return value") );
+ // fall through
+
+ case wxDIR_STOP:
+ cont = false;
+ break;
+
+ case wxDIR_CONTINUE:
+ {
+ wxDir subdir;
+
+ // don't give the error messages for the directories
+ // which we can't open: there can be all sorts of good
+ // reason for this (e.g. insufficient privileges) and
+ // this shouldn't be treated as an error -- instead
+ // let the user code decide what to do
+ bool ok;
+ do
+ {
+ wxLogNull noLog;
+ ok = subdir.Open(fulldirname);
+ if ( !ok )
+ {
+ // ask the user code what to do
+ bool tryagain;
+ switch ( sink.OnOpenError(fulldirname) )
+ {
+ default:
+ wxFAIL_MSG(wxT("unexpected OnOpenError() return value") );
+ // fall through
+
+ case wxDIR_STOP:
+ cont = false;
+ // fall through
+
+ case wxDIR_IGNORE:
+ tryagain = false;
+ break;
+
+ case wxDIR_CONTINUE:
+ tryagain = true;
+ }
+
+ if ( !tryagain )
+ break;
+ }
+ }
+ while ( !ok );
+
+ if ( ok )
+ {
+ nFiles += subdir.Traverse(sink, filespec, flags);
+ }
+ }
+ break;
+
+ case wxDIR_IGNORE:
+ // nothing to do
+ ;
}
-
- cont = GetNext(&dirname);
}
}
break;
wxASSERT_MSG( res == wxDIR_CONTINUE,
- _T("unexpected OnFile() return value") );
+ wxT("unexpected OnFile() return value") );
nFiles++;
virtual wxDirTraverseResult OnFile(const wxString& filename)
{
- m_files.Add(filename);
+ m_files.push_back(filename);
return wxDIR_CONTINUE;
}
private:
wxArrayString& m_files;
+
+ wxDECLARE_NO_COPY_CLASS(wxDirTraverserSimple);
};
/* static */
const wxString& filespec,
int flags)
{
- wxCHECK_MSG( files, (size_t)-1, _T("NULL pointer in wxDir::GetAllFiles") );
+ wxCHECK_MSG( files, (size_t)-1, wxT("NULL pointer in wxDir::GetAllFiles") );
size_t nFiles = 0;
return nFiles;
}
+// ----------------------------------------------------------------------------
+// wxDir::FindFirst()
+// ----------------------------------------------------------------------------
+
+class wxDirTraverserFindFirst : public wxDirTraverser
+{
+public:
+ wxDirTraverserFindFirst() { }
+
+ virtual wxDirTraverseResult OnFile(const wxString& filename)
+ {
+ m_file = filename;
+ return wxDIR_STOP;
+ }
+
+ virtual wxDirTraverseResult OnDir(const wxString& WXUNUSED(dirname))
+ {
+ return wxDIR_CONTINUE;
+ }
+
+ const wxString& GetFile() const
+ {
+ return m_file;
+ }
+
+private:
+ wxString m_file;
+
+ wxDECLARE_NO_COPY_CLASS(wxDirTraverserFindFirst);
+};
+
+/* static */
+wxString wxDir::FindFirst(const wxString& dirname,
+ const wxString& filespec,
+ int flags)
+{
+ wxDir dir(dirname);
+ if ( dir.IsOpened() )
+ {
+ wxDirTraverserFindFirst traverser;
+
+ dir.Traverse(traverser, filespec, flags | wxDIR_FILES);
+ return traverser.GetFile();
+ }
+
+ return wxEmptyString;
+}
+
+
+// ----------------------------------------------------------------------------
+// wxDir::GetTotalSize()
+// ----------------------------------------------------------------------------
+
+#if wxUSE_LONGLONG
+
+class wxDirTraverserSumSize : public wxDirTraverser
+{
+public:
+ wxDirTraverserSumSize() { }
+
+ virtual wxDirTraverseResult OnFile(const wxString& filename)
+ {
+ wxULongLong sz = wxFileName::GetSize(filename);
+
+ // wxFileName::GetSize won't use this class again as
+ // we're passing it a file and not a directory;
+ // thus we are sure to avoid an endless loop
+ if (sz == wxInvalidSize)
+ {
+ // if the GetSize() failed (this can happen because e.g. a
+ // file is locked by another process), we can proceed but
+ // we need to at least warn the user that the resulting
+ // final size could be not reliable (if e.g. the locked
+ // file is very big).
+ m_skippedFiles.Add(filename);
+ return wxDIR_CONTINUE;
+ }
+
+ m_sz += sz;
+ return wxDIR_CONTINUE;
+ }
+
+ virtual wxDirTraverseResult OnDir(const wxString& WXUNUSED(dirname))
+ {
+ return wxDIR_CONTINUE;
+ }
+
+ wxULongLong GetTotalSize() const
+ { return m_sz; }
+ wxArrayString &FilesSkipped()
+ { return m_skippedFiles; }
+
+protected:
+ wxULongLong m_sz;
+ wxArrayString m_skippedFiles;
+};
+
+wxULongLong wxDir::GetTotalSize(const wxString &dirname, wxArrayString *filesSkipped)
+{
+ if (!wxDirExists(dirname))
+ return wxInvalidSize;
+
+ // to get the size of this directory and its contents we need
+ // to recursively walk it...
+ wxDir dir(dirname);
+ if ( !dir.IsOpened() )
+ return wxInvalidSize;
+
+ wxDirTraverserSumSize traverser;
+ if (dir.Traverse(traverser) == (size_t)-1 ||
+ traverser.GetTotalSize() == 0)
+ return wxInvalidSize;
+
+ if (filesSkipped)
+ *filesSkipped = traverser.FilesSkipped();
+
+ return traverser.GetTotalSize();
+}
+
+#endif // wxUSE_LONGLONG
projects / wxWidgets.git / blobdiff