]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGJE/AliAnalysisTaskJetSpectrum2.cxx
fix by Ruediger
[u/mrichter/AliRoot.git] / PWGJE / AliAnalysisTaskJetSpectrum2.cxx
index c76f636ea105614d5b8b17249b528d6468963ece..9dd55cab1e29ac78c0eb90e2a8da7c3e21fab899 100644 (file)
@@ -93,6 +93,8 @@ AliAnalysisTaskSE(),
   fDoMatching(kFALSE),
   fNMatchJets(5),
   fNRPBins(3),
+  fTRP(0),
+  fDebug(0),
   fJetTriggerExcludeMask(AliAODJet::kHighTrackPtTriggered),
   fJetTriggerBestMask(AliAODJet::kHighTrackPtBest),
   fFilterMask(0),
@@ -137,7 +139,9 @@ AliAnalysisTaskSE(),
   fh2MultGen(0x0),
   fh2RPCentrality(0x0),
   fh2PtFGen(0x0),
+  fh2deltaPt1Pt2(0x0),
   fh2RelPtFGen(0x0),
+  fh3RelPtFGenLeadTrkPt(0x0),
   fHistList(0x0)  
 {
 
@@ -200,6 +204,8 @@ AliAnalysisTaskJetSpectrum2::AliAnalysisTaskJetSpectrum2(const char* name):
   fDoMatching(kFALSE),
   fNMatchJets(5),
   fNRPBins(3),
+  fTRP(0),
+  fDebug(0),
   fJetTriggerExcludeMask(AliAODJet::kHighTrackPtTriggered),
   fJetTriggerBestMask(AliAODJet::kHighTrackPtBest),
   fFilterMask(0),
@@ -244,7 +250,9 @@ AliAnalysisTaskJetSpectrum2::AliAnalysisTaskJetSpectrum2(const char* name):
   fh2MultGen(0x0),
   fh2RPCentrality(0x0),
   fh2PtFGen(0x0),
+  fh2deltaPt1Pt2(0x0),
   fh2RelPtFGen(0x0),
+  fh3RelPtFGenLeadTrkPt(0x0),
   fHistList(0x0)
 {
 
@@ -451,9 +459,30 @@ 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);
-  
+
+  const Int_t nBinsLeadingTrackPt = 10;
+  const Double_t binArrayLeadingTrackPt[nBinsLeadingTrackPt+1] = {0.,1.,2.,3.,4.,5.,6.,8.,10.,12.,200.}; //store pT of leading track in jet
+
+  const Int_t nBinDeltaPt = 241;
+  Double_t binLimitsDeltaPt[nBinDeltaPt+1];
+  for(Int_t iDeltaPt = 0;iDeltaPt <= nBinDeltaPt;iDeltaPt++){
+    if(iDeltaPt == 0){
+      binLimitsDeltaPt[iDeltaPt] = -2.41;
+    }
+    else {
+      binLimitsDeltaPt[iDeltaPt] =  binLimitsDeltaPt[iDeltaPt-1] + 0.02;
+    }
+  }
+
+  fh3RelPtFGenLeadTrkPt = new TH3F("fh3RelPtFGenLeadTrkPt",";p_{T,gen};p_{T,rec}-p_{T,gen}/p_{T,Gen};p_{T}^{leading track}",nBinPt,binLimitsPt,nBinDeltaPt,binLimitsDeltaPt,nBinsLeadingTrackPt,binArrayLeadingTrackPt);
+  fHistList->Add(fh3RelPtFGenLeadTrkPt);  
+
   for(int ij = 0;ij <kJetTypes;++ij){    
     TString cAdd = "";
     TString cJetBranch = "";
@@ -512,18 +541,24 @@ void AliAnalysisTaskJetSpectrum2::UserCreateOutputObjects()
 
     // Bins:  Jet number: pTJet, cent, mult, RP, Area, trigger, leading track pT total bins = 4.5M
     const Int_t nBinsSparse1 = 9;
-    const Int_t nBinsLeadingTrackPt = 10;
     const Int_t nBinsArea = 10;
-    Int_t nBins1[nBinsSparse1] = {     kMaxJets+1,120, 10,  fNBinsMult,    fNRPBins, nBinsArea,fNTrigger,fNBinsLeadingTrackPt,fNAcceptance+1};
+    Int_t nbinr5=120;
+    Int_t nbinr5min=-50;
+  
+    if(cJetBranch.Contains("KT05")){
+     nbinr5=132;
+     nbinr5min=-80;}
+         
+    Int_t nBins1[nBinsSparse1] = {     kMaxJets+1,nbinr5, 10,  fNBinsMult,    fNRPBins, nBinsArea,fNTrigger,fNBinsLeadingTrackPt,fNAcceptance+1};
     if(cJetBranch.Contains("RandomCone")){
       nBins1[1] = 600;
       nBins1[5] = 1;
     }
-    const Double_t xmin1[nBinsSparse1]  = {        -0.5,-50,  0,   0,        -0.5, 0.,         -0.5,  0.,           -0.5,};
+    const Double_t xmin1[nBinsSparse1]  = {        -0.5,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]};
-    const Double_t binArrayLeadingTrackPt[nBinsLeadingTrackPt+1] = {xmin1[7],1.,2.,3.,4.,5.,6.,8.,10.,12.,xmax1[7]}; //store pT of leading track in jet
+
 
     fhnJetPt[ij] = new THnSparseF(Form("fhnJetPt%s",cAdd.Data()),";jet number;p_{T,jet};cent;# tracks;RP;area;trigger;leading track p_{T};acceptance bin",nBinsSparse1,nBins1,xmin1,xmax1);
     fhnJetPt[ij]->SetBinEdges(5,binArrayArea);
@@ -545,14 +580,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+0.5,20};
+    Int_t nBins2[nBinsSparse2] = {     kMaxJets+1, 60,   8,  18, 18, 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
@@ -857,6 +892,7 @@ void AliAnalysisTaskJetSpectrum2::UserExec(Option_t */*option*/){
   TList genParticles;
 
   Int_t nT = GetListOfTracks(&recParticles,fTrackTypeRec);
+
   if(fDebug>2)Printf("%s:%d Selected Rec tracks: %d %d",(char*)__FILE__,__LINE__,nT,recParticles.GetEntries());
   nT = GetListOfTracks(&genParticles,fTrackTypeGen);
   if(fDebug>2)Printf("%s:%d Selected Gen tracks: %d %d",(char*)__FILE__,__LINE__,nT,genParticles.GetEntries());
@@ -1018,7 +1054,6 @@ void AliAnalysisTaskJetSpectrum2::FillJetHistos(TList &jetsList,TList &particles
       }
     }
     // fill jet histos for kmax jets
-
     Float_t phiJet = jet->Phi();
     Float_t etaJet = jet->Eta();
     if(phiJet<0)phiJet+=TMath::Pi()*2.;    
@@ -1028,15 +1063,30 @@ void AliAnalysisTaskJetSpectrum2::FillJetHistos(TList &jetsList,TList &particles
     fh1PtIn[iType][kMaxJets]->Fill(ptJet);
     // fill leading jets...
     AliVParticle *leadTrack = LeadingTrackFromJetRefs(jet);
-    //      AliVParticle *leadTrack = LeadingTrackInCone(jet,&particlesList);
+    //AliVParticle *leadTrack = LeadingTrackInCone(jet,&particlesList);
     Int_t phiBin = GetPhiBin(phiJet-fRPAngle);
     Double_t ptLead = jet->GetPtLeading(); //pT of leading jet
+    AliVParticle *tt = (AliVParticle*)particlesList.At(0);
+    Float_t ttphi=tt->Phi();
+    if(ttphi<0)ttphi+=TMath::Pi()*2.;  
+    Float_t ttpt=tt->Pt();
+    Int_t phiBintt = GetPhiBin(ttphi-fRPAngle);
+     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);
+    var1[8] = CheckAcceptance(phiJet,etaJet);}
+    
 
     var2[1] = ptJet;
     var2[3] = etaJet;
