////////////////////////////////////
// MicroBall Event Unpacker
// for the Asy-Eos experiment
// Apr 29, 2010 
// revison 1/2011
// zibi
////////////////////////////////////

#include "TMBALLEvent.h"
#include <iostream>
#include <fstream>
#include <vector>
#define ADC0 0x50000000
#define ADC1 0x58000000
#define ADC2 0x60000000
#define ADC3 0x68000000
#define TDC0 0x70000000
#define TDC1 0x78000000


Bool_t DebugMultiplicity = kFALSE;

TMBALLEvent *gmballevent = 0;

using namespace std;

//Constructor
TMBALLEvent::TMBALLEvent() 
{
 if(gmballevent==0)gmballevent = this;
mMult = 0;
 mRawMult = 0;

 for (int i=0; i<4;i++){
     mMultRing[i] = 0;
     mRawMultRing[i] = 0;
 }
  fMBALLHistoList = 0;
  fMBALLHistoList = new TList();

//---------This var tells us if the halo detector has been fired (Paola 28Oct2011)
fIsHaloFired = kFALSE;
//-----------------
}


//Destructor 
TMBALLEvent::~TMBALLEvent()
{
// cout<<"TMBALLEvent destructor...done"<<endl;
 gmballevent = 0;
 if(fMBALLHistoList!=0)
   {
       delete fMBALLHistoList;
       fMBALLHistoList=0;
   }

}

int TMBALLEvent::GetEvent(int *start)  
{
 int len = *start / 2 + 2;
 return len;
}

//************************************************
//
// MBALL readout -  event structure 
//   
//************************************************

