X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=packages%2Fstd%2Fsrc%2FbbstdFilesFromDirectory.cxx;h=7560fd896335b377d8a6803593fc3d2304aee2a9;hb=8654fac291feca76ace7256d3f38e92edfb35919;hp=fc271c23591dd14977a420714323bd855561b9d4;hpb=de86b2c1152e6932fd8d846702aa5a72b81ded5c;p=bbtk.git

diff --git a/packages/std/src/bbstdFilesFromDirectory.cxx b/packages/std/src/bbstdFilesFromDirectory.cxx
index fc271c2..7560fd8 100644
--- a/packages/std/src/bbstdFilesFromDirectory.cxx
+++ b/packages/std/src/bbstdFilesFromDirectory.cxx
@@ -47,19 +47,21 @@ BBTK_BLACK_BOX_IMPLEMENTATION(FilesFromDirectory,bbtk::AtomicBlackBox);
 
 void FilesFromDirectory::Process()
 {
-   DirName = bbGetInputIn();
-   bool rec = bbGetInputRecursive();
-   /*int nbFiles = */ Explore(DirName, rec);
-   bbSetOutputOut(Filenames);   
-   
-//  for (int i=0; i<Filenames.size(); i++)
-//     std::cout << "Filenames [" << i << "] = [" << Filenames[i] << "]" << std::endl;  
+	Filenames.clear();
+	SimpleFilenames.clear();
+   	DirName 	= bbGetInputIn();
+   	bool rec 	= bbGetInputRecursive();
+	/*int nbFiles = */ Explore(DirName, rec);
+	CleanFilenames( bbGetInputIn() );
+   	bbSetOutputOut(Filenames);   
+  	bbSetOutputOutSimple(SimpleFilenames);   
 }
 
 void FilesFromDirectory::bbUserSetDefaultValues()
 {
     bbSetInputIn(".");
     bbSetInputRecursive(false);  
+    bbSetInputType(0);  
 }
 
 void FilesFromDirectory::bbUserInitializeProcessing() 
@@ -98,6 +100,265 @@ std::string FilesFromDirectory::NormalizePath(std::string const &pathname)
  * @param  dirpath   directory to explore
  * @param  recursive whether we want recursion or not
  */