@@ -1300,22 +1350,24 @@ void AliAnalysisTaskJetSpectrum2::FillMatchHistos(TList &recJetsList,TList &genJ
 
     fhnJetContainer->Fill(containerGen,kStep0); //all generated jets
 
+    if(JetSelected(genJet))
+      fhnJetContainer->Fill(containerGen,kStep1); // all generated jets in eta window
+
     Int_t ir = aRecIndex[ig];
     if(ir>=0&&ir<recJetsList.GetEntries()){   
-      fhnJetContainer->Fill(containerGen,kStep2); // all generated jets with reconstructed partner
-      
-      if(JetSelected(genJet)){
-       fhnJetContainer->Fill(containerGen,kStep1); // all generated jets in eta window
+      AliAODJet* recJet = (AliAODJet*)recJetsList.At(ir); 
 
-       AliAODJet* recJet = (AliAODJet*)recJetsList.At(ir); 
+      fhnJetContainer->Fill(containerGen,kStep2); // all generated jets with reconstructed partner
 
+      if(JetSelected(genJet)){ 
        fhnJetContainer->Fill(containerGen,kStep3); // all generated jets in eta window with reconstructed partner
+       
        if(JetSelected(recJet)) {
          fhnJetContainer->Fill(containerGen,kStep4); // all generated jets in eta window with reconstructed partner in eta window
-
-         containerGen[3] = recJet->GetPtLeading();
-         fhnJetContainer->Fill(containerGen,kStep5); // all generated jets in eta window with reconstructed partner in eta window with leading track on reconstructed level
+         
        }
+       containerGen[3] = recJet->GetPtLeading();
+       fhnJetContainer->Fill(containerGen,kStep5); // all generated jets in eta window with reconstructed partner with leading track on reconstructed level
       }
     }
   }// loop over generated jets used for matching...
@@ -1367,11 +1419,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);
-         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
   }
@@ -1656,6 +1711,24 @@ AliVParticle *AliAnalysisTaskJetSpectrum2::LeadingTrackInCone(AliAODJet* jet,TLi
   return leading;
 }
 
+
+Double_t AliAnalysisTaskJetSpectrum2::RelativePhi(Double_t mphi,Double_t vphi){
+
+  if (vphi < -1*TMath::Pi()) vphi += (2*TMath::Pi());
+  else if (vphi > TMath::Pi()) vphi -= (2*TMath::Pi());
+  if (mphi < -1*TMath::Pi()) mphi += (2*TMath::Pi());
+  else if (mphi > TMath::Pi()) mphi -= (2*TMath::Pi());
+  double dphi = mphi-vphi;
+  if (dphi < -1*TMath::Pi()) dphi += (2*TMath::Pi());
+  else if (dphi > TMath::Pi()) dphi -= (2*TMath::Pi());
+  return dphi;//dphi in [-Pi, Pi]
+}
+
+
+
+
+
+
 Int_t AliAnalysisTaskJetSpectrum2::GetPhiBin(Double_t phi)
 {
   Int_t phibin=-1;