[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEsignalCuts.cxx

/**************************************************************************
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/
//
// Signal cuts
// Checks whether a particle (reconstructed or MC) is coming from MC Signal
// For more information see implementation file
//
// Autor:
//   Markus Fasel <M.Fasel@gsi.de>
//
#include <TClass.h>
#include <TMath.h>
#include <TParticle.h>
#include <TString.h>

#include "AliAODMCParticle.h"
#include "AliLog.h"
#include "AliMCEvent.h"
#include "AliMCParticle.h"
#include "AliVParticle.h"

#include "AliHFEsignalCuts.h"
#include "AliHFEmcQA.h"

ClassImp(AliHFEsignalCuts)

//____________________________________________________________
AliHFEsignalCuts::AliHFEsignalCuts():
  AliAnalysisCuts(),
  fMC(NULL),
  fMCQA(NULL)
{
  //
  // Dummy constructor
  //
}

//____________________________________________________________
AliHFEsignalCuts::AliHFEsignalCuts(const Char_t *name, const Char_t *title):
  AliAnalysisCuts(name, title),
  fMC(NULL),
  fMCQA(NULL)
{
  //
  // Default constructor
  //
  fMCQA = new AliHFEmcQA;
  if(fMCQA) fMCQA->Init();
}

//____________________________________________________________
AliHFEsignalCuts::AliHFEsignalCuts(const AliHFEsignalCuts &ref):
  AliAnalysisCuts(ref),
  fMC(ref.fMC),
  fMCQA(ref.fMCQA)
{
  //
  // Copy constructor
  //
}

//____________________________________________________________
AliHFEsignalCuts &AliHFEsignalCuts::operator=(const AliHFEsignalCuts &ref){
  //
  // Assignment operator
  //
  if(this != &ref){
    fMC = ref.fMC; 
    fMCQA = ref.fMCQA; 
  }
  return *this;
}

//____________________________________________________________
AliHFEsignalCuts::~AliHFEsignalCuts(){
  //
  // Destructor
  //
  if(fMCQA) delete fMCQA;
}

//____________________________________________________________
void AliHFEsignalCuts::SetMCEvent(AliMCEvent *mc){ 
  //
  // Set mc event
  //
  fMC = mc; 
  if(fMCQA) fMCQA->SetMCEvent(mc);
}

//____________________________________________________________
Bool_t AliHFEsignalCuts::IsSelected(TObject *o){
  //
  // Define signal as electron coming from charm or beauty
  // @TODO: Implement setter so that also either of them can be defined
  // as signal alone
  

  return IsCharmElectron(o) || IsBeautyElectron(o);
/*  
  //saving time?
  Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o));
  if(esources>0)printf("esources= %d\n",esources);
  if(esources == AliHFEmcQA::kDirectCharm || esources == AliHFEmcQA::kDirectBeauty || esources == AliHFEmcQA::kBeautyCharm)  // 1: direct D->e, 2: B->e 3: B->D->e
    return kTRUE;
  else
    return kFALSE;
*/

}

//____________________________________________________________
Bool_t AliHFEsignalCuts::IsCharmElectron(const TObject * const o) const {
  //
  // Check if mother is coming from Charm
  //
  Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o));
  if(esources == AliHFEmcQA::kDirectCharm)  // 1: direct D->e
    return kTRUE;
  else
    return kFALSE;
}

//____________________________________________________________
Bool_t AliHFEsignalCuts::IsBeautyElectron(const TObject * const o) const {
  //
  // Check if mother is coming from Beauty
  //
  Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o));
  if(esources == AliHFEmcQA::kDirectBeauty || esources == AliHFEmcQA::kBeautyCharm)  // 2: B->e 3: B->D->e
    return kTRUE;
  else
    return kFALSE;
}

//____________________________________________________________
Bool_t AliHFEsignalCuts::IsGammaElectron(const TObject * const o) const {
  //
  // Check for MC if the electron is coming from Gamma
  //
  Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o));
  if(esources == AliHFEmcQA::kGamma)  // 4: conversion electrons
    return kTRUE;
  else
    return kFALSE;
}

