For fastjet-3.x the defines were changed to have the additional prefix
[u/mrichter/AliRoot.git] / PYTHIA6 / AliPythia.cxx
CommitLineData
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"
12cb0bc0 20#include "AliFastGlauber.h"
21#include "AliQuenchingWeights.h"
03358a32 22#include "AliOmegaDalitz.h"
0f482ae4 23#include "TVector3.h"
03358a32 24#include "TLorentzVector.h"
12cb0bc0 25#include "PyquenCommon.h"
8d2cd130 26
27ClassImp(AliPythia)
28
29#ifndef WIN32
30# define pyclus pyclus_
31# define pycell pycell_
452af8c7 32# define pyshow pyshow_
33# define pyrobo pyrobo_
992f2843 34# define pyquen pyquen_
16a82508 35# define pyevnw pyevnw_
cd07c39b 36# define pyshowq pyshowq_
6c43eccb 37# define qpygin0 qpygin0_
694b39f9 38# define pytune pytune_
9b61ba2a 39# define py2ent py2ent_
8d2cd130 40# define type_of_call
41#else
42# define pyclus PYCLUS
43# define pycell PYCELL
452af8c7 44# define pyrobo PYROBO
992f2843 45# define pyquen PYQUEN
16a82508 46# define pyevnw PYEVNW
cd07c39b 47# define pyshowq PYSHOWQ
6c43eccb 48# define qpygin0 QPYGIN0
9b61ba2a 49# define pytune PYTUNE
50# define py2ent PY2ENT
8d2cd130 51# define type_of_call _stdcall
52#endif
53
54extern "C" void type_of_call pyclus(Int_t & );
55extern "C" void type_of_call pycell(Int_t & );
452af8c7 56extern "C" void type_of_call pyshow(Int_t &, Int_t &, Double_t &);
57extern "C" void type_of_call pyrobo(Int_t &, Int_t &, Double_t &, Double_t &, Double_t &, Double_t &, Double_t &);
992f2843 58extern "C" void type_of_call pyquen(Double_t &, Int_t &, Double_t &);
1bcf6e6c 59extern "C" void type_of_call pyevnw();
cd07c39b 60extern "C" void type_of_call pyshowq(Int_t &, Int_t &, Double_t &);
694b39f9 61extern "C" void type_of_call pytune(Int_t &);
9b61ba2a 62extern "C" void type_of_call py2ent(Int_t &, Int_t&, Int_t&, Double_t&);
6c43eccb 63extern "C" void type_of_call qpygin0();
8d2cd130 64//_____________________________________________________________________________
65
66AliPythia* AliPythia::fgAliPythia=NULL;
67
e8a8adcd 68AliPythia::AliPythia():
69 fProcess(kPyMb),
70 fEcms(0.),
71 fStrucFunc(kCTEQ5L),
72 fXJet(0.),
73 fYJet(0.),
32c8e463 74 fNGmax(30),
75 fZmax(0.97),
e8a8adcd 76 fGlauber(0),
0bd3d7c5 77 fQuenchingWeights(0),
03358a32 78 fItune(-1),
79 fOmegaDalitz()
8d2cd130 80{
81// Default Constructor
82//
83// Set random number
7cdba479 84 if (!AliPythiaRndm::GetPythiaRandom())
85 AliPythiaRndm::SetPythiaRandom(GetRandom());
0f482ae4 86 fGlauber = 0;
87 fQuenchingWeights = 0;
a23e397a 88 Int_t i = 0;
89 for (i = 0; i < 501; i++) fDefMDCY[i] = 0;
90 for (i = 0; i < 2001; i++) fDefMDME[i] = 0;
91 for (i = 0; i < 4; i++) fZQuench[i] = 0;
8d2cd130 92}
93
e8a8adcd 94AliPythia::AliPythia(const AliPythia& pythia):
6b435cde 95 TPythia6(pythia),
96 AliRndm(pythia),
e8a8adcd 97 fProcess(kPyMb),
98 fEcms(0.),
99 fStrucFunc(kCTEQ5L),
100 fXJet(0.),
101 fYJet(0.),
32c8e463 102 fNGmax(30),
103 fZmax(0.97),
e8a8adcd 104 fGlauber(0),
0bd3d7c5 105 fQuenchingWeights(0),
03358a32 106 fItune(-1),
107 fOmegaDalitz()
e8a8adcd 108{
109 // Copy Constructor
a23e397a 110 Int_t i;
111 for (i = 0; i < 501; i++) fDefMDCY[i] = 0;
112 for (i = 0; i < 2001; i++) fDefMDME[i] = 0;
113 for (i = 0; i < 4; i++) fZQuench[i] = 0;
e8a8adcd 114 pythia.Copy(*this);
115}
116
efe3b1cd 117void AliPythia::ProcInit(Process_t process, Float_t energy, StrucFunc_t strucfunc, Int_t itune)
8d2cd130 118{
119// Initialise the process to generate
7cdba479 120 if (!AliPythiaRndm::GetPythiaRandom())
121 AliPythiaRndm::SetPythiaRandom(GetRandom());
8d2cd130 122
0bd3d7c5 123 fItune = itune;
124
8d2cd130 125 fProcess = process;
126 fEcms = energy;
127 fStrucFunc = strucfunc;
1d5b1b20 128//...Switch off decay of pi0, K0S, Lambda, Sigma+-, Xi0-, Omega-.
e2de0ce1 129 SetMDCY(Pycomp(111) ,1,0); // pi0
130 SetMDCY(Pycomp(310) ,1,0); // K0S
131 SetMDCY(Pycomp(3122),1,0); // kLambda
132 SetMDCY(Pycomp(3112),1,0); // sigma -
e2de0ce1 133 SetMDCY(Pycomp(3222),1,0); // sigma +
134 SetMDCY(Pycomp(3312),1,0); // xi -
135 SetMDCY(Pycomp(3322),1,0); // xi 0
136 SetMDCY(Pycomp(3334),1,0); // omega-
1c50ec12 137 // Select structure function
8d2cd130 138 SetMSTP(52,2);
e2de0ce1 139 SetMSTP(51, AliStructFuncType::PDFsetIndex(strucfunc));
1c50ec12 140 // Particles produced in string fragmentation point directly to either of the two endpoints
141 // of the string (depending in the side they were generated from).
142 SetMSTU(16,2);
143
8d2cd130 144//
145// Pythia initialisation for selected processes//
146//
147// Make MSEL clean
148//
149 for (Int_t i=1; i<= 200; i++) {
150 SetMSUB(i,0);
151 }
152// select charm production
153 switch (process)
154 {
65f2626c 155 case kPyOldUEQ2ordered: //Old underlying events with Q2 ordered QCD processes
156// Multiple interactions on.
157 SetMSTP(81,1);
158// Double Gaussian matter distribution.
159 SetMSTP(82,4);
160 SetPARP(83,0.5);
161 SetPARP(84,0.4);
162// pT0.
163 SetPARP(82,2.0);
164// Reference energy for pT0 and energy rescaling pace.
165 SetPARP(89,1800);
166 SetPARP(90,0.25);
167// String drawing almost completely minimizes string length.
168 SetPARP(85,0.9);
169 SetPARP(86,0.95);
170// ISR and FSR activity.
171 SetPARP(67,4);
172 SetPARP(71,4);
173// Lambda_FSR scale.
174 SetPARJ(81,0.29);
175 break;
176 case kPyOldUEQ2ordered2:
177// Old underlying events with Q2 ordered QCD processes
178// Multiple interactions on.
179 SetMSTP(81,1);
180// Double Gaussian matter distribution.
181 SetMSTP(82,4);
182 SetPARP(83,0.5);
183 SetPARP(84,0.4);
184// pT0.
185 SetPARP(82,2.0);
186// Reference energy for pT0 and energy rescaling pace.
187 SetPARP(89,1800);
188 SetPARP(90,0.16); // here is the difference with kPyOldUEQ2ordered
189// String drawing almost completely minimizes string length.
190 SetPARP(85,0.9);
191 SetPARP(86,0.95);
192// ISR and FSR activity.
193 SetPARP(67,4);
194 SetPARP(71,4);
195// Lambda_FSR scale.
196 SetPARJ(81,0.29);
197 break;
198 case kPyOldPopcorn:
199// Old production mechanism: Old Popcorn
200 SetMSEL(1);
201 SetMSTJ(12,3);
202// (D=2) Like MSTJ(12)=2 but added prod ofthe 1er rank baryon
203 SetMSTP(88,2);
204// (D=1)see can be used to form baryons (BARYON JUNCTION)
205 SetMSTJ(1,1);
e0e89f40 206 AtlasTuning();
65f2626c 207 break;
8d2cd130 208 case kPyCharm:
209 SetMSEL(4);
8d2cd130 210// heavy quark masses
211
212 SetPMAS(4,1,1.2);
8d2cd130 213//
214// primordial pT
215 SetMSTP(91,1);
216 SetPARP(91,1.);
217 SetPARP(93,5.);
218//
219 break;
220 case kPyBeauty:
221 SetMSEL(5);
222 SetPMAS(5,1,4.75);
8d2cd130 223 break;
224 case kPyJpsi:
225 SetMSEL(0);
226// gg->J/Psi g
227 SetMSUB(86,1);
228 break;
229 case kPyJpsiChi:
230 SetMSEL(0);
231// gg->J/Psi g
232 SetMSUB(86,1);
233// gg-> chi_0c g
234 SetMSUB(87,1);
235// gg-> chi_1c g
236 SetMSUB(88,1);
237// gg-> chi_2c g
238 SetMSUB(89,1);
239 break;
240 case kPyCharmUnforced:
241 SetMSEL(0);
242// gq->qg
243 SetMSUB(28,1);
244// gg->qq
245 SetMSUB(53,1);
246// gg->gg
247 SetMSUB(68,1);
248 break;
249 case kPyBeautyUnforced:
250 SetMSEL(0);
251// gq->qg
252 SetMSUB(28,1);
253// gg->qq
254 SetMSUB(53,1);
255// gg->gg
256 SetMSUB(68,1);
257 break;
258 case kPyMb:
259// Minimum Bias pp-Collisions
260//
261//
262// select Pythia min. bias model
263 SetMSEL(0);
511db649 264 SetMSUB(92,1); // single diffraction AB-->XB
265 SetMSUB(93,1); // single diffraction AB-->AX
266 SetMSUB(94,1); // double diffraction
267 SetMSUB(95,1); // low pt production
268
e0e89f40 269 AtlasTuning();
511db649 270 break;
0bd3d7c5 271
272 case kPyMbAtlasTuneMC09:
273// Minimum Bias pp-Collisions
274//
275//
276// select Pythia min. bias model
277 SetMSEL(0);
278 SetMSUB(92,1); // single diffraction AB-->XB
279 SetMSUB(93,1); // single diffraction AB-->AX
280 SetMSUB(94,1); // double diffraction
281 SetMSUB(95,1); // low pt production
282
b4d68b2a 283 AtlasTuningMC09();
0bd3d7c5 284 break;
04081a91 285
286 case kPyMbWithDirectPhoton:
287// Minimum Bias pp-Collisions with direct photon processes added
288//
289//
290// select Pythia min. bias model
291 SetMSEL(0);
292 SetMSUB(92,1); // single diffraction AB-->XB
293 SetMSUB(93,1); // single diffraction AB-->AX
294 SetMSUB(94,1); // double diffraction
295 SetMSUB(95,1); // low pt production
296
297 SetMSUB(14,1); //
298 SetMSUB(18,1); //
299 SetMSUB(29,1); //
300 SetMSUB(114,1); //
301 SetMSUB(115,1); //
302
303
304 AtlasTuning();
305 break;
306
f529e69b 307 case kPyMbDefault:
308// Minimum Bias pp-Collisions
309//
310//
311// select Pythia min. bias model
312 SetMSEL(0);
6d2ec66d 313 SetMSUB(92,1); // single diffraction AB-->XB
314 SetMSUB(93,1); // single diffraction AB-->AX
f529e69b 315 SetMSUB(94,1); // double diffraction
316 SetMSUB(95,1); // low pt production
f529e69b 317 break;
0a0cbcfd 318 case kPyLhwgMb:
319// Les Houches Working Group 05 Minimum Bias pp-Collisions: hep-ph/0604120
320// -> Pythia 6.3 or above is needed
321//
322 SetMSEL(0);
323 SetMSUB(92,1); // single diffraction AB-->XB
324 SetMSUB(93,1); // single diffraction AB-->AX
325 SetMSUB(94,1); // double diffraction
326 SetMSUB(95,1); // low pt production
327
e2de0ce1 328 SetMSTP(51,AliStructFuncType::PDFsetIndex(kCTEQ6ll)); // CTEQ6ll pdf
0a0cbcfd 329 SetMSTP(52,2);
330 SetMSTP(68,1);
331 SetMSTP(70,2);
332 SetMSTP(81,1); // Multiple Interactions ON
333 SetMSTP(82,4); // Double Gaussian Model
334 SetMSTP(88,1);
335
336 SetPARP(82,2.3); // [GeV] PT_min at Ref. energy
337 SetPARP(83,0.5); // Core density in proton matter distribution (def.value)
338 SetPARP(84,0.5); // Core radius
339 SetPARP(85,0.9); // Regulates gluon prod. mechanism
340 SetPARP(90,0.2); // 2*epsilon (exponent in power law)
341
342 break;
8d2cd130 343 case kPyMbNonDiffr:
344// Minimum Bias pp-Collisions
345//
346//
347// select Pythia min. bias model
348 SetMSEL(0);
511db649 349 SetMSUB(95,1); // low pt production
0f482ae4 350
e0e89f40 351 AtlasTuning();
8d2cd130 352 break;
d7de4a67 353 case kPyMbMSEL1:
354 ConfigHeavyFlavor();
355// Intrinsic <kT^2>
356 SetMSTP(91,1);// Width (1=gaussian) primordial kT dist. inside hadrons
357 SetPARP(91,1.); // <kT^2> = PARP(91,1.)^2
358 SetPARP(93,5.); // Upper cut-off
359// Set Q-quark mass
360 SetPMAS(4,1,1.2); // Charm quark mass
361 SetPMAS(5,1,4.78); // Beauty quark mass
362 SetPARP(71,4.); // Defaut value
363// Atlas Tuning
364 AtlasTuning();
365 break;
8d2cd130 366 case kPyJets:
367//
368// QCD Jets
369//
370 SetMSEL(1);
65f2626c 371 // Pythia Tune A (CDF)
372 //
4167b79f 373 SetPARP(67,2.5); // Regulates Initial State Radiation (value from best fit to D0 dijet analysis)
65f2626c 374 SetMSTP(82,4); // Double Gaussian Model
375 SetPARP(82,2.0); // [GeV] PT_min at Ref. energy
376 SetPARP(84,0.4); // Core radius
377 SetPARP(85,0.90) ; // Regulates gluon prod. mechanism
378 SetPARP(86,0.95); // Regulates gluon prod. mechanism
379 SetPARP(89,1800.); // [GeV] Ref. energy
0bd3d7c5 380 SetPARP(90,0.25); // 2*epsilon (exponent in power law)
65f2626c 381 break;
8d2cd130 382 case kPyDirectGamma:
383 SetMSEL(10);
384 break;
adf4d898 385 case kPyCharmPbPbMNR:
386 case kPyD0PbPbMNR:
90d7b703 387 case kPyDPlusPbPbMNR:
e0e89f40 388 case kPyDPlusStrangePbPbMNR:
90d7b703 389 // Tuning of Pythia parameters aimed to get a resonable agreement
390 // between with the NLO calculation by Mangano, Nason, Ridolfi for the
391 // c-cbar single inclusive and double differential distributions.
392 // This parameter settings are meant to work with Pb-Pb collisions
393 // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
394 // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
395 // has to be set to 2.1GeV. Example in ConfigCharmPPR.C.
3dc3ec94 396 ConfigHeavyFlavor();
90d7b703 397 // Intrinsic <kT>
398 SetMSTP(91,1);
399 SetPARP(91,1.304);
400 SetPARP(93,6.52);
90d7b703 401 // Set c-quark mass
402 SetPMAS(4,1,1.2);
90d7b703 403 break;
adf4d898 404 case kPyCharmpPbMNR:
405 case kPyD0pPbMNR:
90d7b703 406 case kPyDPluspPbMNR:
e0e89f40 407 case kPyDPlusStrangepPbMNR:
90d7b703 408 // Tuning of Pythia parameters aimed to get a resonable agreement
409 // between with the NLO calculation by Mangano, Nason, Ridolfi for the
410 // c-cbar single inclusive and double differential distributions.
411 // This parameter settings are meant to work with p-Pb collisions
412 // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
413 // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
414 // has to be set to 2.1GeV. Example in ConfigCharmPPR.C.
3dc3ec94 415 ConfigHeavyFlavor();
90d7b703 416 // Intrinsic <kT>
3dc3ec94 417 SetMSTP(91,1);
418 SetPARP(91,1.16);
419 SetPARP(93,5.8);
420
90d7b703 421 // Set c-quark mass
3dc3ec94 422 SetPMAS(4,1,1.2);
90d7b703 423 break;
adf4d898 424 case kPyCharmppMNR:
425 case kPyD0ppMNR:
90d7b703 426 case kPyDPlusppMNR:
e0e89f40 427 case kPyDPlusStrangeppMNR:
68504d42 428 case kPyLambdacppMNR:
90d7b703 429 // Tuning of Pythia parameters aimed to get a resonable agreement
430 // between with the NLO calculation by Mangano, Nason, Ridolfi for the
431 // c-cbar single inclusive and double differential distributions.
432 // This parameter settings are meant to work with pp collisions
433 // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
434 // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
435 // has to be set to 2.1GeV. Example in ConfigCharmPPR.C.
3dc3ec94 436 ConfigHeavyFlavor();
90d7b703 437 // Intrinsic <kT^2>
3dc3ec94 438 SetMSTP(91,1);
439 SetPARP(91,1.);
440 SetPARP(93,5.);
441
90d7b703 442 // Set c-quark mass
3dc3ec94 443 SetPMAS(4,1,1.2);
90d7b703 444 break;
e0e89f40 445 case kPyCharmppMNRwmi:
446 // Tuning of Pythia parameters aimed to get a resonable agreement
447 // between with the NLO calculation by Mangano, Nason, Ridolfi for the
448 // c-cbar single inclusive and double differential distributions.
449 // This parameter settings are meant to work with pp collisions
450 // and with kCTEQ5L PDFs.
451 // Added multiple interactions according to ATLAS tune settings.
452 // To get a "reasonable" agreement with MNR results, events have to be
453 // generated with the minimum ptHard (AliGenPythia::SetPtHard)
454 // set to 2.76 GeV.
455 // To get a "perfect" agreement with MNR results, events have to be
456 // generated in four ptHard bins with the following relative
457 // normalizations:
458 // 2.76-3 GeV: 25%
459 // 3-4 GeV: 40%
460 // 4-8 GeV: 29%
461 // >8 GeV: 6%
462 ConfigHeavyFlavor();
463 // Intrinsic <kT^2>
464 SetMSTP(91,1);
465 SetPARP(91,1.);
466 SetPARP(93,5.);
467
468 // Set c-quark mass
469 SetPMAS(4,1,1.2);
470 AtlasTuning();
471 break;
adf4d898 472 case kPyBeautyPbPbMNR:
8d2cd130 473 // Tuning of Pythia parameters aimed to get a resonable agreement
474 // between with the NLO calculation by Mangano, Nason, Ridolfi for the
475 // b-bbar single inclusive and double differential distributions.
476 // This parameter settings are meant to work with Pb-Pb collisions
477 // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
478 // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
479 // has to be set to 2.75GeV. Example in ConfigBeautyPPR.C.
3dc3ec94 480 ConfigHeavyFlavor();
8d2cd130 481 // QCD scales
3dc3ec94 482 SetPARP(67,1.0);
483 SetPARP(71,1.0);
adf4d898 484 // Intrinsic <kT>
3dc3ec94 485 SetMSTP(91,1);
486 SetPARP(91,2.035);
487 SetPARP(93,10.17);
8d2cd130 488 // Set b-quark mass
3dc3ec94 489 SetPMAS(5,1,4.75);
8d2cd130 490 break;
adf4d898 491 case kPyBeautypPbMNR:
492 // Tuning of Pythia parameters aimed to get a resonable agreement
493 // between with the NLO calculation by Mangano, Nason, Ridolfi for the
494 // b-bbar single inclusive and double differential distributions.
495 // This parameter settings are meant to work with p-Pb collisions
496 // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
497 // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
498 // has to be set to 2.75GeV. Example in ConfigBeautyPPR.C.
3dc3ec94 499 ConfigHeavyFlavor();
adf4d898 500 // QCD scales
3dc3ec94 501 SetPARP(67,1.0);
502 SetPARP(71,1.0);
adf4d898 503 // Intrinsic <kT>
3dc3ec94 504 SetMSTP(91,1);
505 SetPARP(91,1.60);
506 SetPARP(93,8.00);
adf4d898 507 // Set b-quark mass
3dc3ec94 508 SetPMAS(5,1,4.75);
adf4d898 509 break;
510 case kPyBeautyppMNR:
511 // Tuning of Pythia parameters aimed to get a resonable agreement
512 // between with the NLO calculation by Mangano, Nason, Ridolfi for the
513 // b-bbar single inclusive and double differential distributions.
514 // This parameter settings are meant to work with pp collisions
515 // (AliGenPythia::SetNuclei) and with kCTEQ4L PDFs.
516 // To get a good agreement the minimum ptHard (AliGenPythia::SetPtHard)
517 // has to be set to 2.75GeV. Example in ConfigBeautyPPR.C.
3dc3ec94 518 ConfigHeavyFlavor();
adf4d898 519 // QCD scales
3dc3ec94 520 SetPARP(67,1.0);
521 SetPARP(71,1.0);
522
523 // Intrinsic <kT>
524 SetMSTP(91,1);
525 SetPARP(91,1.);
526 SetPARP(93,5.);
527
528 // Set b-quark mass
529 SetPMAS(5,1,4.75);
adf4d898 530 break;
9dfe63b3 531 case kPyBeautyJets:
e0e89f40 532 case kPyBeautyppMNRwmi:
533 // Tuning of Pythia parameters aimed to get a resonable agreement
534 // between with the NLO calculation by Mangano, Nason, Ridolfi for the
535 // b-bbar single inclusive and double differential distributions.
536 // This parameter settings are meant to work with pp collisions
537 // and with kCTEQ5L PDFs.
538 // Added multiple interactions according to ATLAS tune settings.
539 // To get a "reasonable" agreement with MNR results, events have to be
540 // generated with the minimum ptHard (AliGenPythia::SetPtHard)
541 // set to 2.76 GeV.
542 // To get a "perfect" agreement with MNR results, events have to be
543 // generated in four ptHard bins with the following relative
544 // normalizations:
545 // 2.76-4 GeV: 5%
546 // 4-6 GeV: 31%
547 // 6-8 GeV: 28%
548 // >8 GeV: 36%
549 ConfigHeavyFlavor();
550 // QCD scales
551 SetPARP(67,1.0);
552 SetPARP(71,1.0);
553
554 // Intrinsic <kT>
555 SetMSTP(91,1);
556 SetPARP(91,1.);
557 SetPARP(93,5.);
558
559 // Set b-quark mass
560 SetPMAS(5,1,4.75);
561
562 AtlasTuning();
563 break;
589380c6 564 case kPyW:
565
566 //Inclusive production of W+/-
567 SetMSEL(0);
568 //f fbar -> W+
569 SetMSUB(2,1);
570 // //f fbar -> g W+
571 // SetMSUB(16,1);
572 // //f fbar -> gamma W+
573 // SetMSUB(20,1);
574 // //f g -> f W+
575 // SetMSUB(31,1);
576 // //f gamma -> f W+
577 // SetMSUB(36,1);
578
579 // Initial/final parton shower on (Pythia default)
580 // With parton showers on we are generating "W inclusive process"
581 SetMSTP(61,1); //Initial QCD & QED showers on
582 SetMSTP(71,1); //Final QCD & QED showers on
583
584 break;
0f6ee828 585
586 case kPyZ:
587
588 //Inclusive production of Z
589 SetMSEL(0);
590 //f fbar -> Z/gamma
591 SetMSUB(1,1);
592
593 // // f fbar -> g Z/gamma
594 // SetMSUB(15,1);
595 // // f fbar -> gamma Z/gamma
596 // SetMSUB(19,1);
597 // // f g -> f Z/gamma
598 // SetMSUB(30,1);
599 // // f gamma -> f Z/gamma
600 // SetMSUB(35,1);
601
602 //only Z included, not gamma
603 SetMSTP(43,2);
604
605 // Initial/final parton shower on (Pythia default)
606 // With parton showers on we are generating "Z inclusive process"
607 SetMSTP(61,1); //Initial QCD & QED showers on
608 SetMSTP(71,1); //Final QCD & QED showers on
609
610 break;
9a8774a1 611 case kPyMBRSingleDiffraction:
612 case kPyMBRDoubleDiffraction:
613 case kPyMBRCentralDiffraction:
614 break;
8d2cd130 615 }
616//
617// Initialize PYTHIA
efe3b1cd 618//
619// Select the tune
620 if (itune > -1) Pytune(itune);
621
622//
8d2cd130 623 SetMSTP(41,1); // all resonance decays switched on
d95b55fa 624 Initialize("CMS","p","p",fEcms);
03358a32 625 fOmegaDalitz.Init();
8d2cd130 626}
627
628Int_t AliPythia::CheckedLuComp(Int_t kf)
629{
630// Check Lund particle code (for debugging)
631 Int_t kc=Pycomp(kf);
632 printf("\n Lucomp kf,kc %d %d",kf,kc);
633 return kc;
634}
635
20e47f08 636void AliPythia::SetNuclei(Int_t a1, Int_t a2, Int_t pdf)
8d2cd130 637{
638// Treat protons as inside nuclei with mass numbers a1 and a2
639// The MSTP array in the PYPARS common block is used to enable and
640// select the nuclear structure functions.
641// MSTP(52) : (D=1) choice of proton and nuclear structure-function library
642// =1: internal PYTHIA acording to MSTP(51)
643// =2: PDFLIB proton s.f., with MSTP(51) = 1000xNGROUP+NSET
644// If the following mass number both not equal zero, nuclear corrections of the stf are used.
645// MSTP(192) : Mass number of nucleus side 1
646// MSTP(193) : Mass number of nucleus side 2
66f02a7f 647// MSTP(194) : Nuclear structure function: 0: EKS98 8:EPS08 9:EPS09LO 19:EPS09NLO
8d2cd130 648 SetMSTP(52,2);
649 SetMSTP(192, a1);
20e47f08 650 SetMSTP(193, a2);
651 SetMSTP(194, pdf);
8d2cd130 652}
653
654
655AliPythia* AliPythia::Instance()
656{
657// Set random number generator
658 if (fgAliPythia) {
659 return fgAliPythia;
660 } else {
661 fgAliPythia = new AliPythia();
662 return fgAliPythia;
663 }
664}
665
666void AliPythia::PrintParticles()
667{
668// Print list of particl properties
669 Int_t np = 0;
c31f1d37 670 char* name = new char[16];
8d2cd130 671 for (Int_t kf=0; kf<1000000; kf++) {
672 for (Int_t c = 1; c > -2; c-=2) {
8d2cd130 673 Int_t kc = Pycomp(c*kf);
674 if (kc) {
675 Float_t mass = GetPMAS(kc,1);
676 Float_t width = GetPMAS(kc,2);
677 Float_t tau = GetPMAS(kc,4);
c31f1d37 678
8d2cd130 679 Pyname(kf,name);
680
681 np++;
682
683 printf("\n mass, width, tau: %6d %s %10.3f %10.3e %10.3e",
684 c*kf, name, mass, width, tau);
685 }
686 }
687 }
688 printf("\n Number of particles %d \n \n", np);
689}
690
691void AliPythia::ResetDecayTable()
692{
693// Set default values for pythia decay switches
694 Int_t i;
695 for (i = 1; i < 501; i++) SetMDCY(i,1,fDefMDCY[i]);
696 for (i = 1; i < 2001; i++) SetMDME(i,1,fDefMDME[i]);
697}
698
699void AliPythia::SetDecayTable()
700{
701// Set default values for pythia decay switches
702//
703 Int_t i;
704 for (i = 1; i < 501; i++) fDefMDCY[i] = GetMDCY(i,1);
705 for (i = 1; i < 2001; i++) fDefMDME[i] = GetMDME(i,1);
706}
707
708void AliPythia::Pyclus(Int_t& njet)
709{
710// Call Pythia clustering algorithm
711//
712 pyclus(njet);
713}
714
715void AliPythia::Pycell(Int_t& njet)
716{
717// Call Pythia jet reconstruction algorithm
718//
719 pycell(njet);
720}
721
452af8c7 722void AliPythia::Pyshow(Int_t ip1, Int_t ip2, Double_t qmax)
723{
724// Call Pythia jet reconstruction algorithm
725//
452af8c7 726 pyshow(ip1, ip2, qmax);
727}
728
729void AliPythia::Pyrobo(Int_t imi, Int_t ima, Double_t the, Double_t phi, Double_t bex, Double_t bey, Double_t bez)
730{
731 pyrobo(imi, ima, the, phi, bex, bey, bez);
732}
733
694b39f9 734void AliPythia::Pytune(Int_t itune)
735{
c5e2801a 736/*
737C
738C ITUNE NAME (detailed descriptions below)
739C 0 Default : No settings changed => linked Pythia version's defaults.
740C ====== Old UE, Q2-ordered showers ==========================================
741C 100 A : Rick Field's CDF Tune A
742C 101 AW : Rick Field's CDF Tune AW
743C 102 BW : Rick Field's CDF Tune BW
744C 103 DW : Rick Field's CDF Tune DW
745C 104 DWT : Rick Field's CDF Tune DW with slower UE energy scaling
746C 105 QW : Rick Field's CDF Tune QW (NB: needs CTEQ6.1M pdfs externally)
747C 106 ATLAS-DC2: Arthur Moraes' (old) ATLAS tune (ATLAS DC2 / Rome)
748C 107 ACR : Tune A modified with annealing CR
749C 108 D6 : Rick Field's CDF Tune D6 (NB: needs CTEQ6L pdfs externally)
750C 109 D6T : Rick Field's CDF Tune D6T (NB: needs CTEQ6L pdfs externally)
751C ====== Intermediate Models =================================================
752C 200 IM 1 : Intermediate model: new UE, Q2-ordered showers, annealing CR
753C 201 APT : Tune A modified to use pT-ordered final-state showers
754C ====== New UE, interleaved pT-ordered showers, annealing CR ================
755C 300 S0 : Sandhoff-Skands Tune 0
756C 301 S1 : Sandhoff-Skands Tune 1
757C 302 S2 : Sandhoff-Skands Tune 2
758C 303 S0A : S0 with "Tune A" UE energy scaling
759C 304 NOCR : New UE "best try" without colour reconnections
760C 305 Old : New UE, original (primitive) colour reconnections
761C 306 ATLAS-CSC: Arthur Moraes' (new) ATLAS tune (needs CTEQ6L externally)
762C ======= The Uppsala models =================================================
763C ( NB! must be run with special modified Pythia 6.215 version )
764C ( available from http://www.isv.uu.se/thep/MC/scigal/ )
765C 400 GAL 0 : Generalized area-law model. Old parameters
766C 401 SCI 0 : Soft-Colour-Interaction model. Old parameters
767C 402 GAL 1 : Generalized area-law model. Tevatron MB retuned (Skands)
768*/
694b39f9 769 pytune(itune);
770}
771
9b61ba2a 772void AliPythia::Py2ent(Int_t idx, Int_t pdg1, Int_t pdg2, Double_t p){
773 // Inset 2-parton system at line idx
774 py2ent(idx, pdg1, pdg2, p);
775}
452af8c7 776
777
32c8e463 778void AliPythia::InitQuenching(Float_t cMin, Float_t cMax, Float_t k, Int_t iECMethod, Float_t zmax, Int_t ngmax)
0f482ae4 779{
780// Initializes
781// (1) The quenching model using quenching weights according to C. Salgado and U. Wiedemann
782// (2) The nuclear geometry using the Glauber Model
783//
6b435cde 784
18b7a4a1 785 fGlauber = AliFastGlauber::Instance();
0f482ae4 786 fGlauber->Init(2);
787 fGlauber->SetCentralityClass(cMin, cMax);
788
789 fQuenchingWeights = new AliQuenchingWeights();
790 fQuenchingWeights->InitMult();
86b6ad68 791 fQuenchingWeights->SetK(k);
0f482ae4 792 fQuenchingWeights->SetECMethod(AliQuenchingWeights::kECMethod(iECMethod));
32c8e463 793 fNGmax = ngmax;
794 fZmax = zmax;
795
0f482ae4 796}
797
798
452af8c7 799void AliPythia::Quench()
800{
801//
802//
803// Simple Jet Quenching routine:
804// =============================
805// The jet formed by all final state partons radiated by the parton created
0f482ae4 806// in the hard collisions is quenched by a factor (1-z) using light cone variables in
807// the initial parton reference frame:
452af8c7 808// (E + p_z)new = (1-z) (E + p_z)old
809//
0f482ae4 810//
811//
812//
452af8c7 813// The lost momentum is first balanced by one gluon with virtuality > 0.
814// Subsequently the gluon splits to yield two gluons with E = p.
815//
0f482ae4 816//
817//
4e383037 818 static Float_t eMean = 0.;
819 static Int_t icall = 0;
0f482ae4 820
c2c598a3 821 Double_t p0[4][5];
822 Double_t p1[4][5];
823 Double_t p2[4][5];
824 Int_t klast[4] = {-1, -1, -1, -1};
452af8c7 825
826 Int_t numpart = fPyjets->N;
86b6ad68 827 Double_t px = 0., py = 0., pz = 0., e = 0., m = 0., p = 0., pt = 0., theta = 0., phi = 0.;
c2c598a3 828 Double_t pxq[4], pyq[4], pzq[4], eq[4], yq[4], mq[4], pq[4], phiq[4], thetaq[4], ptq[4];
829 Bool_t quenched[4];
a23e397a 830 Double_t wjtKick[4] = {0., 0., 0., 0.};
c2c598a3 831 Int_t nGluon[4];
86b6ad68 832 Int_t qPdg[4];
0f482ae4 833 Int_t imo, kst, pdg;
b280c4cc 834
511db649 835//
c2c598a3 836// Sore information about Primary partons
837//
838// j =
839// 0, 1 partons from hard scattering
840// 2, 3 partons from initial state radiation
841//
842 for (Int_t i = 2; i <= 7; i++) {
843 Int_t j = 0;
844 // Skip gluons that participate in hard scattering
845 if (i == 4 || i == 5) continue;
846 // Gluons from hard Scattering
847 if (i == 6 || i == 7) {
848 j = i - 6;
849 pxq[j] = fPyjets->P[0][i];
850 pyq[j] = fPyjets->P[1][i];
851 pzq[j] = fPyjets->P[2][i];
852 eq[j] = fPyjets->P[3][i];
853 mq[j] = fPyjets->P[4][i];
854 } else {
855 // Gluons from initial state radiation
856 //
857 // Obtain 4-momentum vector from difference between original parton and parton after gluon
858 // radiation. Energy is calculated independently because initial state radition does not
859 // conserve strictly momentum and energy for each partonic system independently.
860 //
861 // Not very clean. Should be improved !
862 //
863 //
864 j = i;
865 pxq[j] = fPyjets->P[0][i] - fPyjets->P[0][i+2];
866 pyq[j] = fPyjets->P[1][i] - fPyjets->P[1][i+2];
867 pzq[j] = fPyjets->P[2][i] - fPyjets->P[2][i+2];
868 mq[j] = fPyjets->P[4][i];
869 eq[j] = TMath::Sqrt(pxq[j] * pxq[j] + pyq[j] * pyq[j] + pzq[j] * pzq[j] + mq[j] * mq[j]);
870 }
871//
872// Calculate some kinematic variables
511db649 873//
4e383037 874 yq[j] = 0.5 * TMath::Log((eq[j] + pzq[j] + 1.e-14) / (eq[j] - pzq[j] + 1.e-14));
0f482ae4 875 pq[j] = TMath::Sqrt(pxq[j] * pxq[j] + pyq[j] * pyq[j] + pzq[j] * pzq[j]);
876 phiq[j] = TMath::Pi()+TMath::ATan2(-pyq[j], -pxq[j]);
877 ptq[j] = TMath::Sqrt(pxq[j] * pxq[j] + pyq[j] * pyq[j]);
878 thetaq[j] = TMath::ATan2(ptq[j], pzq[j]);
86b6ad68 879 qPdg[j] = fPyjets->K[1][i];
880 }
881
882 Double_t int0[4];
883 Double_t int1[4];
86b6ad68 884
b280c4cc 885 fGlauber->GetI0I1ForPythiaAndXY(4, phiq, int0, int1, fXJet, fYJet, 15.);
886
86b6ad68 887 for (Int_t j = 0; j < 4; j++) {
c2c598a3 888 //
889 // Quench only central jets and with E > 10.
890 //
86b6ad68 891
892
893 Int_t itype = (qPdg[j] == 21) ? 2 : 1;
894 Double_t eloss = fQuenchingWeights->GetELossRandomKFast(itype, int0[j], int1[j], eq[j]);
895
c2c598a3 896 if (TMath::Abs(yq[j]) > 2.5 || eq[j] < 10.) {
b280c4cc 897 fZQuench[j] = 0.;
0f482ae4 898 } else {
c2c598a3 899 if (eq[j] > 40. && TMath::Abs(yq[j]) < 0.5) {
4e383037 900 icall ++;
901 eMean += eloss;
902 }
0f482ae4 903 //
904 // Extra pt
86b6ad68 905 Double_t l = fQuenchingWeights->CalcLk(int0[j], int1[j]);
906 wjtKick[j] = TMath::Sqrt(l * fQuenchingWeights->CalcQk(int0[j], int1[j]));
0f482ae4 907 //
908 // Fractional energy loss
b280c4cc 909 fZQuench[j] = eloss / eq[j];
0f482ae4 910 //
911 // Avoid complete loss
912 //
1044c4d8 913 if (fZQuench[j] > fZmax) fZQuench[j] = fZmax;
0f482ae4 914 //
915 // Some debug printing
86b6ad68 916
917
bf9bb016 918// printf("Initial parton # %3d, Type %3d Energy %10.3f Phi %10.3f Length %10.3f Loss %10.3f Kick %10.3f Mean: %10.3f %10.3f\n",
919// j, itype, eq[j], phiq[j], l, eloss, wjtKick[j], eMean / Float_t(icall+1), yq[j]);
4e383037 920
b280c4cc 921// fZQuench[j] = 0.8;
922// while (fZQuench[j] >= 0.95) fZQuench[j] = gRandom->Exp(0.2);
0f482ae4 923 }
4e383037 924
b280c4cc 925 quenched[j] = (fZQuench[j] > 0.01);
4e383037 926 } // primary partons
c2c598a3 927
b280c4cc 928
929
6e90ad26 930 Double_t pNew[1000][4];
931 Int_t kNew[1000];
932 Int_t icount = 0;
b280c4cc 933 Double_t zquench[4];
934
6e90ad26 935//
4e383037 936// System Loop
c2c598a3 937 for (Int_t isys = 0; isys < 4; isys++) {
6e90ad26 938// Skip to next system if not quenched.
4e383037 939 if (!quenched[isys]) continue;
940
b280c4cc 941 nGluon[isys] = 1 + Int_t(fZQuench[isys] / (1. - fZQuench[isys]));
32c8e463 942 if (nGluon[isys] > fNGmax) nGluon[isys] = fNGmax;
b280c4cc 943 zquench[isys] = 1. - TMath::Power(1. - fZQuench[isys], 1./Double_t(nGluon[isys]));
4e383037 944 wjtKick[isys] = wjtKick[isys] / TMath::Sqrt(Double_t(nGluon[isys]));
0f482ae4 945
4e383037 946
947
948 Int_t igMin = -1;
949 Int_t igMax = -1;
950 Double_t pg[4] = {0., 0., 0., 0.};
951
952//
953// Loop on radiation events
954
955 for (Int_t iglu = 0; iglu < nGluon[isys]; iglu++) {
6e90ad26 956 while (1) {
957 icount = 0;
958 for (Int_t k = 0; k < 4; k++)
959 {
960 p0[isys][k] = 0.;
961 p1[isys][k] = 0.;
962 p2[isys][k] = 0.;
963 }
964// Loop over partons
965 for (Int_t i = 0; i < numpart; i++)
966 {
967 imo = fPyjets->K[2][i];
968 kst = fPyjets->K[0][i];
969 pdg = fPyjets->K[1][i];
970
971
972
0f482ae4 973// Quarks and gluons only
6e90ad26 974 if (pdg != 21 && TMath::Abs(pdg) > 6) continue;
0f482ae4 975// Particles from hard scattering only
c2c598a3 976
6e90ad26 977 if (imo > 8 && imo < 1000) imo = fPyjets->K[2][imo - 1];
c2c598a3 978 Int_t imom = imo % 1000;
979 if ((isys == 0 || isys == 1) && ((imom != (isys + 7)))) continue;
980 if ((isys == 2 || isys == 3) && ((imom != (isys + 1)))) continue;
981
6e90ad26 982
0f482ae4 983// Skip comment lines
6e90ad26 984 if (kst != 1 && kst != 2) continue;
0f482ae4 985//
986// Parton kinematic
6e90ad26 987 px = fPyjets->P[0][i];
988 py = fPyjets->P[1][i];
989 pz = fPyjets->P[2][i];
990 e = fPyjets->P[3][i];
991 m = fPyjets->P[4][i];
992 pt = TMath::Sqrt(px * px + py * py);
993 p = TMath::Sqrt(px * px + py * py + pz * pz);
994 phi = TMath::Pi() + TMath::ATan2(-py, -px);
995 theta = TMath::ATan2(pt, pz);
996
0f482ae4 997//
c2c598a3 998// Save 4-momentum sum for balancing
999 Int_t index = isys;
6e90ad26 1000
1001 p0[index][0] += px;
1002 p0[index][1] += py;
1003 p0[index][2] += pz;
1004 p0[index][3] += e;
6e90ad26 1005
1006 klast[index] = i;
1007
0f482ae4 1008//
1009// Fractional energy loss
b280c4cc 1010 Double_t z = zquench[index];
4e383037 1011
c2c598a3 1012
4e383037 1013// Don't fully quench radiated gluons
1014//
1015 if (imo > 1000) {
1016// This small factor makes sure that the gluons are not too close in phase space to avoid recombination
1017//
1018
c2c598a3 1019 z = 0.02;
4e383037 1020 }
c2c598a3 1021// printf("z: %d %f\n", imo, z);
1022
4e383037 1023
1024//
6e90ad26 1025
1026 //
1027 //
1028 // Transform into frame in which initial parton is along z-axis
1029 //
1030 TVector3 v(px, py, pz);
1031 v.RotateZ(-phiq[index]); v.RotateY(-thetaq[index]);
1032 Double_t pxs = v.X(); Double_t pys = v.Y(); Double_t pl = v.Z();
1033
1034 Double_t jt = TMath::Sqrt(pxs * pxs + pys * pys);
1035 Double_t mt2 = jt * jt + m * m;
1036 Double_t zmax = 1.;
1037 //
1038 // Kinematic limit on z
1039 //
4e383037 1040 if (m > 0.) zmax = 1. - m / TMath::Sqrt(m * m + jt * jt);
6e90ad26 1041 //
1042 // Change light-cone kinematics rel. to initial parton
1043 //
1044 Double_t eppzOld = e + pl;
1045 Double_t empzOld = e - pl;
1046
1047 Double_t eppzNew = (1. - z) * eppzOld;
1048 Double_t empzNew = empzOld - mt2 * z / eppzOld;
1049 Double_t eNew = 0.5 * (eppzNew + empzNew);
1050 Double_t plNew = 0.5 * (eppzNew - empzNew);
1051
1052 Double_t jtNew;
1053 //
1054 // if mt very small (or sometimes even < 0 for numerical reasons) set it to 0
1055 Double_t mt2New = eppzNew * empzNew;
1056 if (mt2New < 1.e-8) mt2New = 0.;
4e383037 1057 if (z < zmax) {
1058 if (m * m > mt2New) {
1059 //
1060 // This should not happen
1061 //
1062 Fatal("Quench()", "This should never happen %e %e %e!", m, eppzNew, empzNew);
1063 jtNew = 0;
1064 } else {
1065 jtNew = TMath::Sqrt(mt2New - m * m);
1066 }
6e90ad26 1067 } else {
4e383037 1068 // If pT is to small (probably a leading massive particle) we scale only the energy
1069 // This can cause negative masses of the radiated gluon
1070 // Let's hope for the best ...
1071 jtNew = jt;
1072 eNew = TMath::Sqrt(plNew * plNew + mt2);
1073
6e90ad26 1074 }
6e90ad26 1075 //
1076 // Calculate new px, py
1077 //
b07be423 1078 Double_t pxNew = 0;
1079 Double_t pyNew = 0;
6e90ad26 1080
b07be423 1081 if (jt>0) {
6b118b3c 1082 pxNew = jtNew / jt * pxs;
1083 pyNew = jtNew / jt * pys;
b07be423 1084 }
6e90ad26 1085// Double_t dpx = pxs - pxNew;
1086// Double_t dpy = pys - pyNew;
1087// Double_t dpz = pl - plNew;
1088// Double_t de = e - eNew;
1089// Double_t dmass2 = de * de - dpx * dpx - dpy * dpy - dpz * dpz;
1090// printf("New mass (1) %e %e %e %e %e %e %e \n", dmass2, jt, jtNew, pl, plNew, e, eNew);
1091// printf("New mass (2) %e %e \n", pxNew, pyNew);
1092 //
1093 // Rotate back
1094 //
1095 TVector3 w(pxNew, pyNew, plNew);
1096 w.RotateY(thetaq[index]); w.RotateZ(phiq[index]);
1097 pxNew = w.X(); pyNew = w.Y(); plNew = w.Z();
1098
1099 p1[index][0] += pxNew;
1100 p1[index][1] += pyNew;
1101 p1[index][2] += plNew;
1102 p1[index][3] += eNew;
1103 //
1104 // Updated 4-momentum vectors
1105 //
1106 pNew[icount][0] = pxNew;
1107 pNew[icount][1] = pyNew;
1108 pNew[icount][2] = plNew;
1109 pNew[icount][3] = eNew;
1110 kNew[icount] = i;
1111 icount++;
1112 } // parton loop
0f482ae4 1113 //
6e90ad26 1114 // Check if there was phase-space for quenching
0f482ae4 1115 //
0f482ae4 1116
6e90ad26 1117 if (icount == 0) quenched[isys] = kFALSE;
1118 if (!quenched[isys]) break;
1119
1120 for (Int_t j = 0; j < 4; j++)
1121 {
1122 p2[isys][j] = p0[isys][j] - p1[isys][j];
1123 }
1124 p2[isys][4] = p2[isys][3] * p2[isys][3] - p2[isys][0] * p2[isys][0] - p2[isys][1] * p2[isys][1] - p2[isys][2] * p2[isys][2];
6e90ad26 1125 if (p2[isys][4] > 0.) {
1126 p2[isys][4] = TMath::Sqrt(p2[isys][4]);
1127 break;
1128 } else {
b280c4cc 1129 printf("Warning negative mass squared in system %d %f ! \n", isys, zquench[isys]);
4e383037 1130 printf("4-Momentum: %10.3e %10.3e %10.3e %10.3e %10.3e \n", p2[isys][0], p2[isys][1], p2[isys][2], p2[isys][3], p2[isys][4]);
6e90ad26 1131 if (p2[isys][4] < -0.01) {
4e383037 1132 printf("Negative mass squared !\n");
1133 // Here we have to put the gluon back to mass shell
1134 // This will lead to a small energy imbalance
1135 p2[isys][4] = 0.;
1136 p2[isys][3] = TMath::Sqrt(p2[isys][0] * p2[isys][0] + p2[isys][1] * p2[isys][1] + p2[isys][2] * p2[isys][2]);
1137 break;
6e90ad26 1138 } else {
1139 p2[isys][4] = 0.;
1140 break;
1141 }
1142 }
6e90ad26 1143 /*
6e90ad26 1144 zHeavy *= 0.98;
1145 printf("zHeavy lowered to %f\n", zHeavy);
1146 if (zHeavy < 0.01) {
1147 printf("No success ! \n");
1148 icount = 0;
1149 quenched[isys] = kFALSE;
1150 break;
1151 }
4e383037 1152 */
1153 } // iteration on z (while)
1154
6e90ad26 1155// Update event record
1156 for (Int_t k = 0; k < icount; k++) {
1157// printf("%6d %6d %10.3e %10.3e %10.3e %10.3e\n", k, kNew[k], pNew[k][0],pNew[k][1], pNew[k][2], pNew[k][3] );
1158 fPyjets->P[0][kNew[k]] = pNew[k][0];
1159 fPyjets->P[1][kNew[k]] = pNew[k][1];
1160 fPyjets->P[2][kNew[k]] = pNew[k][2];
1161 fPyjets->P[3][kNew[k]] = pNew[k][3];
0f482ae4 1162 }
4e383037 1163 //
1164 // Add the gluons
1165 //
1166 Int_t ish = 0;
1837e95c 1167 Int_t iGlu;
4e383037 1168 if (!quenched[isys]) continue;
0f482ae4 1169//
1170// Last parton from shower i
4e383037 1171 Int_t in = klast[isys];
0f482ae4 1172//
1173// Continue if no parton in shower i selected
1174 if (in == -1) continue;
1175//
1176// If this is the second initial parton and it is behind the first move pointer by previous ish
4e383037 1177 if (isys == 1 && klast[1] > klast[0]) in += ish;
0f482ae4 1178//
1179// Starting index
452af8c7 1180
4e383037 1181// jmin = in - 1;
0f482ae4 1182// How many additional gluons will be generated
1183 ish = 1;
4e383037 1184 if (p2[isys][4] > 0.05) ish = 2;
0f482ae4 1185//
1186// Position of gluons
4e383037 1187 iGlu = numpart;
1188 if (iglu == 0) igMin = iGlu;
1189 igMax = iGlu;
0f482ae4 1190 numpart += ish;
1191 (fPyjets->N) += ish;
4e383037 1192
0f482ae4 1193 if (ish == 1) {
4e383037 1194 fPyjets->P[0][iGlu] = p2[isys][0];
1195 fPyjets->P[1][iGlu] = p2[isys][1];
1196 fPyjets->P[2][iGlu] = p2[isys][2];
1197 fPyjets->P[3][iGlu] = p2[isys][3];
1198 fPyjets->P[4][iGlu] = p2[isys][4];
0f482ae4 1199
4e383037 1200 fPyjets->K[0][iGlu] = 1;
1201 if (iglu == nGluon[isys] - 1) fPyjets->K[0][iGlu] = 1;
0f482ae4 1202 fPyjets->K[1][iGlu] = 21;
4e383037 1203 fPyjets->K[2][iGlu] = fPyjets->K[2][in] + 1000;
0f482ae4 1204 fPyjets->K[3][iGlu] = -1;
1205 fPyjets->K[4][iGlu] = -1;
4e383037 1206
1207 pg[0] += p2[isys][0];
1208 pg[1] += p2[isys][1];
1209 pg[2] += p2[isys][2];
1210 pg[3] += p2[isys][3];
0f482ae4 1211 } else {
1212 //
1213 // Split gluon in rest frame.
1214 //
4e383037 1215 Double_t bx = p2[isys][0] / p2[isys][3];
1216 Double_t by = p2[isys][1] / p2[isys][3];
1217 Double_t bz = p2[isys][2] / p2[isys][3];
1218 Double_t pst = p2[isys][4] / 2.;
0f482ae4 1219 //
1220 // Isotropic decay ????
1221 Double_t cost = 2. * gRandom->Rndm() - 1.;
60e55aee 1222 Double_t sint = TMath::Sqrt((1.-cost)*(1.+cost));
2ab330c9 1223 Double_t phis = 2. * TMath::Pi() * gRandom->Rndm();
0f482ae4 1224
1225 Double_t pz1 = pst * cost;
1226 Double_t pz2 = -pst * cost;
1227 Double_t pt1 = pst * sint;
1228 Double_t pt2 = -pst * sint;
2ab330c9 1229 Double_t px1 = pt1 * TMath::Cos(phis);
1230 Double_t py1 = pt1 * TMath::Sin(phis);
1231 Double_t px2 = pt2 * TMath::Cos(phis);
1232 Double_t py2 = pt2 * TMath::Sin(phis);
0f482ae4 1233
1234 fPyjets->P[0][iGlu] = px1;
1235 fPyjets->P[1][iGlu] = py1;
1236 fPyjets->P[2][iGlu] = pz1;
1237 fPyjets->P[3][iGlu] = pst;
1238 fPyjets->P[4][iGlu] = 0.;
1239
4e383037 1240 fPyjets->K[0][iGlu] = 1 ;
0f482ae4 1241 fPyjets->K[1][iGlu] = 21;
4e383037 1242 fPyjets->K[2][iGlu] = fPyjets->K[2][in] + 1000;
0f482ae4 1243 fPyjets->K[3][iGlu] = -1;
1244 fPyjets->K[4][iGlu] = -1;
1245
1246 fPyjets->P[0][iGlu+1] = px2;
1247 fPyjets->P[1][iGlu+1] = py2;
1248 fPyjets->P[2][iGlu+1] = pz2;
1249 fPyjets->P[3][iGlu+1] = pst;
1250 fPyjets->P[4][iGlu+1] = 0.;
1251
4e383037 1252 fPyjets->K[0][iGlu+1] = 1;
1253 if (iglu == nGluon[isys] - 1) fPyjets->K[0][iGlu+1] = 1;
0f482ae4 1254 fPyjets->K[1][iGlu+1] = 21;
4e383037 1255 fPyjets->K[2][iGlu+1] = fPyjets->K[2][in] + 1000;
0f482ae4 1256 fPyjets->K[3][iGlu+1] = -1;
1257 fPyjets->K[4][iGlu+1] = -1;
1258 SetMSTU(1,0);
1259 SetMSTU(2,0);
1260 //
1261 // Boost back
1262 //
1263 Pyrobo(iGlu + 1, iGlu + 2, 0., 0., bx, by, bz);
1264 }
4e383037 1265/*
1266 for (Int_t ig = iGlu; ig < iGlu+ish; ig++) {
1267 Double_t px, py, pz;
1268 px = fPyjets->P[0][ig];
1269 py = fPyjets->P[1][ig];
1270 pz = fPyjets->P[2][ig];
1271 TVector3 v(px, py, pz);
1272 v.RotateZ(-phiq[isys]);
1273 v.RotateY(-thetaq[isys]);
1274 Double_t pxs = v.X(); Double_t pys = v.Y(); Double_t pzs = v.Z();
1275 Double_t r = AliPythiaRndm::GetPythiaRandom()->Rndm();
1276 Double_t jtKick = 0.3 * TMath::Sqrt(-TMath::Log(r));
1277 if (ish == 2) jtKick = wjtKick[i] * TMath::Sqrt(-TMath::Log(r)) / TMath::Sqrt(2.);
1278 Double_t phiKick = 2. * TMath::Pi() * AliPythiaRndm::GetPythiaRandom()->Rndm();
1279 pxs += jtKick * TMath::Cos(phiKick);
1280 pys += jtKick * TMath::Sin(phiKick);
1281 TVector3 w(pxs, pys, pzs);
1282 w.RotateY(thetaq[isys]);
1283 w.RotateZ(phiq[isys]);
1284 fPyjets->P[0][ig] = w.X();
1285 fPyjets->P[1][ig] = w.Y();
1286 fPyjets->P[2][ig] = w.Z();
1287 fPyjets->P[2][ig] = w.Mag();
1288 }
1289*/
1290 } // kGluon
1291
6e90ad26 1292
4e383037 1293 // Check energy conservation
0f482ae4 1294 Double_t pxs = 0.;
1295 Double_t pys = 0.;
1296 Double_t pzs = 0.;
1297 Double_t es = 14000.;
1298
1299 for (Int_t i = 0; i < numpart; i++)
1300 {
1301 kst = fPyjets->K[0][i];
1302 if (kst != 1 && kst != 2) continue;
1303 pxs += fPyjets->P[0][i];
1304 pys += fPyjets->P[1][i];
1305 pzs += fPyjets->P[2][i];
1306 es -= fPyjets->P[3][i];
1307 }
1308 if (TMath::Abs(pxs) > 1.e-2 ||
1309 TMath::Abs(pys) > 1.e-2 ||
1310 TMath::Abs(pzs) > 1.e-1) {
1311 printf("%e %e %e %e\n", pxs, pys, pzs, es);
4e383037 1312// Fatal("Quench()", "4-Momentum non-conservation");
452af8c7 1313 }
4e383037 1314
1315 } // end quenching loop (systems)
6e90ad26 1316// Clean-up
0f482ae4 1317 for (Int_t i = 0; i < numpart; i++)
1318 {
4e383037 1319 imo = fPyjets->K[2][i];
1320 if (imo > 1000) {
1321 fPyjets->K[2][i] = fPyjets->K[2][i] % 1000;
1322 }
0f482ae4 1323 }
4e383037 1324// this->Pylist(1);
0f482ae4 1325} // end quench
90d7b703 1326
992f2843 1327
1328void AliPythia::Pyquen(Double_t a, Int_t ibf, Double_t b)
1329{
1330 // Igor Lokthine's quenching routine
12cb0bc0 1331 // http://lokhtin.web.cern.ch/lokhtin/pyquen/pyquen.txt
1332
992f2843 1333 pyquen(a, ibf, b);
1334}
b280c4cc 1335
12cb0bc0 1336void AliPythia::SetPyquenParameters(Double_t t0, Double_t tau0, Int_t nf, Int_t iengl, Int_t iangl)
1337{
1338 // Set the parameters for the PYQUEN package.
1339 // See comments in PyquenCommon.h
1340
1341
1342 PYQPAR.t0 = t0;
1343 PYQPAR.tau0 = tau0;
1344 PYQPAR.nf = nf;
1345 PYQPAR.iengl = iengl;
1346 PYQPAR.iangl = iangl;
1347}
1348
1349
16a82508 1350void AliPythia::Pyevnw()
1351{
1352 // New multiple interaction scenario
1353 pyevnw();
1354}
1355
cd07c39b 1356void AliPythia::Pyshowq(Int_t ip1, Int_t ip2, Double_t qmax)
1357{
1358 // Call medium-modified Pythia jet reconstruction algorithm
1359 //
1360 pyshowq(ip1, ip2, qmax);
1361}
6c43eccb 1362 void AliPythia::Qpygin0()
1363 {
1364 // New multiple interaction scenario
1365 qpygin0();
1366 }
cd07c39b 1367
b280c4cc 1368void AliPythia::GetQuenchingParameters(Double_t& xp, Double_t& yp, Double_t z[4])
1369{
1370 // Return event specific quenching parameters
1371 xp = fXJet;
1372 yp = fYJet;
1373 for (Int_t i = 0; i < 4; i++) z[i] = fZQuench[i];
1374
1375}
1376
3dc3ec94 1377void AliPythia::ConfigHeavyFlavor()
1378{
1379 //
1380 // Default configuration for Heavy Flavor production
1381 //
1382 // All QCD processes
1383 //
1384 SetMSEL(1);
1385
ef185c83 1386
1387 if (fItune < 0) {
1388 // No multiple interactions
1389 SetMSTP(81,0);
1390 SetPARP(81, 0.);
1391 SetPARP(82, 0.);
1392 }
3dc3ec94 1393 // Initial/final parton shower on (Pythia default)
1394 SetMSTP(61,1);
1395 SetMSTP(71,1);
1396
1397 // 2nd order alpha_s
1398 SetMSTP(2,2);
1399
1400 // QCD scales
1401 SetMSTP(32,2);
1402 SetPARP(34,1.0);
1403}
e0e89f40 1404
1405void AliPythia::AtlasTuning()
1406{
1407 //
1408 // Configuration for the ATLAS tuning
0bd3d7c5 1409 if (fItune > -1) return;
1410 printf("ATLAS TUNE \n");
1411
1412 SetMSTP(51, AliStructFuncType::PDFsetIndex(kCTEQ5L)); // CTEQ5L pdf
1413 SetMSTP(81,1); // Multiple Interactions ON
1414 SetMSTP(82,4); // Double Gaussian Model
1415 SetPARP(81,1.9); // Min. pt for multiple interactions (default in 6.2-14)
1416 SetPARP(82,1.8); // [GeV] PT_min at Ref. energy
1417 SetPARP(89,1000.); // [GeV] Ref. energy
1418 SetPARP(90,0.16); // 2*epsilon (exponent in power law)
1419 SetPARP(83,0.5); // Core density in proton matter distribution (def.value)
1420 SetPARP(84,0.5); // Core radius
1421 SetPARP(85,0.33); // Regulates gluon prod. mechanism
1422 SetPARP(86,0.66); // Regulates gluon prod. mechanism
1423 SetPARP(67,1); // Regulates Initial State Radiation
1424}
1425
b4d68b2a 1426void AliPythia::AtlasTuningMC09()
0bd3d7c5 1427{
1428 //
1429 // Configuration for the ATLAS tuning
1430 if (fItune > -1) return;
1431 printf("ATLAS New TUNE MC09\n");
1432 SetMSTP(81,21); // treatment for MI, ISR, FSR and beam remnants: MI on, new model
1433 SetMSTP(82, 4); // Double Gaussian Model
1434 SetMSTP(52, 2); // External PDF
1435 SetMSTP(51, 20650); // MRST LO*
1436
1437
1438 SetMSTP(70, 0); // (was 2: def manual 1, def code 0) virtuality scale for ISR
1439 SetMSTP(72, 1); // (was 0: def 1) maximum scale for FSR
1440 SetMSTP(88, 1); // (was 0: def 1) strategy for qq junction to di-quark or baryon in beam remnant
1441 SetMSTP(90, 0); // (was 1: def 0) strategy of compensate the primordial kT
1442
1443 SetPARP(78, 0.3); // the amount of color reconnection in the final state
1444 SetPARP(80, 0.1); // probability of color partons kicked out from beam remnant
1445 SetPARP(82, 2.3); // [GeV] PT_min at Ref. energy
1446 SetPARP(83, 0.8); // Core density in proton matter distribution (def.value)
1447 SetPARP(84, 0.7); // Core radius
1448 SetPARP(90, 0.25); // 2*epsilon (exponent in power law)
1449 SetPARJ(81, 0.29); // (was 0.14: def 0.29) Labmda value in running alpha_s for parton showers
1450
1451 SetMSTP(95, 6);
1452 SetPARJ(41, 0.3); // a and b parameters of the symmm. Lund FF
1453 SetPARJ(42, 0.58);
1454 SetPARJ(46, 0.75); // mod. of the Lund FF for heavy end-point quarks
1455 SetPARP(89,1800.); // [GeV] Ref. energy
e0e89f40 1456}
e8a8adcd 1457
1458AliPythia& AliPythia::operator=(const AliPythia& rhs)
1459{
1460// Assignment operator
1461 rhs.Copy(*this);
1462 return *this;
1463}
1464
1465 void AliPythia::Copy(TObject&) const
1466{
1467 //
1468 // Copy
1469 //
1470 Fatal("Copy","Not implemented!\n");
1471}
cd07c39b 1472
03358a32 1473void AliPythia::DalitzDecays()
1474{
1475
1476 //
1477 // Replace all omega dalitz decays with the correct matrix element decays
1478 //
1479 Int_t nt = fPyjets->N;
1480 for (Int_t i = 0; i < nt; i++) {
1481 if (fPyjets->K[1][i] != 223) continue;
1482 Int_t fd = fPyjets->K[3][i] - 1;
1483 Int_t ld = fPyjets->K[4][i] - 1;
1484 if (fd < 0) continue;
1485 if ((ld - fd) != 2) continue;
1486 if ((fPyjets->K[1][fd] != 111) ||
ba4e47a0 1487 ((TMath::Abs(fPyjets->K[1][fd+1]) != 11) && (TMath::Abs(fPyjets->K[1][fd+1]) != 13)))
1488 continue;
03358a32 1489 TLorentzVector omega(fPyjets->P[0][i], fPyjets->P[1][i], fPyjets->P[2][i], fPyjets->P[3][i]);
ba4e47a0 1490 Int_t pdg = TMath::Abs(fPyjets->K[1][fd+1]);
1491 fOmegaDalitz.Decay(pdg, &omega);
03358a32 1492 for (Int_t j = 0; j < 3; j++) {
1493 for (Int_t k = 0; k < 4; k++) {
1494 TLorentzVector vec = (fOmegaDalitz.Products())[2-j];
1495 fPyjets->P[k][fd+j] = vec[k];
1496 }
1497 }
1498 }
1499}