]>
Commit | Line | Data |
---|---|---|
4c039060 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
16 | /* | |
17 | $Log$ | |
18 | */ | |
19 | ||
fe4da5cc | 20 | #include "AliGenerator.h" |
21 | #include "AliGenCocktail.h" | |
fe4da5cc | 22 | #include "AliRun.h" |
23 | #include <TDirectory.h> | |
24 | #include <TFile.h> | |
25 | #include <TTree.h> | |
26 | #include <stdlib.h> | |
27 | ClassImp(AliGenCocktailEntry) | |
28 | void AliGenCocktailEntry::PrintInfo() | |
29 | { | |
30 | printf("\n Generator: %s Generated Events: %d First: %d Last: %d", | |
31 | (const char *) fName, fGenerator->NumberParticles(), fFirst, fLast); | |
32 | } | |
33 | ||
34 | ClassImp(AliGenCocktail) | |
35 | ||
36 | AliGenCocktail::AliGenCocktail() | |
37 | :AliGenerator() | |
38 | { | |
39 | fEntries = new TList; | |
40 | fNGenerators=0; | |
41 | } | |
42 | ||
43 | ||
44 | AliGenCocktail::~AliGenCocktail() | |
45 | { | |
46 | delete fEntries; | |
47 | } | |
48 | ||
49 | void AliGenCocktail:: | |
50 | AddGenerator(AliGenerator *Generator, TString Name, Float_t RateExp) | |
51 | { | |
52 | // | |
53 | // Forward parameters to the new generator | |
54 | Generator->SetPtRange(fPtMin,fPtMax); | |
55 | Generator->SetMomentumRange(fPMin,fPMax); | |
56 | Generator->SetYRange(fYMin,fYMax); | |
57 | Generator-> | |
58 | SetPhiRange(fPhiMin*180/TMath::Pi(),fPhiMax*180/TMath::Pi()); | |
59 | Generator-> | |
60 | SetThetaRange(fThetaMin*180/TMath::Pi(),fThetaMax*180/TMath::Pi()); | |
61 | Generator-> | |
62 | SetOrigin(fOrigin[0], fOrigin[1], fOrigin[2]); | |
63 | Generator-> | |
64 | SetSigma(fOsigma[0], fOsigma[1], fOsigma[2]); | |
7a50a130 | 65 | Generator->SetVertexSmear(fVertexSmear); |
66 | Generator->SetTrackingFlag(fTrackIt); | |
fe4da5cc | 67 | // |
68 | // Add generator to list | |
69 | AliGenCocktailEntry *Entry = | |
70 | new AliGenCocktailEntry(Generator, Name, RateExp); | |
71 | fEntries->Add(Entry); | |
72 | fNGenerators++; | |
73 | } | |
74 | ||
75 | void AliGenCocktail::Init() | |
76 | { | |
77 | TIter next(fEntries); | |
78 | AliGenCocktailEntry *Entry; | |
79 | // | |
80 | // Loop over generators and initialize | |
81 | while((Entry = (AliGenCocktailEntry*)next())) { | |
82 | Entry->Generator()->Init(); | |
83 | } | |
84 | } | |
85 | ||
86 | void AliGenCocktail::Generate() | |
87 | { | |
88 | TIter next(fEntries); | |
89 | AliGenCocktailEntry *Entry; | |
90 | AliGenCocktailEntry *e1; | |
91 | AliGenCocktailEntry *e2; | |
92 | TClonesArray *PartArray = gAlice->Particles(); | |
93 | // | |
94 | // Loop over generators and generate events | |
95 | Int_t igen=0; | |
96 | while((Entry = (AliGenCocktailEntry*)next())) { | |
97 | igen++; | |
98 | if (igen ==1) { | |
99 | Entry->SetFirst(0); | |
100 | } else { | |
101 | Entry->SetFirst((PartArray->GetEntriesFast())+1); | |
102 | } | |
103 | Entry->Generator()->Generate(); | |
104 | Entry->SetLast(PartArray->GetEntriesFast()); | |
105 | } | |
106 | next.Reset(); | |
107 | while((Entry = (AliGenCocktailEntry*)next())) { | |
108 | Entry->PrintInfo(); | |
109 | } | |
110 | for (Entry=FirstGenerator(); | |
111 | Entry; | |
112 | Entry=NextGenerator() | |
113 | ) { | |
114 | Entry->PrintInfo(); | |
115 | } | |
116 | for (FirstGeneratorPair(e1,e2); | |
117 | (e1&&e2); | |
118 | NextGeneratorPair(e1,e2) | |
119 | ){ | |
120 | printf("\n -----------------------------"); | |
121 | e1->PrintInfo(); | |
122 | e2->PrintInfo(); | |
123 | } | |
124 | } | |
125 | ||
126 | AliGenCocktailEntry * AliGenCocktail::FirstGenerator() | |
127 | { | |
128 | flnk1 = fEntries->FirstLink(); | |
129 | if (flnk1) { | |
130 | return (AliGenCocktailEntry*) (flnk1->GetObject()); | |
131 | } else { | |
132 | return 0; | |
133 | } | |
134 | } | |
135 | ||
136 | AliGenCocktailEntry* AliGenCocktail::NextGenerator() | |
137 | { | |
138 | flnk1 = flnk1->Next(); | |
139 | if (flnk1) { | |
140 | return (AliGenCocktailEntry*) (flnk1->GetObject()); | |
141 | } else { | |
142 | return 0; | |
143 | } | |
144 | } | |
145 | ||
146 | void AliGenCocktail:: | |
147 | FirstGeneratorPair(AliGenCocktailEntry*& e1, AliGenCocktailEntry*& e2) | |
148 | { | |
149 | flnk2 = flnk1 = fEntries->FirstLink(); | |
150 | if (flnk1) { | |
151 | e2 = e1 = (AliGenCocktailEntry*) (flnk1->GetObject()); | |
152 | } else { | |
153 | e2= e1 = 0; | |
154 | } | |
155 | } | |
156 | ||
157 | void AliGenCocktail:: | |
158 | NextGeneratorPair(AliGenCocktailEntry*& e1, AliGenCocktailEntry*& e2) | |
159 | { | |
160 | flnk2 = flnk2->Next(); | |
161 | if (flnk2) { | |
162 | e1 = (AliGenCocktailEntry*) (flnk1->GetObject()); | |
163 | e2 = (AliGenCocktailEntry*) (flnk2->GetObject()); | |
164 | } else { | |
165 | flnk2 = flnk1 = flnk1->Next(); | |
166 | if (flnk1) { | |
167 | e1 = (AliGenCocktailEntry*) (flnk1->GetObject()); | |
168 | e2 = (AliGenCocktailEntry*) (flnk2->GetObject()); | |
169 | } else { | |
170 | e1=0; | |
171 | e2=0; | |
172 | } | |
173 | } | |
174 | } | |
175 | ||
176 | ||
177 | void AliGenCocktail::Streamer(TBuffer &R__b) | |
178 | { | |
179 | // Stream an object of class AliGenCocktail. | |
180 | TIter next(fEntries); | |
181 | AliGenCocktailEntry *Entry; | |
182 | ||
183 | if (R__b.IsReading()) { | |
184 | Version_t R__v = R__b.ReadVersion(); if (R__v) { } | |
185 | AliGenerator::Streamer(R__b); | |
186 | R__b >> fNGenerators; | |
187 | R__b >> fEntries; | |
188 | // Stream generation related information | |
189 | while((Entry = (AliGenCocktailEntry*)next())) { | |
190 | Entry->Streamer(R__b); | |
191 | } | |
192 | } else { | |
193 | R__b.WriteVersion(AliGenCocktail::IsA()); | |
194 | AliGenerator::Streamer(R__b); | |
195 | R__b << fNGenerators; | |
196 | R__b << fEntries; | |
197 | // Stream generation related information | |
198 | while((Entry = (AliGenCocktailEntry*)next())) { | |
199 | Entry->Streamer(R__b); | |
200 | } | |
201 | } | |
202 | } | |
203 | ||
204 |