/*
//____________________________________________________________
Bool_t AliHFEsignalCuts::IsCharmElectron(const TObject * const o) const {
  //
  // Check if mother is coming from Charm
  //
  if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE;
  Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o)));
  AliDebug(1, Form("Mother PDG %d\n", motherpdg));

  if((motherpdg % 1000) / 100 == 4) return kTRUE;    // charmed meson, 3rd position in pdg code == 4
  if(motherpdg / 1000 == 4) return kTRUE;            // charmed baryon, 4th position in pdg code == 4
  AliDebug(1, "No Charm\n");
  return kFALSE;
}

//____________________________________________________________
Bool_t AliHFEsignalCuts::IsBeautyElectron(const TObject * const o) const {
  //
  // Check if mother is coming from Beauty
  //
  if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE;
  Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o)));
  AliDebug(1, Form("Mother PDG %d\n", motherpdg));

  if((motherpdg % 1000) / 100 == 5) return kTRUE;   // beauty meson, 3rd position in pdg code == 5
  if(motherpdg / 1000 == 5) return kTRUE;           // beauty baryon, 4th position in pdg code == 5   
  AliDebug(1, "No Beauty\n");
  return kFALSE;
}

//____________________________________________________________
Bool_t AliHFEsignalCuts::IsGammaElectron(const TObject * const o) const {
  //
  // Check for MC if the electron is coming from Gamma
  //
  if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE;
  Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o)));
  AliDebug(1, Form("Mother PDG %d\n", motherpdg));

  if(motherpdg!=22){
    AliDebug(1, "No Gamma");
    return kFALSE;
  } else { 
    AliDebug(1, "Gamma");
    return kTRUE;
  }
}
*/

//____________________________________________________________
Int_t AliHFEsignalCuts::GetMotherPDG(const AliVParticle * const track) const {
  //
  // Get Mother Pdg code for reconstructed respectively MC tracks
  // 
  if(!fMC){
    AliDebug(1, "No MC Event Available\n");
    return 0;
  }
  const AliVParticle *motherParticle = NULL, *mctrack = NULL;
  TString objectType = track->IsA()->GetName();
  if(objectType.CompareTo("AliESDtrack") == 0 || objectType.CompareTo("AliAODTrack") == 0){
    // Reconstructed track
    if(track->GetLabel())
      mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel()));
  } else {
    // MCParticle
    mctrack = track;
  }

  if(!mctrack) return 0;
  
  Int_t motherPDG = 0;
  if(TString(mctrack->IsA()->GetName()).CompareTo("AliMCParticle") == 0){
    // case MC Particle
    motherParticle = fMC->GetTrack((dynamic_cast<const AliMCParticle *>(mctrack)->Particle()->GetFirstMother()));
    if(motherParticle)
      motherPDG = TMath::Abs((dynamic_cast<const AliMCParticle *>(motherParticle))->Particle()->GetPdgCode());
  } else {
    // case AODMCParticle
    motherParticle = fMC->GetTrack((dynamic_cast<const AliAODMCParticle *>(mctrack))->GetMother());
    if(motherParticle)
      motherPDG = TMath::Abs((dynamic_cast<const AliAODMCParticle *>(motherParticle))->GetPdgCode());
  }
  return motherPDG;
}

//____________________________________________________________
Int_t AliHFEsignalCuts::GetTrackPDG(const AliVParticle * const track) const {
	//
	// Return PDG code of a particle itself
	//
  if(!fMC){
    AliDebug(1, "No MC Event Available\n");
    return 0;
  }
	TString sourcetype = track->IsA()->GetName();
	const AliVParticle *mctrack = NULL;
	if(!sourcetype.CompareTo("AliESDtrack") || !sourcetype.CompareTo("AliAODTrack")){
		mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel()));
	} else  mctrack = track;
	if(!mctrack) return 0;

	TString mctype = mctrack->IsA()->GetName();
	Int_t trackPdg = 0;
	if(!mctype.CompareTo("AliMCParticle")){
		const AliMCParticle *esdmc = dynamic_cast<const AliMCParticle *>(mctrack);
		trackPdg = esdmc->Particle()->GetPdgCode();
	} else {
		const AliAODMCParticle *aodmc = dynamic_cast< const AliAODMCParticle *>(mctrack);
		trackPdg = aodmc->GetPdgCode();
	}
	return trackPdg;
}

