[u/mrichter/AliRoot.git] / ZDC / AliZDCMerger.cxx

/**************************************************************************
 * 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.                  *
 **************************************************************************/


///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  			ZDC event merging class                              //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

// --- ROOT system
#include <Riostream.h>
#include <TTree.h>
#include <TFile.h>
#include <TDirectory.h>
#include <TNtuple.h>
#include <TSystem.h>
#include <TROOT.h>

// --- AliRoot header files
#include "AliZDCMerger.h"
#include "AliZDC.h"
#include "AliZDCHit.h"
#include "AliZDCMergedHit.h"
#include "AliZDCDigit.h"
#include "AliZDCFragment.h"
#include "AliRun.h"
#include "AliDetector.h"
#include "AliHeader.h"
#include "AliGenEventHeader.h"
#include "AliGenHijingEventHeader.h"

ClassImp(AliZDCMerger)

//int comp(const void *i,const void *j) {return *(int *)i - *(int *)j;}

//____________________________________________________________________________
AliZDCMerger::AliZDCMerger()
{
// Default constructor    
    //fMerge       = kDigitize	-> Only digitization
    //fMerge       = kMerge	-> Digitization + Merging
    fMerge       = kMerge;	
    fFnBgr       = 0;
    fBgrFile     = 0;
    fNEvBgr	 = 0;
    fHitsBgr     = 0;
    fImpPar      = 0;
    fSpecn       = 0;
    fSpecp       = 0;
    fFreeSpn     = 0;
    fFreeSpp     = 0;
    fFnSpecn     = 0;
    fSpecnFile   = 0;
    fFnSpecp     = 0;
    fSpecpFile   = 0;
    fNMhits	 = 0;
     
}

//____________________________________________________________________________
AliZDCMerger::~AliZDCMerger()
{
    // Destructor
    if (fSpecnFile)  delete fSpecnFile;
    if (fSpecpFile)  delete fSpecpFile;
}

//____________________________________________________________________________
void AliZDCMerger::InitMerging()
{
    // Hits tree, impact parameter, num. of spectators n & p 
    // 		in background (full Hijing) event
    Float_t b;
    Background(b, fSpecn, fSpecp);

    // Production of nuclear fragments -> num. of FREE spectators n & p
    Fragmentation(b, fSpecn, fSpecp, fFreeSpn, fFreeSpp);
    
    // Extract from spectators distribution the signal events:
    // NFreeSpectatorN spectator n & NFreeSpectatorP spectator p 
    Mixing();
}

//____________________________________________________________________________
void AliZDCMerger::Background(Float_t &fImpPar, Int_t &fSpecn, Int_t &fSpecp)
{
    
    // --- Open the background file
  if (fMerge && !fBgrFile) fBgrFile = OpenBgrFile();
    
    // --- Read from the TreeE impact parameter (b),
    //     # of spectators n and p (fSpecn, fSpecp)
    fBgrFile->cd();

//    // Get AliRun object from file or create it if not on file
//    gAlice = (AliRun*)fBgrFile->Get("gAlice");
//    if (!gAlice) {
//      gAlice = (AliRun*)fBgrFile->Get("gAlice");
//      if (gAlice) printf("AliRun object found on file\n");
//      if (!gAlice) {
//	  printf("\n create new gAlice object");
//	  gAlice = new AliRun("gAlice","Alice test program");
//      }
//    }
    
    //    gAlice->GetEvent(fNEvBgr);  this is done in the steering macro
    AliHeader *header = gAlice->GetHeader();
    AliGenEventHeader* mcHeader = header->GenEventHeader();
    fImpPar = ((AliGenHijingEventHeader*) mcHeader)->ImpactParameter();
    Int_t dSpecn  = ((AliGenHijingEventHeader*) mcHeader)->Spectatorsn();
    Int_t dSpecp  = ((AliGenHijingEventHeader*) mcHeader)->Spectatorsp();
    // Until there is only 1 ZDC set the # of spectators must be divided by 2!!!
    fSpecn = dSpecn/2;
    fSpecp = dSpecp/2;
    printf("\n	HIJING ev. #%d - b = %f fm, Nspecn = %d, Nspecp = %d\n",
            fNEvBgr,fImpPar,fSpecn,fSpecp);
}

//____________________________________________________________________________
TFile* AliZDCMerger::OpenBgrFile()
{
    // Initialise background event
  TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject(fFnBgr);
  if(!file)cerr<<"AliZDCMerger: background file "<<fFnBgr<<" not found\n";
  //    TFile *file = new TFile(fFnBgr,"UPDATE");
    printf("\n AliZDCMerger --- Background event -> %s file opened \n", fFnBgr);
    fHitsBgr = new TClonesArray("AliZDCHit",1000);
    fMHits   = new TClonesArray("AliZDCMergedHit",1000);
    return file;
}

