* provided "as is" without express or implied warranty. *
**************************************************************************/
+/* $Id$ */
+
/////////////////////////////////////////////////////////////
//
// Class for cuts on AOD reconstructed Lc->pKpi
#include "AliRDHFCutsLctopKpi.h"
#include "AliAODRecoDecayHF3Prong.h"
+#include "AliRDHFCuts.h"
#include "AliAODTrack.h"
#include "AliESDtrack.h"
+#include "AliKFParticle.h"
+#include "AliESDVertex.h"
ClassImp(AliRDHFCutsLctopKpi)
//--------------------------------------------------------------------------
AliRDHFCutsLctopKpi::AliRDHFCutsLctopKpi(const char* name) :
-AliRDHFCuts(name)
+AliRDHFCuts(name),
+fPidObjprot(0),
+fPidObjpion(0),
+fRecoKF(kFALSE),
+fUseImpParProdCorrCut(kFALSE)
{
//
// Default Constructor
//
- Int_t nvars=12;
+ Int_t nvars=13;
SetNVars(nvars);
- TString varNames[12]={"inv. mass [GeV]",
+ TString varNames[13]={"inv. mass [GeV]",
+ "pTK [GeV/c]",
"pTP [GeV/c]",
- "pTPi [GeV/c]",
- "d0P [cm] lower limit!",
+ "d0K [cm] lower limit!",
"d0Pi [cm] lower limit!",
"dist12 (cm)",
"sigmavert (cm)",
"pM=Max{pT1,pT2,pT3} (GeV/c)",
"cosThetaPoint",
"Sum d0^2 (cm^2)",
- "dca cut (cm)"};
- Bool_t isUpperCut[12]={kTRUE,
+ "dca cut (cm)",
+ "cut on pTpion [GeV/c]"};
+ Bool_t isUpperCut[13]={kTRUE,
kFALSE,
kFALSE,
kFALSE,
kFALSE,
kFALSE,
kFALSE,
- kTRUE};
+ kTRUE,
+ kFALSE
+ };
SetVarNames(nvars,varNames,isUpperCut);
- Bool_t forOpt[12]={kFALSE,
- kFALSE,
+ Bool_t forOpt[13]={kFALSE,
+ kTRUE,
+ kTRUE,
kFALSE,
kFALSE,
kFALSE,
kFALSE,
kTRUE,
- kTRUE,
- kTRUE,
- kTRUE,
- kTRUE,
- kFALSE};
- SetVarsForOpt(5,forOpt);
+ kFALSE,
+ kFALSE,
+ kFALSE,
+ kFALSE,
+ kTRUE};
+ SetVarsForOpt(4,forOpt);
Float_t limits[2]={0,999999999.};
SetPtBins(2,limits);
}
//--------------------------------------------------------------------------
AliRDHFCutsLctopKpi::AliRDHFCutsLctopKpi(const AliRDHFCutsLctopKpi &source) :
- AliRDHFCuts(source)
+ AliRDHFCuts(source),
+ fPidObjprot(0),
+ fPidObjpion(0),
+ fRecoKF(source.fRecoKF),
+ fUseImpParProdCorrCut(source.fUseImpParProdCorrCut)
{
//
// Copy constructor
//
-
+ if(source.fPidObjprot) SetPidprot(source.fPidObjprot);
+ if(source.fPidObjpion) SetPidpion(source.fPidObjpion);
}
//--------------------------------------------------------------------------
AliRDHFCutsLctopKpi &AliRDHFCutsLctopKpi::operator=(const AliRDHFCutsLctopKpi &source)
if(&source == this) return *this;
AliRDHFCuts::operator=(source);
+ SetPidprot(source.GetPidprot());
+ SetPidpion(source.GetPidpion());
return *this;
}
+//---------------------------------------------------------------------------
+AliRDHFCutsLctopKpi::~AliRDHFCutsLctopKpi() {
+ //
+ // // Default Destructor
+ //
+ if(fPidObjpion){
+ delete fPidObjpion;
+ fPidObjpion=0;
+ }
+ if(fPidObjprot){
+ delete fPidObjprot;
+ fPidObjprot=0;
+ }
+}
//---------------------------------------------------------------------------
-void AliRDHFCutsLctopKpi::GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters) {
+void AliRDHFCutsLctopKpi::GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters, AliAODEvent *aod) {
//
// Fills in vars the values of the variables
//
if(fVarsForOpt[1]){
iter++;
for(Int_t iprong=0;iprong<3;iprong++){
- if(TMath::Abs(pdgdaughters[iprong])==2212) {
+ if(TMath::Abs(pdgdaughters[iprong])==321) {
vars[iter]=dd->PtProng(iprong);
}
}
if(fVarsForOpt[2]){
iter++;
for(Int_t iprong=0;iprong<3;iprong++){
- if(TMath::Abs(pdgdaughters[iprong])==211) {
+ if(TMath::Abs(pdgdaughters[iprong])==2212) {
vars[iter]=dd->PtProng(iprong);
}
}
}
if(fVarsForOpt[6]){
iter++;
- vars[iter]=dd->GetSigmaVert();
+ vars[iter]=dd->GetSigmaVert(aod);
}
if(fVarsForOpt[7]){
iter++;
iter++;
vars[iter]=dd->GetDCA();
}
+ if(fVarsForOpt[12]){
+ iter++;
+ for(Int_t iprong=0;iprong<3;iprong++){
+ if(TMath::Abs(pdgdaughters[iprong])==211) {
+ vars[iter]=dd->PtProng(iprong);
+ }
+ }
+ }
return;
}
//---------------------------------------------------------------------------
-Int_t AliRDHFCutsLctopKpi::IsSelected(TObject* obj,Int_t selectionLevel) {
+Int_t AliRDHFCutsLctopKpi::IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent *aod) {
//
// Apply selection
//
}
- // selection on daughter tracks
- if(selectionLevel==AliRDHFCuts::kAll ||
- selectionLevel==AliRDHFCuts::kTracks) {
- if(!AreDaughtersSelected(d)) return 0;
- }
-
+ if(fKeepSignalMC) if(IsSignalMC(d,aod,4122)) return 3;
Int_t returnvalue=3;
+ Int_t returnvaluePID=3;
+
+ if(d->Pt()<fMinPtCand) return 0;
+ if(d->Pt()>fMaxPtCand) return 0;
+
+ if(d->HasBadDaughters()) return 0;
+
// selection on candidate
if(selectionLevel==AliRDHFCuts::kAll ||
if(!okLcpKpi && !okLcpiKp) return 0;
if(TMath::Abs(d->PtProng(1)) < fCutsRD[GetGlobalIndex(1,ptbin)] || TMath::Abs(d->Getd0Prong(1))<fCutsRD[GetGlobalIndex(3,ptbin)]) return 0;//Kaon
- //if(TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(2,ptbin)] || TMath::Abs(d->Getd0Prong(0))<fCutsRD[GetGlobalIndex(4,ptbin)]) return 0;//Proton
- //if(TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(2,ptbin)] || TMath::Abs(d->Getd0Prong(2))<fCutsRD[GetGlobalIndex(4,ptbin)]) return 0;//Pion
- if((TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(2,ptbin)]) || (TMath::Abs(d->PtProng(2)) < 0.4)) okLcpKpi=0;
- if((TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(2,ptbin)]) || (TMath::Abs(d->PtProng(0)) < 0.4))okLcpiKp=0;
- if(!okLcpKpi && !okLcpiKp) return 0;
+ if((TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(2,ptbin)]) || (TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(12,ptbin)])) okLcpKpi=0;
+ if((TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(2,ptbin)]) || (TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(12,ptbin)]))okLcpiKp=0;
+ if(!okLcpKpi && !okLcpiKp) return 0;
-
+ if(fRecoKF){
+ Int_t valueTmp=3;
+ if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
+ if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
+ if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
+
+ Int_t valueTotTmp=CombinePIDCuts(valueTmp,returnvaluePID);
+ Int_t pdgs[3]={2212,321,211};
+ if(valueTotTmp>=2) {
+ pdgs[0]=211;
+ pdgs[2]=2212;
+ }
+ if(!d->GetOwnPrimaryVtx()){
+ AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
+ d->SetOwnPrimaryVtx(vtx1);
+ }
+ Double_t field=aod->GetMagneticField();
+ Bool_t outKF=ReconstructKF(d,pdgs,field);
+ if(!outKF) return 0;
+ }
//2track cuts
if(d->GetDist12toPrim()<fCutsRD[GetGlobalIndex(5,ptbin)]|| d->GetDist23toPrim()<fCutsRD[GetGlobalIndex(5,ptbin)]) return 0;
if(d->GetDist12toPrim()>1.) return 0;
if(d->GetDist23toPrim()>1.) return 0;
- if(d->Getd0Prong(0)*d->Getd0Prong(1)<0. && d->Getd0Prong(2)*d->Getd0Prong(1)<0.) return 0;
-
- //sec vert
- if(d->GetSigmaVert()>fCutsRD[GetGlobalIndex(6,ptbin)]) return 0;
+ if(fUseImpParProdCorrCut){
+ if(d->Getd0Prong(0)*d->Getd0Prong(1)<0. && d->Getd0Prong(2)*d->Getd0Prong(1)<0.) return 0;
+ }
+
+ //sec vert
if(d->DecayLength()<fCutsRD[GetGlobalIndex(7,ptbin)]) return 0;
if(d->DecayLength()>0.5) return 0;
if(d->CosPointingAngle()< fCutsRD[GetGlobalIndex(9,ptbin)]) return 0;
Double_t sum2=d->Getd0Prong(0)*d->Getd0Prong(0)+d->Getd0Prong(1)*d->Getd0Prong(1)+d->Getd0Prong(2)*d->Getd0Prong(2);
if(sum2<fCutsRD[GetGlobalIndex(10,ptbin)]) return 0;
+ if(d->GetSigmaVert(aod)>fCutsRD[GetGlobalIndex(6,ptbin)]) return 0;
//DCA
for(Int_t i=0;i<3;i++) if(d->GetDCA(i)>fCutsRD[GetGlobalIndex(11,ptbin)]) return 0;
if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
-
+
}
- return returnvalue;
+ if(selectionLevel==AliRDHFCuts::kAll ||
+ selectionLevel==AliRDHFCuts::kCandidate||
+ selectionLevel==AliRDHFCuts::kPID) {
+ returnvaluePID = IsSelectedPID(d);
+ fIsSelectedPID=returnvaluePID;
+ }
+ // if(fUsePID || selectionLevel==AliRDHFCuts::kPID) returnvaluePID = IsSelectedCombinedPID(d); // to test!!
+ if(returnvaluePID==0) return 0;
+
+ // selection on daughter tracks
+ if(selectionLevel==AliRDHFCuts::kAll ||
+ selectionLevel==AliRDHFCuts::kTracks) {
+ if(!AreDaughtersSelected(d)) return 0;
+ }
+
+
+ Int_t returnvalueTot=CombinePIDCuts(returnvalue,returnvaluePID);
+ return returnvalueTot;
}
//---------------------------------------------------------------------------
+Int_t AliRDHFCutsLctopKpi::IsSelectedPID(AliAODRecoDecayHF* obj) {
+
+
+ if(!fUsePID || !obj) return 3;
+ Int_t okLcpKpi=0,okLcpiKp=0;
+ Int_t returnvalue=0;
+ Bool_t isPeriodd=fPidHF->GetOnePad();
+ Bool_t isMC=fPidHF->GetMC();
+ Bool_t ispion0=kTRUE,ispion2=kTRUE;
+ Bool_t isproton0=kFALSE,isproton2=kFALSE;
+ Bool_t iskaon1=kFALSE;
+ if(isPeriodd) {
+ fPidObjprot->SetOnePad(kTRUE);
+ fPidObjpion->SetOnePad(kTRUE);
+ }
+ if(isMC) {
+ fPidObjprot->SetMC(kTRUE);
+ fPidObjpion->SetMC(kTRUE);
+ }
+
+ for(Int_t i=0;i<3;i++){
+ AliAODTrack *track=(AliAODTrack*)obj->GetDaughter(i);
+ if(!track) return 0;
+ // identify kaon
+ if(i==1) {
+ Int_t isKaon=fPidHF->MakeRawPid(track,3);
+ if(isKaon>=1) {
+ iskaon1=kTRUE;
+ if(fPidHF->MakeRawPid(track,2)>=1) iskaon1=kFALSE;
+ }
+ if(!iskaon1) return 0;
+
+ }else{
+ //pion or proton
+
+ Int_t isProton=fPidObjprot->MakeRawPid(track,4);
+ if(isProton>=1){
+ if(fPidHF->MakeRawPid(track,2)>=1) isProton=-1;
+ if(fPidHF->MakeRawPid(track,3)>=1) isProton=-1;
+ }
+
+ Int_t isPion=fPidObjpion->MakeRawPid(track,2);
+ if(fPidHF->MakeRawPid(track,3)>=1) isPion=-1;
+ if(fPidObjprot->MakeRawPid(track,4)>=1) isPion=-1;
+
+
+ if(i==0) {
+ if(isPion<0) ispion0=kFALSE;
+ if(isProton>=1) isproton0=kTRUE;
+
+ }
+ if(!ispion0 && !isproton0) return 0;
+ if(i==2) {
+ if(isPion<0) ispion2=kFALSE;
+ if(isProton>=1) isproton2=kTRUE;
+ }
+
+ }
+ }
+
+ if(ispion2 && isproton0 && iskaon1) okLcpKpi=1;
+ if(ispion0 && isproton2 && iskaon1) okLcpiKp=1;
+ if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
+ if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
+ if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
+
+ return returnvalue;
+}
+//---------------------------------------------------------------------------
+Int_t AliRDHFCutsLctopKpi::IsSelectedCombinedPID(AliAODRecoDecayHF* obj) {
+
+ // Printf(" -------- IsSelectedCombinedPID --------------");
+
+ if(!obj) {return 3;}
+ Int_t okLcpKpi=0,okLcpiKp=0;
+ Int_t returnvalue=0;
+ Bool_t isPeriodd=fPidHF->GetOnePad();
+ Bool_t isMC=fPidHF->GetMC();
+ Bool_t isKaon = kFALSE;
+ Bool_t isPion = kFALSE;
+ Bool_t isProton = kFALSE;
+
+ if(isPeriodd) {
+ fPidObjprot->SetOnePad(kTRUE);
+ fPidObjpion->SetOnePad(kTRUE);
+ }
+ if(isMC) {
+ fPidObjprot->SetMC(kTRUE);
+ fPidObjpion->SetMC(kTRUE);
+ }
+
+ Double_t probComb[AliPID::kSPECIES]={0.}; // array to store the info for the combined ITS|TPC|TOF probabilities
+ fPidHF->GetPidCombined()->SetDetectorMask(AliPIDResponse::kDetITS|AliPIDResponse::kDetTOF|AliPIDResponse::kDetTPC); // the mask could become a member of the cut object
+
+ for(Int_t i=0;i<3;i++){
+ AliAODTrack *track=(AliAODTrack*)obj->GetDaughter(i);
+ if(!track) return 0;
+ // identify kaon
+ /*UInt_t detUsed =*/ fPidHF->GetPidCombined()->ComputeProbabilities(track, fPidHF->GetPidResponse(), probComb); // for the moment we don't check detUsed...
+ // Printf("[%d] %x %f %f %f %f %f",i,detUsed,probComb[0],probComb[1],probComb[2],probComb[3],probComb[4]);
+ if(i==1) {
+ if(TMath::MaxElement(AliPID::kSPECIES,probComb) == probComb[3]) { // the probability to be a Kaon is the highest
+ isKaon=kTRUE;
+ // if ( probComb[3] > 0.8 ) isKaon=kTRUE;
+ // else return 0;
+ }
+ else return 0; // prong at position 1 is not a Kaon, returning 0
+ }
+ else {
+ //pion or proton
+
+ if(TMath::MaxElement(AliPID::kSPECIES,probComb) == probComb[4]) { // the probability to be a proton is the highest
+ isProton=kTRUE;
+ if (isPion) okLcpiKp = 1; // the pion was already identified, so here we must be at i == 2 --> Lc --> pi K p (K already found)
+ }
+ if(TMath::MaxElement(AliPID::kSPECIES,probComb) == probComb[2]) { // the probability to be a pion is the highest
+ isPion=kTRUE;
+ if (isProton) okLcpKpi = 1; // the proton was already identified, so here we must be at i == 2 --> Lc --> p K pi (K already found)
+ }
+
+ }
+ }
+
+ if(okLcpKpi) returnvalue=1; //cuts passed as Lc->pKpi
+ if(okLcpiKp) returnvalue=2; //cuts passed as Lc->piKp
+ if(okLcpKpi && okLcpiKp) returnvalue=3; //cuts passed as both pKpi and piKp
+
+ return returnvalue;
+}
+//-----------------------
+Int_t AliRDHFCutsLctopKpi::CombinePIDCuts(Int_t returnvalue, Int_t returnvaluePID) const {
+
+ Int_t returnvalueTot=0;
+ Int_t okLcpKpi=0,okLcpiKp=0;
+ if(returnvaluePID==1){
+ if(returnvalue==1 || returnvalue==3) okLcpKpi=1;
+ }
+ if(returnvaluePID==2){
+ if(returnvalue>=2) okLcpiKp=1;
+ }
+ if(returnvaluePID==3 && returnvalue>0){
+ if(returnvalue==1 || returnvalue==3) okLcpKpi=1;
+ if(returnvalue>=2) okLcpiKp=1;
+ }
+
+ if(okLcpKpi) returnvalueTot=1; //cuts passed as Lc->pKpi
+ if(okLcpiKp) returnvalueTot=2; //cuts passed as Lc->piKp
+ if(okLcpKpi && okLcpiKp) returnvalueTot=3; //cuts passed as both pKpi and piKp
+ return returnvalueTot;
+}
+//----------------------------------
+void AliRDHFCutsLctopKpi::SetStandardCutsPP2010() {
+
+ SetName("LctopKpiProdCuts");
+ SetTitle("Production cuts for Lc analysis");
+
+ AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts("AliESDtrackCuts","default");
+ esdTrackCuts->SetRequireSigmaToVertex(kFALSE);
+ esdTrackCuts->SetRequireTPCRefit(kTRUE);
+ esdTrackCuts->SetMinNClustersTPC(70);
+ esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
+ AliESDtrackCuts::kAny);
+ esdTrackCuts->SetRequireITSRefit(kTRUE);
+ esdTrackCuts->SetMinNClustersITS(4);
+ esdTrackCuts->SetMinDCAToVertexXY(0.);
+ esdTrackCuts->SetEtaRange(-0.8,0.8);
+ esdTrackCuts->SetPtRange(0.3,1.e10);
+ AddTrackCuts(esdTrackCuts);
+
+ const Int_t nptbins=4;
+ const Int_t nvars=13;
+ Float_t* ptbins;
+ ptbins=new Float_t[nptbins+1];
+
+ ptbins[0]=0.;
+ ptbins[1]=2.;
+ ptbins[2]=3.;
+ ptbins[3]=4.;
+ ptbins[4]=9999.;
+
+ SetGlobalIndex(nvars,nptbins);
+ SetPtBins(nptbins+1,ptbins);
+
+ Float_t** prodcutsval;
+ prodcutsval=new Float_t*[nvars];
+ for(Int_t iv=0;iv<nvars;iv++){
+ prodcutsval[iv]=new Float_t[nptbins];
+ }
+
+ for(Int_t ipt=0;ipt<nptbins;ipt++){
+ prodcutsval[0][ipt]=0.18;
+ prodcutsval[1][ipt]=0.4;
+ prodcutsval[2][ipt]=0.5;
+ prodcutsval[3][ipt]=0.;
+ prodcutsval[4][ipt]=0.;
+ prodcutsval[5][ipt]=0.01;
+ prodcutsval[6][ipt]=0.06;
+ prodcutsval[7][ipt]=0.005;
+ prodcutsval[8][ipt]=0.;
+ prodcutsval[9][ipt]=0.;
+ prodcutsval[10][ipt]=0.;
+ prodcutsval[11][ipt]=0.05;
+ prodcutsval[12][ipt]=0.4;
+ }
+ SetCuts(nvars,nptbins,prodcutsval);
+
+ AliAODPidHF* pidObjK=new AliAODPidHF();
+ Double_t sigmasK[5]={3.,1.,1.,3.,2.};
+ pidObjK->SetSigma(sigmasK);
+ pidObjK->SetAsym(kTRUE);
+ pidObjK->SetMatch(1);
+ pidObjK->SetTPC(kTRUE);
+ pidObjK->SetTOF(kTRUE);
+ pidObjK->SetITS(kTRUE);
+ Double_t plimK[2]={0.5,0.8};
+ pidObjK->SetPLimit(plimK,2);
+ pidObjK->SetTOFdecide(kTRUE);
+
+ SetPidHF(pidObjK);
+
+ AliAODPidHF* pidObjpi=new AliAODPidHF();
+ pidObjpi->SetTPC(kTRUE);
+ Double_t sigmaspi[5]={3.,0.,0.,0.,0.};
+ pidObjpi->SetSigma(sigmaspi);
+ pidObjpi->SetTOFdecide(kTRUE);
+ SetPidpion(pidObjpi);
+
+ AliAODPidHF* pidObjp=new AliAODPidHF();
+ Double_t sigmasp[5]={3.,1.,1.,3.,2.};
+ pidObjp->SetSigma(sigmasp);
+ pidObjp->SetAsym(kTRUE);
+ pidObjp->SetMatch(1);
+ pidObjp->SetTPC(kTRUE);
+ pidObjp->SetTOF(kTRUE);
+ pidObjp->SetITS(kTRUE);
+ Double_t plimp[2]={1.,2.};
+ pidObjp->SetPLimit(plimp,2);
+ pidObjp->SetTOFdecide(kTRUE);
+
+ SetPidprot(pidObjp);
+
+ SetUsePID(kTRUE);
+
+ PrintAll();
+
+ for(Int_t iiv=0;iiv<nvars;iiv++){
+ delete [] prodcutsval[iiv];
+ }
+ delete [] prodcutsval;
+ prodcutsval=NULL;
+ delete [] ptbins;
+ ptbins=NULL;
+
+ delete pidObjp;
+ pidObjp=NULL;
+
+ return;
+}
+//------------------
+void AliRDHFCutsLctopKpi::SetStandardCutsPbPb2010() {
+
+ SetName("LctopKpiProdCuts");
+ SetTitle("Production cuts for Lc analysis");
+
+ AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts("AliESDtrackCuts","default");
+
+ esdTrackCuts->SetRequireTPCRefit(kTRUE);
+ esdTrackCuts->SetMinNClustersTPC(70);
+ esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
+ AliESDtrackCuts::kAny);
+ esdTrackCuts->SetRequireITSRefit(kTRUE);
+ esdTrackCuts->SetMinNClustersITS(4);
+ esdTrackCuts->SetMinDCAToVertexXYPtDep("0.0100*TMath::Max(0.,(1-TMath::Floor(TMath::Abs(pt)/2.)))");
+ esdTrackCuts->SetEtaRange(-0.8,0.8);
+ esdTrackCuts->SetMaxDCAToVertexXY(1.);
+ esdTrackCuts->SetMaxDCAToVertexZ(1.);
+ esdTrackCuts->SetPtRange(0.8,1.e10);
+ AddTrackCuts(esdTrackCuts);
+
+ const Int_t nptbins=4;
+ const Int_t nvars=13;
+ Float_t* ptbins;
+ ptbins=new Float_t[nptbins+1];
+
+ ptbins[0]=0.;
+ ptbins[1]=2.;
+ ptbins[2]=3.;
+ ptbins[3]=4.;
+ ptbins[4]=9999.;
+
+ SetGlobalIndex(nvars,nptbins);
+ SetPtBins(nptbins+1,ptbins);
+
+ Float_t** prodcutsval;
+ prodcutsval=new Float_t*[nvars];
+ for(Int_t iv=0;iv<nvars;iv++){
+ prodcutsval[iv]=new Float_t[nptbins];
+ }
+
+ for(Int_t ipt=0;ipt<nptbins;ipt++){
+ prodcutsval[0][ipt]=0.13;
+ prodcutsval[1][ipt]=0.5;
+ prodcutsval[2][ipt]=0.6;
+ prodcutsval[3][ipt]=0.;
+ prodcutsval[4][ipt]=0.;
+ prodcutsval[5][ipt]=0.01;
+ prodcutsval[6][ipt]=0.04;
+ prodcutsval[7][ipt]=0.006;
+ prodcutsval[8][ipt]=0.8;
+ prodcutsval[9][ipt]=0.3;
+ prodcutsval[10][ipt]=0.;
+ prodcutsval[11][ipt]=0.05;
+ prodcutsval[12][ipt]=0.4;
+ }
+ SetCuts(nvars,nptbins,prodcutsval);
+
+ AliAODPidHF* pidObjK=new AliAODPidHF();
+ Double_t sigmasK[5]={3.,1.,1.,3.,2.};
+ pidObjK->SetSigma(sigmasK);
+ pidObjK->SetAsym(kTRUE);
+ pidObjK->SetMatch(1);
+ pidObjK->SetTPC(kTRUE);
+ pidObjK->SetTOF(kTRUE);
+ pidObjK->SetITS(kTRUE);
+ Double_t plimK[2]={0.5,0.8};
+ pidObjK->SetPLimit(plimK,2);
+
+ SetPidHF(pidObjK);
+
+ AliAODPidHF* pidObjpi=new AliAODPidHF();
+ pidObjpi->SetTPC(kTRUE);
+ Double_t sigmaspi[5]={3.,0.,0.,0.,0.};
+ pidObjpi->SetSigma(sigmaspi);
+ SetPidpion(pidObjpi);
+
+ AliAODPidHF* pidObjp=new AliAODPidHF();
+ Double_t sigmasp[5]={3.,1.,1.,3.,2.};
+ pidObjp->SetSigma(sigmasp);
+ pidObjp->SetAsym(kTRUE);
+ pidObjp->SetMatch(1);
+ pidObjp->SetTPC(kTRUE);
+ pidObjp->SetTOF(kTRUE);
+ pidObjp->SetITS(kTRUE);
+ Double_t plimp[2]={1.,2.};
+ pidObjp->SetPLimit(plimp,2);
+
+ SetPidprot(pidObjp);
+
+ SetUsePID(kTRUE);
+
+ PrintAll();
+
+ for(Int_t iiv=0;iiv<nvars;iiv++){
+ delete [] prodcutsval[iiv];
+ }
+ delete [] prodcutsval;
+ prodcutsval=NULL;
+ delete [] ptbins;
+ ptbins=NULL;
+
+ delete pidObjp;
+ pidObjp=NULL;
+
+ return;
+}
+//------------------
+Bool_t AliRDHFCutsLctopKpi::ReconstructKF(AliAODRecoDecayHF3Prong *d,Int_t *pdgs,Double_t field) const{
+
+ const Int_t nprongs=d->GetNProngs();
+ if(nprongs==0) return kFALSE;
+
+ Int_t iprongs[nprongs];
+ for(Int_t i=0;i<nprongs;i++) iprongs[i]=i;
+
+ Double_t mass[2]={0.,0.};
+
+ AliKFParticle *decay=d->ApplyVertexingKF(iprongs,nprongs,pdgs,kTRUE,field,mass);
+ if(!decay) return kTRUE;
+ AliESDVertex *vertexESD = new AliESDVertex(decay->Parameters(),
+ decay->CovarianceMatrix(),
+ decay->GetChi2(),
+ nprongs);
+ Double_t pos[3],cov[6],chi2perNDF;
+ vertexESD->GetXYZ(pos);
+ vertexESD->GetCovMatrix(cov);
+ chi2perNDF = vertexESD->GetChi2toNDF();
+ delete vertexESD; vertexESD=NULL;
+ AliAODVertex *vertexAOD = new AliAODVertex(pos,cov,chi2perNDF,0x0,-1,AliAODVertex::kUndef,nprongs);
+ d->SetSecondaryVtx(vertexAOD);
+ return kTRUE;
+}
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff