X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=packages%2Fstd%2Fsrc%2FbbstdFilesFromDirectory.cxx;h=6e097ee1aa0a97815a22c0d0d6bbd6cbcc6c473e;hb=3b9566ad04e2a5cb0891b752fa48fd59631387ac;hp=05283bcc7017b62055760b9e5a0b1e0fe9695fa5;hpb=b459e1177db6455b4a60afe8cbb7ad608231ed95;p=bbtk.git

diff --git a/packages/std/src/bbstdFilesFromDirectory.cxx b/packages/std/src/bbstdFilesFromDirectory.cxx
index 05283bc..6e097ee 100644
--- a/packages/std/src/bbstdFilesFromDirectory.cxx
+++ b/packages/std/src/bbstdFilesFromDirectory.cxx
@@ -139,7 +139,16 @@ int FilesFromDirectory::Explore(std::string const &dirpath, bool recursive)
 			} // Type Lsast_directories
 			
 			numberOfFiles 		= 	numberOfFiles + tmpNumberOfFiles;
-         } // if
+         } // 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;
@@ -188,7 +197,7 @@ int FilesFromDirectory::Explore(std::string const &dirpath, bool recursive)
   // Real POSIX implementation: scandir is a BSD extension only, and doesn't 
   // work on debian for example
 
-   	DIR* dir = opendir(dirName.c_str());
+   	DIR* dir = opendir( dirName.c_str() );
    	if (!dir)
    	{
 		return 0;
@@ -232,7 +241,7 @@ int FilesFromDirectory::Explore(std::string const &dirpath, bool recursive)
 				} // Type files
          	} // d_name
       	} else if ( S_ISDIR(buf.st_mode) ) {  //directory?
-         	if ( d->d_name[0] != '.' && recursive ) //we also skip hidden files
+         	if ( (d->d_name[0]!='.') && recursive ) //we also skip hidden files
          	{
 				if (bbGetInputType()==1)
 				{
@@ -247,7 +256,16 @@ int FilesFromDirectory::Explore(std::string const &dirpath, bool recursive)
    	 	         	numberOfFiles++;
 				} // Type Lsast_directories
            		numberOfFiles 		= numberOfFiles+tmpNumberOfFiles;
-         	} // d_name
+         	}// 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;
@@ -269,12 +287,39 @@ 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)
 		{
@@ -286,6 +331,7 @@ void FilesFromDirectory::CleanFilenames( std::string basePath )
 		} // if spacePosition
 	}
 
+
 // Alphabetical order	
   	for (i=0; i<sizeFilenames; i++)
   	{
@@ -304,7 +350,6 @@ void FilesFromDirectory::CleanFilenames( std::string basePath )
 	unsigned int lenghtBasePath = basePath.length();
 	for (i=0; i<sizeFilenames; i++)
 	{
-
 		SimpleFilenames.push_back( Filenames[i].substr( lenghtBasePath ) );
 	} // for i