//____________________________________________________________
Int_t AliHFEsignalCuts::GetElecSource(const AliVParticle * const track) const {
	//
	// Return PDG code of a particle itself
	//
	
        if(!fMC){
          AliDebug(1, "No MC Event Available\n");
          return 0;
        }
        if(!fMCQA){
          AliDebug(1, "No MCQA Available\n");
          return 0;
        }

        TString sourcetype = track->IsA()->GetName();
        const AliVParticle *mctrack = NULL;
        TParticle *mcpart = NULL;
        if(!sourcetype.CompareTo("AliESDtrack") || !sourcetype.CompareTo("AliAODTrack")){
                mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel()));
        } else  mctrack = track;
        if(!mctrack) return 0;

        TString mctype = mctrack->IsA()->GetName();
        Int_t eSource = 0;
        if(!mctype.CompareTo("AliMCParticle")){
                const AliMCParticle *esdmc = dynamic_cast<const AliMCParticle *>(mctrack);
                mcpart = esdmc->Particle();
	        eSource=fMCQA->GetElecSource(mcpart);
        } else {
                return -1;
        }
          
        return eSource;
}
Commit	Line	Data
3a72645a	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15	//
	16	// Signal cuts
	17	// Checks whether a particle (reconstructed or MC) is coming from MC Signal
	18	// For more information see implementation file
	19	//
	20	// Autor:
	21	// Markus Fasel <M.Fasel@gsi.de>
	22	//
	23	#include <TClass.h>
	24	#include <TMath.h>
	25	#include <TParticle.h>
	26	#include <TString.h>
	27
	28	#include "AliAODMCParticle.h"
	29	#include "AliLog.h"
	30	#include "AliMCEvent.h"
	31	#include "AliMCParticle.h"
	32	#include "AliVParticle.h"
	33
	34	#include "AliHFEsignalCuts.h"
	35	#include "AliHFEmcQA.h"
	36
	37	ClassImp(AliHFEsignalCuts)
	38
	39	//____________________________________________________________
	40	AliHFEsignalCuts::AliHFEsignalCuts():
	41	AliAnalysisCuts(),
	42	fMC(NULL),
	43	fMCQA(NULL)
	44	{
	45	//
	46	// Dummy constructor
	47	//
	48	}
	49
	50	//____________________________________________________________
	51	AliHFEsignalCuts::AliHFEsignalCuts(const Char_t name, const Char_t title):
	52	AliAnalysisCuts(name, title),
	53	fMC(NULL),
	54	fMCQA(NULL)
	55	{
	56	//
	57	// Default constructor
	58	//
	59	fMCQA = new AliHFEmcQA;
	60	if(fMCQA) fMCQA->Init();
	61	}
	62
	63	//____________________________________________________________
	64	AliHFEsignalCuts::AliHFEsignalCuts(const AliHFEsignalCuts &ref):
