#include <string.h>

#include "brpaqu.h"


static inline void bswap4(void* p)
{
  UInt32 tmp = *(UInt32*)p;
  *(UInt32*)p = ((tmp >> 24) & 0x000000FF) |
                ((tmp >> 8)  & 0x0000FF00) |
                ((tmp & 0x0000FF00) << 8)  |
                ((tmp & 0x000000FF) << 24);
}

/* --Methode__ */
BRPaquet::BRPaquet(Byte* src, Byte* dst, int paqsz, BRDataFmtConv fgswap,int binMin,int nbBin )
  // swapall = true -> on swap tout le paquet, sinon swap entete seulement
{
  dst_ = dst;
  sz_ = paqsz;
  if ((src == NULL) || (dst == NULL)) return;
  // Il faut mettre une protection (throw) si dst==NULL ou sz==0

  UInt32* src32 = (UInt32*)src;
  UInt32* dst32 = (UInt32*)dst;
  Byte* tmp = new Byte[sz_];
 
  switch ( fgswap ) {
  case BR_DoNothing :  // rien a faire
    break;
  case BR_Copy : // copie directe 
    memcpy(dst_, src, sz_); 
    break;
  case BR_Copy_Reduc : // copie directe 
   memcpy(dst_, src, HeaderSize()); 
   
    memcpy(dst_+ HeaderSize(), src+(binMin+ HeaderSize()), nbBin); 
   memcpy(dst_+ (HeaderSize()+ nbBin ), src+(binMin+ HeaderSize()+sz_/2), nbBin); 
   memcpy(dst_+( HeaderSize()+( 2* nbBin)),src+(HeaderSize()+DataSize()),TrailerSize());
 
    SetPaqLen( nbBin/2);
   break;
   
  case BR_Swap32 : // On swappe toutes les donnees du paquet 
  //  les bytes sont dans l'ordre par paquet de 4 octets (Int32) , les deux Int32 de 
  // On copie la zone donnees en faisant un byte-swap correspondant a 8 octets (4->8 Reza/firmware SGDMA) 
    for(int ka=0; ka<sz_/4; ka+=2) {
      dst32[ka] = src32[ka+1];
      dst32[ka+1] = src32[ka];
    }   
    break;
  case BR_SwapAll:
  for(int ka=0; ka<sz_; ka+=8) {
      for(int kb=0; kb<4; kb++) {
        dst_[ka+kb] = src[ka+3-kb+4];    
        dst_[ka+kb+4] = src[ka+3-kb];    
      }
    }
  for(int ka=HeaderSize()+DataSize(); ka<sz_; ka+=8) {
      for(int kb=0; kb<4; kb++) {
	dst_[ka+kb] = src[ka+3-kb+4];    
        dst_[ka+kb+4] = src[ka+3-kb];    
      }
    }
    break;
  case BR_SwapHDR :
  case BR_FFTOneChan :
  case BR_FFTTwoChan :
    // Byte swap (sur 8 octets) de l'entete
    // ByteSwap 8 (4->8 Reza/firmware SGDMA) de l'enete 
    for(int ka=0; ka<BRHDRSIZE; ka+=8) {
      for(int kb=0; kb<4; kb++) {
	dst_[ka+kb] = src[ka+3-kb+4];    
        dst_[ka+kb+4] = src[ka+3-kb];    
      }
    }

    // on reoordonne les coeff FFT et on swappe en mem temps  (Byte-swap sur 8 octets)
    if (fgswap == BR_FFTOneChan) ReorderFFTData((SByte*)(src+HeaderSize()), (SByte*)(dst_+HeaderSize()), DataSize());
    else if (fgswap == BR_FFTTwoChan) {
      ReorderFFTData((SByte*)(src+HeaderSize()), (SByte*)(dst_+HeaderSize()), DataSize()/2);
      ReorderFFTData((SByte*)(src+HeaderSize()+DataSize()/2), (SByte*)(dst_+HeaderSize()+DataSize()/2), DataSize()/2);
    }
    // Byte swap (sur 8 octets) du trailer 
    for(int ka=HeaderSize()+DataSize(); ka<sz_; ka+=8) {
      for(int kb=0; kb<4; kb++) {
	dst_[ka+kb] = src[ka+3-kb+4];    
        dst_[ka+kb+4] = src[ka+3-kb];    
      }
    }
    break;
 
  case BR_FFTOneChan32 :
  case BR_FFTTwoChan32 :
    // swap du header uniquement : Echange de deux mots de 4 octets
    for(int ka=0; ka<BRHDRSIZE/4; ka+=2) {
      dst32[ka] = src32[ka+1];
      dst32[ka+1] = src32[ka];
    }
  
    // on reoordonne et on swappe en mem temps (Echange de deux mots de 4 octets)
    if (fgswap == BR_FFTOneChan32) ReorderFFTData32((SByte*)(src+HeaderSize()), (SByte*)(dst_+HeaderSize()), DataSize());
    else if (fgswap == BR_FFTTwoChan32) {
      ReorderFFTData32((SByte*)(src+HeaderSize()), (SByte*)(dst_+HeaderSize()), DataSize()/2);
      ReorderFFTData32((SByte*)(src+HeaderSize()+DataSize()/2), (SByte*)(dst_+HeaderSize()+DataSize()/2), DataSize()/2);
    }
    // swap du trailler uniquement : Echange de deux mots de 4 octets
    for(int ka=(HeaderSize()+DataSize())/4;ka < sz_/4; ka+=2) {
      dst32[ka] = src32[ka+1];
      dst32[ka+1] = src32[ka];
    }
    break;
  case BR_FFTOneChanNoSwap :
  case BR_FFTTwoChanNoSwap :
    // copie du header uniquement 
    for(int ka=0; ka<BRHDRSIZE/4; ka++)  dst32[ka] = src32[ka];

    //DEL  on a plus de swapdonc il faut copier dans dst
    //DEL    memcpy(dst_, src, sz_); 

    // on reoordonne les coeff FFT (pas de swap) 
    if (fgswap == BR_FFTOneChanNoSwap) ReorderFFTDataNoSwap((SByte*)(src+HeaderSize()), (SByte*)(dst_+HeaderSize()), DataSize());
    else if (fgswap == BR_FFTTwoChanNoSwap) {
      ReorderFFTDataNoSwap((SByte*)(src+HeaderSize()), (SByte*)(dst_+HeaderSize()), DataSize()/2);
      ReorderFFTDataNoSwap((SByte*)(src+HeaderSize()+DataSize()/2), (SByte*)(dst_+HeaderSize()+DataSize()/2), DataSize()/2);
    }
    // copie du trailler uniquement : 
     for(int ka=(HeaderSize()+DataSize())/4;ka < sz_/4; ka++)  dst32[ka] = src32[ka];
     break;
  case BR_CpHdrTrl :
    for(int ka=0; ka<BRHDRSIZE/4; ka++)  dst32[ka] = src32[ka];
    for(int ka=(HeaderSize()+DataSize())/4;ka < sz_/4; ka++)  dst32[ka] = src32[ka];
    break;

 case BR_FFTOneChanNoSwapReduct :
  case BR_FFTTwoChanNoSwapReduct :
    // copie du header uniquement 
   
    for(int ka=0; ka<BRHDRSIZE/4; ka++)  dst32[ka] = src32[ka];

    //DEL  on a plus de swapdonc il faut copier dans dst
    //DEL    memcpy(dst_, src, sz_); 

    // on reoordonne les coeff FFT (pas de swap) 
    if (fgswap == BR_FFTOneChanNoSwap) ReorderFFTDataNoSwap((SByte*)(src+HeaderSize()), (SByte*)(tmp+HeaderSize()), DataSize());
    else if (fgswap == BR_FFTTwoChanNoSwap) {
      ReorderFFTDataNoSwap((SByte*)(src+HeaderSize()), (SByte*)(tmp+HeaderSize()), DataSize()/2);
      ReorderFFTDataNoSwap((SByte*)(src+HeaderSize()+DataSize()/2), (SByte*)(tmp+HeaderSize()+DataSize()/2), DataSize()/2);
    }
    memcpy(tmp+ HeaderSize(), src+(binMin+ HeaderSize()), nbBin); 
   memcpy(tmp+ (HeaderSize()+ nbBin ), src+(binMin+ HeaderSize()+sz_/2), nbBin); 
   memcpy(tmp+( HeaderSize()+( 2* nbBin)),src+(HeaderSize()+DataSize()),TrailerSize());
   memcpy(dst_+ HeaderSize(),tmp+ HeaderSize(),( 2* nbBin)+TrailerSize());
    SetPaqLen( nbBin/2);

    // copie du trailler uniquement : 
    //  for(int ka=(HeaderSize()+DataSize())/4;ka < sz_/4; ka++)  dst32[ka] = src32[ka];
     break;
  }  // Fin switch 

}

