]> git.uio.no Git - u/mrichter/AliRoot.git/blame_incremental - EVGEN/AliGenReaderCwn.cxx
Removing remnants of the old Makefile
[u/mrichter/AliRoot.git] / EVGEN / AliGenReaderCwn.cxx
... / ...
CommitLineData
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
17/*
18$Log$
19Revision 1.3 2002/10/21 09:20:51 alibrary
20Introduce Riostream.h and remove unused variables
21
22Revision 1.2 2001/11/12 14:31:00 morsch
23Memory leaks fixed. (M. Bondila)
24
25Revision 1.1 2001/11/09 09:10:46 morsch
26Realisation of AliGenReader that reads the old cwn event format.
27
28*/
29
30// Read the old ALICE event format based on CW-ntuples
31// http://consult.cern.ch/alice/Internal_Notes/1995/32/abstract
32// .cwn file have to be converted to .root using h2root
33// Use SetFileName(file) to read from "file"
34// Author: andreas.morsch@cern.ch
35
36#include <TFile.h>
37#include <TTree.h>
38#include <TParticle.h>
39
40#include "AliGenReaderCwn.h"
41#include "AliMC.h"
42ClassImp(AliGenReaderCwn);
43
44
45AliGenReaderCwn::AliGenReaderCwn()
46{
47// Default constructor
48 fNcurrent = 0;
49 fTreeNtuple = 0;
50}
51
52AliGenReaderCwn::~AliGenReaderCwn()
53{
54 delete fTreeNtuple;
55}
56
57void AliGenReaderCwn::Init()
58{
59//
60// reset the existing file environment and open a new root file if
61// the pointer to the Fluka tree is null
62
63 TFile *pFile=0;
64 if (!pFile) {
65 pFile = new TFile(fFileName);
66 pFile->cd();
67 printf("\n I have opened %s file \n", fFileName);
68 }
69// get the tree address in the Fluka boundary source file
70 fTreeNtuple = (TTree*)gDirectory->Get("h888");
71
72 TTree *h2=fTreeNtuple;
73//Set branch addresses
74 h2->SetBranchAddress("Nihead",&fNihead);
75 h2->SetBranchAddress("Ihead",fIhead);
76 h2->SetBranchAddress("Nrhead",&fNrhead);
77 h2->SetBranchAddress("Rhead",fRhead);
78 h2->SetBranchAddress("Idpart",&fIdpart);
79 h2->SetBranchAddress("Theta",&fTheta);
80 h2->SetBranchAddress("Phi",&fPhi);
81 h2->SetBranchAddress("P",&fP);
82 h2->SetBranchAddress("E",&fE);
83}
84
85Int_t AliGenReaderCwn::NextEvent()
86{
87// Read the next event
88 Int_t nTracks;
89 fNparticle = 0;
90 TFile* pFile = fTreeNtuple->GetCurrentFile();
91 pFile->cd();
92
93 Int_t nentries = (Int_t) fTreeNtuple->GetEntries();
94 if (fNcurrent < nentries) {
95 fNcurrent++;
96
97 Int_t i5=fIhead[4];
98 Int_t i6=fIhead[5];
99 if (i5==0) {
100 printf("\n This should never happen !\n");
101 nTracks = 0;
102 } else {
103 printf("\n Next event contains %d tracks! \n", i6);
104 nTracks = i6;
105 }
106 fNparticleMax = nTracks;
107 return nTracks;
108 }
109
110 return 0;
111}
112
113TParticle* AliGenReaderCwn::NextParticle()
114{
115//
116//
117 Float_t prwn;
118 Float_t p[4];
119// Read the next particle
120 if (fCode == kGEANT3) fIdpart=gMC->PDGFromId(fIdpart);
121 Double_t amass = TDatabasePDG::Instance()->GetParticle(fIdpart)->Mass();
122 if(fE<=amass) {
123 Warning("Generate","Particle %d E = %f mass = %f %f %f \n",
124 fIdpart,fE,amass, fPhi, fTheta);
125 prwn=0;
126 } else {
127 prwn=sqrt((fE+amass)*(fE-amass));
128 }
129
130 fTheta *= TMath::Pi()/180.;
131 fPhi = (fPhi-180)*TMath::Pi()/180.;
132 p[0] = prwn*TMath::Sin(fTheta)*TMath::Cos(fPhi);
133 p[1] = prwn*TMath::Sin(fTheta)*TMath::Sin(fPhi);
134 p[2] = prwn*TMath::Cos(fTheta);
135 p[3] = fE;
136 TParticle* particle = new TParticle(fIdpart, 0, -1, -1, -1, -1, p[0], p[1], p[2], p[3], 0., 0., 0., 0.);
137 fNcurrent++;
138 fNparticle++;
139 return particle;
140}
141
142
143
144AliGenReaderCwn& AliGenReaderCwn::operator=(const AliGenReaderCwn& rhs)
145{
146// Assignment operator
147 return *this;
148}
149
150
151