EVGEN/AliGenBox.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 /*
  17 $Log$
  18 Revision 1.7  2002/02/08 16:50:50  morsch
  19 Add name and title in constructor.
  20
  21 Revision 1.6  2001/07/27 17:09:35  morsch
  22 Use local SetTrack, KeepTrack and SetHighWaterMark methods
  23 to delegate either to local stack or to stack owned by AliRun.
  24 (Piotr Skowronski, A.M.)
  25
  26 Revision 1.5  2000/12/21 16:24:06  morsch
  27 Coding convention clean-up
  28
  29 Revision 1.4  2000/11/30 07:12:49  alibrary
  30 Introducing new Rndm and QA classes
  31
  32 Revision 1.3  2000/10/02 21:28:06  fca
  33 Removal of useless dependecies via forward declarations
  34
  35 Revision 1.2  2000/07/11 18:24:55  fca
  36 Coding convention corrections + few minor bug fixes
  37
  38 Revision 1.1  2000/06/09 20:22:58  morsch
  39 Same class as previously in AliSimpleGen.cxx
  40 All coding rule violations except RS3 corrected (AM)
  41
  42 */
  43
  44
  45
  46 // Generator for particles in a preset
  47 // kinematic range (flat distribution)
  48 // Note that for a given theta pt and p are not independent
  49 // Range for only one variable (pt or p) should be given.
  50 //
  51 // Comments and suggestions: andreas.morsch@cern.ch
  52 //
  53 //Begin_Html
  54 /*
  55 <img src="picts/AliGeneratorClass.gif">
  56 </pre>
  57 <br clear=left>
  58 <font size=+2 color=red>
  59 <p>The responsible person for this module is
  60 <a href="mailto:andreas.morsch@cern.ch">Andreas Morsch</a>.
  61 </font>
  62 <pre>
  63 */
  64 //End_Html
  65 //                                                               //
  66 ///////////////////////////////////////////////////////////////////
  67
  68 #include "TPDGCode.h"
  69
  70 #include "AliConst.h"
  71 #include "AliGenBox.h"
  72 #include "AliRun.h"
  73
  74 ClassImp(AliGenBox)
  75
  76 //_____________________________________________________________________________
  77 AliGenBox::AliGenBox()
  78     :AliGenerator()
  79 {
  80   //
  81   // Default constructor
  82   //
  83   fIpart=0;
  84 }
  85
  86 //_____________________________________________________________________________
  87 AliGenBox::AliGenBox(Int_t npart)
  88   :AliGenerator(npart)
  89 {
  90   //
  91   // Standard constructor
  92   //
  93   fName  = "Box";
  94   fTitle = "Box particle generator";
  95   // Generate Proton by default
  96   fIpart=kProton;
  97 }
  98
  99 //_____________________________________________________________________________
 100
 101 void AliGenBox::Generate()
 102 {
 103   //
 104   // Generate one trigger
 105   //
 106
 107     Float_t polar[3]= {0,0,0};
 108   //
 109     Float_t origin[3];
 110     Float_t p[3];
 111     Int_t i, j, nt;
 112     Double_t pmom, theta, phi, pt;
 113     //
 114     Float_t random[6];
 115   //
 116     for (j=0;j<3;j++) origin[j]=fOrigin[j];
 117     if(fVertexSmear==kPerEvent) {
 118         Rndm(random,6);
 119         for (j=0;j<3;j++) {
 120             origin[j]+=fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())*
 121                 TMath::Sqrt(-2*TMath::Log(random[2*j+1]));
 122         }
 123     }
 124     for(i=0;i<fNpart;i++) {
 125         Rndm(random,3);
 126         theta=fThetaMin+random[0]*(fThetaMax-fThetaMin);
 127         if(TestBit(kMomentumRange)) {
 128             pmom=fPMin+random[1]*(fPMax-fPMin);
 129             pt=pmom*TMath::Sin(theta);
 130         } else {
 131
 132             pt=fPtMin+random[1]*(fPtMax-fPtMin);
 133             pmom=pt/TMath::Sin(theta);
 134         }
 135         phi=fPhiMin+random[2]*(fPhiMax-fPhiMin);
 136         p[0] = pt*TMath::Cos(phi);
 137         p[1] = pt*TMath::Sin(phi);
 138         p[2] = pmom*TMath::Cos(theta);
 139
 140         if(fVertexSmear==kPerTrack) {
 141             Rndm(random,6);
 142             for (j=0;j<3;j++) {
 143                 origin[j]=fOrigin[j]+fOsigma[j]*TMath::Cos(2*random[2*j]*TMath::Pi())*
 144                     TMath::Sqrt(-2*TMath::Log(random[2*j+1]));
 145             }
 146         }
 147         SetTrack(fTrackIt,-1,fIpart,p,origin,polar,0,kPPrimary,nt);
 148     }
 149 }
 150
 151 //_____________________________________________________________________________
 152
 153 void AliGenBox::Init()
 154 {
 155 // Initialisation, check consistency of selected ranges
 156   if(TestBit(kPtRange)&&TestBit(kMomentumRange))
 157     Fatal("Init","You should not set the momentum range and the pt range!\n");
 158   if((!TestBit(kPtRange))&&(!TestBit(kMomentumRange)))
 159     Fatal("Init","You should set either the momentum or the pt range!\n");
 160 }
 161