[u/mrichter/AliRoot.git] / ZDC / AliZDCReconstructor.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.                  *
 **************************************************************************/

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
// class for ZDC reconstruction                                              //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////


#include <TF1.h>

#include "AliRunLoader.h"
#include "AliRawReader.h"
#include "AliESD.h"
#include "AliZDCDigit.h"
#include "AliZDCRawStream.h"
#include "AliZDCReco.h"
#include "AliZDCReconstructor.h"
#include "AliZDCCalibData.h"


ClassImp(AliZDCReconstructor)


//_____________________________________________________________________________
AliZDCReconstructor:: AliZDCReconstructor()
{
  // **** Default constructor
  
  //  ---      Number of generated spectator nucleons and impact parameter
  // --------------------------------------------------------------------------------------------------
  // [1] ### Results from a new production  -> 0<b<18 fm (Apr 2002)
  // Fit results for neutrons (Nspectator n true vs. EZN)
  fZNCen = new TF1("fZNCen",
      "(-2.287920+sqrt(2.287920*2.287920-4*(-0.007629)*(11.921710-x)))/(2*(-0.007629))",0.,164.);
  fZNPer = new TF1("fZNPer",
      "(-37.812280-sqrt(37.812280*37.812280-4*(-0.190932)*(-1709.249672-x)))/(2*(-0.190932))",0.,164.);
  // Fit results for protons (Nspectator p true vs. EZP)
  fZPCen = new TF1("fZPCen",
       "(-1.321353+sqrt(1.321353*1.321353-4*(-0.007283)*(3.550697-x)))/(2*(-0.007283))",0.,60.);
  fZPPer = new TF1("fZPPer",
      "(-42.643308-sqrt(42.643308*42.643308-4*(-0.310786)*(-1402.945615-x)))/(2*(-0.310786))",0.,60.);
  // Fit results for total number of spectators (Nspectators true vs. EZDC)
  fZDCCen = new TF1("fZDCCen",
      "(-1.934991+sqrt(1.934991*1.934991-4*(-0.004080)*(15.111124-x)))/(2*(-0.004080))",0.,225.);
  fZDCPer = new TF1("fZDCPer",
      "(-34.380639-sqrt(34.380639*34.380639-4*(-0.104251)*(-2612.189017-x)))/(2*(-0.104251))",0.,225.);
  // --------------------------------------------------------------------------------------------------
  // Fit results for b (b vs. EZDC)
  // [2] ### Results from a new production  -> 0<b<18 fm (Apr 2002)
  fbCen = new TF1("fbCen","-0.056923+0.079703*x-0.0004301*x*x+0.000001366*x*x*x",0.,220.);
  fbPer = new TF1("fbPer","17.943998-0.046846*x+0.000074*x*x",0.,220.);
  // --------------------------------------------------------------------------------------------------
  // Evaluating Nspectators and b from ZEM energy
  // [2] ### Results from a new production  -> 0<b<18 fm (Apr 2002)
  fZEMn  = new TF1("fZEMn","126.2-0.05399*x+0.000005679*x*x",0.,4000.);
  fZEMp  = new TF1("fZEMp","82.49-0.03611*x+0.00000385*x*x",0.,4000.);
  fZEMsp = new TF1("fZEMsp","208.7-0.09006*x+0.000009526*x*x",0.,4000.);
  fZEMb  = new TF1("fZEMb","16.06-0.01633*x+1.44e-5*x*x-6.778e-9*x*x*x+1.438e-12*x*x*x*x-1.112e-16*x*x*x*x*x",0.,4000.);
  
  // Get calibration data
  fCalibData = GetCalibData(); 
}

//_____________________________________________________________________________
AliZDCReconstructor::AliZDCReconstructor(const AliZDCReconstructor& 
                                         reconstructor):
  AliReconstructor(reconstructor)
{
// copy constructor

  Fatal("AliZDCReconstructor", "copy constructor not implemented");
}

//_____________________________________________________________________________
AliZDCReconstructor& AliZDCReconstructor::operator = 
  (const AliZDCReconstructor& /*reconstructor*/)
{
// assignment operator

  Fatal("operator =", "assignment operator not implemented");
  return *this;
}

//_____________________________________________________________________________
AliZDCReconstructor::~AliZDCReconstructor()
{
// destructor

  delete fZNCen;
  delete fZNPer;
  delete fZPCen;
  delete fZPPer;
  delete fZDCCen;
  delete fZDCPer;
  delete fbCen;
  delete fbPer;
  delete fZEMn;
  delete fZEMp;
  delete fZEMsp;
  delete fZEMb;
}