//____________________________________________________________________________
void AliZDCMerger::Fragmentation(Float_t fImpPar, Int_t fSpecn, Int_t fSpecp,
                                 Int_t &fFreeSpn, Int_t &fFreeSpp)
{
    //printf("\n	Fragmentation -> fSpecn = %d, fSpecp = %d\n",fSpecn,fSpecp);
    Int_t j, zz[100], nn[100], nAlpha, Ztot, Ntot;
    AliZDCFragment *frag = new AliZDCFragment(fImpPar);
    for(j=0; j<=99; j++){
       zz[j] =0;
       nn[j] =0;
    }

    // Fragments generation
    frag->GenerateIMF(zz, nAlpha);

    // Attach neutrons
    Ztot=0;
    Ntot=0;
    frag->AttachNeutrons(zz, nn, Ztot, Ntot);
    fFreeSpn = fSpecn-Ntot-2*nAlpha;
    fFreeSpp = fSpecp-Ztot-2*nAlpha;
    if(fFreeSpn<0) fFreeSpn=0;
    if(fFreeSpp<0) fFreeSpp=0;
    //printf("\n			2*nAlpha = %d, Ztot = %d, Ntot = %d\n",2*nAlpha, Ztot,    Ntot);
    printf("\n	Fragmentation -> FreeSpn = %d, FreeSpp = %d\n",fFreeSpn,fFreeSpp);
}

//____________________________________________________________________________
void AliZDCMerger::Mixing()
{
    
    //printf("\n	AliZDCMerger->Mixing\n");
       
    // ### Background event Hits ###########################################
    fBgrFile->cd();
//    fBgrFile->ls();
    
   AliZDC *ZDC = (AliZDC *)gAlice->GetModule("ZDC");
//    if(ZDC) printf("\n	Ho trovato lo ZDC!\n");

//    fNEvBgr = 0; // Let's suppose to have 1 full Hijing event per file
    // Hits tree
    char treeBgrName[20];
    sprintf(treeBgrName,"TreeH%d",fNEvBgr);
    fTrHBgr = (TTree*)gDirectory->Get(treeBgrName); 
    if(!fTrHBgr){
      printf("\n ERROR -> Can't find TreeH%d in background file\n",fNEvBgr);
    }    
//    fTrHBgr->Print();
    
    // Branch address
    TBranch *branch;
    char branchname[20];
    sprintf(branchname,"%s",ZDC->GetName());
    if(fTrHBgr && fHitsBgr){
//      printf("\n	fTrHBgr!=0 && fHitsBgr!=0\n");
      branch = fTrHBgr->GetBranch(branchname);
      if(branch) branch->SetAddress(&fHitsBgr);
    }
    
    Int_t ntracks =  (Int_t) fTrHBgr->GetEntries();
    //printf("\n	--- ntracks = %d\n\n", ntracks);
    
    Int_t itrack, nhits, ihit, j, sector[2];
    AliZDCHit* zdcHit;
    AliZDCMergedHit *MHit;
    Float_t MHits[7];
    TClonesArray &sdigits = *fMHits;	// SDigits TCArray
    fNMhits = 0;

    // --- Tracks loop
    for(itrack=0; itrack<ntracks; itrack++){
       fTrHBgr->GetEvent(itrack);
       
       nhits = fHitsBgr->GetEntries();
//       printf(" 	      nhits = %d \n", nhits);
       for(ihit=0; ihit<nhits; ihit++){    
	  zdcHit = (AliZDCHit*) fHitsBgr->UncheckedAt(ihit);

	    for(j=0; j<2; j++) sector[j] = zdcHit->GetVolume(j);
	    MHits[0] = zdcHit->GetPrimKinEn();
	    MHits[1] = zdcHit->GetXImpact();
	    MHits[2] = zdcHit->GetYImpact();
	    MHits[3] = zdcHit->GetSFlag();
       	    MHits[4] = zdcHit->GetLightPMQ();
	    MHits[5] = zdcHit->GetLightPMC();
	    MHits[6] = zdcHit->GetEnergy();
	    MHit = new AliZDCMergedHit(sector, MHits);
//	    MHit->Print("");
  	    new((*fMHits)[fNMhits]) AliZDCMergedHit(*MHit);
	    new (sdigits[fNMhits])  AliZDCMergedHit(*MHit);
	    delete MHit;
	    fNMhits++;
	  }//Hits loop
	  
    } // Tracks loop
    //printf("	fNMhits (after bckg) = %d, \n",fNMhits);     
        
    // ### Signal event Hits ###########################################
    // --- Neutrons
    ExtractSignal(1);
        
    // --- Protons
    ExtractSignal(2);
    //printf("	fNMhits (after signal) = %d \n",fNMhits); 
        
}

