Code causing warning messages corrected.
[u/mrichter/AliRoot.git] / EVGEN / AliGenHalo.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 // Read background particles from a boundary source
19 // Very specialized generator to simulate background from beam halo.
20 // The input file is a text file specially prepared 
21 // for this purpose.
22 // Author: andreas.morsch@cern.ch
23
24 #include <stdlib.h>
25
26 #include <TDatabasePDG.h>
27 #include <TPDGCode.h>
28
29 #include "AliGenHalo.h"
30 #include "AliRun.h"
31
32 ClassImp(AliGenHalo)
33
34 AliGenHalo::AliGenHalo()
35     :AliGenerator(-1)
36 {
37 // Constructor
38     fName="Halo";
39     fTitle="Halo from LHC Tunnel";
40 //
41 //  Read all particles
42     fNpart=-1;
43     fp=0;
44 }
45
46 AliGenHalo::AliGenHalo(Int_t npart)
47     :AliGenerator(npart)
48 {
49 // Constructor
50     fName="Halo";
51     fTitle="Halo from LHC Tunnel";
52 //
53 //  Read all particles
54     fNpart=-1;
55     fp=0;
56 }
57
58 AliGenHalo::AliGenHalo(const AliGenHalo & Halo)
59     :AliGenerator(Halo)
60 {
61 // Copy constructor
62     Halo.Copy(*this);
63 }
64
65
66 //____________________________________________________________
67 AliGenHalo::~AliGenHalo()
68 {
69 // Destructor
70 }
71
72 //____________________________________________________________
73 void AliGenHalo::Init() 
74 {
75 // Initialisation
76 }
77
78 //____________________________________________________________
79 void AliGenHalo::Generate()
80 {
81 // Generate from input file
82     FILE *fp = fopen(fFileName,"r");
83     if (fp) {
84         printf("\n File %s opened for reading ! \n ", (char*) &fFileName);
85     } else {
86         printf("\n Opening of file %s failed ! \n ",  (char*) &fFileName);
87     }
88 //
89 // MARS particle codes
90   const Int_t kmars[12]={0,kProton,kNeutron,kPiPlus,kPiMinus,kKPlus,kKMinus,
91                          kMuonPlus,kMuonMinus,kGamma,kElectron,kPositron};
92  
93   Float_t polar[3]= {0,0,0};
94   Float_t origin[3];
95   Float_t p[3], p0;
96   Float_t ekin, wgt, tx, ty, tz, txy;
97   Float_t amass;
98   //
99   Int_t ipart, ncols, nt;
100   
101   Int_t nread=0;
102   origin[2]=2650;
103   
104   while(1) {
105       ncols = fscanf(fp,"%i %f %f %f %f %f %f",
106                      &ipart, &ekin, &wgt, 
107                      &origin[0], &origin[1],
108                      &tx, &ty);
109       if (ncols < 0) break;
110       nread++;
111       if (fNpart !=-1 && nread > fNpart) break;
112       ipart = kmars[ipart];
113       amass = TDatabasePDG::Instance()->GetParticle(ipart)->Mass();
114       p0=sqrt(ekin*ekin + 2.*amass);
115       
116       txy=TMath::Sqrt(tx*tx+ty*ty);
117       if (txy == 1.) {
118           tz=0;
119       } else {
120           tz=-TMath::Sqrt(1.-txy);
121       }
122       p[0]=p0*tx;
123       p[1]=p0*ty;
124       p[2]=p0*tz;
125       fParentWeight=wgt;
126       PushTrack(fTrackIt,-1,ipart,p,origin,polar,0,kPNoProcess,nt,fParentWeight);
127 //    PushTrack(fTrackIt,-1,ipart,p,origin,polar,0,"Halo+",nt,fParentWeight);
128       origin[2]=-origin[2];
129       p[2]=-p[2];
130       PushTrack(fTrackIt,-1,ipart,p,origin,polar,0,kPNoProcess,nt,fParentWeight);
131 //    PushTrack(fTrackIt,-1,ipart,p,origin,polar,0,"Halo-",nt,fParentWeight);
132       origin[2]=-origin[2];
133       p[2]=-p[2];
134   }
135 }
136  
137
138 AliGenHalo& AliGenHalo::operator=(const  AliGenHalo& rhs)
139 {
140 // Assignment operator
141     rhs.Copy(*this);
142     return *this;
143 }
144
145
146 void AliGenHalo::Copy(AliGenHalo&) const
147 {
148     //
149     // Copy 
150   //
151     Fatal("Copy","Not implemented!\n");
152 }
153
154
155
156
157
158
159
160