//_____________________________________________________________________________
void AliZDCReconstructor::Reconstruct(AliRunLoader* runLoader) const
{
  // *** Local ZDC reconstruction for digits
    
  Float_t meanPed[47];
  for(Int_t jj=0; jj<47; jj++) meanPed[jj] = fCalibData->GetMeanPed(jj);

  AliLoader* loader = runLoader->GetLoader("ZDCLoader");
  if (!loader) return;
  loader->LoadDigits("read");
  loader->LoadRecPoints("recreate");
  AliZDCDigit digit;
  AliZDCDigit* pdigit = &digit;

  // Event loop
  for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
    runLoader->GetEvent(iEvent);

    // load digits
    loader->LoadDigits();
    TTree* treeD = loader->TreeD();
    if (!treeD) continue;
    treeD->SetBranchAddress("ZDC", &pdigit);

    // loop over digits
    Float_t zn1corr=0, zp1corr=0, zn2corr=0, zp2corr=0, zemcorr=0;
    for (Int_t iDigit = 0; iDigit < treeD->GetEntries(); iDigit++) {
      treeD->GetEntry(iDigit);
      if (!pdigit) continue;

      if(digit.GetSector(0) == 1)
       	 zn1corr  += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)]);     // high gain ZN1 ADCs
      else if(digit.GetSector(0) == 2)
	 zp1corr  += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+10]);  // high gain ZP1 ADCs
      else if(digit.GetSector(0) == 3){
	 if(digit.GetSector(1)==1)      
	   zemcorr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+20]); // high gain ZEM1 ADCs
	 else if(digit.GetSector(1)==2) 
	   zemcorr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+22]); // high gain ZEM2 ADCs
      }
      else if(digit.GetSector(0) == 4)
       	 zn2corr  += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+24]);  // high gain ZN2 ADCs
      else if(digit.GetSector(0) == 5)
	 zp2corr  += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+34]);  // high gain ZP2 ADCs
    }
    if(zn1corr<0)  zn1corr=0;
    if(zp1corr<0)  zp1corr=0;
    if(zn2corr<0)  zn2corr=0;
    if(zp2corr<0)  zp2corr=0;
    if(zemcorr<0)  zemcorr=0;

    // reconstruct the event
    //printf("\n \t ZDCReco from digits-> Ev.#%d ZN = %.0f, ZP = %.0f, ZEM = %.0f\n",iEvent,zncorr,zpcorr,zemcorr);
    ReconstructEvent(loader, zn1corr, zp1corr, zemcorr, zn2corr, zp2corr);
  }

  loader->UnloadDigits();
  loader->UnloadRecPoints();
}

//_____________________________________________________________________________
void AliZDCReconstructor::Reconstruct(AliRunLoader* runLoader, 
                                      AliRawReader* rawReader) const
{
  // *** Local ZDC reconstruction for raw data
  
  Float_t meanPed[47];
  for(Int_t jj=0; jj<47; jj++) meanPed[jj] = fCalibData->GetMeanPed(jj);

  AliLoader* loader = runLoader->GetLoader("ZDCLoader");
  if (!loader) return;
  loader->LoadRecPoints("recreate");
  // Event loop
  Int_t iEvent = 0;
  while (rawReader->NextEvent()) {
    runLoader->GetEvent(iEvent++);

    // loop over raw data digits
    Float_t zn1corr=0, zp1corr=0,  zn2corr=0, zp2corr=0,zemcorr=0;
    AliZDCRawStream digit(rawReader);
    while (digit.Next()) {
      if(digit.IsADCDataWord()){
        if(digit.GetADCGain() == 0){
          if(digit.GetSector(0) == 1)	   
	    zn1corr  += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)]); // high gain ZN1 ADCs;
          else if(digit.GetSector(0) == 2) 
	    zp1corr  += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+10]); // high gain ZP1 ADCs;
          else if(digit.GetSector(0) == 3) 
	    if(digit.GetSector(1)==1)      
	      zemcorr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+20]); // high gain ZEM1 ADCs
	    else if(digit.GetSector(1)==2) 
	      zemcorr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+22]); // high gain ZEM2 ADCs
          else if(digit.GetSector(0) == 4)	   
	    zn2corr  += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+24]); // high gain ZN2 ADCs;
          else if(digit.GetSector(0) == 5) 
	    zp2corr  += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+34]); // high gain ZP2 ADCs;
	}
      }
    }
    if(zn1corr<0) zn1corr=0;
    if(zp1corr<0) zp1corr=0;
    if(zn2corr<0) zn2corr=0;
    if(zp2corr<0) zp2corr=0;
    if(zemcorr<0) zemcorr=0;
    
    // reconstruct the event
    //printf("\n\t ZDCReco from raw-> Ev.#%d ZN = %.0f, ZP = %.0f, ZEM = %.0f\n",iEvent,zncorr,zpcorr,zemcorr);
    ReconstructEvent(loader, zn1corr, zp1corr, zemcorr, zn2corr, zp2corr);
  }

  loader->UnloadRecPoints();
}

