Provides collision geometry.
[u/mrichter/AliRoot.git] / EVGEN / AliGenHijing.cxx
CommitLineData
110410f9 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$
96a7d299 18Revision 1.45 2002/12/16 09:44:49 morsch
19Default for fRadiation is 3.
20
83a26417 21Revision 1.44 2002/10/14 14:55:35 hristov
22Merging the VirtualMC branch to the main development branch (HEAD)
23
b9d0a01d 24Revision 1.42.4.1 2002/08/28 15:06:50 alibrary
25Updating to v3-09-01
26
27Revision 1.43 2002/08/09 12:09:52 morsch
28Direct gamma trigger correctly included.
29
7a013373 30Revision 1.42 2002/03/12 11:07:08 morsch
31Add status code of particle to SetTrack call.
32
0d7c70a0 33Revision 1.41 2002/03/05 11:25:33 morsch
34- New quenching options
35- Correction in CheckTrigger()
36
9bf3f17b 37Revision 1.40 2002/02/12 11:05:53 morsch
38Get daughter indices right.
39
f477ff09 40Revision 1.39 2002/02/12 09:16:39 morsch
41Correction in SelectFlavor()
42
cdd538f1 43Revision 1.38 2002/02/12 08:53:21 morsch
44SetNoGammas can be used to inhibit writing of gammas and pi0.
45
0b22c984 46Revision 1.37 2002/02/08 16:50:50 morsch
47Add name and title in constructor.
48
8b31bfac 49Revision 1.36 2002/01/31 20:17:55 morsch
50Allow for triggered jets with simplified topology: Exact pT, back-to-back
51
3792091f 52Revision 1.35 2001/12/13 07:56:25 hristov
53Set pointers to zero in the default constructor
54
60b030a9 55Revision 1.34 2001/12/11 16:55:42 morsch
56Correct initialization for jet phi-range.
57
777b4893 58Revision 1.33 2001/12/05 10:18:51 morsch
59Possibility of kinematic biasing of jet phi range. (J. Klay)
60
7a186781 61Revision 1.32 2001/11/28 13:51:11 morsch
62Introduce kinematic biasing (etamin, etamax) of jet trigger. Bookkeeping
63(number of trials) done in AliGenHijingEventHeader.
64
e4ffda63 65Revision 1.31 2001/11/06 12:30:34 morsch
66Add Boost() method to boost all particles to LHC lab frame. Needed for asymmetric collision systems.
67
cc71f2d3 68Revision 1.30 2001/10/21 18:35:56 hristov
69Several pointers were set to zero in the default constructors to avoid memory management problems
70
2685bf00 71Revision 1.29 2001/10/15 08:12:24 morsch
72- Vertex smearing with truncated gaussian.
73- Store triggered jet info before and after final state radiation into mc-heade
74
f8b85aa9 75Revision 1.28 2001/10/08 11:55:25 morsch
76Store 4-momenta of trigegred jets in event header.
77Possibility to switch of initial and final state radiation.
78
332c8db0 79Revision 1.27 2001/10/08 07:13:14 morsch
80Add setter for minimum transverse momentum of triggered jet.
81
ed6361b0 82Revision 1.26 2001/10/04 08:12:24 morsch
83Redefinition of stable condition.
84
79bf94df 85Revision 1.25 2001/07/27 17:09:36 morsch
86Use local SetTrack, KeepTrack and SetHighWaterMark methods
87to delegate either to local stack or to stack owned by AliRun.
88(Piotr Skowronski, A.M.)
89
a99cf51f 90Revision 1.24 2001/07/20 09:34:56 morsch
91Count the number of spectator neutrons and protons and add information
92to the event header. (Chiara Oppedisano)
93
b46f2676 94Revision 1.23 2001/07/13 17:30:22 morsch
95Derive from AliGenMC.
96
55d6adcf 97Revision 1.22 2001/06/11 13:09:23 morsch
98- Store cross-Section and number of binary collisions as a function of impact parameter
99- Pass AliGenHijingEventHeader to gAlice.
100
21ce2e6f 101Revision 1.21 2001/02/28 17:35:24 morsch
102Consider elastic interactions (ks = 1 and ks = 11) as spectator (Chiara Oppedisano)
103
cf6bac7b 104Revision 1.20 2001/02/14 15:50:40 hristov
105The last particle in event marked using SetHighWaterMark
106
28adac24 107Revision 1.19 2000/12/21 16:24:06 morsch
108Coding convention clean-up
109
675e9664 110Revision 1.18 2000/12/06 17:46:30 morsch
111Avoid random numbers 1 and 0.
112
7498b3ad 113Revision 1.17 2000/12/04 11:22:03 morsch
114Init of sRandom as in 1.15
115
097ec7c1 116Revision 1.16 2000/12/02 11:41:39 morsch
117Use SetRandom() to initialize random number generator in constructor.
118
1ed25e11 119Revision 1.15 2000/11/30 20:29:02 morsch
120Initialise static variable sRandom in constructor: sRandom = fRandom;
121
ad1df918 122Revision 1.14 2000/11/30 07:12:50 alibrary
123Introducing new Rndm and QA classes
124
65fb704d 125Revision 1.13 2000/11/09 17:40:27 morsch
126Possibility to select/unselect spectator protons and neutrons.
127Method SetSpectators(Int_t spect) added. (FCA, Ch. Oppedisano)
128
2705454f 129Revision 1.12 2000/10/20 13:38:38 morsch
130Debug printouts commented.
131
1df2195f 132Revision 1.11 2000/10/20 13:22:26 morsch
133- skip particle type 92 (string)
134- Charmed and beauty baryions (5122, 4122) are considered as stable consistent with
135 mesons.
136
608708ba 137Revision 1.10 2000/10/17 15:10:20 morsch
138Write first all the parent particles to the stack and then the final state particles.
139
e06552ba 140Revision 1.9 2000/10/17 13:38:59 morsch
141Protection against division by zero in EvaluateCrossSection() and KinematicSelection(..) (FCA)
142
c007646a 143Revision 1.8 2000/10/17 12:46:31 morsch
144Protect EvaluateCrossSections() against division by zero.
145
73bc6a56 146Revision 1.7 2000/10/02 21:28:06 fca
147Removal of useless dependecies via forward declarations
148
94de3818 149Revision 1.6 2000/09/11 13:23:37 morsch
150Write last seed to file (fortran lun 50) and reed back from same lun using calls to
151luget_hijing and luset_hijing.
152
b78ff279 153Revision 1.5 2000/09/07 16:55:40 morsch
154fHijing->Initialize(); after change of parameters. (Dmitri Yurevitch Peressounko)
155
ba244bac 156Revision 1.4 2000/07/11 18:24:56 fca
157Coding convention corrections + few minor bug fixes
158
aee8290b 159Revision 1.3 2000/06/30 12:08:36 morsch
160In member data: char* replaced by TString, Init takes care of resizing the strings to
1618 characters required by Hijing.
162
3b8bd63f 163Revision 1.2 2000/06/15 14:15:05 morsch
164Add possibility for heavy flavor selection: charm and beauty.
165
eb342d2d 166Revision 1.1 2000/06/09 20:47:27 morsch
167AliGenerator interface class to HIJING using THijing (test version)
168
110410f9 169*/
170
675e9664 171
172
173// Generator using HIJING as an external generator
174// The main HIJING options are accessable for the user through this interface.
175// Uses the THijing implementation of TGenerator.
176//
177// andreas.morsch@cern.ch
178
110410f9 179#include "AliGenHijing.h"
3b8bd63f 180#include "AliGenHijingEventHeader.h"
110410f9 181#include "AliRun.h"
b46f2676 182#include "AliPDG.h"
110410f9 183
184#include <TArrayI.h>
185#include <TParticle.h>
186#include <THijing.h>
21ce2e6f 187#include <TGraph.h>
332c8db0 188#include <TLorentzVector.h>
110410f9 189
ad1df918 190
110410f9 191 ClassImp(AliGenHijing)
192
193AliGenHijing::AliGenHijing()
55d6adcf 194 :AliGenMC()
110410f9 195{
196// Constructor
7a013373 197 fParticles = 0;
198 fHijing = 0;
199 fDsigmaDb = 0;
200 fDnDb = 0;
110410f9 201}
202
203AliGenHijing::AliGenHijing(Int_t npart)
55d6adcf 204 :AliGenMC(npart)
110410f9 205{
206// Default PbPb collisions at 5. 5 TeV
207//
8b31bfac 208 fName = "Hijing";
209 fTitle= "Particle Generator using HIJING";
210
110410f9 211 SetEnergyCMS();
212 SetImpactParameterRange();
213 SetTarget();
214 SetProjectile();
cc71f2d3 215 SetBoostLHC();
e4ffda63 216 SetJetEtaRange();
777b4893 217 SetJetPhiRange();
e4ffda63 218
ed6361b0 219 fKeep = 0;
220 fQuench = 1;
221 fShadowing = 1;
222 fTrigger = 0;
223 fDecaysOff = 1;
224 fEvaluate = 0;
225 fSelectAll = 0;
226 fFlavor = 0;
227 fSpectators = 1;
228 fDsigmaDb = 0;
229 fDnDb = 0;
230 fPtMinJet = -2.5;
83a26417 231 fRadiation = 3;
f8b85aa9 232 fEventVertex.Set(3);
3792091f 233//
234 SetSimpleJets();
0b22c984 235 SetNoGammas();
7a013373 236//
237 fParticles = new TClonesArray("TParticle",10000);
ad1df918 238//
1ed25e11 239// Set random number generator
097ec7c1 240 sRandom = fRandom;
2685bf00 241 fHijing = 0;
242
110410f9 243}
244
245AliGenHijing::AliGenHijing(const AliGenHijing & Hijing)
246{
247// copy constructor
248}
249
250
251AliGenHijing::~AliGenHijing()
252{
253// Destructor
21ce2e6f 254 if ( fDsigmaDb) delete fDsigmaDb;
255 if ( fDnDb) delete fDnDb;
7a013373 256 delete fParticles;
110410f9 257}
258
259void AliGenHijing::Init()
260{
261// Initialisation
3b8bd63f 262 fFrame.Resize(8);
263 fTarget.Resize(8);
264 fProjectile.Resize(8);
265
110410f9 266 SetMC(new THijing(fEnergyCMS, fFrame, fProjectile, fTarget,
267 fAProjectile, fZProjectile, fATarget, fZTarget,
268 fMinImpactParam, fMaxImpactParam));
269
270 fHijing=(THijing*) fgMCEvGen;
332c8db0 271 fHijing->SetIHPR2(2, fRadiation);
110410f9 272 fHijing->SetIHPR2(3, fTrigger);
110410f9 273 fHijing->SetIHPR2(6, fShadowing);
274 fHijing->SetIHPR2(12, fDecaysOff);
275 fHijing->SetIHPR2(21, fKeep);
ed6361b0 276 fHijing->SetHIPR1(10, fPtMinJet);
3792091f 277 fHijing->SetHIPR1(50, fSimpleJet);
9bf3f17b 278//
279// Quenching
280//
281//
282// fQuench = 0: no quenching
283// fQuench = 1: hijing default
284// fQuench = 2: new LHC parameters for HIPR1(11) and HIPR1(14)
285// fQuench = 3: new RHIC parameters for HIPR1(11) and HIPR1(14)
286// fQuench = 4: new LHC parameters with log(e) dependence
287// fQuench = 5: new RHIC parameters with log(e) dependence
288 fHijing->SetIHPR2(50, 0);
289 if (fQuench > 0)
290 fHijing->SetIHPR2(4, 1);
291 else
292 fHijing->SetIHPR2(4, 0);
293// New LHC parameters from Xin-Nian Wang
294 if (fQuench == 2) {
295 fHijing->SetHIPR1(14, 1.1);
296 fHijing->SetHIPR1(11, 3.7);
297 } else if (fQuench == 3) {
298 fHijing->SetHIPR1(14, 0.20);
299 fHijing->SetHIPR1(11, 2.5);
300 } else if (fQuench == 4) {
301 fHijing->SetIHPR2(50, 1);
302 fHijing->SetHIPR1(14, 4.*0.34);
303 fHijing->SetHIPR1(11, 3.7);
304 } else if (fQuench == 5) {
305 fHijing->SetIHPR2(50, 1);
306 fHijing->SetHIPR1(14, 0.34);
307 fHijing->SetHIPR1(11, 2.5);
308 }
309
310
311
312//
313// Initialize Hijing
314//
ba244bac 315 fHijing->Initialize();
110410f9 316//
317 if (fEvaluate) EvaluateCrossSections();
ad1df918 318//
110410f9 319}
320
321void AliGenHijing::Generate()
322{
323// Generate one event
324
21ce2e6f 325 Float_t polar[3] = {0,0,0};
326 Float_t origin[3] = {0,0,0};
327 Float_t origin0[3] = {0,0,0};
28adac24 328 Float_t p[3], random[6];
329 Float_t tof;
110410f9 330
110410f9 331// converts from mm/c to s
21ce2e6f 332 const Float_t kconv = 0.001/2.999792458e8;
110410f9 333//
21ce2e6f 334 Int_t nt = 0;
335 Int_t jev = 0;
28adac24 336 Int_t j, kf, ks, imo;
21ce2e6f 337 kf = 0;
eb342d2d 338
7a013373 339
110410f9 340
21ce2e6f 341 fTrials = 0;
342 for (j = 0;j < 3; j++) origin0[j] = fOrigin[j];
343 if(fVertexSmear == kPerEvent) {
f8b85aa9 344 Float_t dv[3];
345 dv[2] = 1.e10;
346 while(TMath::Abs(dv[2]) > fCutVertexZ*fOsigma[2]) {
347 Rndm(random,6);
348 for (j=0; j < 3; j++) {
349 dv[j] = fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())*
350 TMath::Sqrt(-2*TMath::Log(random[2*j+1]));
351 }
352 }
353 for (j=0; j < 3; j++) origin0[j] += dv[j];
21ce2e6f 354 } else if (fVertexSmear == kPerTrack) {
28adac24 355// fHijing->SetMSTP(151,0);
21ce2e6f 356 for (j = 0; j < 3; j++) {
28adac24 357// fHijing->SetPARP(151+j, fOsigma[j]*10.);
21ce2e6f 358 }
28adac24 359 }
360 while(1)
21ce2e6f 361 {
362// Generate one event
b46f2676 363// --------------------------------------------------------------------------
364 fSpecn = 0;
365 fSpecp = 0;
366// --------------------------------------------------------------------------
28adac24 367 fHijing->GenerateEvent();
368 fTrials++;
7a013373 369 fHijing->ImportParticles(fParticles,"All");
e4ffda63 370 if (fTrigger != kNoTrigger) {
371 if (!CheckTrigger()) continue;
372 }
7a013373 373 if (fLHC) Boost();
cc71f2d3 374
e4ffda63 375
7a013373 376 Int_t np = fParticles->GetEntriesFast();
28adac24 377 printf("\n **************************************************%d\n",np);
21ce2e6f 378 Int_t nc = 0;
28adac24 379 if (np == 0 ) continue;
380 Int_t i;
f477ff09 381 Int_t* newPos = new Int_t[np];
382 Int_t* pSelected = new Int_t[np];
28adac24 383
f477ff09 384 for (i = 0; i < np; i++) {
385 newPos[i] = i;
386 pSelected[i] = 0;
387 }
388
f8b85aa9 389// Get event vertex
390//
7a013373 391 TParticle * iparticle = (TParticle *) fParticles->At(0);
f8b85aa9 392 fEventVertex[0] = origin0[0];
393 fEventVertex[1] = origin0[1];
394 fEventVertex[2] = origin0[2];
395
e06552ba 396//
f477ff09 397// First select parent particles
e06552ba 398//
399
21ce2e6f 400 for (i = 0; i < np; i++) {
7a013373 401 iparticle = (TParticle *) fParticles->At(i);
402
e06552ba 403// Is this a parent particle ?
f8b85aa9 404 if (Stable(iparticle)) continue;
e06552ba 405//
f477ff09 406 Bool_t selected = kTRUE;
407 Bool_t hasSelectedDaughters = kFALSE;
408
409
410 kf = iparticle->GetPdgCode();
411 ks = iparticle->GetStatusCode();
412 if (kf == 92) continue;
e06552ba 413
f477ff09 414 if (!fSelectAll) selected = KinematicSelection(iparticle, 0) &&
415 SelectFlavor(kf);
7a013373 416 hasSelectedDaughters = DaughtersSelection(iparticle);
110410f9 417//
f477ff09 418// Put particle on the stack if it is either selected or
419// it is the mother of at least one seleted particle
e06552ba 420//
f477ff09 421 if (selected || hasSelectedDaughters) {
422 nc++;
423 pSelected[i] = 1;
424 } // selected
28adac24 425 } // particle loop parents
e06552ba 426//
f477ff09 427// Now select the final state particles
e06552ba 428//
429
28adac24 430 for (i = 0; i<np; i++) {
7a013373 431 TParticle * iparticle = (TParticle *) fParticles->At(i);
e06552ba 432// Is this a final state particle ?
f477ff09 433 if (!Stable(iparticle)) continue;
434
435 Bool_t selected = kTRUE;
436 kf = iparticle->GetPdgCode();
437 ks = iparticle->GetStatusCode();
7a013373 438
b46f2676 439// --------------------------------------------------------------------------
440// Count spectator neutrons and protons
f477ff09 441 if(ks == 0 || ks == 1 || ks == 10 || ks == 11){
442 if(kf == kNeutron) fSpecn += 1;
443 if(kf == kProton) fSpecp += 1;
444 }
b46f2676 445// --------------------------------------------------------------------------
446//
f477ff09 447 if (!fSelectAll) {
448 selected = KinematicSelection(iparticle,0)&&SelectFlavor(kf);
449 if (!fSpectators && selected) selected = (ks != 0 && ks != 1 && ks != 10
450 && ks != 11);
451 }
e06552ba 452//
453// Put particle on the stack if selected
454//
f477ff09 455 if (selected) {
456 nc++;
457 pSelected[i] = 1;
458 } // selected
28adac24 459 } // particle loop final state
f477ff09 460//
461// Write particles to stack
462//
463 for (i = 0; i<np; i++) {
7a013373 464 TParticle * iparticle = (TParticle *) fParticles->At(i);
f477ff09 465 Bool_t hasMother = (iparticle->GetFirstMother() >=0);
466 Bool_t hasDaughter = (iparticle->GetFirstDaughter() >=0);
467
468 if (pSelected[i]) {
469 kf = iparticle->GetPdgCode();
0d7c70a0 470 ks = iparticle->GetStatusCode();
f477ff09 471 p[0] = iparticle->Px();
472 p[1] = iparticle->Py();
473 p[2] = iparticle->Pz();
474 origin[0] = origin0[0]+iparticle->Vx()/10;
475 origin[1] = origin0[1]+iparticle->Vy()/10;
476 origin[2] = origin0[2]+iparticle->Vz()/10;
477 tof = kconv*iparticle->T();
478 imo = -1;
479 TParticle* mother = 0;
480 if (hasMother) {
481 imo = iparticle->GetFirstMother();
7a013373 482 mother = (TParticle *) fParticles->At(imo);
f477ff09 483 imo = (mother->GetPdgCode() != 92) ? imo = newPos[imo] : -1;
484 } // if has mother
485 Bool_t tFlag = (fTrackIt && !hasDaughter);
486 SetTrack(tFlag,imo,kf,p,origin,polar,
0d7c70a0 487 tof,kPNoProcess,nt, 1., ks);
f477ff09 488 KeepTrack(nt);
489 newPos[i] = nt;
490 } // if selected
491 } // particle loop
79bf94df 492 delete[] newPos;
f477ff09 493 delete[] pSelected;
494
28adac24 495 printf("\n I've put %i particles on the stack \n",nc);
496 if (nc > 0) {
21ce2e6f 497 jev += nc;
498 if (jev >= fNpart || fNpart == -1) {
499 fKineBias = Float_t(fNpart)/Float_t(fTrials);
500 printf("\n Trials: %i %i %i\n",fTrials, fNpart, jev);
501 break;
502 }
28adac24 503 }
21ce2e6f 504 } // event loop
505 MakeHeader();
a99cf51f 506 SetHighWaterMark(nt);
110410f9 507}
508
110410f9 509void AliGenHijing::KeepFullEvent()
510{
511 fKeep=1;
512}
513
514void AliGenHijing::EvaluateCrossSections()
515{
516// Glauber Calculation of geometrical x-section
517//
21ce2e6f 518 Float_t xTot = 0.; // barn
519 Float_t xTotHard = 0.; // barn
520 Float_t xPart = 0.; // barn
521 Float_t xPartHard = 0.; // barn
522 Float_t sigmaHard = 0.1; // mbarn
523 Float_t bMin = 0.;
524 Float_t bMax = fHijing->GetHIPR1(34)+fHijing->GetHIPR1(35);
525 const Float_t kdib = 0.2;
526 Int_t kMax = Int_t((bMax-bMin)/kdib)+1;
110410f9 527
528
529 printf("\n Projectile Radius (fm): %f \n",fHijing->GetHIPR1(34));
530 printf("\n Target Radius (fm): %f \n",fHijing->GetHIPR1(35));
531 Int_t i;
21ce2e6f 532 Float_t oldvalue= 0.;
533
534 Float_t* b = new Float_t[kMax];
535 Float_t* si1 = new Float_t[kMax];
536 Float_t* si2 = new Float_t[kMax];
110410f9 537
21ce2e6f 538 for (i = 0; i < kMax; i++)
110410f9 539 {
21ce2e6f 540 Float_t xb = bMin+i*kdib;
110410f9 541 Float_t ov;
542 ov=fHijing->Profile(xb);
21ce2e6f 543 Float_t gb = 2.*0.01*fHijing->GetHIPR1(40)*kdib*xb*(1.-TMath::Exp(-fHijing->GetHINT1(12)*ov));
544 Float_t gbh = 2.*0.01*fHijing->GetHIPR1(40)*kdib*xb*sigmaHard*ov;
110410f9 545 xTot+=gb;
21ce2e6f 546 xTotHard += gbh;
110410f9 547 if (xb > fMinImpactParam && xb < fMaxImpactParam)
548 {
21ce2e6f 549 xPart += gb;
550 xPartHard += gbh;
110410f9 551 }
552
c007646a 553 if(oldvalue) if ((xTot-oldvalue)/oldvalue<0.0001) break;
21ce2e6f 554 oldvalue = xTot;
110410f9 555 printf("\n Total cross section (barn): %d %f %f \n",i, xb, xTot);
556 printf("\n Hard cross section (barn): %d %f %f \n\n",i, xb, xTotHard);
21ce2e6f 557 if (i>0) {
558 si1[i] = gb/kdib;
559 si2[i] = gbh/gb;
560 b[i] = xb;
561 }
110410f9 562 }
21ce2e6f 563
110410f9 564 printf("\n Total cross section (barn): %f \n",xTot);
565 printf("\n Hard cross section (barn): %f \n \n",xTotHard);
566 printf("\n Partial cross section (barn): %f %f \n",xPart, xPart/xTot*100.);
567 printf("\n Partial hard cross section (barn): %f %f \n",xPartHard, xPartHard/xTotHard*100.);
21ce2e6f 568
569// Store result as a graph
570 b[0] = 0;
571 si1[0] = 0;
572 si2[0]=si2[1];
573
574 fDsigmaDb = new TGraph(i, b, si1);
575 fDnDb = new TGraph(i, b, si2);
110410f9 576}
577
7a013373 578Bool_t AliGenHijing::DaughtersSelection(TParticle* iparticle)
110410f9 579{
580//
581// Looks recursively if one of the daughters has been selected
582//
583// printf("\n Consider daughters %d:",iparticle->GetPdgCode());
21ce2e6f 584 Int_t imin = -1;
585 Int_t imax = -1;
110410f9 586 Int_t i;
21ce2e6f 587 Bool_t hasDaughters = (iparticle->GetFirstDaughter() >=0);
588 Bool_t selected = kFALSE;
110410f9 589 if (hasDaughters) {
21ce2e6f 590 imin = iparticle->GetFirstDaughter();
591 imax = iparticle->GetLastDaughter();
592 for (i = imin; i <= imax; i++){
7a013373 593 TParticle * jparticle = (TParticle *) fParticles->At(i);
21ce2e6f 594 Int_t ip = jparticle->GetPdgCode();
55d6adcf 595 if (KinematicSelection(jparticle,0)&&SelectFlavor(ip)) {
608708ba 596 selected=kTRUE; break;
597 }
7a013373 598 if (DaughtersSelection(jparticle)) {selected=kTRUE; break; }
110410f9 599 }
600 } else {
601 return kFALSE;
602 }
110410f9 603 return selected;
604}
605
606
eb342d2d 607Bool_t AliGenHijing::SelectFlavor(Int_t pid)
608{
609// Select flavor of particle
610// 0: all
611// 4: charm and beauty
612// 5: beauty
0b22c984 613 Bool_t res = 0;
eb342d2d 614
cdd538f1 615 if (fFlavor == 0) {
616 res = kTRUE;
617 } else {
618 Int_t ifl = TMath::Abs(pid/100);
619 if (ifl > 10) ifl/=10;
620 res = (fFlavor == ifl);
621 }
622//
623// This part if gamma writing is inhibited
624 if (fNoGammas)
625 res = res && (pid != kGamma && pid != kPi0);
626//
0b22c984 627 return res;
608708ba 628}
eb342d2d 629
608708ba 630Bool_t AliGenHijing::Stable(TParticle* particle)
631{
675e9664 632// Return true for a stable particle
633//
0b22c984 634
79bf94df 635 if (particle->GetFirstDaughter() < 0 )
608708ba 636 {
637 return kTRUE;
638 } else {
639 return kFALSE;
640 }
eb342d2d 641}
110410f9 642
cc71f2d3 643
7a013373 644void AliGenHijing::Boost()
cc71f2d3 645{
646//
647// Boost cms into LHC lab frame
648//
649 Double_t dy = - 0.5 * TMath::Log(Double_t(fZProjectile) * Double_t(fATarget) /
650 (Double_t(fZTarget) * Double_t(fAProjectile)));
651 Double_t beta = TMath::TanH(dy);
652 Double_t gamma = 1./TMath::Sqrt(1.-beta*beta);
653 Double_t gb = gamma * beta;
654
655 printf("\n Boosting particles to lab frame %f %f %f", dy, beta, gamma);
656
657 Int_t i;
7a013373 658 Int_t np = fParticles->GetEntriesFast();
cc71f2d3 659 for (i = 0; i < np; i++)
660 {
7a013373 661 TParticle* iparticle = (TParticle*) fParticles->At(i);
cc71f2d3 662
663 Double_t e = iparticle->Energy();
664 Double_t px = iparticle->Px();
665 Double_t py = iparticle->Py();
666 Double_t pz = iparticle->Pz();
667
668 Double_t eb = gamma * e - gb * pz;
669 Double_t pzb = -gb * e + gamma * pz;
670
671 iparticle->SetMomentum(px, py, pzb, eb);
672 }
673}
674
675
3b8bd63f 676void AliGenHijing::MakeHeader()
677{
678// Builds the event header, to be called after each event
21ce2e6f 679 AliGenEventHeader* header = new AliGenHijingEventHeader("Hijing");
332c8db0 680 ((AliGenHijingEventHeader*) header)->SetNProduced(fHijing->GetNATT());
681 ((AliGenHijingEventHeader*) header)->SetImpactParameter(fHijing->GetHINT1(19));
682 ((AliGenHijingEventHeader*) header)->SetTotalEnergy(fHijing->GetEATT());
683 ((AliGenHijingEventHeader*) header)->SetHardScatters(fHijing->GetJATT());
684 ((AliGenHijingEventHeader*) header)->SetParticipants(fHijing->GetNP(), fHijing->GetNT());
685 ((AliGenHijingEventHeader*) header)->SetCollisions(fHijing->GetN0(),
686 fHijing->GetN01(),
687 fHijing->GetN10(),
688 fHijing->GetN11());
689 ((AliGenHijingEventHeader*) header)->SetSpectators(fSpecn, fSpecp);
690
f8b85aa9 691// 4-momentum vectors of the triggered jets.
692//
693// Before final state gluon radiation.
332c8db0 694 TLorentzVector* jet1 = new TLorentzVector(fHijing->GetHINT1(21),
695 fHijing->GetHINT1(22),
696 fHijing->GetHINT1(23),
697 fHijing->GetHINT1(24));
698
699 TLorentzVector* jet2 = new TLorentzVector(fHijing->GetHINT1(31),
700 fHijing->GetHINT1(32),
701 fHijing->GetHINT1(33),
702 fHijing->GetHINT1(34));
f8b85aa9 703// After final state gluon radiation.
704 TLorentzVector* jet3 = new TLorentzVector(fHijing->GetHINT1(26),
705 fHijing->GetHINT1(27),
706 fHijing->GetHINT1(28),
707 fHijing->GetHINT1(29));
708
709 TLorentzVector* jet4 = new TLorentzVector(fHijing->GetHINT1(36),
710 fHijing->GetHINT1(37),
711 fHijing->GetHINT1(38),
712 fHijing->GetHINT1(39));
713 ((AliGenHijingEventHeader*) header)->SetJets(jet1, jet2, jet3, jet4);
7a186781 714// Bookkeeping for kinematic bias
e4ffda63 715 ((AliGenHijingEventHeader*) header)->SetTrials(fTrials);
f8b85aa9 716// Event Vertex
717 header->SetPrimaryVertex(fEventVertex);
96a7d299 718 gAlice->SetGenEventHeader(header);
719 fCollisionGeometry = (AliGenHijingEventHeader*) header;
3b8bd63f 720}
721
e4ffda63 722Bool_t AliGenHijing::CheckTrigger()
723{
724// Check the kinematic trigger condition
7a013373 725//
e4ffda63 726 Bool_t triggered = kFALSE;
7a013373 727
728 if (fTrigger == 1) {
729//
730// jet-jet Trigger
731
732 TLorentzVector* jet1 = new TLorentzVector(fHijing->GetHINT1(26),
733 fHijing->GetHINT1(27),
734 fHijing->GetHINT1(28),
735 fHijing->GetHINT1(29));
736
737 TLorentzVector* jet2 = new TLorentzVector(fHijing->GetHINT1(36),
738 fHijing->GetHINT1(37),
739 fHijing->GetHINT1(38),
740 fHijing->GetHINT1(39));
741 Double_t eta1 = jet1->Eta();
742 Double_t eta2 = jet2->Eta();
743 Double_t phi1 = jet1->Phi();
744 Double_t phi2 = jet2->Phi();
9bf3f17b 745// printf("\n Trigger: %f %f %f %f",
746// fEtaMinJet, fEtaMaxJet, fPhiMinJet, fPhiMaxJet);
7a013373 747 if (
748 (eta1 < fEtaMaxJet && eta1 > fEtaMinJet &&
749 phi1 < fPhiMaxJet && phi1 > fPhiMinJet)
750 ||
751 (eta2 < fEtaMaxJet && eta2 > fEtaMinJet &&
752 phi2 < fPhiMaxJet && phi2 > fPhiMinJet)
753 )
754 triggered = kTRUE;
755 } else if (fTrigger == 2) {
756// Gamma Jet
757//
758 Int_t np = fParticles->GetEntriesFast();
759 for (Int_t i = 0; i < np; i++) {
760 TParticle* part = (TParticle*) fParticles->At(i);
761 Int_t kf = part->GetPdgCode();
762 Int_t ks = part->GetStatusCode();
763 if (kf == 22 && ks == 40) {
764 Float_t phi = part->Phi();
765 Float_t eta = part->Eta();
766 if (eta < fEtaMaxJet &&
767 eta > fEtaMinJet &&
768 phi < fPhiMaxJet &&
769 phi > fPhiMinJet) {
770 triggered = 1;
771 break;
772 } // check phi,eta within limits
773 } // direct gamma ?
774 } // particle loop
775 } // fTrigger == 2
e4ffda63 776 return triggered;
777}
cc71f2d3 778
3792091f 779
780
781
110410f9 782AliGenHijing& AliGenHijing::operator=(const AliGenHijing& rhs)
783{
784// Assignment operator
785 return *this;
786}
608708ba 787
65fb704d 788#ifndef WIN32
789# define rluget_hijing rluget_hijing_
790# define rluset_hijing rluset_hijing_
791# define rlu_hijing rlu_hijing_
792# define type_of_call
793#else
794# define rluget_hijing RLUGET_HIJING
795# define rluset_hijing RLUSET_HIJING
796# define rlu_hijing RLU_HIJING
797# define type_of_call _stdcall
798#endif
799
800
801extern "C" {
802 void type_of_call rluget_hijing(Int_t & /*lfn*/, Int_t & /*move*/)
803 {printf("Dummy version of rluget_hijing reached\n");}
804
805 void type_of_call rluset_hijing(Int_t & /*lfn*/, Int_t & /*move*/)
806 {printf("Dummy version of rluset_hijing reached\n");}
807
808 Double_t type_of_call rlu_hijing(Int_t & /*idum*/)
7498b3ad 809 {
810 Float_t r;
811 do r=sRandom->Rndm(); while(0 >= r || r >= 1);
812 return r;
813 }
65fb704d 814}