969c7896 |
1 | // -*- mode: C++ -*- |
5fe09262 |
2 | #ifndef ALIMCEVENTHANDLER_H |
3 | #define ALIMCEVENTHANDLER_H |
033693d3 |
4 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * |
5 | * See cxx source for full Copyright notice */ |
6 | |
7 | |
8 | /* $Id$ */ |
9 | |
10 | //------------------------------------------------------------------------- |
11 | // Class AliMCEvent |
12 | // This class gives access to MC truth during the analysis. |
415d9f5c |
13 | // Monte Carlo truth is contained in the kinematics tree (produced particles) and |
033693d3 |
14 | // the tree of reference hits. |
15 | // |
16 | // Origin: Andreas Morsch, CERN, andreas.morsch@cern.ch |
17 | //------------------------------------------------------------------------- |
d2f1d9ef |
18 | #include "AliVEventHandler.h" |
47b95647 |
19 | #include "AliHeader.h" |
033693d3 |
20 | class TFile; |
21 | class TTree; |
22 | class TParticle; |
0a05cd41 |
23 | class TString; |
033693d3 |
24 | class TClonesArray; |
5efedd31 |
25 | class TDirectoryFile; |
26 | |
415d9f5c |
27 | class AliMCEvent; |
28 | |
033693d3 |
29 | |
30 | |
276be602 |
31 | class AliMCEventHandler : public AliVEventHandler |
033693d3 |
32 | { |
33 | public: |
5fe09262 |
34 | AliMCEventHandler(); |
35 | AliMCEventHandler(const char* name, const char* title); |
36 | virtual ~AliMCEventHandler(); |
033693d3 |
37 | virtual void SetOutputFileName(char* /* fname */) {;} |
38 | virtual char* GetOutputFileName() {return 0;} |
0a05cd41 |
39 | virtual void SetInputPath(char* fname); |
933fd60f |
40 | virtual void SetInputTree(TTree* /*tree*/) {;} |
0a05cd41 |
41 | virtual TString* GetInputPath() {return fPathName;} |
300d5701 |
42 | virtual Bool_t Init(Option_t* opt); |
43 | virtual Bool_t InitIO(Option_t* opt) {return Init(opt);}; |
44 | virtual Bool_t Init(TTree* /*tree*/, Option_t* /*opt*/) {return kTRUE;} |
ed97dc98 |
45 | virtual Bool_t BeginEvent(Long64_t entry); |
24e577e5 |
46 | virtual Bool_t Notify() { return AliVEventHandler::Notify(); }; |
890126ab |
47 | virtual Bool_t Notify(const char* path); |
033693d3 |
48 | virtual Bool_t FinishEvent(); |
49 | virtual Bool_t Terminate(); |
50 | virtual Bool_t TerminateIO(); |
5fe09262 |
51 | virtual void ResetIO(); |
9aea8469 |
52 | virtual Bool_t GetEvent(Int_t iev); |
969c7896 |
53 | virtual void SetReadTR(Bool_t flag) { fReadTR = flag; } |
033693d3 |
54 | // |
415d9f5c |
55 | AliMCEvent* MCEvent() {return fMCEvent;} |
56 | TTree* TreeTR() {return fTreeTR;} |
57 | Int_t GetParticleAndTR(Int_t i, TParticle*& particle, TClonesArray*& trefs); |
58 | void DrawCheck(Int_t i, Int_t search=0); |
5fe09262 |
59 | private: |
415d9f5c |
60 | Bool_t OpenFile(Int_t i); |
033693d3 |
61 | private: |
415d9f5c |
62 | AliMCEvent *fMCEvent; //! MC Event |
033693d3 |
63 | TFile *fFileE; //! File with TreeE |
64 | TFile *fFileK; //! File with TreeK |
65 | TFile *fFileTR; //! File with TreeTR |
66 | TTree *fTreeE; //! TreeE (Event Headers) |
67 | TTree *fTreeK; //! TreeK (kinematics tree) |
68 | TTree *fTreeTR; //! TreeTR (track references tree) |
5efedd31 |
69 | TDirectoryFile *fDirK; //! Directory for Kine Tree |
70 | TDirectoryFile *fDirTR; //! Directory for TR Tree |
033693d3 |
71 | Int_t fNEvent; //! Number of events |
72 | Int_t fEvent; //! Current event |
0a05cd41 |
73 | TString *fPathName; //! Input file path |
9aea8469 |
74 | char *fExtension; //! File name extension |
75 | Int_t fFileNumber; //! Input file number |
76 | Int_t fEventsPerFile; //! Number of events per file |
969c7896 |
77 | Bool_t fReadTR; // determines if TR shall be read |
78 | ClassDef(AliMCEventHandler,1) //MC Truth EventHandler class |
033693d3 |
79 | }; |
80 | #endif |
81 | |