5 // Author: S.Aiola, C.Loizides
7 #include "AliJetEmbeddingFromGenTask.h"
9 #include <TClonesArray.h>
11 #include <TLorentzVector.h>
12 #include <TParticle.h>
13 #include <TParticlePDG.h>
16 #include "AliAnalysisManager.h"
17 #include "AliEMCALDigit.h"
18 #include "AliEMCALGeometry.h"
19 #include "AliEMCALRecPoint.h"
20 #include "AliGenerator.h"
21 #include "AliHeader.h"
23 #include "AliPicoTrack.h"
25 #include "AliRunLoader.h"
28 #include "AliVCluster.h"
29 #include "AliVEvent.h"
30 #include "AliGenPythiaEventHeader.h"
31 #include "AliStackPartonInfo.h"
33 ClassImp(AliJetEmbeddingFromGenTask)
35 //________________________________________________________________________
36 AliJetEmbeddingFromGenTask::AliJetEmbeddingFromGenTask() :
37 AliJetModelBaseTask("AliJetEmbeddingFromGenTask"),
47 // Default constructor.
48 SetSuffix("EmbeddedFromGen");
51 //________________________________________________________________________
52 AliJetEmbeddingFromGenTask::AliJetEmbeddingFromGenTask(const char *name, Bool_t drawqa) :
53 AliJetModelBaseTask(name,drawqa),
63 // Standard constructor.
64 SetSuffix("EmbeddedFromGen");
67 //________________________________________________________________________
68 AliJetEmbeddingFromGenTask::~AliJetEmbeddingFromGenTask()
73 //________________________________________________________________________
74 void AliJetEmbeddingFromGenTask::UserCreateOutputObjects()
76 // Create user output.
81 AliJetModelBaseTask::UserCreateOutputObjects();
83 fHistPt = new TH1F("fHistpt","fHistPt;#it{p}_{T};N",100,0.,100.);
84 fOutput->Add(fHistPt);
86 fHistEtaPhi = new TH2F("fHistEtapHI","fHistEtaPhi;#eta;#varphi",100,-3.,3.,100.,0.,TMath::TwoPi());
87 fOutput->Add(fHistEtaPhi);
89 fHistTrials = new TH1F("fHistTrials", "fHistTrials", 1, 0, 1);
90 fHistTrials->GetYaxis()->SetTitle("trials");
91 fOutput->Add(fHistTrials);
93 fHistXsection = new TProfile("fHistXsection", "fHistXsection", 1, 0, 1);
94 fHistXsection->GetYaxis()->SetTitle("xsection");
95 fOutput->Add(fHistXsection);
97 fHistPtHard = new TH1F("fHistPtHard", "fHistPtHard", 500, 0., 500.);
98 fHistPtHard->GetXaxis()->SetTitle("p_{T,hard} (GeV/c)");
99 fHistPtHard->GetYaxis()->SetTitle("counts");
100 fOutput->Add(fHistPtHard);
102 PostData(1, fOutput);
105 //________________________________________________________________________
106 Bool_t AliJetEmbeddingFromGenTask::ExecOnce()
111 new AliRun("gAlice","The ALICE Off-line Simulation Framework");
113 gRandom = new TRandom3(0);
116 TFolder *folder = new TFolder(GetName(),GetName());
117 AliRunLoader *rl = new AliRunLoader(folder);
120 AliStack *stack = rl->Stack();
121 fGen->SetStack(stack);
124 if (!(InputEvent()->FindListObject(fTracksName))) {
125 fOutTracks = new TClonesArray("AliPicoTrack", 1000);
126 fOutTracks->SetName(fTracksName);
127 InputEvent()->AddObject(fOutTracks);
131 if (!(InputEvent()->FindListObject(fPartonInfoName))) {
132 fStackPartonInfo = new AliStackPartonInfo("PartonsInfo");
133 fStackPartonInfo->SetName(fPartonInfoName);
134 InputEvent()->AddObject(fStackPartonInfo);
139 //________________________________________________________________________
140 void AliJetEmbeddingFromGenTask::Run()
147 AliStack *stack = fGen->GetStack();
150 const Int_t nprim = stack->GetNprimary();
151 // reject if partons are missing from stack for some reason
152 if(nprim < 8) return;
153 TParticle *part6 = stack->Particle(6);
154 TParticle *part7 = stack->Particle(7);
156 fStackPartonInfo->SetPartonFlag6(TMath::Abs(part6->GetPdgCode()));
157 fStackPartonInfo->SetPartonPt6(part6->Pt());
158 fStackPartonInfo->SetPartonEta6(part6->Eta());
159 fStackPartonInfo->SetPartonPhi6(part6->Phi());
161 fStackPartonInfo->SetPartonFlag7(TMath::Abs(part7->GetPdgCode()));
162 fStackPartonInfo->SetPartonPt7(part7->Pt());
163 fStackPartonInfo->SetPartonEta7(part7->Eta());
164 fStackPartonInfo->SetPartonPhi7(part7->Phi());
166 for (Int_t i=0;i<nprim;++i) {
167 if (!stack->IsPhysicalPrimary(i))
169 TParticle *part = stack->Particle(i);
170 TParticlePDG *pdg = part->GetPDG(1);
173 Int_t c = (Int_t)(TMath::Abs(pdg->Charge()));
174 if (fChargedOnly && c==0) continue;
175 Double_t pt = part->Pt();
176 Double_t eta = part->Eta();
177 Double_t phi = part->Phi();
190 Double_t mass = part->GetMass();
191 if(fMassless) mass = 0.;
193 fHistEtaPhi->Fill(eta,phi);
194 AddTrack(pt, eta, phi,0,0,0,0,0,0,c,mass);
197 FillPythiaHistograms();
200 //________________________________________________________________________
201 void AliJetEmbeddingFromGenTask::FillPythiaHistograms() {
202 //Get PYTHIA info: pt-hard, x-section, trials
207 AliRunLoader *rl = AliRunLoader::Instance();
208 AliGenPythiaEventHeader *genPH = dynamic_cast<AliGenPythiaEventHeader*>(rl->GetHeader()->GenEventHeader());
210 Float_t xsec = genPH->GetXsection();
211 Int_t trials = genPH->Trials();
212 Float_t pthard = genPH->GetPtHard();
214 fHistXsection->Fill(0.5,xsec);
215 fHistTrials->Fill(0.5,trials);
216 fHistPtHard->Fill(pthard);