//_____________________________________________________________________________
void AliZDCReconstructor::ReconstructEvent(AliLoader* loader, Float_t zn1corr, 
	Float_t zp1corr, Float_t zemcorr, Float_t zn2corr, Float_t zp2corr) const
{
  // ***** Reconstruct one event
  
  //  ---      ADCchannel -> photoelectrons
  // NB-> PM gain = 10^(5), ADC resolution = 6.4*10^(-7)
  // Move to V965 (E.S.,15/09/04) NB-> PM gain = 10^(5), ADC resolution = 8*10^(-7)
  Float_t zn1phe, zp1phe, zemphe, zn2phe, zp2phe, convFactor = 0.08;
  zn1phe  = zn1corr/convFactor;
  zp1phe  = zp1corr/convFactor;
  zemphe = zemcorr/convFactor;
  zn2phe  = zn2corr/convFactor;
  zp2phe  = zp2corr/convFactor;
  //if AliDebug(1,Form("\n    znphe = %f, zpphe = %f, zemphe = %f\n",znphe, zpphe, zemphe);
  
  //  ---      Energy calibration
  // Conversion factors for hadronic ZDCs goes from phe yield to TRUE 
  // incident energy (conversion from GeV to TeV is included); while for EM 
  // calos conversion is from light yield to detected energy calculated by
  // GEANT NB -> ZN and ZP conversion factors are constant since incident
  // spectators have all the same energy, ZEM energy is obtained through a
  // fit over the whole range of incident particle energies 
  // (obtained with full HIJING simulations) 
  Float_t zn1energy, zp1energy, zemenergy, zdc1energy, zn2energy, zp2energy, zdc2energy;
  Float_t zn1phexTeV=329., zp1phexTeV=369., zn2phexTeV=329., zp2phexTeV=369.;
  zn1energy  = zn1phe/zn1phexTeV;
  zp1energy  = zp1phe/zp1phexTeV;
  zdc1energy = zn1energy+zp1energy;
  zn2energy  = zn2phe/zn2phexTeV;
  zp2energy  = zp2phe/zp2phexTeV;
  zdc2energy = zn2energy+zp2energy;
  zemenergy = -4.81+0.3238*zemphe;
  if(zemenergy<0) zemenergy=0;
  //  if AliDebug(1,Form("    znenergy = %f TeV, zpenergy = %f TeV, zdcenergy = %f GeV, "
  //			   "\n		zemenergy = %f TeV\n", znenergy, zpenergy, 
  //			   zdcenergy, zemenergy);
  //  if(zdcenergy==0)
  //    if AliDebug(1,Form("\n\n	###	ATTENZIONE!!! -> ev# %d: znenergy = %f TeV, zpenergy = %f TeV, zdcenergy = %f GeV, "
  //			     " zemenergy = %f TeV\n\n", fMerger->EvNum(), znenergy, zpenergy, zdcenergy, zemenergy); 

  //  ---      Number of detected spectator nucleons
  //  *** N.B. -> It works only in Pb-Pb
  Int_t nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight;
  nDetSpecNLeft = (Int_t) (zn1energy/2.760);
  nDetSpecPLeft = (Int_t) (zp1energy/2.760);
  nDetSpecNRight = (Int_t) (zn2energy/2.760);
  nDetSpecPRight = (Int_t) (zp2energy/2.760);

  //  ---      Number of generated spectator nucleons (from HIJING parameterization)
   //  *** N.B. -> Only one side!!!
 Int_t nGenSpecN=0, nGenSpecP=0, nGenSpec=0;
  Double_t impPar=0;
  // Cut value for Ezem (GeV)
  // ### Results from production  -> 0<b<18 fm (Apr 2002)
  Float_t eZEMCut = 420.;
  Float_t deltaEZEMSup = 690.; 
  Float_t deltaEZEMInf = 270.; 
  if(zemenergy > (eZEMCut+deltaEZEMSup)){
    nGenSpecN = (Int_t) (fZNCen->Eval(zn1energy));
    nGenSpecP = (Int_t) (fZPCen->Eval(zp1energy));
    nGenSpec  = (Int_t) (fZDCCen->Eval(zdc1energy));
    impPar    = fbCen->Eval(zdc1energy);
  }
  else if(zemenergy < (eZEMCut-deltaEZEMInf)){
    nGenSpecN = (Int_t) (fZNPer->Eval(zn1energy)); 
    nGenSpecP = (Int_t) (fZPPer->Eval(zp1energy));
    nGenSpec  = (Int_t) (fZDCPer->Eval(zdc1energy));
    impPar    = fbPer->Eval(zdc1energy);
  }
  else if(zemenergy >= (eZEMCut-deltaEZEMInf) && zemenergy <= (eZEMCut+deltaEZEMSup)){
    nGenSpecN = (Int_t) (fZEMn->Eval(zemenergy));
    nGenSpecP = (Int_t) (fZEMp->Eval(zemenergy));
    nGenSpec  = (Int_t)(fZEMsp->Eval(zemenergy));
    impPar    =  fZEMb->Eval(zemenergy);
  }
  // ### Results from production  -> 0<b<18 fm (Apr 2002)
  if(zn1energy>162.)  nGenSpecN = (Int_t) (fZEMn->Eval(zemenergy));
  if(zp1energy>59.75)  nGenSpecP = (Int_t) (fZEMp->Eval(zemenergy));
  if(zdc1energy>221.5) nGenSpec  = (Int_t)(fZEMsp->Eval(zemenergy));
  if(zdc1energy>220.)  impPar    =  fZEMb->Eval(zemenergy);
  
  if(nGenSpecN>125)    nGenSpecN=125;
  else if(nGenSpecN<0) nGenSpecN=0;
  if(nGenSpecP>82)     nGenSpecP=82;
  else if(nGenSpecP<0) nGenSpecP=0;
  if(nGenSpec>207)     nGenSpec=207;
  else if(nGenSpec<0)  nGenSpec=0;
  
  //  ---      Number of generated participants (from HIJING parameterization)
  Int_t nPart, nPartTot;
  nPart = 207-nGenSpecN-nGenSpecP;
  nPartTot = 207-nGenSpec;
  //printf("\t  ZDCeventReco-> ZNEn = %.0f GeV, ZPEn = %.0f GeV, ZEMEn = %.0f GeV\n",
  //	znenergy, zpenergy, zemenergy);

  // create the output tree
  loader->MakeTree("R");
  TTree* treeR = loader->TreeR();
  AliZDCReco reco(zn1energy, zp1energy, zdc1energy, zemenergy,
  		  zn2energy, zp2energy, zdc2energy, 
                  nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight,
		  nGenSpecN, nGenSpecP, nGenSpec,nPartTot, impPar);
  AliZDCReco* preco = &reco;
  const Int_t kBufferSize = 4000;
  treeR->Branch("ZDC", "AliZDCReco", &preco, kBufferSize);

  // write the output tree
  treeR->Fill();
  loader->WriteRecPoints("OVERWRITE");
}

