Switching off the TOF Dz cut in AliESDtrackCuts (cannot be applied on AODTracks)

diff --git a/PWGHF/vertexingHF/AliAnalysisVertexingHF.h b/PWGHF/vertexingHF/AliAnalysisVertexingHF.h
index c412691bc20f09eab70b8df350afbf331c7c72f9..9e7fd06cdbecff5dbafc1ea4c8a254fa8c898b57 100644 (file)
--- a/PWGHF/vertexingHF/AliAnalysisVertexingHF.h
+++ b/PWGHF/vertexingHF/AliAnalysisVertexingHF.h
@@ -14,8 +14,11 @@
 //-------------------------------------------------------------------------
 
 #include <TNamed.h>
+#include <TList.h>
+
+#include "AliAnalysisFilter.h"
+#include "AliESDtrackCuts.h"
 
-class TList;
 class AliESDVertex;
 class AliAODRecoDecay;
 class AliAODRecoDecayHF;
@@ -102,8 +105,20 @@ class AliAnalysisVertexingHF : public TNamed {
     { fRecoPrimVtxSkippingTrks=kFALSE; fRmTrksFromPrimVtx=kFALSE;}
   void SetRmTrksFromPrimVtx() 
     {fRmTrksFromPrimVtx=kTRUE; fRecoPrimVtxSkippingTrks=kFALSE; }
-  void SetTrackFilter(AliAnalysisFilter* trackF) { fTrackFilter = trackF; }
-  void SetTrackFilterSoftPi(AliAnalysisFilter* trackF) { fTrackFilterSoftPi = trackF; }
+  void SetTrackFilter(AliAnalysisFilter* trackF) {
+    // switch off the TOF selection that cannot be applied with AODTracks 
+    TList *l = (TList*)trackF->GetCuts();
+    AliESDtrackCuts *tcuts = (AliESDtrackCuts*)l->FindObject("AliESDtrackCuts");
+    if(tcuts->GetFlagCutTOFdistance()) tcuts->SetFlagCutTOFdistance(kFALSE);
+    fTrackFilter = trackF; 
+  }
+  void SetTrackFilterSoftPi(AliAnalysisFilter* trackF) { 
+    // switch off the TOF selection that cannot be applied with AODTracks 
+    TList *l = (TList*)trackF->GetCuts();
+    AliESDtrackCuts *tcuts = (AliESDtrackCuts*)l->FindObject("AliESDtrackCuts");
+    if(tcuts->GetFlagCutTOFdistance()) tcuts->SetFlagCutTOFdistance(kFALSE);
+    fTrackFilterSoftPi = trackF; 
+  }
   AliAnalysisFilter* GetTrackFilter() const { return fTrackFilter; }
   AliAnalysisFilter* GetTrackFilterSoftPi() const { return fTrackFilterSoftPi; }
   void SetCutsD0toKpi(AliRDHFCutsD0toKpi* cuts) { fCutsD0toKpi = cuts; }

diff --git a/PWGHF/vertexingHF/AliRDHFCuts.cxx b/PWGHF/vertexingHF/AliRDHFCuts.cxx
index 43a1900d5fee2fbc7336f3c02a9190f1f1a31dea..0dae5b3fbf4f4ad5b06dddf9b665f9f514605495 100644 (file)
--- a/PWGHF/vertexingHF/AliRDHFCuts.cxx
+++ b/PWGHF/vertexingHF/AliRDHFCuts.cxx
@@ -417,6 +417,8 @@ Bool_t AliRDHFCuts::IsDaughterSelected(AliAODTrack *track,const AliESDVertex *pr
   //\r
   if(!cuts) return kTRUE;\r
 \r
+  if(cuts->GetFlagCutTOFdistance()) cuts->SetFlagCutTOFdistance(kFALSE);\r
+\r
   Bool_t retval=kTRUE;\r
 \r
   // convert to ESD track here\r