1 #ifndef ALIMCANALYSISUTILS_H
2 #define ALIMCANALYSISUTILS_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
6 //_________________________________________________________________________
7 // Class for analysis utils for MC data
8 // stored in stack or event header.
10 // - method to check the origin of a given track/cluster
11 // - method to obtain the generated jets
13 //*-- Author: Gustavo Conesa (INFN-LNF)
15 // --- ROOT system ---
21 //--- AliRoot system ---
22 class AliCaloTrackReader ;
25 class AliMCAnalysisUtils : public TObject {
28 AliMCAnalysisUtils() ; // ctor
29 virtual ~AliMCAnalysisUtils() ;//virtual dtor
31 //--------------------------------------
32 //Enum with tag for origin of particles
33 //--------------------------------------
35 //"Mostly" photon parent types on line 1,
37 //then "mostly" electron parent types on line 3, (e.g. electrons can
38 //come from pi0 decay)
39 //then charged particles on line 4,
40 //followed by other and unknown on line 5
41 enum mcTypes { kMCPhoton, kMCPrompt, kMCFragmentation, kMCISR,
42 kMCPi0Decay, kMCEtaDecay, kMCOtherDecay, kMCConversion,
43 kMCElectron, kMCEFromCFromB, kMCEFromC, kMCEFromB, kMCZDecay, kMCWDecay,
44 kMCMuon, kMCPion, kMCPi0, kMCKaon, kMCEta, kMCProton,
45 kMCAntiProton, kMCNeutron, kMCAntiNeutron,
46 kMCOther, kMCUnknown, kMCBadLabel } ;
48 //--------------------------------------
49 // Methods to check origin of clusters
50 //--------------------------------------
52 Int_t CheckCommonAncestor(const Int_t index1, const Int_t index2, const AliCaloTrackReader* reader,
53 Int_t & ancPDG, Int_t & ancStatus, TLorentzVector & momentum, TVector3 & v) ;
55 Int_t CheckOrigin(const Int_t label, const AliCaloTrackReader * reader) ;
57 //Check the label of the most significant particle but do checks on the rest of the contributing labels
58 Int_t CheckOrigin (const Int_t *label, const Int_t nlabels, const AliCaloTrackReader * reader) ;
59 Int_t CheckOriginInStack(const Int_t *labels, const Int_t nlabels, AliStack * stack) ; // ESD
60 Int_t CheckOriginInAOD (const Int_t *labels, const Int_t nlabels, const TClonesArray* mcparticles) ; // AOD
62 void CheckOverlapped2GammaDecay(const Int_t *labels, const Int_t nlabels, const Int_t mesonIndex, AliStack * stack, Int_t & tag); // ESD
63 void CheckOverlapped2GammaDecay(const Int_t *labels, const Int_t nlabels, const Int_t mesonIndex, const TClonesArray* mcparticles, Int_t & tag); // AOD
65 TLorentzVector GetMother (const Int_t label,const AliCaloTrackReader* reader, Bool_t & ok);
66 TLorentzVector GetMother (const Int_t label,const AliCaloTrackReader* reader, Int_t & pdg, Int_t & status, Bool_t & ok);
67 TLorentzVector GetMother (const Int_t label,const AliCaloTrackReader* reader, Int_t & pdg, Int_t & status, Bool_t & ok, Int_t & momLabel);
68 TLorentzVector GetGrandMother(const Int_t label,const AliCaloTrackReader* reader,
69 Int_t & pdg, Int_t & status, Bool_t & ok, Int_t & grandMomLabel, Int_t & greatMomLabel);
71 TLorentzVector GetMotherWithPDG(const Int_t label, const Int_t pdg,const AliCaloTrackReader* reader, Bool_t & ok);
73 Float_t GetMCDecayAsymmetryForPDG(const Int_t label, const Int_t pdg,const AliCaloTrackReader* reader, Bool_t & ok);
75 Int_t GetNDaughters(const Int_t label,const AliCaloTrackReader* reader, Bool_t & ok);
76 TLorentzVector GetDaughter (const Int_t daughter, const Int_t label,const AliCaloTrackReader* reader, Int_t & pdg, Int_t & status, Bool_t & ok, Int_t & daugLabel);
78 //Check or set the bits produced in the above methods
79 void SetTagBit(Int_t &tag, const UInt_t set) const {
80 // Set bit of type set (mcTypes) in tag
84 Bool_t CheckTagBit(const Int_t tag, const UInt_t test) const {
85 // Check if in tag the bit test (mcTypes) is set.
86 if (tag & (1<<test) ) return kTRUE ;
90 //--------------------------------------
92 //--------------------------------------
94 // Method to recover MC jets stored in generator
95 TList * GetJets(const AliCaloTrackReader * reader) ;
97 void SetDebug(Int_t deb) { fDebug=deb ; }
98 Int_t GetDebug() const { return fDebug ; }
100 void SetMCGenerator(TString mcgen) { fMCGenerator = mcgen ; }
101 TString GetMCGenerator() const { return fMCGenerator ; }
103 void Print(const Option_t * opt) const;
104 void PrintMCTag(const Int_t tag) const;
107 Int_t fCurrentEvent; // Current Event
108 Int_t fDebug; // Debug level
109 TList * fJetsList; // List of jets
110 TString fMCGenerator; // MC geneator used to generate data in simulation
112 AliMCAnalysisUtils & operator = (const AliMCAnalysisUtils & mcu) ; // cpy assignment
113 AliMCAnalysisUtils( const AliMCAnalysisUtils & mcu) ; // cpy ctor
115 ClassDef(AliMCAnalysisUtils,4)
119 #endif //ALIMCANALYSISUTILS_H