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