Fix of sigmaZ for crossing tracklets from Alex

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEtaggedTrackAnalysis.cxx

diff --git a/PWGHF/hfe/AliHFEtaggedTrackAnalysis.cxx b/PWGHF/hfe/AliHFEtaggedTrackAnalysis.cxx
index ee9d51516c04aa688a779958dda4327c066a5057..5ac7cb2da2f5c3ed4043b4b871f18dbe02cc75db 100644 (file)
--- a/PWGHF/hfe/AliHFEtaggedTrackAnalysis.cxx
+++ b/PWGHF/hfe/AliHFEtaggedTrackAnalysis.cxx
@@ -58,6 +58,7 @@ AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis():
   , fVariablesTRD(kFALSE)
   , fIsPbPb(kFALSE)
   , fIspPb(kFALSE)
+  , fIsAOD(kFALSE)
 {
   //
   // Dummy constructor
@@ -80,6 +81,7 @@ AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis(const char *name):
   , fVariablesTRD(kFALSE)
   , fIsPbPb(kFALSE)
   , fIspPb(kFALSE)
+  , fIsAOD(kFALSE)
 {
   //
   // Default constructor
@@ -111,6 +113,7 @@ AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis(const AliHFEtaggedTrackAnal
   , fVariablesTRD(ref.fVariablesTRD)
   , fIsPbPb(ref.fIsPbPb)
   , fIspPb(ref.fIspPb)
+  , fIsAOD(ref.fIsAOD)
 {
   //
   // Copy constructor
@@ -140,6 +143,7 @@ AliHFEtaggedTrackAnalysis &AliHFEtaggedTrackAnalysis::operator=(const AliHFEtagg
     fVariablesTRD = ref.fVariablesTRD;
     fIsPbPb = ref.fIsPbPb;
     fIspPb = ref.fIspPb;
+    fIsAOD = ref.fIsAOD;
 
     if(ref.fContainer) InitContainer();
    
@@ -215,8 +219,8 @@ void AliHFEtaggedTrackAnalysis::ProcessTrack(AliVTrack *track, Int_t abinitioPID
       
       const AliExternalTrackParam *trueparam = NULL;
       if(esdtrackc->GetOuterParam()) {
-             trueparam = esdtrackc->GetOuterParam();
-             fVarManager->NewTrack((AliVParticle *)trueparam, NULL, fCentralityF, abinitioPID, kTRUE);
+       trueparam = esdtrackc->GetOuterParam();
+       fVarManager->NewTrack((AliVParticle *)trueparam, NULL, fCentralityF, abinitioPID, kTRUE);
       }
       else return;
     }
@@ -277,17 +281,16 @@ void AliHFEtaggedTrackAnalysis::ProcessTrack(AliVTrack *track, Int_t abinitioPID
      AliDebug(2, "Use track in the PID");
      // Apply PID
      AliHFEpidObject hfetrack;
-     if(track->IsA() == AliESDtrack::Class())
-      hfetrack.SetAnalysisType(AliHFEpidObject::kESDanalysis);
-     else
-      hfetrack.SetAnalysisType(AliHFEpidObject::kAODanalysis);
+     if(!fIsAOD) hfetrack.SetAnalysisType(AliHFEpidObject::kESDanalysis);
+     else hfetrack.SetAnalysisType(AliHFEpidObject::kAODanalysis);
      hfetrack.SetRecTrack(track);
      hfetrack.SetAbInitioPID(abinitioPID);
      hfetrack.SetCentrality(fCentralityF);
      if(fIsPbPb) hfetrack.SetPbPb();
-     else{
-      if(fIspPb) hfetrack.SetpPb();
-           else hfetrack.SetPP();
+     else
+     {
+        if(fIspPb) hfetrack.SetpPb();
+        else hfetrack.SetPP();
      }
      fPID->SetVarManager(fVarManager);
      fPID->IsSelected(&hfetrack, fContainer, "taggedTrackContainer", fPIDqa);