65	AliAnalysisCuts(ref),
66	fMC(ref.fMC),
67	fMCQA(ref.fMCQA)
68	{
69	//
70	// Copy constructor
71	//
72	}
73
74	//____________________________________________________________
75	AliHFEsignalCuts &AliHFEsignalCuts::operator=(const AliHFEsignalCuts &ref){
76	//
77	// Assignment operator
78	//
79	if(this != &ref){
80	fMC = ref.fMC;
81	fMCQA = ref.fMCQA;
82	}
83	return *this;
84	}
85
86	//____________________________________________________________
87	AliHFEsignalCuts::~AliHFEsignalCuts(){
88	//
89	// Destructor
90	//
91	if(fMCQA) delete fMCQA;
92	}
93
94	//____________________________________________________________
95	void AliHFEsignalCuts::SetMCEvent(AliMCEvent *mc){
96	//
97	// Set mc event
98	//
99	fMC = mc;
100	if(fMCQA) fMCQA->SetMCEvent(mc);
101	}
102
103	//____________________________________________________________
104	Bool_t AliHFEsignalCuts::IsSelected(TObject *o){
105	//
106	// Define signal as electron coming from charm or beauty
107	// @TODO: Implement setter so that also either of them can be defined
108	// as signal alone
109
110
111	return IsCharmElectron(o) \|\| IsBeautyElectron(o);
112	/*
113	//saving time?
114	Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o));
115	if(esources>0)printf("esources= %d\n",esources);
116	if(esources == AliHFEmcQA::kDirectCharm \|\| esources == AliHFEmcQA::kDirectBeauty \|\| esources == AliHFEmcQA::kBeautyCharm) // 1: direct D->e, 2: B->e 3: B->D->e
117	return kTRUE;
118	else
119	return kFALSE;
120	*/
121
122	}
123
124	//____________________________________________________________
125	Bool_t AliHFEsignalCuts::IsCharmElectron(const TObject * const o) const {
126	//
127	// Check if mother is coming from Charm
128	//
129	Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o));
130	if(esources == AliHFEmcQA::kDirectCharm) // 1: direct D->e
131	return kTRUE;
132	else
133	return kFALSE;
134	}
135
136	//____________________________________________________________
137	Bool_t AliHFEsignalCuts::IsBeautyElectron(const TObject * const o) const {
138	//
139	// Check if mother is coming from Beauty
140	//
141	Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o));
142	if(esources == AliHFEmcQA::kDirectBeauty \|\| esources == AliHFEmcQA::kBeautyCharm) // 2: B->e 3: B->D->e
143	return kTRUE;
144	else
145	return kFALSE;
146	}
147
148	//____________________________________________________________
149	Bool_t AliHFEsignalCuts::IsGammaElectron(const TObject * const o) const {
150	//
151	// Check for MC if the electron is coming from Gamma
152	//
153	Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o));
154	if(esources == AliHFEmcQA::kGamma) // 4: conversion electrons
155	return kTRUE;
156	else
157	return kFALSE;
158	}
159
160	/*
161	//____________________________________________________________
162	Bool_t AliHFEsignalCuts::IsCharmElectron(const TObject * const o) const {
163	//
164	// Check if mother is coming from Charm
165	//
166	if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE;
167	Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o)));
168	AliDebug(1, Form("Mother PDG %d\n", motherpdg));
169
170	if((motherpdg % 1000) / 100 == 4) return kTRUE; // charmed meson, 3rd position in pdg code == 4
171	if(motherpdg / 1000 == 4) return kTRUE; // charmed baryon, 4th position in pdg code == 4
172	AliDebug(1, "No Charm\n");
173	return kFALSE;
174	}
175
176	//____________________________________________________________
177	Bool_t AliHFEsignalCuts::IsBeautyElectron(const TObject * const o) const {
178	//
179	// Check if mother is coming from Beauty
180	//
181	if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE;
182	Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o)));
183	AliDebug(1, Form("Mother PDG %d\n", motherpdg));
184
185	if((motherpdg % 1000) / 100 == 5) return kTRUE; // beauty meson, 3rd position in pdg code == 5
186	if(motherpdg / 1000 == 5) return kTRUE; // beauty baryon, 4th position in pdg code == 5
187	AliDebug(1, "No Beauty\n");
188	return kFALSE;
189	}
190
191	//____________________________________________________________
192	Bool_t AliHFEsignalCuts::IsGammaElectron(const TObject * const o) const {
193	//
194	// Check for MC if the electron is coming from Gamma
195	//
196	if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE;
197	Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o)));
198	AliDebug(1, Form("Mother PDG %d\n", motherpdg));
199
200	if(motherpdg!=22){
201	AliDebug(1, "No Gamma");
202	return kFALSE;
203	} else {
204	AliDebug(1, "Gamma");
205	return kTRUE;
206	}
207	}
208	*/
209
210	//____________________________________________________________
211	Int_t AliHFEsignalCuts::GetMotherPDG(const AliVParticle * const track) const {
212	//
213	// Get Mother Pdg code for reconstructed respectively MC tracks
214	//
215	if(!fMC){
216	AliDebug(1, "No MC Event Available\n");
217	return 0;
218	}
219	const AliVParticle motherParticle = NULL, mctrack = NULL;
220	TString objectType = track->IsA()->GetName();
221	if(objectType.CompareTo("AliESDtrack") == 0 \|\| objectType.CompareTo("AliAODTrack") == 0){
222	// Reconstructed track
223	if(track->GetLabel())
224	mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel()));
225	} else {
226	// MCParticle
227	mctrack = track;
228	}
229
230	if(!mctrack) return 0;
231
232	Int_t motherPDG = 0;
233	if(TString(mctrack->IsA()->GetName()).CompareTo("AliMCParticle") == 0){
234	// case MC Particle
235	motherParticle = fMC->GetTrack((dynamic_cast<const AliMCParticle *>(mctrack)->Particle()->GetFirstMother()));
236	if(motherParticle)
237	motherPDG = TMath::Abs((dynamic_cast<const AliMCParticle *>(motherParticle))->Particle()->GetPdgCode());
238	} else {
239	// case AODMCParticle
240	motherParticle = fMC->GetTrack((dynamic_cast<const AliAODMCParticle *>(mctrack))->GetMother());
241	if(motherParticle)
242	motherPDG = TMath::Abs((dynamic_cast<const AliAODMCParticle *>(motherParticle))->GetPdgCode());
243	}
244	return motherPDG;
245	}
246
247	//____________________________________________________________
248	Int_t AliHFEsignalCuts::GetTrackPDG(const AliVParticle * const track) const {
249	//
250	// Return PDG code of a particle itself
251	//
252	if(!fMC){
253	AliDebug(1, "No MC Event Available\n");
254	return 0;
255	}
256	TString sourcetype = track->IsA()->GetName();
257	const AliVParticle *mctrack = NULL;
258	if(!sourcetype.CompareTo("AliESDtrack") \|\| !sourcetype.CompareTo("AliAODTrack")){
259	mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel()));
260	} else mctrack = track;
261	if(!mctrack) return 0;
262
263	TString mctype = mctrack->IsA()->GetName();
264	Int_t trackPdg = 0;
265	if(!mctype.CompareTo("AliMCParticle")){
266	const AliMCParticle esdmc = dynamic_cast<const AliMCParticle >(mctrack);
267	trackPdg = esdmc->Particle()->GetPdgCode();
268	} else {
269	const AliAODMCParticle aodmc = dynamic_cast< const AliAODMCParticle >(mctrack);
270	trackPdg = aodmc->GetPdgCode();
271	}
272	return trackPdg;
273	}
274
275	//____________________________________________________________
276	Int_t AliHFEsignalCuts::GetElecSource(const AliVParticle * const track) const {
277	//
278	// Return PDG code of a particle itself
279	//
280
281	if(!fMC){
282	AliDebug(1, "No MC Event Available\n");
283	return 0;
284	}
285	if(!fMCQA){
286	AliDebug(1, "No MCQA Available\n");
287	return 0;
288	}
289
290	TString sourcetype = track->IsA()->GetName();
291	const AliVParticle *mctrack = NULL;
292	TParticle *mcpart = NULL;
293	if(!sourcetype.CompareTo("AliESDtrack") \|\| !sourcetype.CompareTo("AliAODTrack")){
294	mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel()));
295	} else mctrack = track;
296	if(!mctrack) return 0;
297
298	TString mctype = mctrack->IsA()->GetName();
299	Int_t eSource = 0;
300	if(!mctype.CompareTo("AliMCParticle")){
301	const AliMCParticle esdmc = dynamic_cast<const AliMCParticle >(mctrack);
302	mcpart = esdmc->Particle();
303	eSource=fMCQA->GetElecSource(mcpart);
304	} else {
305	return -1;
306	}
307
308	return eSource;
309	}