/* --Methode__ */
BRPaquet::BRPaquet(Byte* srcdst, int paqsz)
{
  dst_ = srcdst;
  sz_ = paqsz;
  // Il faut mettre une protection (throw) si srcdst==NULL ou sz==0
}


/* --Methode__ */
UInt16 BRPaquet::ChannelID()
{

  UInt16 ChnId=ChanId();
  UInt16 ChpId=ChipId();
 
     
  if (ChpId == 2) 
    {
      if (ChnId == 1) ChnId = Ch3;
      if (ChnId == 2) ChnId = Ch4;
      if (ChnId == 3) ChnId = Ch3_4;
    }
  return(ChnId);
}

/* --Methode__ */
UInt16 BRPaquet::ModeAcquisition()
{
  UInt16 ModAq;
//DEL  printf("Mod Acq %x \n",ModeAcq());
  ModAq = ((ModeAcq() & 0x30)>> 4);
  return(ModAq);
}

/* --Methode__ */
void BRPaquet::SetHDRMarker64(UInt64 htag)
{
  *((UInt64*)(dst_+OFFSET)) = htag;
  return;  	
}

/* --Methode__ */
void BRPaquet::SetTRLMarker64(UInt64 ttag)
{
  *((UInt64*)(dst_+(sz_-BRTRLSIZE+OFFSET+1))) = 0;	
  *((UInt64*)(dst_+(sz_-BRTRLSIZE+OFFSET))) = ttag; 
  return;         	          	   	
}

