- Updates necessary for newio.
[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythia.cxx
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 /* $Id$ */
17
18 //
19 // Generator using the TPythia interface (via AliPythia)
20 // to generate pp collisions.
21 // Using SetNuclei() also nuclear modifications to the structure functions
22 // can be taken into account. This makes, of course, only sense for the
23 // generation of the products of hard processes (heavy flavor, jets ...)
24 //
25 // andreas.morsch@cern.ch
26 //
27
28 #include <TDatabasePDG.h>
29 #include <TParticle.h>
30 #include <TPDGCode.h>
31 #include <TSystem.h>
32 #include <TTree.h>
33
34 #include "AliConst.h"
35 #include "AliDecayerPythia.h"
36 #include "AliGenPythia.h"
37 #include "AliGenPythiaEventHeader.h"
38 #include "AliPythia.h"
39 #include "AliPythiaRndm.h"
40 #include "AliRun.h"
41 #include "AliStack.h"
42 #include "AliRunLoader.h"
43
44 ClassImp(AliGenPythia)
45
46 AliGenPythia::AliGenPythia()
47                  :AliGenMC()
48 {
49 // Default Constructor
50   fParticles = 0;
51   fPythia    = 0;
52   fHeader = 0;
53   fDecayer   = new AliDecayerPythia();
54   SetEventListRange();
55   SetJetPhiRange();
56   SetJetEtaRange();
57   SetJetEtRange();
58   SetGammaPhiRange();
59   SetGammaEtaRange();
60   SetPtKick();
61   SetQuench();
62   
63   fSetNuclei = kFALSE;
64   if (!AliPythiaRndm::GetPythiaRandom()) 
65     AliPythiaRndm::SetPythiaRandom(GetRandom());
66 }
67
68 AliGenPythia::AliGenPythia(Int_t npart)
69                  :AliGenMC(npart)
70 {
71 // default charm production at 5. 5 TeV
72 // semimuonic decay
73 // structure function GRVHO
74 //
75     fName = "Pythia";
76     fTitle= "Particle Generator using PYTHIA";
77     fXsection  = 0.;
78     SetProcess();
79     SetStrucFunc();
80     SetForceDecay();
81     SetPtHard();
82     SetYHard();
83     SetEnergyCMS();
84     fDecayer = new AliDecayerPythia();
85     // Set random number generator 
86     if (!AliPythiaRndm::GetPythiaRandom()) 
87       AliPythiaRndm::SetPythiaRandom(GetRandom());
88     fFlavorSelect   = 0;
89     // Produced particles  
90     fParticles = new TClonesArray("TParticle",1000);
91     fHeader = 0;
92     SetEventListRange();
93     SetJetPhiRange();
94     SetJetEtaRange();
95     SetJetEtRange();
96     SetGammaPhiRange();
97     SetGammaEtaRange();
98     SetJetReconstructionMode();
99     SetQuench();
100     SetPtKick();
101     // Options determining what to keep in the stack (Heavy flavour generation)
102     fStackFillOpt = kFlavorSelection; // Keep particle with selected flavor
103     fFeedDownOpt = kTRUE;             // allow feed down from higher family
104     // Fragmentation on/off
105     fFragmentation = kTRUE;
106     // Default counting mode
107     fCountMode = kCountAll;
108     // Pycel
109     SetPycellParameters();
110     fSetNuclei = kFALSE;
111 }
112
113 AliGenPythia::AliGenPythia(const AliGenPythia & Pythia)
114     :AliGenMC(Pythia)
115 {
116 // copy constructor
117     Pythia.Copy(*this);
118 }
119
120 AliGenPythia::~AliGenPythia()
121 {
122 // Destructor
123 }
124
125 void AliGenPythia::SetPycellParameters(Float_t etamax, Int_t neta, Int_t nphi,
126                                        Float_t thresh, Float_t etseed, Float_t minet, Float_t r)
127 {
128 // Set pycell parameters
129     fPycellEtaMax    =  etamax;
130     fPycellNEta      =  neta;
131     fPycellNPhi      =  nphi;
132     fPycellThreshold =  thresh;
133     fPycellEtSeed    =  etseed;
134     fPycellMinEtJet  =  minet;
135     fPycellMaxRadius =  r;
136 }
137
138
139
140 void AliGenPythia::SetEventListRange(Int_t eventFirst, Int_t eventLast)
141 {
142   // Set a range of event numbers, for which a table
143   // of generated particle will be printed
144   fDebugEventFirst = eventFirst;
145   fDebugEventLast  = eventLast;
146   if (fDebugEventLast==-1) fDebugEventLast=fDebugEventFirst;
147 }
148
149 void AliGenPythia::Init()
150 {
151 // Initialisation
152     
153     SetMC(AliPythia::Instance());
154     fPythia=(AliPythia*) fMCEvGen;
155 //
156     fParentWeight=1./Float_t(fNpart);
157 //
158 //  Forward Paramters to the AliPythia object
159     fDecayer->SetForceDecay(fForceDecay);    
160     fDecayer->Init();
161
162
163     fPythia->SetCKIN(3,fPtHardMin);
164     fPythia->SetCKIN(4,fPtHardMax);
165     fPythia->SetCKIN(7,fYHardMin);
166     fPythia->SetCKIN(8,fYHardMax);
167     
168     if (fAProjectile > 0 && fATarget > 0) fPythia->SetNuclei(fAProjectile, fATarget);  
169     // Fragmentation?
170     if (fFragmentation) {
171       fPythia->SetMSTP(111,1);
172     } else {
173       fPythia->SetMSTP(111,0);
174     }
175
176
177 //  initial state radiation   
178     fPythia->SetMSTP(61,fGinit);
179 //  final state radiation
180     fPythia->SetMSTP(71,fGfinal);
181 //  pt - kick
182     if (fPtKick > 0.) {
183         fPythia->SetMSTP(91,1);
184         fPythia->SetPARP(91,fPtKick);
185     } else {
186         fPythia->SetMSTP(91,0);
187     }
188
189  //   fPythia->SetMSTJ(1,2);
190  //
191     fPythia->ProcInit(fProcess,fEnergyCMS,fStrucFunc);
192
193 //  Parent and Children Selection
194     switch (fProcess) 
195     {
196     case kPyCharm:
197     case kPyCharmUnforced:
198     case kPyCharmPbPbMNR:
199     case kPyCharmppMNR:
200     case kPyCharmpPbMNR:
201         fParentSelect[0] =   411;
202         fParentSelect[1] =   421;
203         fParentSelect[2] =   431;
204         fParentSelect[3] =  4122;
205         fFlavorSelect    =  4;  
206         break;
207     case kPyD0PbPbMNR:
208     case kPyD0pPbMNR:
209     case kPyD0ppMNR:
210         fParentSelect[0] =   421;
211         fFlavorSelect    =   4; 
212         break;
213     case kPyBeauty:
214     case kPyBeautyPbPbMNR:
215     case kPyBeautypPbMNR:
216     case kPyBeautyppMNR:
217         fParentSelect[0]=  511;
218         fParentSelect[1]=  521;
219         fParentSelect[2]=  531;
220         fParentSelect[3]= 5122;
221         fParentSelect[4]= 5132;
222         fParentSelect[5]= 5232;
223         fParentSelect[6]= 5332;
224         fFlavorSelect   = 5;    
225         break;
226     case kPyBeautyUnforced:
227         fParentSelect[0] =  511;
228         fParentSelect[1] =  521;
229         fParentSelect[2] =  531;
230         fParentSelect[3] = 5122;
231         fParentSelect[4] = 5132;
232         fParentSelect[5] = 5232;
233         fParentSelect[6] = 5332;
234         fFlavorSelect    = 5;   
235         break;
236     case kPyJpsiChi:
237     case kPyJpsi:
238         fParentSelect[0] = 443;
239         break;
240     case kPyMb:
241     case kPyMbNonDiffr:
242     case kPyJets:
243     case kPyDirectGamma:
244         break;
245     }
246 //
247 //
248 //  JetFinder for Trigger
249 //
250 //  Configure detector (EMCAL like)
251 //
252         fPythia->SetPARU(51, fPycellEtaMax);
253         fPythia->SetMSTU(51, fPycellNEta);
254         fPythia->SetMSTU(52, fPycellNPhi);
255 //
256 //  Configure Jet Finder
257 //  
258         fPythia->SetPARU(58,  fPycellThreshold);
259         fPythia->SetPARU(52,  fPycellEtSeed);
260         fPythia->SetPARU(53,  fPycellMinEtJet);
261         fPythia->SetPARU(54,  fPycellMaxRadius);
262         fPythia->SetMSTU(54,  2);
263 //
264 //  This counts the total number of calls to Pyevnt() per run.
265     fTrialsRun = 0;
266     fQ         = 0.;
267     fX1        = 0.;
268     fX2        = 0.;    
269     fNev       = 0 ;
270 //    
271 //
272 //
273     AliGenMC::Init();
274 //
275 //
276 //  
277     if (fSetNuclei) {
278         fDyBoost = 0;
279         Warning("Init","SetNuclei used. Use SetProjectile + SetTarget instead. fDyBoost has been reset to 0\n");
280     }
281 }
282
283 void AliGenPythia::Generate()
284 {
285 // Generate one event
286     
287     fDecayer->ForceDecay();
288
289     Float_t polar[3]   =   {0,0,0};
290     Float_t origin[3]  =   {0,0,0};
291     Float_t p[3];
292 //  converts from mm/c to s
293     const Float_t kconv=0.001/2.999792458e8;
294 //
295     Int_t nt=0;
296     Int_t jev=0;
297     Int_t j, kf;
298     fTrials=0;
299
300     //  Set collision vertex position 
301     if(fVertexSmear==kPerEvent) {
302         fPythia->SetMSTP(151,1);
303         for (j=0;j<3;j++) {
304             fPythia->SetPARP(151+j, fOsigma[j]*10.);
305         }
306     } else if (fVertexSmear==kPerTrack) {
307         fPythia->SetMSTP(151,0);
308     }
309 //  event loop    
310     while(1)
311     {
312         if (fQuench) {
313             fPythia->SetMSTJ(1, 0);
314         }
315
316         fPythia->Pyevnt();
317
318         if (fQuench) {
319             fPythia->Quench();
320             fPythia->SetMSTJ(1, 1);
321             fPythia->Pyexec();
322         }
323         
324         
325         if (gAlice->GetEvNumber()>=fDebugEventFirst &&
326             gAlice->GetEvNumber()<=fDebugEventLast) fPythia->Pylist(1);
327         fTrials++;
328         
329         fPythia->ImportParticles(fParticles,"All");
330
331         Boost();
332 //
333 //
334 //
335         Int_t i;
336         
337         Int_t np = fParticles->GetEntriesFast();
338         if (np == 0 ) continue;
339 // Get event vertex and discard the event if the Z coord. is too big    
340         TParticle *iparticle = (TParticle *) fParticles->At(0);
341         Float_t distz = iparticle->Vz()/10.;
342         if(TMath::Abs(distz)>fCutVertexZ*fOsigma[2]) continue;
343 //
344         fVertex[0] = iparticle->Vx()/10.+fOrigin.At(0);
345         fVertex[1] = iparticle->Vy()/10.+fOrigin.At(1);
346         fVertex[2] = iparticle->Vz()/10.+fOrigin.At(2);
347 //
348         Int_t* pParent   = new Int_t[np];
349         Int_t* pSelected = new Int_t[np];
350         Int_t* trackIt   = new Int_t[np];
351         for (i=0; i< np; i++) {
352             pParent[i]   = -1;
353             pSelected[i] =  0;
354             trackIt[i]   =  0;
355         }
356
357         Int_t nc = 0;        // Total n. of selected particles
358         Int_t nParents = 0;  // Selected parents
359         Int_t nTkbles = 0;   // Trackable particles
360         if (fProcess != kPyMb && fProcess != kPyJets && 
361             fProcess != kPyDirectGamma &&
362             fProcess != kPyMbNonDiffr) {
363             
364             for (i = 0; i<np; i++) {
365                 iparticle = (TParticle *) fParticles->At(i);
366                 Int_t ks = iparticle->GetStatusCode();
367                 kf = CheckPDGCode(iparticle->GetPdgCode());
368 // No initial state partons
369                 if (ks==21) continue;
370 //
371 // Heavy Flavor Selection
372 //
373                 // quark ?
374                 kf = TMath::Abs(kf);
375                 Int_t kfl = kf;
376                 // meson ?
377                 if  (kfl > 10) kfl/=100;
378                 // baryon
379                 if (kfl > 10) kfl/=10;
380                 if (kfl > 10) kfl/=10;
381
382                 Int_t ipa = iparticle->GetFirstMother()-1;
383                 Int_t kfMo = 0;
384                 
385                 if (ipa > -1) {
386                     TParticle *  mother = (TParticle *) fParticles->At(ipa);
387                     kfMo = TMath::Abs(mother->GetPdgCode());
388                 }
389                 // What to keep in Stack?
390                 Bool_t flavorOK = kFALSE;
391                 Bool_t selectOK = kFALSE;
392                 if (fFeedDownOpt) {
393                   if (kfl >= fFlavorSelect) flavorOK = kTRUE;
394                 } else {
395                   if (kfl > fFlavorSelect) {
396                     nc = -1;
397                     break;
398                   }
399                   if (kfl == fFlavorSelect) flavorOK = kTRUE;
400                 }
401                 switch (fStackFillOpt) {
402                 case kFlavorSelection:
403                   selectOK = kTRUE;
404                   break;
405                 case kParentSelection:
406                   if (ParentSelected(kf) || kf <= 10) selectOK = kTRUE;
407                   break;
408                 }
409                 if (flavorOK && selectOK) { 
410 //
411 // Heavy flavor hadron or quark
412 //
413 // Kinematic seletion on final state heavy flavor mesons
414                     if (ParentSelected(kf) && !KinematicSelection(iparticle, 0)) 
415                     {
416                       continue;
417                     }
418                     pSelected[i] = 1;
419                     if (ParentSelected(kf)) ++nParents; // Update parent count
420 //                  printf("\n particle (HF)  %d %d %d", i, pSelected[i], kf);
421                 } else {
422 // Kinematic seletion on decay products
423                     if (fCutOnChild && ParentSelected(kfMo) && ChildSelected(kf) 
424                         && !KinematicSelection(iparticle, 1))
425                     {
426                       continue;
427                     }
428 //
429 // Decay products 
430 // Select if mother was selected and is not tracked
431
432                     if (pSelected[ipa] && 
433                         !trackIt[ipa]  &&     // mother will be  tracked ?
434                         kfMo !=  5 &&         // mother is b-quark, don't store fragments          
435                         kfMo !=  4 &&         // mother is c-quark, don't store fragments 
436                         kf   != 92)           // don't store string
437                     {
438 //
439 // Semi-stable or de-selected: diselect decay products:
440 // 
441 //
442                         if (pSelected[i] == -1 ||  fDecayer->GetLifetime(kf) > fMaxLifeTime)
443                         {
444                             Int_t ipF = iparticle->GetFirstDaughter();
445                             Int_t ipL = iparticle->GetLastDaughter();   
446                             if (ipF > 0) for (j = ipF-1; j < ipL; j++) pSelected[j] = -1;
447                         }
448 //                      printf("\n particle (decay)  %d %d %d", i, pSelected[i], kf);
449                         pSelected[i] = (pSelected[i] == -1) ? 0 : 1;
450                     }
451                 }
452                 if (pSelected[i] == -1) pSelected[i] = 0;
453                 if (!pSelected[i]) continue;
454                 // Count quarks only if you did not include fragmentation
455                 if (fFragmentation && kf <= 10) continue;
456                 nc++;
457 // Decision on tracking
458                 trackIt[i] = 0;
459 //
460 // Track final state particle
461                 if (ks == 1) trackIt[i] = 1;
462 // Track semi-stable particles
463                 if ((ks == 1) || (fDecayer->GetLifetime(kf) > fMaxLifeTime))  trackIt[i] = 1;
464 // Track particles selected by process if undecayed. 
465                 if (fForceDecay == kNoDecay) {
466                     if (ParentSelected(kf)) trackIt[i] = 1;
467                 } else {
468                     if (ParentSelected(kf)) trackIt[i] = 0;
469                 }
470                 if (trackIt[i] == 1) ++nTkbles; // Update trackable counter
471 //
472 //
473
474             } // particle selection loop
475             if (nc > 0) {
476                 for (i = 0; i<np; i++) {
477                     if (!pSelected[i]) continue;
478                     TParticle *  iparticle = (TParticle *) fParticles->At(i);
479                     kf = CheckPDGCode(iparticle->GetPdgCode());
480                     Int_t ks = iparticle->GetStatusCode();  
481                     p[0] = iparticle->Px();
482                     p[1] = iparticle->Py();
483                     p[2] = iparticle->Pz();
484                     origin[0] = fOrigin[0]+iparticle->Vx()/10.;
485                     origin[1] = fOrigin[1]+iparticle->Vy()/10.;
486                     origin[2] = fOrigin[2]+iparticle->Vz()/10.;
487                     Float_t tof   = kconv*iparticle->T();
488                     Int_t ipa     = iparticle->GetFirstMother()-1;
489                     Int_t iparent = (ipa > -1) ? pParent[ipa] : -1;
490                     PushTrack(fTrackIt*trackIt[i] ,
491                                      iparent, kf, p, origin, polar, tof, kPPrimary, nt, 1., ks);
492                     pParent[i] = nt;
493                     KeepTrack(nt); 
494                 } //  PushTrack loop
495             }
496         } else {
497             nc = GenerateMB();
498         } // mb ?
499
500         if (pParent)   delete[] pParent;
501         if (pSelected) delete[] pSelected;
502         if (trackIt)   delete[] trackIt;
503
504         if (nc > 0) {
505           switch (fCountMode) {
506           case kCountAll:
507             // printf(" Count all \n");
508             jev += nc;
509             break;
510           case kCountParents:
511             // printf(" Count parents \n");
512             jev += nParents;
513             break;
514           case kCountTrackables:
515             // printf(" Count trackable \n");
516             jev += nTkbles;
517             break;
518           }
519             if (jev >= fNpart || fNpart == -1) {
520                 fKineBias=Float_t(fNpart)/Float_t(fTrials);
521                 printf("\n Trials: %i %i %i\n",fTrials, fNpart, jev);
522
523                 fQ  += fPythia->GetVINT(51);
524                 fX1 += fPythia->GetVINT(41);
525                 fX2 += fPythia->GetVINT(42);
526                 fTrialsRun += fTrials;
527                 fNev++;
528                 MakeHeader();
529                 break;
530             }
531         }
532     } // event loop
533     SetHighWaterMark(nt);
534 //  adjust weight due to kinematic selection
535 //    AdjustWeights();
536 //  get cross-section
537     fXsection=fPythia->GetPARI(1);
538 }
539
540 Int_t  AliGenPythia::GenerateMB()
541 {
542 //
543 // Min Bias selection and other global selections
544 //
545     Int_t i, kf, nt, iparent;
546     Int_t nc = 0;
547     Float_t p[3];
548     Float_t polar[3]   =   {0,0,0};
549     Float_t origin[3]  =   {0,0,0};
550 //  converts from mm/c to s
551     const Float_t kconv=0.001/2.999792458e8;
552     
553     Int_t np = fParticles->GetEntriesFast();
554     Int_t* pParent = new Int_t[np];
555     for (i=0; i< np; i++) pParent[i] = -1;
556     if (fProcess == kPyJets || fProcess == kPyDirectGamma) {
557         TParticle* jet1 = (TParticle *) fParticles->At(6);
558         TParticle* jet2 = (TParticle *) fParticles->At(7);
559         if (!CheckTrigger(jet1, jet2)) return 0;
560     }
561     
562     for (i = 0; i<np; i++) {
563         Int_t trackIt = 0;
564         TParticle *  iparticle = (TParticle *) fParticles->At(i);
565         kf = CheckPDGCode(iparticle->GetPdgCode());
566         Int_t ks = iparticle->GetStatusCode();
567         Int_t km = iparticle->GetFirstMother();
568         if ((ks == 1  && kf!=0 && KinematicSelection(iparticle, 0)) ||
569             (ks != 1) ||
570             (fProcess == kPyJets && ks == 21 && km == 0 && i>1)) {
571             nc++;
572             if (ks == 1) trackIt = 1;
573             Int_t ipa = iparticle->GetFirstMother()-1;
574             
575             iparent = (ipa > -1) ? pParent[ipa] : -1;
576             
577 //
578 // store track information
579             p[0] = iparticle->Px();
580             p[1] = iparticle->Py();
581             p[2] = iparticle->Pz();
582             origin[0] = fOrigin[0]+iparticle->Vx()/10.;
583             origin[1] = fOrigin[1]+iparticle->Vy()/10.;
584             origin[2] = fOrigin[2]+iparticle->Vz()/10.;
585             Float_t tof=kconv*iparticle->T();
586             PushTrack(fTrackIt*trackIt, iparent, kf, p, origin, polar,
587                      tof, kPPrimary, nt, 1., ks);
588             KeepTrack(nt);
589             pParent[i] = nt;
590         } // select particle
591     } // particle loop 
592
593     if (pParent) delete[] pParent;
594     
595     printf("\n I've put %i particles on the stack \n",nc);
596     return nc;
597 }
598
599
600 void AliGenPythia::FinishRun()
601 {
602 // Print x-section summary
603     fPythia->Pystat(1);
604     fQ  /= fNev;
605     fX1 /= fNev;
606     fX2 /= fNev;    
607     printf("\nTotal number of Pyevnt() calls %d\n", fTrialsRun);
608     printf("\nMean Q, x1, x2: %f %f %f\n", fQ, fX1, fX2);
609     
610
611 }
612
613 void AliGenPythia::AdjustWeights()
614 {
615 // Adjust the weights after generation of all events
616 //
617     TParticle *part;
618     Int_t ntrack=gAlice->GetNtrack();
619     for (Int_t i=0; i<ntrack; i++) {
620         part= gAlice->Particle(i);
621         part->SetWeight(part->GetWeight()*fKineBias);
622     }
623 }
624     
625 void AliGenPythia::SetNuclei(Int_t a1, Int_t a2)
626 {
627 // Treat protons as inside nuclei with mass numbers a1 and a2  
628
629     fAProjectile = a1;
630     fATarget     = a2;
631     fSetNuclei   = kTRUE;
632 }
633
634
635 void AliGenPythia::MakeHeader()
636 {
637 // Builds the event header, to be called after each event
638     if (fHeader) delete fHeader;
639     fHeader = new AliGenPythiaEventHeader("Pythia");
640 //
641 // Event type  
642     ((AliGenPythiaEventHeader*) fHeader)->SetProcessType(fPythia->GetMSTI(1));
643 //
644 // Number of trials
645     ((AliGenPythiaEventHeader*) fHeader)->SetTrials(fTrials);
646 //
647 // Event Vertex 
648     fHeader->SetPrimaryVertex(fVertex);
649 //
650 // Jets that have triggered
651     if (fProcess == kPyJets)
652     {
653         Int_t ntrig, njet;
654         Float_t jets[4][10];
655         GetJets(njet, ntrig, jets);
656         
657         for (Int_t i = 0; i < ntrig; i++) {
658             ((AliGenPythiaEventHeader*) fHeader)->AddJet(jets[0][i], jets[1][i], jets[2][i], 
659                                                         jets[3][i]);
660         }
661     }
662     gAlice->SetGenEventHeader(fHeader);
663 }
664         
665
666 Bool_t AliGenPythia::CheckTrigger(TParticle* jet1, TParticle* jet2)
667 {
668 // Check the kinematic trigger condition
669 //
670     Double_t eta[2];
671     eta[0] = jet1->Eta();
672     eta[1] = jet2->Eta();
673     Double_t phi[2];
674     phi[0] = jet1->Phi();
675     phi[1] = jet2->Phi();
676     Int_t    pdg[2]; 
677     pdg[0] = jet1->GetPdgCode();
678     pdg[1] = jet2->GetPdgCode();    
679     Bool_t   triggered = kFALSE;
680
681     if (fProcess == kPyJets) {
682         Int_t njets = 0;
683         Int_t ntrig = 0;
684         Float_t jets[4][10];
685 //
686 // Use Pythia clustering on parton level to determine jet axis
687 //
688         GetJets(njets, ntrig, jets);
689         
690         if (ntrig) triggered = kTRUE;
691 //
692     } else {
693         Int_t ij = 0;
694         Int_t ig = 1;
695         if (pdg[0] == kGamma) {
696             ij = 1;
697             ig = 0;
698         }
699         //Check eta range first...
700         if ((eta[ij] < fEtaMaxJet   && eta[ij] > fEtaMinJet) &&
701             (eta[ig] < fEtaMaxGamma && eta[ig] > fEtaMinGamma))
702         {
703             //Eta is okay, now check phi range
704             if ((phi[ij] < fPhiMaxJet   && phi[ij] > fPhiMinJet) &&
705                 (phi[ig] < fPhiMaxGamma && phi[ig] > fPhiMinGamma))
706             {
707                 triggered = kTRUE;
708             }
709         }
710     }
711     return triggered;
712 }
713           
714 AliGenPythia& AliGenPythia::operator=(const  AliGenPythia& rhs)
715 {
716 // Assignment operator
717     rhs.Copy(*this);
718     return *this;
719 }
720
721 void  AliGenPythia::LoadEvent()
722 {
723 //
724 // Load event into Pythia Common Block
725 //
726  
727     AliRunLoader* rl = AliRunLoader::GetRunLoader();
728     Int_t npart = (rl->Stack())-> GetNprimary();
729    (fPythia->GetPyjets())->N = npart;
730
731     for (Int_t part = 0; part < npart; part++) {
732         TParticle *MPart = (rl->Stack())->Particle(part);
733         Int_t kf     = MPart->GetPdgCode();
734         Int_t ks     = MPart->GetStatusCode();
735         Float_t px = MPart->Px();
736         Float_t py = MPart->Py();
737         Float_t pz = MPart->Pz();
738         Float_t e  = MPart->Energy();
739         Float_t p  = TMath::Sqrt(px * px + py * py + pz * pz);
740         Float_t m  = TMath::Sqrt(e * e - p * p);
741         
742         
743         (fPythia->GetPyjets())->P[0][part] = px;
744         (fPythia->GetPyjets())->P[1][part] = py;
745         (fPythia->GetPyjets())->P[2][part] = pz;
746         (fPythia->GetPyjets())->P[3][part] = e;
747         (fPythia->GetPyjets())->P[4][part] = m;
748         
749         (fPythia->GetPyjets())->K[1][part] = kf;
750         (fPythia->GetPyjets())->K[0][part] = ks;
751     }
752 }
753
754 void AliGenPythia::RecJetsUA1(Int_t& njets, Float_t jets [4][50])
755 {
756 //
757 //  Calls the Pythia jet finding algorithm to find jets in the current event
758 //
759 //
760 //
761 //  Save jets
762     Int_t n     = fPythia->GetN();
763
764 //
765 //  Run Jet Finder
766     fPythia->Pycell(njets);
767     Int_t i;
768     for (i = 0; i < njets; i++) {
769         Float_t px    = (fPythia->GetPyjets())->P[0][n+i];
770         Float_t py    = (fPythia->GetPyjets())->P[1][n+i];
771         Float_t pz    = (fPythia->GetPyjets())->P[2][n+i];
772         Float_t e     = (fPythia->GetPyjets())->P[3][n+i];
773
774         jets[0][i] = px;
775         jets[1][i] = py;
776         jets[2][i] = pz;
777         jets[3][i] = e;
778     }
779 }
780
781
782
783 void  AliGenPythia::GetJets(Int_t& nJets, Int_t& nJetsTrig, Float_t jets[4][10])
784 {
785 //
786 //  Calls the Pythia clustering algorithm to find jets in the current event
787 //
788     Int_t n     = fPythia->GetN();
789     nJets       = 0;
790     nJetsTrig   = 0;
791     if (fJetReconstruction == kCluster) {
792 //
793 //  Configure cluster algorithm
794 //    
795         fPythia->SetPARU(43, 2.);
796         fPythia->SetMSTU(41, 1);
797 //
798 //  Call cluster algorithm
799 //    
800         fPythia->Pyclus(nJets);
801 //
802 //  Loading jets from common block
803 //
804     } else {
805
806 //
807 //  Run Jet Finder
808         fPythia->Pycell(nJets);
809     }
810
811     Int_t i;
812     for (i = 0; i < nJets; i++) {
813         Float_t px    = (fPythia->GetPyjets())->P[0][n+i];
814         Float_t py    = (fPythia->GetPyjets())->P[1][n+i];
815         Float_t pz    = (fPythia->GetPyjets())->P[2][n+i];
816         Float_t e     = (fPythia->GetPyjets())->P[3][n+i];
817         Float_t pt    = TMath::Sqrt(px * px + py * py);
818         Float_t phi   = TMath::ATan2(py,px);
819         Float_t theta = TMath::ATan2(pt,pz);
820         Float_t et    = e * TMath::Sin(theta);
821         Float_t eta   = -TMath::Log(TMath::Tan(theta / 2.));
822
823         if (
824             eta > fEtaMinJet && eta < fEtaMaxJet && 
825             phi > fPhiMinJet && phi < fPhiMaxJet &&
826             et  > fEtMinJet  && et  < fEtMaxJet     
827             ) 
828         {
829             jets[0][nJetsTrig] = px;
830             jets[1][nJetsTrig] = py;
831             jets[2][nJetsTrig] = pz;
832             jets[3][nJetsTrig] = e;
833             nJetsTrig++;
834             
835         } else {
836 //          printf("\n........-Jet #%d: %10.3f %10.3f %10.3f %10.3f \n", i, pt, et, eta, phi * kRaddeg);
837         }
838     }
839 }
840
841
842 #ifdef never
843 void AliGenPythia::Streamer(TBuffer &R__b)
844 {
845    // Stream an object of class AliGenPythia.
846
847    if (R__b.IsReading()) {
848       Version_t R__v = R__b.ReadVersion(); if (R__v) { }
849       AliGenerator::Streamer(R__b);
850       R__b >> (Int_t&)fProcess;
851       R__b >> (Int_t&)fStrucFunc;
852       R__b >> (Int_t&)fForceDecay;
853       R__b >> fEnergyCMS;
854       R__b >> fKineBias;
855       R__b >> fTrials;
856       fParentSelect.Streamer(R__b);
857       fChildSelect.Streamer(R__b);
858       R__b >> fXsection;
859 //      (AliPythia::Instance())->Streamer(R__b);
860       R__b >> fPtHardMin;
861       R__b >> fPtHardMax;
862 //      if (fDecayer) fDecayer->Streamer(R__b);
863    } else {
864       R__b.WriteVersion(AliGenPythia::IsA());
865       AliGenerator::Streamer(R__b);
866       R__b << (Int_t)fProcess;
867       R__b << (Int_t)fStrucFunc;
868       R__b << (Int_t)fForceDecay;
869       R__b << fEnergyCMS;
870       R__b << fKineBias;
871       R__b << fTrials;
872       fParentSelect.Streamer(R__b);
873       fChildSelect.Streamer(R__b);
874       R__b << fXsection;
875 //      R__b << fPythia;
876       R__b << fPtHardMin;
877       R__b << fPtHardMax;
878       //     fDecayer->Streamer(R__b);
879    }
880 }
881 #endif
882