//____________________________________________________________________________
void AliZDCMerger::ExtractSignal(Int_t SpecType)
{

// printf("\n	Entering in Extract Signal\n");
 
 Int_t NumEvents = 0;
 if(SpecType == 1){		// --- Signal for spectator neutrons
   fFnSpecn = gSystem->ExpandPathName("$ALICE/$ALICE_LEVEL/ZDC/ZNsignalntu.root");
   fSpecnFile = TFile::Open(fFnSpecn,"R");
   fSpecnFile->cd();
   printf("\n	--- ExtractSignal x n: file %s opened\n", fFnSpecn);
   NumEvents = fFreeSpn;
 }
 else if(SpecType == 2){	// --- Signal for spectator protons
   fFnSpecp = gSystem->ExpandPathName("$ALICE/$ALICE_LEVEL/ZDC/ZPsignalntu.root");
   fSpecpFile = TFile::Open(fFnSpecp,"R");
   fSpecpFile->cd();
   printf("\n	--- ExtractSignal x p: file %s opened\n", fFnSpecp);
   NumEvents = fFreeSpp;
 }
 //printf("\n		# of free spectators = %d\n", NumEvents);
 //printf("\n         fNMhits (before adding signal) = %d\n",fNMhits);

  TNtuple *ZDCSignal = (TNtuple*) gDirectory->Get("ZDCSignal");
  Int_t nentries = (Int_t) ZDCSignal->GetEntries();
  //printf("\n   # entries = %d\n", nentries);
  
  AliZDCMergedHit *MHit; 
  Float_t *entry, HitsSpec[7];
  Int_t pl, i, j, k, iev=0, rnd[125], Volume[2];
  for(pl=0;pl<125;pl++){
     rnd[pl] = 0;
  }
  for(pl=0;pl<NumEvents;pl++){
     rnd[pl] = (Int_t) (9999*gRandom->Rndm());
     if(rnd[pl] >= 9998) rnd[pl] = 9997;
     //printf("	rnd[%d] = %d\n",pl,rnd[pl]);     
  }
  // Sorting vector in ascending order with C function QSORT 
  qsort((void*)rnd,NumEvents,sizeof(Int_t),comp);
  //for(pl=0;pl<NumEvents;pl++){
  ////printf("	rnd[%d] = %d\n",pl,rnd[pl]);     
  //}
  do{
     for(i=0; i<nentries; i++){  
  	ZDCSignal->GetEvent(i);
  	entry = ZDCSignal->GetArgs();
  	if(entry[0] == rnd[iev]){
          for(k=0; k<2; k++) Volume[k] = (Int_t) entry[k+1];
          for(j=0; j<7; j++){
             HitsSpec[j] = entry[j+3];
          }
          //printf("\n   i = %d, iev = %d, entry[0] = %f, rnd[%d] = %d ",i,iev,entry[0],iev,rnd[iev]);
	  MHit = new AliZDCMergedHit(Volume, HitsSpec);
	  new((*fMHits)[fNMhits++]) AliZDCMergedHit(*MHit);
	  delete MHit;
  	}
  	else if(entry[0] > rnd[iev]){
	  iev++;
	  continue;
	}
     }
  }while(iev<NumEvents);
  
  if(SpecType ==1){
    //printf("\n         fNMhits (after n signal) = %d\n",fNMhits);
    fSpecnFile->Close();
  }
  else if(SpecType == 2){
    //printf("\n         fNMhits (after p signal) = %d\n",fNMhits);
    fSpecpFile->Close();
  }
      
}

//____________________________________________________________________________
void AliZDCMerger::Digitize(Int_t fNMhits, TClonesArray *fMHits)
{

  printf("\n	AliZDCMerger->Digitize()");

  AliZDC *ZDC = (AliZDC *)gAlice->GetModule("ZDC");
//  if(ZDC) printf("\n 	Ho trovato lo ZDC!\n");
  Int_t lightQ, lightC, sector[2], digit;
  Int_t PMCZN = 0, PMCZP = 0, PMQZN[4], PMQZP[4], PMZEM1 = 0, PMZEM2 = 0;
  Int_t i;
  for(i=0; i<4; i++){
     PMQZN[i] = 0;
     PMQZP[i] = 0;
  }
  
  AliZDCMergedHit *MHit;
  Int_t imhit;
  printf("	   fNMHits = %d\n", fNMhits);
  // Loop over SDigits
  for(imhit=0; imhit<fNMhits; imhit++){
     
     MHit = (AliZDCMergedHit*) fMHits->UncheckedAt(imhit);
     sector[0] = MHit->GetSector(0);
     sector[1] = MHit->GetSector(1);
   // tmp -> c'erano i quadranti cannati!
  if((sector[1]!=1) && (sector[1]!=2) && (sector[1]!=3) && (sector[1]!=4)){
     printf("\n  *** ERROR!!! sector[0] = %d, sector[1] = %d\n",
 	      sector[0], sector[1]);
       sector[1] = 0;
   }
     lightQ    = Int_t(MHit->GetLightPMQ());
     lightC    = Int_t(MHit->GetLightPMC());
//     printf("    imhit = %d -> DET. = %d, quad = %d,PMQ = %d, PMC = %d\n", 
//      imhit,sector[0], sector[1],lightQ, lightC);
     
     if(sector[0] == 1){   	   //ZN 
       PMCZN = PMCZN + lightC;
       PMQZN[sector[1]-1] = PMQZN[sector[1]-1] + lightQ;
     }
     else if(sector[0] == 2){	   //ZP 
       PMCZP = PMCZP + lightC;
       PMQZP[sector[1]-1] = PMQZP[sector[1]-1] + lightQ;
     }
     else if(sector[0] == 3){	   //ZEM 
       if(sector[1] ==1) PMZEM1 = PMZEM1 + lightC;
       else              PMZEM2 = PMZEM2 + lightQ;
     }
  } // SDigits loop
      
  // ### Digits creation ###############################################
  // Create digits for ZN
  Int_t PedValue;
  sector[0] = 1; // Detector = ZN
  sector[1] = 0; // Common PM ADC
  digit = Phe2ADCch(1, 0, PMCZN);
  printf("\n\n	ZN ###	PMCZN = %d	ADCZN = %d",PMCZN, digit);
  PedValue = AddPedestal();
  digit += PedValue;
//  printf("	PedValue = %d",PedValue);
  ZDC->AddDigit(sector, digit);
  Int_t j;
  for(j=0; j<4; j++){
    sector[1] = j+1; // Towers PM ADCs
    digit = Phe2ADCch(1, j+1, PMQZN[j]);
    printf("\n		PMQZN[%d] = %d	phe	ADCZN[%d] = %d ADCch",j,PMQZN[j],j,digit);
    PedValue = AddPedestal();
    digit += PedValue;
//    printf("	PedValue = %d",PedValue);
    ZDC->AddDigit(sector, digit);
  }
  printf("\n");
  
  // Create digits for ZP
  sector[0] = 2; // Detector = ZP
  sector[1] = 0; // Common PM ADC
  digit = Phe2ADCch(2, 0, PMCZP);
  printf("\n	ZP --- PMCZP = %d	phe	ADCZP = %d ADCch",PMCZP,digit);
  PedValue = AddPedestal();
  digit += PedValue;
//  printf("	PedValue = %d",PedValue);
  ZDC->AddDigit(sector, digit);
  for(j=0; j<4; j++){
    sector[1] = j+1; // Towers PM ADCs
    digit = Phe2ADCch(2, j+1, PMQZP[j]);
    printf("\n	       PMQZP[%d] = %d	phe	ADCZP[%d] = %d ADCch",j,PMQZP[j],j,digit);
    PedValue = AddPedestal();
    digit += PedValue;
//    printf("	PedValue = %d",PedValue);
    ZDC->AddDigit(sector, digit);
  }
  printf("\n");
  
  // Create digits for ZEM
  sector[0] = 3; 
  sector[1] = 1; // Detector = ZEM1
  digit  = Phe2ADCch(3, 1, PMZEM1);
  printf("\n	ZEM *** PMZEM1 = %d      phe     ADCZEM1 = %d ADCch",PMZEM1,digit);
  PedValue = AddPedestal();
  digit += PedValue;
//  printf("  PedValue = %d\n",PedValue);
  ZDC->AddDigit(sector, digit); 
  sector[1] = 2; // Detector = ZEM2
  digit  = Phe2ADCch(3, 2, PMZEM2);
  printf("\n	ZEM *** PMZEM2 = %d      phe     ADCZEM2 = %d ADCch\n",PMZEM2,digit);
  PedValue = AddPedestal();
  digit += PedValue;
//  printf("  PedValue = %d\n",PedValue);
  ZDC->AddDigit(sector, digit); 
}

