#include <creaImageIOVtkImageReader.h>
#include <vtkImageReader2.h>
#include <creaImageIOSystem.h>
#include "boost/filesystem/path.hpp"

namespace creaImageIO{
  //=====================================================================
  VtkImageReader::VtkImageReader(vtkImageReader2* r, 
				       const std::string& name,
				       const std::string& extensions)
    : mReader(r), mExtensions(extensions)
  {
    if (name.size() == 0) 
      {
		  const char *test =mReader->GetDescriptiveName();
		  if(test != "")
		  {
			SetName ( "toto");// mReader->GetDescriptiveName());
		  }
		
      }
    else 
      {
	SetName ( name );
      }
    GimmickDebugMessage(5,"Constructing vtkImageReader : "<<GetName()
			<<std::endl);
      
  }
  //=====================================================================
  
  //=====================================================================
  VtkImageReader::~VtkImageReader()
  {

    mReader->Delete();
  }
  //=====================================================================

  //=====================================================================
  bool VtkImageReader::CanRead(const std::string& filename)
  { 

    return (mReader->CanReadFile(filename.c_str())!=0);
/*	  if(filename != "")
	  {
		return (mReader->CanReadFile(filename.c_str())!=0);
	  }
	  else
	  {
		  return false;
	  }*/
  }
  //=====================================================================
  
  //=====================================================================
  vtkImageData* VtkImageReader::ReadImage(const std::string& filename)
  {
    vtkImageData* im = 0;
    try
      {
	mReader->SetFileName(filename.c_str());
	mReader->Update();
	im = vtkImageData::New();
	im->ShallowCopy(mReader->GetOutput());
      }
    catch (...)
      {
	if (im!=0) im->Delete();
	im = 0;
      }
    return im;
  }
  //=====================================================================
  
  //=====================================================================
  void SplitExtensionsString ( const std::string& str, 
			       const std::string& delimiters, 
			       std::vector<std::string>& tokens)
  {
    // Skip delimiters at beginning.
    std::string::size_type lastPos = str.find_first_not_of(delimiters, 0);
    // Find first delimiter.
    std::string::size_type pos     = str.find_first_of(delimiters, lastPos);
    
    while (std::string::npos != pos || std::string::npos != lastPos)
      {
	// Found a token, add it to the vector.
	// SPECIFIC : REMOVE INITIAL DOT (lastPos + 1)
	tokens.push_back(str.substr(lastPos+1, pos - lastPos));
	// Skip delimiters.  Note the "not_of"
	lastPos = str.find_first_not_of(delimiters, pos);
	// Find next delimiter
	pos = str.find_first_of(delimiters, lastPos);
      }
    
    }
  //=====================================================================
  
  //=====================================================================
  void VtkImageReader::PushBackExtensions(std::vector<std::string>& v)
  {
    std::string ext = mExtensions;
    if (ext.size()==0) ext = mReader->GetFileExtensions ();
    
    SplitExtensionsString(ext," ",v);
  }
  //=====================================================================
 


  //=====================================================================
  void VtkImageReader::ReadAttributes(const std::string& filename, 
				      std::map<std::string,std::string>& attr)
  {
    GimmickMessage(2,"Reading attributes from '"<<filename<<std::endl);
    // Get image dimensions
    // How to get the image info without loading it in vtk ?
    mReader->SetFileName(filename.c_str());
    mReader->Update(); //OpenFile();
    int ext[6];
    mReader->GetDataExtent(ext);
    // Columns
    char cols[128];
    sprintf(cols,"%i",ext[1]-ext[0]);
    // Rows
    char rows[128];
    sprintf(rows,"%i",ext[3]-ext[2]);
    // Planes 
    char planes[128];
    sprintf(planes,"%i",ext[5]-ext[4]);
   
	std::map<std::string,std::string>::iterator i;
    if ( (i = attr.find("FullFileName")) != attr.end())
      {
	i->second = filename;
      }
    if ( (i = attr.find("D0004_1500")) != attr.end())
      {
	boost::filesystem::path full_path(filename);
	std::string f = full_path.leaf();
	i->second = f;
      }
    if ( (i = attr.find("D0028_0010")) != attr.end())
      {
	i->second = rows;
      }
    if ( (i = attr.find("D0028_0011")) != attr.end())
      {
	i->second = cols;
      }
    
    if ( (i = attr.find("D0028_0012")) != attr.end())
      {
		i->second = planes;
      }
	  if ( (i = attr.find("FullFileDirectory")) != attr.end())
      {
    	 std::string::size_type last_pos = filename.find_last_of("//");
		 i->second = filename.substr(0, last_pos);
	  }

    GimmickMessage(2,"Attributes map:"<<std::endl<<attr<<std::endl);
  }
  //=====================================================================

} // namespace creaImageIO