Coding Violations Removed
[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 "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->GetModelType());
102   Int_t ctrl; Float_t psiRMean = 0; Float_t psiRStDev = 0;
103   fConfig->GetRectPlane(ctrl,psiRMean,psiRStDev);
104   mevsim->SetReacPlaneCntrl(ctrl);
105   mevsim->SetPsiRParams(psiRMean, psiRStDev);
106   Float_t mean; Float_t stDev;
107   fConfig->GetMultFac(mean,stDev);
108   mevsim->SetMultFacParams(mean,stDev);
109   // mevsim initialisation
110
111   mevsim->Initialize();
112 }
113
114 //____________________________________________________________________________
115 void AliGenMevSim::Generate() 
116 {
117   //
118   // Read the formatted output file and load the particles
119   // Temporary solution
120   //
121
122   Int_t i;
123
124   PDG_t pdg;
125   Float_t orgin[3] = {0,0,0};
126   Float_t polar[3] = {0,0,0};
127   Float_t p[3] = {1,1,1};
128   Float_t time = 0;
129   
130   const Int_t parent = -1;
131   Int_t id;
132
133   // vertexing 
134
135   VertexInternal();
136
137   orgin[0] = fVertex[0];
138   orgin[1] = fVertex[1];
139   orgin[2] = fVertex[2];
140
141   cout << "Vertex ";
142   for (i =0; i<3; i++)
143     cout << orgin[i] << "\t";
144   cout << endl;
145
146   Int_t nParticles = 0;
147
148   TClonesArray *particles = new TClonesArray("TParticle");
149   TParticle *particle;
150
151   fMevSim->GenerateEvent();
152   fNpart= fMevSim->ImportParticles(particles,"");
153
154   cout << "Found " << fNpart << " particles ..." << endl;
155   nParticles = fNpart;
156
157   for (i=0; i<nParticles; i++) {
158     
159     particle = (TParticle*) (*particles)[i];
160
161     pdg = (PDG_t) particle->GetPdgCode();
162     p[0] = particle->Px();
163     p[1] = particle->Py();
164     p[2] = particle->Pz();
165     
166     PushTrack(fTrackIt, parent, pdg, p, orgin, polar, time, kPPrimary, id);
167
168   }  
169  
170   particles->Clear();
171   if (particles) delete particles;
172 }
173 //____________________________________________________________________________
174 //____________________________________________________________________________
175
176
177 #ifndef WIN32
178 # define ran ran_
179 # define type_of_call
180 #else
181 # define ran RAN
182 # define type_of_call _stdcall
183 #endif
184
185 extern "C" Float_t type_of_call ran(Int_t &)
186 {
187   return sRandom->Rndm(); 
188 }
189
190 //____________________________________________________________________________