]> Creatis software - crea.git/blob - src/creaFilesFromDirectory.cxx
Feature #1711 CreaDevManager application implementation
[crea.git] / src / creaFilesFromDirectory.cxx
1 /*
2 # ---------------------------------------------------------------------
3 #
4 # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image 
5 #                        pour la Santé)
6 # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
7 # Previous Authors : Laurent Guigues, Jean-Pierre Roux
8 # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
9 #
10 #  This software is governed by the CeCILL-B license under French law and 
11 #  abiding by the rules of distribution of free software. You can  use, 
12 #  modify and/ or redistribute the software under the terms of the CeCILL-B 
13 #  license as circulated by CEA, CNRS and INRIA at the following URL 
14 #  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html 
15 #  or in the file LICENSE.txt.
16 #
17 #  As a counterpart to the access to the source code and  rights to copy,
18 #  modify and redistribute granted by the license, users are provided only
19 #  with a limited warranty  and the software's author,  the holder of the
20 #  economic rights,  and the successive licensors  have only  limited
21 #  liability. 
22 #
23 #  The fact that you are presently reading this means that you have had
24 #  knowledge of the CeCILL-B license and that you accept its terms.
25 # ------------------------------------------------------------------------ 
26 */ 
27
28 /*=========================================================================
29                                                                                 
30   Program:   crea
31   Module:    $RCSfile: creaFilesFromDirectory.cxx,v $
32   Language:  C++
33   Date:      $Date: 2012/11/15 10:43:26 $
34   Version:   $Revision: 1.7 $
35                                                                                 
36    
37      This software is distributed WITHOUT ANY WARRANTY; without even
38      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
39      PURPOSE.  See the above copyright notices for more information.
40                                                                                 
41 =========================================================================*/
42
43
44 /**
45  *  \file 
46  *  \brief  class Exception:generic class for throwing any exception (header) 
47  *
48  *    Long description:
49  */
50 #ifdef _MSC_VER
51    #include <windows.h>
52    #include <direct.h>
53 #else
54    #include <dirent.h>   
55    #include <sys/types.h>
56 #endif
57
58 #include <sys/stat.h>  //stat function
59
60  #include "creaFilesFromDirectory.h"
61  #include "creaMessageManager.h"
62
63 namespace crea
64
65
66 /**
67  * \brief   Add a SEPARATOR to the end of the name if necessary
68  * @param   pathname file/directory name to normalize 
69  */
70 std::string DirList::NormalizePath(std::string const &pathname)
71 {
72
73
74 #ifdef _WIN32
75    const char FILESEPARATOR = '\\';
76 #else
77    const char FILESEPARATOR = '/';
78 #endif
79
80         
81    std::string name = pathname;
82    int size = name.size();
83
84 //   if ( name[size-1] != SEPARATOR_X && name[size-1] != SEPARATOR_WIN )
85    if ( name[size-1] != FILESEPARATOR )
86    {
87       name += FILESEPARATOR;
88    }
89    return name;
90 }  
91 /**
92  * \brief   Explore a directory with possibility of recursion
93  *          return number of files read
94  * @param  dirpath   directory to explore
95  * @param  recursive whether we want recursion or not
96  */
97 int DirList::Explore(std::string const &dirpath, bool recursive)
98 {
99    int numberOfFiles = 0;
100    std::string fileName;
101    std::string dirName = NormalizePath(dirpath);
102 #ifdef _MSC_VER
103    WIN32_FIND_DATA fileData;
104    //assert( dirName[dirName.size()-1] == '' );
105    HANDLE hFile = FindFirstFile((dirName+"*").c_str(), &fileData);
106
107    for(BOOL b = (hFile != INVALID_HANDLE_VALUE); b;
108        b = FindNextFile(hFile, &fileData))
109    {
110       fileName = fileData.cFileName;
111       if ( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
112       {
113          // Need to check for . and .. to avoid infinite loop
114          if ( fileName != "." && fileName != ".." && recursive )
115          {
116             numberOfFiles += Explore(dirName+fileName,recursive);
117          }
118       }
119       else
120       {
121          Filenames.push_back(dirName+fileName);
122          numberOfFiles++;
123       }
124    }
125    DWORD dwError = GetLastError();
126    if (hFile != INVALID_HANDLE_VALUE) 
127       FindClose(hFile);
128    if (dwError != ERROR_NO_MORE_FILES) 
129    {
130       LPVOID lpMsgBuf;
131       FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|
132                     FORMAT_MESSAGE_FROM_SYSTEM|
133                     FORMAT_MESSAGE_IGNORE_INSERTS,
134                     NULL,GetLastError(),
135                     MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
136                     (LPTSTR) &lpMsgBuf,0,NULL);
137
138       creaError("FindNextFile error. Error is " << (char *)lpMsgBuf
139                    <<" for the directory : "<<dirName);
140       return -1;
141    }
142
143 #else
144   // Real POSIX implementation: scandir is a BSD extension only, and doesn't 
145   // work on debian for example
146
147    DIR* dir = opendir(dirName.c_str());
148    if (!dir)
149    {
150       return 0;
151    }
152
153    // According to POSIX, the dirent structure contains a field char d_name[]
154    // of unspecified size, with at most NAME_MAX characters preceeding the
155    // terminating null character. Use of other fields will harm the  porta-
156    // bility of your programs.
157
158    struct stat buf;
159    dirent *d;
160    for (d = readdir(dir); d; d = readdir(dir))
161    {
162            
163       fileName = dirName + d->d_name;
164       if( stat(fileName.c_str(), &buf) != 0 )
165       {
166          //gdcmErrorMacro( strerror(errno) );
167       }
168       if ( S_ISREG(buf.st_mode) )    //is it a regular file?
169       {
170                   //printf("EED DirList::Explore [%s]\n" , d->d_name );
171                   if ( d->d_name[0]!='.')
172                   {     
173                  Filenames.push_back( fileName );
174                  numberOfFiles++;
175                   }
176       }
177       else if ( S_ISDIR(buf.st_mode) ) //directory?
178       {
179          if ( d->d_name[0] != '.' && recursive ) //we also skip hidden files
180          {
181             numberOfFiles += Explore( fileName, recursive);
182          }
183       }
184       else
185       {
186          //gdcmErrorMacro( "Unexpected error" );
187          return -1;
188       }
189    }
190    if( closedir(dir) != 0 )
191    {
192       //gdcmErrorMacro( strerror(errno) );
193    }
194 #endif
195
196   return numberOfFiles;
197 }
198
199 }