fe4da5cc |
1 | #include "AliGenerator.h" |
2 | #include "AliGenCocktail.h" |
fe4da5cc |
3 | #include "AliRun.h" |
4 | #include <TDirectory.h> |
5 | #include <TFile.h> |
6 | #include <TTree.h> |
7 | #include <stdlib.h> |
8 | ClassImp(AliGenCocktailEntry) |
9 | void AliGenCocktailEntry::PrintInfo() |
10 | { |
11 | printf("\n Generator: %s Generated Events: %d First: %d Last: %d", |
12 | (const char *) fName, fGenerator->NumberParticles(), fFirst, fLast); |
13 | } |
14 | |
15 | ClassImp(AliGenCocktail) |
16 | |
17 | AliGenCocktail::AliGenCocktail() |
18 | :AliGenerator() |
19 | { |
20 | fEntries = new TList; |
21 | fNGenerators=0; |
22 | } |
23 | |
24 | |
25 | AliGenCocktail::~AliGenCocktail() |
26 | { |
27 | delete fEntries; |
28 | } |
29 | |
30 | void AliGenCocktail:: |
31 | AddGenerator(AliGenerator *Generator, TString Name, Float_t RateExp) |
32 | { |
33 | // |
34 | // Forward parameters to the new generator |
35 | Generator->SetPtRange(fPtMin,fPtMax); |
36 | Generator->SetMomentumRange(fPMin,fPMax); |
37 | Generator->SetYRange(fYMin,fYMax); |
38 | Generator-> |
39 | SetPhiRange(fPhiMin*180/TMath::Pi(),fPhiMax*180/TMath::Pi()); |
40 | Generator-> |
41 | SetThetaRange(fThetaMin*180/TMath::Pi(),fThetaMax*180/TMath::Pi()); |
42 | Generator-> |
43 | SetOrigin(fOrigin[0], fOrigin[1], fOrigin[2]); |
44 | Generator-> |
45 | SetSigma(fOsigma[0], fOsigma[1], fOsigma[2]); |
7a50a130 |
46 | Generator->SetVertexSmear(fVertexSmear); |
47 | Generator->SetTrackingFlag(fTrackIt); |
fe4da5cc |
48 | // |
49 | // Add generator to list |
50 | AliGenCocktailEntry *Entry = |
51 | new AliGenCocktailEntry(Generator, Name, RateExp); |
52 | fEntries->Add(Entry); |
53 | fNGenerators++; |
54 | } |
55 | |
56 | void AliGenCocktail::Init() |
57 | { |
58 | TIter next(fEntries); |
59 | AliGenCocktailEntry *Entry; |
60 | // |
61 | // Loop over generators and initialize |
62 | while((Entry = (AliGenCocktailEntry*)next())) { |
63 | Entry->Generator()->Init(); |
64 | } |
65 | } |
66 | |
67 | void AliGenCocktail::Generate() |
68 | { |
69 | TIter next(fEntries); |
70 | AliGenCocktailEntry *Entry; |
71 | AliGenCocktailEntry *e1; |
72 | AliGenCocktailEntry *e2; |
73 | TClonesArray *PartArray = gAlice->Particles(); |
74 | // |
75 | // Loop over generators and generate events |
76 | Int_t igen=0; |
77 | while((Entry = (AliGenCocktailEntry*)next())) { |
78 | igen++; |
79 | if (igen ==1) { |
80 | Entry->SetFirst(0); |
81 | } else { |
82 | Entry->SetFirst((PartArray->GetEntriesFast())+1); |
83 | } |
84 | Entry->Generator()->Generate(); |
85 | Entry->SetLast(PartArray->GetEntriesFast()); |
86 | } |
87 | next.Reset(); |
88 | while((Entry = (AliGenCocktailEntry*)next())) { |
89 | Entry->PrintInfo(); |
90 | } |
91 | for (Entry=FirstGenerator(); |
92 | Entry; |
93 | Entry=NextGenerator() |
94 | ) { |
95 | Entry->PrintInfo(); |
96 | } |
97 | for (FirstGeneratorPair(e1,e2); |
98 | (e1&&e2); |
99 | NextGeneratorPair(e1,e2) |
100 | ){ |
101 | printf("\n -----------------------------"); |
102 | e1->PrintInfo(); |
103 | e2->PrintInfo(); |
104 | } |
105 | } |
106 | |
107 | AliGenCocktailEntry * AliGenCocktail::FirstGenerator() |
108 | { |
109 | flnk1 = fEntries->FirstLink(); |
110 | if (flnk1) { |
111 | return (AliGenCocktailEntry*) (flnk1->GetObject()); |
112 | } else { |
113 | return 0; |
114 | } |
115 | } |
116 | |
117 | AliGenCocktailEntry* AliGenCocktail::NextGenerator() |
118 | { |
119 | flnk1 = flnk1->Next(); |
120 | if (flnk1) { |
121 | return (AliGenCocktailEntry*) (flnk1->GetObject()); |
122 | } else { |
123 | return 0; |
124 | } |
125 | } |
126 | |
127 | void AliGenCocktail:: |
128 | FirstGeneratorPair(AliGenCocktailEntry*& e1, AliGenCocktailEntry*& e2) |
129 | { |
130 | flnk2 = flnk1 = fEntries->FirstLink(); |
131 | if (flnk1) { |
132 | e2 = e1 = (AliGenCocktailEntry*) (flnk1->GetObject()); |
133 | } else { |
134 | e2= e1 = 0; |
135 | } |
136 | } |
137 | |
138 | void AliGenCocktail:: |
139 | NextGeneratorPair(AliGenCocktailEntry*& e1, AliGenCocktailEntry*& e2) |
140 | { |
141 | flnk2 = flnk2->Next(); |
142 | if (flnk2) { |
143 | e1 = (AliGenCocktailEntry*) (flnk1->GetObject()); |
144 | e2 = (AliGenCocktailEntry*) (flnk2->GetObject()); |
145 | } else { |
146 | flnk2 = flnk1 = flnk1->Next(); |
147 | if (flnk1) { |
148 | e1 = (AliGenCocktailEntry*) (flnk1->GetObject()); |
149 | e2 = (AliGenCocktailEntry*) (flnk2->GetObject()); |
150 | } else { |
151 | e1=0; |
152 | e2=0; |
153 | } |
154 | } |
155 | } |
156 | |
157 | |
158 | void AliGenCocktail::Streamer(TBuffer &R__b) |
159 | { |
160 | // Stream an object of class AliGenCocktail. |
161 | TIter next(fEntries); |
162 | AliGenCocktailEntry *Entry; |
163 | |
164 | if (R__b.IsReading()) { |
165 | Version_t R__v = R__b.ReadVersion(); if (R__v) { } |
166 | AliGenerator::Streamer(R__b); |
167 | R__b >> fNGenerators; |
168 | R__b >> fEntries; |
169 | // Stream generation related information |
170 | while((Entry = (AliGenCocktailEntry*)next())) { |
171 | Entry->Streamer(R__b); |
172 | } |
173 | } else { |
174 | R__b.WriteVersion(AliGenCocktail::IsA()); |
175 | AliGenerator::Streamer(R__b); |
176 | R__b << fNGenerators; |
177 | R__b << fEntries; |
178 | // Stream generation related information |
179 | while((Entry = (AliGenCocktailEntry*)next())) { |
180 | Entry->Streamer(R__b); |
181 | } |
182 | } |
183 | } |
184 | |
185 | |