#include "creaImageIOSystem.h"
#include "creaImageIOUltrasonixImageReader.h"
#include <creaVtk.h>
#include <boost/filesystem/path.hpp>
namespace creaImageIO
{
#define HEADER_SIZE	19
#define TYPE_RF		16
#define TYPE_B8		4
#define TYPE_B32	8


  //=====================================================================
  UltrasonixImageReader::UltrasonixImageReader()
  {
    SetName("Ultrasonix");
  }
  //=====================================================================
  
  //=====================================================================
  UltrasonixImageReader::~UltrasonixImageReader()
  {
  }
  //=====================================================================

  //=====================================================================
  struct Ultrasonix_header
  {
    // frames, width, height, ultrasounds frequency, sampling rate
    int type, frame, width, height, frequency, samplingRate;
  };
  //=====================================================================


  //=====================================================================
  bool ReadHeader( FILE *Ultrasonix_file, Ultrasonix_header& h )
  {
    //int *header=(int*)malloc(sizeof(int)*HEADER_SIZE);
    int header[HEADER_SIZE];
    fread(header, sizeof(int), HEADER_SIZE, Ultrasonix_file);
    if (ferror(Ultrasonix_file))
    	return false;
    h.type         = header[0];
    h.frame        = header[1];
    h.height       = header[2];
    h.width        = header[3];
    h.frequency    = header[14];
    h.samplingRate = header[15];
    //free(header);  
    return true;
  }
  //=====================================================================

  //=====================================================================
  bool UltrasonixImageReader::CanRead(const std::string& filename)
  { 
    long size;
    bool ok = false;
    FILE *Ultrasonix_file=fopen(filename.c_str(), "rb");
    if (Ultrasonix_file) 
    {
	Ultrasonix_header h;
	if (!ReadHeader(Ultrasonix_file, h) )
	{
		fclose(Ultrasonix_file);
        	std::cout << "cannot read Ultrasonix header for file [" << filename << "]" << std::endl;				
		return false;	
	}

	fseek(Ultrasonix_file,0,SEEK_END);		// go to end of file
	if (h.type == TYPE_RF)
		size = (ftell(Ultrasonix_file) - (HEADER_SIZE+h.frame) * sizeof(int)) / sizeof(short);
	else if (h.type == TYPE_B8)
		size = (ftell(Ultrasonix_file) - HEADER_SIZE * sizeof(int)) / sizeof(char);
	else if (h.type == TYPE_B32)
		size = (ftell(Ultrasonix_file) - HEADER_SIZE * sizeof(int)) / sizeof(int);

	// check if the data size corresponds to the dimensions of the images
	if (size == h.width * h.height * h.frame)
		ok = true;
		
	fclose(Ultrasonix_file);
    }
    return ok;
  }
  //=====================================================================
  
  //=====================================================================
  vtkImageData* UltrasonixImageReader::ReadImage(const std::string& filename)
  {
    FILE *Ultrasonix_file=fopen(filename.c_str(),"rb");
    if (!Ultrasonix_file) 
    {
    	std::cout << "cannot open file [" << filename << "]" << std::endl;
    	return 0;
    }
    Ultrasonix_header h;
    if (!ReadHeader(Ultrasonix_file,h)) 
    {
        std::cout << "cannot read Ultrasonix header for file [" << filename << "]" << std::endl;
	fclose(Ultrasonix_file);  
	return 0;
    }

    long frame_size = h.height   * h.width;      
    long im_size    = frame_size * h.frame;

    short *dataRF,  *ptrRF;
    char  *dataB8,  *ptrB8;
    int   *dataB32, *ptrB32;
    vtkImageData* im;
    int temp;

    switch (h.type)
    {
	case TYPE_RF:
		dataRF = (short*)malloc(sizeof(short)*im_size);
		ptrRF  = dataRF;

		for (int k=0; k<h.frame; k++)
		{
			int frame_number;
			fread(&frame_number, sizeof(int), 1, Ultrasonix_file);
			fread(ptrRF,sizeof(short), frame_size, Ultrasonix_file);
			ptrRF += frame_size;
		}
		fclose(Ultrasonix_file);  

		im = crea::NewVtkImageDataFromRaw( dataRF, h.width, h.height, h.frame);
	break;

	case TYPE_B8:
		dataB8 = (char*)malloc(sizeof(char)*im_size);
		ptrB8  = dataB8;
		for (int k=0; k<h.frame; k++)
		{
			fread(ptrB8,sizeof(char), frame_size, Ultrasonix_file);
			ptrB8 += frame_size;
		}
		// in mode b frames width and height are inverted
		temp     = h.width;
		h.width  = h.height;
		h.height = temp;

		fclose(Ultrasonix_file);  

	im = crea::NewVtkImageDataFromRaw( dataB8, h.width, h.height, h.frame);
	break;

	case TYPE_B32:
		dataB32 = (int*)malloc(sizeof(int)*im_size);
		ptrB32  = dataB32;
		for (int k=0; k<h.frame; k++)
		{
			fread(ptrB32, sizeof(int), frame_size, Ultrasonix_file);
			ptrB32 += frame_size;
		}
		// in B mode frames width and height are inverted
		temp     = h.width;
		h.width  = h.height;
		h.height = temp;

		fclose(Ultrasonix_file);  

		im = crea::NewVtkImageDataFromRaw( dataB32, h.width, h.height, h.frame);
	break;
    }

    return im;
}
  //=====================================================================
  

  //=====================================================================
  void UltrasonixImageReader::PushBackExtensions(std::vector<std::string>& v)
  {
    v.push_back("Ultrasonix");
  }
  //=====================================================================
 


  //=====================================================================
 void UltrasonixImageReader::ReadAttributes(const std::string& filename, 
				      std::map<std::string,std::string>& attr)
 {
    GimmickMessage(2,"Reading attributes from '" << filename << std::endl);

    FILE *Ultrasonix_file = fopen(filename.c_str(), "rb");
    if (!Ultrasonix_file)
    {
    	std::cout << "cannot open RF file [" << filename << "]" << std::endl;
    	return;
    }

    Ultrasonix_header h;
    if (!ReadHeader(Ultrasonix_file, h)) 
    {
        fclose(Ultrasonix_file);
        std::cout << "cannot read Ultrasonix Attributes for RF file [" << filename << "]" << std::endl;  
	return;
    }

    fclose(Ultrasonix_file);  
  
    // Columns
    char cols[128];
    sprintf(cols,"%i", h.width);
    // Rows
    char rows[128];
    sprintf(rows,"%i", h.height);
    // Planes 
    char planes[128];
    sprintf(planes,"%i", h.frame);
    // Sampling frequency
    char samplingFrequency[128];
    sprintf(samplingFrequency,"%i", h.samplingRate);
    // Transducer frequency
    char transducerFrequency[128];
    sprintf(transducerFrequency,"%i", h.frequency);
   
    // 
    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("D003a_001a")) != attr.end())
    {
       i->second = samplingFrequency;
    }
    if ( (i = attr.find("D0018_6030")) != attr.end())
    {
	   i->second = transducerFrequency;
    }
    
    GimmickMessage(2,"Attributes map:"<<std::endl<<attr<<std::endl);
    return;
}
  //=====================================================================

} // namespace creaImageIO