/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/*




Author: R. GUERNANE LPSC Grenoble CNRS/IN2P3
*/

#include "AliEMCALTriggerSTU.h"
#include "AliCDBManager.h"
#include "AliCDBEntry.h"
#include "AliEMCALCalibData.h"
#include "AliVZEROCalibData.h"
#include "AliVZEROdigit.h"
#include "AliEMCALTriggerPatch.h"
#include "AliESDVZERO.h"
#include "AliLog.h"

#include <TClonesArray.h>
#include <TSystem.h>
#include <TH2F.h>
#include <TFile.h>
#include <TTree.h>

#include <fstream>
#include <Riostream.h>
#include <cstdlib>

ClassImp(AliEMCALTriggerSTU)

//_______________
AliEMCALTriggerSTU::AliEMCALTriggerSTU() : AliEMCALTriggerBoard()
{
	//
	fV0M[0] = fV0M[1] = 0;
}

//_______________
AliEMCALTriggerSTU::AliEMCALTriggerSTU(AliEMCALCalibData *calibData, const TVector2& RS) : 
AliEMCALTriggerBoard(calibData, RS)
{
	//
	fV0M[0] = fV0M[1] = 0;
}

//_______________
AliEMCALTriggerSTU::~AliEMCALTriggerSTU()
{
	//
}

//_______________
void AliEMCALTriggerSTU::BuildMap( Int_t iTRU, Int_t** M, const TVector2* rSize )
{
	//
	if ( iTRU == 31 ) iTRU = 35;
	
	Int_t i2y = iTRU / 3 / 2;
	
	if ( ( iTRU / 3 ) % 2 ) // odd (z<0) C side
	{
		Int_t i1y = 2 - ( iTRU - int( iTRU / 3 ) * 3 ); // 0 1 2 w/ increasing phi

		for (Int_t i=0; i<rSize->X(); i++) 
			for (Int_t j=0; j<rSize->Y(); j++) fMap[24+i][j + i1y * 4 + i2y * 12] = M[i][j];
	}
	else                   // A side
	{
		Int_t i1y =       iTRU - int( iTRU / 3 ) * 3;
		
		for (Int_t i=0; i<rSize->X(); i++)
			for (Int_t j=0; j<rSize->Y(); j++) fMap[   i][j + i1y * 4 + i2y * 12] = M[i][j];
	}	
}

//_______________
void AliEMCALTriggerSTU::L1( L1TriggerType_t type )
{
	//
	SlidingWindow( type, int(ThresholdFromV0( type )) );	
}

//________________
void AliEMCALTriggerSTU::PrintADC( L1TriggerType_t type, TVector2& pos, TVector2& idx )
{
	//
        Int_t ix = (Int_t) (( pos.X() + fPatchSize->X() ) * fSubRegionSize->X());
        Int_t iy = (Int_t) (( pos.Y() + fPatchSize->Y() ) * fSubRegionSize->Y());
	
	TString subRegionADC[] = {"0->15", "16->31", "32->47", "48->63", "64->79", "80->95"};
	
	switch ( type )
	{
		case kGamma:
		{
			Int_t iTRU = ( (ix-1) < 24 ) ? 31 - int(pos.Y() / 4) : 15 - int(pos.Y() / 4);
			
			printf("TRU #%d row #%d col #%d fastor: ",iTRU,int(idx.Y()),int(idx.X()));
			
			for (Int_t i=(Int_t)(pos.X() * fSubRegionSize->X());i<ix;i++) 
			{
			  for (Int_t j=(Int_t) (pos.Y() * fSubRegionSize->Y());j<iy;j++) 
				{
					Int_t jtru = ( i < 24 ) ? 31 - j / 4 : 15 - j / 4;
					printf("TRU#%d_%d ",jtru,fMap[i][j]);
				}
			}
			
			cout << endl;
		}	
		break;
		case kJet:
		{
			//Int_t jTRU = ( (ix-1) < 24 ) ? 31 - (iy-1) / 4 : 15 - (iy-1) / 4;
			
			printf("jet found at row : %d and col : %d",int(idx.X()),int(idx.Y()));
			
			Char_t* vPair = 0x0;
			Int_t nSubRegion = 0;
			
			for (Int_t i=(Int_t)(pos.X() * fSubRegionSize->X());i<ix;i++) 
			{
			  for (Int_t j=(Int_t)(pos.Y() * fSubRegionSize->Y());j<iy;j++) 
				{
					Int_t itru = ( i < 24 ) ? 31 - j / 4 : 15 - j / 4;
					
					Int_t idSR = fMap[i][j]/16;
					
					Char_t value = ((itru << 3) & 0xF8) | (idSR & 0x7);
					
					Bool_t isFound = kFALSE;
					
					for (Int_t k=0;k<nSubRegion && nSubRegion;k++)
					{
						if (vPair[k] == value) isFound = kTRUE;
					}
					
					if (!isFound) 
					{	
						nSubRegion++;
						vPair = (Char_t*)realloc(vPair, nSubRegion * sizeof(Char_t));
						if (vPair == NULL) {AliError("Error (re)allocating PrintADC() memory");}
						
						vPair[nSubRegion-1] = value;
					}
				}
			}
			
			cout << " fastor:"; 
			for (Int_t i=0;i<nSubRegion;i++)
			{
				cout << " TRU#" << ((vPair[i] & 0xF8) >> 3) << "_" << subRegionADC[(vPair[i] & 0x7)];
			}
		
			cout << endl;
			if (vPair) delete[] vPair;
		}
		break;
		default:
			AliError("AliEMCALTriggerSTU::PrintADC(): Undefined trigger type, pls check!");
	}
}