/* --Methode__ */
void BRPaquet::SetFrameCounter(UInt32 fc)
{
  UInt32* wp =  (UInt32*)(dst_+(BRFRCPTOFF+OFFSET));
  *wp = (*wp & 0x0000FFFF) | ((fc<<16) & 0xFFFF0000); 
  return;         	          	
}

/* --Methode__ */
void BRPaquet::SetTimeTag(UInt64 timtag)
{
  UInt32* wp =  (UInt32*)(dst_+(BRFRCPTOFF+OFFSET));
  UInt32 fc = *wp;
  *((UInt64*)(dst_+(BRTMTAGOFF+OFFSET))) = timtag;
  *wp = (*wp & 0x0000FFFF) | (fc & 0xFFFF0000);
  return;         	
}

/* --Methode__ */
ostream& BRPaquet::Print(ostream & os, int nelt, bool prht)
{
  os << endl << "BRPaquet::Print() PaqSz=" << PaquetSize() << " DataSize=" << DataSize() 
     << " dst_pointer=(hex)" << hex << (unsigned long)dst_ << dec << endl; 
  if (dst_ == NULL) {
    os << " ...NULL paquet " << endl;
    return os; 
  }
  os << " BR AcqMode: " << ModeAcquisition() << " Channel: " << ChannelID()  
     << " FrameCounter=" << FrameCounter() << " FrameDataLen=" << PaqLen() << endl;
  if (TrailerSize() > 0) 
    os << " ...HDRMarker(hex)=" << hex <<  HDRMarker() << " TRLMarker=" << TRLMarker() << dec << endl;
  else 
    os << " ...HDRMarker(hex)=" << hex <<  HDRMarker() << " NO TRLMarker=" << dec << endl;
  UInt32 tt1, tt2; 
  tt2 = TimeTag1(); 
  tt1 = TimeTag2();
  os << " ...TimeTag (hex)=" << hex << " TT1= " << tt1 << " TT2=" << tt2 
     << dec << " ->TimeTag()=" << TimeTag() << endl;
  // os << " ...Position Chariot (hex)= " << hex << PositionChariot() << endl;
  if (nelt > DataSize()/2) nelt = DataSize()/2;
  os << " ...Data[1.." << nelt << "]= ";

  for(int k=0; k<nelt; k++) os << (int)(*(Data1()+k)) << " , ";
  os << endl;
  os << " ...Data[" << DataSize()-nelt << ".." << DataSize()-1 << "]= ";
  for(int k=DataSize()-nelt; k<DataSize(); k++) os << (int)(*(Data1()+k)) << " , ";
  os << endl;
  if (prht) {   // Impression header / trailer
    UInt32* hdr = (UInt32*)Header();
    os << " ...Header (hex):" << hex ;
    for(int k=0; k<HeaderSize()/sizeof(UInt32); k++) 
      os << hdr[k] << " , " ;
    os << dec << endl;
    if (TrailerSize() > 0) {
      UInt32* trl = (UInt32*)Trailer();
      os << " ...Trailer (hex):" << hex ;
      for(int k=0; k<TrailerSize()/sizeof(UInt32); k++) 
        os << trl[k] << " , " ;
      os << dec << endl;
    }
  }
  return os;
}


