1 /*=========================================================================
4 Module: $RCSfile: gdcmDirList.cxx,v $
6 Date: $Date: 2006/11/15 15:53:08 $
7 Version: $Revision: 1.60 $
9 Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
10 l'Image). All rights reserved. See Doc/License.txt or
11 http://www.creatis.insa-lyon.fr/Public/Gdcm/License.html for details.
13 This software is distributed WITHOUT ANY WARRANTY; without even
14 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
15 PURPOSE. See the above copyright notices for more information.
17 =========================================================================*/
19 #include "gdcmDirList.h"
21 #include "gdcmDebug.h"
26 #include <sys/stat.h> //stat function
33 #include <sys/types.h>
38 //-----------------------------------------------------------------------------
39 // Constructor / Destructor
42 * @param dirName root directory name
43 * @param recursive whether we want to explore recursively or not
45 DirList::DirList(std::string const &dirName, bool recursive)
48 Explore(dirName, recursive);
58 //-----------------------------------------------------------------------------
61 * \brief Tells us if file name corresponds to a Directory
62 * @param dirName file name to check
63 * @return true if the file IS a Directory
65 bool DirList::IsDirectory(std::string const &dirName)
68 // std::cout << "dirName[dirName.size()-1] [" << dirName[dirName.size()-1] << "]"
70 //assert( dirName[dirName.size()-1] != GDCM_FILESEPARATOR );
71 if ( stat(dirName.c_str(), &fs) == 0 )
74 return ((fs.st_mode & _S_IFDIR) != 0);
76 return S_ISDIR(fs.st_mode);
81 gdcmStaticErrorMacro( strerror(errno) );
87 * \brief Get the first entry while visiting Filenames
88 * \return The first if found, otherwhise ""
90 std::string DirList::GetFirst()
92 ItDirList = Filenames.begin();
93 if (ItDirList != Filenames.end())
99 * \brief Get the next entry while visiting Filenames
100 * \return The next if found, otherwhise ""
102 std::string DirList::GetNext()
104 gdcmAssertMacro (ItDirList != Filenames.end())
107 if (ItDirList != Filenames.end())
113 //-----------------------------------------------------------------------------
116 //-----------------------------------------------------------------------------
119 * \brief Explore a directory with possibility of recursion
120 * return number of files read
121 * @param dirpath directory to explore
122 * @param recursive whether we want recursion or not
124 int DirList::Explore(std::string const &dirpath, bool recursive)
126 int numberOfFiles = 0;
127 std::string fileName;
128 std::string dirName = Util::NormalizePath(dirpath);
130 WIN32_FIND_DATA fileData;
131 //assert( dirName[dirName.size()-1] == '' );
132 HANDLE hFile = FindFirstFile((dirName+"*").c_str(), &fileData);
134 for(BOOL b = (hFile != INVALID_HANDLE_VALUE); b;
135 b = FindNextFile(hFile, &fileData))
137 fileName = fileData.cFileName;
138 if ( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
140 // Need to check for . and .. to avoid infinite loop
141 if ( fileName != "." && fileName != ".." && recursive )
143 numberOfFiles += Explore(dirName+fileName,recursive);
148 Filenames.push_back(dirName+fileName);
152 DWORD dwError = GetLastError();
153 if (hFile != INVALID_HANDLE_VALUE)
155 if (dwError != ERROR_NO_MORE_FILES)
158 FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|
159 FORMAT_MESSAGE_FROM_SYSTEM|
160 FORMAT_MESSAGE_IGNORE_INSERTS,
162 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
163 (LPTSTR) &lpMsgBuf,0,NULL);
165 gdcmErrorMacro("FindNextFile error. Error is " << (char *)lpMsgBuf
166 <<" for the directory : "<<dirName);
171 // Real POSIX implementation: scandir is a BSD extension only, and doesn't
172 // work on debian for example
174 DIR* dir = opendir(dirName.c_str());
180 // According to POSIX, the dirent structure contains a field char d_name[]
181 // of unspecified size, with at most NAME_MAX characters preceeding the
182 // terminating null character. Use of other fields will harm the porta-
183 // bility of your programs.
187 for (d = readdir(dir); d; d = readdir(dir))
189 fileName = dirName + d->d_name;
190 if( stat(fileName.c_str(), &buf) != 0 )
192 gdcmErrorMacro( strerror(errno) );
194 if ( S_ISREG(buf.st_mode) ) //is it a regular file?
196 Filenames.push_back( fileName );
199 else if ( S_ISDIR(buf.st_mode) ) //directory?
201 if ( d->d_name[0] != '.' && recursive ) //we also skip hidden files
203 numberOfFiles += Explore( fileName, recursive);
208 gdcmErrorMacro( "Unexpected error" );
212 if( closedir(dir) != 0 )
214 gdcmErrorMacro( strerror(errno) );
218 return numberOfFiles;
221 //-----------------------------------------------------------------------------
224 * \brief Print method
225 * @param os ostream to write to
227 void DirList::Print(std::ostream &os, std::string const &)
229 std::copy(Filenames.begin(), Filenames.end(),
230 std::ostream_iterator<std::string>(os, "\n"));
233 //-----------------------------------------------------------------------------
234 } // end namespace gdcm