pp jet spectra with pileup patch

diff --git a/PWGJE/AliAnalysisTaskJetSpectrum2.cxx b/PWGJE/AliAnalysisTaskJetSpectrum2.cxx
index cddda504e8b767d54dd3698c623acc183c8b3872..14a9819993e804e0fc4abe8c5f848761b42b501a 100644 (file)
--- a/PWGJE/AliAnalysisTaskJetSpectrum2.cxx
+++ b/PWGJE/AliAnalysisTaskJetSpectrum2.cxx
@@ -18,7 +18,6 @@
  * about the suitability of this software for any purpose. It is          *
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
-
  
 #include <TROOT.h>
 #include <TRandom.h>
@@ -142,7 +141,12 @@ AliAnalysisTaskSE(),
   fh2deltaPt1Pt2(0x0),
   fh2RelPtFGen(0x0),
   fh3RelPtFGenLeadTrkPt(0x0),
-  fHistList(0x0)  
+  fHistList(0x0),
+  fh1EvtSelection(0),
+  fMaxVertexZ(100.),
+  fMinNcontributors(0),
+  fRejectPileup(0)
+  
 {
 
   for(int ij = 0;ij <kJetTypes;++ij){    
@@ -253,7 +257,11 @@ AliAnalysisTaskJetSpectrum2::AliAnalysisTaskJetSpectrum2(const char* name):
   fh2deltaPt1Pt2(0x0),
   fh2RelPtFGen(0x0),
   fh3RelPtFGenLeadTrkPt(0x0),
-  fHistList(0x0)
+  fHistList(0x0),
+  fh1EvtSelection(0),
+  fMaxVertexZ(100.),
+  fMinNcontributors(0),
+  fRejectPileup(0)
 {
 
   for(int ij = 0;ij <kJetTypes;++ij){    
@@ -387,7 +395,17 @@ void AliAnalysisTaskJetSpectrum2::UserCreateOutputObjects()
   //  Histogram
     
 
+  fh1EvtSelection            = new TH1F("fh1EvtSelection", "Event Selection", 7, -0.5, 6.5);
+  fh1EvtSelection->GetXaxis()->SetBinLabel(1,"ACCEPTED");
+  fh1EvtSelection->GetXaxis()->SetBinLabel(2,"event selection: rejected");
+  fh1EvtSelection->GetXaxis()->SetBinLabel(3,"event class: rejected");
+  fh1EvtSelection->GetXaxis()->SetBinLabel(4,"vertex Ncontr: rejected");
+  fh1EvtSelection->GetXaxis()->SetBinLabel(5,"vertex z: rejected");
+  fh1EvtSelection->GetXaxis()->SetBinLabel(6,"vertex type: rejected");
+  fh1EvtSelection->GetXaxis()->SetBinLabel(7,"pileup: rejected");
 
+  fHistList->Add(fh1EvtSelection);
+ 
   const Int_t nBinPt = 150;
   Double_t binLimitsPt[nBinPt+1];
   for(Int_t iPt = 0;iPt <= nBinPt;iPt++){
@@ -699,6 +717,14 @@ void AliAnalysisTaskJetSpectrum2::UserExec(Option_t */*option*/){
     selected = AliAnalysisHelperJetTasks::TestSelectInfo(fEventSelectionMask);
   }
 
+  if(!selected){
+    // no selection by the service task, we continue
+    if (fDebug > 1)Printf("Not selected %s:%d SelectInfo %d  Class %d",(char*)__FILE__,__LINE__, AliAnalysisHelperJetTasks::Selected(),AliAnalysisHelperJetTasks::EventClass());
+    fh1EvtSelection->Fill(1.); 
+    PostData(1, fHistList);
+    return;
+  }
+
   if(fEventClass>0){
     selected = selected&&(AliAnalysisHelperJetTasks::EventClass()==fEventClass);
   }
@@ -706,6 +732,7 @@ void AliAnalysisTaskJetSpectrum2::UserExec(Option_t */*option*/){
   if(!selected){
     // no selection by the service task, we continue
     if (fDebug > 1)Printf("Not selected %s:%d SelectInfo %d  Class %d",(char*)__FILE__,__LINE__, AliAnalysisHelperJetTasks::Selected(),AliAnalysisHelperJetTasks::EventClass());
+    fh1EvtSelection->Fill(2.);
     PostData(1, fHistList);
     return;
   }
@@ -821,6 +848,44 @@ void AliAnalysisTaskJetSpectrum2::UserExec(Option_t */*option*/){
     }
   }
 
+  AliAODVertex* primVtx = aod->GetPrimaryVertex();
+  Int_t nTracksPrim = primVtx->GetNContributors();
+  
+  
+  if (fDebug > 1) Printf("%s:%d primary vertex selection: %d", (char*)__FILE__,__LINE__,nTracksPrim);
+  if(nTracksPrim < fMinNcontributors){
+    if (fDebug > 1) Printf("%s:%d primary vertex selection: event REJECTED...",(char*)__FILE__,__LINE__); 
+    fh1EvtSelection->Fill(3.);
+    PostData(1, fHistList);
+    return;
+  }
+  
+  if(TMath::Abs(primVtx->GetZ())>fMaxVertexZ){
+    if (fDebug > 1) Printf("%s:%d primary vertex z = %f: event REJECTED...",(char*)__FILE__,__LINE__,primVtx->GetZ()); 
+    fh1EvtSelection->Fill(4.);
+    PostData(1, fHistList);
+    return; 
+  }
+  
+  TString primVtxName(primVtx->GetName());
+
+  if(primVtxName.CompareTo("TPCVertex",TString::kIgnoreCase) == 1){
+    if (fDebug > 1) Printf("%s:%d primary vertex selection: TPC vertex, event REJECTED...",(char*)__FILE__,__LINE__);
+    fh1EvtSelection->Fill(5.);
+    PostData(1, fHistList);
+    return;
+  }
+
+  if(fRejectPileup && AliAnalysisHelperJetTasks::IsPileUp()){
+    if (fDebug > 1) Printf("%s:%d SPD pileup: event REJECTED...",(char*)__FILE__,__LINE__);
+    fh1EvtSelection->Fill(6.);
+    PostData(1, fHistList);
+    return;
+  }
+
+  if (fDebug > 1) Printf("%s:%d event ACCEPTED ...",(char*)__FILE__,__LINE__); 
+  fh1EvtSelection->Fill(0.);
+
  
   // new Scheme
   // first fill all the pure  histograms, i.e. full jets 

diff --git a/PWGJE/AliAnalysisTaskJetSpectrum2.h b/PWGJE/AliAnalysisTaskJetSpectrum2.h
index 0da884a9799c9422d3d4a15914594dbe2085ee65..d53ed7470b33c19ce4083ae8797c12d3bc84c928 100644 (file)
--- a/PWGJE/AliAnalysisTaskJetSpectrum2.h
+++ b/PWGJE/AliAnalysisTaskJetSpectrum2.h
@@ -35,7 +35,6 @@ class TProfile;
 class TProfile2D;
 class TSTring;
 
-
 class AliAnalysisTaskJetSpectrum2 : public AliAnalysisTaskSE
 {
  public:
@@ -93,6 +92,14 @@ class AliAnalysisTaskJetSpectrum2 : public AliAnalysisTaskSE
     virtual void SetNAcceptance(Int_t n);
     virtual void SetAcceptance(Int_t i,Float_t phiMin,Float_t phiMax,Float_t etaMin,Float_t etaMax);// non overlapping regions
     virtual Int_t CheckAcceptance(Float_t phi,Float_t eta);
+
+
+    void    SetMaxVertexZ(Float_t a)      {fMaxVertexZ = a;}
+    void    SetMinNcontributors(Int_t i)  {fMinNcontributors = i;}
+    void    SetPileupRejection(Bool_t a)  {fRejectPileup = a;}
+
+
+
     // Helper
     //
 
@@ -149,7 +156,6 @@ class AliAnalysisTaskJetSpectrum2 : public AliAnalysisTaskSE
     AliVParticle *LeadingTrackFromJetRefs(AliAODJet* jet);
     AliVParticle *LeadingTrackInCone(AliAODJet* jet,TList *list,Float_t r = 0.4);
 
-
     AliJetHeader *fJetHeaderRec;//! The jet header that can be fetched from the userinfo
     AliJetHeader *fJetHeaderGen;//! The jet header that can fetched from the userinfo
     AliAODEvent  *fAODIn; //! where we take the jets from 
@@ -264,6 +270,11 @@ class AliAnalysisTaskJetSpectrum2 : public AliAnalysisTaskSE
     TH2F*   fh2DijetPt2vsPt1[kJetTypes];        //! dijet pt2 vs pt1
     TH2F*   fh2DijetDifvsSum[kJetTypes];        //! dijet dif vs sum
 
+    TH1F*   fh1EvtSelection;
+    Float_t fMaxVertexZ;
+    Int_t   fMinNcontributors;
+    Bool_t  fRejectPileup;
+
 
     TList *fHistList;                  //! Output list
    

diff --git a/PWGJE/CMakelibPWGJE.pkg b/PWGJE/CMakelibPWGJE.pkg
index 6bd156d8bbbf8b145c95fd7f29b657424b6a0b78..40c1f035455b8e46b14bd49825854ce1662b92a8 100644 (file)
--- a/PWGJE/CMakelibPWGJE.pkg
+++ b/PWGJE/CMakelibPWGJE.pkg
@@ -52,6 +52,7 @@ set ( SRCS
     UserTasks/AliAnalysisTaskIDFragmentationFunction.cxx
     UserTasks/AliAnalysisTaskIDFFTCF.cxx
     UserTasks/AliIDFFUtils.cxx
+    UserTasks/AliAnalysisTaskPPJetSpectra.cxx
     )
 
 # Add code that needs fastjet or FJWrapper here 

diff --git a/PWGJE/PWGJELinkDef.h b/PWGJE/PWGJELinkDef.h
index 441b12605a104859a44a0a453deb8bd51bd1b40a..31d1dfc450c1ef01e91dafe7eb1d547196edce4c 100644 (file)
--- a/PWGJE/PWGJELinkDef.h
+++ b/PWGJE/PWGJELinkDef.h
@@ -42,6 +42,7 @@
 #pragma link C++ class AliAnalysisTaskIDFFTCF::AliFragFuncQATrackHistos+;
 #pragma link C++ class AliAnalysisTaskIDFFTCF::AliFragFuncQAJetHistos+;
 #pragma link C++ class AliIDFFUtils+;
+#pragma link C++ class AliAnalysisTaskPPJetSpectra+;
 
 #ifdef HAVE_FASTJET
 #pragma link C++ class AliAnalysisTaskCheckSingleTrackJetRejection+;