TMEVSIM/AliGenMevSim.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.3  2001/07/27 17:24:28  morsch
  19 Some bugs corrected. ( Piotr Skowronski)
  20
  21 Revision 1.2  2001/03/28 07:31:48  hristov
  22 Loop variables declared only once (HP,Sun)
  23
  24 Revision 1.1  2001/03/24 10:04:44  morsch
  25 MevSim interfaced through AliGenerator, first commit (Sylwester Radomski et al.)
  26 //Piotr Skowronski Line 104: fThetaMin-->fThetaMax
  27 */
  28
  29 //
  30 // Wrapper for MEVSIM generator.
  31 // It is using TMevSim to comunicate with fortarn code
  32 //
  33 //
  34 // Sylwester Radomski <radomski@if.pw.edu.pl>
  35 //
  36
  37 //#include "TSystem.h"
  38 //#include "TUnixSystem.h"
  39 #include "TParticle.h"
  40 #include "TMevSim.h"
  41
  42 #include "AliGenMevSim.h"
  43 #include "AliRun.h"
  44
  45
  46 ClassImp(AliGenMevSim)
  47
  48 //____________________________________________________________________________
  49 AliGenMevSim::AliGenMevSim() : AliGenerator(-1)
  50 {
  51   //
  52   // Standard creator
  53   //
  54
  55   fConfig = new AliMevSimConfig();
  56   fMevSim = new TMevSim();
  57   sRandom = fRandom;
  58
  59 }
  60 //____________________________________________________________________________
  61 AliGenMevSim::AliGenMevSim(AliMevSimConfig *config): AliGenerator(-1) {
  62
  63   fConfig = config;
  64   fMevSim = new TMevSim();
  65   sRandom = fRandom;
  66 }
  67
  68 //____________________________________________________________________________
  69 AliGenMevSim::~AliGenMevSim()
  70 {
  71   //
  72   // Standard destructor
  73   //
  74   if (fMevSim) delete fMevSim;
  75 }
  76 //____________________________________________________________________________
  77 void AliGenMevSim::SetConfig(AliMevSimConfig *config) {
  78
  79   fConfig = config;
  80 }
  81 //____________________________________________________________________________
  82 void AliGenMevSim::AddParticleType(AliMevSimParticle *type) {
  83
  84   fMevSim->AddPartTypeParams((TMevSimPartTypeParams*)type);
  85 }
  86 //____________________________________________________________________________
  87 void AliGenMevSim::Init()
  88 {
  89   //
  90   // Generator initialisation method
  91   //
  92
  93   // fill data from AliMevSimConfig;
  94
  95   TMevSim *mevsim = fMevSim;
  96
  97   // geometry & momentum cut
  98
  99   if (TestBit(kPtRange))  mevsim->SetPtCutRange(fPtMin, fPtMax);
 100
 101   if (TestBit(kPhiRange)) // from radians to 0 - 360 deg
 102     mevsim->SetPhiCutRange( fPhiMin * 180 / TMath::Pi() , fPhiMax * 180 / TMath::Pi() );
 103
 104   if (TestBit(kThetaRange)) // from theta to eta
 105   {
 106     mevsim->SetEtaCutRange( -TMath::Log( TMath::Tan(fThetaMax/2)) , -TMath::Log( TMath::Tan(fThetaMin/2)) );
 107   }
 108
 109   // mevsim specyfic parameters
 110
 111   mevsim->SetModelType(fConfig->fModelType);
 112   mevsim->SetReacPlaneCntrl(fConfig->fReacPlaneCntrl);
 113   mevsim->SetPsiRParams(fConfig->fPsiRMean, fConfig->fPsiRStDev);
 114   mevsim->SetMultFacParams(fConfig->fMultFacMean, fConfig->fMultFacStDev);
 115
 116   // mevsim initialisation
 117
 118   mevsim->Initialize();
 119 }
 120
 121 //____________________________________________________________________________
 122 void AliGenMevSim::Generate()
 123 {
 124   //
 125   // Read the formatted output file and load the particles
 126   // Temporary solution
 127   //
 128
 129   Int_t i;
 130
 131   PDG_t pdg;
 132   Float_t orgin[3] = {0,0,0};
 133   Float_t polar[3] = {0,0,0};
 134   Float_t p[3] = {1,1,1};
 135   Float_t time = 0;
 136
 137   const Int_t parent = -1;
 138   Int_t id;
 139
 140   // vertexing
 141
 142   VertexInternal();
 143
 144   orgin[0] = fVertex[0];
 145   orgin[1] = fVertex[1];
 146   orgin[2] = fVertex[2];
 147
 148   cout << "Vertex ";
 149   for (i =0; i<3; i++)
 150     cout << orgin[i] << "\t";
 151   cout << endl;
 152
 153   Int_t nParticles = 0;
 154
 155   TClonesArray *particles = new TClonesArray("TParticle");
 156   TParticle *particle;
 157
 158   fMevSim->GenerateEvent();
 159   fNpart= fMevSim->ImportParticles(particles,"");
 160
 161   cout << "Found " << fNpart << " particles ..." << endl;
 162   nParticles = fNpart;
 163
 164   for (i=0; i<nParticles; i++) {
 165
 166     particle = (TParticle*) (*particles)[i];
 167
 168     pdg = (PDG_t) particle->GetPdgCode();
 169     p[0] = particle->Px();
 170     p[1] = particle->Py();
 171     p[2] = particle->Pz();
 172
 173     SetTrack(fTrackIt, parent, pdg, p, orgin, polar, time, kPPrimary, id);
 174
 175   }
 176
 177   particles->Clear();
 178   if (particles) delete particles;
 179 }
 180 //____________________________________________________________________________
 181 //____________________________________________________________________________
 182
 183
 184 #ifndef WIN32
 185 # define ran ran_
 186 # define type_of_call
 187 #else
 188 # define ran RAN
 189 # define type_of_call _stdcall
 190 #endif
 191
 192 extern "C" Float_t type_of_call ran(Int_t &)
 193 {
 194   return sRandom->Rndm();
 195 }
 196
 197 //____________________________________________________________________________