-// gdcmDirList.cxx
-//-----------------------------------------------------------------------------
+/*=========================================================================
+
+ Program: gdcm
+ Module: $RCSfile: gdcmDirList.cxx,v $
+ Language: C++
+ Date: $Date: 2005/01/14 21:52:06 $
+ Version: $Revision: 1.33 $
+
+ Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
+ l'Image). All rights reserved. See Doc/License.txt or
+ http://www.creatis.insa-lyon.fr/Public/Gdcm/License.html for details.
+
+ This software is distributed WITHOUT ANY WARRANTY; without even
+ the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
+ PURPOSE. See the above copyright notices for more information.
+
+=========================================================================*/
+
#include "gdcmDirList.h"
#include "gdcmUtil.h"
-#include <iostream>
-#include <algorithm>
+#include <iterator>
-#ifdef _MSC_VER
+#ifdef _MSC_VER
#include <windows.h>
#include <direct.h>
#else
#include <dirent.h>
- #include <unistd.h>
+ #include <sys/types.h>
+ #include <sys/stat.h>
#endif
+namespace gdcm
+{
// Constructor / Destructor
/**
- * \ingroup gdcmDirList
+ * \ingroup DirList
* \brief Constructor
* @param dirName root directory name
* @param recursive whether we want to explore recursively or not
*/
-gdcmDirList::gdcmDirList(std::string dirName,bool recursive)
+DirList::DirList(std::string const &dirName, bool recursive)
{
- name=dirName;
-
- NormalizePath(name);
- Explore(name,recursive);
+ DirName = dirName;
+ Explore(dirName, recursive);
}
/**
- * \ingroup gdcmDirList
+ * \ingroup DirList
* \brief Destructor
*/
-gdcmDirList::~gdcmDirList(void)
+DirList::~DirList()
{
}
//-----------------------------------------------------------------------------
// Public
/**
- * \ingroup gdcmDirList
+ * \ingroup DirList
* \brief Get the directory name
* @return the directory name
*/
-std::string gdcmDirList::GetDirName(void)
+std::string const &DirList::GetDirName() const
{
- return(name);
+ return DirName;
+}
+
+void DirList::Print(std::ostream &os)
+{
+ copy(begin(), end(), std::ostream_iterator<std::string>(os, "\n"));
}
//-----------------------------------------------------------------------------
// Private
/**
- * \ingroup gdcmDirList
* \brief Explore a directory with possibility of recursion
- * @param dirName directory to explore
+ * return number of files read
+ * @param dirpath directory to explore
* @param recursive whether we want recursion or not
*/
-void gdcmDirList::Explore(std::string dirName,bool recursive)
+int DirList::Explore(std::string const &dirpath, bool recursive)
{
+ int numberOfFiles = 0;
std::string fileName;
-
- NormalizePath(dirName);
-
-#ifdef _MSC_VER
+ std::string dirName = Util::NormalizePath(dirpath);
+#ifdef _MSC_VER
WIN32_FIND_DATA fileData;
HANDLE hFile=FindFirstFile((dirName+"*").c_str(),&fileData);
- int found=true;
+ int found = true;
- while( (hFile!=INVALID_HANDLE_VALUE) && (found) )
+ while( hFile != INVALID_HANDLE_VALUE && found )
{
- fileName=fileData.cFileName;
- if(fileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)
+ fileName = fileData.cFileName;
+ if( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
{
- if( (fileName!=".") && (fileName!="..") && (recursive) )
- Explore(dirName+fileName,recursive);
+ // Is the '.' and '..' usefull ?
+ if( fileName != "." && fileName != ".." && recursive )
+ {
+ numberOfFiles += Explore(dirName+fileName,recursive);
+ }
}
else
{
- this->push_back(dirName+fileName);
+ push_back(dirName+fileName);
+ numberOfFiles++;
}
- found=FindNextFile(hFile,&fileData);
+ found = FindNextFile(hFile, &fileData);
}
#else
- struct dirent **namelist;
- int n=scandir(dirName.c_str(), &namelist, 0, alphasort);
+ // Real POSIX implementation: scandir is a BSD extension only, and doesn't
+ // work on debian for example
+
+ DIR* dir = opendir(dirName.c_str());
+ if (!dir)
+ {
+ return 0;
+ }
- for (int i= 0;i < n; i++)
+ // 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.
+
+ struct stat buf;
+ dirent *d = 0;
+ for (d = readdir(dir); d; d = readdir(dir))
{
- fileName=namelist[i]->d_name;
- if(namelist[i]->d_type==DT_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?
+ {
+ push_back( fileName );
+ numberOfFiles++;
+ }
+ else if( S_ISDIR(buf.st_mode) ) //directory?
{
- if( (fileName!=".") && (fileName!="..") && (recursive) )
- Explore(dirName+fileName,recursive);
+ if( d->d_name[0] != '.' && recursive ) //we are also skipping hidden files
+ {
+ numberOfFiles += Explore( fileName, recursive);
+ }
}
else
{
- this->push_back(dirName+fileName);
+ // we might need to do a different treament
+ //abort();
}
}
+ closedir(dir);
#endif
+
+ return numberOfFiles;
}
+} // end namespace gdcm
//-----------------------------------------------------------------------------