]> Creatis software - gdcm.git/blobdiff - src/gdcmDirList.cxx
Fix bug in windows part
[gdcm.git] / src / gdcmDirList.cxx
index 938dbb3fb948af4b9e65da419ad48bc8110a64c9..641c11b63935c249baabc10342a306e82577320e 100644 (file)
@@ -3,8 +3,8 @@
   Program:   gdcm
   Module:    $RCSfile: gdcmDirList.cxx,v $
   Language:  C++
-  Date:      $Date: 2004/11/16 10:25:53 $
-  Version:   $Revision: 1.27 $
+  Date:      $Date: 2009/05/19 15:04:15 $
+  Version:   $Revision: 1.67 $
                                                                                 
   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
   l'Image). All rights reserved. See Doc/License.txt or
 
 #include "gdcmDirList.h"
 #include "gdcmUtil.h"
+#include "gdcmDebug.h"
 
-#include <iostream>
-#include <algorithm>
+#include "gdcmDicomDirImage.h"
 
-#if defined(_MSC_VER) || defined (__CYGWIN__) || defined(__BORLANDC__)
-   #include <windows.h> 
-#if defined _MSC_VER || defined(__BORLANDC__)
+#include <iterator>
+#include <assert.h>
+#include <errno.h>
+#include <sys/stat.h>  //stat function
+
+#ifdef _MSC_VER
+   #include <windows.h>
    #include <direct.h>
-#endif //_MSC_VER
 #else
    #include <dirent.h>   
-   #include <unistd.h>
-   #include <sys/stat.h>
    #include <sys/types.h>
 #endif
 