int TMBALLEvent::ProcessCurrentEventMBS(int *buffer, int nWords)
{
    nWords /= 2;
    nWords -= 1;

    mRawMult = 0;
    mMult = 0;
    for (int i=0; i<4; i++){
	mRawMultRing[i] = 0;
	mMultRing[i] = 0;
    }
    

  // # of words
  int adc_channel;
  int adc_data;
  int module=0;
  int tag;
  
  int mNumEvts;

  int tWords = nWords;
  tWords -= 1;
  
  //  cout <<"nWords = "<<nWords<<endl;
  
  if (nWords < 2) return 0;
  int k=0;
  int baseadd;
  int baseadd_word;
  /*  
  int *buffcopy = buffer;
  for (int w = 0; w < nWords; w++)
      {
      cout <<(hex)<< *buffcopy << "\t";
      buffcopy++;
	  k++;
	  if (k==34)
	      { 
		  cout<<endl<<endl; k =0;	  
	      }
      }
  cout<<"------------------------------------------------"<<endl;
  */ 

  int isheader, isdata, isfooter;
  int tRing, tDet;
  int temp;

//---------This var tells us if the halo detector has been fired (Paola 28Oct2011)
//-------- I think it has to be reset here
 fIsHaloFired = kFALSE;
//------------------------------------


  
 while (tWords > 0) //this makes you continue to the following ADC once the fired channels of one ADC have been read (Paola)
    {
	buffer++;
	tWords--;
	
      tag = *buffer;

      isheader = (tag & 0x07000000)>> 24;
      isfooter = (tag & 0x07000000)>> 24;  // we don't really have to check whether it's a footer

      if (isheader!=2) continue;   // only if it's a header, otherwise continue
      
      mNumEvts = tag & 0x0000003F;
      baseadd  = tag & 0xF8000000;

            
     switch(baseadd)
	{
	case ADC0: 
	    module = 0; 
	    break;
	case ADC1: 
	    module = 1; 
	    break;
	case ADC2: 
	    module = 2; 
	    break;
	case ADC3: 
	    module = 3; 
	    break;
	case TDC0: 
	    module = 4; 
	    break;
	case TDC1: 
	    module = 5; 
	    break;
	default:
	    module = 1000;
	    mNumEvts = 0;
	    break;
	    
	}
	  
     if(DebugMultiplicity) cout<<"number of fired channels for THAT adc (also TDC) for this event : "<<mNumEvts<<endl; // mNumEvts is for THAT adc, how many words need to be read! It is not the number of fired adcs, but the number of fired channels for THAT adc!!!! (Paola)

     for (int i=0;i<mNumEvts;i++) //Reading the words recorded for THAT adc
        {
	    buffer++;
	    tWords--;

	    
	    isdata = ((*buffer)&0x07000000)>> 24;
	    if (isdata != 0) continue;

	    adc_channel = (int)(((*buffer)&(0x001f0000))>>16);
	    adc_data = (int)((*buffer)&0x00000fff);
	    baseadd_word = (int)((*buffer)&0xF8000000);
	  
	    // if ((module==4) || (module ==5))
	    // cout<<"module = "<<module<<"\t adc channel = "<<adc_channel<<"\t adc_data = "<<adc_data<<endl;

	  if (baseadd_word != baseadd) 
	      cout <<i<<" ERRRRRRRROOOOOORRRRR baseadd_word/baseadd/tag/buff word: "<<hex << baseadd_word << " / "<< baseadd<<" / "<<tag<<" / "<<buffer<<dec<<endl;
	  
	  switch(baseadd)
	      {
	      case ADC0:
		  module =0;
		  break;
	      case ADC1:
		  module =1;
		  break;
	      case ADC2:
		  module =2;
		  break;
	      case ADC3:
		  module= 3;
		  break;
	      case TDC0:
		  module=4;
		  break;
	      case TDC1:
		  module=5;
		  break;
	      }


	  temp = GetDetector(module,adc_channel);
	  tRing = temp/100;
	  tDet  = temp  - tRing *100;
	  if (!((tRing>4) && (tRing < 9))) continue;

	  //	  if(DebugMultiplicity)  PrintEvent(module,adc_channel, adc_data, tRing, tDet);

//---------Paola: the halo detector, at least at the beginning, was on ADC 0 (module 0) channels 22 and 23. 
// The pedestal for those 2 channels are 140 and 123, respectively (this are the values used durng the 
// experiment). I'm hardcoding these values here now, but we will change it

	  if(module==0){
	    if(adc_channel==22&&adc_data>140) fIsHaloFired=kTRUE;  //looking at ADC of halo det
	    if(adc_channel==23&&adc_data>123) fIsHaloFired=kTRUE;
	  }
//---------done with looking at the Halo detector;
	  
	  
	  if ((module==0) || (module==1)) {

	    if(DebugMultiplicity)  PrintEvent(module,adc_channel, adc_data, tRing, tDet);


	    mRawMult++;
	    mRawMultRing[tRing-5]++;
	    
	    int r = GetDetector(module,adc_channel)/100;
	    int d =  GetDetector(module,adc_channel) - r*100;
	    
	    
	    if (adc_data > mLookupAdcThreshTable[module*32+adc_channel][2])
	      {
		
		mMult++;
		mMultRing[tRing-5]++;
		if(DebugMultiplicity) 
		  {  cout<<"good : saving event ->";
		    cout << "module=" << module << "   adc channel=" << adc_channel << "    adc data =" << adc_data << endl;
		  }
		uBallModule.push_back(module);
		uBallChannel.push_back(adc_channel);
		uBallData.push_back(adc_data);
	      }
	    
	  }
	  
	} // for (int i=0;i<mNumEvts;i++): nNumEvts is for THAT adc, how many words need to be read! It is not the number of fired adcs, but the number of fired channels for THAT adc!!!!

    } // while (tWords > 0)
    
 if(DebugMultiplicity) cout<<"Done loop on fired ADCs in the event"<<endl;
 
 if(DebugMultiplicity)
   {
     cout<<" Raw M : "<< mRawMult
         <<" M : "<<mMult<<endl;
     for(int t=0;t<4;t++)
       cout<<" Ring "<<t+5<<" M : "<<mRawMultRing[t]<<" "<<mMultRing[t]<<endl;
   }


  return mRawMult;

}

//---------Paola: we add this method to pass the information if the halo detector has been fired or not
Bool_t TMBALLEvent::IsHaloFired()
{
return fIsHaloFired;
}


void TMBALLEvent::GetTiMeStamp(int *tsl, int *tsm, int *tsh)
{
 int ftsl=1,ftsm=1,ftsh=1;
 *tsl = ftsl;
 *tsm = ftsm;
 *tsh = ftsh;
}

int TMBALLEvent::GetMBALLMultiplicity()
{
 return mMult;
}

int TMBALLEvent::GetMBALLRingMultiplicity(int ring)
{
  int pos = ring-5;
 return mMultRing[pos];
}

int TMBALLEvent::GetMBALLRawMultiplicity()
{
 return mRawMult;
}

int TMBALLEvent::GetMBALLRingRawMultiplicity(int ring)
{
 return (mRawMultRing[ring-5]);
}