//_____________________________________________________________________________
void AliZDCReconstructor::FillESD(AliRunLoader* runLoader, 
				  AliESD* esd) const
{
// fill energies and number of participants to the ESD

  AliLoader* loader = runLoader->GetLoader("ZDCLoader");
  if (!loader) return;
  loader->LoadRecPoints();

  TTree* treeR = loader->TreeR();
  if (!treeR) return;
  AliZDCReco reco;
  AliZDCReco* preco = &reco;
  treeR->SetBranchAddress("ZDC", &preco);

  treeR->GetEntry(0);
  esd->SetZDC(reco.GetZN1energy(), reco.GetZP1energy(), reco.GetZEMenergy(),
	      reco.GetZN2energy(), reco.GetZP2energy(), reco.GetNPart());

  loader->UnloadRecPoints();
}

//_____________________________________________________________________________
AliCDBStorage* AliZDCReconstructor::SetStorage(const char *uri) 
{
  //printf("\n\t AliZDCReconstructor::SetStorage \n");

  Bool_t deleteManager = kFALSE;
  
  AliCDBManager *manager = AliCDBManager::Instance();
  AliCDBStorage *defstorage = manager->GetDefaultStorage();
  
  if(!defstorage || !(defstorage->Contains("ZDC"))){ 
     AliWarning("No default storage set or default storage doesn't contain ZDC!");
     manager->SetDefaultStorage(uri);
     deleteManager = kTRUE;
  }
 
  AliCDBStorage *storage = manager->GetDefaultStorage();

  if(deleteManager){
    AliCDBManager::Instance()->UnsetDefaultStorage();
    defstorage = 0;   // the storage is killed by AliCDBManager::Instance()->Destroy()
  }

  return storage; 
}

//_____________________________________________________________________________
AliZDCCalibData* AliZDCReconstructor::GetCalibData() const
{

  // Getting calibration object for ZDC set

  AliCDBEntry  *entry = AliCDBManager::Instance()->Get("ZDC/Calib/Data");
  AliZDCCalibData *calibdata = (AliZDCCalibData*) entry->GetObject();

  if (!calibdata)  AliWarning("No calibration data from calibration database !");

  return calibdata;
}
Commit	Line	Data
8309c1ab	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	/* $Id$ */
	17
	18	///////////////////////////////////////////////////////////////////////////////
	19	// //
	20	// class for ZDC reconstruction //
	21	// //
	22	///////////////////////////////////////////////////////////////////////////////
	23
	24
	25	#include <TF1.h>
	26
	27	#include "AliRunLoader.h"
	28	#include "AliRawReader.h"
	29	#include "AliESD.h"
	30	#include "AliZDCDigit.h"
	31	#include "AliZDCRawStream.h"
	32	#include "AliZDCReco.h"
	33	#include "AliZDCReconstructor.h"
