]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGJE/AliAnalysisTaskJetSpectrum2.cxx
change axis title style to italics
[u/mrichter/AliRoot.git] / PWGJE / AliAnalysisTaskJetSpectrum2.cxx
index 8adcb87298513e3d48dab3fc5ac97344e3792ff3..fc28f6fc845714cd4bf66d471cb5d9f406d55371 100644 (file)
@@ -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>
@@ -139,9 +138,15 @@ AliAnalysisTaskSE(),
   fh2MultGen(0x0),
   fh2RPCentrality(0x0),
   fh2PtFGen(0x0),
+  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){    
@@ -249,9 +254,14 @@ AliAnalysisTaskJetSpectrum2::AliAnalysisTaskJetSpectrum2(const char* name):
   fh2MultGen(0x0),
   fh2RPCentrality(0x0),
   fh2PtFGen(0x0),
+  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){    
@@ -385,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++){
@@ -457,6 +477,10 @@ void AliAnalysisTaskJetSpectrum2::UserCreateOutputObjects()
   fh2PtFGen = new TH2F("fh2PtFGen",Form("%s vs. %s;p_{T,gen};p_{T,rec}",fBranchRec.Data(),fBranchGen.Data()),nBinPt,binLimitsPt,nBinPt,binLimitsPt);
   fHistList->Add(fh2PtFGen);
 
+  //  fh2deltaPt1Pt2(0x0),
+  fh2deltaPt1Pt2 = new TH3F("fh2deltaPt1Pt2",Form("%s vs. %s;delta;p_{T,gen};p_{T,rec}",fBranchRec.Data(),fBranchGen.Data()),nBinPt,binLimitsPt,nBinPt,binLimitsPt,nBinPt,binLimitsPt);
+  fHistList->Add(fh2deltaPt1Pt2);
+
   fh2RelPtFGen = new TH2F("fh2RelPtFGen",";p_{T,gen};p_{T,rec}-p_{T,gen}/p_{T,Gen}",nBinPt,binLimitsPt,241,-2.41,2.41);
   fHistList->Add(fh2RelPtFGen);
 
@@ -548,7 +572,7 @@ void AliAnalysisTaskJetSpectrum2::UserCreateOutputObjects()
       nBins1[1] = 600;
       nBins1[5] = 1;
     }
-    const Double_t xmin1[nBinsSparse1]  = {        -0.5,nbinr5min,  0,   0,        -0.5, 0.,         -0.5,  0.,           -0.5,};
+    const Double_t xmin1[nBinsSparse1]  = {        -0.5, static_cast<Double_t>(nbinr5min),  0,   0,        -0.5, 0.,         -0.5,  0.,           -0.5,};
     const Double_t xmax1[nBinsSparse1]  = {kMaxJets+0.5,250,100,4000,fNRPBins-0.5,1.0,fNTrigger-0.5,200.,fNAcceptance+0.5};
      
     const Double_t binArrayArea[nBinsArea+1] = {xmin1[5],0.07,0.1,0.15,0.2,0.25,0.3,0.4,0.5,0.6,xmax1[5]};
@@ -574,14 +598,14 @@ void AliAnalysisTaskJetSpectrum2::UserCreateOutputObjects()
     
     // Bins:  Jet number: pTJet, cent, eta, phi, Area, trigger, acceptance, signed pT leading
     const Int_t nBinsSparse2 = 9;
-    Int_t nBins2[nBinsSparse2] = {     kMaxJets+1, 60,   8,  18, 72, 10,fNTrigger,fNAcceptance+1,20};
+    Int_t nBins2[nBinsSparse2] = {     kMaxJets+1, 60,   8,  6, 90, 100,fNTrigger,fNAcceptance+1,20};
     if(cJetBranch.Contains("RandomCone")){
       nBins2[5] = 1;
     }
     const Double_t xmin2[nBinsSparse2]  = {        -0.5, -50,  0,-0.9,              0,  0.,        -0.5,            -0.5, -100};
     const Double_t xmax2[nBinsSparse2]  = {kMaxJets+0.5, 250, 80, 0.9, 2.*TMath::Pi(),1.0,fNTrigger-0.5,fNAcceptance+0.5,  100};
     fhnJetPtQA[ij] = new THnSparseF(Form("fhnJetPtQA%s",cAdd.Data()),";jet number;p_{T,jet};cent;#eta;#phi;area;trigger;acceptance bin;signed pt leading",nBinsSparse2,nBins2,xmin2,xmax2);
