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