48642b09	34	#include "AliZDCCalibData.h"
8309c1ab	35
	36
	37	ClassImp(AliZDCReconstructor)
	38
	39
	40	//_____________________________________________________________________________
	41	AliZDCReconstructor:: AliZDCReconstructor()
	42	{
48642b09	43	// **** Default constructor
4fda3ba1	44
8309c1ab	45	// --- Number of generated spectator nucleons and impact parameter
8309c1ab	46	// --------------------------------------------------------------------------------------------------
8309c1ab	47	// [1] ### Results from a new production -> 0<b<18 fm (Apr 2002)
	48	// Fit results for neutrons (Nspectator n true vs. EZN)
	49	fZNCen = new TF1("fZNCen",
	50	"(-2.287920+sqrt(2.2879202.287920-4(-0.007629)(11.921710-x)))/(2(-0.007629))",0.,164.);
	51	fZNPer = new TF1("fZNPer",
	52	"(-37.812280-sqrt(37.81228037.812280-4(-0.190932)(-1709.249672-x)))/(2(-0.190932))",0.,164.);
	53	// Fit results for protons (Nspectator p true vs. EZP)
	54	fZPCen = new TF1("fZPCen",
	55	"(-1.321353+sqrt(1.3213531.321353-4(-0.007283)(3.550697-x)))/(2(-0.007283))",0.,60.);
	56	fZPPer = new TF1("fZPPer",
	57	"(-42.643308-sqrt(42.64330842.643308-4(-0.310786)(-1402.945615-x)))/(2(-0.310786))",0.,60.);
	58	// Fit results for total number of spectators (Nspectators true vs. EZDC)
	59	fZDCCen = new TF1("fZDCCen",
	60	"(-1.934991+sqrt(1.9349911.934991-4(-0.004080)(15.111124-x)))/(2(-0.004080))",0.,225.);
	61	fZDCPer = new TF1("fZDCPer",
	62	"(-34.380639-sqrt(34.38063934.380639-4(-0.104251)(-2612.189017-x)))/(2(-0.104251))",0.,225.);
	63	// --------------------------------------------------------------------------------------------------
48642b09	64	// Fit results for b (b vs. EZDC)
8309c1ab	65	// [2] ### Results from a new production -> 0<b<18 fm (Apr 2002)
	66	fbCen = new TF1("fbCen","-0.056923+0.079703x-0.0004301xx+0.000001366xxx",0.,220.);
	67	fbPer = new TF1("fbPer","17.943998-0.046846x+0.000074x*x",0.,220.);
	68	// --------------------------------------------------------------------------------------------------
	69	// Evaluating Nspectators and b from ZEM energy
8309c1ab	70	// [2] ### Results from a new production -> 0<b<18 fm (Apr 2002)
	71	fZEMn = new TF1("fZEMn","126.2-0.05399x+0.000005679x*x",0.,4000.);
	72	fZEMp = new TF1("fZEMp","82.49-0.03611x+0.00000385x*x",0.,4000.);
	73	fZEMsp = new TF1("fZEMsp","208.7-0.09006x+0.000009526x*x",0.,4000.);
	74	fZEMb = new TF1("fZEMb","16.06-0.01633x+1.44e-5xx-6.778e-9xxx+1.438e-12xxxx-1.112e-16xxxx*x",0.,4000.);
78d18275	75
78d18275	76	// Get calibration data
4fda3ba1	77	fCalibData = GetCalibData();
8309c1ab	78	}
	79
	80	//_____________________________________________________________________________
	81	AliZDCReconstructor::AliZDCReconstructor(const AliZDCReconstructor&
	82	reconstructor):
	83	AliReconstructor(reconstructor)
	84	{
	85	// copy constructor
	86
	87	Fatal("AliZDCReconstructor", "copy constructor not implemented");
	88	}
	89
	90	//_____________________________________________________________________________
	91	AliZDCReconstructor& AliZDCReconstructor::operator =
	92	(const AliZDCReconstructor& /reconstructor/)
	93	{
	94	// assignment operator
	95
	96	Fatal("operator =", "assignment operator not implemented");
	97	return *this;
	98	}
	99
	100	//_____________________________________________________________________________
	101	AliZDCReconstructor::~AliZDCReconstructor()
	102	{
	103	// destructor
	104
	105	delete fZNCen;
	106	delete fZNPer;
	107	delete fZPCen;
	108	delete fZPPer;
	109	delete fZDCCen;
	110	delete fZDCPer;
	111	delete fbCen;
	112	delete fbPer;
	113	delete fZEMn;
	114	delete fZEMp;
	115	delete fZEMsp;
	116	delete fZEMb;
	117	}
	118
	119
	120	//_____________________________________________________________________________
	121	void AliZDCReconstructor::Reconstruct(AliRunLoader* runLoader) const
	122	{
48642b09	123	// *** Local ZDC reconstruction for digits
78d18275	124
48642b09	125	Float_t meanPed[47];
78d18275	126	for(Int_t jj=0; jj<47; jj++) meanPed[jj] = fCalibData->GetMeanPed(jj);
8309c1ab	127
	128	AliLoader* loader = runLoader->GetLoader("ZDCLoader");
	129	if (!loader) return;
	130	loader->LoadDigits("read");
	131	loader->LoadRecPoints("recreate");
	132	AliZDCDigit digit;
	133	AliZDCDigit* pdigit = &digit;
	134
	135	// Event loop
	136	for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
	137	runLoader->GetEvent(iEvent);
	138
	139	// load digits
	140	loader->LoadDigits();
	141	TTree* treeD = loader->TreeD();
	142	if (!treeD) continue;
	143	treeD->SetBranchAddress("ZDC", &pdigit);
	144
	145	// loop over digits
980685f2	146	Float_t zn1corr=0, zp1corr=0, zn2corr=0, zp2corr=0, zemcorr=0;
8309c1ab	147	for (Int_t iDigit = 0; iDigit < treeD->GetEntries(); iDigit++) {
	148	treeD->GetEntry(iDigit);
	149	if (!pdigit) continue;
	150
48642b09	151	if(digit.GetSector(0) == 1)
980685f2	152	zn1corr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)]); // high gain ZN1 ADCs
48642b09	153	else if(digit.GetSector(0) == 2)
980685f2	154	zp1corr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+10]); // high gain ZP1 ADCs
48642b09	155	else if(digit.GetSector(0) == 3){
78d18275	156	if(digit.GetSector(1)==1)
980685f2	157	zemcorr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+20]); // high gain ZEM1 ADCs
78d18275	158	else if(digit.GetSector(1)==2)
980685f2	159	zemcorr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+22]); // high gain ZEM2 ADCs
48642b09	160	}
980685f2	161	else if(digit.GetSector(0) == 4)
	162	zn2corr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+24]); // high gain ZN2 ADCs
	163	else if(digit.GetSector(0) == 5)
	164	zp2corr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+34]); // high gain ZP2 ADCs