//________________
void AliEMCALTriggerSTU::V0Multiplicity( TTree& treeV0 )
{  
	//
    AliCDBManager *man = AliCDBManager::Instance();
    AliCDBEntry *entry = man->Get("VZERO/Calib/Data");
    AliVZEROCalibData *calibdata = (AliVZEROCalibData*)entry->GetObject();
	
    TClonesArray* digitsArray = 0x0;
    treeV0.SetBranchAddress("VZERODigit",&digitsArray);

	Float_t mult[64];
	Short_t  adc[64];
   
	for (Int_t i=0; i<64; i++)
	{
		adc[i]    = 0;
		mult[i]   = 0.0;
	}

	Int_t nEntries = (Int_t)treeV0.GetEntries();

	for (Int_t e=0; e<nEntries; e++) 
	{
		treeV0.GetEvent(e);

		Int_t nDigits = digitsArray->GetEntriesFast();
          
		for (Int_t d=0; d<nDigits; d++) 
		{    
			AliVZEROdigit* digit = (AliVZEROdigit*)digitsArray->At(d);      
			Int_t  pmNumber      = digit->PMNumber();
			
			if (adc[pmNumber] > (int(1.0/calibdata->GetMIPperADC(pmNumber)) /2) ) 
				mult[pmNumber] += float(adc[pmNumber])*calibdata->GetMIPperADC(pmNumber);
		}
	}
  
	//  0..31 V0C 
	// 32..63 V0A
	for (Int_t j=0; j<32; j++) 
	{
		fV0M[0] += short(mult[j   ]+0.5); 
		fV0M[1] += short(mult[j+32]+0.5); 
	}
}

//________________
void AliEMCALTriggerSTU::FetchFOR( Int_t iTRU, Int_t **R, const TVector2* rSize )
{
	// L1 triggers run over the whole EMCal surface
	// STU builds its own fRegion aggregating TRU fRegion into one 
	
	// STU I from TRUs O in Olivier's coordinate system
	
	if ( iTRU == 31 ) iTRU = 35;
		
	Int_t i2y = iTRU / 3 / 2;	
	
	if ( ( iTRU / 3 ) % 2 ) // C side odd (z<0)
	{
		Int_t i1y = 2 - ( iTRU - int( iTRU / 3 ) * 3 ); // 0 1 2 w/ increasing phi
		
		for (Int_t i=0; i<rSize->X(); i++)                                            //  0:23 0:4 
			for (Int_t j=0; j<rSize->Y(); j++) fRegion[24+i][j + i1y * 4 + i2y * 12] = R[i][j];
	}
	else                    // A side
	{
		Int_t i1y =       iTRU - int( iTRU / 3 ) * 3;
		
		for (Int_t i=0; i<rSize->X(); i++)
			for (Int_t j=0; j<rSize->Y(); j++) fRegion[   i][j + i1y * 4 + i2y * 12] = R[i][j];
	}
}

//___________
void AliEMCALTriggerSTU::PatchGenerator(const TClonesArray* lpos, Int_t val)
{
	ZeroRegion();
	
	Int_t vTRU[32][24][4] ;//= {0};
	
	for (Int_t i = 0; i < 32; i++) 
		for (Int_t j = 0; j < 24; j++) 
			for (Int_t k = 0; k < 4; k++) vTRU[i][j][k]=0;
	
	
	AliWarning("AliEMCALTriggerSTU::PatchGenerator(): STU region has been reset!");
	
	// Fill the patch FOR at 'pos' w/ value 'val'
	
	TIter NextPosition( lpos );
	
	while ( TVector2 *pos = (TVector2*)NextPosition() )
	{
		pos->Print();
		for (Int_t i=(Int_t)(pos->X()*fSubRegionSize->X()); i<int((pos->X()+fPatchSize->X())*fSubRegionSize->X()); i++)
		  for (Int_t j=(Int_t)(pos->Y()*fSubRegionSize->Y()); j<int((pos->Y()+fPatchSize->Y())*fSubRegionSize->Y()); j++) 
				fRegion[i][j] = val;
	}
	
	for (Int_t i=0; i<2; i++)
	{
		for (Int_t j=0; j<16; j++)
		{
			Int_t iTRU = ( !i ) ? 31 - j : 15 - j;
			
			for (Int_t i1=24*i; i1<24*(i+1); i1++)
			{
				for (Int_t j1=4*j; j1<4*(j+1); j1++)
				{
					vTRU[iTRU][i1%24][j1%4] = fRegion[i1][j1];
				}
			}
		}
	}
	
	gSystem->Exec(Form("mkdir -p GP"));

	for (Int_t i=0; i<32; i++)
	{
		std::ofstream outfile(Form("GP/data_TRU%d.txt",i),std::ios::trunc);
	
		for (Int_t j=0;j<96;j++) 
		{
			Int_t ietam, iphim;
			
			if ( int(i / 16) )
			{
				ietam =      j/4;
				iphim =  3 - j%4;
			}
			else
			{					
				ietam = 23 - j/4;
				iphim =      j%4;
			}
			
			outfile << vTRU[i][ietam][iphim] << endl;
		}
	
		outfile.close();
	}
}

//___________
Float_t AliEMCALTriggerSTU::ThresholdFromV0( L1TriggerType_t type )
{	
	// Compute threshold FIXME: need an access to the OCDB
	// to get f(V0) parameters depending on trigger type
	
	switch ( type )
	{
		case kGamma:
			break;
		case kJet:
//			return 15.;
			break;
		default:
			AliError("AliEMCALTriggerSTU::ThresholdFromV0(): Undefined trigger type, pls check!");
	}
	
	return 0.;
}

//__________
void AliEMCALTriggerSTU::Reset()
{
	fPatches->Delete();
}