6 #include <TClonesArray.h>
10 #include <THnSparse.h>
13 #include <TLorentzVector.h>
19 #include <AliAnalysisDataSlot.h>
20 #include <AliAnalysisDataContainer.h>
22 #include "TMatrixDSym.h"
23 #include "TMatrixDSymEigen.h"
26 #include "AliVCluster.h"
27 #include "AliVTrack.h"
32 #include "AliMCEvent.h"
33 #include "AliGenPythiaEventHeader.h"
34 #include "AliAODMCHeader.h"
35 #include "AliMCEvent.h"
36 #include "AliAnalysisManager.h"
44 #include "AliAODEvent.h"
56 fMinFractionShared(0),
59 fJetSelection(kInclusive),
65 fangWindowRecoil(0.6),
72 fOneConstSelectOn(kFALSE),
87 fTreeObservableTagging(0)
90 for(
Int_t i=0;i<17;i++){
92 SetMakeGeneralHistograms(kTRUE);
93 DefineOutput(1, TList::Class());
94 DefineOutput(2, TTree::Class());
101 fMinFractionShared(0),
102 fJetShapeType(
kData),
103 fJetShapeSub(kNoSub),
104 fJetSelection(kInclusive),
105 fPtThreshold(-9999.),
110 fangWindowRecoil(0.6),
114 fCentSelectOn(kTRUE),
117 fOneConstSelectOn(kFALSE),
130 fHLundIterative(0x0),
132 fTreeObservableTagging(0)
136 for(
Int_t i=0;i<17;i++){
140 DefineOutput(1, TList::Class());
141 DefineOutput(2, TTree::Class());
157 Bool_t oldStatus = TH1::AddDirectoryStatus();
158 TH1::AddDirectory(kFALSE);
163 fh2ResponseUW=
new TH2F(
"fh2ResponseUW",
"fh2ResponseUW", 100, 0, 200, 100, 0, 200);
165 fh2ResponseW=
new TH2F(
"fh2ResponseW",
"fh2ResponseW", 100, 0, 200, 100, 0, 200);
167 fPhiJetCorr6=
new TH2F(
"fPhiJetCorr6",
"fPhiJetCorr6", 50, 0, 2*TMath::Pi(), 50, 0, 2*TMath::Pi());
169 fEtaJetCorr6=
new TH2F(
"fEtaJetCorr6",
"fEtaJetCorr6", 50, -1.5, 1.5, 50, -1.5, 1.5);
172 fPhiJetCorr7=
new TH2F(
"fPhiJetCorr7",
"fPhiJetCorr7", 50, 0, 2*TMath::Pi(), 50, 0, 2*TMath::Pi());
174 fEtaJetCorr7=
new TH2F(
"fEtaJetCorr7",
"fEtaJetCorr7", 50, -1.5, 1.5, 50, -1.5, 1.5);
177 fPtJetCorr=
new TH2F(
"fPtJetCorr",
"fPtJetCorr", 100, 0, 200, 100, 0, 200);
179 fPtJet=
new TH1F(
"fPtJet",
"fPtJet", 100, 0, 200);
182 fhpTjetpT=
new TH2F(
"fhpTjetpT",
"fhpTjetpT", 200, 0, 200, 200, 0, 200);
184 fhPt=
new TH1F(
"fhPt",
"fhPt", 200, 0, 200);
186 fhPhi=
new TH1F(
"fhPhi",
"fhPhi", 100, -TMath::Pi(), TMath::Pi());
190 const Int_t dimSpec = 5;
191 const Int_t nBinsSpec[5] = {50,50,10,3,10};
192 const Double_t lowBinSpec[5] = {0.0,-10, 0,0,0};
193 const Double_t hiBinSpec[5] = {5.0, 0,200,3,10};
195 "LundIterativePlot [log(1/theta),log(z*theta),pTjet,algo]",
196 dimSpec,nBinsSpec,lowBinSpec,hiBinSpec);
199 fNbOfConstvspT=
new TH2F(
"fNbOfConstvspT",
"fNbOfConstvspT", 100, 0, 100, 200, 0, 200);
209 THnSparse *hn =
dynamic_cast<THnSparse*
>(
fOutput->At(i));
214 TH1::AddDirectory(oldStatus);
216 const char* nameoutput = GetOutputSlot(2)->GetContainer()->GetName();
222 fShapesVarNames[0] =
"partonCode";
223 fShapesVarNames[1] =
"ptJet";
224 fShapesVarNames[2] =
"ptDJet";
225 fShapesVarNames[3] =
"mJet";
227 fShapesVarNames[4] =
"angularity";
229 fShapesVarNames[5] =
"lesub";
232 fShapesVarNames[6] =
"ptJetMatch";
233 fShapesVarNames[7] =
"ptDJetMatch";
234 fShapesVarNames[8] =
"mJetMatch";
236 fShapesVarNames[9] =
"angularityMatch";
238 fShapesVarNames[10] =
"lesubMatch";
240 fShapesVarNames[11]=
"weightPythia";
247 cout<<
"looping over variables"<<endl;
253 delete [] fShapesVarNames;
276 AliAODTrack *triggerHadron = 0x0;
283 if (triggerHadronLabel==-99999) {
298 TClonesArray *TrackArray = NULL;
299 TClonesArray *TrackArrayMC = NULL;
301 else TrackArray = PartCont->GetArray();
303 else triggerHadron =
static_cast<AliAODTrack*
>(TrackArray->At(triggerHadronLabel));
309 if (!triggerHadron) {
320 fhPt->Fill(triggerHadron->Pt());
326 TClonesArray *trackArrayAn = partContAn->GetArray();
327 Int_t ntracksEvt = trackArrayAn->GetEntriesFast();
329 Float_t rhoVal=0, rhoMassVal = 0.;
332 jetCont->ResetCurrentID();
337 Printf(
"%s: Could not retrieve rho %s (some histograms will be filled with zero)!", GetName(), jetCont->
GetRhoName().Data());
338 }
else rhoVal = rhoParam->GetVal();
342 Printf(
"%s: Could not retrieve rho_m %s (some histograms will be filled with zero)!", GetName(), jetCont->
GetRhoMassName().Data());
343 }
else rhoMassVal = rhomParam->GetVal();
352 Int_t ifound=0, jfound=0;
353 Int_t ilab=-1, jlab=-1;
383 jetUS = jetContUS->
GetJet(i);
386 if(ifound==1) ilab = i;
389 if(ilab==-1)
continue;
390 jetUS=jetContUS->
GetJet(ilab);
396 Printf(
"jet2 does not exist, returning");
404 Printf(
"jet3 does not exist, returning");
407 cout<<
"jet 3 exists"<<jet3->
Pt()<<endl;
433 jetUS = jetContUS->
GetJet(i);
436 if(ifound==1) ilab = i;
439 if(ilab==-1)
continue;
440 jetUS=jetContUS->
GetJet(ilab);
444 Printf(
"jet2 does not exist, returning");
450 jet3 = jetContPart->
GetJet(j);
454 if(jfound==1) jlab = j;
457 if(jlab==-1)
continue;
458 jet3=jetContPart->
GetJet(jlab);
460 Printf(
"jet3 does not exist, returning");
466 Printf(
"jet3 does not exist, returning");
485 Float_t dRp1 = TMath::Sqrt(jp1 * jp1 + detap1 * detap1);
495 dRp1 = TMath::Sqrt(jp1 * jp1 + detap1 * detap1);
539 Float_t ptMatch=0., ptDMatch=0., massMatch=0., constMatch=0.,angulMatch=0.,circMatch=0., lesubMatch=0., sigma2Match=0., coronnaMatch=0;
634 AliVParticle *vp1 = 0x0;
639 Printf(
"AliVParticle associated to constituent not found");
644 Double_t dr2 = (vp1->Eta()-jet->
Eta())*(vp1->Eta()-jet->
Eta()) + dphi*dphi;
646 num=num+vp1->Pt()*dr;
680 TClonesArray *TracksArray = NULL;
681 TClonesArray *TracksArrayMC = NULL;
684 else TracksArray = PartCont->GetArray();
687 if(!PartContMC || !TracksArrayMC)
return -2;
690 if(!PartCont || !TracksArray)
return -2;
694 AliAODTrack *Track = 0x0;
698 else NTracks = TracksArray->GetEntriesFast();
700 for(
Int_t i=0; i < NTracks; i++){
703 if (!Track)
continue;
704 if(TMath::Abs(Track->Eta())>0.9)
continue;
706 Double_t dr2 = (Track->Eta()-jet->
Eta())*(Track->Eta()-jet->
Eta()) + dphi*dphi;
708 if((dr>=0.8) && (dr<1)) sumpt=sumpt+Track->Pt();
712 if((Track = static_cast<AliAODTrack*>(PartCont->
GetAcceptTrack(i)))){
713 if (!Track)
continue;
714 if(TMath::Abs(Track->Eta())>0.9)
continue;
715 if (Track->Pt()<0.15)
continue;
717 Double_t dr2 = (Track->Eta()-jet->
Eta())*(Track->Eta()-jet->
Eta()) + dphi*dphi;
719 if((dr>=0.8) && (dr<1)) sumpt=sumpt+Track->Pt();
739 return Coronna(jet, jetContNb);
755 AliVParticle *vp1 = 0x0;
760 Printf(
"AliVParticle associated to constituent not found");
764 num=num+vp1->Pt()*vp1->Pt();
767 return TMath::Sqrt(num)/den;
777 return PTD(jet, jetContNb);
798 TVector3 ppJ1(pxjet, pyjet, pzjet);
799 TVector3 ppJ3(- pxjet* pzjet, - pyjet * pzjet, pxjet * pxjet + pyjet * pyjet);
801 TVector3 ppJ2(-pyjet, pxjet, 0);
803 AliVParticle *vp1 = 0x0;
808 Printf(
"AliVParticle associated to constituent not found");
812 TVector3 pp(vp1->Px(), vp1->Py(), vp1->Pz());
815 TVector3 pLong = pp.Dot(ppJ1) / ppJ1.Mag2() * ppJ1;
816 TVector3 pPerp = pp - pLong;
819 Float_t ppjX = pPerp.Dot(ppJ2);
820 Float_t ppjY = pPerp.Dot(ppJ3);
821 Float_t ppjT = TMath::Sqrt(ppjX * ppjX + ppjY * ppjY);
822 if(ppjT<=0)
return 0;
824 mxx += (ppjX * ppjX / ppjT);
825 myy += (ppjY * ppjY / ppjT);
826 mxy += (ppjX * ppjY / ppjT);
831 if(sump2==0)
return 0;
833 Double_t ele[4] = {mxx / sump2, mxy / sump2, mxy / sump2, myy / sump2};
834 TMatrixDSym m0(2,ele);
837 TMatrixDSymEigen m(m0);
839 TMatrixD evecm = m.GetEigenVectors();
840 eval = m.GetEigenValues();
844 if (eval[0] < eval[1]) jev = 1;
847 evec0 = TMatrixDColumn(evecm, jev);
850 TVector2 evec(compx, compy);
852 if(jev==1) circ=2*eval[0];
853 if(jev==0) circ=2*eval[1];
884 AliVParticle *vp1 = 0x0;
885 AliVParticle *vp2 = 0x0;
886 std::vector<int> ordindex;
891 if(ordindex.size()<2)
return -1;
895 Printf(
"AliVParticle associated to Leading constituent not found");
901 Printf(
"AliVParticle associated to Subleading constituent not found");
921 return LeSub(jet, jetContNb);
950 AliVParticle *vp1 = 0x0;
955 Printf(
"AliVParticle associated to constituent not found");
963 mxx += ppt*ppt*deta*deta;
964 myy += ppt*ppt*dphi*dphi;
965 mxy -= ppt*ppt*deta*TMath::Abs(dphi);
971 if(sump2==0)
return 0;
973 Double_t ele[4] = {mxx , mxy , mxy , myy };
974 TMatrixDSym m0(2,ele);
977 TMatrixDSymEigen m(m0);
979 TMatrixD evecm = m.GetEigenVectors();
980 eval = m.GetEigenValues();
984 if (eval[0] < eval[1]) jev = 1;
987 evec0 = TMatrixDColumn(evecm, jev);
990 TVector2 evec(compx, compy);
992 if(jev==1) sig=TMath::Sqrt(TMath::Abs(eval[0])/sump2);
993 if(jev==0) sig=TMath::Sqrt(TMath::Abs(eval[1])/sump2);
1007 return Sigma2(jet, jetContNb);
1027 TClonesArray *TracksArray = NULL;
1028 TClonesArray *TracksArrayMC = NULL;
1031 else TracksArray = PartCont->GetArray();
1034 if(!PartContMC || !TracksArrayMC)
return -99999;
1037 if(!PartCont || !TracksArray)
return -99999;
1041 AliAODTrack *Track = 0x0;
1046 Int_t triggers[100];
1047 for (
Int_t iTrigger=0; iTrigger<100; iTrigger++) triggers[iTrigger] = 0;
1051 else NTracks = TracksArray->GetEntriesFast();
1053 for(
Int_t i=0; i < NTracks; i++){
1056 if (!Track)
continue;
1057 if(TMath::Abs(Track->Eta())>0.9)
continue;
1058 if (Track->Pt()<0.15)
continue;
1059 if ((Track->Pt() >= minpT) && (Track->Pt()< maxpT)) {
1066 if((Track = static_cast<AliAODTrack*>(PartCont->
GetAcceptTrack(i)))){
1067 if (!Track)
continue;
1068 if(TMath::Abs(Track->Eta())>0.9)
continue;
1069 if (Track->Pt()<0.15)
continue;
1070 if ((Track->Pt() >= minpT) && (Track->Pt()< maxpT)) {
1079 if (iTT == 0)
return -99999;
1080 Int_t nbRn = 0, index = 0 ;
1081 TRandom3* random =
new TRandom3(0);
1082 nbRn = random->Integer(iTT);
1083 index = triggers[nbRn];
1092 if (vphi < -1*TMath::Pi()) vphi += (2*TMath::Pi());
1093 else if (vphi > TMath::Pi()) vphi -= (2*TMath::Pi());
1094 if (mphi < -1*TMath::Pi()) mphi += (2*TMath::Pi());
1095 else if (mphi > TMath::Pi()) mphi -= (2*TMath::Pi());
1096 double dphi = mphi-vphi;
1097 if (dphi < -1*TMath::Pi()) dphi += (2*TMath::Pi());
1098 else if (dphi > TMath::Pi()) dphi -= (2*TMath::Pi());
1106 std::vector<fastjet::PseudoJet> fInputVectors;
1107 fInputVectors.clear();
1108 fastjet::PseudoJet PseudoTracks;
1113 AliVParticle *fTrk = fJet->
TrackAt(i, fTrackCont->GetArray());
1114 if (!fTrk)
continue;
1115 PseudoTracks.reset(fTrk->Px(), fTrk->Py(), fTrk->Pz(),fTrk->E());
1116 PseudoTracks.set_user_index(fJet->
TrackAt(i)+100);
1117 fInputVectors.push_back(PseudoTracks);
1120 fastjet::JetAlgorithm jetalgo(fastjet::antikt_algorithm);
1122 if(ReclusterAlgo==0){ xflagalgo=0.5;
1123 jetalgo=fastjet::kt_algorithm ;}
1125 if(ReclusterAlgo==1){ xflagalgo=1.5;
1126 jetalgo=fastjet::cambridge_algorithm;}
1127 if(ReclusterAlgo==2){ xflagalgo=2.5;
1128 jetalgo=fastjet::antikt_algorithm;}
1130 fastjet::JetDefinition fJetDef(jetalgo, 1., static_cast<fastjet::RecombinationScheme>(0), fastjet::BestFJ30 );
1133 fastjet::ClusterSequence fClustSeqSA(fInputVectors, fJetDef);
1134 std::vector<fastjet::PseudoJet> fOutputJets;
1135 fOutputJets.clear();
1136 fOutputJets=fClustSeqSA.inclusive_jets(0);
1138 fastjet::PseudoJet jj;
1139 fastjet::PseudoJet j1;
1140 fastjet::PseudoJet j2;
1143 while(jj.has_parents(j1,j2)){
1145 if(j1.perp() < j2.perp())
swap(j1,j2);
1146 double delta_R=j1.delta_R(j2);
1147 double z=j2.perp()/(j1.perp()+j2.perp());
1148 double y =log(1.0/delta_R);
1149 double lnpt_rel=log(z*delta_R);
1150 Double_t LundEntries[5] = {y,lnpt_rel,fOutputJets[0].perp(),xflagalgo,ndepth};
1157 }
catch (fastjet::Error) {
1158 AliError(
" [w] FJ Exception caught.");
Int_t SelectTrigger(Float_t minpT, Float_t maxpT)
Float_t Coronna(AliEmcalJet *jet, Int_t jetContNb)
JetShapeType fJetShapeType
Double_t GetFirstOrderSubtractedAngularity() const
const TString & GetRhoName() const
Double_t GetSecondOrderSubtractedSigma2() const
Bool_t RetrieveEventObjects()
Retrieve common objects from event.
Float_t GetPartonEta6() const
AliEmcalJet * ClosestJet() const
virtual ~AliAnalysisTaskEmcalQGTagging()
AliJetContainer * GetJetContainer(Int_t i=0) const
Bool_t FillHistograms()
Function filling histograms.
Float_t GetPartonEta7() const
Double_t GetSecondOrderSubtractedConstituent() const
Float_t GetJetMass(AliEmcalJet *jet, Int_t jetContNb)
Float_t GetPythiaEventWeight() const
Container with name, TClonesArray and cuts for particles.
void UserCreateOutputObjects()
THnSparse * fHLundIterative
Float_t GetPartonPhi7() const
Double_t GetSecondOrderSubtractedpTD() const
Bool_t RetrieveEventObjects()
Double_t GetFirstOrderSubtractedLeSub() const
Float_t GetJetAngularity(AliEmcalJet *jet, Int_t jetContNb)
Int_t GetPartonFlag7() const
Container for particles within the EMCAL framework.
Float_t GetPartonPhi6() const
Int_t TrackAt(Int_t idx) const
Double_t GetFirstOrderSubtractedpTD() const
UShort_t GetNumberOfTracks() const
Float_t GetJetCoronna(AliEmcalJet *jet, Int_t jetContNb)
TString kData
Declare data MC or deltaAOD.
Float_t fMinFractionShared
AliParticleContainer * GetParticleContainer(Int_t i=0) const
Get particle container attached to this task.
const TString & GetRhoMassName() const
Double_t GetSecondOrderSubtractedLeSub() const
Int_t GetPartonFlag6() const
AliParticleContainer * GetParticleContainer() const
Double_t GetFirstOrderSubtractedConstituent() const
Float_t GetJetCircularity(AliEmcalJet *jet, Int_t jetContNb)
Float_t PTD(AliEmcalJet *jet, Int_t jetContNb)
Double_t GetSecondOrderSubtractedAngularity() const
Float_t GetSigma2(AliEmcalJet *jet, Int_t jetContNb)
virtual AliVParticle * GetAcceptParticle(Int_t i=-1) const
Double_t fCent
!event centrality
void RecursiveParents(AliEmcalJet *fJet, AliJetContainer *fJetCont, Int_t ReclusterAlgo)
Double_t GetSecondOrderSubtracted() const
AliEmcalJet * GetNextAcceptJet()
Float_t Angularity(AliEmcalJet *jet, Int_t jetContNb)
Bool_t Data(TH1F *h, Double_t *rangefit, Bool_t writefit, Double_t &sgn, Double_t &errsgn, Double_t &bkg, Double_t &errbkg, Double_t &sgnf, Double_t &errsgnf, Double_t &sigmafit, Int_t &status)
Float_t LeSub(AliEmcalJet *jet, Int_t jetContNb)
AliAnalysisTaskEmcalQGTagging()
Double_t GetRhoVal(Int_t i=0) const
Double_t GetFirstOrderSubtractedCircularity() const
AliEmcalList * fOutput
!output list
JetSelectionType fJetSelection
Float_t Sigma2(AliEmcalJet *jet, Int_t jetContNb)
Double_t RelativePhi(Double_t mphi, Double_t vphi)
void Terminate(Option_t *option)
Double_t GetSecondOrderSubtractedCircularity() const
Float_t GetJetNumberOfConstituents(AliEmcalJet *jet, Int_t jetContNb)
AliTrackContainer * GetTrackContainer(Int_t i=0) const
Store some informaion about a Pythia eventThis class is used to store some information about a Pythia...
void SetMakeGeneralHistograms(Bool_t g)
virtual AliVTrack * GetAcceptTrack(Int_t i=-1) const
Base task in the EMCAL jet framework.
Represent a jet reconstructed using the EMCal jet framework.
const AliEmcalPythiaInfo * GetPythiaInfo() const
void swap(AliEmcalContainerIndexMap< X, Y > &first, AliEmcalContainerIndexMap< X, Y > &second)
Float_t GetJetpTD(AliEmcalJet *jet, Int_t jetContNb)
Double_t GetFractionSharedPt(const AliEmcalJet *jet, AliParticleContainer *cont2=0x0) const
Declaration of class AliEmcalPythiaInfo.
void UserCreateOutputObjects()
Main initialization function on the worker.
Double_t GetFirstOrderSubtractedSigma2() const
Float_t Circularity(AliEmcalJet *jet, Int_t jetContNb)
AliEmcalJetShapeProperties * GetShapeProperties() const
Float_t GetJetLeSub(AliEmcalJet *jet, Int_t jetContNb)
Float_t GetPartonPt7() const
Double_t GetFirstOrderSubtracted() const
Container for jet within the EMCAL jet framework.
Bool_t Run()
Run function. This is the core function of the analysis and contains the user code. Therefore users have to implement this function.
std::vector< int > GetPtSortedTrackConstituentIndexes(TClonesArray *tracks) const
Float_t GetPartonPt6() const
AliEmcalJet * GetJet(Int_t i) const
TTree * fTreeObservableTagging