| 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 | |
git://git.uio.no / u / mrichter / AliRoot.git / blame