-    fhnJetPt[ij]->SetBinEdges(5,binArrayArea);
+    //    fhnJetPt[ij]->SetBinEdges(5,binArrayArea);
     fHistList->Add(fhnJetPtQA[ij]);
     
     // Bins:track number  pTtrack, cent, mult, RP.  charge total bins = 224 k
@@ -693,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);
   }
@@ -700,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;
   }
@@ -815,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 
@@ -978,8 +1049,6 @@ void AliAnalysisTaskJetSpectrum2::FillJetHistos(TList &jetsList,TList &particles
 
   if(nJets<=0)return;
   
-  Float_t ptOld = 1.E+32;
-  Float_t pT0 = 0;
   Float_t pT1 = 0;
   Float_t phi0 = 0;
   Float_t phi1 = 0;
@@ -1026,12 +1095,10 @@ void AliAnalysisTaskJetSpectrum2::FillJetHistos(TList &jetsList,TList &particles
       continue;
     }
     fh1PtJetsIn[iType]->Fill(ptJet);
-    ptOld = ptJet;
     
     // find the dijets assume sorting and acceptance cut...
     if(ij==0){
       ij0 = ij;
-      pT0 = ptJet;
       phi0 = jet->Phi();
       if(phi0<0)phi0+=TMath::Pi()*2.;
     }
@@ -1065,23 +1132,25 @@ void AliAnalysisTaskJetSpectrum2::FillJetHistos(TList &jetsList,TList &particles
     if(ttphi<0)ttphi+=TMath::Pi()*2.;  
     Float_t ttpt=tt->Pt();
     Int_t phiBintt = GetPhiBin(ttphi-fRPAngle);
-     Double_t dphitrigjet=RelativePhi(ttphi,phiJet);
+    Double_t dphitrigjet=RelativePhi(ttphi,phiJet);
     if(fTRP==1){
-    if(TMath::Abs(dphitrigjet)<TMath::Pi()-0.6) continue; 
-    var1[1] = ptJet;
-    var1[4] = phiBintt;
-    var1[5] = jet->EffectiveAreaCharged();
-    var1[7] = ttpt;
-    var1[8] = CheckAcceptance(phiJet,etaJet);}
-
-     if(fTRP==0){
-    var1[1] = ptJet;
-    var1[4] = phiBin;
-    var1[5] = jet->EffectiveAreaCharged();
-    var1[7] = ptLead;
-    var1[8] = CheckAcceptance(phiJet,etaJet);}
+      if(TMath::Abs(dphitrigjet)<TMath::Pi()-0.6) continue; 
+      var1[1] = ptJet;
+      var1[4] = phiBintt;
+      var1[5] = jet->EffectiveAreaCharged();
+      var1[7] = ttpt;
+      var1[8] = CheckAcceptance(phiJet,etaJet);
+    }
     
-
+    if(fTRP==0){
+      var1[1] = ptJet;
+      var1[4] = phiBin;
+      var1[5] = jet->EffectiveAreaCharged();
+      var1[7] = ptLead;
+      var1[8] = CheckAcceptance(phiJet,etaJet);
+    }
+    
+    //jet number;p_{T,jet};cent;#eta;#phi;area;trigger;acceptance bin;signed pt leading
     var2[1] = ptJet;
     var2[3] = etaJet;
     var2[4] = phiJet;
@@ -1413,12 +1482,14 @@ void AliAnalysisTaskJetSpectrum2::FillMatchHistos(TList &recJetsList,TList &genJ
        fhnJetContainer->Fill(containerGen,kStep6); // all rec jets in eta window with generated partner
 
        fhnCorrelation->Fill(container);
-       if(ptGen>0){
-         Float_t delta = (ptRec-ptGen)/ptGen;
-         fh2RelPtFGen->Fill(ptGen,delta);
-         fh3RelPtFGenLeadTrkPt->Fill(ptGen,delta,recJet->GetPtLeading());
-         fh2PtFGen->Fill(ptGen,ptRec);
-       }
+       
+       Float_t delta = (ptRec-ptGen)/ptGen;
+       fh2RelPtFGen->Fill(ptGen,delta);
+       fh3RelPtFGenLeadTrkPt->Fill(ptGen,delta,recJet->GetPtLeading());
+       fh2PtFGen->Fill(ptGen,ptRec);
+
+       fh2deltaPt1Pt2->Fill(ptRec-ptGen,ptRec,ptGen);
+       
       } 
     }// loop over reconstructed jets
   }