8309c1ab	165	}
980685f2	166	if(zn1corr<0) zn1corr=0;
	167	if(zp1corr<0) zp1corr=0;
	168	if(zn2corr<0) zn2corr=0;
	169	if(zp2corr<0) zp2corr=0;
	170	if(zemcorr<0) zemcorr=0;
8309c1ab	171
8309c1ab	172	// reconstruct the event
78d18275	173	//printf("\n \t ZDCReco from digits-> Ev.#%d ZN = %.0f, ZP = %.0f, ZEM = %.0f\n",iEvent,zncorr,zpcorr,zemcorr);
980685f2	174	ReconstructEvent(loader, zn1corr, zp1corr, zemcorr, zn2corr, zp2corr);
8309c1ab	175	}
	176
	177	loader->UnloadDigits();
	178	loader->UnloadRecPoints();
	179	}
	180
	181	//_____________________________________________________________________________
	182	void AliZDCReconstructor::Reconstruct(AliRunLoader* runLoader,
	183	AliRawReader* rawReader) const
	184	{
48642b09	185	// *** Local ZDC reconstruction for raw data
48642b09	186
48642b09	187	Float_t meanPed[47];
78d18275	188	for(Int_t jj=0; jj<47; jj++) meanPed[jj] = fCalibData->GetMeanPed(jj);
8309c1ab	189
	190	AliLoader* loader = runLoader->GetLoader("ZDCLoader");
	191	if (!loader) return;
	192	loader->LoadRecPoints("recreate");
	193	// Event loop
	194	Int_t iEvent = 0;
	195	while (rawReader->NextEvent()) {
	196	runLoader->GetEvent(iEvent++);
	197
	198	// loop over raw data digits
980685f2	199	Float_t zn1corr=0, zp1corr=0, zn2corr=0, zp2corr=0,zemcorr=0;
8309c1ab	200	AliZDCRawStream digit(rawReader);
	201	while (digit.Next()) {
	202	if(digit.IsADCDataWord()){
	203	if(digit.GetADCGain() == 0){
980685f2	204	if(digit.GetSector(0) == 1)
	205	zn1corr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)]); // high gain ZN1 ADCs;
	206	else if(digit.GetSector(0) == 2)
	207	zp1corr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+10]); // high gain ZP1 ADCs;
	208	else if(digit.GetSector(0) == 3)
	209	if(digit.GetSector(1)==1)
	210	zemcorr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+20]); // high gain ZEM1 ADCs
	211	else if(digit.GetSector(1)==2)
	212	zemcorr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+22]); // high gain ZEM2 ADCs
	213	else if(digit.GetSector(0) == 4)
	214	zn2corr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+24]); // high gain ZN2 ADCs;
	215	else if(digit.GetSector(0) == 5)
	216	zp2corr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+34]); // high gain ZP2 ADCs;
48642b09	217	}
8309c1ab	218	}
8309c1ab	219	}
980685f2	220	if(zn1corr<0) zn1corr=0;
	221	if(zp1corr<0) zp1corr=0;
	222	if(zn2corr<0) zn2corr=0;
	223	if(zp2corr<0) zp2corr=0;
48642b09	224	if(zemcorr<0) zemcorr=0;
48642b09	225
8309c1ab	226	// reconstruct the event
78d18275	227	//printf("\n\t ZDCReco from raw-> Ev.#%d ZN = %.0f, ZP = %.0f, ZEM = %.0f\n",iEvent,zncorr,zpcorr,zemcorr);
980685f2	228	ReconstructEvent(loader, zn1corr, zp1corr, zemcorr, zn2corr, zp2corr);
8309c1ab	229	}
	230
	231	loader->UnloadRecPoints();
	232	}
	233
	234	//_____________________________________________________________________________
