[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / AliRsnCutDaughterD0.cxx

//
// All cuts for single track in D0 analysis,
// based on track quality and particle identification
// with TPC and TOF.
// Author: Massimo Venaruzzo
//
//

#include <Riostream.h>

#include "AliPID.h"
#include "AliPIDResponse.h"
#include "AliRsnCutDaughterD0.h"

ClassImp(AliRsnCutDaughterD0)

//__________________________________________________________________________________________________
AliRsnCutDaughterD0::AliRsnCutDaughterD0(const char *name, AliPID::EParticleType pid) :
AliRsnCut(name, AliRsnTarget::kDaughter),
  fNoPID(kFALSE),
  fPID(pid),
  fCutQuality(Form("%sQuality", name)),
  fPionTPCPIDCut(3.0),
  fKaonTPCPIDCut(3.0),
  fPionTOFPIDCut(3.0),
  fKaonTOFPIDCut(3.0),
  fPtDepPIDCut(kFALSE) 
{
  //
  // Constructor
  // Initialize track quality cuts to 2010 defaults
  //
  fCutQuality.SetPtRange(0.15, 1E+20);
  fCutQuality.SetEtaRange(-0.8, 0.8);
  fCutQuality.SetDCARPtFormula("0.0105+0.0350/pt^1.1");
  fCutQuality.SetDCARmin(0.0);
  fCutQuality.SetDCAZmax(2.0);
  fCutQuality.SetSPDminNClusters(1);
  fCutQuality.SetITSminNClusters(0);
  fCutQuality.SetITSmaxChi2(1E+20);
  fCutQuality.SetTPCminNClusters(70);
  fCutQuality.SetTPCmaxChi2(4.0);
  fCutQuality.SetRejectKinkDaughters();
  fCutQuality.SetAODTestFilterBit(5);
}

//__________________________________________________________________________________________________
Bool_t AliRsnCutDaughterD0::IsSelected(TObject *obj)
{
  //
  // Global check
  //

  // coherence check
  if (!TargetOK(obj)) return kFALSE;

  // check track
  AliVTrack *track = dynamic_cast<AliVTrack *>(fDaughter->GetRef());
  if (!track) return kFALSE;
   
  AliDebugClass(2, "Checking status...");
  // check flags
  if ((track->GetStatus() & AliESDtrack::kTPCin   ) == 0) return kFALSE;
  if ((track->GetStatus() & AliESDtrack::kTPCrefit) == 0) return kFALSE;
  if ((track->GetStatus() & AliESDtrack::kITSrefit) == 0) return kFALSE;
  AliDebugClass(2, "...passed");
  
  // quality
  AliDebugClass(2, "Checking quality cuts...");
  if (!fCutQuality.IsSelected(obj)) return kFALSE;
  AliDebugClass(2, "...passed");

  // if no PID is required, accept
  if (fNoPID) return kTRUE;
  
  // check initialization of PID object
  AliPIDResponse *pid = fEvent->GetPIDResponse();
  if (!pid) {
    AliFatal("NULL PID response");
    return kFALSE;
  }
  
  AliDebugClass(2, "Checking TOF Matching..."); 
  // check if TOF is matched
  // and computes all values used in the PID cut
  Bool_t   isTOF  = MatchTOF(track);   
  AliDebugClass(2, "...passed");
   
  Double_t pTPC   = track->GetTPCmomentum();
  Double_t p      = track->P();
  Double_t nsTPC  = TMath::Abs(pid->NumberOfSigmasTPC(track, fPID));
  Double_t nsTOF  = isTOF ? TMath::Abs(pid->NumberOfSigmasTOF(track, fPID)) : 1E20;
  Double_t maxTPC = 1E20;
  Double_t maxTOF = 1E20;
  AliDebugClass(2, "Checking PID...");

  if(!fPtDepPIDCut){
    // applies the cut differently depending on the PID and the momentum
    if (isTOF) {
      if (fPID == AliPID::kPion) {maxTPC = fPionTPCPIDCut; maxTOF = fPionTOFPIDCut;}
      if (fPID == AliPID::kKaon) {maxTPC = fKaonTPCPIDCut; maxTOF = fKaonTOFPIDCut;}
      return (nsTPC <= maxTPC && nsTOF <= maxTOF);
    } else {
      if (fPID == AliPID::kPion) maxTPC = fPionTPCPIDCut;
      if (fPID == AliPID::kKaon) maxTPC = fKaonTPCPIDCut;
      return (nsTPC <= maxTPC); 
    }
  } else {
    // applies the cut differently depending on the PID and the momentum
    if (isTOF) {
      // TPC: 5sigma cut for all
      if (nsTPC > 5.0) return kFALSE;
      // TOF: 3sigma below 1.5 GeV, 2sigma above
      if (p < 1.5) maxTOF = 3.0; else maxTOF = 2.0;
      return (nsTOF <= maxTOF);
    } else {
      // TPC:
      // all   below   350         MeV: 5sigma
      // all   between 350 and 500 MeV: 3sigma
      // pions above   500         MeV: 2sigma
      // kaons between 500 and 700 MeV: 2sigma
      // kaons above   700         MeV: rejected
      if (pTPC <= 0.35)
	maxTPC = 5.0;
      else if (pTPC > 0.35 && pTPC <= 0.5)
	maxTPC = 3.0;
      else {
	if (fPID == AliPID::kPion)
	  maxTPC = 2.0;
	else if (fPID == AliPID::kKaon) {
	  if (pTPC <= 0.7)
	    maxTPC = 2.0;
	  else
	    return kFALSE;
	}
      }
      return (nsTPC <= maxTPC);
    } 
  }    
  
  AliDebugClass(2, "...passed"); 
  // if we reach this point, all checks were successful
  AliDebugClass(2, "Good Pion/Kaon Candidate Found!!");
}
Commit	Line	Data
3e6c5c16	1	//
	2	// All cuts for single track in D0 analysis,
	3	// based on track quality and particle identification
	4	// with TPC and TOF.
	5	// Author: Massimo Venaruzzo
	6	//
	7	//
	8
	9	#include <Riostream.h>
	10
	11	#include "AliPID.h"
	12	#include "AliPIDResponse.h"
	13	#include "AliRsnCutDaughterD0.h"
	14
	15	ClassImp(AliRsnCutDaughterD0)
	16
	17	//__________________________________________________________________________________________________
	18	AliRsnCutDaughterD0::AliRsnCutDaughterD0(const char *name, AliPID::EParticleType pid) :
	19	AliRsnCut(name, AliRsnTarget::kDaughter),
	20	fNoPID(kFALSE),
	21	fPID(pid),
	22	fCutQuality(Form("%sQuality", name)),
	23	fPionTPCPIDCut(3.0),
	24	fKaonTPCPIDCut(3.0),
	25	fPionTOFPIDCut(3.0),
8ed7b07f	26	fKaonTOFPIDCut(3.0),
8ed7b07f	27	fPtDepPIDCut(kFALSE)
3e6c5c16	28	{
	29	//
	30	// Constructor
	31	// Initialize track quality cuts to 2010 defaults
	32	//
3e6c5c16	33	fCutQuality.SetPtRange(0.15, 1E+20);
	34	fCutQuality.SetEtaRange(-0.8, 0.8);
	35	fCutQuality.SetDCARPtFormula("0.0105+0.0350/pt^1.1");
	36	fCutQuality.SetDCARmin(0.0);
	37	fCutQuality.SetDCAZmax(2.0);
	38	fCutQuality.SetSPDminNClusters(1);
	39	fCutQuality.SetITSminNClusters(0);
	40	fCutQuality.SetITSmaxChi2(1E+20);
	41	fCutQuality.SetTPCminNClusters(70);
	42	fCutQuality.SetTPCmaxChi2(4.0);
	43	fCutQuality.SetRejectKinkDaughters();
	44	fCutQuality.SetAODTestFilterBit(5);
	45	}
	46
	47	//__________________________________________________________________________________________________
	48	Bool_t AliRsnCutDaughterD0::IsSelected(TObject *obj)
	49	{
	50	//
	51	// Global check
	52	//
	53
	54	// coherence check
	55	if (!TargetOK(obj)) return kFALSE;
	56
	57	// check track
	58	AliVTrack track = dynamic_cast<AliVTrack >(fDaughter->GetRef());
	59	if (!track) return kFALSE;
	60
	61	AliDebugClass(2, "Checking status...");
	62	// check flags
	63	if ((track->GetStatus() & AliESDtrack::kTPCin ) == 0) return kFALSE;
	64	if ((track->GetStatus() & AliESDtrack::kTPCrefit) == 0) return kFALSE;
	65	if ((track->GetStatus() & AliESDtrack::kITSrefit) == 0) return kFALSE;
	66	AliDebugClass(2, "...passed");
	67
	68	// quality
	69	AliDebugClass(2, "Checking quality cuts...");
	70	if (!fCutQuality.IsSelected(obj)) return kFALSE;
	71	AliDebugClass(2, "...passed");
	72
	73	// if no PID is required, accept
	74	if (fNoPID) return kTRUE;
	75
	76	// check initialization of PID object
	77	AliPIDResponse *pid = fEvent->GetPIDResponse();
	78	if (!pid) {
	79	AliFatal("NULL PID response");
	80	return kFALSE;
	81	}
	82
	83	AliDebugClass(2, "Checking TOF Matching...");
	84	// check if TOF is matched
	85	// and computes all values used in the PID cut
	86	Bool_t isTOF = MatchTOF(track);
	87	AliDebugClass(2, "...passed");
	88
670857a6	89	Double_t pTPC = track->GetTPCmomentum();
670857a6	90	Double_t p = track->P();
3e6c5c16	91	Double_t nsTPC = TMath::Abs(pid->NumberOfSigmasTPC(track, fPID));
	92	Double_t nsTOF = isTOF ? TMath::Abs(pid->NumberOfSigmasTOF(track, fPID)) : 1E20;
	93	Double_t maxTPC = 1E20;
	94	Double_t maxTOF = 1E20;
	95	AliDebugClass(2, "Checking PID...");
	96
670857a6	97	if(!fPtDepPIDCut){
	98	// applies the cut differently depending on the PID and the momentum
	99	if (isTOF) {
	100	if (fPID == AliPID::kPion) {maxTPC = fPionTPCPIDCut; maxTOF = fPionTOFPIDCut;}
	101	if (fPID == AliPID::kKaon) {maxTPC = fKaonTPCPIDCut; maxTOF = fKaonTOFPIDCut;}
	102	return (nsTPC <= maxTPC && nsTOF <= maxTOF);
	103	} else {
	104	if (fPID == AliPID::kPion) maxTPC = fPionTPCPIDCut;
	105	if (fPID == AliPID::kKaon) maxTPC = fKaonTPCPIDCut;
	106	return (nsTPC <= maxTPC);
	107	}
3e6c5c16	108	} else {
670857a6	109	// applies the cut differently depending on the PID and the momentum
	110	if (isTOF) {
	111	// TPC: 5sigma cut for all
	112	if (nsTPC > 5.0) return kFALSE;
	113	// TOF: 3sigma below 1.5 GeV, 2sigma above
	114	if (p < 1.5) maxTOF = 3.0; else maxTOF = 2.0;
	115	return (nsTOF <= maxTOF);
	116	} else {
	117	// TPC:
	118	// all below 350 MeV: 5sigma
	119	// all between 350 and 500 MeV: 3sigma
	120	// pions above 500 MeV: 2sigma
	121	// kaons between 500 and 700 MeV: 2sigma
	122	// kaons above 700 MeV: rejected
	123	if (pTPC <= 0.35)
	124	maxTPC = 5.0;
	125	else if (pTPC > 0.35 && pTPC <= 0.5)
	126	maxTPC = 3.0;
	127	else {
	128	if (fPID == AliPID::kPion)
	129	maxTPC = 2.0;
	130	else if (fPID == AliPID::kKaon) {
	131	if (pTPC <= 0.7)
	132	maxTPC = 2.0;
	133	else
	134	return kFALSE;
	135	}
	136	}
	137	return (nsTPC <= maxTPC);
	138	}
	139	}
3e6c5c16	140
	141	AliDebugClass(2, "...passed");
	142	// if we reach this point, all checks were successful
	143	AliDebugClass(2, "Good Pion/Kaon Candidate Found!!");
	144	}