AliMCAnalysisUtils.h

Go to the documentation of this file.

#ifndef ALIMCANALYSISUTILS_H
#define ALIMCANALYSISUTILS_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */

//_________________________________________________________________________
//_________________________________________________________________________

// --- ROOT system ---
#include <TObject.h>
#include <TString.h>
#include <TLorentzVector.h>
class TList ;
class TVector3;
class TClonesArray;

//--- AliRoot system ---
class AliMCEvent;
class AliGenEventHeader;
class AliGenPythiaEventHeader;

class AliMCAnalysisUtils : public TObject {
  
 public: 
  
  AliMCAnalysisUtils() ;          // ctor
  
  virtual ~AliMCAnalysisUtils() ; // virtual dtor
    
  //--------------------------------------
  // Enum with tag for origin of particles
  //--------------------------------------
  
  enum mcTypes { kMCPhoton,     kMCPrompt,      kMCFragmentation, kMCISR,    
                 kMCPi0Decay,   kMCEtaDecay,    kMCOtherDecay,
                 kMCDecayPairLost, kMCDecayPairInCalo,
                 kMCDecayDalitz,kMCConversion,  kMCElectron,
                 kMCEFromCFromB,kMCEFromC,      kMCEFromB,        kMCZDecay, kMCWDecay,
                 kMCMuon,       kMCPion,        kMCPi0,           kMCKaon,   kMCEta,      kMCProton,   
                 kMCAntiProton, kMCNeutron,     kMCAntiNeutron,
                 kMCUnknown,    kMCBadLabel                                                         } ;
  
  //--------------------------------------
  // Methods to check origin of clusters
  //--------------------------------------
  
  Int_t   CheckCommonAncestor(Int_t index1, Int_t index2, const AliMCEvent* mcevent, 
            Int_t & ancPDG, Int_t & ancStatus, TLorentzVector & momentum, TVector3 & prodVertex) ;
  
  Int_t   CheckOrigin(Int_t label, AliMCEvent* mcevent, TString selectHeaderName, Float_t clusE) ;  
  Int_t   CheckOrigin(const Int_t *labels, const UShort_t * edepFrac, Int_t nlabels, 
                      AliMCEvent* mcevent, TString selectHeaderName, Float_t clusE,
                      const TObjArray *arrayCluster = 0x0) ; 
  
  void    CheckOverlapped2GammaDecay(const Int_t *labels, const UShort_t * edepFrac, Int_t nlabels, 
                                     Int_t mesonIndex, Float_t clusE, const AliMCEvent* mcevent, Int_t & tag); 
  
  void    CheckLostDecayPair(const TObjArray *arrayCluster, Int_t iMom, Int_t iParent, const AliMCEvent* mcevent, Int_t & tag); 
  
  TLorentzVector GetMother     (Int_t label,const AliMCEvent* mcevent, Bool_t & ok);
  TLorentzVector GetMother     (Int_t label,const AliMCEvent* mcevent, Int_t & pdg, Int_t & status, Bool_t & ok);
  TLorentzVector GetMother     (Int_t label,const AliMCEvent* mcevent, Int_t & pdg, Int_t & status, Bool_t & ok, Int_t & momLabel);
  TLorentzVector GetGrandMother(Int_t label,const AliMCEvent* mcevent, Int_t & pdg, Int_t & status, Bool_t & ok, Int_t & grandMomLabel, Int_t & greatMomLabel);

  TLorentzVector GetMotherWithPDG     (Int_t label, Int_t pdg,const AliMCEvent* mcevent, Bool_t & ok, Int_t & momLabel);
  TLorentzVector GetFirstMotherWithPDG(Int_t label, Int_t pdg,const AliMCEvent* mcevent, Bool_t & ok, Int_t & momLabel, Int_t & gparentlabel);
  
  void GetMCDecayAsymmetryAngleForPDG(Int_t label, Int_t pdg,const AliMCEvent* mcevent,
                                      Float_t & asy, Float_t & angle, Bool_t & ok);

