[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronMC.h

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

//#####################################################
//#                                                   # 
//#              Class AliDielectronMC                #
//#       Cut Class for Jpsi->e+e- analysis           #
//#                                                   #
//#   by WooJin J. Park, GSI / W.J.Park@gsi.de        #
//#                                                   #
//#####################################################

#ifndef ROOT_TObject
#include <TObject.h>
#endif
class AliESDEvent;
class AliHFEpid;
class AliStack;
class AliMCEvent;
class AliESDtrack;
class AliAODTrack;
class TParticle;
class AliMCParticle;
class AliAODMCParticle;
class AliAODMCHeader;

#include "AliDielectronSignalMC.h"
#include "AliDielectronPair.h"

class AliDielectronMC : public TObject{
  
public:
  enum AnalysisType {kUNSET=0, kESD, kAOD};
  
  AliDielectronMC(AnalysisType type=kUNSET);
  virtual ~AliDielectronMC();

  void SetHasMC(Bool_t hasMC) { fHasMC=hasMC; }
  Bool_t HasMC() const { return fHasMC; }
  
  static AliDielectronMC* Instance();
  
  void Initialize();                              // initialization
  Int_t GetNMCTracks();                                     // return number of generated tracks
  Int_t GetNMCTracksFromStack();                            // return number of generated tracks from stack
  Int_t GetNPrimary() const;                                      // return number of primary tracks
  Int_t GetNPrimaryFromStack();                                   // return number of primary tracks from stack
  Int_t GetMCPID(const AliESDtrack* _track);                      // return MC PID
  Int_t GetMCPID(const AliAODTrack* _track);                      // return MC PID for AODtrack
  Int_t GetMCPIDFromStack(const AliESDtrack* _track);             // return MC PID
  Int_t GetMotherPDG(const AliESDtrack* _track);                  // return mother PID from the MC stack
  Int_t GetMotherPDG(const AliAODTrack* _track);                  // return mother PID from the MC stack
  Int_t GetMotherPDG(const AliMCParticle* _track);                  // return mother PID from the MC stack
  Int_t GetMotherPDG(const AliAODMCParticle* _track);                  // return mother PID from the MC stack
  Int_t GetMotherPDGFromStack(const AliESDtrack* _track);         // return mother PID from the MC stack
  Int_t GetMCProcess(const AliESDtrack* _track);                  // return process number
  Int_t GetMCProcessFromStack(const AliESDtrack* _track);         // return process number
  Int_t GetMCProcessMother(const AliESDtrack* _track);            // return process number of the mother track
  Int_t GetMCProcessMotherFromStack(const AliESDtrack* _track);   // return process number of the mother track
  
  Bool_t ConnectMCEvent();
  Bool_t UpdateStack();

  Bool_t IsMotherPdg(const AliDielectronPair* pair, Int_t pdgMother);
  Bool_t IsMotherPdg(const AliVParticle *particle1, const AliVParticle *particle2, Int_t pdgMother);
  Bool_t IsMCMotherToEE(const AliVParticle *particle, Int_t pdgMother);
  Bool_t IsMCTruth(const AliDielectronPair* pair, const AliDielectronSignalMC* signalMC) const;
  Bool_t IsMCTruth(Int_t label, AliDielectronSignalMC* signalMC, Int_t branch) const;
  Int_t GetMothersLabel(Int_t daughterLabel) const;
  Int_t GetPdgFromLabel(Int_t label) const;

  Bool_t IsPhysicalPrimary(Int_t label) const;  // checks if a particle is physical primary

  Bool_t HaveSameMother(const AliDielectronPair *pair) const;
  
  Int_t GetLabelMotherWithPdg(const AliDielectronPair* pair, Int_t pdgMother);
  Int_t GetLabelMotherWithPdg(const AliVParticle *particle1, const AliVParticle *particle2, Int_t pdgMother);
  
//   AliVParticle* GetMCTrackFromMCEvent(const AliVParticle *track);   // return MC track directly from MC event
  AliVParticle* GetMCTrackFromMCEvent(Int_t label) const;           // return MC track directly from MC event
  TParticle* GetMCTrackFromStack(const AliESDtrack* _track);        // return MC track from stack
  AliMCParticle* GetMCTrack(const AliESDtrack* _track);             // return MC track associated with reco track
  AliAODMCParticle* GetMCTrack( const AliAODTrack* _track);          // return MC track associated with reco AOD track
  
  TParticle* GetMCTrackMotherFromStack(const AliESDtrack* _track);  // return MC mother track from stack
  AliMCParticle* GetMCTrackMother(const AliESDtrack* _track);       // return MC mother track from stack
  AliAODMCParticle* GetMCTrackMother(const AliAODTrack* _track);       // return MC mother fot track AODTrack
  AliMCParticle* GetMCTrackMother(const AliMCParticle* _particle);       // return MC mother track from stack
  AliAODMCParticle* GetMCTrackMother(const AliAODMCParticle* _particle);       // return MC mother track from stack

  Int_t NumberOfDaughters(const AliESDtrack* track);                 // return number of daughters
  Int_t NumberOfDaughters(const AliAODTrack* track);                 // return number of daughters for AOD track
  Int_t NumberOfDaughters(const AliMCParticle* particle);                 // return number of daughters
  Int_t NumberOfDaughters(const AliAODMCParticle* particle);                 // return number of daughters

  void GetDaughters(const TObject *mother, AliVParticle* &d1, AliVParticle* &d2);
  Int_t IsJpsiPrimary(const AliDielectronPair * pair);
  Int_t IsJpsiPrimary(const AliVParticle * pair);
  Bool_t CheckParticleSource(Int_t label, AliDielectronSignalMC::ESource source) const;
  Bool_t GetPrimaryVertex(Double_t &primVtxX, Double_t &primVtxY, Double_t &primVtxZ);

  AliMCEvent* GetMCEvent() { return fMCEvent; }         // return the AliMCEvent
  
private:
  AliMCEvent    *fMCEvent;  // MC event object
  AliStack      *fStack;    // MC stack

  AnalysisType fAnaType;    // Analysis type
  Bool_t fHasMC;            // Do we have an MC handler?
  
  static AliDielectronMC* fgInstance; //! singleton pointer
  TClonesArray* fMcArray; //mcArray for AOD MC particles 

 
  AliDielectronMC(const AliDielectronMC &c);
  AliDielectronMC &operator=(const AliDielectronMC &c);

  Bool_t IsMCMotherToEEesd(const AliMCParticle *particle, Int_t pdgMother);
  Bool_t IsMCMotherToEEaod(const AliAODMCParticle *particle, Int_t pdgMother);

  Int_t GetLabelMotherWithPdgESD(const AliVParticle *particle1, const AliVParticle *particle2, Int_t pdgMother);
  Int_t GetLabelMotherWithPdgAOD(const AliVParticle *particle1, const AliVParticle *particle2, Int_t pdgMother);
  
  Bool_t ComparePDG(Int_t particlePDG, Int_t requiredPDG, Bool_t pdgExclusion, Bool_t checkBothCharges) const;
  Bool_t CheckIsRadiative(Int_t label) const;
  Bool_t CheckRadiativeDecision(Int_t mLabel, const AliDielectronSignalMC * const signalMC) const;

  ClassDef(AliDielectronMC, 0)
};

//
// inline functions
//
inline Bool_t AliDielectronMC::IsMotherPdg(const AliDielectronPair* pair, Int_t pdgMother)
{
  return IsMotherPdg(pair->GetFirstDaughter(),pair->GetSecondDaughter(),pdgMother);
}
//___________________________________________________________
inline Bool_t AliDielectronMC::IsMotherPdg(const AliVParticle *particle1, const AliVParticle *particle2, Int_t pdgMother){
  return GetLabelMotherWithPdg(particle1,particle2,pdgMother)>=0;
}
//___________________________________________________________
inline Int_t AliDielectronMC::GetLabelMotherWithPdg(const AliDielectronPair* pair, Int_t pdgMother){
  return GetLabelMotherWithPdg(pair->GetFirstDaughter(),pair->GetSecondDaughter(),pdgMother);
}

#endif
Commit	Line	Data
b2a297fa	1	#ifndef ALIDIELECTRONMC_H
	2	#define ALIDIELECTRONMC_H
	3	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	//#####################################################
	7	//# #
	8	//# Class AliDielectronMC #
	9	//# Cut Class for Jpsi->e+e- analysis #
	10	//# #
	11	//# by WooJin J. Park, GSI / W.J.Park@gsi.de #
	12	//# #
	13	//#####################################################
	14
	15	#ifndef ROOT_TObject
	16	#include <TObject.h>
	17	#endif
	18	class AliESDEvent;
	19	class AliHFEpid;
	20	class AliStack;
	21	class AliMCEvent;
	22	class AliESDtrack;
fb7d2d99	23	class AliAODTrack;
b2a297fa	24	class TParticle;
	25	class AliMCParticle;
	26	class AliAODMCParticle;
4533e78e	27	class AliAODMCHeader;
b2a297fa	28
ba15fdfb	29	#include "AliDielectronSignalMC.h"
b2a297fa	30	#include "AliDielectronPair.h"
	31
	32	class AliDielectronMC : public TObject{
	33
	34	public:
	35	enum AnalysisType {kUNSET=0, kESD, kAOD};
	36
	37	AliDielectronMC(AnalysisType type=kUNSET);
	38	virtual ~AliDielectronMC();
	39
572b0139	40	void SetHasMC(Bool_t hasMC) { fHasMC=hasMC; }
	41	Bool_t HasMC() const { return fHasMC; }
	42
b2a297fa	43	static AliDielectronMC* Instance();
	44
	45	void Initialize(); // initialization
	46	Int_t GetNMCTracks(); // return number of generated tracks
	47	Int_t GetNMCTracksFromStack(); // return number of generated tracks from stack
c8f0f810	48	Int_t GetNPrimary() const; // return number of primary tracks
2a14a7b1	49	Int_t GetNPrimaryFromStack(); // return number of primary tracks from stack
8df8e382	50	Int_t GetMCPID(const AliESDtrack* _track); // return MC PID
fb7d2d99	51	Int_t GetMCPID(const AliAODTrack* _track); // return MC PID for AODtrack
8df8e382	52	Int_t GetMCPIDFromStack(const AliESDtrack* _track); // return MC PID
8df8e382	53	Int_t GetMotherPDG(const AliESDtrack* _track); // return mother PID from the MC stack
fb7d2d99	54	Int_t GetMotherPDG(const AliAODTrack* _track); // return mother PID from the MC stack
d191916c	55	Int_t GetMotherPDG(const AliMCParticle* _track); // return mother PID from the MC stack
d191916c	56	Int_t GetMotherPDG(const AliAODMCParticle* _track); // return mother PID from the MC stack
8df8e382	57	Int_t GetMotherPDGFromStack(const AliESDtrack* _track); // return mother PID from the MC stack
	58	Int_t GetMCProcess(const AliESDtrack* _track); // return process number
	59	Int_t GetMCProcessFromStack(const AliESDtrack* _track); // return process number
	60	Int_t GetMCProcessMother(const AliESDtrack* _track); // return process number of the mother track
	61	Int_t GetMCProcessMotherFromStack(const AliESDtrack* _track); // return process number of the mother track
b2a297fa	62
	63	Bool_t ConnectMCEvent();
	64	Bool_t UpdateStack();
	65
a655b716	66	Bool_t IsMotherPdg(const AliDielectronPair* pair, Int_t pdgMother);
b2a297fa	67	Bool_t IsMotherPdg(const AliVParticle particle1, const AliVParticle particle2, Int_t pdgMother);
b2a297fa	68	Bool_t IsMCMotherToEE(const AliVParticle *particle, Int_t pdgMother);
c8f0f810	69	Bool_t IsMCTruth(const AliDielectronPair* pair, const AliDielectronSignalMC* signalMC) const;
d191916c	70	Bool_t IsMCTruth(Int_t label, AliDielectronSignalMC* signalMC, Int_t branch) const;
c8f0f810	71	Int_t GetMothersLabel(Int_t daughterLabel) const;
c8f0f810	72	Int_t GetPdgFromLabel(Int_t label) const;
8df8e382	73
5720c765	74	Bool_t IsPhysicalPrimary(Int_t label) const; // checks if a particle is physical primary
5720c765	75
c8f0f810	76	Bool_t HaveSameMother(const AliDielectronPair *pair) const;
b2a297fa	77
a655b716	78	Int_t GetLabelMotherWithPdg(const AliDielectronPair* pair, Int_t pdgMother);
	79	Int_t GetLabelMotherWithPdg(const AliVParticle particle1, const AliVParticle particle2, Int_t pdgMother);
	80
5720c765	81	// AliVParticle* GetMCTrackFromMCEvent(const AliVParticle *track); // return MC track directly from MC event
d191916c	82	AliVParticle* GetMCTrackFromMCEvent(Int_t label) const; // return MC track directly from MC event
8df8e382	83	TParticle* GetMCTrackFromStack(const AliESDtrack* _track); // return MC track from stack
8df8e382	84	AliMCParticle* GetMCTrack(const AliESDtrack* _track); // return MC track associated with reco track
fb7d2d99	85	AliAODMCParticle* GetMCTrack( const AliAODTrack* _track); // return MC track associated with reco AOD track
8df8e382	86
	87	TParticle* GetMCTrackMotherFromStack(const AliESDtrack* _track); // return MC mother track from stack
	88	AliMCParticle* GetMCTrackMother(const AliESDtrack* _track); // return MC mother track from stack
fb7d2d99	89	AliAODMCParticle* GetMCTrackMother(const AliAODTrack* _track); // return MC mother fot track AODTrack
8df8e382	90	AliMCParticle* GetMCTrackMother(const AliMCParticle* _particle); // return MC mother track from stack
	91	AliAODMCParticle* GetMCTrackMother(const AliAODMCParticle* _particle); // return MC mother track from stack
	92
	93	Int_t NumberOfDaughters(const AliESDtrack* track); // return number of daughters
fb7d2d99	94	Int_t NumberOfDaughters(const AliAODTrack* track); // return number of daughters for AOD track
8df8e382	95	Int_t NumberOfDaughters(const AliMCParticle* particle); // return number of daughters
8df8e382	96	Int_t NumberOfDaughters(const AliAODMCParticle* particle); // return number of daughters
6551594b	97
6551594b	98	void GetDaughters(const TObject mother, AliVParticle &d1, AliVParticle* &d2);
fb7d2d99	99	Int_t IsJpsiPrimary(const AliDielectronPair * pair);
fb7d2d99	100	Int_t IsJpsiPrimary(const AliVParticle * pair);
1bb1fef1	101	Bool_t CheckParticleSource(Int_t label, AliDielectronSignalMC::ESource source) const;
4533e78e	102	Bool_t GetPrimaryVertex(Double_t &primVtxX, Double_t &primVtxY, Double_t &primVtxZ);
ba15fdfb	103
ba15fdfb	104	AliMCEvent* GetMCEvent() { return fMCEvent; } // return the AliMCEvent
b2a297fa	105
	106	private:
	107	AliMCEvent *fMCEvent; // MC event object
	108	AliStack *fStack; // MC stack
	109
	110	AnalysisType fAnaType; // Analysis type
572b0139	111	Bool_t fHasMC; // Do we have an MC handler?
b2a297fa	112
b2a297fa	113	static AliDielectronMC* fgInstance; //! singleton pointer
fb7d2d99	114	TClonesArray* fMcArray; //mcArray for AOD MC particles
	115
	116
b2a297fa	117	AliDielectronMC(const AliDielectronMC &c);
	118	AliDielectronMC &operator=(const AliDielectronMC &c);
	119
	120	Bool_t IsMCMotherToEEesd(const AliMCParticle *particle, Int_t pdgMother);
	121	Bool_t IsMCMotherToEEaod(const AliAODMCParticle *particle, Int_t pdgMother);
	122
a655b716	123	Int_t GetLabelMotherWithPdgESD(const AliVParticle particle1, const AliVParticle particle2, Int_t pdgMother);
a655b716	124	Int_t GetLabelMotherWithPdgAOD(const AliVParticle particle1, const AliVParticle particle2, Int_t pdgMother);
b2a297fa	125
5720c765	126	Bool_t ComparePDG(Int_t particlePDG, Int_t requiredPDG, Bool_t pdgExclusion, Bool_t checkBothCharges) const;
ee877a49	127	Bool_t CheckIsRadiative(Int_t label) const;
	128	Bool_t CheckRadiativeDecision(Int_t mLabel, const AliDielectronSignalMC * const signalMC) const;
	129
b2a297fa	130	ClassDef(AliDielectronMC, 0)
	131	};
	132
a655b716	133	//
	134	// inline functions
	135	//
	136	inline Bool_t AliDielectronMC::IsMotherPdg(const AliDielectronPair* pair, Int_t pdgMother)
b2a297fa	137	{
b2a297fa	138	return IsMotherPdg(pair->GetFirstDaughter(),pair->GetSecondDaughter(),pdgMother);
b2a297fa	139	}
a655b716	140	//___________________________________________________________
	141	inline Bool_t AliDielectronMC::IsMotherPdg(const AliVParticle particle1, const AliVParticle particle2, Int_t pdgMother){
	142	return GetLabelMotherWithPdg(particle1,particle2,pdgMother)>=0;
	143	}
	144	//___________________________________________________________
	145	inline Int_t AliDielectronMC::GetLabelMotherWithPdg(const AliDielectronPair* pair, Int_t pdgMother){
	146	return GetLabelMotherWithPdg(pair->GetFirstDaughter(),pair->GetSecondDaughter(),pdgMother);
	147	}
b2a297fa	148
fb7d2d99	149	#endif
fb7d2d99	150