// ---------------------------------------------------------
//  **** REMARQUE   N/2+1 complexes -> N/2 complexes *****
// N = Nb d'echantillon en temps -> N/2 paires (real, imag)
// Il y a le continu, et N/2 frequences ---> N/2+1 nombres complexes,
// mais avec la contrainte Z(0).imag = 0  Z(N/2).imag = 0 
// f(i) i=0...N-1   ===> Z(k) ( Z complexe , k=0...N/2 )
// mais avec la contrainte Z(0).imag = 0  Z(N/2).imag = 0 
// On peut donc tout mettre ds N/2 complexes en choisissant 
// de mettre ds Z(0).imag  Z(N/2).real 
// ---------------------------------------------------------- 

// Fonction magique qui donne le pointeur permettant de tenir compte du byte-swp sur 8 octets
static inline int IndexByteSwap8(int idx) 
{
  return ( (idx-(idx%8))+(7-idx%8) ) ;
}

/* --Methode__ */
void BRPaquet::ReorderFFTData(SByte* src, SByte* dst, int N)
{
  // Code recopie depuis /Dev/DisplayData/HistoWindow.cc 
  // fonction TraceWind::DisplayBaoDatasFFT() et adapte aux structures BRPaquet et Cie 
  // Modif par rapport au code de Bruno : N/2 elements complexes au lieu de N/2+1 - Remarque ci-dessus
 
  int nCoef = N / 2; // to change
  int debutIndex = N / 4 + 1;
  int fifoSize =  N / 4 - 1;
  int i; 

  TwoByteComplex* dstcmplx = (TwoByteComplex*)dst;
  
  //   cout << " Display BAO Datas FFT (" << N << ")" << " : from 0 to "<< nCoef << endl;
  //   cout << " Variables : debutIndex, fifoSize " << debutIndex << ", " << fifoSize << endl;

  
  // Sortie 1
  for (i = 0; i < fifoSize ; i++)
    {
      dstcmplx[debutIndex + i].realB() = src[IndexByteSwap8(2*i)];
      dstcmplx[debutIndex + i].imagB() = src[IndexByteSwap8(2*i + 1)];
    }

  // element au milieu
  dstcmplx[N / 4].realB() = src[IndexByteSwap8(2*fifoSize)];
  dstcmplx[N / 4].imagB() = src[IndexByteSwap8(2*fifoSize + 1)];

  // Sortie 2
  for (i = 0; i < fifoSize ; i++)
    {
      dstcmplx[fifoSize - i].realB() = src[IndexByteSwap8(nCoef + 2*i)];
      dstcmplx[fifoSize - i].imagB() = src[IndexByteSwap8(nCoef + 2*i + 1)];
    }

  // k = 0 et k = N/2  
  dstcmplx[0].realB() = src[IndexByteSwap8(N - 2)];
  // Voir Remarque ci-dessus Z(N/2).real -> Z(0).image
  dstcmplx[0].imagB() = src[IndexByteSwap8(N - 1)];  // Attention, on met ici la real(fmax)

  return ; 
}

static inline int IndexByteSwap8_32(int idx) 
{
  return ( (idx-(idx%8))+((4+idx%8)%8) ) ;
}

void BRPaquet::ReorderFFTData32(SByte* src, SByte* dst, int N)
{
  // Code recopie depuis /Dev/DisplayData/HistoWindow.cc 
  // fonction TraceWind::DisplayBaoDatasFFT() et adapte aux structures BRPaquet et Cie 
  // Modif par rapport au code de Bruno : N/2 elements complexes au lieu de N/2+1 - Remarque ci-dessus

  int nCoef = N / 2; // to change
  int debutIndex = N / 4 + 1;
  int fifoSize =  N / 4 - 1;
  int i; 

  TwoByteComplex* dstcmplx = (TwoByteComplex*)dst;
  
  //   cout << " Display BAO Datas FFT (" << N << ")" << " : from 0 to "<< nCoef << endl;
  //   cout << " Variables : debutIndex, fifoSize " << debutIndex << ", " << fifoSize << endl;

  
  // Sortie 1
  for (i = 0; i < fifoSize ; i++)
    {
      dstcmplx[debutIndex + i].realB() = src[IndexByteSwap8_32(2*i)];
      dstcmplx[debutIndex + i].imagB() = src[IndexByteSwap8_32(2*i + 1)];
    }

  // element au milieu
  dstcmplx[N / 4].realB() = src[IndexByteSwap8_32(2*fifoSize)];
  dstcmplx[N / 4].imagB() = src[IndexByteSwap8_32(2*fifoSize + 1)];

  // Sortie 2
  for (i = 0; i < fifoSize ; i++)
    {
      dstcmplx[fifoSize - i].realB() = src[IndexByteSwap8_32(nCoef + 2*i)];
      dstcmplx[fifoSize - i].imagB() = src[IndexByteSwap8_32(nCoef + 2*i + 1)];
    }

  // k = 0 et k = N/2  
  dstcmplx[0].realB() = src[IndexByteSwap8_32(N - 2)];
  // Voir Remarque ci-dessus Z(N/2).real -> Z(0).image
  dstcmplx[0].imagB() = src[IndexByteSwap8_32(N - 1)];  // Attention, on met ici la real(fmax)

  return ; 
}
void BRPaquet::ReorderFFTDataNoSwap(SByte* src, SByte* dst, int N)
{
  // Code recopie depuis /Dev/DisplayData/HistoWindow.cc 
  // fonction TraceWind::DisplayBaoDatasFFT() et adapte aux structures BRPaquet et Cie 
  // Modif par rapport au code de Bruno : N/2 elements complexes au lieu de N/2+1 - Remarque ci-dessus
 
  int nCoef = N / 2; // to change
  int debutIndex = N / 4 + 1;
  int fifoSize =  N / 4 - 1;
  int i; 

  TwoByteComplex* dstcmplx = (TwoByteComplex*)dst;
  
  //   cout << " Display BAO Datas FFT (" << N << ")" << " : from 0 to "<< nCoef << endl;
  //   cout << " Variables : debutIndex, fifoSize " << debutIndex << ", " << fifoSize << endl;

  
  // Sortie 1
  for (i = 0; i < fifoSize ; i++)
    {
      dstcmplx[debutIndex + i].realB() = src[(2*i)];
      dstcmplx[debutIndex + i].imagB() = src[(2*i + 1)];
    }

  // element au milieu
  dstcmplx[N / 4].realB() = src[(2*fifoSize)];
  dstcmplx[N / 4].imagB() = src[(2*fifoSize + 1)];

  // Sortie 2
  for (i = 0; i < fifoSize ; i++)
    {
      dstcmplx[fifoSize - i].realB() = src[(nCoef + 2*i)];
      dstcmplx[fifoSize - i].imagB() = src[(nCoef + 2*i + 1)];
    }

  // k = 0 et k = N/2  
  dstcmplx[0].realB() = src[(N - 2)];
  // Voir Remarque ci-dessus Z(N/2).real -> Z(0).image
  dstcmplx[0].imagB() = src[(N - 1)];  // Attention, on met ici la real(fmax)

  return ; 
}

