#include <stdlib.h>
#include <string.h>
#include "minifits.h"

// #include <iostream>

//////////////////////////////////////////////////////////////////////
//     Classe  MiniFITSException
//////////////////////////////////////////////////////////////////////
/*!
   \class MiniFITSException
   \ingroup TAcq

   \brief Exception class for FITS I/O services provided by the MiniFITSFile class.
*/ 


/* --Methode-- */
MiniFITSException::MiniFITSException(const char * m)   throw()
{
  if (m!=NULL) { 
    strncpy(msg_, m, MFEX_MAXMSGLEN-1);
    msg_[MFEX_MAXMSGLEN-1] = '\0';
  }
  else msg_[0] = '\0';
}

/* --Methode-- */
MiniFITSException::MiniFITSException(const string& m)   throw()
{
  strncpy(msg_, m.c_str(), MFEX_MAXMSGLEN-1);
  msg_[MFEX_MAXMSGLEN-1] = '\0';
}

/* --Methode-- */
MiniFITSException::~MiniFITSException()  throw()
{
}

/* --Methode-- */
const char* MiniFITSException::what() const throw()
{
  return msg_;
}

/* --Methode-- */
string const MiniFITSException::Msg() const 
{
  return (string(msg_));
}

//////////////////////////////////////////////////////////////////////
//     Classe  MiniFITSFile
//////////////////////////////////////////////////////////////////////
/*!
   \class MiniFITSFile
   \ingroup TAcq

   \brief This class provides a subset of I/O services in FITS format (IMAGE_HDU only).
*/ 

#define MFITSHLEN 2880

/* --Methode-- */
MiniFITSFile::MiniFITSFile()
{
  Init();
}

/* --Methode-- */
MiniFITSFile::MiniFITSFile(string const & nom, MiniFITS_Mode rwm)
{
  Init();
  Open(nom, rwm);
}

/* --Methode-- */
MiniFITSFile::MiniFITSFile(const char* nom, MiniFITS_Mode rwm)
{
  Init();
  Open(nom, rwm);
}


/* --Methode-- */
MiniFITSFile::~MiniFITSFile()
{
  Close();
  delete[] header;
}

/* --Methode-- */
void MiniFITSFile::Init()
{
  fip = NULL;
  rwmode = MF_Read;
  dtype = MF_Byte;
  nax1 = 1;
  nax2 = 1; 
  nax3 = 1; 
  fgnax3 = false;
  totwsz = 0;
  header = new char[MFITSHLEN];
  for(int i=0; i<MFITSHLEN; i++)  header[i]=' ';
  nkeya_ = 0;
}

/* --Methode-- */
void MiniFITSFile::Open(const char* nom, MiniFITS_Mode rwm)
{
  if (fip != NULL) throw MiniFITSException("MiniFITSFile::Open() - fip != NULL");
  if (rwm == MF_Write) {
    for(int i=0; i<MFITSHLEN; i++)  header[i]=' ';
    nkeya_ = 0;
    FillHeader();
    fip = fopen(nom, "w");
    if (fip == NULL) 
      throw MiniFITSException("MiniFITSFile::Open()/ failed fopen() for write");
    fwrite(header, 1, MFITSHLEN, fip);
    rwmode = MF_Write;
  }
  else {
    fip = fopen(nom, "r");
    if (fip == NULL) 
      throw MiniFITSException("MiniFITSFile::Open()/ failed fopen() for read");
    fread(header, 1, MFITSHLEN, fip);
    DecodeHeader();
    rwmode = MF_Read;
  }
  return;
}

/* --Methode-- */
void MiniFITSFile::Close()
{
  if (fip) { 
    if (rwmode == MF_Write)  {  
      // on remplit avec des zeros pour avoir une longueur multiple de 2880	
      size_t padsz = MFITSHLEN-(totwsz%MFITSHLEN);
      char zeros[160];
      for(size_t k=0; k<160; k++) zeros[k]=0;
      while(padsz>160) {
	fwrite(zeros, 1, 160, fip);
	padsz-=160;
      }
      if (padsz>0)  fwrite(zeros, 1, padsz, fip);
      // On reecrit l'entete 
      FillHeader();
      fseek(fip, 0, SEEK_SET);
      fwrite(header, 1, MFITSHLEN, fip);
    }	
    fclose(fip);
  }
  fip = NULL;
  return;
}