void TMBALLEvent::PassLists(TList *list)
{
  fMBALLHistoList = list;
}
void TMBALLEvent::LoadMapping(TString inF)
{
    
    ifstream inFile(inF,ios::in);
    
    if (!inFile) {
	
cout<<"failed to find inFile......."<<endl;
 
//	exit(1);
	
    }

    else cout<<"inFile exists"<<endl;
    
    int num=0;//, iembank, iemch, ring, det, preamp, preampch, adc, adcch, tdc, tdcch;    double theta, phi;
    string line;
    getline(inFile,line);

    //    mLookupTable[];
    int r=0;
    while (!inFile.eof())
    {
	inFile >> num;
	for (int c=0; c<12; c++)
	    inFile >> mLookupTable[r][c];
	
	cout << num << "  ";
	for (int c=0; c<12; c++)
	    cout << mLookupTable[r][c] <<" ";
	cout<<endl;

	r++;
    }
}

int TMBALLEvent::GetDetector(int module, int adcch)
{
    bool found = 0;
    int c=0;
    //    cout<<"adc/adcch/isEnergy = "<<adc<<" / "<< adcch <<" / "<< isEnergy <<endl;
    int col_adc=0, col_adcch=0;
    int mod=0; 
    
    switch (module)
	{
	case 0: 
	    col_adc = 6;
	    col_adcch  = 7;
	    mod = module;
	    break;
	case 1:
	    col_adc = 6;
	    col_adcch  = 7;
	    mod = module;
	    break;
	case 2:
	    col_adc = 6;
	    col_adcch  = 7;
	    mod = module-2;
	    break;
	case 3:
	    col_adc = 6;
	    col_adcch  = 7;
	    mod = module-2;
	    break;
	case 4: 
	    col_adc = 10;
	    col_adcch  = 11;
	    mod = module-4;
	    break;
	case 5:
	    col_adc = 10;
	    col_adcch  = 11;
	    mod = module-4;
	    break;
	default:
	    col_adc = 99;
	    col_adcch  = 99;
	    mod = 99;
	    break;
	    
	}
    
    while ((found!=1) && (c<49))
    {
	if ((mLookupTable[c][col_adc]==mod) && (mLookupTable[c][col_adcch]==adcch))
	    found=1;
	else c++;
    }
    if (found==0) return 999;
    else
    return int(mLookupTable[c][2]*100 + mLookupTable[c][3]);
}


int TMBALLEvent::GetDetectorRing(int module, int adcch)
{
  int temp = GetDetector(module,adcch);
  int ring = temp/100; //it' not a mistake, it has to be divided by 100 without point because we want the integer!!
  return ring;
}
int TMBALLEvent::GetDetectorNumber(int module, int adcch)
{
  int temp = GetDetector(module,adcch);
  int ring = temp/100; //it' not a mistake, it has to be divided by 100 without point because we want the integer!!
  int number = temp - ring*100;
  return number;
}

void TMBALLEvent::LoadThresholds(TString inF)
{
    ifstream inFile(inF,ios::in);
    if (!inFile) {
      cout<<"MICROBALL: failed to find threshold file......."<<endl;
	exit(1);
    }
    else cout<<"MICROBALL threshold file exists"<<endl;
    
    int num=0;//, iembank, iemch, ring, det, preamp, preampch, adc, adcch, tdc, tdcch;    double theta, phi;
    string line;
    getline(inFile,line);

    int r=0;
    cout<<"MICROBALL: ADC Thresholds"<<endl;
    while (!inFile.eof())
    {
	inFile >> mLookupAdcThreshTable[r][0] >> mLookupAdcThreshTable[r][1] >> mLookupAdcThreshTable[r][2];
	cout << r<< "\t"<<mLookupAdcThreshTable[r][0] << "\t" << mLookupAdcThreshTable[r][1] << "\t" << mLookupAdcThreshTable[r][2]<<endl;
	r++;
    }
}


int TMBALLEvent::GetAdcPedestal(int module, int channel)
{
  return mLookupAdcThreshTable[module*32+channel][2];
}
int TMBALLEvent::GetDetectorPedestal(int ring, int det)
{
  int module = GetADCModule(ring, det);
  int channel = GetADCChannel(ring, det);
  return mLookupAdcThreshTable[module*32+channel][2];
}


