ClassImp(AliRDHFCutsDstoKKpi)
//--------------------------------------------------------------------------
-AliRDHFCutsDstoKKpi::AliRDHFCutsDstoKKpi() :
-AliRDHFCuts()
+AliRDHFCutsDstoKKpi::AliRDHFCutsDstoKKpi(const char* name) :
+AliRDHFCuts(name)
{
//
// Default Constructor
//
- Int_t nvars=9;
+ Int_t nvars=14;
SetNVars(nvars);
- TString varNames[9]={"inv. mass [GeV]",
- "dca [cm]",
- "cosThetaStar",
- "pTK [GeV/c]",
- "pTPi [GeV/c]",
- "d0K [cm]",
- "d0Pi [cm]",
- "d0d0 [cm^2]",
- "cosThetaPoint"};
- Bool_t isUpperCut[9]={kTRUE,
- kTRUE,
- kTRUE,
- kFALSE,
- kFALSE,
- kTRUE,
- kTRUE,
- kTRUE,
- kFALSE};
- SetVarNames(nvars,varNames,isUpperCut);
- Bool_t forOpt[9]={kFALSE,
- kTRUE,
- kTRUE,
+ TString varNames[14]={"inv. mass [GeV]",
+ "pTK [GeV/c]",
+ "pTPi [GeV/c]",
+ "d0K [cm]",
+ "d0Pi [cm]",
+ "dist12 [cm]",
+ "sigmavert [cm]",
+ "decLen [cm]",
+ "ptMax [GeV/c]",
+ "cosThetaPoint",
+ "Sum d0^2 (cm^2)",
+ "dca [cm]",
+ "inv. mass (Mphi-MKK) [GeV]",
+ "inv. mass (MKo*-MKpi) [GeV]"};
+ Bool_t isUpperCut[14]={kTRUE,
+ kFALSE,
+ kFALSE,
+ kFALSE,
+ kFALSE,
+ kFALSE,
+ kTRUE,
+ kFALSE,
+ kFALSE,
+ kFALSE,
+ kFALSE,
+ kTRUE,
+ kTRUE,
+ kTRUE};
+ SetVarNames(14,varNames,isUpperCut);
+ Bool_t forOpt[14]={kFALSE,
+ kFALSE,
kFALSE,
kFALSE,
kFALSE,
kFALSE,
kTRUE,
+ kTRUE,
+ kTRUE,
+ kTRUE,
+ kTRUE,
+ kFALSE,
+ kTRUE,
kTRUE};
- SetVarsForOpt(4,forOpt);
+ SetVarsForOpt(7,forOpt);
Float_t limits[2]={0,999999999.};
SetPtBins(2,limits);
}
AliAODRecoDecayHF3Prong *dd = (AliAODRecoDecayHF3Prong*)d;
+ Int_t iter=-1;
+ if(fVarsForOpt[0]){
+ iter++;
+ if(TMath::Abs(pdgdaughters[0]==321)){
+ vars[iter]=dd->InvMassDsKKpi();
+ }else{
+ vars[iter]=dd->InvMassDspiKK();
+ }
+ }
+ if(fVarsForOpt[1]){
+ iter++;
+ Float_t minPtDau=99999.;
+ for(Int_t iprong=0;iprong<3;iprong++){
+ if(TMath::Abs(pdgdaughters[iprong])==321 &&
+ dd->PtProng(iprong)<minPtDau) minPtDau=dd->PtProng(iprong);
+ }
+ vars[iter]=minPtDau;
+ }
+ if(fVarsForOpt[2]){
+ iter++;
+ for(Int_t iprong=0;iprong<3;iprong++){
+ if(TMath::Abs(pdgdaughters[iprong])==211) {
+ vars[iter]=dd->PtProng(iprong);
+ }
+ }
+ }
+ if(fVarsForOpt[3]){
+ iter++;
+ Float_t minImpParDau=99999.;
+ for(Int_t iprong=0;iprong<3;iprong++){
+ if(TMath::Abs(pdgdaughters[iprong])==321 &&
+ dd->Getd0Prong(iprong)<minImpParDau) minImpParDau=dd->Getd0Prong(iprong);
+ }
+ vars[iter]=minImpParDau;
+ }
+ if(fVarsForOpt[4]){
+ iter++;
+ for(Int_t iprong=0;iprong<3;iprong++){
+ if(TMath::Abs(pdgdaughters[iprong])==211) {
+ vars[iter]=dd->Getd0Prong(iprong);
+ }
+ }
+ }
+ if(fVarsForOpt[5]){
+ iter++;
+ Float_t minDistPair=TMath::Min(dd->GetDist12toPrim(),dd->GetDist23toPrim());
+ vars[iter]=minDistPair;
+ }
+ if(fVarsForOpt[6]){
+ iter++;
+ vars[iter]=dd->GetSigmaVert();
+ }
+ if(fVarsForOpt[7]){
+ iter++;
+ vars[iter] = dd->DecayLength();
+ }
+ if(fVarsForOpt[8]){
+ iter++;
+ Float_t ptmax=0;
+ for(Int_t i=0;i<3;i++){
+ if(dd->PtProng(i)>ptmax)ptmax=dd->PtProng(i);
+ }
+ vars[iter]=ptmax;
+ }
+ if(fVarsForOpt[9]){
+ iter++;
+ vars[iter]=dd->CosPointingAngle();
+ }
+ if(fVarsForOpt[10]){
+ iter++;
+ vars[iter]=dd->Getd0Prong(0)*dd->Getd0Prong(0)+dd->Getd0Prong(1)*dd->Getd0Prong(1)+dd->Getd0Prong(2)*dd->Getd0Prong(2);
+ }
+ if(fVarsForOpt[11]){
+ iter++;
+ Float_t maxDCA=0.;
+ for(Int_t i=0;i<3;i++){
+ if(d->GetDCA(i)>maxDCA) maxDCA=d->GetDCA(i);
+ }
+ vars[iter]=maxDCA;
+ }
+ if(fVarsForOpt[12]){
+ iter++;
+ Double_t mPDGPhi = TDatabasePDG::Instance()->GetParticle(333)->Mass();
+ if(TMath::Abs(pdgdaughters[0]==321)){
+
+ Double_t phimass01=d->InvMass2Prongs(0,1,321,321);
+ vars[iter]=TMath::Abs(phimass01-mPDGPhi);
+ // vars[iter]=dd->InvMass2Prongs(0,1,321,321);
+ }else{
+ Double_t phimass12=d->InvMass2Prongs(1,2,321,321);
+ vars[iter]=TMath::Abs(phimass12-mPDGPhi);
+ // vars[iter]=dd->InvMass2Prongs(1,2,321,321);
+ }
+ }
+ if(fVarsForOpt[13]){
+ iter++;
+ Double_t mPDGK0star = TDatabasePDG::Instance()->GetParticle(313)->Mass();
+ if(TMath::Abs(pdgdaughters[0]==321)){
+
+ Double_t mass12kpi=d->InvMass2Prongs(1,2,321,211);
+ vars[iter]=TMath::Abs(mass12kpi-mPDGK0star);
+ // vars[iter]=dd->InvMass2Prongs(1,2,321,211);
+ }else{
+ Double_t mass01pik=d->InvMass2Prongs(0,1,211,321);
+ vars[iter]=TMath::Abs(mass01pik-mPDGK0star);
+ // vars[iter]=dd->InvMass2Prongs(0,1,211,321);
+ }
+ }
+
return;
}
+//---------------------------------------------------------------------------
+Int_t AliRDHFCutsDstoKKpi::IsSelectedPID(AliAODRecoDecayHF *rd) {
+ // PID selection
+ // return values: 0->NOT OK, 1->OK as KKpi, 2->OK as piKK, 3->OK as both
+ Int_t retCode=3;
+ Bool_t okKKpi=kTRUE;
+ Bool_t okpiKK=kTRUE;
+ if(!fUsePID || !rd) return retCode;
+ if(!fPidHF){
+ AliWarning("AliAODPidHF not created!");
+ return retCode;
+ }
+ Int_t nKaons=0;
+ Int_t nNotKaons=0;
+ Int_t sign= rd->GetCharge();
+ for(Int_t iDaught=0; iDaught<3; iDaught++){
+ AliAODTrack *track=(AliAODTrack*)rd->GetDaughter(iDaught);
+ Int_t isPion=fPidHF->MakeRawPid(track,AliPID::kPion);
+ Int_t isKaon=fPidHF->MakeRawPid(track,AliPID::kKaon);
+ Int_t isProton=fPidHF->MakeRawPid(track,AliPID::kProton);
+
+ if(isProton>0 && isKaon<0 && isPion<0) return 0;
+ if(sign!=track->Charge()){// must be kaon
+ if(isKaon<0) return 0;
+ }
+ if(isKaon>0 && isPion<0) nKaons++;
+ if(isKaon<0) nNotKaons++;
+ if(iDaught==0){
+ if(isKaon<0) okKKpi=kFALSE;
+ if(isPion<0) okpiKK=kFALSE;
+ }
+ else if(iDaught==2){
+ if(isKaon<0) okpiKK=kFALSE;
+ if(isPion<0) okKKpi=kFALSE;
+ }
+ }
+
+ if(nKaons>2)return 0;
+ if(nNotKaons>1) return 0;
+
+ if(!okKKpi) retCode-=1;
+ if(!okpiKK) retCode-=2;
+
+ return retCode;
+}
+
//---------------------------------------------------------------------------
Int_t AliRDHFCutsDstoKKpi::IsSelected(TObject* obj,Int_t selectionLevel) {
//
}
- Int_t returnvalue=1;
+ // PID selection
+ Int_t returnvaluePID=3;
+ if(selectionLevel==AliRDHFCuts::kAll ||
+ selectionLevel==AliRDHFCuts::kCandidate ||
+ selectionLevel==AliRDHFCuts::kPID) {
+ returnvaluePID = IsSelectedPID(d);
+ }
+ if(returnvaluePID==0)return 0;
+ Bool_t okPidDsKKpi=returnvaluePID&1;
+ Bool_t okPidDspiKK=returnvaluePID&2;
+
+ Int_t returnvalue=1;
// selection on candidate
if(selectionLevel==AliRDHFCuts::kAll ||
selectionLevel==AliRDHFCuts::kCandidate) {
- }
+ Int_t okDsKKpi=1;
+ Int_t okDspiKK=1;
+ Int_t okMassPhi=0;
+ Int_t okMassK0star=0;
+
+ Double_t pt=d->Pt();
+ Int_t ptbin=PtBin(pt);
+
+ Double_t mDsPDG = TDatabasePDG::Instance()->GetParticle(431)->Mass();
+ Double_t mDsKKpi=d->InvMassDsKKpi();
+ Double_t mDspiKK=d->InvMassDspiKK();
+ if(TMath::Abs(mDsKKpi-mDsPDG)>fCutsRD[GetGlobalIndex(0,ptbin)]) okDsKKpi = 0;
+ if(TMath::Abs(mDspiKK-mDsPDG)>fCutsRD[GetGlobalIndex(0,ptbin)]) okDspiKK = 0;
+ if(!okDsKKpi && !okDspiKK) return 0;
+ if(okPidDsKKpi && !okDsKKpi) return 0;
+ if(okPidDspiKK && !okDspiKK) return 0;
+
+ //single track
+ if(TMath::Abs(d->PtProng(1)) < fCutsRD[GetGlobalIndex(1,ptbin)] ||
+ TMath::Abs(d->Getd0Prong(1))<fCutsRD[GetGlobalIndex(3,ptbin)]) return 0;
+ if(okDsKKpi){
+ if(TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(1,ptbin)] ||
+ TMath::Abs(d->Getd0Prong(0))<fCutsRD[GetGlobalIndex(3,ptbin)]) okDsKKpi=0;
+ if(TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(2,ptbin)] ||
+ TMath::Abs(d->Getd0Prong(2))<fCutsRD[GetGlobalIndex(4,ptbin)]) okDsKKpi=0;
+ }
+ if(okDspiKK){
+ if(TMath::Abs(d->PtProng(0)) < fCutsRD[GetGlobalIndex(2,ptbin)] ||
+ TMath::Abs(d->Getd0Prong(0))<fCutsRD[GetGlobalIndex(4,ptbin)]) okDspiKK=0;
+ if(TMath::Abs(d->PtProng(2)) < fCutsRD[GetGlobalIndex(1,ptbin)] ||
+ TMath::Abs(d->Getd0Prong(2))<fCutsRD[GetGlobalIndex(3,ptbin)]) okDspiKK=0;
+ }
+ if(!okDsKKpi && !okDspiKK) return 0;
+ // cuts on resonant decays (via Phi or K0*)
+ Double_t mPhiPDG = TDatabasePDG::Instance()->GetParticle(333)->Mass();
+ Double_t mK0starPDG = TDatabasePDG::Instance()->GetParticle(313)->Mass();
+ if(okDsKKpi){
+ Double_t mass01phi=d->InvMass2Prongs(0,1,321,321);
+ Double_t mass12K0s=d->InvMass2Prongs(1,2,321,211);
+ if(TMath::Abs(mass01phi-mPhiPDG)<fCutsRD[GetGlobalIndex(12,ptbin)]) okMassPhi=1;
+ if(TMath::Abs(mass12K0s-mK0starPDG)<fCutsRD[GetGlobalIndex(13,ptbin)]) okMassK0star = 1;
+ if(!okMassPhi && !okMassK0star) okDsKKpi=0;
+ }
+ if(okDspiKK){
+ Double_t mass01K0s=d->InvMass2Prongs(0,1,211,321);
+ Double_t mass12phi=d->InvMass2Prongs(1,2,321,321);
+ if(TMath::Abs(mass01K0s-mK0starPDG)<fCutsRD[GetGlobalIndex(13,ptbin)]) okMassK0star = 1;
+ if(TMath::Abs(mass12phi-mPhiPDG)<fCutsRD[GetGlobalIndex(12,ptbin)]) okMassPhi=1;
+ if(!okMassPhi && !okMassK0star) okDspiKK=0;
+ }
+ if(!okDsKKpi && !okDspiKK) return 0;
+
+ // Cuts on track pairs
+ for(Int_t i=0;i<3;i++) if(d->GetDCA(i)>fCutsRD[GetGlobalIndex(11,ptbin)]) return 0;
+ if(d->GetDist12toPrim()<fCutsRD[GetGlobalIndex(5,ptbin)] ||
+ d->GetDist23toPrim()<fCutsRD[GetGlobalIndex(5,ptbin)]) return 0;
+
+
+ // Cuts on candidate triplet
+ if(d->GetSigmaVert()>fCutsRD[GetGlobalIndex(6,ptbin)]) return 0;
+ if(d->DecayLength()<fCutsRD[GetGlobalIndex(7,ptbin)]) return 0;
+ 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;
+ 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;
+
+ returnvalue=0;
+ if(okDsKKpi) returnvalue+=1;
+ if(okDspiKK) returnvalue+=2;
+ if(okMassPhi) returnvalue+=4;
+ if(okMassK0star) returnvalue+=8;
+
+ }
return returnvalue;
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff