//-------------------------------------------------------------------------
#include "AliVEventHandler.h"
#include "AliHeader.h"
+#include <TExMap.h>
+
class TFile;
class TTree;
+class TList;
+
class TParticle;
class TString;
class TClonesArray;
virtual void ResetIO();
virtual Bool_t GetEvent(Int_t iev);
virtual void SetReadTR(Bool_t flag) { fReadTR = flag; }
+ virtual void AddSubsidiaryHandler(AliMCEventHandler* handler);
+ virtual void SetNumberOfEventsInContainer(Int_t nev) {fEventsInContainer = nev;}
//
AliMCEvent* MCEvent() const {return fMCEvent;}
TTree* TreeTR() const {return fTreeTR;}
TTree* TreeK() const {return fTreeK;}
Int_t GetParticleAndTR(Int_t i, TParticle*& particle, TClonesArray*& trefs);
void DrawCheck(Int_t i, Int_t search=0);
+ Bool_t InitOk() {return fInitOk;}
+ // Label manipulation
+ void SelectParticle(Int_t i);
+ Bool_t IsParticleSelected(Int_t i);
+ void CreateLabelMap();
+ Int_t GetNewLabel(Int_t i);
+
private:
Bool_t OpenFile(Int_t i);
+ void VerifySelectedParticles();
AliMCEventHandler(const AliMCEventHandler& handler);
AliMCEventHandler& operator=(const AliMCEventHandler& handler);
private:
- AliMCEvent *fMCEvent; //! MC Event
- TFile *fFileE; //! File with TreeE
- TFile *fFileK; //! File with TreeK
- TFile *fFileTR; //! File with TreeTR
- TTree *fTreeE; //! TreeE (Event Headers)
- TTree *fTreeK; //! TreeK (kinematics tree)
- TTree *fTreeTR; //! TreeTR (track references tree)
- TDirectoryFile *fDirK; //! Directory for Kine Tree
- TDirectoryFile *fDirTR; //! Directory for TR Tree
- Int_t fNEvent; //! Number of events
- Int_t fEvent; //! Current event
- TString *fPathName; //! Input file path
- char *fExtension; //! File name extension
- Int_t fFileNumber; //! Input file number
- Int_t fEventsPerFile; //! Number of events per file
- Bool_t fReadTR; // determines if TR shall be read
- Bool_t fInitOk; // Initialization ok
+ AliMCEvent *fMCEvent; //! MC Event
+ TFile *fFileE; //! File with TreeE
+ TFile *fFileK; //! File with TreeK
+ TFile *fFileTR; //! File with TreeTR
+ TTree *fTreeE; //! TreeE (Event Headers)
+ TTree *fTreeK; //! TreeK (kinematics tree)
+ TTree *fTreeTR; //! TreeTR (track references tree)
+ TDirectoryFile *fDirK; //! Directory for Kine Tree
+ TDirectoryFile *fDirTR; //! Directory for TR Tree
+ TExMap fParticleSelected; //! List of selected MC particles for t
+ TExMap fLabelMap; //! Stores the Map of MC (ESDLabel,AODlabel)
+ Int_t fNEvent; //! Number of events
+ Int_t fEvent; //! Current event
+ TString *fPathName; //! Input file path
+ const Char_t *fExtension; //! File name extension
+ Int_t fFileNumber; //! Input file number
+ Int_t fEventsPerFile; //! Number of events per file
+ Bool_t fReadTR; // determines if TR shall be read
+ Bool_t fInitOk; // Initialization ok
+ TList *fSubsidiaryHandlers; //! List of subsidiary MC handlers (for example for Background)
+ Int_t fEventsInContainer; //! Number of events in container class
ClassDef(AliMCEventHandler,1) //MC Truth EventHandler class
};
#endif