This commit was generated by cvs2svn to compensate for changes in r23244,
[u/mrichter/AliRoot.git] / TTherminator / Therminator / Event.cxx
CommitLineData
2e967919 1/******************************************************************************
2 * T H E R M I N A T O R *
3 * THERMal heavy-IoN generATOR *
4 * version 1.0 *
5 * *
6 * Authors of the model: Wojciech Broniowski, Wojciech.Broniowski@ifj.edu.pl, *
7 * Wojciech Florkowski, Wojciech.Florkowski@ifj.edu.pl *
8 * Authors of the code: Adam Kisiel, kisiel@if.pw.edu.pl *
9 * Tomasz Taluc, ttaluc@if.pw.edu.pl *
10 * Code designers: Adam Kisiel, Tomasz Taluc, Wojciech Broniowski, *
11 * Wojciech Florkowski *
12 * *
13 * For the detailed description of the program and furhter references *
14 * to the description of the model plesase refer to: nucl-th/0504047, *
15 * accessibile at: http://www.arxiv.org/nucl-th/0504047 *
16 * *
17 * Homepage: http://hirg.if.pw.edu.pl/en/therminator/ *
18 * *
19 * This code can be freely used and redistributed. However if you decide to *
20 * make modifications to the code, please contact the authors, especially *
21 * if you plan to publish the results obtained with such modified code. *
22 * Any publication of results obtained using this code must include the *
23 * reference to nucl-th/0504047 and the published version of it, when *
24 * available. *
25 * *
26 *****************************************************************************/
27#include <iostream>
28#include <TFile.h>
29#include <TH1D.h>
30#include "Particle.h"
31#include "Event.h"
32#include "Integrator.h"
33#include "ParticleDecayer.h"
34#include "DecayTable.h"
35#include "ReadPar.h"
36
37extern ReadPar* sRPInstance;
38
39Event::Event(ParticleDB *aDB, Integrator* aInteg) :
40 mCurIter(NULL),
41 mScanStarted(0)
42{
43 mDB = aDB;
44 mInteg = aInteg;
45 mRandom = new TRandom2();
46#ifdef _ROOT_4_
47 mRandom->SetSeed2(31851, 14327);
48#else
49 mRandom->SetSeed(31851);
50#endif
51 mAverageMultiplicities.resize(mDB->GetParticleTypeCount());
52 mMultiplicities.resize(mDB->GetParticleTypeCount());
53 ReadParameters();
54 ReadMultiplicities();
55
56
57}
58
59Event::~Event()
60{
61
62 mParticles.clear();
63 delete mRandom;
64}
65
66void
67Event::GenerateEvent(int aSeed)
68{
69 Particle **tPartBuf;
70
71 mParticles.clear();
72#ifdef _ROOT_4_
73 if (aSeed) mRandom->SetSeed2(aSeed, (aSeed*2) % (7*11*23*31));
74#else
75 if (aSeed) mRandom->SetSeed(aSeed);
76#endif
77 GenerateMultiplicities();
78 for (unsigned int tIter=0; tIter<mAverageMultiplicities.size(); tIter++) {
79 mInteg->Generate(mDB->GetParticleType(tIter), mMultiplicities[tIter], &tPartBuf);
80 for (int tBufIter=0; tBufIter<mMultiplicities[tIter]; tBufIter++) {
81 AddParticle(tPartBuf[tBufIter]);
82 delete tPartBuf[tBufIter];
83 }
84 free(tPartBuf);
85 }
86}
87
88int
89Event::GetParticleCount()
90{
91 return mParticles.size();
92}
93
94Particle*
95Event::GetNextParticle()
96{
97 if (mScanStarted)
98 InitParticleScan();
99 if (mCurIter == mParticles.end())
100 return 0;
101
102 Particle *tBuf;
103 tBuf = &(*mCurIter);
104 mCurIter++;
105
106 return tBuf;
107}
108
109void
110Event::InitParticleScan()
111{
112 mCurIter = mParticles.begin();
113 mScanStarted = 1;
114}
115
116Particle*
117Event::GetParticleOfCount(int aCount)
118{
119 if (aCount == 0) {
120 InitParticleScan();
121 }
122 else if (aCount == 1) {
123 InitParticleScan();
124 mCurIter++;
125 }
126 else {
127 mCurIter--;
128 mCurIter++;
129 mCurIter++;
130 }
131
132 if (mCurIter == mParticles.end())
133 return 0;
134
135 Particle *tBuf;
136 tBuf = &(*mCurIter);
137 return tBuf;
138}
139
140void
141Event::WriteEvent(int nEvent)
142{
143 ParticleListIter tPLIter;
144 tPLIter = mParticles.begin();
145 int tCount = 0;
146 ofstream os;
147
148 if (nEvent == 0) {
149 try {
150 os.open(sRPInstance->getPar("EventOutputFile").Data());
151 }
152 catch (STR e) {
153 PRINT_MESSAGE("Very strange: No event output filename ???");
154 exit(0);
155 }
156 }
157 else {
158 try {
159 os.open(sRPInstance->getPar("EventOutputFile").Data(), ios::app);
160 }
161 catch (STR e) {
162 PRINT_MESSAGE("Very strange: No event output filename ???");
163 exit(0);
164 }
165 }
166
167
168
169 if (nEvent == 0)
170 {
171 os << "Therminator text output\nall_particles_p_x\nTherminator " << endl;
172 }
173 os << nEvent+1 << " " << GetParticleCount() << " " << "0.000" << " " << "0.000" << endl;
174
175 while (tPLIter != mParticles.end())
176 {
177 os.flags(ios::right);
178 os.precision(6);
179 os.width(6);
180 os << tCount << " ";
181 (*tPLIter).WriteParticle(&os);
182 os << endl;
183
184 tCount++;
185 tPLIter++;
186 }
187}
188
189
190void
191Event::AddParticle(Particle* aParticle)
192{
193 if (0) {
194 ParticleListIter tPLIter;
195 tPLIter = mParticles.begin();
196 while (tPLIter != mParticles.end())
197 {
198 if (((*tPLIter).GetMass() == aParticle->GetMass()) &&
199 ((*tPLIter).GetI3() == aParticle->GetI3()) &&
200 ((*tPLIter).GetBarionN() == aParticle->GetBarionN()) &&
201 ((*tPLIter).GetStrangeness() == aParticle->GetStrangeness()))
202 break;
203 if ((*tPLIter).GetMass() < aParticle->GetMass())
204 break;
205 else if ((*tPLIter).GetMass() == aParticle->GetMass())
206 if ((*tPLIter).GetI3() < aParticle->GetI3())
207 break;
208 else if ((*tPLIter).GetI3() == aParticle->GetI3())
209 if ((*tPLIter).GetBarionN() < aParticle->GetBarionN())
210 break;
211 else if ((*tPLIter).GetBarionN() == aParticle->GetBarionN())
212 if ((*tPLIter).GetStrangeness() < aParticle->GetStrangeness())
213 break;
214 tPLIter++;
215 }
216 mParticles.insert(tPLIter, *aParticle);
217 }
218 else
219 mParticles.push_back(*aParticle);
220}
221
222void
223Event::ReadMultiplicities()
224{
225 char tName[200];
226 double tMult;
227
228 char *tHash;
229 char tMultName[100];
230 ifstream *fin = NULL;
231
232 tHash = mInteg->ParameterHash();
233
234 strcpy(tMultName, "fmultiplicity_");
235 strcat(tMultName, tHash);
236 strcat(tMultName, ".txt");
237
238 fin = new ifstream(tMultName);
239 if ((fin) && (fin->is_open())) {
240 PRINT_MESSAGE("Reading Multiplicities from " << tMultName);
241
242 while (!fin->eof())
243 {
244 (*fin) >> tName >> tMult;
245 PRINT_DEBUG_2(tName << " " << mDB->GetParticleTypeIndex(strdup(tName)) << " " << tMult);
246 mAverageMultiplicities[mDB->GetParticleTypeIndex(strdup(tName))] = tMult;
247 }
248 fin->close();
249 }
250}
251
252void
253Event::GenerateMultiplicities()
254{
255 for (unsigned int tIter=0; tIter<mAverageMultiplicities.size(); tIter++)
256 mMultiplicities[tIter] = mRandom->Poisson(mAverageMultiplicities[tIter]);
257}
258
259void
260Event::DecayParticles()
261{
262 Particle *tPart1, *tPart2, *tPart3, *tFather;
263 ParticleDecayer* tDecayer;
264 int tCount = 0;
265
266 tDecayer = new ParticleDecayer(mDB);
267 tDecayer->SeedSet(mRandom->Integer(100000000));
268
269 // InitParticleScan();
270 while ((tFather = GetParticleOfCount(tCount)))
271 {
272 if (tFather->GetParticleType()->GetGamma() >= 0.0)
273 if ((tFather->GetParticleType()->GetTable()) && (((DecayTable *) tFather->GetParticleType()->GetTable())->GetChannelCount()+1 > 0))
274 {
275 tDecayer->DecayParticle(tFather, &tPart1, &tPart2, &tPart3);
276#ifndef _RESCALE_CHANNELS_
277 if (!tPart1) {
278 tCount++;
279 continue;
280 }
281#endif
282#ifdef _NO_THREE_BODY_DECAYS_
283 if (tPart3){
284 tCount++;
285 continue;
286 }
287#endif
288#ifdef _OMIT_TWO_BODY_
289 if (tPart1 && tPart2 && !tPart3){
290 tCount++;
291 continue;
292 }
293#endif
294
295 tPart1->SetFather(tCount);
296 tPart2->SetFather(tCount);
297 if (tPart3)
298 tPart3->SetFather(tCount);
299
300 AddParticle(tPart1);
301 AddParticle(tPart2);
302
303 delete tPart1;
304 delete tPart2;
305
306 if (tPart3) {
307 AddParticle(tPart3);
308 delete tPart3;
309 }
310 }
311 else
312 {
313 }
314 tCount++;
315 }
316
317 delete tDecayer;
318}
319
320void
321Event::Randomize()
322{
323 TDatime dat;
324
325#ifdef _ROOT_4_
326 mRandom->SetSeed2(dat.Get() / 2 * 3, dat.Get() / 11 * 9);
327#else
328 mRandom->SetSeed(dat.Get() / 2 * 3);
329#endif
330}
331
332void
333Event::ReadParameters()
334{
335 STR tEvFile;
336 STR tUseNegativeBinomial;
337
338 try {
339 tEvFile = sRPInstance->getPar("EventOutputFile");
340 }
341 catch (STR e) {
342 PRINT_DEBUG_1("Event::ReadParameters - Caught exception " << e);
343 PRINT_MESSAGE("Did not find one of the neccessary parameters in the parameters file.");
344 exit(0);
345 }
346 try {
347 tUseNegativeBinomial = sRPInstance->getPar("MultiplicityDistribution");
348 if (tUseNegativeBinomial.Contains("NegativeBinomial"))
349 mNegBin = 1;
350 else
351 mNegBin = 0;
352 }
353 catch (STR e) {
354 PRINT_MESSAGE("Using default multiplicty distribution: Poissonian");
355 mNegBin = 0;
356 }
357
358}
359