//_____________________________________________________________________________
Int_t AliZDCMerger::Phe2ADCch(Int_t Det, Int_t Quad, Int_t Light)
{
  // Evaluation of the ADC channel corresponding to the light yield Light

  if(gAlice->GetDebug() > 0){
//    printf("\n  Phe2ADCch -> Detector = %d, Quadrant = %d, Light = %d\n", Det, Quad, Light);
  }
  
  Int_t ADCch = 0;
  
  Int_t j,i;
  for(i=0; i<3; i++){
     for(j=0; j<5; j++){
        fPMGain[i][j]   = 100000.;
     }
  }
  fADCRes   = 0.00000064; // ADC Resolution: 250 fC/ADCch
  
  ADCch = (Int_t) (Light*fPMGain[Det-1][Quad]*fADCRes);
     
  return ADCch;
}

//_____________________________________________________________________________
Int_t AliZDCMerger::AddPedestal()
{
  // --- Pedestal value -> extracted from a gaussian distribution
  // obtained from the beam test on the ZEM prototype (Aug. 2000)
  
  Int_t PedValue;
  Float_t PedMean  = 50.;
  Float_t PedWidth = 5.;
  
  PedValue    = (Int_t) gRandom->Gaus(PedMean,PedWidth);
  
  return PedValue;
}
Commit	Line	Data
359cdddc	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16
	17	///////////////////////////////////////////////////////////////////////////////
	18	// //
	19	// ZDC event merging class //
	20	// //
	21	///////////////////////////////////////////////////////////////////////////////
	22
	23	// --- ROOT system
ef0750c2	24	#include <Riostream.h>
359cdddc	25	#include <TTree.h>
	26	#include <TFile.h>
	27	#include <TDirectory.h>
1acbaaa8	28	#include <TNtuple.h>
53c30603	29	#include <TSystem.h>
fe107ada	30	#include <TROOT.h>
359cdddc	31
	32	// --- AliRoot header files
	33	#include "AliZDCMerger.h"
	34	#include "AliZDC.h"
	35	#include "AliZDCHit.h"
1acbaaa8	36	#include "AliZDCMergedHit.h"
359cdddc	37	#include "AliZDCDigit.h"
	38	#include "AliZDCFragment.h"
	39	#include "AliRun.h"
	40	#include "AliDetector.h"
	41	#include "AliHeader.h"
	42	#include "AliGenEventHeader.h"
53c30603	43	#include "AliGenHijingEventHeader.h"
359cdddc	44
	45	ClassImp(AliZDCMerger)
	46
