jet test update
authorlmilano <lmilano@cern.ch>
Tue, 14 Oct 2014 09:15:49 +0000 (11:15 +0200)
committerlmilano <lmilano@cern.ch>
Tue, 14 Oct 2014 09:16:04 +0000 (11:16 +0200)
PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskJetSpectraAOD.cxx
PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskJetSpectraAOD.h

index 4b617c1..e67e55c 100644 (file)
@@ -86,7 +86,9 @@ AliAnalysisTaskJetSpectraAOD::AliAnalysisTaskJetSpectraAOD(const char *name) : A
   fDebug(0),
   fMinNcontributors(0),
   fRejectPileup(0),
-  fR(0.4)
+  fR(0.4),
+  fZvertexDiff(1),
+  fZvertex(10.)
 {
   // Default constructor
   
@@ -259,7 +261,33 @@ void AliAnalysisTaskJetSpectraAOD::UserExec(Option_t *)
     PostData(3, fTrackCuts);
     return;
   }
+    
+  //cut on difference of the z-vertex
+  if (fZvertexDiff || (fZvertex>0)) {
+    
+    Double_t vzPRI = +999;
+    Double_t vzSPD = -999;
+    
+    const AliVVertex *pv = fAOD->GetPrimaryVertex(); // AOD
+    if (pv && pv->GetNContributors()>0) {
+      vzPRI = pv->GetZ();
+    }
+    
+    const AliVVertex *sv = fAOD->GetPrimaryVertexSPD();
+    if (sv && sv->GetNContributors()>0) {
+      vzSPD = sv->GetZ();
+    }
+    
+    Double_t  dvertex = TMath::Abs(vzPRI-vzSPD);
 
+    if (fZvertexDiff && (dvertex>0.1)) {
+      if (fDebug) Printf("%s:%d ZvertexDiff Event selection: event REJECTED...",(char*)__FILE__,__LINE__);
+      return;}
+    if ((fZvertex>0) && (TMath::Abs(vzPRI)>fZvertex)) {
+      if (fDebug) Printf("%s:%d ZvertexDiff Event selection: event REJECTED...",(char*)__FILE__,__LINE__);
+      return;}
+  }
+  
   //ESE event cuts
   if(!fEventCuts->IsSelected(fAOD,fTrackCuts)){
     if (fDebug) Printf("%s:%d ESE Event selection: event REJECTED...",(char*)__FILE__,__LINE__);
index d551e80..db39819 100644 (file)
@@ -45,7 +45,9 @@ class AliAnalysisTaskJetSpectraAOD : public AliAnalysisTaskSE
      fDebug(0),
      fMinNcontributors(0),
      fRejectPileup(0),
-     fR(0.4)
+     fR(0.4),
+     fZvertexDiff(1),
+     fZvertex(10.)
        {}
   AliAnalysisTaskJetSpectraAOD(const char *name);
   virtual ~AliAnalysisTaskJetSpectraAOD();
@@ -93,6 +95,8 @@ class AliAnalysisTaskJetSpectraAOD : public AliAnalysisTaskSE
   void    SetMinNcontributors(Int_t val)  {fMinNcontributors = val;}
   void    SetPileupRejection(Bool_t val)  {fRejectPileup = val;}
   
+  void    SetZvertexDiff (Bool_t val) { fZvertexDiff = val; }
+  
   void SetJetParameterR(Double_t val) {fR = val;}
 
   void   UserCreateOutputObjects();
@@ -141,10 +145,13 @@ class AliAnalysisTaskJetSpectraAOD : public AliAnalysisTaskSE
   
   Double_t fR; // cone radius
   
+  Bool_t fZvertexDiff;
+  Double_t fZvertex;
+  
   AliAnalysisTaskJetSpectraAOD(const AliAnalysisTaskJetSpectraAOD&);
   AliAnalysisTaskJetSpectraAOD& operator=(const AliAnalysisTaskJetSpectraAOD&);
   
-  ClassDef(AliAnalysisTaskJetSpectraAOD, 3);
+  ClassDef(AliAnalysisTaskJetSpectraAOD, 4);
 };
 
 #endif