Helper method to create input chain AOD with friend AODVertexingHF (A.Rossi)

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

diff --git a/ZDC/AliZDCReconstructor.cxx b/ZDC/AliZDCReconstructor.cxx
index 9acf51ca0f74ec6dc7a231875068813ef00237e8..ceeeb9c1877e57887f5e23eae4a9a3b824d5858f 100644 (file)
--- a/ZDC/AliZDCReconstructor.cxx
+++ b/ZDC/AliZDCReconstructor.cxx
@@ -23,10 +23,11 @@
 
 
 #include <TF1.h>
+#include <TMap.h>
 
 #include "AliRunLoader.h"
-#include "AliRunInfo.h"
 #include "AliRawReader.h"
+#include "AliGRPObject.h"
 #include "AliESDEvent.h"
 #include "AliESDZDC.h"
 #include "AliZDCDigit.h"
@@ -46,9 +47,12 @@ AliZDCRecoParam *AliZDCReconstructor::fRecoParam=0;  //reconstruction parameters
 //_____________________________________________________________________________
 AliZDCReconstructor:: AliZDCReconstructor() :
   fPedData(GetPedData()),
-  fECalibData(GetECalibData())
+  fECalibData(GetECalibData()),
+  fRecoMode(0),
+  fBeamEnergy(0.)
 {
   // **** Default constructor
+  SetRecoMode();
 
 }
 
@@ -62,6 +66,54 @@ AliZDCReconstructor::~AliZDCReconstructor()
    if(fECalibData) delete fECalibData;
 }
 
+//____________________________________________________________________________
+void AliZDCReconstructor::SetRecoMode()
+{
+  // Setting reconstruction mode
+
+  // Initialization of the GRP entry 
+  AliGRPObject* grpData;
+  AliCDBEntry*  entry = AliCDBManager::Instance()->Get("GRP/GRP/Data");
+  if(entry){
+    TMap* m = dynamic_cast<TMap*>(entry->GetObject());  // old GRP entry
+    if(m){
+       m->Print();
+       grpData = new AliGRPObject();
+       grpData->ReadValuesFromMap(m);
+    }
+    else{
+       grpData = dynamic_cast<AliGRPObject*>(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(TTree* digitsTree, TTree* clustersTree) const
 {
@@ -169,19 +221,14 @@ void AliZDCReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree) co
    }
   }
 
-  // Reconstruct the event according to beam type from AliInfo
-  AliRunInfo *runInfo;
-  if((strcmp(runInfo->GetBeamType(),"p-p") == 0)){
-    printf("\n\t ZDC reconstructing p-p event\n");
+  // reconstruct the event
+  if(fRecoMode==0)
     ReconstructEventpp(clustersTree, tZN1Corr, tZP1Corr, tZN2Corr, tZP2Corr, 
        dZEM1Corr, dZEM2Corr, PMRef1, PMRef2);
-  }
-  else if((strcmp(runInfo->GetBeamType(),"A-A") == 0)){
-    printf("\n\t ZDC reconstructing A-A event\n");
+  else if(fRecoMode==1)
     ReconstructEventPbPb(clustersTree, tZN1Corr, tZP1Corr, tZN2Corr, tZP2Corr, 
        dZEM1Corr, dZEM2Corr, PMRef1, PMRef2);
-  }
-  else printf("\n\t ZDC does not reconstruct event! Beam type != p-p || A-A\n");
+
 }
 
 //_____________________________________________________________________________
@@ -262,19 +309,13 @@ void AliZDCReconstructor::Reconstruct(AliRawReader* rawReader, TTree* clustersTr
     }//IsADCDataWord
   }
     
-  // Reconstruct the event according to beam type from AliInfo
-  AliRunInfo *runInfo;
-  if((strcmp(runInfo->GetBeamType(),"p-p") == 0)){
-    printf("\n\t ZDC reconstructing p-p event\n");
+  // reconstruct the event
+  if(fRecoMode==0)
     ReconstructEventpp(clustersTree, tZN1Corr, tZP1Corr, tZN2Corr, tZP2Corr, 
        dZEM1Corr, dZEM2Corr, PMRef1, PMRef2);
-  }
-  else if((strcmp(runInfo->GetBeamType(),"A-A") == 0)){
-    printf("\n\t ZDC reconstructing A-A event\n");
+  else if(fRecoMode==1)
     ReconstructEventPbPb(clustersTree, tZN1Corr, tZP1Corr, tZN2Corr, tZP2Corr, 
        dZEM1Corr, dZEM2Corr, PMRef1, PMRef2);
-  }
-  else printf("\n\t ZDC does not reconstruct event! Beam type != p-p || A-A\n");
 
 }
 
@@ -342,19 +383,17 @@ void AliZDCReconstructor::ReconstructEventpp(TTree *clustersTree, Float_t* ZN1AD
        calibSumZP2[1] += calibTowZP2[gi];
      }
   }
-
-  //
-  // --- Reconstruction parameters ------------------ 
-  if(!fRecoParam)  fRecoParam = (AliZDCRecoParampp*) AliZDCRecoParampp::GetppRecoParam();
   
   //  ---      Number of detected spectator nucleons
   //  *** N.B. -> It works only in Pb-Pb
   Int_t nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight;
-  Float_t beamE = fRecoParam->GetBeamEnergy();
-  nDetSpecNLeft = (Int_t) (calibSumZN1[0]/beamE);
-  nDetSpecPLeft = (Int_t) (calibSumZP1[0]/beamE);
-  nDetSpecNRight = (Int_t) (calibSumZN2[0]/beamE);
-  nDetSpecPRight = (Int_t) (calibSumZP2[0]/beamE);
+  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);*/
@@ -466,11 +505,13 @@ void AliZDCReconstructor::ReconstructEventPbPb(TTree *clustersTree, Float_t* ZN1
   //  ---      Number of detected spectator nucleons
   //  *** N.B. -> It works only in Pb-Pb
   Int_t nDetSpecNLeft, nDetSpecPLeft, nDetSpecNRight, nDetSpecPRight;
-  Float_t beamE = fRecoParam->GetBeamEnergy();
-  nDetSpecNLeft = (Int_t) (calibSumZN1[0]/beamE);
-  nDetSpecPLeft = (Int_t) (calibSumZP1[0]/beamE);
-  nDetSpecNRight = (Int_t) (calibSumZN2[0]/beamE);
-  nDetSpecPRight = (Int_t) (calibSumZP2[0]/beamE);
+  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);*/
@@ -593,7 +634,7 @@ void AliZDCReconstructor::FillZDCintoESD(TTree *clustersTree, AliESDEvent* esd)
   // 
   esd->SetZDC(reco.GetZN1HREnergy(), reco.GetZP1HREnergy(), reco.GetZEM1HRsignal(), 
              reco.GetZEM2HRsignal(), reco.GetZN2HREnergy(), reco.GetZP2HREnergy(), 
-             reco.GetNPartLeft(),reco.GetNPartRight());
+             reco.GetNPartLeft(), reco.GetNPartRight());
   //
   
 }