1450a7cd	47	//int comp(const void i,const void j) {return (int )i - (int )j;}
1450a7cd	48
359cdddc	49	//____________________________________________________________________________
	50	AliZDCMerger::AliZDCMerger()
	51	{
	52	// Default constructor
1450a7cd	53	//fMerge = kDigitize -> Only digitization
	54	//fMerge = kMerge -> Digitization + Merging
	55	fMerge = kMerge;
359cdddc	56	fFnBgr = 0;
359cdddc	57	fBgrFile = 0;
1acbaaa8	58	fNEvBgr = 0;
2685bf00	59	fHitsBgr = 0;
359cdddc	60	fImpPar = 0;
	61	fSpecn = 0;
	62	fSpecp = 0;
	63	fFreeSpn = 0;
	64	fFreeSpp = 0;
	65	fFnSpecn = 0;
359cdddc	66	fSpecnFile = 0;
359cdddc	67	fFnSpecp = 0;
359cdddc	68	fSpecpFile = 0;
1acbaaa8	69	fNMhits = 0;
1450a7cd	70
359cdddc	71	}
	72
	73	//____________________________________________________________________________
	74	AliZDCMerger::~AliZDCMerger()
	75	{
fe107ada	76	// Destructor
359cdddc	77	if (fSpecnFile) delete fSpecnFile;
	78	if (fSpecpFile) delete fSpecpFile;
	79	}
	80
	81	//____________________________________________________________________________
	82	void AliZDCMerger::InitMerging()
	83	{
	84	// Hits tree, impact parameter, num. of spectators n & p
	85	// in background (full Hijing) event
	86	Float_t b;
1450a7cd	87	Background(b, fSpecn, fSpecp);
359cdddc	88
359cdddc	89	// Production of nuclear fragments -> num. of FREE spectators n & p
1450a7cd	90	Fragmentation(b, fSpecn, fSpecp, fFreeSpn, fFreeSpp);
359cdddc	91
	92	// Extract from spectators distribution the signal events:
	93	// NFreeSpectatorN spectator n & NFreeSpectatorP spectator p
1acbaaa8	94	Mixing();
359cdddc	95	}
	96
	97	//____________________________________________________________________________
	98	void AliZDCMerger::Background(Float_t &fImpPar, Int_t &fSpecn, Int_t &fSpecp)
	99	{
	100
	101	// --- Open the background file
fe107ada	102	if (fMerge && !fBgrFile) fBgrFile = OpenBgrFile();
359cdddc	103
	104	// --- Read from the TreeE impact parameter (b),
	105	// # of spectators n and p (fSpecn, fSpecp)
	106	fBgrFile->cd();
	107
1acbaaa8	108	// // Get AliRun object from file or create it if not on file
359cdddc	109	// gAlice = (AliRun*)fBgrFile->Get("gAlice");
1acbaaa8	110	// if (!gAlice) {
	111	// gAlice = (AliRun*)fBgrFile->Get("gAlice");
	112	// if (gAlice) printf("AliRun object found on file\n");
	113	// if (!gAlice) {
	114	// printf("\n create new gAlice object");
	115	// gAlice = new AliRun("gAlice","Alice test program");
	116	// }
	117	// }
1450a7cd	118
fe107ada	119	// gAlice->GetEvent(fNEvBgr); this is done in the steering macro
359cdddc	120	AliHeader *header = gAlice->GetHeader();
	121	AliGenEventHeader* mcHeader = header->GenEventHeader();
	122	fImpPar = ((AliGenHijingEventHeader*) mcHeader)->ImpactParameter();
1450a7cd	123	Int_t dSpecn = ((AliGenHijingEventHeader*) mcHeader)->Spectatorsn();
	124	Int_t dSpecp = ((AliGenHijingEventHeader*) mcHeader)->Spectatorsp();
	125	// Until there is only 1 ZDC set the # of spectators must be divided by 2!!!
	126	fSpecn = dSpecn/2;
	127	fSpecp = dSpecp/2;
	128	printf("\n HIJING ev. #%d - b = %f fm, Nspecn = %d, Nspecp = %d\n",
	129	fNEvBgr,fImpPar,fSpecn,fSpecp);
359cdddc	130	}
	131
	132	//____________________________________________________________________________
	133	TFile* AliZDCMerger::OpenBgrFile()
	134	{
	135	// Initialise background event
fe107ada	136	TFile file = (TFile)gROOT->GetListOfFiles()->FindObject(fFnBgr);
	137	if(!file)cerr<<"AliZDCMerger: background file "<<fFnBgr<<" not found\n";
	138	// TFile *file = new TFile(fFnBgr,"UPDATE");
1acbaaa8	139	printf("\n AliZDCMerger --- Background event -> %s file opened \n", fFnBgr);
359cdddc	140	fHitsBgr = new TClonesArray("AliZDCHit",1000);
1acbaaa8	141	fMHits = new TClonesArray("AliZDCMergedHit",1000);
359cdddc	142	return file;
	143	}
	144
	145	//____________________________________________________________________________
	146	void AliZDCMerger::Fragmentation(Float_t fImpPar, Int_t fSpecn, Int_t fSpecp,
	147	Int_t &fFreeSpn, Int_t &fFreeSpp)
	148	{
1450a7cd	149	//printf("\n Fragmentation -> fSpecn = %d, fSpecp = %d\n",fSpecn,fSpecp);
359cdddc	150	Int_t j, zz[100], nn[100], nAlpha, Ztot, Ntot;
	151	AliZDCFragment *frag = new AliZDCFragment(fImpPar);
	152	for(j=0; j<=99; j++){
	153	zz[j] =0;
	154	nn[j] =0;
	155	}
	156
	157	// Fragments generation
	158	frag->GenerateIMF(zz, nAlpha);
	159
	160	// Attach neutrons
	161	Ztot=0;
	162	Ntot=0;
	163	frag->AttachNeutrons(zz, nn, Ztot, Ntot);
1450a7cd	164	fFreeSpn = fSpecn-Ntot-2*nAlpha;
	165	fFreeSpp = fSpecp-Ztot-2*nAlpha;
	166	if(fFreeSpn<0) fFreeSpn=0;
	167	if(fFreeSpp<0) fFreeSpp=0;
	168	//printf("\n 2nAlpha = %d, Ztot = %d, Ntot = %d\n",2nAlpha, Ztot, Ntot);
1acbaaa8	169	printf("\n Fragmentation -> FreeSpn = %d, FreeSpp = %d\n",fFreeSpn,fFreeSpp);
359cdddc	170	}
	171
	172	//____________________________________________________________________________
