X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ZDC%2FAliZDCReconstructor.cxx;h=ceeeb9c1877e57887f5e23eae4a9a3b824d5858f;hb=496aee550629be3d4e8973b04eb85208d470279b;hp=6bc66751bd3db57bee18083eec6168e2f9ea5890;hpb=659f58788005dfef851eed9da0dfb0db7270f08d;p=u%2Fmrichter%2FAliRoot.git diff --git a/ZDC/AliZDCReconstructor.cxx b/ZDC/AliZDCReconstructor.cxx index 6bc66751bd3..ceeeb9c1877 100644 --- a/ZDC/AliZDCReconstructor.cxx +++ b/ZDC/AliZDCReconstructor.cxx @@ -23,343 +23,675 @@ #include +#include #include "AliRunLoader.h" #include "AliRawReader.h" -#include "AliESD.h" +#include "AliGRPObject.h" +#include "AliESDEvent.h" +#include "AliESDZDC.h" #include "AliZDCDigit.h" #include "AliZDCRawStream.h" #include "AliZDCReco.h" #include "AliZDCReconstructor.h" -#include "AliZDCCalibData.h" +#include "AliZDCPedestals.h" +#include "AliZDCCalib.h" +#include "AliZDCRecoParam.h" +#include "AliZDCRecoParampp.h" +#include "AliZDCRecoParamPbPb.h" ClassImp(AliZDCReconstructor) - +AliZDCRecoParam *AliZDCReconstructor::fRecoParam=0; //reconstruction parameters //_____________________________________________________________________________ -AliZDCReconstructor:: AliZDCReconstructor() +AliZDCReconstructor:: AliZDCReconstructor() : + fPedData(GetPedData()), + fECalibData(GetECalibData()), + fRecoMode(0), + fBeamEnergy(0.) { // **** Default constructor - // if(!fStorage) fStorage = AliCDBManager::Instance()->GetStorage("local://DBlocal"); - - // --- Number of generated spectator nucleons and impact parameter - // -------------------------------------------------------------------------------------------------- - // [1] ### Results from a new production -> 0 0 0Get("GRP/GRP/Data"); + if(entry){ + TMap* m = dynamic_cast(entry->GetObject()); // old GRP entry + if(m){ + m->Print(); + grpData = new AliGRPObject(); + grpData->ReadValuesFromMap(m); + } + else{ + grpData = dynamic_cast(entry->GetObject()); // new GRP entry + entry->SetOwner(0); + } + AliCDBManager::Instance()->UnloadFromCache("GRP/GRP/Data"); + } + if(!grpData) AliError("No GRP entry found in OCDB!"); + + TString beamType = grpData->GetBeamType(); + if(beamType==AliGRPObject::GetInvalidString()){ + AliError("GRP/GRP/Data entry: missing value for the beam energy !"); + AliError("\t ZDC does not reconstruct event 4 UNKNOWN beam type\n"); + return; + } + // + if((beamType.CompareTo("p-p")) == 0){ + fRecoMode=0; + fRecoParam = (AliZDCRecoParampp*) AliZDCRecoParampp::GetppRecoParam(); + } + else if((beamType.CompareTo("A-A")) == 0){ + fRecoMode=1; + fRecoParam = (AliZDCRecoParamPbPb*) AliZDCRecoParamPbPb::GetPbPbRecoParam(); + } + + fBeamEnergy = grpData->GetBeamEnergy(); + if(fBeamEnergy==AliGRPObject::GetInvalidFloat()) { + AliError("GRP/GRP/Data entry: missing value for the beam energy ! Using 0"); + fBeamEnergy = 0.; + } + + printf("\n ***** ZDC reconstruction initialized for %s @ %1.3f GeV\n\n",beamType.Data(), fBeamEnergy); +} //_____________________________________________________________________________ -void AliZDCReconstructor::Reconstruct(AliRunLoader* runLoader) const +void AliZDCReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree) const { // *** Local ZDC reconstruction for digits - - // Get calibration data - int runNumber = 0; - AliZDCCalibData *calibda = GetCalibData(runNumber); - - Float_t meanPed[47]; - for(Int_t jj=0; jj<47; jj++) meanPed[jj] = calibda->GetMeanPed(jj); + // Works on the current event + + // Retrieving calibration data + Float_t meanPed[48]; + for(Int_t jj=0; jj<48; jj++) meanPed[jj] = fPedData->GetMeanPed(jj); - AliLoader* loader = runLoader->GetLoader("ZDCLoader"); - if (!loader) return; - loader->LoadDigits("read"); - loader->LoadRecPoints("recreate"); + // get digits 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 zncorr=0, zpcorr=0, zemcorr=0; - for (Int_t iDigit = 0; iDigit < treeD->GetEntries(); iDigit++) { - treeD->GetEntry(iDigit); - if (!pdigit) continue; - - if(digit.GetSector(0) == 1) - zncorr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)]); // ped 4 high gain ZN ADCs - else if(digit.GetSector(0) == 2) - zpcorr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+10]); // ped 4 high gain ZP ADCs - else if(digit.GetSector(0) == 3){ - if(digit.GetSector(1)==1) zemcorr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+20]); // ped 4 high gain ZEM1 ADCs - else if(digit.GetSector(1)==2) zemcorr += (Float_t) (digit.GetADCValue(0)-meanPed[digit.GetSector(1)+22]); // ped 4 high gain ZEM2 ADCs - } + digitsTree->SetBranchAddress("ZDC", &pdigit); + //printf("\n\t # of digits in tree: %d\n",(Int_t) digitsTree->GetEntries()); + + // loop over digits + Float_t tZN1Corr[10], tZP1Corr[10], tZN2Corr[10], tZP2Corr[10]; + Float_t dZEM1Corr[2], dZEM2Corr[2], PMRef1[2], PMRef2[2]; + for(Int_t i=0; i<10; i++){ + tZN1Corr[i] = tZP1Corr[i] = tZN2Corr[i] = tZP2Corr[i] = 0.; + if(i<2) dZEM1Corr[i] = dZEM2Corr[i] = PMRef1[i] = PMRef2[i] = 0.; + } + // + for (Int_t iDigit = 0; iDigit < (digitsTree->GetEntries()/2); iDigit++) { + digitsTree->GetEntry(iDigit); + if (!pdigit) continue; + // + Int_t det = digit.GetSector(0); + Int_t quad = digit.GetSector(1); + Int_t pedindex = -1, kNch = 24; + //printf("\n\t Digit #%d det %d quad %d", iDigit, det, quad); + // + if(quad != 5){ // ZDC (not reference PTMs!) + if(det == 1){ // *** ZNC + pedindex = quad; + tZN1Corr[quad] = (Float_t) (digit.GetADCValue(0)-meanPed[pedindex]); + if(tZN1Corr[quad]<0.) tZN1Corr[quad] = 0.; + tZN1Corr[quad+5] = (Float_t) (digit.GetADCValue(1)-meanPed[pedindex+kNch]); + if(tZN1Corr[quad+5]<0.) tZN1Corr[quad+5] = 0.; + //printf("\t pedindex %d tZN1Corr[%d] = %1.0f tZN1Corr[%d] = %1.0f", + // pedindex, quad, tZN1Corr[quad], quad+5, tZN1Corr[quad+5]); } - if(zncorr<0) zncorr=0; - if(zpcorr<0) zpcorr=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, zncorr, zpcorr, zemcorr); + else if(det == 2){ // *** ZP1 + pedindex = quad+5; + tZP1Corr[quad] = (Float_t) (digit.GetADCValue(0)-meanPed[pedindex]); + if(tZP1Corr[quad]<0.) tZP1Corr[quad] = 0.; + tZP1Corr[quad+5] = (Float_t) (digit.GetADCValue(1)-meanPed[pedindex+kNch]); + if(tZP1Corr[quad+5]<0.) tZP1Corr[quad+5] = 0.; + //printf("\t pedindex %d tZP1Corr[%d] = %1.0f tZP1Corr[%d] = %1.0f", + // pedindex, quad, tZP1Corr[quad], quad+5, tZP1Corr[quad+5]); + } + else if(det == 3){ + pedindex = quad+9; + if(quad == 1){ // *** ZEM1 + dZEM1Corr[0] += (Float_t) (digit.GetADCValue(0)-meanPed[pedindex]); + if(dZEM1Corr[0]<0.) dZEM1Corr[0] = 0.; + dZEM1Corr[1] += (Float_t) (digit.GetADCValue(1)-meanPed[pedindex+kNch]); + if(dZEM1Corr[1]<0.) dZEM1Corr[1] = 0.; + //printf("\t pedindex %d tZEM1Corr[%d] = %1.0f tZEM1Corr[%d] = %1.0f", + // pedindex, quad, tZEM1Corr[quad], quad+1, tZEM1Corr[quad+1]); + } + else if(quad == 2){ // *** ZEM2 + dZEM2Corr[0] += (Float_t) (digit.GetADCValue(0)-meanPed[pedindex]); + if(dZEM2Corr[0]<0.) dZEM2Corr[0] = 0.; + dZEM2Corr[1] += (Float_t) (digit.GetADCValue(1)-meanPed[pedindex+kNch]); + if(dZEM2Corr[1]<0.) dZEM2Corr[1] = 0.; + //printf("\t pedindex %d tZEM2Corr[%d] = %1.0f tZEM2Corr[%d] = %1.0f", + // pedindex, quad, tZEM2Corr[quad], quad+1, tZEM2Corr[quad+1]); + } + } + else if(det == 4){ // *** ZN2 + pedindex = quad+12; + tZN2Corr[quad] = (Float_t) (digit.GetADCValue(0)-meanPed[pedindex]); + if(tZN2Corr[quad]<0.) tZN2Corr[quad] = 0.; + tZN2Corr[quad+5] = (Float_t) (digit.GetADCValue(1)-meanPed[pedindex+kNch]); + if(tZN2Corr[quad+5]<0.) tZN2Corr[quad+5] = 0.; + //printf("\t pedindex %d tZN2Corr[%d] = %1.0f tZN2Corr[%d] = %1.0f\n", + // pedindex, quad, tZN2Corr[quad], quad+5, tZN2Corr[quad+5]); + } + else if(det == 5){ // *** ZP2 + pedindex = quad+17; + tZP2Corr[quad] = (Float_t) (digit.GetADCValue(0)-meanPed[pedindex]); + if(tZP2Corr[quad]<0.) tZP2Corr[quad] = 0.; + tZP2Corr[quad+5] = (Float_t) (digit.GetADCValue(1)-meanPed[pedindex+kNch]); + if(tZP2Corr[quad+5]<0.) tZP2Corr[quad+5] = 0.; + //printf("\t pedindex %d tZP2Corr[%d] = %1.0f tZP2Corr[%d] = %1.0f\n", + // pedindex, quad, tZP2Corr[quad], quad+5, tZP2Corr[quad+5]); + } + } + else{ // Reference PMs + pedindex = (det-1)/3+22; + if(det == 1){ + PMRef1[0] = (Float_t) (digit.GetADCValue(0)-meanPed[pedindex]); + if(PMRef1[0]<0.) PMRef1[0] = 0.; + PMRef1[1] = (Float_t) (digit.GetADCValue(1)-meanPed[pedindex+kNch]); + if(PMRef2[1]<0.) PMRef1[1] = 0.; + } + else if(det == 4){ + PMRef2[0] = (Float_t) (digit.GetADCValue(0)-meanPed[pedindex]); + if(PMRef2[0]<0.) PMRef2[0] = 0.; + PMRef2[1] = (Float_t) (digit.GetADCValue(1)-meanPed[pedindex+kNch]); + if(PMRef2[1]<0.) PMRef2[1] = 0.; + } + } } - loader->UnloadDigits(); - loader->UnloadRecPoints(); + // reconstruct the event + if(fRecoMode==0) + ReconstructEventpp(clustersTree, tZN1Corr, tZP1Corr, tZN2Corr, tZP2Corr, + dZEM1Corr, dZEM2Corr, PMRef1, PMRef2); + else if(fRecoMode==1) + ReconstructEventPbPb(clustersTree, tZN1Corr, tZP1Corr, tZN2Corr, tZP2Corr, + dZEM1Corr, dZEM2Corr, PMRef1, PMRef2); + } //_____________________________________________________________________________ -void AliZDCReconstructor::Reconstruct(AliRunLoader* runLoader, - AliRawReader* rawReader) const +void AliZDCReconstructor::Reconstruct(AliRawReader* rawReader, TTree* clustersTree) const { - // *** Local ZDC reconstruction for raw data + // *** ZDC raw data reconstruction + // Works on the current event - // Calibration data - int runNumber = 0; - AliZDCCalibData *calibda = GetCalibData(runNumber); + // Retrieving calibration data + Float_t meanPed[48]; + for(Int_t jj=0; jj<48; jj++) meanPed[jj] = fPedData->GetMeanPed(jj); + + rawReader->Reset(); - Float_t meanPed[47]; - for(Int_t jj=0; jj<47; jj++) meanPed[jj] = calibda->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 zncorr=0, zpcorr=0, zemcorr=0; - AliZDCRawStream digit(rawReader); - while (digit.Next()) { - if(digit.IsADCDataWord()){ - if(digit.GetADCGain() == 0){ - if(digit.GetSector(0) == 1) zncorr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)]); // pedestals for high gain ZN ADCs; - else if(digit.GetSector(0) == 2) zpcorr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+10]); // pedestals for high gain ZP ADCs; - else if(digit.GetSector(0) == 3) zemcorr += (Float_t) (digit.GetADCValue()-meanPed[digit.GetSector(1)+20]); // pedestals for high gain ZEM ADCs; - } + // loop over raw data + Float_t tZN1Corr[10], tZP1Corr[10], tZN2Corr[10], tZP2Corr[10]; + Float_t dZEM1Corr[2], dZEM2Corr[2], PMRef1[2], PMRef2[2]; + for(Int_t i=0; i<10; i++){ + tZN1Corr[i] = tZP1Corr[i] = tZN2Corr[i] = tZP2Corr[i] = 0.; + if(i<2) dZEM1Corr[i] = dZEM2Corr[i] = PMRef1[i] = PMRef2[i] = 0.; + } + // + AliZDCRawStream rawData(rawReader); + Int_t kNch = 24; + while (rawData.Next()) { + if(rawData.IsADCDataWord()){ + Int_t det = rawData.GetSector(0); + Int_t quad = rawData.GetSector(1); + Int_t gain = rawData.GetADCGain(); + Int_t pedindex=0; + // + if(quad !=5){ // ZDCs (not reference PTMs) + if(det == 1){ + pedindex = quad; + if(gain == 0) tZN1Corr[quad] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + else tZN1Corr[quad+5] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex+kNch]); } - } - if(zncorr<0) zncorr=0; - if(zpcorr<0) zpcorr=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, zncorr, zpcorr, zemcorr); + else if(det == 2){ + pedindex = quad+5; + if(gain == 0) tZP1Corr[quad] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + else tZP1Corr[quad+5] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex+kNch]); + } + else if(det == 3){ + pedindex = quad+9; + if(quad==1){ + if(gain == 0) dZEM1Corr[0] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + else dZEM1Corr[1] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex+kNch]); + } + else if(quad==2){ + if(gain == 0) dZEM2Corr[0] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + else dZEM2Corr[1] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex+kNch]); + } + } + else if(det == 4){ + pedindex = quad+12; + if(gain == 0) tZN2Corr[quad] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + else tZN2Corr[quad+5] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex+kNch]); + } + else if(det == 5){ + pedindex = quad+17; + if(gain == 0) tZP2Corr[quad] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + else tZP2Corr[quad+5] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex+kNch]); + } + //printf("\t AliZDCReconstructor - det %d quad %d res %d -> Ped[%d] = %1.0f\n", + // det,quad,gain, pedindex, meanPed[pedindex]); + } + else{ // reference PM + pedindex = (det-1)/3 + 22; + if(det == 1){ + if(gain==0) PMRef1[0] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + else PMRef1[1] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + } + else if(det ==4){ + if(gain==0) PMRef2[0] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + else PMRef2[1] += (Float_t) (rawData.GetADCValue()-meanPed[pedindex]); + } + } + }//IsADCDataWord } + + // reconstruct the event + if(fRecoMode==0) + ReconstructEventpp(clustersTree, tZN1Corr, tZP1Corr, tZN2Corr, tZP2Corr, + dZEM1Corr, dZEM2Corr, PMRef1, PMRef2); + else if(fRecoMode==1) + ReconstructEventPbPb(clustersTree, tZN1Corr, tZP1Corr, tZN2Corr, tZP2Corr, + dZEM1Corr, dZEM2Corr, PMRef1, PMRef2); - loader->UnloadRecPoints(); } //_____________________________________________________________________________ -void AliZDCReconstructor::ReconstructEvent(AliLoader* loader, Float_t zncorr, - Float_t zpcorr, Float_t zemcorr) const +void AliZDCReconstructor::ReconstructEventpp(TTree *clustersTree, Float_t* ZN1ADCCorr, + Float_t* ZP1ADCCorr, Float_t* ZN2ADCCorr, Float_t* ZP2ADCCorr, + Float_t* ZEM1ADCCorr, Float_t* ZEM2ADCCorr, Float_t* PMRef1, Float_t* PMRef2) 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 znphe, zpphe, zemphe, convFactor = 0.08; - znphe = zncorr/convFactor; - zpphe = zpcorr/convFactor; - zemphe = zemcorr/convFactor; - //if AliDebug(1,Form("\n znphe = %f, zpphe = %f, zemphe = %f\n",znphe, zpphe, zemphe); + // *** RECONSTRUCTION FROM "REAL" DATA + // + // Retrieving calibration data + // --- Equalization coefficients --------------------------------------------- + Float_t equalCoeffZN1[5], equalCoeffZP1[5], equalCoeffZN2[5], equalCoeffZP2[5]; + for(Int_t ji=0; ji<5; ji++){ + equalCoeffZN1[ji] = fECalibData->GetZN1EqualCoeff(ji); + equalCoeffZP1[ji] = fECalibData->GetZP1EqualCoeff(ji); + equalCoeffZN2[ji] = fECalibData->GetZN2EqualCoeff(ji); + equalCoeffZP2[ji] = fECalibData->GetZP2EqualCoeff(ji); + } + // --- Energy calibration factors ------------------------------------ + Float_t calibEne[4]; + // ********************************************************************* + // **** Until the beam type info isn't known @ reconstruction level **** + // **** the energy calibration coefficient are manually set to 1 **** + // **** as it will be in real life for pp data taking **** + // ********************************************************************* + //for(Int_t ij=0; ij<4; ij++) calibEne[ij] = fECalibData->GetEnCalib(ij); + for(Int_t ij=0; ij<4; ij++) calibEne[ij] = 1.; - // --- 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 znenergy, zpenergy, zemenergy, zdcenergy; - Float_t znphexTeV=329., zpphexTeV=369.; - znenergy = znphe/znphexTeV; - zpenergy = zpphe/zpphexTeV; - zdcenergy = znenergy+zpenergy; - 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); + // Equalization of detector responses + Float_t equalTowZN1[10], equalTowZN2[10], equalTowZP1[10], equalTowZP2[10]; + for(Int_t gi=0; gi<5; gi++){ + equalTowZN1[gi] = ZN1ADCCorr[gi]*equalCoeffZN1[gi]; + equalTowZN1[gi+5] = ZN1ADCCorr[gi+5]*equalCoeffZN1[gi]; + equalTowZP1[gi] = ZP1ADCCorr[gi]*equalCoeffZP1[gi]; + equalTowZP1[gi+5] = ZP1ADCCorr[gi+5]*equalCoeffZP1[gi]; + equalTowZN2[gi] = ZN2ADCCorr[gi]*equalCoeffZN2[gi]; + equalTowZN2[gi+5] = ZN2ADCCorr[gi+5]*equalCoeffZN2[gi]; + equalTowZP2[gi] = ZP2ADCCorr[gi]*equalCoeffZP2[gi]; + equalTowZP2[gi+5] = ZP2ADCCorr[gi+5]*equalCoeffZP2[gi]; + } - // 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 incident spectator nucleons - Int_t nDetSpecN, nDetSpecP; - nDetSpecN = (Int_t) (znenergy/2.760); - nDetSpecP = (Int_t) (zpenergy/2.760); -// if AliDebug(1,Form("\n nDetSpecN = %d, nDetSpecP = %d\n",nDetSpecN, nDetSpecP); - - Int_t nGenSpecN=0, nGenSpecP=0, nGenSpec=0; - Double_t impPar=0; - // Cut value for Ezem (GeV) - // [2] ### Results from a new production -> 0 (eZEMCut+deltaEZEMSup)){ - nGenSpecN = (Int_t) (fZNCen->Eval(znenergy)); - nGenSpecP = (Int_t) (fZPCen->Eval(zpenergy)); - nGenSpec = (Int_t) (fZDCCen->Eval(zdcenergy)); - impPar = fbCen->Eval(zdcenergy); + // Energy calibration of detector responses + Float_t calibTowZN1[10], calibTowZN2[10], calibTowZP1[10], calibTowZP2[10]; + Float_t calibSumZN1[]={0,0}, calibSumZN2[]={0,0}, calibSumZP1[]={0,0}, calibSumZP2[]={0,0}; + for(Int_t gi=0; gi<10; gi++){ + calibTowZN1[gi] = equalTowZN1[gi]*calibEne[0]; + calibTowZP1[gi] = equalTowZP1[gi]*calibEne[1]; + calibTowZN2[gi] = equalTowZN2[gi]*calibEne[2]; + calibTowZP2[gi] = equalTowZP2[gi]*calibEne[3]; + // + if(gi<5){ + calibSumZN1[0] += calibTowZN1[gi]; + calibSumZP1[0] += calibTowZP1[gi]; + calibSumZN2[0] += calibTowZN2[gi]; + calibSumZP2[0] += calibTowZP2[gi]; + } + // + else{ + calibSumZN1[1] += calibTowZN1[gi]; + calibSumZP1[1] += calibTowZP1[gi]; + calibSumZN2[1] += calibTowZN2[gi]; + calibSumZP2[1] += calibTowZP2[gi]; + } } - else if(zemenergy < (eZEMCut-deltaEZEMInf)){ - nGenSpecN = (Int_t) (fZNPer->Eval(znenergy)); - nGenSpecP = (Int_t) (fZPPer->Eval(zpenergy)); - nGenSpec = (Int_t) (fZDCPer->Eval(zdcenergy)); - impPar = fbPer->Eval(zdcenergy); + + // --- Number of detected spectator nucleons + // *** N.B. -> It works only in Pb-Pb + Int_t nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight; + if(fBeamEnergy!=0){ + nDetSpecNLeft = (Int_t) (calibSumZN1[0]/fBeamEnergy); + nDetSpecPLeft = (Int_t) (calibSumZP1[0]/fBeamEnergy); + nDetSpecNRight = (Int_t) (calibSumZN2[0]/fBeamEnergy); + nDetSpecPRight = (Int_t) (calibSumZP2[0]/fBeamEnergy); } - 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); + else AliWarning(" ATTENTION -> fBeamEnergy = 0\n"); + /*printf("\n\t AliZDCReconstructor -> nDetSpecNLeft %d, nDetSpecPLeft %d," + " nDetSpecNRight %d, nDetSpecPRight %d\n",nDetSpecNLeft, nDetSpecPLeft, + nDetSpecNRight, nDetSpecPRight);*/ + + // --- Number of generated spectator nucleons (from HIJING parameterization) + Int_t nGenSpecNLeft=0, nGenSpecPLeft=0, nGenSpecLeft=0; + Int_t nGenSpecNRight=0, nGenSpecPRight=0, nGenSpecRight=0; + Int_t nPartTotLeft=0, nPartTotRight=0; + Double_t impPar=0.; + + // create the output tree + AliZDCReco reco(calibSumZN1, calibSumZP1, calibSumZN2, calibSumZP2, + calibTowZN1, calibTowZP1, calibTowZN2, calibTowZP2, + ZEM1ADCCorr, ZEM2ADCCorr, PMRef1, PMRef2, + nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight, + nGenSpecNLeft, nGenSpecPLeft, nGenSpecLeft, nGenSpecNRight, + nGenSpecPRight, nGenSpecRight, nPartTotLeft, nPartTotRight, impPar); + + AliZDCReco* preco = &reco; + const Int_t kBufferSize = 4000; + clustersTree->Branch("ZDC", "AliZDCReco", &preco, kBufferSize); + + // write the output tree + clustersTree->Fill(); +} + +//_____________________________________________________________________________ +void AliZDCReconstructor::ReconstructEventPbPb(TTree *clustersTree, Float_t* ZN1ADCCorr, + Float_t* ZP1ADCCorr, Float_t* ZN2ADCCorr, Float_t* ZP2ADCCorr, + Float_t* ZEM1ADCCorr, Float_t* ZEM2ADCCorr, Float_t* PMRef1, Float_t* PMRef2) const +{ + // ***** Reconstruct one event + + // *** RECONSTRUCTION FROM "REAL" DATA + // + // Retrieving calibration data + // --- Equalization coefficients --------------------------------------------- + Float_t equalCoeffZN1[5], equalCoeffZP1[5], equalCoeffZN2[5], equalCoeffZP2[5]; + for(Int_t ji=0; ji<5; ji++){ + equalCoeffZN1[ji] = fECalibData->GetZN1EqualCoeff(ji); + equalCoeffZP1[ji] = fECalibData->GetZP1EqualCoeff(ji); + equalCoeffZN2[ji] = fECalibData->GetZN2EqualCoeff(ji); + equalCoeffZP2[ji] = fECalibData->GetZP2EqualCoeff(ji); } - // [2] ### Results from a new production -> 0162.) nGenSpecN = (Int_t) (fZEMn->Eval(zemenergy)); - if(zpenergy>59.75) nGenSpecP = (Int_t) (fZEMp->Eval(zemenergy)); - if(zdcenergy>221.5) nGenSpec = (Int_t)(fZEMsp->Eval(zemenergy)); - if(zdcenergy>220.) impPar = fZEMb->Eval(zemenergy); + // --- Energy calibration factors ------------------------------------ + Float_t calibEne[4]; + for(Int_t ij=0; ij<4; ij++) calibEne[ij] = fECalibData->GetEnCalib(ij); - 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; + // Equalization of detector responses + Float_t equalTowZN1[10], equalTowZN2[10], equalTowZP1[10], equalTowZP2[10]; + for(Int_t gi=0; gi<5; gi++){ + equalTowZN1[gi] = ZN1ADCCorr[gi]*equalCoeffZN1[gi]; + equalTowZN1[gi+5] = ZN1ADCCorr[gi+5]*equalCoeffZN1[gi]; + equalTowZP1[gi] = ZP1ADCCorr[gi]*equalCoeffZP1[gi]; + equalTowZP1[gi+5] = ZP1ADCCorr[gi+5]*equalCoeffZP1[gi]; + equalTowZN2[gi] = ZN2ADCCorr[gi]*equalCoeffZN2[gi]; + equalTowZN2[gi+5] = ZN2ADCCorr[gi+5]*equalCoeffZN2[gi]; + equalTowZP2[gi] = ZP2ADCCorr[gi]*equalCoeffZP2[gi]; + equalTowZP2[gi+5] = ZP2ADCCorr[gi+5]*equalCoeffZP2[gi]; + } - // --- Number of participants - 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); + // Energy calibration of detector responses + Float_t calibTowZN1[10], calibTowZN2[10], calibTowZP1[10], calibTowZP2[10]; + Float_t calibSumZN1[]={0,0}, calibSumZN2[]={0,0}, calibSumZP1[]={0,0}, calibSumZP2[]={0,0}; + for(Int_t gi=0; gi<10; gi++){ + calibTowZN1[gi] = equalTowZN1[gi]*calibEne[0]; + calibTowZP1[gi] = equalTowZP1[gi]*calibEne[1]; + calibTowZN2[gi] = equalTowZN2[gi]*calibEne[2]; + calibTowZP2[gi] = equalTowZP2[gi]*calibEne[3]; + // + if(gi<5){ + calibSumZN1[0] += calibTowZN1[gi]; + calibSumZP1[0] += calibTowZP1[gi]; + calibSumZN2[0] += calibTowZN2[gi]; + calibSumZP2[0] += calibTowZP2[gi]; + } + // + else{ + calibSumZN1[1] += calibTowZN1[gi]; + calibSumZP1[1] += calibTowZP1[gi]; + calibSumZN2[1] += calibTowZN2[gi]; + calibSumZP2[1] += calibTowZP2[gi]; + } + } + // + // --- Reconstruction parameters ------------------ + if(!fRecoParam) fRecoParam = (AliZDCRecoParamPbPb*) AliZDCRecoParamPbPb::GetPbPbRecoParam(); + // + Float_t endPointZEM = fRecoParam->GetZEMEndValue(); + Float_t cutFractionZEM = fRecoParam->GetZEMCutFraction(); + Float_t dZEMSup = fRecoParam->GetDZEMSup(); + Float_t dZEMInf = fRecoParam->GetDZEMInf(); + // + Float_t cutValueZEM = endPointZEM*cutFractionZEM; + Float_t supValueZEM = cutValueZEM+(endPointZEM*dZEMSup); + Float_t infValueZEM = cutValueZEM-(endPointZEM*dZEMInf); + // + Float_t maxValEZN1 = fRecoParam->GetEZN1MaxValue(); + Float_t maxValEZP1 = fRecoParam->GetEZP1MaxValue(); + Float_t maxValEZDC1 = fRecoParam->GetEZDC1MaxValue(); + Float_t maxValEZN2 = fRecoParam->GetEZN2MaxValue(); + Float_t maxValEZP2 = fRecoParam->GetEZP2MaxValue(); + Float_t maxValEZDC2 = fRecoParam->GetEZDC2MaxValue(); + // + //printf("\n\t AliZDCReconstructor -> ZEMEndPoint %1.0f, ZEMCutValue %1.0f," + // " ZEMSupValue %1.0f, ZEMInfValue %1.0f\n",endPointZEM,cutValueZEM,supValueZEM,infValueZEM); + + // --- Number of detected spectator nucleons + // *** N.B. -> It works only in Pb-Pb + Int_t nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight; + if(fBeamEnergy!=0){ + nDetSpecNLeft = (Int_t) (calibSumZN1[0]/fBeamEnergy); + nDetSpecPLeft = (Int_t) (calibSumZP1[0]/fBeamEnergy); + nDetSpecNRight = (Int_t) (calibSumZN2[0]/fBeamEnergy); + nDetSpecPRight = (Int_t) (calibSumZP2[0]/fBeamEnergy); + } + else AliWarning(" ATTENTION -> fBeamEnergy = 0\n"); + /*printf("\n\t AliZDCReconstructor -> nDetSpecNLeft %d, nDetSpecPLeft %d," + " nDetSpecNRight %d, nDetSpecPRight %d\n",nDetSpecNLeft, nDetSpecPLeft, + nDetSpecNRight, nDetSpecPRight);*/ + + // --- Number of generated spectator nucleons (from HIJING parameterization) + Int_t nGenSpecNLeft=0, nGenSpecPLeft=0, nGenSpecLeft=0; + Int_t nGenSpecNRight=0, nGenSpecPRight=0, nGenSpecRight=0; + Double_t impPar=0.; + // + Float_t corrADCZEMHG = ZEM1ADCCorr[0] + ZEM2ADCCorr[0]; + // + if(corrADCZEMHG > supValueZEM){ + nGenSpecNLeft = (Int_t) ((fRecoParam->GetfZNCen())->Eval(calibSumZN1[0])); + nGenSpecPLeft = (Int_t) ((fRecoParam->GetfZPCen())->Eval(calibSumZP1[0])); + nGenSpecLeft = (Int_t) ((fRecoParam->GetfZDCCen())->Eval(calibSumZN1[0]+calibSumZP1[0])); + nGenSpecNRight = (Int_t) ((fRecoParam->GetfZNCen())->Eval(calibSumZN2[0])); + nGenSpecPRight = (Int_t) ((fRecoParam->GetfZNCen())->Eval(calibSumZP2[0])); + nGenSpecRight = (Int_t) ((fRecoParam->GetfZNCen())->Eval(calibSumZN2[0]+calibSumZP2[0])); + impPar = (fRecoParam->GetfbCen())->Eval(calibSumZN1[0]+calibSumZP1[0]); + } + else if(corrADCZEMHG < infValueZEM){ + nGenSpecNLeft = (Int_t) ((fRecoParam->GetfZNPer())->Eval(calibSumZN1[0])); + nGenSpecPLeft = (Int_t) ((fRecoParam->GetfZPPer())->Eval(calibSumZP1[0])); + nGenSpecLeft = (Int_t) ((fRecoParam->GetfZDCPer())->Eval(calibSumZN1[0]+calibSumZP1[0])); + impPar = (fRecoParam->GetfbPer())->Eval(calibSumZN1[0]+calibSumZP1[0]); + } + else if(corrADCZEMHG >= infValueZEM && corrADCZEMHG <= supValueZEM){ + nGenSpecNLeft = (Int_t) ((fRecoParam->GetfZEMn())->Eval(corrADCZEMHG)); + nGenSpecPLeft = (Int_t) ((fRecoParam->GetfZEMp())->Eval(corrADCZEMHG)); + nGenSpecLeft = (Int_t)((fRecoParam->GetfZEMsp())->Eval(corrADCZEMHG)); + impPar = (fRecoParam->GetfZEMb())->Eval(corrADCZEMHG); + } + // + if(calibSumZN1[0]/maxValEZN1>1.) nGenSpecNLeft = (Int_t) ((fRecoParam->GetfZEMn())->Eval(corrADCZEMHG)); + if(calibSumZP1[0]/maxValEZP1>1.) nGenSpecPLeft = (Int_t) ((fRecoParam->GetfZEMp())->Eval(corrADCZEMHG)); + if((calibSumZN1[0]+calibSumZP1[0]/maxValEZDC1)>1.){ + nGenSpecLeft = (Int_t)((fRecoParam->GetfZEMsp())->Eval(corrADCZEMHG)); + impPar = (fRecoParam->GetfZEMb())->Eval(corrADCZEMHG); + } + if(calibSumZN2[0]/maxValEZN2>1.) nGenSpecNRight = (Int_t) ((fRecoParam->GetfZEMn())->Eval(corrADCZEMHG)); + if(calibSumZP2[0]/maxValEZP2>1.) nGenSpecPRight = (Int_t) ((fRecoParam->GetfZEMp())->Eval(corrADCZEMHG)); + if((calibSumZN2[0]+calibSumZP2[0]/maxValEZDC2)>1.) nGenSpecRight = (Int_t)((fRecoParam->GetfZEMsp())->Eval(corrADCZEMHG)); + // + if(nGenSpecNLeft>125) nGenSpecNLeft=125; + else if(nGenSpecNLeft<0) nGenSpecNLeft=0; + if(nGenSpecPLeft>82) nGenSpecPLeft=82; + else if(nGenSpecPLeft<0) nGenSpecPLeft=0; + if(nGenSpecLeft>207) nGenSpecLeft=207; + else if(nGenSpecLeft<0) nGenSpecLeft=0; + + // --- Number of generated participants (from HIJING parameterization) + Int_t nPart, nPartTotLeft, nPartTotRight; + nPart = 207-nGenSpecNLeft-nGenSpecPLeft; + nPartTotLeft = 207-nGenSpecLeft; + nPartTotRight = 207-nGenSpecRight; + if(nPart<0) nPart=0; + if(nPartTotLeft<0) nPartTotLeft=0; + if(nPartTotRight<0) nPartTotRight=0; + // + // *** DEBUG *** + /*printf("\n\t AliZDCReconstructor -> calibSumZN1[0] %1.0f, calibSumZP1[0] %1.0f," + " calibSumZN2[0] %1.0f, calibSumZP2[0] %1.0f, corrADCZEMHG %1.0f\n", + calibSumZN1[0],calibSumZP1[0],calibSumZN2[0],calibSumZP2[0],corrADCZEMHG); + printf("\t AliZDCReconstructor -> nGenSpecNLeft %d, nGenSpecPLeft %d, nGenSpecLeft %d\n" + "\t\t nGenSpecNRight %d, nGenSpecPRight %d, nGenSpecRight %d\n", + nGenSpecNLeft, nGenSpecPLeft, nGenSpecLeft, + nGenSpecNRight, nGenSpecPRight, nGenSpecRight); + printf("\t AliZDCReconstructor -> NpartL %d, NpartR %d, b %1.2f fm\n\n",nPartTotLeft, nPartTotRight, impPar); + */ + // create the output tree - loader->MakeTree("R"); - TTree* treeR = loader->TreeR(); - AliZDCReco reco(znenergy, zpenergy, zdcenergy, zemenergy, - nDetSpecN, nDetSpecP, nGenSpecN, nGenSpecP, nGenSpec, - nPartTot, impPar); + AliZDCReco reco(calibSumZN1, calibSumZP1, calibSumZN2, calibSumZP2, + calibTowZN1, calibTowZP1, calibTowZN2, calibTowZP2, + ZEM1ADCCorr, ZEM2ADCCorr, PMRef1, PMRef2, + nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight, + nGenSpecNLeft, nGenSpecPLeft, nGenSpecLeft, nGenSpecNRight, + nGenSpecPRight, nGenSpecRight, nPartTotLeft, nPartTotRight, impPar); + AliZDCReco* preco = &reco; const Int_t kBufferSize = 4000; - treeR->Branch("ZDC", "AliZDCReco", &preco, kBufferSize); + clustersTree->Branch("ZDC", "AliZDCReco", &preco, kBufferSize); // write the output tree - treeR->Fill(); - loader->WriteRecPoints("OVERWRITE"); + clustersTree->Fill(); } //_____________________________________________________________________________ -void AliZDCReconstructor::FillESD(AliRunLoader* runLoader, - AliESD* esd) const +void AliZDCReconstructor::FillZDCintoESD(TTree *clustersTree, AliESDEvent* esd) const { -// fill energies and number of participants to the ESD + // 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); + clustersTree->SetBranchAddress("ZDC", &preco); + + clustersTree->GetEntry(0); + // + AliESDZDC * esdzdc = esd->GetESDZDC(); + Float_t tZN1Ene[5], tZN2Ene[5], tZP1Ene[5], tZP2Ene[5]; + Float_t tZN1EneLR[5], tZN2EneLR[5], tZP1EneLR[5], tZP2EneLR[5]; + for(Int_t i=0; i<5; i++){ + tZN1Ene[i] = reco.GetZN1HREnTow(i); + tZN2Ene[i] = reco.GetZN2HREnTow(i); + tZP1Ene[i] = reco.GetZP1HREnTow(i); + tZP2Ene[i] = reco.GetZP2HREnTow(i); + // + tZN1EneLR[i] = reco.GetZN1LREnTow(i); + tZN2EneLR[i] = reco.GetZN2LREnTow(i); + tZP1EneLR[i] = reco.GetZP1LREnTow(i); + tZP2EneLR[i] = reco.GetZP2LREnTow(i); + } + esdzdc->SetZN1TowerEnergy(tZN1Ene); + esdzdc->SetZN2TowerEnergy(tZN2Ene); + esdzdc->SetZP1TowerEnergy(tZP1Ene); + esdzdc->SetZP2TowerEnergy(tZP2Ene); + esdzdc->SetZN1TowerEnergyLR(tZN1EneLR); + esdzdc->SetZN2TowerEnergyLR(tZN2EneLR); + esdzdc->SetZP1TowerEnergyLR(tZP1EneLR); + esdzdc->SetZP2TowerEnergyLR(tZP2EneLR); + // + esd->SetZDC(reco.GetZN1HREnergy(), reco.GetZP1HREnergy(), reco.GetZEM1HRsignal(), + reco.GetZEM2HRsignal(), reco.GetZN2HREnergy(), reco.GetZP2HREnergy(), + reco.GetNPartLeft(), reco.GetNPartRight()); + // + +} + +//_____________________________________________________________________________ +AliCDBStorage* AliZDCReconstructor::SetStorage(const char *uri) +{ + // Setting the storage + + 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(); - treeR->GetEntry(0); - esd->SetZDC(reco.GetZNenergy(), reco.GetZPenergy(), reco.GetZEMenergy(), - reco.GetNPart()); + if(deleteManager){ + AliCDBManager::Instance()->UnsetDefaultStorage(); + defstorage = 0; // the storage is killed by AliCDBManager::Instance()->Destroy() + } - loader->UnloadRecPoints(); + return storage; } //_____________________________________________________________________________ -AliZDCCalibData* AliZDCReconstructor::GetCalibData(int runNumber) const +AliZDCPedestals* AliZDCReconstructor::GetPedData() const { - //printf("\n\t AliZDCReconstructor::GetCalibData \n"); - //fStorage->PrintSelectionList(); - //AliCDBEntry *entry = fStorage->Get("DBlocal/ZDC/Calib/Data",runNumber); + // Getting pedestal calibration object for ZDC set - - AliCDBStorage *fStorage = AliCDBManager::Instance()->GetStorage("local://$ALICE_ROOT"); - AliCDBEntry *entry = fStorage->Get("ZDC/Calib/Data",0); - - AliZDCCalibData *calibda = (AliZDCCalibData*) entry->GetObject(); + AliCDBEntry *entry = AliCDBManager::Instance()->Get("ZDC/Calib/Pedestals"); + if(!entry) AliFatal("No calibration data loaded!"); - //AliCDBManager::Instance()->Destroy(); + AliZDCPedestals *calibdata = dynamic_cast (entry->GetObject()); + if(!calibdata) AliFatal("Wrong calibration object in calibration file!"); - return calibda; + return calibdata; +} +//_____________________________________________________________________________ +AliZDCCalib* AliZDCReconstructor::GetECalibData() const +{ + + // Getting energy and equalization calibration object for ZDC set + + AliCDBEntry *entry = AliCDBManager::Instance()->Get("ZDC/Calib/EMDCalib"); + if(!entry) AliFatal("No calibration data loaded!"); + + AliZDCCalib *calibdata = dynamic_cast (entry->GetObject()); + if(!calibdata) AliFatal("Wrong calibration object in calibration file!"); + + return calibdata; } +