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
 
#include <TClonesArray.h>
 
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