Removing AliMCProcess and AliMC
[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 /*
17 $Log$
18 Revision 1.2  2002/10/30 13:39:06  hristov
19 Missing initialization added (Alpha)
20
21 Revision 1.1  2002/01/08 09:59:34  morsch
22 Readers for EMCAL primary particle input.
23
24 */
25
26 #include <TFile.h>
27 #include <TTree.h>
28 #include <TParticle.h>
29
30 #include "AliGenReaderEcalJets.h"
31 ClassImp(AliGenReaderEcalJets)
32
33
34 AliGenReaderEcalJets::AliGenReaderEcalJets() 
35 {
36 // Default constructor
37     fNcurrent   = 0;
38     fTreeNtuple = 0;
39 }
40
41 void AliGenReaderEcalJets::Init() 
42 {
43 //
44 // reset the existing file environment and open a new root file if
45 // the pointer to the Fluka tree is null
46     
47     TFile *pFile=0;
48     if (!pFile) {
49         pFile = new TFile(fFileName);
50         pFile->cd();
51         printf("\n I have opened %s file \n", fFileName);
52     }
53 // get the tree address in the Fluka boundary source file
54     fTreeNtuple = (TTree*)gDirectory->Get("h1");
55     TTree *h1=fTreeNtuple;
56     h1->SetMakeClass(1);
57 //Set branch addresses
58     h1->SetBranchAddress("nev",   &fNev);
59     h1->SetBranchAddress("x",      fX);
60     h1->SetBranchAddress("xtyp",   fXtyp);
61     h1->SetBranchAddress("npart", &fNpart);
62     h1->SetBranchAddress("xpt",    fXpt);
63     h1->SetBranchAddress("xeta",   fXeta);
64     h1->SetBranchAddress("xphi",   fXphi);
65     h1->SetBranchAddress("xid",    fXid);
66     h1->SetBranchAddress("njet",  &fNjet);
67     h1->SetBranchAddress("jet",    fJet);
68     h1->SetBranchAddress("jeta",   fJeta);
69     h1->SetBranchAddress("jphi",   fJphi);
70     h1->SetBranchAddress("nsjet", &fNsjet);
71     h1->SetBranchAddress("jset",   fJset);
72     h1->SetBranchAddress("jseta",  fJseta);
73     h1->SetBranchAddress("jsphi",  fJsphi);
74     h1->SetBranchAddress("npjet", &fNpjet);
75     h1->SetBranchAddress("jpet",   fJpet);
76     h1->SetBranchAddress("jpeta",  fJpeta);
77     h1->SetBranchAddress("jpphi",  fJpphi);
78 }
79
80 Int_t AliGenReaderEcalJets::NextEvent() 
81 {
82 // Read the next event  
83     Int_t nTracks=0, nread=0;
84     
85     TFile* pFile = fTreeNtuple->GetCurrentFile();
86     pFile->cd();
87
88     Int_t nentries = (Int_t) fTreeNtuple->GetEntries();
89     if (fNcurrent < nentries) {
90         Int_t nb = (Int_t)fTreeNtuple->GetEvent(fNcurrent);
91         nread += nb;
92         fNcurrent++;
93         printf("\n Next event contains %d tracks! \n", fNpjet);
94         nTracks    = fNpjet;
95         fNparticle = 0;
96         return nTracks;
97     }
98     return 0;
99 }
100
101 TParticle* AliGenReaderEcalJets::NextParticle() 
102 {
103     Float_t p[4];
104 // Read the next particle
105     Int_t    ipart  = fXid[fNparticle];
106     Float_t  pt     = fXpt[fNparticle];
107     Float_t  eta    = fXeta[fNparticle];
108     Float_t  phi    = fXphi[fNparticle];
109     Float_t  theta  = 2.*TMath::ATan(TMath::Exp(-eta));
110     Double_t amass  = TDatabasePDG::Instance()->GetParticle(ipart)->Mass();
111
112     p[0] = pt*TMath::Sin(phi);
113     p[1] = pt*TMath::Cos(phi);      
114     p[2] = pt/TMath::Cos(theta);
115     p[3] = TMath::Sqrt(pt*pt+p[2]*p[2]+amass*amass);
116     
117
118     TParticle* particle = 
119         new TParticle(ipart, 0, -1, -1, -1, -1, p[0], p[1], p[2], p[3], 
120                       0., 0., 0., 0.);
121     fNparticle++;
122     return particle;
123 }
124
125
126
127 AliGenReaderEcalJets& AliGenReaderEcalJets::operator=(const  AliGenReaderEcalJets& rhs)
128 {
129 // Assignment operator
130     return *this;
131 }
132
133
134
135
136
137