Clean-up
[u/mrichter/AliRoot.git] / EVGEN / AliGenReaderEcalJets.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 // Realisation of AliGenReader to be used with AliGenExtFile
19 // It reads Pythia Jet events from a ntuple like event structure.
20 // The event format is defined in Init()
21 // NextEvent() is used to loop over events and NextParticle() to loop over particles.  
22 // Author: andreas.morsch@cern.ch
23 //
24 #include <TFile.h>
25 #include <TParticle.h>
26 #include <TDatabasePDG.h>
27 #include <TTree.h>
28
29 #include "AliGenReaderEcalJets.h"
30
31 ClassImp(AliGenReaderEcalJets)
32
33
34 AliGenReaderEcalJets::AliGenReaderEcalJets():
35     fNcurrent(0),
36     fNparticle(0),
37     fNev(0),
38     fNpart(0),
39     fNjet(0),     
40     fNsjet(0),
41     fNpjet(0),
42     fTreeNtuple(0) 
43 {
44 // Default constructor
45 }
46
47  AliGenReaderEcalJets::AliGenReaderEcalJets(const AliGenReaderEcalJets &reader)
48      :AliGenReader(reader),
49       fNcurrent(0),
50       fNparticle(0),
51       fNev(0),
52       fNpart(0),
53       fNjet(0),     
54       fNsjet(0),
55       fNpjet(0),
56       fTreeNtuple(0) 
57 {
58     // Copy Constructor
59     reader.Copy(*this);
60 }
61
62 void AliGenReaderEcalJets::Init() 
63 {
64 //
65 // reset the existing file environment and open a new root file if
66 // the pointer to the Fluka tree is null
67     
68     TFile *pFile=0;
69     if (!pFile) {
70         pFile = new TFile(fFileName);
71         pFile->cd();
72         printf("\n I have opened %s file \n", fFileName);
73     }
74 // get the tree address in the Fluka boundary source file
75     fTreeNtuple = (TTree*)gDirectory->Get("h1");
76     TTree *h1=fTreeNtuple;
77     h1->SetMakeClass(1);
78 //Set branch addresses
79     h1->SetBranchAddress("nev",   &fNev);
80     h1->SetBranchAddress("x",      fX);
81     h1->SetBranchAddress("xtyp",   fXtyp);
82     h1->SetBranchAddress("npart", &fNpart);
83     h1->SetBranchAddress("xpt",    fXpt);
84     h1->SetBranchAddress("xeta",   fXeta);
85     h1->SetBranchAddress("xphi",   fXphi);
86     h1->SetBranchAddress("xid",    fXid);
87     h1->SetBranchAddress("njet",  &fNjet);
88     h1->SetBranchAddress("jet",    fJet);
89     h1->SetBranchAddress("jeta",   fJeta);
90     h1->SetBranchAddress("jphi",   fJphi);
91     h1->SetBranchAddress("nsjet", &fNsjet);
92     h1->SetBranchAddress("jset",   fJset);
93     h1->SetBranchAddress("jseta",  fJseta);
94     h1->SetBranchAddress("jsphi",  fJsphi);
95     h1->SetBranchAddress("npjet", &fNpjet);
96     h1->SetBranchAddress("jpet",   fJpet);
97     h1->SetBranchAddress("jpeta",  fJpeta);
98     h1->SetBranchAddress("jpphi",  fJpphi);
99 }
100
101 Int_t AliGenReaderEcalJets::NextEvent() 
102 {
103 // Read the next event  
104     Int_t nTracks=0, nread=0;
105     
106     TFile* pFile = fTreeNtuple->GetCurrentFile();
107     pFile->cd();
108
109     Int_t nentries = (Int_t) fTreeNtuple->GetEntries();
110     if (fNcurrent < nentries) {
111         Int_t nb = (Int_t)fTreeNtuple->GetEvent(fNcurrent);
112         nread += nb;
113         fNcurrent++;
114         printf("\n Next event contains %d tracks! \n", fNpjet);
115         nTracks    = fNpjet;
116         fNparticle = 0;
117         return nTracks;
118     }
119     return 0;
120 }
121
122 TParticle* AliGenReaderEcalJets::NextParticle() 
123 {
124 // Read the next particle
125
126     Float_t p[4];
127     Int_t    ipart  = fXid[fNparticle];
128     Float_t  pt     = fXpt[fNparticle];
129     Float_t  eta    = fXeta[fNparticle];
130     Float_t  phi    = fXphi[fNparticle];
131     Float_t  theta  = 2.*TMath::ATan(TMath::Exp(-eta));
132     Double_t amass  = TDatabasePDG::Instance()->GetParticle(ipart)->Mass();
133
134     p[0] = pt*TMath::Sin(phi);
135     p[1] = pt*TMath::Cos(phi);      
136     p[2] = pt/TMath::Cos(theta);
137     p[3] = TMath::Sqrt(pt*pt+p[2]*p[2]+amass*amass);
138     
139
140     TParticle* particle = 
141         new TParticle(ipart, 0, -1, -1, -1, -1, p[0], p[1], p[2], p[3], 
142                       0., 0., 0., 0.);
143     fNparticle++;
144     return particle;
145 }
146
147
148
149 AliGenReaderEcalJets& AliGenReaderEcalJets::operator=(const  AliGenReaderEcalJets& rhs)
150 {
151 // Assignment operator
152     rhs.Copy(*this);
153     return (*this);
154 }
155
156 void AliGenReaderEcalJets::Copy(TObject&) const
157 {
158     //
159     // Copy 
160     //
161     Fatal("Copy","Not implemented!\n");
162 }
163
164
165
166
167
168
169