1 /**************************************************************************
2 * Copyright(c) 1998-2010, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
18 /////////////////////////////////////////////////////////////
20 // Class for cuts on AOD reconstructed Lc->pKpi
22 // Author: A.Dainese, andrea.dainese@pd.infn.it
23 /////////////////////////////////////////////////////////////
25 #include <TDatabasePDG.h>
26 #include <Riostream.h>
28 #include "AliRDHFCutsLctopKpi.h"
29 #include "AliAODRecoDecayHF3Prong.h"
30 #include "AliRDHFCuts.h"
31 #include "AliAODTrack.h"
32 #include "AliESDtrack.h"
33 #include "AliKFParticle.h"
34 #include "AliESDVertex.h"
36 ClassImp(AliRDHFCutsLctopKpi)
38 //--------------------------------------------------------------------------
39 AliRDHFCutsLctopKpi::AliRDHFCutsLctopKpi(const char* name) :
46 // Default Constructor
50 TString varNames[13]={"inv. mass [GeV]",
53 "d0K [cm] lower limit!",
54 "d0Pi [cm] lower limit!",
58 "pM=Max{pT1,pT2,pT3} (GeV/c)",
62 "cut on pTpion [GeV/c]"};
63 Bool_t isUpperCut[13]={kTRUE,
77 SetVarNames(nvars,varNames,isUpperCut);
78 Bool_t forOpt[13]={kFALSE,
91 SetVarsForOpt(4,forOpt);
92 Float_t limits[2]={0,999999999.};
95 //--------------------------------------------------------------------------
96 AliRDHFCutsLctopKpi::AliRDHFCutsLctopKpi(const AliRDHFCutsLctopKpi &source) :
105 if(source.fPidObjprot) SetPidprot(source.fPidObjprot);
106 if(source.fPidObjpion) SetPidpion(source.fPidObjpion);
108 //--------------------------------------------------------------------------
109 AliRDHFCutsLctopKpi &AliRDHFCutsLctopKpi::operator=(const AliRDHFCutsLctopKpi &source)
112 // assignment operator
114 if(&source == this) return *this;
116 AliRDHFCuts::operator=(source);
117 SetPidprot(source.GetPidprot());
118 SetPidpion(source.GetPidpion());
122 //---------------------------------------------------------------------------
123 AliRDHFCutsLctopKpi::~AliRDHFCutsLctopKpi() {
125 // // Default Destructor
138 //---------------------------------------------------------------------------
139 void AliRDHFCutsLctopKpi::GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters) {
141 // Fills in vars the values of the variables
144 if(nvars!=fnVarsForOpt) {
145 printf("AliRDHFCutsLctopKpi::GetCutsVarsForOpt: wrong number of variables\n");
149 AliAODRecoDecayHF3Prong *dd = (AliAODRecoDecayHF3Prong*)d;
154 vars[iter]=dd->InvMassLcpKpi();
158 for(Int_t iprong=0;iprong<3;iprong++){
159 if(TMath::Abs(pdgdaughters[iprong])==321) {
160 vars[iter]=dd->PtProng(iprong);
166 for(Int_t iprong=0;iprong<3;iprong++){
167 if(TMath::Abs(pdgdaughters[iprong])==2212) {
168 vars[iter]=dd->PtProng(iprong);
174 for(Int_t iprong=0;iprong<3;iprong++){
175 if(TMath::Abs(pdgdaughters[iprong])==2212) {
176 vars[iter]=dd->Getd0Prong(iprong);
182 for(Int_t iprong=0;iprong<3;iprong++){
183 if(TMath::Abs(pdgdaughters[iprong])==211) {
184 vars[iter]=dd->Getd0Prong(iprong);
190 vars[iter]=dd->GetDist12toPrim();
194 vars[iter]=dd->GetSigmaVert();
198 vars[iter] = dd->DecayLength();
203 for(Int_t i=0;i<3;i++){
204 if(dd->PtProng(i)>ptmax)ptmax=dd->PtProng(i);
210 vars[iter]=dd->CosPointingAngle();
214 vars[iter]=dd->Getd0Prong(0)*dd->Getd0Prong(0)+dd->Getd0Prong(1)*dd->Getd0Prong(1)+dd->Getd0Prong(2)*dd->Getd0Prong(2);
218 vars[iter]=dd->GetDCA();
222 for(Int_t iprong=0;iprong<3;iprong++){
223 if(TMath::Abs(pdgdaughters[iprong])==211) {
224 vars[iter]=dd->PtProng(iprong);
231 //---------------------------------------------------------------------------
232 Int_t AliRDHFCutsLctopKpi::IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent *aod) {
238 cout<<"Cut matrice not inizialized. Exit..."<<endl;
242 AliAODRecoDecayHF3Prong* d=(AliAODRecoDecayHF3Prong*)obj;
245 cout<<"AliAODRecoDecayHF3Prong null"<<endl;
252 Int_t returnvaluePID=3;
254 if(d->Pt()<fMinPtCand) return 0;
255 if(d->Pt()>fMaxPtCand) return 0;
257 // selection on candidate
258 if(selectionLevel==AliRDHFCuts::kAll ||
259 selectionLevel==AliRDHFCuts::kCandidate) {
263 Int_t ptbin=PtBin(pt);
265 Double_t mLcpKpi,mLcpiKp;
266 Int_t okLcpKpi=1,okLcpiKp=1;
268 Double_t mLcPDG = TDatabasePDG::Instance()->GetParticle(4122)->Mass();
270 mLcpKpi=d->InvMassLcpKpi();
271 mLcpiKp=d->InvMassLcpiKp();
273 if(TMath::Abs(mLcpKpi-mLcPDG)>fCutsRD[GetGlobalIndex(0,ptbin)]) okLcpKpi = 0;
274 if(TMath::Abs(mLcpiKp-mLcPDG)>fCutsRD[GetGlobalIndex(0,ptbin)]) okLcpiKp = 0;
275 if(!okLcpKpi && !okLcpiKp) return 0;
277 if(TMath::Abs(d->PtProng(1)) < fCutsRD[GetGlobalIndex(1,ptbin)] || TMath::Abs(d->Getd0Prong(1))<fCutsRD[GetGlobalIndex(3,ptbin)]) return 0;//Kaon
278 if((TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(2,ptbin)]) || (TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(12,ptbin)])) okLcpKpi=0;
279 if((TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(2,ptbin)]) || (TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(12,ptbin)]))okLcpiKp=0;
280 if(!okLcpKpi && !okLcpiKp) return 0;
285 if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
286 if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
287 if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
289 Int_t valueTotTmp=CombinePIDCuts(valueTmp,returnvaluePID);
290 Int_t pdgs[3]={2212,321,211};
295 if(!d->GetOwnPrimaryVtx()){
296 AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
297 d->SetOwnPrimaryVtx(vtx1);
299 Double_t field=aod->GetMagneticField();
300 ReconstructKF(d,pdgs,field);
303 if(d->GetDist12toPrim()<fCutsRD[GetGlobalIndex(5,ptbin)]|| d->GetDist23toPrim()<fCutsRD[GetGlobalIndex(5,ptbin)]) return 0;
304 if(d->GetDist12toPrim()>1.) return 0;
305 if(d->GetDist23toPrim()>1.) return 0;
306 if(d->Getd0Prong(0)*d->Getd0Prong(1)<0. && d->Getd0Prong(2)*d->Getd0Prong(1)<0.) return 0;
309 if(d->GetSigmaVert()>fCutsRD[GetGlobalIndex(6,ptbin)]) return 0;
311 if(d->DecayLength()<fCutsRD[GetGlobalIndex(7,ptbin)]) return 0;
312 if(d->DecayLength()>0.5) return 0;
314 if(TMath::Abs(d->PtProng(0))<fCutsRD[GetGlobalIndex(8,ptbin)] && TMath::Abs(d->PtProng(1))<fCutsRD[GetGlobalIndex(8,ptbin)] && TMath::Abs(d->PtProng(2))<fCutsRD[GetGlobalIndex(8,ptbin)]) return 0;
315 if(d->CosPointingAngle()< fCutsRD[GetGlobalIndex(9,ptbin)]) return 0;
316 Double_t sum2=d->Getd0Prong(0)*d->Getd0Prong(0)+d->Getd0Prong(1)*d->Getd0Prong(1)+d->Getd0Prong(2)*d->Getd0Prong(2);
317 if(sum2<fCutsRD[GetGlobalIndex(10,ptbin)]) return 0;
320 for(Int_t i=0;i<3;i++) if(d->GetDCA(i)>fCutsRD[GetGlobalIndex(11,ptbin)]) return 0;
323 if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
324 if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
325 if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
329 if(fUsePID || selectionLevel==AliRDHFCuts::kPID) returnvaluePID = IsSelectedPID(d);
330 if(returnvaluePID==0) return 0;
332 // selection on daughter tracks
333 if(selectionLevel==AliRDHFCuts::kAll ||
334 selectionLevel==AliRDHFCuts::kTracks) {
335 if(!AreDaughtersSelected(d)) return 0;
339 Int_t returnvalueTot=CombinePIDCuts(returnvalue,returnvaluePID);
340 return returnvalueTot;
342 //---------------------------------------------------------------------------
343 Int_t AliRDHFCutsLctopKpi::IsSelectedPID(AliAODRecoDecayHF* obj) {
347 Int_t okLcpKpi=0,okLcpiKp=0;
349 Bool_t isPeriodd=fPidHF->GetOnePad();
350 Bool_t isMC=fPidHF->GetMC();
351 Bool_t ispion0=kTRUE,ispion2=kTRUE;
352 Bool_t isproton0=kFALSE,isproton2=kFALSE;
353 Bool_t iskaon1=kFALSE;
355 fPidObjprot->SetOnePad(kTRUE);
356 fPidObjpion->SetOnePad(kTRUE);
359 fPidObjprot->SetMC(kTRUE);
360 fPidObjpion->SetMC(kTRUE);
363 for(Int_t i=0;i<3;i++){
364 AliAODTrack *track=(AliAODTrack*)obj->GetDaughter(i);
368 Int_t isKaon=fPidHF->MakeRawPid(track,3);
371 if(fPidHF->MakeRawPid(track,2)>=1) iskaon1=kFALSE;
373 if(!iskaon1) return 0;
378 Int_t isProton=fPidObjprot->MakeRawPid(track,4);
380 if(fPidHF->MakeRawPid(track,2)>=1) isProton=-1;
381 if(fPidHF->MakeRawPid(track,3)>=1) isProton=-1;
384 Int_t isPion=fPidObjpion->MakeRawPid(track,2);
385 if(fPidHF->MakeRawPid(track,3)>=1) isPion=-1;
386 if(fPidObjprot->MakeRawPid(track,4)>=1) isPion=-1;
390 if(isPion<0) ispion0=kFALSE;
391 if(isProton>=1) isproton0=kTRUE;
394 if(!ispion0 && !isproton0) return 0;
396 if(isPion<0) ispion2=kFALSE;
397 if(isProton>=1) isproton2=kTRUE;
403 if(ispion2 && isproton0 && iskaon1) okLcpKpi=1;
404 if(ispion0 && isproton2 && iskaon1) okLcpiKp=1;
405 if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
406 if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
407 if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
411 //-----------------------
412 Int_t AliRDHFCutsLctopKpi::CombinePIDCuts(Int_t returnvalue, Int_t returnvaluePID) const {
414 Int_t returnvalueTot=0;
415 Int_t okLcpKpi=0,okLcpiKp=0;
416 if(returnvaluePID==1){
417 if(returnvalue==1 || returnvalue==3) okLcpKpi=1;
419 if(returnvaluePID==2){
420 if(returnvalue>=2) okLcpiKp=1;
422 if(returnvaluePID==3 && returnvalue>0){
423 if(returnvalue==1 || returnvalue==3) okLcpKpi=1;
424 if(returnvalue>=2) okLcpiKp=1;
427 if(okLcpKpi) returnvalueTot=1; //cuts passed as Lc->pKpi
428 if(okLcpiKp) returnvalueTot=2; //cuts passed as Lc->piKp
429 if(okLcpKpi && okLcpiKp) returnvalueTot=3; //cuts passed as both pKpi and piKp
430 return returnvalueTot;
432 //----------------------------------
433 void AliRDHFCutsLctopKpi::SetStandardCutsPP2010() {
435 SetName("LctopKpiProdCuts");
436 SetTitle("Production cuts for Lc analysis");
438 AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts("AliESDtrackCuts","default");
439 esdTrackCuts->SetRequireSigmaToVertex(kFALSE);
440 esdTrackCuts->SetRequireTPCRefit(kTRUE);
441 esdTrackCuts->SetMinNClustersTPC(70);
442 esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
443 AliESDtrackCuts::kAny);
444 esdTrackCuts->SetRequireITSRefit(kTRUE);
445 esdTrackCuts->SetMinNClustersITS(4);
446 esdTrackCuts->SetMinDCAToVertexXY(0.);
447 esdTrackCuts->SetEtaRange(-0.8,0.8);
448 esdTrackCuts->SetPtRange(0.3,1.e10);
449 AddTrackCuts(esdTrackCuts);
451 const Int_t nptbins=4;
452 const Int_t nvars=13;
454 ptbins=new Float_t[nptbins+1];
462 SetGlobalIndex(nvars,nptbins);
463 SetPtBins(nptbins+1,ptbins);
465 Float_t** prodcutsval;
466 prodcutsval=new Float_t*[nvars];
467 for(Int_t iv=0;iv<nvars;iv++){
468 prodcutsval[iv]=new Float_t[nptbins];
471 for(Int_t ipt=0;ipt<nptbins;ipt++){
472 prodcutsval[0][ipt]=0.18;
473 prodcutsval[1][ipt]=0.4;
474 prodcutsval[2][ipt]=0.5;
475 prodcutsval[3][ipt]=0.;
476 prodcutsval[4][ipt]=0.;
477 prodcutsval[5][ipt]=0.01;
478 prodcutsval[6][ipt]=0.06;
479 prodcutsval[7][ipt]=0.005;
480 prodcutsval[8][ipt]=0.;
481 prodcutsval[9][ipt]=0.;
482 prodcutsval[10][ipt]=0.;
483 prodcutsval[11][ipt]=0.05;
484 prodcutsval[12][ipt]=0.4;
486 SetCuts(nvars,nptbins,prodcutsval);
488 AliAODPidHF* pidObjK=new AliAODPidHF();
489 Double_t sigmasK[5]={3.,1.,1.,3.,2.};
490 pidObjK->SetSigma(sigmasK);
491 pidObjK->SetAsym(kTRUE);
492 pidObjK->SetMatch(1);
493 pidObjK->SetTPC(kTRUE);
494 pidObjK->SetTOF(kTRUE);
495 pidObjK->SetITS(kTRUE);
496 Double_t plimK[2]={0.5,0.8};
497 pidObjK->SetPLimit(plimK,2);
498 pidObjK->SetTOFdecide(kTRUE);
502 AliAODPidHF* pidObjpi=new AliAODPidHF();
503 pidObjpi->SetTPC(kTRUE);
504 Double_t sigmaspi[5]={3.,0.,0.,0.,0.};
505 pidObjpi->SetSigma(sigmaspi);
506 pidObjpi->SetTOFdecide(kTRUE);
507 SetPidpion(pidObjpi);
509 AliAODPidHF* pidObjp=new AliAODPidHF();
510 Double_t sigmasp[5]={3.,1.,1.,3.,2.};
511 pidObjp->SetSigma(sigmasp);
512 pidObjp->SetAsym(kTRUE);
513 pidObjp->SetMatch(1);
514 pidObjp->SetTPC(kTRUE);
515 pidObjp->SetTOF(kTRUE);
516 pidObjp->SetITS(kTRUE);
517 Double_t plimp[2]={1.,2.};
518 pidObjp->SetPLimit(plimp,2);
519 pidObjp->SetTOFdecide(kTRUE);
527 for(Int_t iiv=0;iiv<nvars;iiv++){
528 delete [] prodcutsval[iiv];
530 delete [] prodcutsval;
541 void AliRDHFCutsLctopKpi::SetStandardCutsPbPb2010() {
543 SetName("LctopKpiProdCuts");
544 SetTitle("Production cuts for Lc analysis");
546 AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts("AliESDtrackCuts","default");
548 esdTrackCuts->SetRequireTPCRefit(kTRUE);
549 esdTrackCuts->SetMinNClustersTPC(70);
550 esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
551 AliESDtrackCuts::kAny);
552 esdTrackCuts->SetRequireITSRefit(kTRUE);
553 esdTrackCuts->SetMinNClustersITS(4);
554 esdTrackCuts->SetMinDCAToVertexXYPtDep("0.0100*TMath::Max(0.,(1-TMath::Floor(TMath::Abs(pt)/2.)))");
555 esdTrackCuts->SetEtaRange(-0.8,0.8);
556 esdTrackCuts->SetMaxDCAToVertexXY(1.);
557 esdTrackCuts->SetMaxDCAToVertexZ(1.);
558 esdTrackCuts->SetPtRange(0.8,1.e10);
559 AddTrackCuts(esdTrackCuts);
561 const Int_t nptbins=4;
562 const Int_t nvars=13;
564 ptbins=new Float_t[nptbins+1];
572 SetGlobalIndex(nvars,nptbins);
573 SetPtBins(nptbins+1,ptbins);
575 Float_t** prodcutsval;
576 prodcutsval=new Float_t*[nvars];
577 for(Int_t iv=0;iv<nvars;iv++){
578 prodcutsval[iv]=new Float_t[nptbins];
581 for(Int_t ipt=0;ipt<nptbins;ipt++){
582 prodcutsval[0][ipt]=0.13;
583 prodcutsval[1][ipt]=0.5;
584 prodcutsval[2][ipt]=0.6;
585 prodcutsval[3][ipt]=0.;
586 prodcutsval[4][ipt]=0.;
587 prodcutsval[5][ipt]=0.01;
588 prodcutsval[6][ipt]=0.04;
589 prodcutsval[7][ipt]=0.006;
590 prodcutsval[8][ipt]=0.8;
591 prodcutsval[9][ipt]=0.3;
592 prodcutsval[10][ipt]=0.;
593 prodcutsval[11][ipt]=0.05;
594 prodcutsval[12][ipt]=0.4;
596 SetCuts(nvars,nptbins,prodcutsval);
598 AliAODPidHF* pidObjK=new AliAODPidHF();
599 Double_t sigmasK[5]={3.,1.,1.,3.,2.};
600 pidObjK->SetSigma(sigmasK);
601 pidObjK->SetAsym(kTRUE);
602 pidObjK->SetMatch(1);
603 pidObjK->SetTPC(kTRUE);
604 pidObjK->SetTOF(kTRUE);
605 pidObjK->SetITS(kTRUE);
606 Double_t plimK[2]={0.5,0.8};
607 pidObjK->SetPLimit(plimK,2);
611 AliAODPidHF* pidObjpi=new AliAODPidHF();
612 pidObjpi->SetTPC(kTRUE);
613 Double_t sigmaspi[5]={3.,0.,0.,0.,0.};
614 pidObjpi->SetSigma(sigmaspi);
615 SetPidpion(pidObjpi);
617 AliAODPidHF* pidObjp=new AliAODPidHF();
618 Double_t sigmasp[5]={3.,1.,1.,3.,2.};
619 pidObjp->SetSigma(sigmasp);
620 pidObjp->SetAsym(kTRUE);
621 pidObjp->SetMatch(1);
622 pidObjp->SetTPC(kTRUE);
623 pidObjp->SetTOF(kTRUE);
624 pidObjp->SetITS(kTRUE);
625 Double_t plimp[2]={1.,2.};
626 pidObjp->SetPLimit(plimp,2);
634 for(Int_t iiv=0;iiv<nvars;iiv++){
635 delete [] prodcutsval[iiv];
637 delete [] prodcutsval;
648 Bool_t AliRDHFCutsLctopKpi::ReconstructKF(AliAODRecoDecayHF3Prong *d,Int_t *pdgs,Double_t field) const{
650 Int_t nprongs=d->GetNProngs();
651 Int_t iprongs[nprongs];
652 for(Int_t i=0;i<nprongs;i++) iprongs[i]=i;
654 Double_t mass[2]={0.,0.};
656 AliKFParticle *decay=d->ApplyVertexingKF(iprongs,nprongs,pdgs,kTRUE,field,mass);
657 if(!decay) return kTRUE;
658 AliESDVertex *vertexESD = new AliESDVertex(decay->Parameters(),
659 decay->CovarianceMatrix(),
662 Double_t pos[3],cov[6],chi2perNDF;
663 vertexESD->GetXYZ(pos);
664 vertexESD->GetCovMatrix(cov);
665 chi2perNDF = vertexESD->GetChi2toNDF();
666 delete vertexESD; vertexESD=NULL;
667 AliAODVertex *vertexAOD = new AliAODVertex(pos,cov,chi2perNDF,0x0,-1,AliAODVertex::kUndef,nprongs);
668 d->SetSecondaryVtx(vertexAOD);