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"
32 ClassImp(AliJetEmbeddingFromGenTask)
34 //________________________________________________________________________
35 AliJetEmbeddingFromGenTask::AliJetEmbeddingFromGenTask() :
36 AliJetModelBaseTask("AliJetEmbeddingFromGenTask"),
46 // Default constructor.
47 SetSuffix("EmbeddedFromGen");
50 //________________________________________________________________________
51 AliJetEmbeddingFromGenTask::AliJetEmbeddingFromGenTask(const char *name, Bool_t drawqa) :
52 AliJetModelBaseTask(name,drawqa),
62 // Standard constructor.
63 SetSuffix("EmbeddedFromGen");
66 //________________________________________________________________________
67 AliJetEmbeddingFromGenTask::~AliJetEmbeddingFromGenTask()
72 //________________________________________________________________________
73 void AliJetEmbeddingFromGenTask::UserCreateOutputObjects()
75 // Create user output.
80 AliJetModelBaseTask::UserCreateOutputObjects();
82 fHistPt = new TH1F("fHistpt","fHistPt;#it{p}_{T};N",100,0.,100.);
83 fOutput->Add(fHistPt);
85 fHistEtaPhi = new TH2F("fHistEtaPhi","fHistEtaPhi;#eta;#varphi",100,-3.,3.,100.,0.,TMath::TwoPi());
86 fOutput->Add(fHistEtaPhi);
88 fHistTrials = new TH1F("fHistTrials", "fHistTrials", 1, 0, 1);
89 fHistTrials->GetYaxis()->SetTitle("trials");
90 fOutput->Add(fHistTrials);
92 fHistXsection = new TProfile("fHistXsection", "fHistXsection", 1, 0, 1);
93 fHistXsection->GetYaxis()->SetTitle("xsection");
94 fOutput->Add(fHistXsection);
96 fHistPtHard = new TH1F("fHistPtHard", "fHistPtHard", 500, 0., 500.);
97 fHistPtHard->GetXaxis()->SetTitle("p_{T,hard} (GeV/c)");
98 fHistPtHard->GetYaxis()->SetTitle("counts");
99 fOutput->Add(fHistPtHard);
101 PostData(1, fOutput);
104 //________________________________________________________________________
105 Bool_t AliJetEmbeddingFromGenTask::ExecOnce()
110 new AliRun("gAlice","The ALICE Off-line Simulation Framework");
112 gRandom = new TRandom3(0);
115 TFolder *folder = new TFolder(GetName(),GetName());
116 AliRunLoader *rl = new AliRunLoader(folder);
119 AliStack *stack = rl->Stack();
120 fGen->SetStack(stack);
123 if (!(InputEvent()->FindListObject(fTracksName))) {
124 fOutTracks = new TClonesArray("AliPicoTrack", 1000);
125 fOutTracks->SetName(fTracksName);
126 InputEvent()->AddObject(fOutTracks);
133 //________________________________________________________________________
134 void AliJetEmbeddingFromGenTask::Run()
141 AliStack *stack = fGen->GetStack();
144 const Int_t nprim = stack->GetNprimary();
145 for (Int_t i=0;i<nprim;++i) {
146 if (!stack->IsPhysicalPrimary(i))
148 TParticle *part = stack->Particle(i);
149 TParticlePDG *pdg = part->GetPDG(1);
152 Int_t c = (Int_t)(TMath::Abs(pdg->Charge()));
153 if (fChargedOnly && c==0) continue;
154 Double_t pt = part->Pt();
155 Double_t eta = part->Eta();
156 Double_t phi = part->Phi();
169 Double_t mass = part->GetMass();
170 if(fMassless) mass = 0.;
172 fHistEtaPhi->Fill(eta,phi);
173 AddTrack(pt, eta, phi,0,0,0,0,0,0,c,mass);
176 FillPythiaHistograms();
179 //________________________________________________________________________
180 void AliJetEmbeddingFromGenTask::FillPythiaHistograms() {
181 //Get PYTHIA info: pt-hard, x-section, trials
186 AliRunLoader *rl = AliRunLoader::Instance();
187 AliGenPythiaEventHeader *genPH = dynamic_cast<AliGenPythiaEventHeader*>(rl->GetHeader()->GenEventHeader());
189 Float_t xsec = genPH->GetXsection();
190 Int_t trials = genPH->Trials();
191 Float_t pthard = genPH->GetPtHard();
193 fHistXsection->Fill(0.5,xsec);
194 fHistTrials->Fill(0.5,trials);
195 fHistPtHard->Fill(pthard);