TuneA settings removed from kPyJets
[u/mrichter/AliRoot.git] / EVGEN / AliGenReaderCwn.cxx
CommitLineData
f24650c5 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
88cb7938 17/* $Id$ */
f24650c5 18
19// Read the old ALICE event format based on CW-ntuples
20// http://consult.cern.ch/alice/Internal_Notes/1995/32/abstract
21// .cwn file have to be converted to .root using h2root
22// Use SetFileName(file) to read from "file"
23// Author: andreas.morsch@cern.ch
24
25#include <TFile.h>
f24650c5 26#include <TParticle.h>
c180f65d 27#include <TDatabasePDG.h>
88cb7938 28#include <TTree.h>
acc86a24 29#include <TVirtualMC.h>
f24650c5 30
31#include "AliGenReaderCwn.h"
88cb7938 32
706938e6 33ClassImp(AliGenReaderCwn)
f24650c5 34
1c56e311 35AliGenReaderCwn::AliGenReaderCwn():
36 fNcurrent(0),
37 fNparticle(0),
38 fNparticleMax(0),
39 fTreeNtuple(0),
40 fNihead(0),
41 fNrhead(0),
42 fIdpart(0),
43 fTheta(0.),
44 fPhi(0.),
45 fP(0.),
46 fE(0.)
f24650c5 47{
48// Default constructor
5eebf646 49 Int_t i;
50 for (i = 0; i < 6; i++) fRhead[i] = 0.;
51 for (i = 0; i < 12; i++) fIhead[i] = 0;
f24650c5 52}
53
5eebf646 54
1c56e311 55AliGenReaderCwn::AliGenReaderCwn(const AliGenReaderCwn &reader):
56 AliGenReader(reader),
57 fNcurrent(0),
58 fNparticle(0),
59 fNparticleMax(0),
60 fTreeNtuple(0),
61 fNihead(0),
62 fNrhead(0),
63 fIdpart(0),
64 fTheta(0.),
65 fPhi(0.),
66 fP(0.),
67 fE(0.)
68{
69 // Copy constructor
5eebf646 70 Int_t i;
71 for (i = 0; i < 6; i++) fRhead[i] = 0.;
72 for (i = 0; i < 12; i++) fIhead[i] = 0;
1c56e311 73 reader.Copy(*this);
74}
75
76
ae872676 77AliGenReaderCwn::~AliGenReaderCwn()
78{
79 delete fTreeNtuple;
80}
81
f24650c5 82void AliGenReaderCwn::Init()
83{
84//
85// reset the existing file environment and open a new root file if
86// the pointer to the Fluka tree is null
87
88 TFile *pFile=0;
89 if (!pFile) {
90 pFile = new TFile(fFileName);
91 pFile->cd();
92 printf("\n I have opened %s file \n", fFileName);
93 }
94// get the tree address in the Fluka boundary source file
95 fTreeNtuple = (TTree*)gDirectory->Get("h888");
96
97 TTree *h2=fTreeNtuple;
98//Set branch addresses
99 h2->SetBranchAddress("Nihead",&fNihead);
100 h2->SetBranchAddress("Ihead",fIhead);
101 h2->SetBranchAddress("Nrhead",&fNrhead);
102 h2->SetBranchAddress("Rhead",fRhead);
103 h2->SetBranchAddress("Idpart",&fIdpart);
104 h2->SetBranchAddress("Theta",&fTheta);
105 h2->SetBranchAddress("Phi",&fPhi);
106 h2->SetBranchAddress("P",&fP);
107 h2->SetBranchAddress("E",&fE);
108}
109
110Int_t AliGenReaderCwn::NextEvent()
111{
112// Read the next event
113 Int_t nTracks;
114 fNparticle = 0;
115 TFile* pFile = fTreeNtuple->GetCurrentFile();
116 pFile->cd();
117
118 Int_t nentries = (Int_t) fTreeNtuple->GetEntries();
119 if (fNcurrent < nentries) {
f24650c5 120 fNcurrent++;
121
122 Int_t i5=fIhead[4];
123 Int_t i6=fIhead[5];
124 if (i5==0) {
125 printf("\n This should never happen !\n");
126 nTracks = 0;
127 } else {
128 printf("\n Next event contains %d tracks! \n", i6);
129 nTracks = i6;
130 }
131 fNparticleMax = nTracks;
132 return nTracks;
f24650c5 133 }
4facea7f 134
f24650c5 135 return 0;
136}
137
138TParticle* AliGenReaderCwn::NextParticle()
139{
4a33c50d 140// Read next particle
f24650c5 141//
142 Float_t prwn;
143 Float_t p[4];
144// Read the next particle
2942f542 145 if (fCode == kGEANT3) fIdpart=TVirtualMC::GetMC()->PDGFromId(fIdpart);
f24650c5 146 Double_t amass = TDatabasePDG::Instance()->GetParticle(fIdpart)->Mass();
147 if(fE<=amass) {
148 Warning("Generate","Particle %d E = %f mass = %f %f %f \n",
149 fIdpart,fE,amass, fPhi, fTheta);
150 prwn=0;
151 } else {
152 prwn=sqrt((fE+amass)*(fE-amass));
153 }
154
155 fTheta *= TMath::Pi()/180.;
156 fPhi = (fPhi-180)*TMath::Pi()/180.;
157 p[0] = prwn*TMath::Sin(fTheta)*TMath::Cos(fPhi);
158 p[1] = prwn*TMath::Sin(fTheta)*TMath::Sin(fPhi);
159 p[2] = prwn*TMath::Cos(fTheta);
160 p[3] = fE;
159ec319 161 TParticle* particle = new TParticle(fIdpart, 0, -1, -1, -1, -1, p[0], p[1], p[2], p[3], 0., 0., 0., 0.);
f24650c5 162 fNcurrent++;
163 fNparticle++;
164 return particle;
165}
166
167
168
ae872676 169AliGenReaderCwn& AliGenReaderCwn::operator=(const AliGenReaderCwn& rhs)
170{
171// Assignment operator
198bb1c7 172 rhs.Copy(*this);
ae872676 173 return *this;
174}
f24650c5 175
dc1d768c 176void AliGenReaderCwn::Copy(TObject&) const
198bb1c7 177{
178 //
179 // Copy
180 //
181 Fatal("Copy","Not implemented!\n");
182}
183
184
f24650c5 185
186