]>
Commit | Line | Data |
---|---|---|
8d2cd130 | 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 | ||
7cdba479 | 16 | /* $Id$ */ |
8d2cd130 | 17 | |
18 | #include "AliPythia.h" | |
7cdba479 | 19 | #include "AliPythiaRndm.h" |
8d2cd130 | 20 | |
21 | ClassImp(AliPythia) | |
22 | ||
23 | #ifndef WIN32 | |
24 | # define pyclus pyclus_ | |
25 | # define pycell pycell_ | |
452af8c7 | 26 | # define pyshow pyshow_ |
27 | # define pyrobo pyrobo_ | |
8d2cd130 | 28 | # define type_of_call |
29 | #else | |
30 | # define pyclus PYCLUS | |
31 | # define pycell PYCELL | |
452af8c7 | 32 | # define pyrobo PYROBO |
8d2cd130 | 33 | # define type_of_call _stdcall |
34 | #endif | |
35 | ||
36 | extern "C" void type_of_call pyclus(Int_t & ); | |
37 | extern "C" void type_of_call pycell(Int_t & ); | |
452af8c7 | 38 | extern "C" void type_of_call pyshow(Int_t &, Int_t &, Double_t &); |
39 | extern "C" void type_of_call pyrobo(Int_t &, Int_t &, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &); | |
8d2cd130 | 40 | |
41 | //_____________________________________________________________________________ | |
42 | ||
43 | AliPythia* AliPythia::fgAliPythia=NULL; | |
44 | ||
45 | AliPythia::AliPythia() | |
46 | { | |
47 | // Default Constructor | |
48 | // | |
49 | // Set random number | |
7cdba479 | 50 | if (!AliPythiaRndm::GetPythiaRandom()) |
51 | AliPythiaRndm::SetPythiaRandom(GetRandom()); | |
8d2cd130 | 52 | |
53 | } | |
54 | ||
55 | void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfunc) | |
56 | { | |
57 | // Initialise the process to generate | |
7cdba479 | 58 | if (!AliPythiaRndm::GetPythiaRandom()) |
59 | AliPythiaRndm::SetPythiaRandom(GetRandom()); | |
8d2cd130 | 60 | |
61 | fProcess = process; | |
62 | fEcms = energy; | |
63 | fStrucFunc = strucfunc; | |
64 | // don't decay p0 | |
65 | SetMDCY(Pycomp(111),1,0); | |
66 | // select structure function | |
67 | SetMSTP(52,2); | |
68 | SetMSTP(51,strucfunc); | |
69 | // | |
70 | // Pythia initialisation for selected processes// | |
71 | // | |
72 | // Make MSEL clean | |
73 | // | |
74 | for (Int_t i=1; i<= 200; i++) { | |
75 | SetMSUB(i,0); | |
76 | } | |
77 | // select charm production | |
78 | switch (process) | |
79 | { | |
80 | case kPyCharm: | |
81 | SetMSEL(4); | |
82 | // | |
83 | // heavy quark masses | |
84 | ||
85 | SetPMAS(4,1,1.2); | |
86 | SetMSTU(16,2); | |
87 | // | |
88 | // primordial pT | |
89 | SetMSTP(91,1); | |
90 | SetPARP(91,1.); | |
91 | SetPARP(93,5.); | |
92 | // | |
93 | break; | |
94 | case kPyBeauty: | |
95 | SetMSEL(5); | |
96 | SetPMAS(5,1,4.75); | |
97 | SetMSTU(16,2); | |
98 | break; | |
99 | case kPyJpsi: | |
100 | SetMSEL(0); | |
101 | // gg->J/Psi g | |
102 | SetMSUB(86,1); | |
103 | break; | |
104 | case kPyJpsiChi: | |
105 | SetMSEL(0); | |
106 | // gg->J/Psi g | |
107 | SetMSUB(86,1); | |
108 | // gg-> chi_0c g | |
109 | SetMSUB(87,1); | |
110 | // gg-> chi_1c g | |
111 | SetMSUB(88,1); | |
112 | // gg-> chi_2c g | |
113 | SetMSUB(89,1); | |
114 | break; | |
115 | case kPyCharmUnforced: | |
116 | SetMSEL(0); | |
117 | // gq->qg | |
118 | SetMSUB(28,1); | |
119 | // gg->qq | |
120 | SetMSUB(53,1); | |
121 | // gg->gg | |
122 | SetMSUB(68,1); | |
123 | break; | |
124 | case kPyBeautyUnforced: | |
125 | SetMSEL(0); | |
126 | // gq->qg | |
127 | SetMSUB(28,1); | |
128 | // gg->qq | |
129 | SetMSUB(53,1); | |
130 | // gg->gg | |
131 | SetMSUB(68,1); | |
132 | break; | |
133 | case kPyMb: | |
134 | // Minimum Bias pp-Collisions | |
135 | // | |
136 | // | |
137 | // select Pythia min. bias model | |
138 | SetMSEL(0); | |
511db649 | 139 | SetMSUB(92,1); // single diffraction AB-->XB |
140 | SetMSUB(93,1); // single diffraction AB-->AX | |
141 | SetMSUB(94,1); // double diffraction | |
142 | SetMSUB(95,1); // low pt production | |
143 | ||
144 | // | |
145 | // ATLAS Tuning | |
146 | // | |
147 | SetMSTP(51, 7); // CTEQ5L pdf | |
148 | SetMSTP(81,1); // Multiple Interactions ON | |
149 | SetMSTP(82,4); // Double Gaussian Model | |
150 | ||
151 | SetPARP(82,1.8); // [GeV] PT_min at Ref. energy | |
152 | SetPARP(89,1000.); // [GeV] Ref. energy | |
153 | SetPARP(90,0.16); // 2*epsilon (exponent in power law) | |
154 | SetPARP(83,0.5); // Core density in proton matter distribution (def.value) | |
155 | SetPARP(84,0.5); // Core radius | |
156 | SetPARP(85,0.33); // Regulates gluon prod. mechanism | |
157 | SetPARP(86,0.66); // Regulates gluon prod. mechanism | |
158 | SetPARP(67,1); // Regulates Initial State Radiation | |
159 | break; | |
8d2cd130 | 160 | case kPyMbNonDiffr: |
161 | // Minimum Bias pp-Collisions | |
162 | // | |
163 | // | |
164 | // select Pythia min. bias model | |
165 | SetMSEL(0); | |
511db649 | 166 | SetMSUB(95,1); // low pt production |
167 | ||
168 | SetMSTP(51, 7); // CTEQ5L pdf | |
169 | SetMSTP(81,1); // Multiple Interactions ON | |
170 | SetMSTP(82,4); // Double Gaussian Model | |
171 | ||
172 | SetPARP(82,1.8); // [GeV] PT_min at Ref. energy | |
173 | SetPARP(89,1000.); // [GeV] Ref. energy | |
174 | SetPARP(90,0.16); // 2*epsilon (exponent in power law) | |
175 | SetPARP(83,0.5); // Core density in proton matter distribution (def.value) | |
176 | SetPARP(84,0.5); // Core radius | |
177 | SetPARP(85,0.33); // Regulates gluon prod. mechanism | |
178 | SetPARP(86,0.66); // Regulates gluon prod. mechanism | |
179 | SetPARP(67,1); // Regulates Initial State Radiation | |
8d2cd130 | 180 | break; |
181 | case kPyJets: | |
182 | // | |
183 | // QCD Jets | |
184 | // | |
185 | SetMSEL(1); | |
186 | break; | |
187 | case kPyDirectGamma: | |
188 | SetMSEL(10); | |
189 | break; | |
adf4d898 | 190 | case kPyCharmPbPbMNR: |
191 | case kPyD0PbPbMNR: | |
8d2cd130 | 192 | // Tuning of Pythia parameters aimed to get a resonable agreement |
193 | // between with the NLO calculation by Mangano, Nason, Ridolfi for the | |
194 | // c-cbar single inclusive and double differential distributions. | |
195 | // This parameter settings are meant to work with Pb-Pb collisions | |
adf4d898 | 196 | // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs. |
8d2cd130 | 197 | // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard) |
198 | // has to be set to 2.1GeV. Example in ConfigCharmPPR.C. | |
199 | ||
200 | // All QCD processes | |
201 | SetMSEL(1); | |
202 | ||
203 | // No multiple interactions | |
204 | SetMSTP(81,0); | |
205 | SetPARP(81,0.0); | |
206 | SetPARP(82,0.0); | |
207 | ||
208 | // Initial/final parton shower on (Pythia default) | |
209 | SetMSTP(61,1); | |
210 | SetMSTP(71,1); | |
211 | ||
212 | // 2nd order alpha_s | |
213 | SetMSTP(2,2); | |
214 | ||
215 | // QCD scales | |
216 | SetMSTP(32,2); | |
217 | SetPARP(34,1.0); | |
218 | ||
adf4d898 | 219 | // Intrinsic <kT> |
8d2cd130 | 220 | SetMSTP(91,1); |
221 | SetPARP(91,1.304); | |
222 | SetPARP(93,6.52); | |
223 | ||
224 | // Set c-quark mass | |
225 | SetPMAS(4,1,1.2); | |
226 | ||
227 | break; | |
adf4d898 | 228 | case kPyCharmpPbMNR: |
229 | case kPyD0pPbMNR: | |
230 | // Tuning of Pythia parameters aimed to get a resonable agreement | |
231 | // between with the NLO calculation by Mangano, Nason, Ridolfi for the | |
232 | // c-cbar single inclusive and double differential distributions. | |
233 | // This parameter settings are meant to work with p-Pb collisions | |
234 | // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs. | |
235 | // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard) | |
236 | // has to be set to 2.1GeV. Example in ConfigCharmPPR.C. | |
237 | ||
238 | // All QCD processes | |
239 | SetMSEL(1); | |
240 | ||
241 | // No multiple interactions | |
242 | SetMSTP(81,0); | |
243 | SetPARP(81,0.0); | |
244 | SetPARP(82,0.0); | |
245 | ||
246 | // Initial/final parton shower on (Pythia default) | |
247 | SetMSTP(61,1); | |
248 | SetMSTP(71,1); | |
249 | ||
250 | // 2nd order alpha_s | |
251 | SetMSTP(2,2); | |
252 | ||
253 | // QCD scales | |
254 | SetMSTP(32,2); | |
255 | SetPARP(34,1.0); | |
256 | ||
257 | // Intrinsic <kT> | |
258 | SetMSTP(91,1); | |
259 | SetPARP(91,1.16); | |
260 | SetPARP(93,5.8); | |
261 | ||
262 | // Set c-quark mass | |
263 | SetPMAS(4,1,1.2); | |
264 | ||
265 | break; | |
266 | case kPyCharmppMNR: | |
267 | case kPyD0ppMNR: | |
268 | // Tuning of Pythia parameters aimed to get a resonable agreement | |
269 | // between with the NLO calculation by Mangano, Nason, Ridolfi for the | |
270 | // c-cbar single inclusive and double differential distributions. | |
271 | // This parameter settings are meant to work with pp collisions | |
272 | // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs. | |
273 | // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard) | |
274 | // has to be set to 2.1GeV. Example in ConfigCharmPPR.C. | |
275 | ||
276 | // All QCD processes | |
277 | SetMSEL(1); | |
278 | ||
279 | // No multiple interactions | |
280 | SetMSTP(81,0); | |
281 | SetPARP(81,0.0); | |
282 | SetPARP(82,0.0); | |
283 | ||
284 | // Initial/final parton shower on (Pythia default) | |
285 | SetMSTP(61,1); | |
286 | SetMSTP(71,1); | |
287 | ||
288 | // 2nd order alpha_s | |
289 | SetMSTP(2,2); | |
290 | ||
291 | // QCD scales | |
292 | SetMSTP(32,2); | |
293 | SetPARP(34,1.0); | |
294 | ||
295 | // Intrinsic <kT^2> | |
296 | SetMSTP(91,1); | |
297 | SetPARP(91,1.); | |
298 | SetPARP(93,5.); | |
299 | ||
300 | // Set c-quark mass | |
301 | SetPMAS(4,1,1.2); | |
302 | ||
303 | break; | |
304 | case kPyBeautyPbPbMNR: | |
8d2cd130 | 305 | // Tuning of Pythia parameters aimed to get a resonable agreement |
306 | // between with the NLO calculation by Mangano, Nason, Ridolfi for the | |
307 | // b-bbar single inclusive and double differential distributions. | |
308 | // This parameter settings are meant to work with Pb-Pb collisions | |
309 | // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs. | |
310 | // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard) | |
311 | // has to be set to 2.75GeV. Example in ConfigBeautyPPR.C. | |
312 | ||
313 | // All QCD processes | |
314 | SetMSEL(1); | |
315 | ||
316 | // No multiple interactions | |
317 | SetMSTP(81,0); | |
318 | SetPARP(81,0.0); | |
319 | SetPARP(82,0.0); | |
320 | ||
321 | // Initial/final parton shower on (Pythia default) | |
322 | SetMSTP(61,1); | |
323 | SetMSTP(71,1); | |
324 | ||
325 | // 2nd order alpha_s | |
326 | SetMSTP(2,2); | |
327 | ||
328 | // QCD scales | |
329 | SetMSTP(32,2); | |
330 | SetPARP(34,1.0); | |
331 | SetPARP(67,1.0); | |
332 | SetPARP(71,1.0); | |
333 | ||
adf4d898 | 334 | // Intrinsic <kT> |
8d2cd130 | 335 | SetMSTP(91,1); |
336 | SetPARP(91,2.035); | |
337 | SetPARP(93,10.17); | |
338 | ||
339 | // Set b-quark mass | |
340 | SetPMAS(5,1,4.75); | |
341 | ||
adf4d898 | 342 | break; |
343 | case kPyBeautypPbMNR: | |
344 | // Tuning of Pythia parameters aimed to get a resonable agreement | |
345 | // between with the NLO calculation by Mangano, Nason, Ridolfi for the | |
346 | // b-bbar single inclusive and double differential distributions. | |
347 | // This parameter settings are meant to work with p-Pb collisions | |
348 | // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs. | |
349 | // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard) | |
350 | // has to be set to 2.75GeV. Example in ConfigBeautyPPR.C. | |
351 | ||
352 | // All QCD processes | |
353 | SetMSEL(1); | |
354 | ||
355 | // No multiple interactions | |
356 | SetMSTP(81,0); | |
357 | SetPARP(81,0.0); | |
358 | SetPARP(82,0.0); | |
359 | ||
360 | // Initial/final parton shower on (Pythia default) | |
361 | SetMSTP(61,1); | |
362 | SetMSTP(71,1); | |
363 | ||
364 | // 2nd order alpha_s | |
365 | SetMSTP(2,2); | |
366 | ||
367 | // QCD scales | |
368 | SetMSTP(32,2); | |
369 | SetPARP(34,1.0); | |
370 | SetPARP(67,1.0); | |
371 | SetPARP(71,1.0); | |
372 | ||
373 | // Intrinsic <kT> | |
374 | SetMSTP(91,1); | |
375 | SetPARP(91,1.60); | |
376 | SetPARP(93,8.00); | |
377 | ||
378 | // Set b-quark mass | |
379 | SetPMAS(5,1,4.75); | |
380 | ||
381 | break; | |
382 | case kPyBeautyppMNR: | |
383 | // Tuning of Pythia parameters aimed to get a resonable agreement | |
384 | // between with the NLO calculation by Mangano, Nason, Ridolfi for the | |
385 | // b-bbar single inclusive and double differential distributions. | |
386 | // This parameter settings are meant to work with pp collisions | |
387 | // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs. | |
388 | // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard) | |
389 | // has to be set to 2.75GeV. Example in ConfigBeautyPPR.C. | |
390 | ||
391 | // All QCD processes | |
392 | SetMSEL(1); | |
393 | ||
394 | // No multiple interactions | |
395 | SetMSTP(81,0); | |
396 | SetPARP(81,0.0); | |
397 | SetPARP(82,0.0); | |
398 | ||
399 | // Initial/final parton shower on (Pythia default) | |
400 | SetMSTP(61,1); | |
401 | SetMSTP(71,1); | |
402 | ||
403 | // 2nd order alpha_s | |
404 | SetMSTP(2,2); | |
405 | ||
406 | // QCD scales | |
407 | SetMSTP(32,2); | |
408 | SetPARP(34,1.0); | |
409 | SetPARP(67,1.0); | |
410 | SetPARP(71,1.0); | |
411 | ||
412 | // Intrinsic <kT> | |
413 | SetMSTP(91,1); | |
414 | SetPARP(91,1.); | |
415 | SetPARP(93,5.); | |
416 | ||
417 | // Set b-quark mass | |
418 | SetPMAS(5,1,4.75); | |
419 | ||
8d2cd130 | 420 | break; |
421 | } | |
422 | // | |
423 | // Initialize PYTHIA | |
424 | SetMSTP(41,1); // all resonance decays switched on | |
425 | ||
426 | Initialize("CMS","p","p",fEcms); | |
427 | ||
428 | } | |
429 | ||
430 | Int_t AliPythia::CheckedLuComp(Int_t kf) | |
431 | { | |
432 | // Check Lund particle code (for debugging) | |
433 | Int_t kc=Pycomp(kf); | |
434 | printf("\n Lucomp kf,kc %d %d",kf,kc); | |
435 | return kc; | |
436 | } | |
437 | ||
438 | void AliPythia::SetNuclei(Int_t a1, Int_t a2) | |
439 | { | |
440 | // Treat protons as inside nuclei with mass numbers a1 and a2 | |
441 | // The MSTP array in the PYPARS common block is used to enable and | |
442 | // select the nuclear structure functions. | |
443 | // MSTP(52) : (D=1) choice of proton and nuclear structure-function library | |
444 | // =1: internal PYTHIA acording to MSTP(51) | |
445 | // =2: PDFLIB proton s.f., with MSTP(51) = 1000xNGROUP+NSET | |
446 | // If the following mass number both not equal zero, nuclear corrections of the stf are used. | |
447 | // MSTP(192) : Mass number of nucleus side 1 | |
448 | // MSTP(193) : Mass number of nucleus side 2 | |
449 | SetMSTP(52,2); | |
450 | SetMSTP(192, a1); | |
451 | SetMSTP(193, a2); | |
452 | } | |
453 | ||
454 | ||
455 | AliPythia* AliPythia::Instance() | |
456 | { | |
457 | // Set random number generator | |
458 | if (fgAliPythia) { | |
459 | return fgAliPythia; | |
460 | } else { | |
461 | fgAliPythia = new AliPythia(); | |
462 | return fgAliPythia; | |
463 | } | |
464 | } | |
465 | ||
466 | void AliPythia::PrintParticles() | |
467 | { | |
468 | // Print list of particl properties | |
469 | Int_t np = 0; | |
c31f1d37 | 470 | char* name = new char[16]; |
8d2cd130 | 471 | for (Int_t kf=0; kf<1000000; kf++) { |
472 | for (Int_t c = 1; c > -2; c-=2) { | |
8d2cd130 | 473 | Int_t kc = Pycomp(c*kf); |
474 | if (kc) { | |
475 | Float_t mass = GetPMAS(kc,1); | |
476 | Float_t width = GetPMAS(kc,2); | |
477 | Float_t tau = GetPMAS(kc,4); | |
c31f1d37 | 478 | |
8d2cd130 | 479 | Pyname(kf,name); |
480 | ||
481 | np++; | |
482 | ||
483 | printf("\n mass, width, tau: %6d %s %10.3f %10.3e %10.3e", | |
484 | c*kf, name, mass, width, tau); | |
485 | } | |
486 | } | |
487 | } | |
488 | printf("\n Number of particles %d \n \n", np); | |
489 | } | |
490 | ||
491 | void AliPythia::ResetDecayTable() | |
492 | { | |
493 | // Set default values for pythia decay switches | |
494 | Int_t i; | |
495 | for (i = 1; i < 501; i++) SetMDCY(i,1,fDefMDCY[i]); | |
496 | for (i = 1; i < 2001; i++) SetMDME(i,1,fDefMDME[i]); | |
497 | } | |
498 | ||
499 | void AliPythia::SetDecayTable() | |
500 | { | |
501 | // Set default values for pythia decay switches | |
502 | // | |
503 | Int_t i; | |
504 | for (i = 1; i < 501; i++) fDefMDCY[i] = GetMDCY(i,1); | |
505 | for (i = 1; i < 2001; i++) fDefMDME[i] = GetMDME(i,1); | |
506 | } | |
507 | ||
508 | void AliPythia::Pyclus(Int_t& njet) | |
509 | { | |
510 | // Call Pythia clustering algorithm | |
511 | // | |
512 | pyclus(njet); | |
513 | } | |
514 | ||
515 | void AliPythia::Pycell(Int_t& njet) | |
516 | { | |
517 | // Call Pythia jet reconstruction algorithm | |
518 | // | |
519 | pycell(njet); | |
520 | } | |
521 | ||
452af8c7 | 522 | void AliPythia::Pyshow(Int_t ip1, Int_t ip2, Double_t qmax) |
523 | { | |
524 | // Call Pythia jet reconstruction algorithm | |
525 | // | |
526 | Int_t numpart = fPyjets->N; | |
527 | for (Int_t i = 0; i < numpart; i++) | |
528 | { | |
529 | if (fPyjets->K[2][i] == 7) ip1 = i+1; | |
530 | if (fPyjets->K[2][i] == 8) ip2 = i+1; | |
531 | } | |
532 | ||
533 | ||
534 | qmax = 2. * GetVINT(51); | |
535 | printf("Pyshow %d %d %f", ip1, ip2, qmax); | |
536 | ||
537 | pyshow(ip1, ip2, qmax); | |
538 | } | |
539 | ||
540 | void AliPythia::Pyrobo(Int_t imi, Int_t ima, Double_t the, Double_t phi, Double_t bex, Double_t bey, Double_t bez) | |
541 | { | |
542 | pyrobo(imi, ima, the, phi, bex, bey, bez); | |
543 | } | |
544 | ||
545 | ||
546 | ||
547 | void AliPythia::Quench() | |
548 | { | |
549 | // | |
550 | // | |
551 | // Simple Jet Quenching routine: | |
552 | // ============================= | |
553 | // The jet formed by all final state partons radiated by the parton created | |
554 | // in the hard collisions is quenched by a factor z using: | |
555 | // (E + p_z)new = (1-z) (E + p_z)old | |
556 | // | |
557 | // The lost momentum is first balanced by one gluon with virtuality > 0. | |
558 | // Subsequently the gluon splits to yield two gluons with E = p. | |
559 | // | |
560 | Float_t p0[2][5]; | |
561 | Float_t p1[2][5]; | |
562 | Float_t p2[2][5]; | |
563 | Int_t klast[2] = {-1, -1}; | |
564 | Int_t kglu[2]; | |
565 | for (Int_t i = 0; i < 4; i++) | |
566 | { | |
567 | p0[0][i] = 0.; | |
568 | p0[1][i] = 0.; | |
569 | p1[0][i] = 0.; | |
570 | p1[1][i] = 0.; | |
571 | p2[0][i] = 0.; | |
572 | p2[1][i] = 0.; | |
573 | } | |
574 | ||
575 | Int_t numpart = fPyjets->N; | |
576 | ||
577 | for (Int_t i = 0; i < numpart; i++) | |
578 | { | |
579 | Int_t imo = fPyjets->K[2][i]; | |
580 | Int_t kst = fPyjets->K[0][i]; | |
581 | Int_t pdg = fPyjets->K[1][i]; | |
582 | ||
583 | // Quarks and gluons only | |
584 | if (pdg != 21 && TMath::Abs(pdg) > 6) continue; | |
585 | ||
586 | // Particles from hard scattering only | |
587 | ||
588 | ||
589 | Float_t px = fPyjets->P[0][i]; | |
590 | Float_t py = fPyjets->P[1][i]; | |
591 | Float_t pz = fPyjets->P[2][i]; | |
592 | Float_t e = fPyjets->P[3][i]; | |
593 | Float_t m = fPyjets->P[4][i]; | |
594 | Float_t pt = TMath::Sqrt(px * px + py * py); | |
595 | // Skip comment lines | |
596 | if (kst != 1 && kst != 2) continue; | |
597 | ||
598 | Float_t mt = TMath::Sqrt(px * px + py * py + m * m); | |
599 | ||
600 | // | |
601 | // Some cuts to be in a save kinematic region | |
602 | // | |
603 | if (imo != 7 && imo != 8) continue; | |
604 | Int_t index = imo - 7; | |
605 | klast[index] = i; | |
606 | ||
607 | p0[index][0] += px; | |
608 | p0[index][1] += py; | |
609 | p0[index][2] += pz; | |
511db649 | 610 | p0[index][3] += e; |
611 | // | |
612 | // Fix z | |
613 | // | |
452af8c7 | 614 | |
452af8c7 | 615 | Float_t z = 0.2; |
616 | Float_t eppzOld = e + pz; | |
617 | Float_t empzOld = e - pz; | |
618 | ||
619 | ||
620 | // | |
621 | // Kinematics of the original parton | |
622 | // | |
623 | ||
624 | Float_t eppzNew = (1. - z) * eppzOld; | |
625 | Float_t empzNew = empzOld - mt * mt * z / eppzOld; | |
626 | Float_t eNew0 = 0.5 * (eppzNew + empzNew); | |
627 | Float_t pzNew0 = 0.5 * (eppzNew - empzNew); | |
628 | // | |
629 | // Skip if pt too small | |
630 | // | |
631 | if (m * m > eppzNew * empzNew) continue; | |
632 | Float_t ptNew = TMath::Sqrt(eppzNew * empzNew - m * m); | |
633 | Float_t pxNew0 = ptNew / pt * px; | |
634 | Float_t pyNew0 = ptNew / pt * py; | |
635 | ||
636 | p1[index][0] += pxNew0; | |
637 | p1[index][1] += pyNew0; | |
638 | p1[index][2] += pzNew0; | |
639 | p1[index][3] += eNew0; | |
640 | // | |
641 | // Update event record | |
642 | // | |
643 | fPyjets->P[0][i] = pxNew0; | |
644 | fPyjets->P[1][i] = pyNew0; | |
645 | fPyjets->P[2][i] = pzNew0; | |
646 | fPyjets->P[3][i] = eNew0; | |
647 | ||
648 | } | |
649 | ||
650 | // | |
651 | // Gluons | |
652 | // | |
653 | ||
654 | for (Int_t k = 0; k < 2; k++) | |
655 | { | |
656 | for (Int_t j = 0; j < 4; j++) | |
657 | { | |
658 | p2[k][j] = p0[k][j] - p1[k][j]; | |
659 | } | |
660 | p2[k][4] = p2[k][3] * p2[k][3] - p2[k][0] * p2[k][0] - p2[k][1] * p2[k][1] - p2[k][2] * p2[k][2]; | |
661 | ||
662 | if (p2[k][4] > 0.) | |
663 | { | |
664 | ||
665 | // | |
666 | // Bring gluon back to mass shell via momentum scaling | |
667 | // (momentum will not be conserved, but energy) | |
668 | // | |
669 | // not used anymore | |
670 | /* | |
671 | Float_t psq = p2[k][0] * p2[k][0] + p2[k][1] * p2[k][1] + p2[k][2] * p2[k][2]; | |
672 | Float_t fact = TMath::Sqrt(1. + p2[k][4] / psq); | |
673 | p2[k][0] *= fact; | |
674 | p2[k][1] *= fact; | |
675 | p2[k][2] *= fact; | |
676 | p2[k][3] = TMath::Sqrt(psq) * fact; | |
677 | p2[k][4] = 0.; | |
678 | */ | |
679 | } | |
680 | } | |
681 | ||
682 | if (p2[0][4] > 0.) { | |
683 | p2[0][4] = TMath::Sqrt(p2[0][4]); | |
684 | } else { | |
685 | printf("Warning negative mass squared ! \n"); | |
686 | } | |
687 | ||
688 | if (p2[1][4] > 0.) { | |
689 | p2[1][4] = TMath::Sqrt(p2[1][4]); | |
690 | } else { | |
691 | printf("Warning negative mass squared ! \n"); | |
692 | } | |
693 | ||
694 | // | |
695 | // Add the gluons | |
696 | // | |
697 | ||
698 | ||
699 | for (Int_t i = 0; i < 2; i++) { | |
700 | Int_t ish, jmin, jmax, iGlu, iNew; | |
701 | Int_t in = klast[i]; | |
702 | ish = 0; | |
703 | ||
704 | if (in == -1) continue; | |
705 | if (i == 1 && klast[1] > klast[0]) in += ish; | |
706 | ||
707 | jmin = in - 1; | |
708 | ish = 1; | |
709 | ||
710 | if (p2[i][4] > 0) ish = 2; | |
711 | ||
712 | iGlu = in; | |
713 | iNew = in + ish; | |
714 | jmax = numpart + ish - 1; | |
715 | ||
716 | if (fPyjets->K[0][in-1] == 1 || fPyjets->K[0][in-1] == 21 || fPyjets->K[0][in-1] == 11) { | |
717 | jmin = in; | |
718 | iGlu = in + 1; | |
719 | iNew = in; | |
720 | } | |
721 | ||
722 | kglu[i] = iGlu; | |
723 | ||
724 | for (Int_t j = jmax; j > jmin; j--) | |
725 | { | |
726 | for (Int_t k = 0; k < 5; k++) { | |
727 | fPyjets->K[k][j] = fPyjets->K[k][j-ish]; | |
728 | fPyjets->P[k][j] = fPyjets->P[k][j-ish]; | |
729 | fPyjets->V[k][j] = fPyjets->V[k][j-ish]; | |
730 | } | |
731 | } // end shifting | |
732 | numpart += ish; | |
733 | (fPyjets->N) += ish; | |
734 | ||
735 | if (ish == 1) { | |
736 | fPyjets->P[0][iGlu] = p2[i][0]; | |
737 | fPyjets->P[1][iGlu] = p2[i][1]; | |
738 | fPyjets->P[2][iGlu] = p2[i][2]; | |
739 | fPyjets->P[3][iGlu] = p2[i][3]; | |
740 | fPyjets->P[4][iGlu] = p2[i][4]; | |
741 | ||
742 | fPyjets->K[0][iGlu] = 2; | |
743 | fPyjets->K[1][iGlu] = 21; | |
744 | fPyjets->K[2][iGlu] = fPyjets->K[2][iNew]; | |
745 | fPyjets->K[3][iGlu] = -1; | |
746 | fPyjets->K[4][iGlu] = -1; | |
747 | } else { | |
748 | // | |
749 | // Split gluon in rest frame. | |
750 | // | |
751 | Double_t bx = p2[i][0] / p2[i][3]; | |
752 | Double_t by = p2[i][1] / p2[i][3]; | |
753 | Double_t bz = p2[i][2] / p2[i][3]; | |
754 | ||
755 | Float_t pst = p2[i][4] / 2.; | |
756 | ||
757 | Float_t cost = 2. * gRandom->Rndm() - 1.; | |
758 | Float_t sint = TMath::Sqrt(1. - cost * cost); | |
759 | Float_t phi = 2. * TMath::Pi() * gRandom->Rndm(); | |
760 | ||
761 | Float_t pz1 = pst * cost; | |
762 | Float_t pz2 = -pst * cost; | |
763 | Float_t pt1 = pst * sint; | |
764 | Float_t pt2 = -pst * sint; | |
765 | Float_t px1 = pt1 * TMath::Cos(phi); | |
766 | Float_t py1 = pt1 * TMath::Sin(phi); | |
767 | Float_t px2 = pt2 * TMath::Cos(phi); | |
768 | Float_t py2 = pt2 * TMath::Sin(phi); | |
769 | ||
770 | fPyjets->P[0][iGlu] = px1; | |
771 | fPyjets->P[1][iGlu] = py1; | |
772 | fPyjets->P[2][iGlu] = pz1; | |
773 | fPyjets->P[3][iGlu] = pst; | |
774 | fPyjets->P[4][iGlu] = 0.; | |
775 | ||
776 | fPyjets->K[0][iGlu] = 2; | |
777 | fPyjets->K[1][iGlu] = 21; | |
778 | fPyjets->K[2][iGlu] = fPyjets->K[2][iNew]; | |
779 | fPyjets->K[3][iGlu] = -1; | |
780 | fPyjets->K[4][iGlu] = -1; | |
781 | ||
782 | fPyjets->P[0][iGlu+1] = px2; | |
783 | fPyjets->P[1][iGlu+1] = py2; | |
784 | fPyjets->P[2][iGlu+1] = pz2; | |
785 | fPyjets->P[3][iGlu+1] = pst; | |
786 | fPyjets->P[4][iGlu+1] = 0.; | |
787 | ||
788 | fPyjets->K[0][iGlu+1] = 2; | |
789 | fPyjets->K[1][iGlu+1] = 21; | |
790 | fPyjets->K[2][iGlu+1] = fPyjets->K[2][iNew]; | |
791 | fPyjets->K[3][iGlu+1] = -1; | |
792 | fPyjets->K[4][iGlu+1] = -1; | |
793 | SetMSTU(1,0); | |
794 | SetMSTU(2,0); | |
795 | ||
796 | // | |
797 | // Boost back | |
798 | // | |
799 | Pyrobo(iGlu + 1, iGlu + 2, 0., 0., bx, by, bz); | |
800 | ||
801 | } | |
802 | } // end adding gluons | |
803 | } // end quench | |
804 | ||
805 |