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] =
"phiJet";
227 fShapesVarNames[4] =
"angularity";
229 fShapesVarNames[5] =
"lesub";
232 fShapesVarNames[6] =
"ptJetMatch";
233 fShapesVarNames[7] =
"ptDJetMatch";
234 fShapesVarNames[8] =
"phiJetMatch";
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);
541 Float_t ptMatch=0., ptDMatch=0., massMatch=0., constMatch=0.,angulMatch=0.,circMatch=0., lesubMatch=0., sigma2Match=0., coronnaMatch=0;
550 massMatch=jet3->
Phi();
565 massMatch=jet3->
Phi();
592 fShapesVar[6] = ptMatch;
593 fShapesVar[7] = ptDMatch;
594 fShapesVar[8] = massMatch;
595 fShapesVar[9] = angulMatch;
597 fShapesVar[10] = lesubMatch;
599 fShapesVar[11] = kWeight;
638 AliVParticle *vp1 = 0x0;
643 Printf(
"AliVParticle associated to constituent not found");
648 Double_t dr2 = (vp1->Eta()-jet->
Eta())*(vp1->Eta()-jet->
Eta()) + dphi*dphi;
650 num=num+vp1->Pt()*dr;
684 TClonesArray *TracksArray = NULL;
685 TClonesArray *TracksArrayMC = NULL;
688 else TracksArray = PartCont->GetArray();
691 if(!PartContMC || !TracksArrayMC)
return -2;
694 if(!PartCont || !TracksArray)
return -2;
698 AliAODTrack *Track = 0x0;
702 else NTracks = TracksArray->GetEntriesFast();
704 for(
Int_t i=0; i < NTracks; i++){
707 if (!Track)
continue;
708 if(TMath::Abs(Track->Eta())>0.9)
continue;
710 Double_t dr2 = (Track->Eta()-jet->
Eta())*(Track->Eta()-jet->
Eta()) + dphi*dphi;
712 if((dr>=0.8) && (dr<1)) sumpt=sumpt+Track->Pt();
716 if((Track = static_cast<AliAODTrack*>(PartCont->
GetAcceptTrack(i)))){
717 if (!Track)
continue;
718 if(TMath::Abs(Track->Eta())>0.9)
continue;
719 if (Track->Pt()<0.15)
continue;
721 Double_t dr2 = (Track->Eta()-jet->
Eta())*(Track->Eta()-jet->
Eta()) + dphi*dphi;
723 if((dr>=0.8) && (dr<1)) sumpt=sumpt+Track->Pt();
743 return Coronna(jet, jetContNb);
759 AliVParticle *vp1 = 0x0;
764 Printf(
"AliVParticle associated to constituent not found");
768 num=num+vp1->Pt()*vp1->Pt();
771 return TMath::Sqrt(num)/den;
781 return PTD(jet, jetContNb);
802 TVector3 ppJ1(pxjet, pyjet, pzjet);
803 TVector3 ppJ3(- pxjet* pzjet, - pyjet * pzjet, pxjet * pxjet + pyjet * pyjet);
805 TVector3 ppJ2(-pyjet, pxjet, 0);
807 AliVParticle *vp1 = 0x0;
812 Printf(
"AliVParticle associated to constituent not found");
816 TVector3 pp(vp1->Px(), vp1->Py(), vp1->Pz());
819 TVector3 pLong = pp.Dot(ppJ1) / ppJ1.Mag2() * ppJ1;
820 TVector3 pPerp = pp - pLong;
823 Float_t ppjX = pPerp.Dot(ppJ2);
824 Float_t ppjY = pPerp.Dot(ppJ3);
825 Float_t ppjT = TMath::Sqrt(ppjX * ppjX + ppjY * ppjY);
826 if(ppjT<=0)
return 0;
828 mxx += (ppjX * ppjX / ppjT);
829 myy += (ppjY * ppjY / ppjT);
830 mxy += (ppjX * ppjY / ppjT);
835 if(sump2==0)
return 0;
837 Double_t ele[4] = {mxx / sump2, mxy / sump2, mxy / sump2, myy / sump2};
838 TMatrixDSym m0(2,ele);
841 TMatrixDSymEigen m(m0);
843 TMatrixD evecm = m.GetEigenVectors();
844 eval = m.GetEigenValues();
848 if (eval[0] < eval[1]) jev = 1;
851 evec0 = TMatrixDColumn(evecm, jev);
854 TVector2 evec(compx, compy);
856 if(jev==1) circ=2*eval[0];
857 if(jev==0) circ=2*eval[1];
888 AliVParticle *vp1 = 0x0;
889 AliVParticle *vp2 = 0x0;
890 std::vector<int> ordindex;
895 if(ordindex.size()<2)
return -1;
899 Printf(
"AliVParticle associated to Leading constituent not found");
905 Printf(
"AliVParticle associated to Subleading constituent not found");
925 return LeSub(jet, jetContNb);
954 AliVParticle *vp1 = 0x0;
959 Printf(
"AliVParticle associated to constituent not found");
967 mxx += ppt*ppt*deta*deta;
968 myy += ppt*ppt*dphi*dphi;
969 mxy -= ppt*ppt*deta*TMath::Abs(dphi);
975 if(sump2==0)
return 0;
977 Double_t ele[4] = {mxx , mxy , mxy , myy };
978 TMatrixDSym m0(2,ele);
981 TMatrixDSymEigen m(m0);
983 TMatrixD evecm = m.GetEigenVectors();
984 eval = m.GetEigenValues();
988 if (eval[0] < eval[1]) jev = 1;
991 evec0 = TMatrixDColumn(evecm, jev);
994 TVector2 evec(compx, compy);
996 if(jev==1) sig=TMath::Sqrt(TMath::Abs(eval[0])/sump2);
997 if(jev==0) sig=TMath::Sqrt(TMath::Abs(eval[1])/sump2);
1011 return Sigma2(jet, jetContNb);
1031 TClonesArray *TracksArray = NULL;
1032 TClonesArray *TracksArrayMC = NULL;
1035 else TracksArray = PartCont->GetArray();
1038 if(!PartContMC || !TracksArrayMC)
return -99999;
1041 if(!PartCont || !TracksArray)
return -99999;
1045 AliAODTrack *Track = 0x0;
1050 Int_t triggers[100];
1051 for (
Int_t iTrigger=0; iTrigger<100; iTrigger++) triggers[iTrigger] = 0;
1055 else NTracks = TracksArray->GetEntriesFast();
1057 for(
Int_t i=0; i < NTracks; i++){
1060 if (!Track)
continue;
1061 if(TMath::Abs(Track->Eta())>0.9)
continue;
1062 if (Track->Pt()<0.15)
continue;
1063 if ((Track->Pt() >= minpT) && (Track->Pt()< maxpT)) {
1070 if((Track = static_cast<AliAODTrack*>(PartCont->
GetAcceptTrack(i)))){
1071 if (!Track)
continue;
1072 if(TMath::Abs(Track->Eta())>0.9)
continue;
1073 if (Track->Pt()<0.15)
continue;
1074 if ((Track->Pt() >= minpT) && (Track->Pt()< maxpT)) {
1083 if (iTT == 0)
return -99999;
1084 Int_t nbRn = 0, index = 0 ;
1085 TRandom3* random =
new TRandom3(0);
1086 nbRn = random->Integer(iTT);
1087 index = triggers[nbRn];
1096 if (vphi < -1*TMath::Pi()) vphi += (2*TMath::Pi());
1097 else if (vphi > TMath::Pi()) vphi -= (2*TMath::Pi());
1098 if (mphi < -1*TMath::Pi()) mphi += (2*TMath::Pi());
1099 else if (mphi > TMath::Pi()) mphi -= (2*TMath::Pi());
1100 double dphi = mphi-vphi;
1101 if (dphi < -1*TMath::Pi()) dphi += (2*TMath::Pi());
1102 else if (dphi > TMath::Pi()) dphi -= (2*TMath::Pi());
1110 std::vector<fastjet::PseudoJet> fInputVectors;
1111 fInputVectors.clear();
1112 fastjet::PseudoJet PseudoTracks;
1117 AliVParticle *fTrk = fJet->
TrackAt(i, fTrackCont->GetArray());
1118 if (!fTrk)
continue;
1119 PseudoTracks.reset(fTrk->Px(), fTrk->Py(), fTrk->Pz(),fTrk->E());
1120 PseudoTracks.set_user_index(fJet->
TrackAt(i)+100);
1121 fInputVectors.push_back(PseudoTracks);
1124 fastjet::JetAlgorithm jetalgo(fastjet::antikt_algorithm);
1126 if(ReclusterAlgo==0){ xflagalgo=0.5;
1127 jetalgo=fastjet::kt_algorithm ;}
1129 if(ReclusterAlgo==1){ xflagalgo=1.5;
1130 jetalgo=fastjet::cambridge_algorithm;}
1131 if(ReclusterAlgo==2){ xflagalgo=2.5;
1132 jetalgo=fastjet::antikt_algorithm;}
1134 fastjet::JetDefinition fJetDef(jetalgo, 1., static_cast<fastjet::RecombinationScheme>(0), fastjet::BestFJ30 );
1137 fastjet::ClusterSequence fClustSeqSA(fInputVectors, fJetDef);
1138 std::vector<fastjet::PseudoJet> fOutputJets;
1139 fOutputJets.clear();
1140 fOutputJets=fClustSeqSA.inclusive_jets(0);
1142 fastjet::PseudoJet jj;
1143 fastjet::PseudoJet j1;
1144 fastjet::PseudoJet j2;
1147 while(jj.has_parents(j1,j2)){
1149 if(j1.perp() < j2.perp())
swap(j1,j2);
1150 double delta_R=j1.delta_R(j2);
1151 double z=j2.perp()/(j1.perp()+j2.perp());
1152 double y =log(1.0/delta_R);
1153 double lnpt_rel=log(z*delta_R);
1154 Double_t LundEntries[5] = {y,lnpt_rel,fOutputJets[0].perp(),xflagalgo,ndepth};
1161 }
catch (fastjet::Error) {
1162 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