New representation of the raw TRD PID signal in ESD (Alex, Yuri)
[u/mrichter/AliRoot.git] / STEER / AliMCEventHandler.cxx
CommitLineData
5fe09262 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// Class AliMCEventHandler
19// This class gives access to MC truth during the analysis.
20// Monte Carlo truth is containe in the kinematics tree (produced particles) and
21// the tree of reference hits.
22//
23// Origin: Andreas Morsch, CERN, andreas.morsch@cern.ch
24//---------------------------------------------------------------------------------
25
26
27
28#include "AliMCEventHandler.h"
415d9f5c 29#include "AliMCEvent.h"
5fe09262 30#include "AliTrackReference.h"
31#include "AliHeader.h"
32#include "AliStack.h"
2c36081e 33#include "AliLog.h"
5fe09262 34
35#include <TTree.h>
36#include <TFile.h>
37#include <TParticle.h>
0a05cd41 38#include <TString.h>
5fe09262 39#include <TClonesArray.h>
40#include <TDirectoryFile.h>
5fe09262 41
42ClassImp(AliMCEventHandler)
43
44AliMCEventHandler::AliMCEventHandler() :
d2f1d9ef 45 AliVEventHandler(),
415d9f5c 46 fMCEvent(new AliMCEvent()),
5fe09262 47 fFileE(0),
48 fFileK(0),
49 fFileTR(0),
5fe09262 50 fTreeE(0),
51 fTreeK(0),
52 fTreeTR(0),
5efedd31 53 fDirK(0),
54 fDirTR(0),
5fe09262 55 fNEvent(-1),
56 fEvent(-1),
0a05cd41 57 fPathName(new TString("./")),
9aea8469 58 fExtension(""),
59 fFileNumber(0),
969c7896 60 fEventsPerFile(0),
61 fReadTR(kTRUE)
5fe09262 62{
63 // Default constructor
64}
65
66AliMCEventHandler::AliMCEventHandler(const char* name, const char* title) :
d2f1d9ef 67 AliVEventHandler(name, title),
415d9f5c 68 fMCEvent(new AliMCEvent()),
5fe09262 69 fFileE(0),
70 fFileK(0),
71 fFileTR(0),
5fe09262 72 fTreeE(0),
73 fTreeK(0),
74 fTreeTR(0),
5efedd31 75 fDirK(0),
76 fDirTR(0),
5fe09262 77 fNEvent(-1),
78 fEvent(-1),
0a05cd41 79 fPathName(new TString("./")),
9aea8469 80 fExtension(""),
81 fFileNumber(0),
0cd61c1d 82 fEventsPerFile(0),
83 fReadTR(kTRUE)
5fe09262 84{
85 // Constructor
86}
87AliMCEventHandler::~AliMCEventHandler()
88{
89 // Destructor
415d9f5c 90 delete fMCEvent;
5fe09262 91 delete fFileE;
92 delete fFileK;
93 delete fFileTR;
94}
95
300d5701 96Bool_t AliMCEventHandler::Init(Option_t* opt)
5fe09262 97{
98 // Initialize input
99 //
6073f8c9 100 if (!(strcmp(opt, "proof")) || !(strcmp(opt, "local"))) return kTRUE;
101 //
0a05cd41 102 fFileE = TFile::Open(Form("%sgalice.root", fPathName->Data()));
103 if (!fFileE) AliFatal(Form("AliMCEventHandler:galice.root not found in directory %s ! \n", fPathName->Data()));
5fe09262 104
415d9f5c 105 //
106 // Tree E
5fe09262 107 fFileE->GetObject("TE", fTreeE);
415d9f5c 108 // Connect Tree E to the MCEvent
109 fMCEvent->ConnectTreeE(fTreeE);
5fe09262 110 fNEvent = fTreeE->GetEntries();
111 //
112 // Tree K
0a05cd41 113 fFileK = TFile::Open(Form("%sKinematics%s.root", fPathName->Data(), fExtension));
2c36081e 114 if (!fFileK) AliFatal(Form("AliMCEventHandler:Kinematics.root not found in directory %s ! \n", fPathName));
9aea8469 115 fEventsPerFile = fFileK->GetNkeys() - fFileK->GetNProcessIDs();
5fe09262 116 //
117 // Tree TR
969c7896 118 if (fReadTR) {
119 fFileTR = TFile::Open(Form("%sTrackRefs%s.root", fPathName->Data(), fExtension));
120 if (!fFileTR) AliWarning(Form("AliMCEventHandler:TrackRefs.root not found in directory %s ! \n", fPathName->Data()));
121 }
5fe09262 122 //
123 // Reset the event number
2c36081e 124 fEvent = -1;
125 fFileNumber = 0;
f6065654 126 printf("AliMCEvenHandler::Init() %d\n",__LINE__);
2c36081e 127 AliInfo(Form("AliMCEventHandler:Number of events in this directory %5d \n", fNEvent));
5fe09262 128 return kTRUE;
5fe09262 129}
130
9aea8469 131Bool_t AliMCEventHandler::GetEvent(Int_t iev)
132{
133 // Load the event number iev
2c36081e 134 //
135 // Calculate the file number
415d9f5c 136 Int_t inew = iev / fEventsPerFile;
9aea8469 137 if (inew != fFileNumber) {
138 fFileNumber = inew;
139 if (!OpenFile(fFileNumber)){
140 return kFALSE;
141 }
142 }
2c36081e 143 // Folder name
5fe09262 144 char folder[20];
9aea8469 145 sprintf(folder, "Event%d", iev);
5fe09262 146 // TreeE
9aea8469 147 fTreeE->GetEntry(iev);
5fe09262 148 // Tree K
5efedd31 149 fFileK->GetObject(folder, fDirK);
150 if (!fDirK) {
2c36081e 151 AliWarning(Form("AliMCEventHandler: Event #%5d not found\n", iev));
9aea8469 152 return kFALSE;
153 }
5efedd31 154 fDirK ->GetObject("TreeK", fTreeK);
415d9f5c 155 // Connect TreeK to MCEvent
156 fMCEvent->ConnectTreeK(fTreeK);
5fe09262 157 //Tree TR
07bd4750 158 if (fFileTR) {
5efedd31 159 // Check which format has been read
160 fFileTR->GetObject(folder, fDirTR);
161 fDirTR->GetObject("TreeTR", fTreeTR);
415d9f5c 162 //
163 // Connect TR to MCEvent
164 fMCEvent->ConnectTreeTR(fTreeTR);
5fe09262 165 }
5fe09262 166 //
5fe09262 167 return kTRUE;
9aea8469 168}
169
170Bool_t AliMCEventHandler::OpenFile(Int_t i)
171{
172 // Open file i
173 Bool_t ok = kTRUE;
174 if (i > 0) {
175 fExtension = Form("%d", i);
176 } else {
177 fExtension = "";
178 }
179
180
181 delete fFileK;
0a05cd41 182 fFileK = TFile::Open(Form("%sKinematics%s.root", fPathName->Data(), fExtension));
9aea8469 183 if (!fFileK) {
0a05cd41 184 AliFatal(Form("AliMCEventHandler:Kinematics%s.root not found in directory %s ! \n", fExtension, fPathName->Data()));
9aea8469 185 ok = kFALSE;
186 }
187
188 delete fFileTR;
0a05cd41 189 fFileTR = TFile::Open(Form("%sTrackRefs%s.root", fPathName->Data(), fExtension));
9aea8469 190 if (!fFileTR) {
0a05cd41 191 AliWarning(Form("AliMCEventHandler:TrackRefs%s.root not found in directory %s ! \n", fExtension, fPathName->Data()));
9aea8469 192 ok = kFALSE;
193 }
194
195 return ok;
196}
197
ed97dc98 198Bool_t AliMCEventHandler::BeginEvent(Long64_t entry)
9aea8469 199{
200 // Read the next event
ed97dc98 201 if (entry == -1) {
202 fEvent++;
203 entry = fEvent;
204 } else {
205 fEvent = entry;
206 }
207
208 if (entry >= fNEvent) {
f6065654 209 AliWarning(Form("AliMCEventHandler: Event number out of range %5d %5d\n", entry,fNEvent));
9aea8469 210 return kFALSE;
211 }
ed97dc98 212 return GetEvent(entry);
5fe09262 213}
214
215Int_t AliMCEventHandler::GetParticleAndTR(Int_t i, TParticle*& particle, TClonesArray*& trefs)
216{
217 // Retrieve entry i
415d9f5c 218 return (fMCEvent->GetParticleAndTR(i, particle, trefs));
5fe09262 219}
220
2d8f26f6 221void AliMCEventHandler::DrawCheck(Int_t i, Int_t search)
5fe09262 222{
223 // Retrieve entry i and draw momentum vector and hits
415d9f5c 224 fMCEvent->DrawCheck(i, search);
5fe09262 225}
226
890126ab 227Bool_t AliMCEventHandler::Notify(const char *path)
5fe09262 228{
53faeca4 229 // Notify about directory change
230 // The directory is taken from the 'path' argument
231 // Reconnect trees
232 TString fileName(path);
233 if(fileName.Contains("AliESDs.root")){
234 fileName.ReplaceAll("AliESDs.root", "");
235 }
236 else if(fileName.Contains("galice.root")){
237 // for running with galice and kinematics alone...
238 fileName.ReplaceAll("galice.root", "");
239 }
240
241 *fPathName = fileName;
242 printf("AliMCEventHandler::Notify() Path: %s\n", fPathName->Data());
243
f6065654 244 ResetIO();
245 InitIO("");
246
5fe09262 247 return kTRUE;
248}
36e82a52 249
5fe09262 250void AliMCEventHandler::ResetIO()
251{
5efedd31 252// Clear header and stack
415d9f5c 253 fMCEvent->Clean();
5efedd31 254
36e82a52 255// Delete Tree E
415d9f5c 256 delete fTreeE; fTreeE = 0;
36e82a52 257
5efedd31 258// Reset files
1febc514 259 if (fFileE) {fFileE->Close(); delete fFileE; fFileE = 0;}
260 if (fFileK) {fFileK->Close(); delete fFileK; fFileK = 0;}
261 if (fFileTR) {fFileTR->Close(); delete fFileTR; fFileTR = 0;}
19cfde04 262 fExtension="";
5fe09262 263}
264
265
266Bool_t AliMCEventHandler::FinishEvent()
267{
5efedd31 268 // Clean-up after each event
269 delete fDirTR; fDirTR = 0;
270 delete fDirK; fDirK = 0;
415d9f5c 271 fMCEvent->FinishEvent();
5fe09262 272 return kTRUE;
273}
274
275Bool_t AliMCEventHandler::Terminate()
276{
277 // Dummy
278 return kTRUE;
279}
280
281Bool_t AliMCEventHandler::TerminateIO()
282{
283 // Dummy
284 return kTRUE;
285}
286
0a05cd41 287
0931e76a 288void AliMCEventHandler::SetInputPath(const char* fname)
0a05cd41 289{
290 // Set the input path name
291 delete fPathName;
292 fPathName = new TString(fname);
293}