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