-namespace gdcm 
+namespace GDCM_NAME_SPACE
 {
+//-----------------------------------------------------------------------------
 // Constructor / Destructor
 /**
- * \ingroup DirList
  * \brief Constructor  
  * @param  dirName root directory name
  * @param  recursive whether we want to explore recursively or not 
  */
-DirList::DirList(std::string const & dirName, bool recursive)
+DirList::DirList(std::string const &dirName, bool recursive, bool all)
 {
-   name = dirName;
-   Util::NormalizePath(name);
-   Explore(name, recursive);
+   DirName = dirName;
+   Explore(dirName, recursive, all);
 }
 
 /**
- * \ingroup DirList
+ * \brief Constructor  
+ * @param  se DicomDirSerie we want to explore
+ */
+DirList::DirList(DicomDirSerie *se)
+{
+   Explore(se);
+}
+/**
  * \brief  Destructor
  */
 DirList::~DirList()
 {
 }
 
-//-----------------------------------------------------------------------------
-// Print
-
 //-----------------------------------------------------------------------------
 // Public
 /**
- * \ingroup DirList
- * \brief   Get the directory name
- * @return the directory name 
+ * \brief Tells us if file name corresponds to a Directory   
+ * @param  dirName file name to check
+ * @return true if the file IS a Directory
  */
-std::string const & DirList::GetDirName() const
+bool DirList::IsDirectory(std::string const &dirName)
 {
-   return name;
+   struct stat fs;
+   // std::cout << "dirName[dirName.size()-1] [" << dirName[dirName.size()-1] << "]"
+   //           << std::endl;
+   //assert( dirName[dirName.size()-1] != GDCM_FILESEPARATOR );
+   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
+   {
+      gdcmStaticWarningMacro("[" << dirName << "] not found or is not a Directory");
+      //gdcmStaticErrorMacro( strerror(errno) );
+      return false;
+   }
 }
 
+/**
+ * \brief   Get the first entry while visiting Filenames
+ * \return  The first  if found, otherwhise ""
+ */ 
+std::string DirList::GetFirst()
+{
+   ItDirList = Filenames.begin();
+   if (ItDirList != Filenames.end())
+      return *ItDirList;
+   return "";
+} 
+
+/**
+ * \brief   Get the next entry while visiting Filenames
+ * \return  The next  if found, otherwhise ""
+ */ 
+std::string DirList::GetNext()
+{
+   gdcmAssertMacro (ItDirList != Filenames.end())
+   {
+      ++ItDirList;
+      if (ItDirList != Filenames.end())
+         return *ItDirList;      
+   }
+   return "";
+} 
+
 //-----------------------------------------------------------------------------
 // Protected
 
 //-----------------------------------------------------------------------------
 // Private
 
+/**
+ * \brief   Explores a DicomDirSerie
+ *          return number of files found
+ * @param  se DicomDirSerie to explore
+ */
+int DirList::Explore(DicomDirSerie *se)
+{
+   int numberOfFiles = 0;
+
+   DicomDirImage *im = se->GetFirstImage();
+   while ( im ) 
+   { 
+      Filenames.push_back( im->GetEntryString(0x0004, 0x1500) );// File name (Referenced File ID)
+      numberOfFiles++;           
+      im = se->GetNextImage();   
+   }
+   return numberOfFiles;
+}
+   
 /**
  * \brief   Explore a directory with possibility of recursion
  *          return number of files read
  * @param  dirpath   directory to explore
  * @param  recursive whether we want recursion or not
+ * @param all whether we want all (i.e; File names + Directory names) default=false
  */
-int DirList::Explore(std::string const & dirpath, bool recursive)
+int DirList::Explore(std::string const &dirpath, bool recursive, bool all)
 {
    int numberOfFiles = 0;
    std::string fileName;
    std::string dirName = Util::NormalizePath(dirpath);
-#if defined(_MSC_VER) || defined(__CYGWIN__) || defined(__BORLANDC__)
-   WIN32_FIND_DATA fileData; 
-   HANDLE hFile=FindFirstFile((dirName+"*").c_str(),&fileData);
-   int found = true;
+#ifdef _MSC_VER
+   WIN32_FIND_DATA fileData;
+   //assert( dirName[dirName.size()-1] == '' );
+   HANDLE hFile = FindFirstFile((dirName+"*").c_str(), &fileData);
 
-   while( hFile != INVALID_HANDLE_VALUE && found )
+   for(BOOL b = (hFile != INVALID_HANDLE_VALUE);
+       b = FindNextFile(hFile, &fileData))
    {
-      fileName = fileData.cFileName;
-      if( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
+      fileName = dirName + fileData.cFileName;
+      // avoid infinite loop! 
+      if ( GDCM_NAME_SPACE::Util::GetName(fileName) == "." || GDCM_NAME_SPACE::Util::GetName(fileName) == "..")
+         continue;
+
+      if (all)  // meaningfull only when recursive=false
       {
-         // Is the '.' and '..' usefull ?
-         if( fileName != "." && fileName != ".." && recursive )
-         {
-            numberOfFiles += Explore(dirName+fileName,recursive);
-         }
+         Filenames.push_back( fileName );
+         numberOfFiles++;
       }
       else
       {
-         push_back(dirName+fileName);
-         numberOfFiles++;
-      }
+        if ( !( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) )    //is it a regular file?
+        {
+           Filenames.push_back( fileName );
+           numberOfFiles++;
+
+        }
+        else if ( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) //directory?
+        {
+           if ( GDCM_NAME_SPACE::Util::GetName(fileName)[0] != '.' && recursive ) //we also skip hidden files
+           {
+              numberOfFiles += Explore( fileName, recursive);
+           }
+        }
+        else
+        {
+           gdcmErrorMacro( "Unexpected error" );
+           return -1;
+        }
+     }
+   }
+
+   DWORD dwError = GetLastError();
+   if (hFile != INVALID_HANDLE_VALUE) 
+      FindClose(hFile);
+   if (dwError != ERROR_NO_MORE_FILES) 
+   {
+      LPVOID lpMsgBuf;
+      FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|
+                    FORMAT_MESSAGE_FROM_SYSTEM|
+                    FORMAT_MESSAGE_IGNORE_INSERTS,
+                    NULL,GetLastError(),
+                    MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
+                    (LPTSTR) &lpMsgBuf,0,NULL);
 
-      found = FindNextFile(hFile,&fileData);
+      gdcmErrorMacro("FindNextFile error. Error is " << (char *)lpMsgBuf
+                   <<" for the directory : "<<dirName);
+      return -1;
    }
 
 #else
@@ -126,39 +228,72 @@ int DirList::Explore(std::string const & dirpath, bool 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 preceeding 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?
+      //std::cout << d->d_name << std::endl;
+      // avoid infinite loop!  
+      if ( GDCM_NAME_SPACE::Util::GetName(d->d_name) == "." || GDCM_NAME_SPACE::Util::GetName(d->d_name) == "..")
+         continue;      
+      
+      if( stat(fileName.c_str(), &buf) != 0 )
       {
-         push_back( fileName );
-         numberOfFiles++;
+         gdcmErrorMacro( strerror(errno) );
       }
-      else if( S_ISDIR(buf.st_mode) ) //directory?
+      
+      if (all)  // meaningfull only when recursive=false
       {
-         if( d->d_name[0] != '.' && recursive ) //we are also skipping hidden files
-         {
-            numberOfFiles += Explore( fileName, recursive);
-         }
+         Filenames.push_back( fileName );
+         numberOfFiles++;      
       }
       else
       {
-         // we might need to do a different treament
-         //abort();
-      }
+        if ( S_ISREG(buf.st_mode) )    //is it a regular file?
+        {
+           Filenames.push_back( fileName );
+           numberOfFiles++;
+        }
+        else if ( S_ISDIR(buf.st_mode) ) //directory?
+        {
+           if ( d->d_name[0] != '.' && recursive ) //we also skip hidden files
+           {
+              numberOfFiles += Explore( fileName, recursive);
+           }
+        }
+        else
+        {
+           gdcmErrorMacro( "Unexpected error" );
+           return -1;
+        }
+     }
+   }
+   
+   if( closedir(dir) != 0 )
+   {
+      gdcmErrorMacro( strerror(errno) );
    }
-  closedir(dir);
 #endif
 
   return numberOfFiles;
 }
-} // end namespace gdcm
 
 //-----------------------------------------------------------------------------
+// Print
+/**
+ * \brief   Print method
+ * @param os ostream to write to 
+ */
+void DirList::Print(std::ostream &os, std::string const &)
+{
+   std::copy(Filenames.begin(), Filenames.end(), 
+             std::ostream_iterator<std::string>(os, "\n"));
+}
+
+//-----------------------------------------------------------------------------
+} // end namespace gdcm