PairAnalysisMC.h

Go to the documentation of this file.

#ifndef PAIRANALYSISMC_H
#define PAIRANALYSISMC_H
/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
//#####################################################
//#                                                   #
//#              Class PairAnalysisMC                   #
//#                                                   #
//#   by Julian Book, Uni Ffm, Julian.Book@cern.ch    #
//#                                                   #
//#####################################################
 
#ifndef ROOT_TObject
#include <TMCProcess.h>
#include <TObject.h>
#endif
class TParticle;
class CbmMCTrack;
class PairAnalysisTrack;
 
#include "PairAnalysisPair.h"
#include "PairAnalysisSignalMC.h"
 
class PairAnalysisMC : public TObject {
 
public:
  PairAnalysisMC();
  virtual ~PairAnalysisMC();
 
  void SetHasMC(Bool_t hasMC) { fHasMC = hasMC; }
  Bool_t HasMC() const { return fHasMC; }
 
  static PairAnalysisMC* Instance();
 
  Int_t GetNMCTracks();  // return number of generated tracks
  Int_t GetMotherPDG(
    const PairAnalysisTrack* _track);  // return mother PID from the MC stack
  Int_t GetMotherPDG(
    const CbmMCTrack* _track);  // return mother PID from the MC stack
 
  Bool_t ConnectMCEvent();
 
  Bool_t IsMotherPdg(const PairAnalysisPair* pair, Int_t pdgMother);
  Bool_t IsMotherPdg(const PairAnalysisTrack* particle1,
                     const PairAnalysisTrack* particle2,
                     Int_t pdgMother);
 
  Bool_t IsMCTruth(const PairAnalysisPair* pair,
                   const PairAnalysisSignalMC* signalMC) const;
  Bool_t IsMCTruth(const PairAnalysisTrack* trk,
                   PairAnalysisSignalMC* signalMC,
                   Int_t branch) const;
  Bool_t
  IsMCTruth(Int_t label, PairAnalysisSignalMC* signalMC, Int_t branch) const;
 
  Int_t GetMothersLabel(Int_t daughterLabel) const;
  Int_t GetPdgFromLabel(Int_t label) const;
 
  Bool_t CheckGEANTProcess(Int_t label, TMCProcess process) const;
  //  Bool_t IsPhysicalPrimary(Int_t label) const;  // checks if a particle is physical primary
  //  Bool_t IsSecondaryFromWeakDecay(Int_t label) const;
  //  Bool_t IsSecondaryFromMaterial(Int_t label) const;
 
  Bool_t HaveSameMother(const PairAnalysisPair* pair) const;
 
  Int_t GetLabelMotherWithPdg(const PairAnalysisPair* pair, Int_t pdgMother);
  Int_t GetLabelMotherWithPdg(const PairAnalysisTrack* particle1,
                              const PairAnalysisTrack* particle2,
                              Int_t pdgMother);
 
  CbmMCTrack* GetMCTrackFromMCEvent(
    Int_t label) const;  // return MC track directly from MC event
  CbmMCTrack*
  GetMCTrack(const PairAnalysisTrack*
               _track);  // return MC track associated with reco track
 
  CbmMCTrack* GetMCTrackMother(
    const PairAnalysisTrack* _track);  // return MC mother for reco track
  CbmMCTrack*
  GetMCTrackMother(const CbmMCTrack* _track);  // return MC mother for mc rtack
 
  Int_t
  NumberOfDaughters(const CbmMCTrack* particle);  // return number of daughters
 
  void GetDaughters(const TObject* mother, CbmMCTrack*& d1, CbmMCTrack*& d2);
  Bool_t CheckParticleSource(Int_t label,
                             PairAnalysisSignalMC::ESource source) const;
  //  Bool_t GetPrimaryVertex(Double_t &primVtxX, Double_t &primVtxY, Double_t &primVtxZ);
  //  AliMCEvent* GetMCEvent() { return fMCEvent; }         // return the AliMCEvent
 
private:
  TObject* fMCEvent;       // MC event object
  Bool_t fHasMC;           // Do we have an MC handler?
  TClonesArray* fMCArray;  //mcArray for AOD MC particles
 
  static PairAnalysisMC* fgInstance;  
 
  PairAnalysisMC(const PairAnalysisMC& c);
  PairAnalysisMC& operator=(const PairAnalysisMC& c);
 
  Bool_t ComparePDG(Int_t particlePDG,
                    Int_t requiredPDG,
                    Bool_t pdgExclusion,
                    Bool_t checkBothCharges) const;
 
  Bool_t CheckIsDalitz(Int_t label,
                       const PairAnalysisSignalMC* const signalMC) const;
  Bool_t CheckDalitzDecision(Int_t mLabel,
                             const PairAnalysisSignalMC* const signalMC) const;
  Bool_t IsPhysicalPrimary(Int_t label, UInt_t processID) const;
  Bool_t IsSecondaryFromWeakDecay(Int_t label, UInt_t processID) const;
  Bool_t IsSecondaryFromMaterial(Int_t label, UInt_t processID) const;
 
  ClassDef(PairAnalysisMC, 0)  // MC signal finder
};
 
//
// inline functions
//
inline Bool_t PairAnalysisMC::IsMotherPdg(const PairAnalysisPair* pair,
                                          Int_t pdgMother) {
  return IsMotherPdg(
    pair->GetFirstDaughter(), pair->GetSecondDaughter(), pdgMother);
}
//___________________________________________________________
inline Bool_t PairAnalysisMC::IsMotherPdg(const PairAnalysisTrack* particle1,
                                          const PairAnalysisTrack* particle2,
                                          Int_t pdgMother) {
  return GetLabelMotherWithPdg(particle1, particle2, pdgMother) >= 0;
}
//___________________________________________________________
inline Int_t PairAnalysisMC::GetLabelMotherWithPdg(const PairAnalysisPair* pair,
                                                   Int_t pdgMother) {
  return GetLabelMotherWithPdg(
    pair->GetFirstDaughter(), pair->GetSecondDaughter(), pdgMother);
}
 
#endif