/* --Methode__ */
const char* BRPaquet::FmtConvToString(BRDataFmtConv fgswap)
{
  const char * rs="";
  switch ( fgswap ) {
  case BR_DoNothing :  
    rs = "BR_DoNothing";
    break;
  case BR_Copy : 
    rs = "BR_Copy";
    break;
  case BR_SwapAll : 
    rs = "BR_SwapAll";
    break;
  case BR_SwapHDR :
    rs = "BR_SwapHDR";
    break;
  case BR_FFTOneChan :
    rs = "BR_FFTOneChan";
    break;
  case BR_FFTTwoChan :
    rs = "BR_FFTTwoChan";
    break;
  case BR_Swap32 :
    rs = "BR_Swap32";
    break;
  case BR_FFTOneChan32 :
    rs = "BR_FFTOneChan32";
    break;
  case BR_FFTTwoChan32 :
    rs = "BR_FFTTwoChan32";
    break;
  case BR_FFTOneChanNoSwap :
    rs = "BR_FFTOneChanNoSwap";
    break;
  case BR_FFTTwoChanNoSwap :
    rs = "BR_FFTTwoChanNoSwap";
    break;
  default:
    rs = "?????";
    break;
  }  // Fin switch 
  return rs;
}

// --------------------------------------------------------------------------
// Classe pour effectuer des verifications d'integrite sur les paquets/frames
// --------------------------------------------------------------------------

BRPaqChecker::BRPaqChecker(bool cktrl, int maxprt)
{
  cktrl_ = cktrl;	
  totnframes = 0;
  nframeok = 0;
  lostframes = 0;
  frclst = 0;
  lastframenum = 0;
  cnt_saut = 0;
  maxprt_ = maxprt;
  DefineHDRTag();
  DefineTRLTag();
}

BRPaqChecker::~BRPaqChecker()
{
}

UInt64 BRPaqChecker::DefineHDRTag(UInt32 hdr1, UInt32 hdr2)
{
  hdrtag_ = (UInt64)hdr1 + ((UInt64)hdr2 << 32);
  return hdrtag_;
}

UInt64 BRPaqChecker::DefineTRLTag(UInt32 trl1, UInt32 trl2)
{
  trltag_ = (UInt64)trl1 + ((UInt64)trl2 << 32);
  return trltag_;
}


bool BRPaqChecker::Check(BRPaquet& paq, UInt64& numframe)
{
  totnframes++; 
  numframe=0;
  if (paq.HDRMarker64() != HDRTag()) return false;
  if (cktrl_&&(paq.TRLMarker64() != TRLTag())) return false;
  /*  DBG 
  if (paq.TRLMarker64() != TRLTag()) { 
    cnt_pb++; 
    if (cnt_pb < 5) paq.Print();
    return false;
  }
  */
  unsigned int curfc = paq.FrameCounter();
  unsigned int delfc = 0;
  if (nframeok > 0) {
    if (curfc>frclst)  delfc = (curfc-frclst);
    else delfc = (65536-frclst+curfc);
    lostframes += (unsigned long long)delfc - 1;
    if (delfc != 1) { 
      cnt_saut++; 
      if (cnt_saut < maxprt_) { 
        cout << "BRPaqChecker::Check([NumFrameOK=" << nframeok 
             << ")/Debug  FrameCounter Cur=" << curfc 
             << " Last=" << frclst << " -> delta=" << delfc << endl;  
        paq.Print();
      }
    }
  }
  nframeok++; frclst=curfc; 
  lastframenum+=(unsigned long long)delfc;  // numero de frame sans repliement a 65535 
  numframe=lastframenum;
//DBG  if (cnt_pb<5) { cnt_pb++; paq.Print(); }
  return true;
}

ostream& BRPaqChecker::Print(ostream& os) const
{
  //  os << "BRPaqChecker:  HDRTag=" << hex << HDRTag() << " TRLTag=" << TRLTag() << dec << "\n"
  //     << " ... Tot.Nb.Frames.Proc=" << totnframes << " NbFrame HDR/TRL OK=" << nframeok 
  os << "BRPaqChecker: Tot.Nb.Frames.Proc=" << totnframes << " Nb.HDR/TRL OK=" << nframeok;
  if (cktrl_) os << " (Check Header AND Trailer)" << endl;
  else os << " (Header Check only)" << endl;
  float meangap = (cnt_saut>0)?((float)lostframes/(float)cnt_saut):0.;
  os   << " ... LostFrames=" << lostframes 
       << " LossRate=" << (double)lostframes*100./(double)totnframes << " %" 
       << " NbGaps=" << cnt_saut << " MeanGap=" << meangap << endl;
  return os;
}