Program: gdcm
Module: $RCSfile: gdcmDirList.cxx,v $
Language: C++
- Date: $Date: 2005/02/01 09:46:15 $
- Version: $Revision: 1.40 $
+ Date: $Date: 2005/06/24 10:55:59 $
+ Version: $Revision: 1.51 $
Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
l'Image). All rights reserved. See Doc/License.txt or
#include "gdcmUtil.h"
#include <iterator>
+#include <assert.h>
+#include <sys/stat.h> //stat function
#ifdef _MSC_VER
#include <windows.h>
#else
#include <dirent.h>
#include <sys/types.h>
- #include <sys/stat.h>
#endif
-namespace gdcm
+namespace gdcm
{
+//-----------------------------------------------------------------------------
// Constructor / Destructor
/**
* \brief Constructor
//-----------------------------------------------------------------------------
// Public
-
/**
- * \brief Print method
- * @param os ostream to write to
+ * \brief Tells us if file name corresponds to a Directory
+ * @param dirName file name to check
+ * @return true if the file IS a Directory
*/
-void DirList::Print(std::ostream &os)
+bool DirList::IsDirectory(std::string const &dirName)
{
- std::copy(Filenames.begin(), Filenames.end(),
- std::ostream_iterator<std::string>(os, "\n"));
+ struct stat fs;
+ assert( dirName[dirName.size()-1] != '/' );
+ if ( stat(dirName.c_str(), &fs) == 0 )
+ {
+#if _WIN32
+ return ((fs.st_mode & _S_IFDIR) != 0);
+#else
+ return S_ISDIR(fs.st_mode);
+#endif
+ }
+ else
+ {
+ return false;
+ }
}
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
// Private
-
/**
* \brief Explore a directory with possibility of recursion
* return number of files read
HANDLE hFile = FindFirstFile((dirName+"*").c_str(), &fileData);
for(BOOL b = (hFile != INVALID_HANDLE_VALUE); b;
- b = FindNextFile(hFile,&fileData))
+ b = FindNextFile(hFile, &fileData))
{
fileName = fileData.cFileName;
- if( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
+ if ( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
{
// Need to check for . and .. to avoid infinite loop
- if( fileName != "." && fileName != ".." && recursive )
+ if ( fileName != "." && fileName != ".." && recursive )
{
numberOfFiles += Explore(dirName+fileName,recursive);
}
}
// According to POSIX, the dirent structure contains a field char d_name[]
- // of unspecified size, with at most NAME_MAX characters preceding the
- // terminating null character. Use of other fields will harm the porta-
- // bility of your programs.
+ // of unspecified size, with at most NAME_MAX characters preceding the
+ // terminating null character. Use of other fields will harm the porta-
+ // bility of your programs.
struct stat buf;
- dirent *d = 0;
+ dirent *d;
for (d = readdir(dir); d; d = readdir(dir))
{
fileName = dirName + d->d_name;
stat(fileName.c_str(), &buf); //really discard output ?
- if( S_ISREG(buf.st_mode) ) //is it a regular file?
+ if ( S_ISREG(buf.st_mode) ) //is it a regular file?
{
Filenames.push_back( fileName );
numberOfFiles++;
}
- else if( S_ISDIR(buf.st_mode) ) //directory?
+ else if ( S_ISDIR(buf.st_mode) ) //directory?
{
- if( d->d_name[0] != '.' && recursive ) //we are also skipping hidden files
+ if ( d->d_name[0] != '.' && recursive ) //we are also skipping hidden files
{
numberOfFiles += Explore( fileName, recursive);
}
return numberOfFiles;
}
-bool DirList::IsDirectory(std::string const &dirName)
+//-----------------------------------------------------------------------------
+// Print
+/**
+ * \brief Print method
+ * @param os ostream to write to
+ */
+void DirList::Print(std::ostream &os)
{
-#ifndef _MSC_VER
- std::ifstream testDir(dirName.c_str(), std::ios::in | std::ios::binary);
- return testDir != 0;
-#else
- return (GetFileAttributes(dirName.c_str()) & FILE_ATTRIBUTE_DIRECTORY) != 0;
-#endif
+ std::copy(Filenames.begin(), Filenames.end(),
+ std::ostream_iterator<std::string>(os, "\n"));
}
-} // end namespace gdcm
-
//-----------------------------------------------------------------------------
+} // end namespace gdcm