double* TMBALLEvent::GetADCAngles(int module, int adcch)
{
  double *angle = new double[2];
    bool found = 0;
    int c=0;
    //    cout<<"adc/adcch/isEnergy = "<<adc<<" / "<< adcch <<" / "<< isEnergy <<endl;
    int col_adc=0, col_adcch=0;
    int mod=0; 
    
    switch (module)
	{
	case 0: 
	    col_adc = 6;
	    col_adcch  = 7;
	    mod = module;
	    break;
	case 1:
	    col_adc = 6;
	    col_adcch  = 7;
	    mod = module;
	    break;
	case 2:
	    col_adc = 6;
	    col_adcch  = 7;
	    mod = module-2;
	    break;
	case 3:
	    col_adc = 6;
	    col_adcch  = 7;
	    mod = module-2;
	    break;
	case 4: 
	    col_adc = 10;
	    col_adcch  = 11;
	    mod = module-4;
	    break;
	case 5:
	    col_adc = 10;
	    col_adcch  = 11;
	    mod = module-4;
	    break;
	default:
	    col_adc = 99;
	    col_adcch  = 99;
	    mod = 99;
	    break;
	    
	}
    
    while ((found!=1) && (c<49))
    {
	if ((mLookupTable[c][col_adc]==mod) && (mLookupTable[c][col_adcch]==adcch))
	    found=1;
	else c++;
    }
    if (found==0) 
      {angle[0] = 999;
	angle[1] = 999;}
    else
      {angle[0] =  mLookupTable[c][8]; //theta
	angle[1] =  mLookupTable[c][9];} //phi

    return angle;
}



int TMBALLEvent::GetADCModule(int ring, int det)
{
    bool found = 0;
    int c=0;
    //    cout<<"adc/adcch/isEnergy = "<<adc<<" / "<< adcch <<" / "<< isEnergy <<endl;
    int col_ring = 2;
    int col_det = 3;
 
    while ((found!=1) && (c<49))
    {
	if ((mLookupTable[c][col_ring]==ring) && (mLookupTable[c][col_det]==det))
	    found=1;
	else c++;
    }


    return mLookupTable[c][6];
}

int TMBALLEvent::GetADCChannel(int ring, int det)
{
    bool found = 0;
    int c=0;
    //    cout<<"adc/adcch/isEnergy = "<<adc<<" / "<< adcch <<" / "<< isEnergy <<endl;
    int col_ring = 2;
    int col_det = 3;
 
    while ((found!=1) && (c<49))
    {
	if ((mLookupTable[c][col_ring]==ring) && (mLookupTable[c][col_det]==det))
	    found=1;
	else c++;
    }


    return mLookupTable[c][7];
}


int TMBALLEvent::GetTDCModule(int ring, int det)
{
    bool found = 0;
    int c=0;
    //    cout<<"adc/adcch/isEnergy = "<<adc<<" / "<< adcch <<" / "<< isEnergy <<endl;
    int col_ring = 2;
    int col_det = 3;
 
    while ((found!=1) && (c<49))
    {
	if ((mLookupTable[c][col_ring]==ring) && (mLookupTable[c][col_det]==det))
	    found=1;
	else c++;
    }


    return mLookupTable[c][10];
}

int TMBALLEvent::GetTDCChannel(int ring, int det)
{
    bool found = 0;
    int c=0;
    //    cout<<"adc/adcch/isEnergy = "<<adc<<" / "<< adcch <<" / "<< isEnergy <<endl;
    int col_ring = 2;
    int col_det = 3;
 
    while ((found!=1) && (c<49))
    {
	if ((mLookupTable[c][col_ring]==ring) && (mLookupTable[c][col_det]==det))
	    found=1;
	else c++;
    }


    return mLookupTable[c][11];
}



double TMBALLEvent::GetTheta(int ring, int det)
{
    bool found = 0;
    int c=0;
    //    cout<<"adc/adcch/isEnergy = "<<adc<<" / "<< adcch <<" / "<< isEnergy <<endl;
    int col_ring = 2;
    int col_det = 3;
 
    while ((found!=1) && (c<49))
    {
	if ((mLookupTable[c][col_ring]==ring) && (mLookupTable[c][col_det]==det))
	    found=1;
	else c++;
    }


    return mLookupTable[c][8];
}

double TMBALLEvent::GetPhi(int ring, int det)
{
    bool found = 0;
    int c=0;
    //    cout<<"adc/adcch/isEnergy = "<<adc<<" / "<< adcch <<" / "<< isEnergy <<endl;
    int col_ring = 2;
    int col_det = 3;
 
    while ((found!=1) && (c<49))
    {
	if ((mLookupTable[c][col_ring]==ring) && (mLookupTable[c][col_det]==det))
	    found=1;
	else c++;
    }


    return mLookupTable[c][9];
}

void TMBALLEvent::PrintEvent(int adc,int adc_ch, int adc_data, int Ring,int Det)
{
  cout<<"Ring "<<Ring<<" Det : "<<Det<<" ADC : "<<adc<<" ch : "<<adc_ch<<" signal : "<<adc_data<<endl;

}