1acbaaa8	173	void AliZDCMerger::Mixing()
359cdddc	174	{
1acbaaa8	175
1450a7cd	176	//printf("\n AliZDCMerger->Mixing\n");
1acbaaa8	177
359cdddc	178	// ### Background event Hits ###########################################
	179	fBgrFile->cd();
	180	// fBgrFile->ls();
	181
1acbaaa8	182	AliZDC ZDC = (AliZDC )gAlice->GetModule("ZDC");
359cdddc	183	// if(ZDC) printf("\n Ho trovato lo ZDC!\n");
359cdddc	184
1acbaaa8	185	// fNEvBgr = 0; // Let's suppose to have 1 full Hijing event per file
1acbaaa8	186	// Hits tree
359cdddc	187	char treeBgrName[20];
359cdddc	188	sprintf(treeBgrName,"TreeH%d",fNEvBgr);
1acbaaa8	189	fTrHBgr = (TTree*)gDirectory->Get(treeBgrName);
359cdddc	190	if(!fTrHBgr){
	191	printf("\n ERROR -> Can't find TreeH%d in background file\n",fNEvBgr);
	192	}
1acbaaa8	193	// fTrHBgr->Print();
	194
	195	// Branch address
	196	TBranch *branch;
	197	char branchname[20];
	198	sprintf(branchname,"%s",ZDC->GetName());
	199	if(fTrHBgr && fHitsBgr){
	200	// printf("\n fTrHBgr!=0 && fHitsBgr!=0\n");
	201	branch = fTrHBgr->GetBranch(branchname);
	202	if(branch) branch->SetAddress(&fHitsBgr);
	203	}
	204
359cdddc	205	Int_t ntracks = (Int_t) fTrHBgr->GetEntries();
1450a7cd	206	//printf("\n --- ntracks = %d\n\n", ntracks);
1acbaaa8	207
	208	Int_t itrack, nhits, ihit, j, sector[2];
	209	AliZDCHit* zdcHit;
	210	AliZDCMergedHit *MHit;
	211	Float_t MHits[7];
	212	TClonesArray &sdigits = *fMHits; // SDigits TCArray
	213	fNMhits = 0;
359cdddc	214
359cdddc	215	// --- Tracks loop
359cdddc	216	for(itrack=0; itrack<ntracks; itrack++){
359cdddc	217	fTrHBgr->GetEvent(itrack);
359cdddc	218
1acbaaa8	219	nhits = fHitsBgr->GetEntries();
1acbaaa8	220	// printf(" nhits = %d \n", nhits);
	221	for(ihit=0; ihit<nhits; ihit++){
	222	zdcHit = (AliZDCHit*) fHitsBgr->UncheckedAt(ihit);
	223
	224	for(j=0; j<2; j++) sector[j] = zdcHit->GetVolume(j);
	225	MHits[0] = zdcHit->GetPrimKinEn();
	226	MHits[1] = zdcHit->GetXImpact();
	227	MHits[2] = zdcHit->GetYImpact();
	228	MHits[3] = zdcHit->GetSFlag();
1450a7cd	229	MHits[4] = zdcHit->GetLightPMQ();
1acbaaa8	230	MHits[5] = zdcHit->GetLightPMC();
	231	MHits[6] = zdcHit->GetEnergy();
	232	MHit = new AliZDCMergedHit(sector, MHits);
	233	// MHit->Print("");
	234	new((fMHits)[fNMhits]) AliZDCMergedHit(MHit);
	235	new (sdigits[fNMhits]) AliZDCMergedHit(*MHit);
	236	delete MHit;
	237	fNMhits++;
	238	}//Hits loop
	239
359cdddc	240	} // Tracks loop
1450a7cd	241	//printf(" fNMhits (after bckg) = %d, \n",fNMhits);
1acbaaa8	242
	243	// ### Signal event Hits ###########################################
	244	// --- Neutrons
	245	ExtractSignal(1);
359cdddc	246
1acbaaa8	247	// --- Protons
1acbaaa8	248	ExtractSignal(2);
1450a7cd	249	//printf(" fNMhits (after signal) = %d \n",fNMhits);
359cdddc	250
	251	}
	252
	253	//____________________________________________________________________________