  Int_t          GetNDaughters(Int_t label,const AliMCEvent* mcevent, Bool_t & ok);
  TLorentzVector GetDaughter  (Int_t daughter, Int_t label,const AliMCEvent* mcevent,
                               Int_t & pdg, Int_t & status, Bool_t & ok, Int_t & daugLabel, TVector3 & prodVertex);

  Int_t          GetNOverlaps(const Int_t * label, UInt_t nlabels,
                              Int_t mctag, Int_t mesonLabel,
                              AliMCEvent* mcevent,
                              Int_t *overpdg, Int_t *overlabel);
  
  //Check or set the bits produced in the above methods
  void    SetTagBit(Int_t &tag, UInt_t set) const {
    // Set bit of type set (mcTypes) in tag
    tag |= (1<<set) ; 
  } 
  
  Bool_t  CheckTagBit(Int_t tag, UInt_t test) const {
    // Check if in tag the bit test (mcTypes) is set.
    if (tag & (1<<test) ) return  kTRUE ;    
    else return kFALSE ;
  }
  
  //--------------------------------------
  // Other methods
  //--------------------------------------
    
  // Method to recover MC jets stored in generator
  TList * GetJets(AliMCEvent* mcevent, Bool_t check) ;
  
  static AliGenPythiaEventHeader * GetPythiaEventHeader
  (AliMCEvent* mcevent, TString selecHeaderName, 
   TString & genName, TString & processName, 
   Int_t   & process, Int_t & firstParticle, 
   Int_t   & pythiaVersion);

  AliGenPythiaEventHeader * CheckAndGetPythiaEventHeader
  (AliMCEvent* mcevent, TString selecHeaderName);
  TString GetPythiaHeaderName()   const { return fPyGenName       ; }
  TString GetPythiaProcessName()  const { return fPyProcessName   ; } 
  Int_t   GetPythiaProcess()      const { return fPyProcess       ; }
  Int_t   GetPythiaFirstParticle()const { return fPyFirstParticle ; } 
  Int_t   GetPythiaVersion()      const { return fPyVersion       ; }
  Int_t   GetPythiaMinPartParent()const { return fMinPartonicParent;}
  Int_t   GetPythiaMaxPartParent()const { return fMaxPartonicParent;}
  
  
  void    SetDebug(Int_t deb)           { fDebug=deb           ; }
  Int_t   GetDebug()              const { return fDebug        ; }  
  
  enum generator {kPythia = 0, kHerwig = 1, kHijing = 2, kBoxLike = 3 } ;
  void    SetMCGenerator(Int_t   mcgen) ;
  void    SetMCGenerator(TString mcgen) ;
  Int_t   GetMCGenerator()        const { return fMCGenerator  ; }
  TString GetMCGeneratorString()  const { return fMCGeneratorString ; }
  
  void    Print(const Option_t * opt) const;
  void    PrintAncestry(AliMCEvent* mcevent, Int_t label, Int_t nGenerMax = 1000) const;
  void    PrintMCTag(Int_t tag) const;

 private:

  Int_t          fCurrentEvent;        
  
  Int_t          fDebug;               
  
  TList        * fJetsList;            
  
  Int_t          fMCGenerator;         
  
  TString        fMCGeneratorString;   
  
  TLorentzVector fDaughMom;            
  
  TLorentzVector fDaughMom2;           
  
  TLorentzVector fMotherMom;           
  
  TLorentzVector fGMotherMom;          
  
  // Specific to pythia events and header recovery
  // avoid multiple times checks
  AliGenPythiaEventHeader 
              * fPyGenHead;           
  TString       fPyGenName;           
  TString       fPyProcessName;       
  Int_t         fPyProcess;           
  Int_t         fPyFirstParticle;     
  Int_t         fPyVersion;           
  
  Int_t         fMinPartonicParent;   
  Int_t         fMaxPartonicParent;   
  
  AliMCAnalysisUtils & operator = (const AliMCAnalysisUtils & mcu) ; 
  
  AliMCAnalysisUtils(              const AliMCAnalysisUtils & mcu) ; 
  
  ClassDef(AliMCAnalysisUtils,8) ;

} ;

#endif //ALIMCANALYSISUTILS_H