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 c72b0d86d126d921ab271894627a59d4512e1b1f..5ac7cb2da2f5c3ed4043b4b871f18dbe02cc75db 100644 (file)
--- a/PWGHF/hfe/AliHFEtaggedTrackAnalysis.cxx
+++ b/PWGHF/hfe/AliHFEtaggedTrackAnalysis.cxx
@@ -57,6 +57,8 @@ AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis():
   , fMagneticField(0.0)
   , fVariablesTRD(kFALSE)
   , fIsPbPb(kFALSE)
+  , fIspPb(kFALSE)
+  , fIsAOD(kFALSE)
 {
   //
   // Dummy constructor
@@ -78,6 +80,8 @@ AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis(const char *name):
   , fMagneticField(0.0)
   , fVariablesTRD(kFALSE)
   , fIsPbPb(kFALSE)
+  , fIspPb(kFALSE)
+  , fIsAOD(kFALSE)
 {
   //
   // Default constructor
@@ -108,6 +112,8 @@ AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis(const AliHFEtaggedTrackAnal
   , fMagneticField(ref.fMagneticField)
   , fVariablesTRD(ref.fVariablesTRD)
   , fIsPbPb(ref.fIsPbPb)
+  , fIspPb(ref.fIspPb)
+  , fIsAOD(ref.fIsAOD)
 {
   //
   // Copy constructor
@@ -136,6 +142,8 @@ AliHFEtaggedTrackAnalysis &AliHFEtaggedTrackAnalysis::operator=(const AliHFEtagg
     fMagneticField = ref.fMagneticField;
     fVariablesTRD = ref.fVariablesTRD;
     fIsPbPb = ref.fIsPbPb;
+    fIspPb = ref.fIspPb;
+    fIsAOD = ref.fIsAOD;
 
     if(ref.fContainer) InitContainer();
    
@@ -195,7 +203,7 @@ void AliHFEtaggedTrackAnalysis::InitContainer(){
 }
 
 //____________________________________________________________
-void AliHFEtaggedTrackAnalysis::ProcessTrack(AliVParticle *track, Int_t abinitioPID){
+void AliHFEtaggedTrackAnalysis::ProcessTrack(AliVTrack *track, Int_t abinitioPID){
   //
   // Filter tracks tagged by V0 PID class
   //
@@ -273,12 +281,17 @@ void AliHFEtaggedTrackAnalysis::ProcessTrack(AliVParticle *track, Int_t abinitio
      AliDebug(2, "Use track in the PID");
      // Apply PID
      AliHFEpidObject hfetrack;
-     hfetrack.SetAnalysisType(AliHFEpidObject::kESDanalysis);
+     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 hfetrack.SetPP();
+     else
+     {
+        if(fIspPb) hfetrack.SetpPb();
+        else hfetrack.SetPP();
+     }
      fPID->SetVarManager(fVarManager);
      fPID->IsSelected(&hfetrack, fContainer, "taggedTrackContainer", fPIDqa);
    }