1acbaaa8	254	void AliZDCMerger::ExtractSignal(Int_t SpecType)
359cdddc	255	{
359cdddc	256
1acbaaa8	257	// printf("\n Entering in Extract Signal\n");
	258
	259	Int_t NumEvents = 0;
	260	if(SpecType == 1){ // --- Signal for spectator neutrons
	261	fFnSpecn = gSystem->ExpandPathName("$ALICE/$ALICE_LEVEL/ZDC/ZNsignalntu.root");
	262	fSpecnFile = TFile::Open(fFnSpecn,"R");
	263	fSpecnFile->cd();
1450a7cd	264	printf("\n --- ExtractSignal x n: file %s opened\n", fFnSpecn);
1acbaaa8	265	NumEvents = fFreeSpn;
	266	}
	267	else if(SpecType == 2){ // --- Signal for spectator protons
	268	fFnSpecp = gSystem->ExpandPathName("$ALICE/$ALICE_LEVEL/ZDC/ZPsignalntu.root");
	269	fSpecpFile = TFile::Open(fFnSpecp,"R");
	270	fSpecpFile->cd();
1450a7cd	271	printf("\n --- ExtractSignal x p: file %s opened\n", fFnSpecp);
1acbaaa8	272	NumEvents = fFreeSpp;
1acbaaa8	273	}
1450a7cd	274	//printf("\n # of free spectators = %d\n", NumEvents);
1450a7cd	275	//printf("\n fNMhits (before adding signal) = %d\n",fNMhits);
1acbaaa8	276
	277	TNtuple ZDCSignal = (TNtuple) gDirectory->Get("ZDCSignal");
	278	Int_t nentries = (Int_t) ZDCSignal->GetEntries();
1450a7cd	279	//printf("\n # entries = %d\n", nentries);
1acbaaa8	280
	281	AliZDCMergedHit *MHit;
	282	Float_t *entry, HitsSpec[7];
1450a7cd	283	Int_t pl, i, j, k, iev=0, rnd[125], Volume[2];
	284	for(pl=0;pl<125;pl++){
	285	rnd[pl] = 0;
	286	}
	287	for(pl=0;pl<NumEvents;pl++){
	288	rnd[pl] = (Int_t) (9999*gRandom->Rndm());
	289	if(rnd[pl] >= 9998) rnd[pl] = 9997;
	290	//printf(" rnd[%d] = %d\n",pl,rnd[pl]);
	291	}
	292	// Sorting vector in ascending order with C function QSORT
	293	qsort((void*)rnd,NumEvents,sizeof(Int_t),comp);
	294	//for(pl=0;pl<NumEvents;pl++){
	295	////printf(" rnd[%d] = %d\n",pl,rnd[pl]);
	296	//}
	297	do{
1acbaaa8	298	for(i=0; i<nentries; i++){
	299	ZDCSignal->GetEvent(i);
	300	entry = ZDCSignal->GetArgs();
1450a7cd	301	if(entry[0] == rnd[iev]){
1acbaaa8	302	for(k=0; k<2; k++) Volume[k] = (Int_t) entry[k+1];
	303	for(j=0; j<7; j++){
	304	HitsSpec[j] = entry[j+3];
	305	}
1450a7cd	306	//printf("\n i = %d, iev = %d, entry[0] = %f, rnd[%d] = %d ",i,iev,entry[0],iev,rnd[iev]);
1acbaaa8	307	MHit = new AliZDCMergedHit(Volume, HitsSpec);
	308	new((fMHits)[fNMhits++]) AliZDCMergedHit(MHit);
	309	delete MHit;
	310	}
1450a7cd	311	else if(entry[0] > rnd[iev]){
	312	iev++;
	313	continue;
	314	}
1acbaaa8	315	}
1450a7cd	316	}while(iev<NumEvents);
1acbaaa8	317
1acbaaa8	318	if(SpecType ==1){
1450a7cd	319	//printf("\n fNMhits (after n signal) = %d\n",fNMhits);
1acbaaa8	320	fSpecnFile->Close();
	321	}
	322	else if(SpecType == 2){
1450a7cd	323	//printf("\n fNMhits (after p signal) = %d\n",fNMhits);
1acbaaa8	324	fSpecpFile->Close();
	325	}
	326
359cdddc	327	}
	328
	329	//____________________________________________________________________________