+
+
+
+int FilesFromDirectory::Explore(std::string const &dirpath, bool recursive)
+{
+   	int numberOfFiles 		= 0;
+   	std::string dirName 	= NormalizePath(dirpath);
+   	int tmpNumberOfFiles;
+   	std::string fileName;
+#ifdef _MSC_VER
+   	WIN32_FIND_DATA fileData;
+   	//assert( dirName[dirName.size()-1] == '' );
+   	HANDLE hFile = FindFirstFile((dirName+"*").c_str(), &fileData);
+
+   	for(BOOL b = (hFile != INVALID_HANDLE_VALUE); b;
+       b = FindNextFile(hFile, &fileData))
+   	{
+      fileName = fileData.cFileName;
+      if ( fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
+      {
+         // Need to check for . and .. to avoid infinite loop
+         if ( fileName != "." && fileName != ".." && recursive )
+         {
+			if (bbGetInputType()==1)
+			{
+		        std::string temp = dirName+fileName;
+	       		Filenames.push_back(temp);	 
+   	 	     	numberOfFiles++;
+			} // Type ALL_directories
+
+			tmpNumberOfFiles	=	Explore(dirName+fileName, recursive);
+			if ((bbGetInputType()==2)&&tmpNumberOfFiles==0)
+			{
+		        std::string temp = dirName+fileName;
+	       		Filenames.push_back(temp);	 
+   	 	        numberOfFiles++;
+			} // Type Lsast_directories
+			
+			numberOfFiles 		= 	numberOfFiles + tmpNumberOfFiles;
+         } // if recursive
+         if ( fileName != "." && fileName != ".." && !recursive )
+			if ((bbGetInputType()==1) || (bbGetInputType()==2))
+			{
+		        std::string temp = dirName+fileName;
+	           	Filenames.push_back(temp);	 
+   	 	       	numberOfFiles++;
+			} // Type All_directories
+		}// fileName && !recursive
+      } else  {
+//      std::string temp = "\"" +dirName+fileName + "\"";
+        std::string temp = dirName+fileName;
+
+/*
+        std::string::size_type spacePosition = temp.find_first_of(' ');
+		if (spacePosition != std::string::npos) 
+	 	{
+   			std::cout << "=========================================== File name : [" <<temp << 
+              "] contains space(s); Discarted !" << std::endl;
+	    	temp.insert(spacePosition, "\\");
+   			continue;  /// \TODO : fix the trouble (vtk?)
+	 	} // if !npos
+*/
+
+
+		if (bbGetInputType()==0)
+		{
+	   		Filenames.push_back(temp);	 
+   			numberOfFiles++;
+		} // Type files
+
+      } // if !directory
+   	} // for
+	DWORD dwError = GetLastError();
+	if (hFile != INVALID_HANDLE_VALUE) 
+	{
+   		FindClose(hFile);
+	}// 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);
+
+      //gdcmErrorMacro("FindNextFile error. Error is " << (char *)lpMsgBuf
+      //             <<" for the directory : "<<dirName);
+      return -1;
+   	} // dwError
+
+#else
+  // 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;
+   	}
+
+   // According to POSIX, the dirent structure contains a field char d_name[]
+   // 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;
+   	for (d = readdir(dir); d; d = readdir(dir))
+   	{
+     	fileName = dirName + d->d_name;
+      	std::string temp = fileName;
+      	if( stat(fileName.c_str(), &buf) != 0 )
+      	{
+        	 //gdcmErrorMacro( strerror(errno) );
+      	} // stat
+      	if ( S_ISREG(buf.st_mode) )    //is it a regular file?
+      	{
+         	if ( d->d_name[0]!='.')
+         	{
+
+/*
+          		std::string::size_type  spacePosition = temp.find_first_of(' ');
+             	if (spacePosition != std::string::npos)
+	     		{
+   					std::cout << "=========================================== File name : [" <<temp << 
+              			"] contains space(s); Discarted !" << std::endl;
+                 	temp.insert(spacePosition, "\\");
+   					continue;   /// \TODO : fix the trouble (vtk?)
+             	} // if spacePosition
+*/
+
+				if (bbGetInputType()==0)
+				{
+	             	Filenames.push_back(temp);	 
+   	 	         	numberOfFiles++;
+				} // Type files
+         	} // d_name
+      	} else if ( S_ISDIR(buf.st_mode) ) {  //directory?
+         	if ( (d->d_name[0]!='.') && recursive ) //we also skip hidden files
+         	{
+				if (bbGetInputType()==1)
+				{
+	             	Filenames.push_back(fileName);	 
+   	 	         	numberOfFiles++;
+				} // Type All_directories
+
+				tmpNumberOfFiles	= Explore( fileName, recursive);
+				if ((bbGetInputType()==2)&&tmpNumberOfFiles==0)
+				{
+			  		Filenames.push_back(fileName);	 
+   	 	         	numberOfFiles++;
+				} // Type Lsast_directories
+           		numberOfFiles 		= numberOfFiles+tmpNumberOfFiles;
+         	}// d_name && recursive
+         	if ( (d->d_name[0]!='.') && !recursive ) { //we also skip hidden files
+				if ((bbGetInputType()==1) || (bbGetInputType()==2))
+				{
+	             	Filenames.push_back(fileName);	 
+   	 	         	numberOfFiles++;
+				} // Type All_directories
+			
+			}// d_name && !recursive
+
+		} else {
+         	//gdcmErrorMacro( "Unexpected error" );
+         	return -1;
+      	} // Regular FILE
+   }
+   if( closedir(dir) != 0 )
+   {
+      //gdcmErrorMacro( strerror(errno) );
+   }// closedir
+#endif
+
+
+  	return numberOfFiles;
+}
+
+
+//------------------------------------------------------------------------------
+void FilesFromDirectory::CleanFilenames( std::string basePath )
+{
+  	std::string tmpString;
+  	int i,ii,sizeFilenames = Filenames.size();
+	int j,sizeFileEnd = bbGetInputFileEnd().size();
+
+	int okEraseElement;
+	
+//Selecting just the files in the FileEnd List
+	if (bbGetInputFileEnd().size()!=0)
+	{
+		for (i=sizeFilenames-1; i>=0; i--)
+		{
+			okEraseElement=0;
+			for (j=0; j<sizeFileEnd; j++)
+			{
+				std::string tmp=Filenames[i].substr( (Filenames[i].length()) - (bbGetInputFileEnd()[j]).length() ) ;
+				if (bbGetInputFileEnd()[j].compare( tmp )==0 )
+				{
+					okEraseElement++;
+				} // if FileEnd
+			} // for j					
+
+			if (okEraseElement==0) 
+			{ 
+				Filenames.erase( Filenames.begin()+i );
+			}
+		} // for i	
+	} // if FileEnd !=0
+
+	sizeFilenames = Filenames.size();
+
+// Cleanning paths with spaces	
+	for (i=0; i<sizeFilenames; i++)
+	{
+		tmpString=Filenames[i];
+
+		std::string::size_type  spacePosition = tmpString.find_first_of(' ');
+		if (spacePosition != std::string::npos)
+		{
+			std::cout << "=========================================== File name : [" <<tmpString << 
+            	"] contains space(s); Discarted !" << std::endl;
+			tmpString.insert(spacePosition, "\\");
+			Filenames[i]=tmpString;
+//			continue;   /// \TODO : fix the trouble (vtk?)
+		} // if spacePosition
+	}
+
+
+// Alphabetical order	
+  	for (i=0; i<sizeFilenames; i++)
+  	{
+    	for (ii=i; ii<sizeFilenames; ii++)
+   	 	{
+			if (Filenames[i]>Filenames[ii]) 
+        	{
+          		tmpString=Filenames[i];
+          		Filenames[i]=Filenames[ii];
+          		Filenames[ii]=tmpString;
+        	} // if 
+    	} // for ii
+  	} // for i
+
+// Creating SimpleFilenames
+	unsigned int lenghtBasePath = basePath.length();
+	for (i=0; i<sizeFilenames; i++)
+	{
+		SimpleFilenames.push_back( Filenames[i].substr( lenghtBasePath ) );
+	} // for i
+
+}
+
+
+
+
+/*
+
 int FilesFromDirectory::Explore(std::string const &dirpath, bool recursive)
 {
    Filenames.clear();
@@ -185,7 +446,7 @@ int FilesFromDirectory::Explore(std::string const &dirpath, bool recursive)
          if ( d->d_name[0]!='.')
          {
 	 
-             std::string::size_type /* long int */ spacePosition = temp.find_first_of(' ');
+             std::string::size_type  spacePosition = temp.find_first_of(' ');
              if (spacePosition != std::string::npos)
 	     {
    std::cout << "=========================================== File name : [" <<temp << 
@@ -233,7 +494,7 @@ int FilesFromDirectory::Explore(std::string const &dirpath, bool recursive)
 
   return numberOfFiles;
 }
-
+*/
 
 }
 // EO namespace bbstd