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"
42 #include "AliAODEvent.h"
54 fMinFractionShared(0),
55 fJetShapeType(kGenShapes),
57 fJetSelection(kInclusive),
65 fangWindowRecoil(0.6),
72 fOneConstSelectOn(kFALSE),
84 fTreeObservableTagging(0x0)
87 for(
Int_t i=0;i<33;i++){
89 SetMakeGeneralHistograms(kTRUE);
96 fMinFractionShared(0),
97 fJetShapeType(kGenShapes),
99 fJetSelection(kInclusive),
100 fPtThreshold(-9999.),
105 fangWindowRecoil(0.6),
109 fCentSelectOn(kTRUE),
112 fOneConstSelectOn(kFALSE),
124 fTreeObservableTagging(0x0)
130 for(
Int_t i=0;i<33;i++){
135 DefineOutput(1, TList::Class());
136 DefineOutput(2, TTree::Class());
157 Bool_t oldStatus = TH1::AddDirectoryStatus();
158 TH1::AddDirectory(oldStatus);
164 const char* nameoutput = GetOutputSlot(2)->GetContainer()->GetName();
171 fShapesVarNames[0] =
"partonCode";
172 fShapesVarNames[1] =
"ptJet";
173 fShapesVarNames[2] =
"ptDJet";
174 fShapesVarNames[3] =
"mJet";
175 fShapesVarNames[4] =
"nbOfConst";
176 fShapesVarNames[5] =
"angularity";
177 fShapesVarNames[6] =
"Nsubjet1kt";
178 fShapesVarNames[7] =
"Nsubjet2kt";
179 fShapesVarNames[8] =
"Nsubjet1Min";
180 fShapesVarNames[9] =
"Nsubjet2Min";
181 fShapesVarNames[10] =
"DeltaRkt";
182 fShapesVarNames[11] =
"DeltaRMin";
183 fShapesVarNames[12] =
"SDSymm";
184 fShapesVarNames[13] =
"SDDeltaR";
185 fShapesVarNames[14] =
"SDGroomedFrac";
186 fShapesVarNames[15] =
"SDGroomedN";
187 fShapesVarNames[16] =
"SDSymmkt";
188 fShapesVarNames[17] =
"SDDeltaRkt";
189 fShapesVarNames[18] =
"SDGroomedFrackt";
190 fShapesVarNames[19] =
"SDGroomedNkt";
191 fShapesVarNames[20] =
"SDSymmAkt";
192 fShapesVarNames[21] =
"SDDeltaRAkt";
193 fShapesVarNames[22] =
"SDGroomedFracAkt";
194 fShapesVarNames[23] =
"SDGroomedNAkt";
195 fShapesVarNames[24] =
"SDSymmBeta1";
196 fShapesVarNames[25] =
"SDDeltaRBeta1";
197 fShapesVarNames[26] =
"SDGroomedFracBeta1";
198 fShapesVarNames[27] =
"SDGroomedNBeta1";
199 fShapesVarNames[28] =
"SDSymmBeta2";
200 fShapesVarNames[29] =
"SDDeltaRBeta2";
201 fShapesVarNames[30] =
"SDGroomedFracBeta2";
202 fShapesVarNames[31] =
"SDGroomedNBeta2";
203 fShapesVarNames[32] =
"weightPythia";
228 cout<<
"looping over variables"<<endl;
233 fPhiJetCorr6=
new TH2F(
"fPhiJetCorr6",
"fPhiJetCorr6", 50, 0, 2*TMath::Pi(), 50, 0, 2*TMath::Pi());
235 fEtaJetCorr6=
new TH2F(
"fEtaJetCorr6",
"fEtaJetCorr6", 50, -1.5, 1.5, 50, -1.5, 1.5);
238 fPhiJetCorr7=
new TH2F(
"fPhiJetCorr7",
"fPhiJetCorr7", 50, 0, 2*TMath::Pi(), 50, 0, 2*TMath::Pi());
240 fEtaJetCorr7=
new TH2F(
"fEtaJetCorr7",
"fEtaJetCorr7", 50, -1.5, 1.5, 50, -1.5, 1.5);
243 fPtJetCorr=
new TH2F(
"fPtJetCorr",
"fPtJetCorr", 100, 0, 200, 100, 0, 200);
245 fPtJet=
new TH1F(
"fPtJet",
"fPtJet", 100, 0, 200);
248 fhpTjetpT=
new TH2F(
"fhpTjetpT",
"fhpTjetpT", 200, 0, 200, 200, 0, 200);
250 fhPt=
new TH1F(
"fhPt",
"fhPt", 200, 0, 200);
252 fhPhi=
new TH1F(
"fhPhi",
"fhPhi", 100, -TMath::Pi(), TMath::Pi());
255 fNbOfConstvspT=
new TH2F(
"fNbOfConstvspT",
"fNbOfConstvspT", 100, 0, 100, 200, 0, 200);
264 delete [] fShapesVarNames;
288 AliAODTrack *triggerHadron = 0x0;
295 if (triggerHadronLabel==-99999) {
301 TClonesArray *trackArrayAn = partContAn->GetArray();
302 triggerHadron =
static_cast<AliAODTrack*
>(trackArrayAn->At(triggerHadronLabel));
304 if (!triggerHadron) {
315 fhPt->Fill(triggerHadron->Pt());
320 jetCont->ResetCurrentID();
328 Int_t ifound=0, jfound=0;
329 Int_t ilab=-1, jlab=-1;
365 Float_t dRp1 = TMath::Sqrt(jp1 * jp1 + detap1 * detap1);
375 dRp1 = TMath::Sqrt(jp1 * jp1 + detap1 * detap1);
387 ptSubtracted= jet1->
Pt();
466 AliVParticle *vp1 = 0x0;
471 Printf(
"AliVParticle associated to constituent not found");
476 Double_t dr2 = (vp1->Eta()-jet->
Eta())*(vp1->Eta()-jet->
Eta()) + dphi*dphi;
478 num=num+vp1->Pt()*dr;
504 AliVParticle *vp1 = 0x0;
509 Printf(
"AliVParticle associated to constituent not found");
513 num=num+vp1->Pt()*vp1->Pt();
516 return TMath::Sqrt(num)/den;
526 return PTD(jet, jetContNb);
547 TVector3 ppJ1(pxjet, pyjet, pzjet);
548 TVector3 ppJ3(- pxjet* pzjet, - pyjet * pzjet, pxjet * pxjet + pyjet * pyjet);
550 TVector3 ppJ2(-pyjet, pxjet, 0);
552 AliVParticle *vp1 = 0x0;
557 Printf(
"AliVParticle associated to constituent not found");
561 TVector3 pp(vp1->Px(), vp1->Py(), vp1->Pz());
564 TVector3 pLong = pp.Dot(ppJ1) / ppJ1.Mag2() * ppJ1;
565 TVector3 pPerp = pp - pLong;
568 Float_t ppjX = pPerp.Dot(ppJ2);
569 Float_t ppjY = pPerp.Dot(ppJ3);
570 Float_t ppjT = TMath::Sqrt(ppjX * ppjX + ppjY * ppjY);
571 if(ppjT<=0)
return 0;
573 mxx += (ppjX * ppjX / ppjT);
574 myy += (ppjY * ppjY / ppjT);
575 mxy += (ppjX * ppjY / ppjT);
580 if(sump2==0)
return 0;
582 Double_t ele[4] = {mxx / sump2, mxy / sump2, mxy / sump2, myy / sump2};
583 TMatrixDSym m0(2,ele);
586 TMatrixDSymEigen m(m0);
588 TMatrixD evecm = m.GetEigenVectors();
589 eval = m.GetEigenValues();
593 if (eval[0] < eval[1]) jev = 1;
596 evec0 = TMatrixDColumn(evecm, jev);
599 TVector2 evec(compx, compy);
601 if(jev==1) circ=2*eval[0];
602 if(jev==0) circ=2*eval[1];
633 AliVParticle *vp1 = 0x0;
634 AliVParticle *vp2 = 0x0;
635 std::vector<int> ordindex;
640 if(ordindex.size()<2)
return -1;
644 Printf(
"AliVParticle associated to Leading constituent not found");
650 Printf(
"AliVParticle associated to Subleading constituent not found");
670 return LeSub(jet, jetContNb);
697 const AliVVertex *vert = InputEvent()->GetPrimaryVertex();
701 if(Reclusterer->AliEmcalJetFinder::Filter(Jet, JetCont, dVtx)){;}
714 AliVParticle *JetParticle=0x0;
716 Double_t SubJetiness_Denominator = 0;
720 if(Index==-999)
return -2;
721 SubJetiness_Numerator=(Reclusterer->
GetJet(Index)->
Pt());
722 SubJetiness_Denominator=Jet->
Pt();
723 return SubJetiness_Numerator/SubJetiness_Denominator;
735 AliVParticle *vp1 = 0x0;
740 Printf(
"AliVParticle associated to constituent not found");
745 Double_t dr2 = (vp1->Eta()-jet->
Eta())*(vp1->Eta()-jet->
Eta()) + dphi*dphi;
750 return num/jet->
Pt();
775 Int_t ArraySize =N+1;
778 for (
Int_t i=0; i<ArraySize; i++){
779 JetSorter->SetAt(0,i);
781 for (
Int_t i=0; i<ArraySize; i++){
782 JetIndexSorter->SetAt(0,i);
786 SubJet=Reclusterer->
GetJet(i);
787 if (Type==0) SortingVariable=SubJet->
Pt();
788 else if (Type==1) SortingVariable=SubJet->
E();
789 else if (Type==2) SortingVariable=SubJet->
M();
790 for (
Int_t j=0; j<N; j++){
791 if (SortingVariable>JetSorter->GetAt(j)){
792 for (
Int_t k=N-1; k>=j; k--){
793 JetSorter->SetAt(JetSorter->GetAt(k),k+1);
794 JetIndexSorter->SetAt(JetIndexSorter->GetAt(k),k+1);
796 JetSorter->SetAt(SortingVariable,j);
797 JetIndexSorter->SetAt(i,j);
802 if (!Index)
return JetSorter->GetAt(N-1);
803 else return JetIndexSorter->GetAt(N-1);
828 AliVParticle *vp1 = 0x0;
833 Printf(
"AliVParticle associated to constituent not found");
841 mxx += ppt*ppt*deta*deta;
842 myy += ppt*ppt*dphi*dphi;
843 mxy -= ppt*ppt*deta*TMath::Abs(dphi);
849 if(sump2==0)
return 0;
851 Double_t ele[4] = {mxx , mxy , mxy , myy };
852 TMatrixDSym m0(2,ele);
855 TMatrixDSymEigen m(m0);
857 TMatrixD evecm = m.GetEigenVectors();
858 eval = m.GetEigenValues();
862 if (eval[0] < eval[1]) jev = 1;
865 evec0 = TMatrixDColumn(evecm, jev);
868 TVector2 evec(compx, compy);
870 if(jev==1) sig=TMath::Sqrt(TMath::Abs(eval[0])/sump2);
871 if(jev==0) sig=TMath::Sqrt(TMath::Abs(eval[1])/sump2);
885 return Sigma2(jet, jetContNb);
899 AliVParticle *vp1 = 0x0;
900 std::vector<int> ordindex;
910 Printf(
"AliVParticle associated to Leading constituent not found");
914 if (nTFractions[0] <= 0.7*ptJet){
915 nTFractions[0] += vp1->Pt();
919 if (nTFractions[2] <= 0.8*ptJet){
920 nTFractions[2] += vp1->Pt();
924 if (nTFractions[4] <= 0.9*ptJet){
925 nTFractions[4] += vp1->Pt();
929 if (nTFractions[6] <= 0.95*ptJet){
930 nTFractions[6] += vp1->Pt();
968 return JetFinder->
Nsubjettiness(Jet,JetCont,dVtx,N,Algorithm,0.2,Beta,Option,0,0,0);
980 TClonesArray *tracksArray = partCont->GetArray();
982 if(!partCont || !tracksArray)
return -99999;
983 AliAODTrack *track = 0x0;
989 for (
Int_t iTrigger=0; iTrigger<100; iTrigger++) triggers[iTrigger] = 0;
992 for(
Int_t iTrack=0; iTrack <= tracksArray->GetEntriesFast(); iTrack++){
997 if (!emcPart)
continue;
998 if(TMath::Abs(emcPart->
Eta())>0.9)
continue;
999 if (emcPart->
Pt()<0.15)
continue;
1001 if ((emcPart->
Pt() >= minpT) && (emcPart->
Pt()< maxpT)) {
1002 trackList->Add(emcPart);
1003 triggers[iTT] = iTrack;
1008 track =
static_cast<AliAODTrack*
>(tracksArray->At(iTrack));
1009 if (!track)
continue;
1010 if(TMath::Abs(track->Eta())>0.9)
continue;
1011 if (track->Pt()<0.15)
continue;
1012 if (!(track->TestFilterBit(768)))
continue;
1014 if ((track->Pt() >= minpT) && (track->Pt()< maxpT)) {
1015 trackList->Add(track);
1016 triggers[iTT] = iTrack;
1023 if (iTT == 0)
return -99999;
1024 Int_t nbRn = 0, index = 0 ;
1025 TRandom3* random =
new TRandom3(0);
1026 nbRn = random->Integer(iTT);
1028 index = triggers[nbRn];
1037 if (vphi < -1*TMath::Pi()) vphi += (2*TMath::Pi());
1038 else if (vphi > TMath::Pi()) vphi -= (2*TMath::Pi());
1039 if (mphi < -1*TMath::Pi()) mphi += (2*TMath::Pi());
1040 else if (mphi > TMath::Pi()) mphi -= (2*TMath::Pi());
1041 double dphi = mphi-vphi;
1042 if (dphi < -1*TMath::Pi()) dphi += (2*TMath::Pi());
1043 else if (dphi > TMath::Pi()) dphi -= (2*TMath::Pi());
1064 AliInfo(
"Terminate");
1065 AliAnalysisTaskSE::Terminate();
1069 AliError(
"fOutput not available");
1075 Printf(
"ERROR: fTreeObservableTagging not available");
1084 std::vector<fastjet::PseudoJet> fInputVectors;
1085 Double_t JetInvMass=0, PseudJetInvMass=0, TrackMom = 0, TrackEnergy = 0;
1090 Double_t FJTrackEta[9999],FJTrackPhi[9999],FJTrackPt[9999],EmcalJetTrackEta[9999],EmcalJetTrackPhi[9999],EmcalJetTrackPt[9999];
1091 UShort_t FJNTracks=0,EmcalJetNTracks=0;
1094 AliVParticle *fTrk = fJet->
TrackAt(i, fTrackCont->GetArray());
1095 if (!fTrk)
continue;
1096 JetInvMass += fTrk->M();
1098 fastjet::PseudoJet PseudoTracks(fTrk->Px(), fTrk->Py(), fTrk->Pz(),fTrk->E());
1099 TrackMom += TMath::Sqrt(TMath::Power(fTrk->Px(),2)+TMath::Power(fTrk->Py(),2)+TMath::Power(fTrk->Pz(),2));
1100 TrackEnergy += fTrk->E();
1101 PseudoTracks.set_user_index(fJet->
TrackAt(i)+100);
1102 PseudJetInvMass += PseudoTracks.m();
1103 fInputVectors.push_back(PseudoTracks);
1104 EmcalJetTrackEta[i]=fTrk->Eta();
1105 EmcalJetTrackPhi[i]=fTrk->Phi();
1106 EmcalJetTrackPt[i]=fTrk->Pt();
1109 fastjet::JetDefinition *fJetDef;
1110 fastjet::ClusterSequence *fClustSeqSA;
1114 fJetDef =
new fastjet::JetDefinition(fastjet::antikt_algorithm,
fJetRadius*2, static_cast<fastjet::RecombinationScheme>(0), fastjet::BestFJ30 );
1117 fClustSeqSA =
new fastjet::ClusterSequence(fInputVectors, *fJetDef);
1118 }
catch (fastjet::Error) {
1119 AliError(
" [w] FJ Exception caught.");
1123 std::vector<fastjet::PseudoJet> fOutputJets;
1124 fOutputJets.clear();
1125 fOutputJets=fClustSeqSA->inclusive_jets(0);
1128 std::vector<fastjet::PseudoJet> jet_constituents = fOutputJets[0].constituents();
1129 fastjet::contrib::SoftDrop softdrop(beta, zcut);
1131 softdrop.set_verbose_structure(kTRUE);
1134 fastjet::contrib::Recluster *recluster;
1135 if(ReclusterAlgo == 1) recluster =
new fastjet::contrib::Recluster(fastjet::kt_algorithm,1,
true);
1136 if(ReclusterAlgo == 2) recluster =
new fastjet::contrib::Recluster(fastjet::antikt_algorithm,1,
true);
1137 if(ReclusterAlgo == 0) recluster =
new fastjet::contrib::Recluster(fastjet::cambridge_algorithm,1,
true);
1138 softdrop.set_reclustering(
true,recluster);
1139 fastjet::PseudoJet finaljet = softdrop(fOutputJets[0]);
1163 Int_t NGroomedBranches;
1164 SymParam=(finaljet.structure_of<fastjet::contrib::SoftDrop>().symmetry());
1165 Mu=(finaljet.structure_of<fastjet::contrib::SoftDrop>().mu());
1166 DeltaR=(finaljet.structure_of<fastjet::contrib::SoftDrop>().delta_R());
1167 NGroomedBranches=finaljet.structure_of<fastjet::contrib::SoftDrop>().dropped_count();
1168 GroomedPt=finaljet.perp();
1170 if(ReclusterAlgo==0){
1175 if(ReclusterAlgo==1){
1181 if(ReclusterAlgo==2){
void SetRadius(Double_t val)
Float_t GetSigma2(AliEmcalJet *jet, Int_t jetContNb=0)
Double_t GetFirstOrderSubtractedAngularity() const
Float_t GetJetMass(AliEmcalJet *jet, Int_t jetContNb=0)
void SetJetMinPt(Double_t val)
Double_t GetSecondOrderSubtractedSigma2() const
Float_t GetPartonEta6() const
AliJetContainer * GetJetContainer(Int_t i=0) const
Float_t GetPartonEta7() const
Double_t GetSecondOrderSubtractedConstituent() const
ClassImp(AliAnalysisTaskEmcalJetShapesMC) AliAnalysisTaskEmcalJetShapesMC
Float_t GetPythiaEventWeight() const
Float_t GetPartonPhi7() const
JetShapeType fJetShapeType
Double_t GetSecondOrderSubtractedpTD() const
Bool_t RetrieveEventObjects()
Double_t Nsubjettiness(AliEmcalJet *pJet, AliJetContainer *pContJets, Double_t dVtx[3], Int_t N, Int_t Algorithm, Double_t Radius, Double_t Beta, Int_t Option=0, Int_t Measure=0, Double_t Beta_SD=0, Double_t ZCut=0.1)
Double_t GetFirstOrderSubtractedLeSub() const
Float_t CoreFrac(AliEmcalJet *jet, Int_t jetContNb=0)
Float_t GetJetCircularity(AliEmcalJet *jet, Int_t jetContNb=0)
Float_t GetJetpTD(AliEmcalJet *jet, Int_t jetContNb=0)
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 GetJetCoreFrac(AliEmcalJet *jet, Int_t jetContNb=0)
AliParticleContainer * GetParticleContainer(Int_t i=0) const
Get particle container attached to this task.
Double_t fjNSubJettiness(AliEmcalJet *Jet, Int_t JetContNb, Int_t N, Int_t Algorithm, Double_t Beta, Int_t Option)
Double_t GetSecondOrderSubtractedLeSub() const
Int_t GetPartonFlag6() const
AliParticleContainer * GetParticleContainer() const
Double_t GetFirstOrderSubtractedConstituent() const
void SetRecombSheme(Int_t val)
void SetJetAlgorithm(Int_t val)
Float_t GetJetNumberOfConstituents(AliEmcalJet *jet, Int_t jetContNb=0)
Task to store and correlate the MC shapes.
TTree * fTreeObservableTagging
! Tree with tagging variables subtracted MC or true MC or raw
Double_t RelativePhi(Double_t mphi, Double_t vphi)
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.
Double_t GetSecondOrderSubtractedAngularity() const
Double_t GetSubjetFraction(AliEmcalJet *Jet, Int_t JetContNb, Double_t JetRadius, AliEmcalJetFinder *Reclusterer)
void SoftDrop(AliEmcalJet *fJet, AliJetContainer *fJetCont, double zcut, double beta, Int_t ReclusterAlgo)
AliAnalysisTaskEmcalJetShapesMC()
AliEmcalJetFinder * Recluster(AliEmcalJet *Jet, Int_t JetContNb, Double_t JetRadius, Double_t SubJetRadius, Double_t SubJetMinPt, Int_t Algorithm, const char *Name)
Double_t fCent
!event centrality
Double_t GetSecondOrderSubtracted() const
Float_t PTD(AliEmcalJet *jet, Int_t jetContNb=0)
void SetJetMaxEta(Double_t val)
void Terminate(Option_t *option)
AliEmcalJet * GetNextAcceptJet()
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 GetJetAngularity(AliEmcalJet *jet, Int_t jetContNb=0)
Int_t SelectTrigger(Float_t minpT, Float_t maxpT)
AliEmcalJet * GetJet(Int_t index)
Enhanced TList-derived class that implements correct merging for pt_hard binned production.
Float_t Angularity(AliEmcalJet *jet, Int_t jetContNb=0)
Bool_t FillHistograms()
Function filling histograms.
Double_t GetFirstOrderSubtractedCircularity() const
JetSelectionType fJetSelection
Bool_t RetrieveEventObjects()
Retrieve common objects from event.
AliEmcalList * fOutput
!output list
Double_t SubJetOrdering(AliEmcalJet *Jet, AliEmcalJetFinder *Reclusterer, Int_t N, Int_t Type, Bool_t Index)
void NTValues(AliEmcalJet *jet, Int_t jetContNb, Float_t *nTFractions)
Double_t GetSecondOrderSubtractedCircularity() const
Double_t DeltaR(const AliVParticle *part1, const AliVParticle *part2)
Float_t GetJetLeSub(AliEmcalJet *jet, Int_t jetContNb=0)
void UserCreateOutputObjects()
Store some informaion about a Pythia eventThis class is used to store some information about a Pythia...
void SetMakeGeneralHistograms(Bool_t g)
Base task in the EMCAL jet framework.
Represent a jet reconstructed using the EMCal jet framework.
const AliEmcalPythiaInfo * GetPythiaInfo() const
Float_t Circularity(AliEmcalJet *jet, Int_t jetContNb=0)
Declaration of class AliEmcalPythiaInfo.
virtual ~AliAnalysisTaskEmcalJetShapesMC()
void UserCreateOutputObjects()
Main initialization function on the worker.
Double_t GetFirstOrderSubtractedSigma2() const
Float_t Sigma2(AliEmcalJet *jet, Int_t jetContNb=0)
AliEmcalJetShapeProperties * GetShapeProperties() const
Float_t GetPartonPt7() const
Float_t LeSub(AliEmcalJet *jet, Int_t jetContNb=0)
Double_t GetFirstOrderSubtracted() const
Container for jet within the EMCAL jet framework.
std::vector< int > GetPtSortedTrackConstituentIndexes(TClonesArray *tracks) const
Float_t GetPartonPt6() const