1acbaaa8	330	void AliZDCMerger::Digitize(Int_t fNMhits, TClonesArray *fMHits)
359cdddc	331	{
359cdddc	332
1450a7cd	333	printf("\n AliZDCMerger->Digitize()");
359cdddc	334
	335	AliZDC ZDC = (AliZDC )gAlice->GetModule("ZDC");
	336	// if(ZDC) printf("\n Ho trovato lo ZDC!\n");
1acbaaa8	337	Int_t lightQ, lightC, sector[2], digit;
1acbaaa8	338	Int_t PMCZN = 0, PMCZP = 0, PMQZN[4], PMQZP[4], PMZEM1 = 0, PMZEM2 = 0;
359cdddc	339	Int_t i;
	340	for(i=0; i<4; i++){
	341	PMQZN[i] = 0;
	342	PMQZP[i] = 0;
	343	}
1acbaaa8	344
	345	AliZDCMergedHit *MHit;
	346	Int_t imhit;
1450a7cd	347	printf(" fNMHits = %d\n", fNMhits);
1acbaaa8	348	// Loop over SDigits
	349	for(imhit=0; imhit<fNMhits; imhit++){
	350
	351	MHit = (AliZDCMergedHit*) fMHits->UncheckedAt(imhit);
	352	sector[0] = MHit->GetSector(0);
	353	sector[1] = MHit->GetSector(1);
1450a7cd	354	// tmp -> c'erano i quadranti cannati!
1acbaaa8	355	if((sector[1]!=1) && (sector[1]!=2) && (sector[1]!=3) && (sector[1]!=4)){
1450a7cd	356	printf("\n *** ERROR!!! sector[0] = %d, sector[1] = %d\n",
1450a7cd	357	sector[0], sector[1]);
1acbaaa8	358	sector[1] = 0;
	359	}
	360	lightQ = Int_t(MHit->GetLightPMQ());
	361	lightC = Int_t(MHit->GetLightPMC());
1450a7cd	362	// printf(" imhit = %d -> DET. = %d, quad = %d,PMQ = %d, PMC = %d\n",
1acbaaa8	363	// imhit,sector[0], sector[1],lightQ, lightC);
	364
	365	if(sector[0] == 1){ //ZN
	366	PMCZN = PMCZN + lightC;
	367	PMQZN[sector[1]-1] = PMQZN[sector[1]-1] + lightQ;
	368	}
	369	else if(sector[0] == 2){ //ZP
	370	PMCZP = PMCZP + lightC;
	371	PMQZP[sector[1]-1] = PMQZP[sector[1]-1] + lightQ;
	372	}
	373	else if(sector[0] == 3){ //ZEM
	374	if(sector[1] ==1) PMZEM1 = PMZEM1 + lightC;
	375	else PMZEM2 = PMZEM2 + lightQ;
	376	}
	377	} // SDigits loop
359cdddc	378
1acbaaa8	379	// ### Digits creation ###############################################
359cdddc	380	// Create digits for ZN
	381	Int_t PedValue;
	382	sector[0] = 1; // Detector = ZN
	383	sector[1] = 0; // Common PM ADC
	384	digit = Phe2ADCch(1, 0, PMCZN);
1450a7cd	385	printf("\n\n ZN ### PMCZN = %d ADCZN = %d",PMCZN, digit);
1acbaaa8	386	PedValue = AddPedestal();
	387	digit += PedValue;
	388	// printf(" PedValue = %d",PedValue);
359cdddc	389	ZDC->AddDigit(sector, digit);
	390	Int_t j;
	391	for(j=0; j<4; j++){
	392	sector[1] = j+1; // Towers PM ADCs
	393	digit = Phe2ADCch(1, j+1, PMQZN[j]);
1450a7cd	394	printf("\n PMQZN[%d] = %d phe ADCZN[%d] = %d ADCch",j,PMQZN[j],j,digit);
359cdddc	395	PedValue = AddPedestal();
	396	digit += PedValue;
	397	// printf(" PedValue = %d",PedValue);
	398	ZDC->AddDigit(sector, digit);
	399	}
1450a7cd	400	printf("\n");
359cdddc	401
	402	// Create digits for ZP
	403	sector[0] = 2; // Detector = ZP
	404	sector[1] = 0; // Common PM ADC
	405	digit = Phe2ADCch(2, 0, PMCZP);
1450a7cd	406	printf("\n ZP --- PMCZP = %d phe ADCZP = %d ADCch",PMCZP,digit);
1acbaaa8	407	PedValue = AddPedestal();
1acbaaa8	408	digit += PedValue;
1450a7cd	409	// printf(" PedValue = %d",PedValue);
359cdddc	410	ZDC->AddDigit(sector, digit);
	411	for(j=0; j<4; j++){
	412	sector[1] = j+1; // Towers PM ADCs
	413	digit = Phe2ADCch(2, j+1, PMQZP[j]);
1450a7cd	414	printf("\n PMQZP[%d] = %d phe ADCZP[%d] = %d ADCch",j,PMQZP[j],j,digit);
359cdddc	415	PedValue = AddPedestal();
	416	digit += PedValue;
	417	// printf(" PedValue = %d",PedValue);
	418	ZDC->AddDigit(sector, digit);
	419	}
1450a7cd	420	printf("\n");
359cdddc	421
359cdddc	422	// Create digits for ZEM
1acbaaa8	423	sector[0] = 3;
	424	sector[1] = 1; // Detector = ZEM1
	425	digit = Phe2ADCch(3, 1, PMZEM1);
1450a7cd	426	printf("\n ZEM *** PMZEM1 = %d phe ADCZEM1 = %d ADCch",PMZEM1,digit);
1acbaaa8	427	PedValue = AddPedestal();
	428	digit += PedValue;
	429	// printf(" PedValue = %d\n",PedValue);
	430	ZDC->AddDigit(sector, digit);
	431	sector[1] = 2; // Detector = ZEM2
	432	digit = Phe2ADCch(3, 2, PMZEM2);
1450a7cd	433	printf("\n ZEM *** PMZEM2 = %d phe ADCZEM2 = %d ADCch\n",PMZEM2,digit);
1acbaaa8	434	PedValue = AddPedestal();
	435	digit += PedValue;
	436	// printf(" PedValue = %d\n",PedValue);
359cdddc	437	ZDC->AddDigit(sector, digit);
359cdddc	438	}
	439
	440	//_____________________________________________________________________________
	441	Int_t AliZDCMerger::Phe2ADCch(Int_t Det, Int_t Quad, Int_t Light)
	442	{
	443	// Evaluation of the ADC channel corresponding to the light yield Light
	444
	445	if(gAlice->GetDebug() > 0){
	446	// printf("\n Phe2ADCch -> Detector = %d, Quadrant = %d, Light = %d\n", Det, Quad, Light);
	447	}
	448
	449	Int_t ADCch = 0;
359cdddc	450
	451	Int_t j,i;
	452	for(i=0; i<3; i++){
	453	for(j=0; j<5; j++){
	454	fPMGain[i][j] = 100000.;
	455	}
	456	}
	457	fADCRes = 0.00000064; // ADC Resolution: 250 fC/ADCch
	458
	459	ADCch = (Int_t) (LightfPMGain[Det-1][Quad]fADCRes);
	460
	461	return ADCch;
	462	}
	463
	464	//_____________________________________________________________________________
	465	Int_t AliZDCMerger::AddPedestal()
	466	{
	467	// --- Pedestal value -> extracted from a gaussian distribution
	468	// obtained from the beam test on the ZEM prototype (Aug. 2000)
	469
	470	Int_t PedValue;
	471	Float_t PedMean = 50.;
1450a7cd	472	Float_t PedWidth = 5.;
359cdddc	473
	474	PedValue = (Int_t) gRandom->Gaus(PedMean,PedWidth);
	475
	476	return PedValue;
	477	}