980685f2	235	void AliZDCReconstructor::ReconstructEvent(AliLoader* loader, Float_t zn1corr,
980685f2	236	Float_t zp1corr, Float_t zemcorr, Float_t zn2corr, Float_t zp2corr) const
8309c1ab	237	{
48642b09	238	// ***** Reconstruct one event
8309c1ab	239
	240	// --- ADCchannel -> photoelectrons
	241	// NB-> PM gain = 10^(5), ADC resolution = 6.4*10^(-7)
	242	// Move to V965 (E.S.,15/09/04) NB-> PM gain = 10^(5), ADC resolution = 8*10^(-7)
980685f2	243	Float_t zn1phe, zp1phe, zemphe, zn2phe, zp2phe, convFactor = 0.08;
	244	zn1phe = zn1corr/convFactor;
	245	zp1phe = zp1corr/convFactor;
8309c1ab	246	zemphe = zemcorr/convFactor;
980685f2	247	zn2phe = zn2corr/convFactor;
980685f2	248	zp2phe = zp2corr/convFactor;
48642b09	249	//if AliDebug(1,Form("\n znphe = %f, zpphe = %f, zemphe = %f\n",znphe, zpphe, zemphe);
8309c1ab	250
	251	// --- Energy calibration
	252	// Conversion factors for hadronic ZDCs goes from phe yield to TRUE
	253	// incident energy (conversion from GeV to TeV is included); while for EM
	254	// calos conversion is from light yield to detected energy calculated by
	255	// GEANT NB -> ZN and ZP conversion factors are constant since incident
	256	// spectators have all the same energy, ZEM energy is obtained through a
	257	// fit over the whole range of incident particle energies
	258	// (obtained with full HIJING simulations)
980685f2	259	Float_t zn1energy, zp1energy, zemenergy, zdc1energy, zn2energy, zp2energy, zdc2energy;
	260	Float_t zn1phexTeV=329., zp1phexTeV=369., zn2phexTeV=329., zp2phexTeV=369.;
	261	zn1energy = zn1phe/zn1phexTeV;
	262	zp1energy = zp1phe/zp1phexTeV;
	263	zdc1energy = zn1energy+zp1energy;
	264	zn2energy = zn2phe/zn2phexTeV;
	265	zp2energy = zp2phe/zp2phexTeV;
	266	zdc2energy = zn2energy+zp2energy;
8309c1ab	267	zemenergy = -4.81+0.3238*zemphe;
8309c1ab	268	if(zemenergy<0) zemenergy=0;
48642b09	269	// if AliDebug(1,Form(" znenergy = %f TeV, zpenergy = %f TeV, zdcenergy = %f GeV, "
	270	// "\n zemenergy = %f TeV\n", znenergy, zpenergy,
	271	// zdcenergy, zemenergy);
48642b09	272	// if(zdcenergy==0)
	273	// if AliDebug(1,Form("\n\n ### ATTENZIONE!!! -> ev# %d: znenergy = %f TeV, zpenergy = %f TeV, zdcenergy = %f GeV, "
	274	// " zemenergy = %f TeV\n\n", fMerger->EvNum(), znenergy, zpenergy, zdcenergy, zemenergy);
8309c1ab	275
980685f2	276	// --- Number of detected spectator nucleons
	277	// *** N.B. -> It works only in Pb-Pb
	278	Int_t nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight;
	279	nDetSpecNLeft = (Int_t) (zn1energy/2.760);
	280	nDetSpecPLeft = (Int_t) (zp1energy/2.760);
	281	nDetSpecNRight = (Int_t) (zn2energy/2.760);
	282	nDetSpecPRight = (Int_t) (zp2energy/2.760);
	283
	284	// --- Number of generated spectator nucleons (from HIJING parameterization)
	285	// *** N.B. -> Only one side!!!
	286	Int_t nGenSpecN=0, nGenSpecP=0, nGenSpec=0;
8309c1ab	287	Double_t impPar=0;
8309c1ab	288	// Cut value for Ezem (GeV)
980685f2	289	// ### Results from production -> 0<b<18 fm (Apr 2002)
8309c1ab	290	Float_t eZEMCut = 420.;
	291	Float_t deltaEZEMSup = 690.;
	292	Float_t deltaEZEMInf = 270.;
	293	if(zemenergy > (eZEMCut+deltaEZEMSup)){
980685f2	294	nGenSpecN = (Int_t) (fZNCen->Eval(zn1energy));
	295	nGenSpecP = (Int_t) (fZPCen->Eval(zp1energy));
	296	nGenSpec = (Int_t) (fZDCCen->Eval(zdc1energy));
	297	impPar = fbCen->Eval(zdc1energy);
8309c1ab	298	}
8309c1ab	299	else if(zemenergy < (eZEMCut-deltaEZEMInf)){
980685f2	300	nGenSpecN = (Int_t) (fZNPer->Eval(zn1energy));
	301	nGenSpecP = (Int_t) (fZPPer->Eval(zp1energy));
	302	nGenSpec = (Int_t) (fZDCPer->Eval(zdc1energy));
	303	impPar = fbPer->Eval(zdc1energy);
8309c1ab	304	}
	305	else if(zemenergy >= (eZEMCut-deltaEZEMInf) && zemenergy <= (eZEMCut+deltaEZEMSup)){
	306	nGenSpecN = (Int_t) (fZEMn->Eval(zemenergy));
	307	nGenSpecP = (Int_t) (fZEMp->Eval(zemenergy));
	308	nGenSpec = (Int_t)(fZEMsp->Eval(zemenergy));
	309	impPar = fZEMb->Eval(zemenergy);
8309c1ab	310	}
980685f2	311	// ### Results from production -> 0<b<18 fm (Apr 2002)
	312	if(zn1energy>162.) nGenSpecN = (Int_t) (fZEMn->Eval(zemenergy));
	313	if(zp1energy>59.75) nGenSpecP = (Int_t) (fZEMp->Eval(zemenergy));
	314	if(zdc1energy>221.5) nGenSpec = (Int_t)(fZEMsp->Eval(zemenergy));
	315	if(zdc1energy>220.) impPar = fZEMb->Eval(zemenergy);
8309c1ab	316
	317	if(nGenSpecN>125) nGenSpecN=125;
	318	else if(nGenSpecN<0) nGenSpecN=0;
	319	if(nGenSpecP>82) nGenSpecP=82;
	320	else if(nGenSpecP<0) nGenSpecP=0;
	321	if(nGenSpec>207) nGenSpec=207;
	322	else if(nGenSpec<0) nGenSpec=0;
8309c1ab	323
980685f2	324	// --- Number of generated participants (from HIJING parameterization)
8309c1ab	325	Int_t nPart, nPartTot;
	326	nPart = 207-nGenSpecN-nGenSpecP;
	327	nPartTot = 207-nGenSpec;
78d18275	328	//printf("\t ZDCeventReco-> ZNEn = %.0f GeV, ZPEn = %.0f GeV, ZEMEn = %.0f GeV\n",
78d18275	329	// znenergy, zpenergy, zemenergy);
8309c1ab	330
	331	// create the output tree
	332	loader->MakeTree("R");
	333	TTree* treeR = loader->TreeR();
980685f2	334	AliZDCReco reco(zn1energy, zp1energy, zdc1energy, zemenergy,
	335	zn2energy, zp2energy, zdc2energy,
	336	nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight,
	337	nGenSpecN, nGenSpecP, nGenSpec,nPartTot, impPar);
8309c1ab	338	AliZDCReco* preco = &reco;
	339	const Int_t kBufferSize = 4000;
	340	treeR->Branch("ZDC", "AliZDCReco", &preco, kBufferSize);
	341
	342	// write the output tree
	343	treeR->Fill();
	344	loader->WriteRecPoints("OVERWRITE");
	345	}
	346
	347	//_____________________________________________________________________________
	348	void AliZDCReconstructor::FillESD(AliRunLoader* runLoader,
	349	AliESD* esd) const
	350	{
	351	// fill energies and number of participants to the ESD
	352
	353	AliLoader* loader = runLoader->GetLoader("ZDCLoader");
	354	if (!loader) return;
	355	loader->LoadRecPoints();
	356
	357	TTree* treeR = loader->TreeR();
	358	if (!treeR) return;
	359	AliZDCReco reco;
	360	AliZDCReco* preco = &reco;
	361	treeR->SetBranchAddress("ZDC", &preco);
	362
	363	treeR->GetEntry(0);
980685f2	364	esd->SetZDC(reco.GetZN1energy(), reco.GetZP1energy(), reco.GetZEMenergy(),
980685f2	365	reco.GetZN2energy(), reco.GetZP2energy(), reco.GetNPart());
8309c1ab	366
	367	loader->UnloadRecPoints();
	368	}