/* --Methode-- */
void MiniFITSFile::setDTypeNaxis(MiniFITS_DT dt, size_t na1, size_t na2)
{
  // Interdit si fichier ouvert en lecture ...
  if ((fip!=NULL)&&(rwmode == MF_Read))  
     throw MiniFITSException("MiniFITSFile::setDTypeNaxis()/Error ReadOnly file");

  dtype = dt;
  nax1 = na1; 
  nax2 = na2;
  nax3 = 1;
  fgnax3 = false;
}

/* --Methode-- */
void MiniFITSFile::setDTypeNaxis(MiniFITS_DT dt, size_t na1, size_t na2, size_t na3)
{
  // Interdit si fichier ouvert en lecture ...
  if ((fip!=NULL)&&(rwmode == MF_Read))  
     throw MiniFITSException("MiniFITSFile::setDTypeNaxis()/Error ReadOnly file");

  dtype = dt;
  nax1 = na1; 
  nax2 = na2;
  nax3 = na3;
  fgnax3 = true;
}


/* --Methode-- */
string MiniFITSFile::DataTypeToString()
{
  if (dtype == MF_Byte) return "MF_Byte";
  else if (dtype == MF_Int16) return "MF_Int16";
  else if (dtype == MF_Float32) return "MF_Float32";
  else return "Unknown??";
}

/* --Methode-- */
int MiniFITSFile::Write(void* data, size_t sz)
{
  fwrite(data, 1, sz, fip);
  totwsz += sz;
  return 0;
}

/* --Methode-- */
int MiniFITSFile::Read(void* data, size_t sz, size_t offset)
{
  fseek(fip, offset+MFITSHLEN, SEEK_SET);
  fread(data, 1, sz, fip);
  return 0;
}

/* --Methode-- */
void MiniFITSFile::FillHeader()
{
  strcpy(header, "SIMPLE  =                    T / file does conform to FITS standard");
  header[strlen(header)] = ' ';
  int bpix = 8;
  if (dtype == MF_Byte) bpix = 8;
  else if (dtype == MF_Int16) bpix = 16;
  else if (dtype == MF_Float32) bpix = -32;
  char * buff = header+80;
  sprintf(buff, "BITPIX  = %20d / number of bits per data pixel", bpix);
  buff[strlen(buff)] = ' ';
  buff = header+160;
  if (fgnax3) 
    strcpy(buff, "NAXIS   =                    3 / number of data axes");
  else 
    strcpy(buff, "NAXIS   =                    2 / number of data axes");
  buff[strlen(buff)] = ' ';
  buff = header+240;
  sprintf(buff, "NAXIS1  = %20ld / nb of pixels along X = PaquetSize", (long)nax1);
  buff[strlen(buff)] = ' ';
  buff = header+320;
  if (fgnax3) 
    sprintf(buff, "NAXIS2  = %20ld / Number of fibers ", (long)nax2);
  else 
    sprintf(buff, "NAXIS2  = %20ld / NumberOfPaquets", (long)nax2);
  buff[strlen(buff)] = ' ';
  buff = header+400;
  if (fgnax3) 
    sprintf(buff, "NAXIS3  = %20ld / Number of paquets ", (long)nax3);
  else 
    strcpy(buff,"COMMENT  BAO-Radio / MiniFITSFile ");
  buff[strlen(buff)] = ' ';
  
  buff = header+480+nkeya_*80;
  strcpy(buff,"COMMENT  BAO-Radio / MiniFITSFile ");
  buff[strlen(buff)] = ' ';
  buff = header+560+nkeya_*80;
  strcpy(buff,"END");
  buff[strlen(buff)] = ' ';

  return;
}

/* --Methode-- */
int MiniFITSFile::AddKeyI(const char* key, long val, const char* comm)
{
  if (nkeya_ >= 28) return 0;
  char cle[10];
  strncpy(cle,key,8);
  cle[8]='=';
  for(int i=0;i<8;i++) 
    if (cle[i]=='\0') cle[i]=' ';
  cle[9]='\0';
  char* buff=header+480+nkeya_*80;
  if (comm!=NULL) {
    char tcom[50];
    strncpy(tcom,comm,48);  
    tcom[48]='\0';
    sprintf(buff,"%s %20ld / %s", cle, val, tcom);
  }
  else  sprintf(buff,"%s %20ld / ", cle, val);
  buff[strlen(buff)]=' ';
  nkeya_++;
  return nkeya_;
}

/* --Methode-- */
int MiniFITSFile::AddKeyD(const char* key, double val, const char* comm)
{
  if (nkeya_ >= 28) return 0;
  char cle[10];
  strncpy(cle,key,8);
  cle[8]='=';
  for(int i=0;i<8;i++) 
    if (cle[i]=='\0') cle[i]=' ';
  cle[9]='\0';
  char* buff=header+480+nkeya_*80;
  if (comm!=NULL) {
    char tcom[50];
    strncpy(tcom,comm,48);  
    tcom[48]='\0';
    sprintf(buff,"%s %20lg / %s", cle, val, tcom);
  }
  else  sprintf(buff,"%s %20lg / ", cle, val);
  buff[strlen(buff)] = ' ';
  nkeya_++;
  return nkeya_;
}

