/*
 # ---------------------------------------------------------------------
 #
 # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
 #                        pour la SantÈ)
 # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
 # Previous Authors : Laurent Guigues, Jean-Pierre Roux
 # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
 #
 #  This software is governed by the CeCILL-B license under French law and
 #  abiding by the rules of distribution of free software. You can  use,
 #  modify and/ or redistribute the software under the terms of the CeCILL-B
 #  license as circulated by CEA, CNRS and INRIA at the following URL
 #  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
 #  or in the file LICENSE.txt.
 #
 #  As a counterpart to the access to the source code and  rights to copy,
 #  modify and redistribute granted by the license, users are provided only
 #  with a limited warranty  and the software's author,  the holder of the
 #  economic rights,  and the successive licensors  have only  limited
 #  liability.
 #
 #  The fact that you are presently reading this means that you have had
 #  knowledge of the CeCILL-B license and that you accept its terms.
 # ------------------------------------------------------------------------ */


#include "bbstdFilesFromDirectory.h"
#include "bbstdPackage.h"
#include <string>

#ifdef _MSC_VER
   #include <windows.h>
   #include <direct.h>
#else
   #include <dirent.h>   
   #include <sys/types.h>
#endif

#include <sys/stat.h>  //stat function

namespace bbstd
{

BBTK_ADD_BLACK_BOX_TO_PACKAGE(std,FilesFromDirectory)
BBTK_BLACK_BOX_IMPLEMENTATION(FilesFromDirectory,bbtk::AtomicBlackBox);

void FilesFromDirectory::Process()
{
	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() 
{ 
}

void FilesFromDirectory::bbUserFinalizeProcessing() 
{
}
  
/**
 * \brief   Add a SEPARATOR to the end of the name if necessary
 * @param   pathname file/directory name to normalize 
 */
std::string FilesFromDirectory::NormalizePath(std::string const &pathname)
{
#ifdef _WIN32
   const char FILESEPARATOR = '\\';
#else
   const char FILESEPARATOR = '/';
#endif

   std::string name = pathname;
   int size = name.size();

   if ( name[size-1] != FILESEPARATOR )
   {
      name += FILESEPARATOR;
   }
   return name;
} 

/**
 * \brief   Explores a directory with possibility of recursion
 *          return number of files read
 * @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();
   int numberOfFiles = 0;
   std::string fileName;
   std::string dirName = NormalizePath(dirpath);
#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 )
         {
            numberOfFiles += Explore(dirName+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?)
	 }      
         Filenames.push_back(temp);
         numberOfFiles++;
      }
   }
   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);

      //gdcmErrorMacro("FindNextFile error. Error is " << (char *)lpMsgBuf
      //             <<" for the directory : "<<dirName);
      return -1;
   }

#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) );
      }
      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?)
             }
             Filenames.push_back(temp);	 
             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) );
   }
#endif

  std::string tmpString;
  int i,ii,sizeFilenames = Filenames.size();
  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

  return numberOfFiles;
}
*/

}
// EO namespace bbstd