48642b09	369
48642b09	370	//_____________________________________________________________________________
78d18275	371	AliCDBStorage* AliZDCReconstructor::SetStorage(const char *uri)
48642b09	372	{
78d18275	373	//printf("\n\t AliZDCReconstructor::SetStorage \n");
48642b09	374
78d18275	375	Bool_t deleteManager = kFALSE;
48642b09	376
78d18275	377	AliCDBManager *manager = AliCDBManager::Instance();
78d18275	378	AliCDBStorage *defstorage = manager->GetDefaultStorage();
48642b09	379
78d18275	380	if(!defstorage \|\| !(defstorage->Contains("ZDC"))){
	381	AliWarning("No default storage set or default storage doesn't contain ZDC!");
	382	manager->SetDefaultStorage(uri);
	383	deleteManager = kTRUE;
	384	}
	385
	386	AliCDBStorage *storage = manager->GetDefaultStorage();
	387
	388	if(deleteManager){
	389	AliCDBManager::Instance()->UnsetDefaultStorage();
	390	defstorage = 0; // the storage is killed by AliCDBManager::Instance()->Destroy()
	391	}
	392
	393	return storage;
	394	}
48642b09	395
78d18275	396	//_____________________________________________________________________________
4fda3ba1	397	AliZDCCalibData* AliZDCReconstructor::GetCalibData() const
78d18275	398	{
48642b09	399
4fda3ba1	400	// Getting calibration object for ZDC set
	401
	402	AliCDBEntry *entry = AliCDBManager::Instance()->Get("ZDC/Calib/Data");
78d18275	403	AliZDCCalibData calibdata = (AliZDCCalibData) entry->GetObject();
4fda3ba1	404
78d18275	405	if (!calibdata) AliWarning("No calibration data from calibration database !");
48642b09	406
78d18275	407	return calibdata;
48642b09	408	}