/* --Methode-- */
int MiniFITSFile::AddKeyS(const char* key, const char* val, const char* comm)
{
  if (nkeya_ >= 28) return 0;
  char cle[10];
  strncpy(cle,key,8);
  cle[8]='=';
  for(int i=0;i<8;i++) 
    if (cle[i]=='\0') cle[i]=' ';
  cle[9]='\0';
  char tcom[72];
  tcom[0]='\'';
  strncpy(tcom+1,val,65);
  int l=strlen(tcom);
  strcpy(tcom+l,"' / ");
  l+=4;
  if ((l<70)&&(comm!=NULL)) strncpy(tcom+l,comm,70-l);
  tcom[70]='\0';
  char* buff=header+480+nkeya_*80;
  sprintf(buff,"%s %s", cle, tcom);
  buff[strlen(buff)] = ' ';
  nkeya_++;
  return nkeya_;
}

/* --Methode-- */
string MiniFITSFile::GetKey(const char* key)
{
  char rs[80];
  rs[0]='\0';
  char cle[10];
  strncpy(cle,key,8);
  cle[8]='=';
  for(int i=0;i<8;i++) 
    if (cle[i]=='\0') cle[i]=' ';
  cle[9]='\0';
  for(int kh=80; kh<2800; kh+=80) {
    char * buff = header+kh;
    if (strncmp(buff, cle, 9) == 0)  {
      strncpy(rs, buff, 79);
      rs[79]='\0';
      break;
    }
  }
  return rs;
}

/* --Methode-- */
string MiniFITSFile::GetKeyValue(const char* key)
{
  string line = GetKey(key);
  if(line.size()==0) return "";
  char rs[84], sep = '?';
  bool foundeq = false;
  unsigned int ip = 0;
  for(unsigned int i=0;i<line.size();i++) {
    if(!foundeq) {if(line[i] == '=') foundeq = true; continue;}
    if(sep=='?') {  // recherche du separateur de debut
      if(line[i]==' ') continue;
      if(line[i]=='\'' || line[i]=='\"') {sep = line[i]; continue;}
      else sep = ' ';
    }
    if(line[i]==sep) break;
    rs[ip] = line[i];
    ip++;
  }
  rs[ip] = '\0';
  return rs;
}

/* --Methode-- */
void MiniFITSFile::DecodeHeader()
{
  // AMELIORER le decodage de l'entete, remplissage dtype, nax1, nax2
  char * buff = header;
  if (strncmp(buff, "SIMPLE  =", 9) != 0) 
     throw MiniFITSException("MiniFITSFile::DecodeHeader()/Error - NO SIMPLE keyword");
  bool fgokt=false;
  bool fgok1=false;
  bool fgok2=false;
  bool fgok3=false;
  for(int kh=80; kh<2800; kh+=80) {
    buff = header+kh;
    if (strncmp(buff, "NAXIS1  =", 9) == 0)  {
      nax1 = atol(buff+10);
      fgok1 = true;
      //  	  cout << " FOUND : NAXIS1= " << nax1 << endl;
    } 
    else if (strncmp(buff, "NAXIS2  =", 9) == 0)  {
      nax2 = atol(buff+10);
      fgok2 = true;
      //  	  cout << " FOUND : NAXIS2= " << nax2 << endl;  	  
    } 
    else if (strncmp(buff, "NAXIS3  =", 9) == 0)  {
      nax3 = atol(buff+10);
      fgok3 = true;
      //  	  cout << " FOUND : NAXIS2= " << nax2 << endl;  	  
    } 
    else if (strncmp(buff, "BITPIX  =", 9) == 0)  {
      int bpix = atoi(buff+10);
      fgokt = true;
      if (bpix == 8)  dtype = MF_Byte;
      else if (bpix == 16)  dtype = MF_Int16;
      else if (bpix == -32)  dtype = MF_Float32;
      else fgokt = false;
      //  	  cout << " FOUND : bpix= " << bpix << endl;  	  	  
    } 
  }
  if (!(fgok1&&fgok2&&fgokt))  
    throw MiniFITSException("MiniFITSFile::DecodeHeader()/Error- Missing/wrong NAXIS1/2,BITPIX");
  if (fgok3)  fgnax3=true;
  return;
}