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),
73 fTrackCheckPlots(kFALSE),
88 fTreeObservableTagging(0)
91 for(
Int_t i=0;i<17;i++){
93 SetMakeGeneralHistograms(kTRUE);
94 DefineOutput(1, TList::Class());
95 DefineOutput(2, TTree::Class());
102 fMinFractionShared(0),
103 fJetShapeType(
kData),
104 fJetShapeSub(kNoSub),
105 fJetSelection(kInclusive),
106 fPtThreshold(-9999.),
111 fangWindowRecoil(0.6),
115 fCentSelectOn(kTRUE),
118 fOneConstSelectOn(kFALSE),
119 fTrackCheckPlots(kFALSE),
132 fHLundIterative(0x0),
134 fTreeObservableTagging(0)
138 for(
Int_t i=0;i<17;i++){
142 DefineOutput(1, TList::Class());
143 DefineOutput(2, TTree::Class());
159 Bool_t oldStatus = TH1::AddDirectoryStatus();
160 TH1::AddDirectory(kFALSE);
165 fh2ResponseUW=
new TH2F(
"fh2ResponseUW",
"fh2ResponseUW", 100, 0, 200, 100, 0, 200);
167 fh2ResponseW=
new TH2F(
"fh2ResponseW",
"fh2ResponseW", 100, 0, 200, 100, 0, 200);
169 fPhiJetCorr6=
new TH2F(
"fPhiJetCorr6",
"fPhiJetCorr6", 50, 0, 2*TMath::Pi(), 50, 0, 2*TMath::Pi());
171 fEtaJetCorr6=
new TH2F(
"fEtaJetCorr6",
"fEtaJetCorr6", 50, -1.5, 1.5, 50, -1.5, 1.5);
174 fPhiJetCorr7=
new TH2F(
"fPhiJetCorr7",
"fPhiJetCorr7", 50, 0, 2*TMath::Pi(), 50, 0, 2*TMath::Pi());
176 fEtaJetCorr7=
new TH2F(
"fEtaJetCorr7",
"fEtaJetCorr7", 50, -1.5, 1.5, 50, -1.5, 1.5);
179 fPtJetCorr=
new TH2F(
"fPtJetCorr",
"fPtJetCorr", 100, 0, 200, 100, 0, 200);
181 fPtJet=
new TH1F(
"fPtJet",
"fPtJet", 100, 0, 200);
184 fhpTjetpT=
new TH2F(
"fhpTjetpT",
"fhpTjetpT", 200, 0, 200, 200, 0, 200);
186 fhPt=
new TH1F(
"fhPt",
"fhPt", 200, 0, 200);
188 fhPhi=
new TH1F(
"fhPhi",
"fhPhi", 100, -TMath::Pi(), TMath::Pi());
190 fhTrackPhi=
new TH1F(
"fhTrackPhi",
"fhTrackPhi", 100, 0, 2*TMath::Pi());
196 const Int_t dimSpec = 5;
197 const Int_t nBinsSpec[5] = {50,50,10,3,10};
198 const Double_t lowBinSpec[5] = {0.0,-10, 0,0,0};
199 const Double_t hiBinSpec[5] = {5.0, 0,200,3,10};
201 "LundIterativePlot [log(1/theta),log(z*theta),pTjet,algo]",
202 dimSpec,nBinsSpec,lowBinSpec,hiBinSpec);
205 fNbOfConstvspT=
new TH2F(
"fNbOfConstvspT",
"fNbOfConstvspT", 100, 0, 100, 200, 0, 200);
215 THnSparse *hn =
dynamic_cast<THnSparse*
>(
fOutput->At(i));
220 TH1::AddDirectory(oldStatus);
222 const char* nameoutput = GetOutputSlot(2)->GetContainer()->GetName();
228 fShapesVarNames[0] =
"partonCode";
229 fShapesVarNames[1] =
"ptJet";
230 fShapesVarNames[2] =
"ptDJet";
231 fShapesVarNames[3] =
"phiJet";
233 fShapesVarNames[4] =
"angularity";
235 fShapesVarNames[5] =
"lesub";
238 fShapesVarNames[6] =
"ptJetMatch";
239 fShapesVarNames[7] =
"ptDJetMatch";
240 fShapesVarNames[8] =
"phiJetMatch";
242 fShapesVarNames[9] =
"angularityMatch";
244 fShapesVarNames[10] =
"lesubMatch";
246 fShapesVarNames[11]=
"weightPythia";
253 cout<<
"looping over variables"<<endl;
259 delete [] fShapesVarNames;
282 AliAODTrack *triggerHadron = 0x0;
296 TClonesArray *TrackArray = NULL;
297 TClonesArray *TrackArrayMC = NULL;
299 else TrackArray = PartCont->GetArray();
303 else NTracks = TrackArray->GetEntriesFast();
313 if (triggerHadronLabel==-99999) {
318 else triggerHadron =
static_cast<AliAODTrack*
>(TrackArray->At(triggerHadronLabel));
321 if (!triggerHadron) {
333 else triggerHadron =
static_cast<AliAODTrack*
>(TrackArray->At(triggerHadronLabel));
334 fhPt->Fill(triggerHadron->Pt());
340 AliAODTrack *Track = 0x0;
341 for(
Int_t i=0; i < NTracks; i++){
344 if (!Track)
continue;
345 if(TMath::Abs(Track->Eta())>0.9)
continue;
346 if (Track->Pt()<0.15)
continue;
351 if((Track = static_cast<AliAODTrack*>(PartCont->
GetAcceptTrack(i)))){
352 if (!Track)
continue;
353 if(TMath::Abs(Track->Eta())>0.9)
continue;
354 if (Track->Pt()<0.15)
continue;
363 TClonesArray *trackArrayAn = partContAn->GetArray();
364 Int_t ntracksEvt = trackArrayAn->GetEntriesFast();
366 Float_t rhoVal=0, rhoMassVal = 0.;
369 jetCont->ResetCurrentID();
374 Printf(
"%s: Could not retrieve rho %s (some histograms will be filled with zero)!", GetName(), jetCont->
GetRhoName().Data());
375 }
else rhoVal = rhoParam->GetVal();
379 Printf(
"%s: Could not retrieve rho_m %s (some histograms will be filled with zero)!", GetName(), jetCont->
GetRhoMassName().Data());
380 }
else rhoMassVal = rhomParam->GetVal();
389 Int_t ifound=0, jfound=0;
390 Int_t ilab=-1, jlab=-1;
420 jetUS = jetContUS->
GetJet(i);
423 if(ifound==1) ilab = i;
426 if(ilab==-1)
continue;
427 jetUS=jetContUS->
GetJet(ilab);
433 Printf(
"jet2 does not exist, returning");
441 Printf(
"jet3 does not exist, returning");
444 cout<<
"jet 3 exists"<<jet3->
Pt()<<endl;
470 jetUS = jetContUS->
GetJet(i);
473 if(ifound==1) ilab = i;
476 if(ilab==-1)
continue;
477 jetUS=jetContUS->
GetJet(ilab);
481 Printf(
"jet2 does not exist, returning");
487 jet3 = jetContPart->
GetJet(j);
491 if(jfound==1) jlab = j;
494 if(jlab==-1)
continue;
495 jet3=jetContPart->
GetJet(jlab);
497 Printf(
"jet3 does not exist, returning");
503 Printf(
"jet3 does not exist, returning");
522 Float_t dRp1 = TMath::Sqrt(jp1 * jp1 + detap1 * detap1);
532 dRp1 = TMath::Sqrt(jp1 * jp1 + detap1 * detap1);
578 Float_t ptMatch=0., ptDMatch=0., massMatch=0., constMatch=0.,angulMatch=0.,circMatch=0., lesubMatch=0., sigma2Match=0., coronnaMatch=0;
587 massMatch=jet3->
Phi();
602 massMatch=jet3->
Phi();
629 fShapesVar[6] = ptMatch;
630 fShapesVar[7] = ptDMatch;
631 fShapesVar[8] = massMatch;
632 fShapesVar[9] = angulMatch;
634 fShapesVar[10] = lesubMatch;
636 fShapesVar[11] = kWeight;
675 AliVParticle *vp1 = 0x0;
680 Printf(
"AliVParticle associated to constituent not found");
685 Double_t dr2 = (vp1->Eta()-jet->
Eta())*(vp1->Eta()-jet->
Eta()) + dphi*dphi;
687 num=num+vp1->Pt()*dr;
721 TClonesArray *TracksArray = NULL;
722 TClonesArray *TracksArrayMC = NULL;
725 else TracksArray = PartCont->GetArray();
728 if(!PartContMC || !TracksArrayMC)
return -2;
731 if(!PartCont || !TracksArray)
return -2;
735 AliAODTrack *Track = 0x0;
739 else NTracks = TracksArray->GetEntriesFast();
741 for(
Int_t i=0; i < NTracks; i++){
744 if (!Track)
continue;
745 if(TMath::Abs(Track->Eta())>0.9)
continue;
747 Double_t dr2 = (Track->Eta()-jet->
Eta())*(Track->Eta()-jet->
Eta()) + dphi*dphi;
749 if((dr>=0.8) && (dr<1)) sumpt=sumpt+Track->Pt();
753 if((Track = static_cast<AliAODTrack*>(PartCont->
GetAcceptTrack(i)))){
754 if (!Track)
continue;
755 if(TMath::Abs(Track->Eta())>0.9)
continue;
756 if (Track->Pt()<0.15)
continue;
758 Double_t dr2 = (Track->Eta()-jet->
Eta())*(Track->Eta()-jet->
Eta()) + dphi*dphi;
760 if((dr>=0.8) && (dr<1)) sumpt=sumpt+Track->Pt();
780 return Coronna(jet, jetContNb);
796 AliVParticle *vp1 = 0x0;
801 Printf(
"AliVParticle associated to constituent not found");
805 num=num+vp1->Pt()*vp1->Pt();
808 return TMath::Sqrt(num)/den;
818 return PTD(jet, jetContNb);
839 TVector3 ppJ1(pxjet, pyjet, pzjet);
840 TVector3 ppJ3(- pxjet* pzjet, - pyjet * pzjet, pxjet * pxjet + pyjet * pyjet);
842 TVector3 ppJ2(-pyjet, pxjet, 0);
844 AliVParticle *vp1 = 0x0;
849 Printf(
"AliVParticle associated to constituent not found");
853 TVector3 pp(vp1->Px(), vp1->Py(), vp1->Pz());
856 TVector3 pLong = pp.Dot(ppJ1) / ppJ1.Mag2() * ppJ1;
857 TVector3 pPerp = pp - pLong;
860 Float_t ppjX = pPerp.Dot(ppJ2);
861 Float_t ppjY = pPerp.Dot(ppJ3);
862 Float_t ppjT = TMath::Sqrt(ppjX * ppjX + ppjY * ppjY);
863 if(ppjT<=0)
return 0;
865 mxx += (ppjX * ppjX / ppjT);
866 myy += (ppjY * ppjY / ppjT);
867 mxy += (ppjX * ppjY / ppjT);
872 if(sump2==0)
return 0;
874 Double_t ele[4] = {mxx / sump2, mxy / sump2, mxy / sump2, myy / sump2};
875 TMatrixDSym m0(2,ele);
878 TMatrixDSymEigen m(m0);
880 TMatrixD evecm = m.GetEigenVectors();
881 eval = m.GetEigenValues();
885 if (eval[0] < eval[1]) jev = 1;
888 evec0 = TMatrixDColumn(evecm, jev);
891 TVector2 evec(compx, compy);
893 if(jev==1) circ=2*eval[0];
894 if(jev==0) circ=2*eval[1];
925 AliVParticle *vp1 = 0x0;
926 AliVParticle *vp2 = 0x0;
927 std::vector<int> ordindex;
932 if(ordindex.size()<2)
return -1;
936 Printf(
"AliVParticle associated to Leading constituent not found");
942 Printf(
"AliVParticle associated to Subleading constituent not found");
962 return LeSub(jet, jetContNb);
991 AliVParticle *vp1 = 0x0;
996 Printf(
"AliVParticle associated to constituent not found");
1004 mxx += ppt*ppt*deta*deta;
1005 myy += ppt*ppt*dphi*dphi;
1006 mxy -= ppt*ppt*deta*TMath::Abs(dphi);
1012 if(sump2==0)
return 0;
1014 Double_t ele[4] = {mxx , mxy , mxy , myy };
1015 TMatrixDSym m0(2,ele);
1018 TMatrixDSymEigen m(m0);
1020 TMatrixD evecm = m.GetEigenVectors();
1021 eval = m.GetEigenValues();
1025 if (eval[0] < eval[1]) jev = 1;
1028 evec0 = TMatrixDColumn(evecm, jev);
1031 TVector2 evec(compx, compy);
1033 if(jev==1) sig=TMath::Sqrt(TMath::Abs(eval[0])/sump2);
1034 if(jev==0) sig=TMath::Sqrt(TMath::Abs(eval[1])/sump2);
1048 return Sigma2(jet, jetContNb);
1068 TClonesArray *TracksArray = NULL;
1069 TClonesArray *TracksArrayMC = NULL;
1072 else TracksArray = PartCont->GetArray();
1075 if(!PartContMC || !TracksArrayMC)
return -99999;
1078 if(!PartCont || !TracksArray)
return -99999;
1082 AliAODTrack *Track = 0x0;
1087 Int_t triggers[100];
1088 for (
Int_t iTrigger=0; iTrigger<100; iTrigger++) triggers[iTrigger] = 0;
1092 else NTracks = TracksArray->GetEntriesFast();
1094 for(
Int_t i=0; i < NTracks; i++){
1097 if (!Track)
continue;
1098 if(TMath::Abs(Track->Eta())>0.9)
continue;
1099 if (Track->Pt()<0.15)
continue;
1100 if ((Track->Pt() >= minpT) && (Track->Pt()< maxpT)) {
1107 if((Track = static_cast<AliAODTrack*>(PartCont->
GetAcceptTrack(i)))){
1108 if (!Track)
continue;
1109 if(TMath::Abs(Track->Eta())>0.9)
continue;
1110 if (Track->Pt()<0.15)
continue;
1111 if ((Track->Pt() >= minpT) && (Track->Pt()< maxpT)) {
1120 if (iTT == 0)
return -99999;
1121 Int_t nbRn = 0, index = 0 ;
1123 nbRn = random.Integer(iTT);
1124 index = triggers[nbRn];
1133 if (vphi < -1*TMath::Pi()) vphi += (2*TMath::Pi());
1134 else if (vphi > TMath::Pi()) vphi -= (2*TMath::Pi());
1135 if (mphi < -1*TMath::Pi()) mphi += (2*TMath::Pi());
1136 else if (mphi > TMath::Pi()) mphi -= (2*TMath::Pi());
1137 double dphi = mphi-vphi;
1138 if (dphi < -1*TMath::Pi()) dphi += (2*TMath::Pi());
1139 else if (dphi > TMath::Pi()) dphi -= (2*TMath::Pi());
1147 std::vector<fastjet::PseudoJet> fInputVectors;
1148 fInputVectors.clear();
1149 fastjet::PseudoJet PseudoTracks;
1154 AliVParticle *fTrk = fJet->
TrackAt(i, fTrackCont->GetArray());
1155 if (!fTrk)
continue;
1156 PseudoTracks.reset(fTrk->Px(), fTrk->Py(), fTrk->Pz(),fTrk->E());
1157 PseudoTracks.set_user_index(fJet->
TrackAt(i)+100);
1158 fInputVectors.push_back(PseudoTracks);
1161 fastjet::JetAlgorithm jetalgo(fastjet::antikt_algorithm);
1163 if(ReclusterAlgo==0){ xflagalgo=0.5;
1164 jetalgo=fastjet::kt_algorithm ;}
1166 if(ReclusterAlgo==1){ xflagalgo=1.5;
1167 jetalgo=fastjet::cambridge_algorithm;}
1168 if(ReclusterAlgo==2){ xflagalgo=2.5;
1169 jetalgo=fastjet::antikt_algorithm;}
1171 fastjet::JetDefinition fJetDef(jetalgo, 1., static_cast<fastjet::RecombinationScheme>(0), fastjet::BestFJ30 );
1174 fastjet::ClusterSequence fClustSeqSA(fInputVectors, fJetDef);
1175 std::vector<fastjet::PseudoJet> fOutputJets;
1176 fOutputJets.clear();
1177 fOutputJets=fClustSeqSA.inclusive_jets(0);
1179 fastjet::PseudoJet jj;
1180 fastjet::PseudoJet j1;
1181 fastjet::PseudoJet j2;
1184 while(jj.has_parents(j1,j2)){
1186 if(j1.perp() < j2.perp())
swap(j1,j2);
1187 double delta_R=j1.delta_R(j2);
1188 double z=j2.perp()/(j1.perp()+j2.perp());
1189 double y =log(1.0/delta_R);
1190 double lnpt_rel=log(z*delta_R);
1191 Double_t LundEntries[5] = {y,lnpt_rel,fOutputJets[0].perp(),xflagalgo,ndepth};
1198 }
catch (fastjet::Error) {
1199 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