PWGHF/vertexingHF/AliRDHFCutsLctoV0.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2010, 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 /* $Id$ */
  17
  18 /////////////////////////////////////////////////////////////
  19 //
  20 // Class for cuts on AOD reconstructed Lc->V0+X
  21 //
  22 // Modified by A.De Caro - decaro@sa.infn.it
  23 //
  24 /////////////////////////////////////////////////////////////
  25
  26 #include <Riostream.h>
  27
  28 #include <TDatabasePDG.h>
  29 #include <TMath.h>
  30
  31 #include "AliAnalysisManager.h"
  32 #include "AliInputEventHandler.h"
  33 #include "AliPIDResponse.h"
  34 #include "AliRDHFCutsLctoV0.h"
  35 #include "AliAODRecoCascadeHF.h"
  36 #include "AliAODTrack.h"
  37 #include "AliESDtrack.h"
  38 #include "AliESDVertex.h"
  39 #include "AliAODVertex.h"
  40 #include "AliAODv0.h"
  41 #include "AliESDv0.h"
  42
  43 using std::cout;
  44 using std::endl;
  45
  46 ClassImp(AliRDHFCutsLctoV0)
  47
  48 //--------------------------------------------------------------------------
  49   AliRDHFCutsLctoV0::AliRDHFCutsLctoV0(const char* name, Short_t /*v0channel*/) :
  50   AliRDHFCuts(name),
  51   fPidSelectionFlag(0),
  52   fPidHFV0pos(0),
  53   fPidHFV0neg(0),
  54   fV0daughtersCuts(0),
  55   fV0Type(0)
  56 {
  57   //
  58   // Default Constructor
  59   //
  60
  61   const Int_t nvars=17;
  62   SetNVars(nvars);
  63   TString varNames[nvars]={"Lc inv. mass if K0S [GeV/c2]",                   //  0
  64                            "Lc inv. mass if Lambda [GeV/c2]",                //  1
  65                            "K0S inv. mass [GeV/c2]",                         //  2
  66                            "Lambda/LambdaBar inv. mass[GeV/c2]",             //  3
  67                            "pT min bachelor track [GeV/c]",                  //  4
  68                            "pT min V0-positive track [GeV/c]",               //  5
  69                            "pT min V0-negative track [GeV/c]",               //  6
  70                            "dca cascade (prong-to-prong) cut [cm]",          //  7
  71                            "dca V0 (prong-to-prong) cut [number of sigmas]", //  8
  72                            "V0 cosPA min wrt PV",                            //  9
  73                            "d0 max bachelor wrt PV [cm]",                    // 10
  74                            "d0 max V0 wrt PV [cm]",                          // 11
  75                            "mass K0S veto [GeV/c2]",                         // 12
  76                            "mass Lambda/LambdaBar veto [GeV/c2]",            // 13
  77                            "mass Gamma veto [GeV/c2]",                       // 14
  78                            "pT min V0 track [GeV/c]",                        // 15
  79                            "V0 type"                                         // 16
  80   };
  81
  82   Bool_t isUpperCut[nvars]={kTRUE,  //  0
  83                             kTRUE,  //  1
  84                             kTRUE,  //  2
  85                             kTRUE,  //  3
  86                             kFALSE, //  4
  87                             kFALSE, //  5
  88                             kFALSE, //  6
  89                             kTRUE,  //  7
  90                             kTRUE,  //  8
  91                             kFALSE, //  9
  92                             kTRUE,  // 10
  93                             kTRUE,  // 11
  94                             kFALSE, // 12
  95                             kFALSE, // 13
  96                             kFALSE, // 14
  97                             kFALSE, // 15
  98                             kFALSE  // 16
  99   };
 100   SetVarNames(nvars,varNames,isUpperCut);
 101   Bool_t forOpt[nvars]={kFALSE, //  0
 102                         kFALSE, //  1
 103                         kTRUE,  //  2
 104                         kTRUE,  //  3
 105                         kTRUE,  //  4
 106                         kTRUE,  //  5
 107                         kTRUE,  //  6
 108                         kTRUE,  //  7
 109                         kTRUE,  //  8
 110                         kTRUE,  //  9
 111                         kTRUE,  // 10
 112                         kTRUE,  // 11
 113                         kTRUE,  // 12
 114                         kTRUE,  // 13
 115                         kTRUE,  // 14
 116                         kTRUE,  // 15
 117                         kFALSE  // 16
 118   };
 119   SetVarsForOpt(nvars,forOpt);
 120
 121   Float_t limits[2]={0,999999999.};
 122   SetPtBins(2,limits);
 123
 124   /*
 125   switch (v0channel) {
 126   case 0:
 127     fV0channel = 0x0001;
 128     break;
 129   case 1:
 130     fV0channel = 0x0002;
 131     break;
 132   case 2:
 133     fV0channel = 0x0004;
 134     break;
 135   }
 136   */
 137
 138 }
 139 //--------------------------------------------------------------------------
 140 AliRDHFCutsLctoV0::AliRDHFCutsLctoV0(const AliRDHFCutsLctoV0 &source) :
 141   AliRDHFCuts(source),
 142   fPidSelectionFlag(source.fPidSelectionFlag),
 143   fPidHFV0pos(0),
 144   fPidHFV0neg(0),
 145   fV0daughtersCuts(0),
 146   fV0Type(source.fV0Type)
 147     /*fV0channel(source.fV0channel)*/
 148 {
 149   //
 150   // Copy constructor
 151   //
 152
 153   if (source.fPidHFV0pos) fPidHFV0pos = new AliAODPidHF(*(source.fPidHFV0pos));
 154   else fPidHFV0pos = new AliAODPidHF();
 155   if (source.fPidHFV0neg) fPidHFV0neg = new AliAODPidHF(*(source.fPidHFV0neg));
 156   else fPidHFV0neg = new AliAODPidHF();
 157
 158   if (source.fV0daughtersCuts) fV0daughtersCuts = new AliESDtrackCuts(*(source.fV0daughtersCuts));
 159   else fV0daughtersCuts = new AliESDtrackCuts();
 160
 161 }
 162 //--------------------------------------------------------------------------
 163 AliRDHFCutsLctoV0 &AliRDHFCutsLctoV0::operator=(const AliRDHFCutsLctoV0 &source)
 164 {
 165   //
 166   // assignment operator
 167   //
 168
 169   if (this != &source) {
 170
 171     AliRDHFCuts::operator=(source);
 172     fPidSelectionFlag = source.fPidSelectionFlag;
 173     delete fPidHFV0pos;
 174     fPidHFV0pos = new AliAODPidHF(*(source.fPidHFV0pos));
 175     delete fPidHFV0neg;
 176     fPidHFV0neg = new AliAODPidHF(*(source.fPidHFV0neg));
 177
 178     delete fV0daughtersCuts;
 179     fV0daughtersCuts = new AliESDtrackCuts(*(source.fV0daughtersCuts));
 180
 181     fV0Type  = source.fV0Type;
 182
 183   }
 184
 185   return *this;
 186 }
 187
 188
 189 //---------------------------------------------------------------------------
 190 AliRDHFCutsLctoV0::~AliRDHFCutsLctoV0() {
 191  //
 192  //  Default Destructor
 193  //
 194
 195  if (fPidHFV0pos) {
 196   delete fPidHFV0pos;
 197   fPidHFV0pos=0;
 198  }
 199  if (fPidHFV0neg) {
 200   delete fPidHFV0neg;
 201   fPidHFV0neg=0;
 202  }
 203
 204  if (fV0daughtersCuts) {
 205   delete fV0daughtersCuts;
 206   fV0daughtersCuts=0;
 207  }
 208
 209 }
 210
 211 //---------------------------------------------------------------------------
 212 void AliRDHFCutsLctoV0::GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters) {
 213   //
 214   // Fills in vars the values of the variables
 215   //
 216
 217   if (pdgdaughters[0]==-9999) return; // dummy
 218
 219   if (nvars!=fnVarsForOpt) {
 220     printf("AliRDHFCutsLctoV0::GetCutVarsForOpt: wrong number of variables\n");
 221     return;
 222   }
 223
 224   Double_t mLcPDG  = TDatabasePDG::Instance()->GetParticle(4122)->Mass();
 225   Double_t mk0sPDG = TDatabasePDG::Instance()->GetParticle(310)->Mass();
 226   Double_t mLPDG   = TDatabasePDG::Instance()->GetParticle(3122)->Mass();
 227
 228   AliAODRecoCascadeHF *dd = (AliAODRecoCascadeHF*)d;
 229
 230   // Get the v0 and all daughter tracks
 231   AliAODTrack *bachelorTrack = (AliAODTrack*)dd->GetBachelor();
 232   AliAODv0 *v0 = (AliAODv0*)dd->Getv0();
 233   AliAODTrack *v0positiveTrack = (AliAODTrack*)dd->Getv0PositiveTrack();
 234   AliAODTrack *v0negativeTrack = (AliAODTrack*)dd->Getv0NegativeTrack();
 235
 236   Int_t iter=-1;
 237   // cut on cascade mass, if K0S + p
 238   if (fVarsForOpt[0]) {
 239     iter++;
 240     vars[iter]=TMath::Abs(dd->InvMassLctoK0sP()-mLcPDG);
 241   }
 242   // cut on cascade mass, if Lambda/LambdaBar + pi
 243   if (fVarsForOpt[1]) {
 244     iter++;
 245     vars[iter]=TMath::Abs(dd->InvMassLctoLambdaPi()-mLcPDG);
 246   }
 247
 248   // cut on V0 mass if K0S
 249   if (fVarsForOpt[2]) {
 250     iter++;
 251     vars[iter]=TMath::Abs(v0->MassK0Short()-mk0sPDG);
 252   }
 253
 254   // cut on V0 mass if Lambda/LambdaBar
 255   if (fVarsForOpt[3]) {
 256
 257     if (bachelorTrack->Charge()==1) {
 258       iter++;
 259       vars[iter]=TMath::Abs(v0->MassLambda()-mLPDG);
 260     } else if (bachelorTrack->Charge()==-1) {
 261       iter++;
 262       vars[iter]=TMath::Abs(v0->MassAntiLambda()-mLPDG);
 263     }
 264
 265   }
 266
 267   // cut bachelor min pt
 268   if (fVarsForOpt[4]) {
 269     iter++;
 270     vars[iter]=bachelorTrack->Pt();
 271   }
 272
 273   // cut on V0-positive min pt
 274   if (fVarsForOpt[5]) {
 275     iter++;
 276     vars[iter]=v0positiveTrack->Pt();
 277   }
 278
 279   // cut on V0-negative min pt
 280   if (fVarsForOpt[6]) {
 281     iter++;
 282     vars[iter]=v0negativeTrack->Pt();
 283   }
 284
 285   // cut on cascade dca (prong-to-prong)
 286   if (fVarsForOpt[7]) {
 287     iter++;
 288     vars[iter]=dd->GetDCA(); // prong-to-prong cascade DCA
 289   }
 290
 291   // cut on V0 dca (prong-to-prong)
 292   if (fVarsForOpt[8]) {
 293     iter++;
 294     vars[iter]=v0->GetDCA(); // prong-to-prong V0 DCA
 295   }
 296
 297   // cut on V0 cosPA wrt PV
 298   if (fVarsForOpt[9]) {
 299     iter++;
 300     vars[iter]=dd->CosV0PointingAngle(); // cosine of V0 pointing angle wrt primary vertex
 301   }
 302
 303   // cut on bachelor transverse impact parameter wrt PV
 304   if (fVarsForOpt[10]) {
 305     iter++;
 306     vars[iter]=dd->Getd0Prong(0); // bachelor transverse impact parameter wrt primary vertex
 307   }
 308
 309   // cut on V0 transverse impact parameter wrt PV
 310   if (fVarsForOpt[11]) {
 311     iter++;
 312     vars[iter]=dd->Getd0Prong(1); // V0 transverse impact parameter wrt primary vertex
 313   }
 314
 315   // cut on K0S invariant mass veto
 316   if (fVarsForOpt[12]) {
 317     iter++;
 318     vars[iter]=TMath::Abs(v0->MassK0Short()-mk0sPDG); // K0S invariant mass veto
 319   }
 320
 321   // cut on Lambda/LambdaBar invariant mass veto
 322   if (fVarsForOpt[13]) {
 323
 324     if (bachelorTrack->Charge()==1) {
 325       iter++;
 326       vars[iter]=TMath::Abs(v0->MassLambda()-mLPDG);
 327     } else if (bachelorTrack->Charge()==-1) {
 328       iter++;
 329       vars[iter]=TMath::Abs(v0->MassAntiLambda()-mLPDG);
 330     }
 331
 332   }
 333
 334   // cut on gamma invariant mass veto
 335   if (fVarsForOpt[14]) {
 336     iter++;
 337     vars[iter]= v0->InvMass2Prongs(0,1,11,11); // gamma invariant mass veto
 338   }
 339
 340
 341   // cut on V0 pT min
 342   if (fVarsForOpt[15]) {
 343     iter++;
 344     vars[iter]= v0->Pt(); // V0 pT min
 345   }
 346
 347
 348   return;
 349 }
 350 //---------------------------------------------------------------------------
 351 Int_t AliRDHFCutsLctoV0::IsSelected(TObject* obj,Int_t selectionLevel) {
 352   //
 353   // Apply selection
 354   //
 355
 356   if (!fCutsRD) {
 357     cout<<"Cut matrice not inizialized. Exit..."<<endl;
 358     return 0;
 359   }
 360
 361   //PrintAll();
 362
 363   AliAODRecoCascadeHF* d = (AliAODRecoCascadeHF*)obj;
 364   if (!d) {
 365     AliDebug(2,"AliAODRecoCascadeHF null");
 366     return 0;
 367   }
 368
 369   if (!d->GetSecondaryVtx()) {
 370     AliDebug(2,"No secondary vertex for cascade");
 371     return 0;
 372   }
 373
 374   if (d->GetNDaughters()!=2) {
 375     AliDebug(2,Form("No 2 daughters for current cascade (nDaughters=%d)",d->GetNDaughters()));
 376     return 0;
 377   }
 378
 379   AliVTrack *cascTrk0 = dynamic_cast<AliVTrack*>(d->GetDaughter(0));
 380   AliVTrack *cascTrk1 = dynamic_cast<AliVTrack*>(d->GetDaughter(1));
 381   if (!cascTrk0 || !cascTrk1) {
 382     AliDebug(2,"At least one of V0daughters doesn't exist");
 383     return 0;
 384   }
 385
 386   AliAODv0 * v0 = dynamic_cast<AliAODv0*>(d->Getv0());
 387   AliAODTrack * bachelorTrack = dynamic_cast<AliAODTrack*>(d->GetBachelor());
 388   if (!v0 || !bachelorTrack) {
 389     AliDebug(2,"No V0 or no bachelor for current cascade");
 390     return 0;
 391   }
 392
 393   if (bachelorTrack->GetID()<0) {
 394     AliDebug(2,Form("Bachelor has negative ID %d",bachelorTrack->GetID()));
 395     return 0;
 396   }
 397
 398   if (!v0->GetSecondaryVtx()) {
 399     AliDebug(2,"No secondary vertex for V0 by cascade");
 400     return 0;
 401   }
 402
 403   if (v0->GetNDaughters()!=2) {
 404     AliDebug(2,Form("No 2 daughters for V0 of current cascade (onTheFly=%d, nDaughters=%d)",v0->GetOnFlyStatus(),v0->GetNDaughters()));
 405     return 0;
 406   }
 407
 408
 409   // Get the V0 daughter tracks
 410   AliAODTrack *v0positiveTrack = dynamic_cast<AliAODTrack*>(d->Getv0PositiveTrack());
 411   AliAODTrack *v0negativeTrack = dynamic_cast<AliAODTrack*>(d->Getv0NegativeTrack());
 412   //AliVTrack *trk0 = dynamic_cast<AliVTrack*>(v0->GetDaughter(0));
 413   //AliVTrack *trk1 = dynamic_cast<AliVTrack*>(v0->GetDaughter(1));
 414   if (!v0positiveTrack || !v0negativeTrack ) {
 415     AliDebug(2,"No V0 daughters' objects");
 416     return 0;
 417   }
 418
 419   if (v0positiveTrack->GetLabel()<0 || v0negativeTrack->GetLabel()<0) {
 420     AliDebug(2,Form("At least one of V0daughters has label negative (%d %d)",v0positiveTrack->GetLabel(),v0negativeTrack->GetLabel()));
 421     return 0;
 422   }
 423
 424   if (v0positiveTrack->GetID()<0 || v0negativeTrack->GetID()<0) {
 425     AliDebug(2,Form("At least one of V0 daughters has negative ID %d %d",v0positiveTrack->GetID(),v0negativeTrack->GetID()));
 426     return 0;
 427   }
 428
 429   //if(fUseTrackSelectionWithFilterBits && d->HasBadDaughters()) return 0;
 430   if ( fUseTrackSelectionWithFilterBits && !(bachelorTrack->TestFilterMask(BIT(4))) ) return 0;
 431
 432   // selection on daughter tracks
 433   if (selectionLevel==AliRDHFCuts::kAll ||
 434       selectionLevel==AliRDHFCuts::kTracks) {
 435
 436     if (fIsCandTrackSPDFirst && d->Pt()<fMaxPtCandTrackSPDFirst) {
 437       if (!bachelorTrack->HasPointOnITSLayer(0)) return 0;
 438     }
 439     if (fTrackCuts && fV0daughtersCuts) {
 440       AliAODVertex *vAOD = (AliAODVertex*)d->GetPrimaryVtx();
 441       Double_t pos[3]; vAOD->GetXYZ(pos);
 442       Double_t cov[6]; vAOD->GetCovarianceMatrix(cov);
 443       const AliESDVertex vESD(pos,cov,100.,100);
 444       if ( !(IsDaughterSelected(bachelorTrack,&vESD,fTrackCuts)) ||
 445            !(IsDaughterSelected(v0negativeTrack,&vESD,fV0daughtersCuts)) ||
 446            !(IsDaughterSelected(v0positiveTrack,&vESD,fV0daughtersCuts)) ) return 0;
 447     }
 448
 449   }
 450
 451   Bool_t okLck0sp=kTRUE, okLcLpi=kTRUE, okLcLBarpi=kTRUE;
 452   Bool_t okK0spipi=kTRUE, okLppi=kTRUE, okLBarpip=kTRUE;
 453
 454   // selection on candidate
 455   if (selectionLevel==AliRDHFCuts::kAll ||
 456       selectionLevel==AliRDHFCuts::kCandidate) {
 457
 458     Double_t pt = d->Pt();
 459     Int_t ptbin = PtBin(pt);
 460
 461     Double_t mLcPDG  = TDatabasePDG::Instance()->GetParticle(4122)->Mass();
 462     Double_t mk0sPDG = TDatabasePDG::Instance()->GetParticle(310)->Mass();
 463     Double_t mLPDG   = TDatabasePDG::Instance()->GetParticle(3122)->Mass();
 464
 465     // K0S + p
 466     Double_t mk0s    = v0->MassK0Short();
 467     Double_t mLck0sp = d->InvMassLctoK0sP();
 468
 469     // Lambda + pi
 470     Double_t mlambda  = v0->MassLambda();
 471     Double_t malambda = v0->MassAntiLambda();
 472     Double_t mLcLpi   = d->InvMassLctoLambdaPi();
 473
 474     // cut on Lc mass with K0S+p hypothesis
 475     if (TMath::Abs(mLck0sp-mLcPDG) > fCutsRD[GetGlobalIndex(0,ptbin)]) {
 476       okLck0sp = kFALSE;
 477       AliDebug(4,Form(" cascade mass is %2.2e and does not correspond to Lambda_c into K0S+p cut",mLck0sp));
 478     }
 479
 480     // cuts on the V0 mass: K0S case
 481     if (TMath::Abs(mk0s-mk0sPDG) > fCutsRD[GetGlobalIndex(2,ptbin)]) {
 482       okK0spipi = kFALSE;
 483       AliDebug(4,Form(" V0 mass is %2.2e and does not correspond to K0S cut",mk0s));
 484     }
 485
 486     // cut on Lc mass with Lambda+pi hypothesis
 487     if (TMath::Abs(mLcLpi-mLcPDG) > fCutsRD[GetGlobalIndex(1,ptbin)]) {
 488       okLcLpi = kFALSE;
 489       okLcLBarpi = kFALSE;
 490       AliDebug(4,Form(" cascade mass is %2.2e and does not correspond to Lambda_c into Lambda+pi cut",mLcLpi));
 491     }
 492
 493     // cuts on the V0 mass: Lambda/LambdaBar case
 494     //if ( !(bachelorTrack->Charge()==+1 && TMath::Abs(mlambda-mLPDG) <= fCutsRD[GetGlobalIndex(3,ptbin)] ) ) {
 495     if ( TMath::Abs(mlambda-mLPDG) > fCutsRD[GetGlobalIndex(3,ptbin)] ) {
 496       okLppi = kFALSE;
 497       AliDebug(4,Form(" V0 mass is %2.2e and does not correspond to LambdaBar cut",mlambda));
 498     }
 499
 500     //if ( !(bachelorTrack->Charge()==-1 && TMath::Abs(malambda-mLPDG) <= fCutsRD[GetGlobalIndex(3,ptbin)] ) ) {
 501     if ( TMath::Abs(malambda-mLPDG) > fCutsRD[GetGlobalIndex(3,ptbin)] ) {
 502       okLBarpip = kFALSE;
 503       AliDebug(4,Form(" V0 mass is %2.2e and does not correspond to LambdaBar cut",malambda));
 504     }
 505
 506     okLck0sp   = okLck0sp   && okK0spipi;
 507     okLcLpi    = okLcLpi    && okLppi;
 508     okLcLBarpi = okLcLBarpi && okLBarpip;
 509
 510     if (!okLck0sp && !okLcLpi && !okLcLBarpi) return 0;
 511
 512     // cuts on the minimum pt of the tracks
 513     if (TMath::Abs(bachelorTrack->Pt()) < fCutsRD[GetGlobalIndex(4,ptbin)]) {
 514       AliDebug(4,Form(" bachelor track Pt=%2.2e > %2.2e",bachelorTrack->Pt(),fCutsRD[GetGlobalIndex(4,ptbin)]));
 515       return 0;
 516     }
 517     if (TMath::Abs(v0positiveTrack->Pt()) < fCutsRD[GetGlobalIndex(5,ptbin)]) {
 518       AliDebug(4,Form(" V0-positive track Pt=%2.2e > %2.2e",v0positiveTrack->Pt(),fCutsRD[GetGlobalIndex(5,ptbin)]));
 519       return 0;
 520     }
 521     if (TMath::Abs(v0negativeTrack->Pt()) < fCutsRD[GetGlobalIndex(6,ptbin)]) {
 522       AliDebug(4,Form(" V0-negative track Pt=%2.2e > %2.2e",v0negativeTrack->Pt(),fCutsRD[GetGlobalIndex(6,ptbin)]));
 523       return 0;
 524     }
 525
 526     // cut on cascade dca (prong-to-prong)
 527     if ( TMath::Abs(d->GetDCA()) > fCutsRD[GetGlobalIndex(7,ptbin)] ) { // prong-to-prong cascade DCA
 528       AliDebug(4,Form(" cascade tracks DCA don't pass the cut"));
 529       return 0;
 530     }
 531
 532     // cut on V0 dca (prong-to-prong)
 533     if ( TMath::Abs(v0->GetDCA()) > fCutsRD[GetGlobalIndex(8,ptbin)] ) { // prong-to-prong V0 DCA
 534       AliDebug(4,Form(" V0 DCA don't pass the cut"));
 535       return 0;
 536     }
 537
 538     // cut on V0 cosine of pointing angle wrt PV
 539     if (d->CosV0PointingAngle() < fCutsRD[GetGlobalIndex(9,ptbin)]) { // cosine of V0 pointing angle wrt primary vertex
 540       AliDebug(4,Form(" V0 cosine of pointing angle doesn't pass the cut"));
 541       return 0;
 542     }
 543
 544     // cut on bachelor transverse impact parameter wrt PV
 545     if (TMath::Abs(d->Getd0Prong(0)) > fCutsRD[GetGlobalIndex(10,ptbin)]) { // bachelor transverse impact parameter wrt PV
 546       AliDebug(4,Form(" bachelor transverse impact parameter doesn't pass the cut"));
 547       return 0;
 548     }
 549
 550     // cut on V0 transverse impact parameter wrt PV
 551     if (TMath::Abs(d->Getd0Prong(1)) > fCutsRD[GetGlobalIndex(11,ptbin)]) { // V0 transverse impact parameter wrt PV
 552       AliDebug(4,Form(" V0 transverse impact parameter doesn't pass the cut"));
 553       return 0;
 554     }
 555
 556     // cut on K0S invariant mass veto
 557     if (TMath::Abs(v0->MassK0Short()-mk0sPDG) < fCutsRD[GetGlobalIndex(12,ptbin)]) { // K0S invariant mass veto
 558       AliDebug(4,Form(" veto on K0S invariant mass doesn't pass the cut"));
 559       return 0;
 560     }
 561
 562     // cut on Lambda/LambdaBar invariant mass veto
 563     if (TMath::Abs(v0->MassLambda()-mLPDG) < fCutsRD[GetGlobalIndex(13,ptbin)] ||
 564         TMath::Abs(v0->MassAntiLambda()-mLPDG) < fCutsRD[GetGlobalIndex(13,ptbin)] ) { // Lambda/LambdaBar invariant mass veto
 565       AliDebug(4,Form(" veto on K0S invariant mass doesn't pass the cut"));
 566       return 0;
 567     }
 568
 569     // cut on gamma invariant mass veto
 570     if (v0->InvMass2Prongs(0,1,11,11) < fCutsRD[GetGlobalIndex(14,ptbin)]) { // K0S invariant mass veto
 571       AliDebug(4,Form(" veto on gamma invariant mass doesn't pass the cut"));
 572       return 0;
 573     }
 574
 575     // cut on V0 pT min
 576     if (v0->Pt() < fCutsRD[GetGlobalIndex(15,ptbin)]) { // V0 pT min
 577       AliDebug(4,Form(" V0 track Pt=%2.2e > %2.2e",v0->Pt(),fCutsRD[GetGlobalIndex(15,ptbin)]));
 578       return 0;
 579     }
 580
 581   }
 582
 583   Int_t returnvalue = okLck0sp+2*okLcLBarpi+4*okLcLpi;
 584   /*
 585     retvalue case
 586           1  Lc->K0S + p
 587           2  Lc->LambdaBar + pi
 588           3  Lc->K0S + p AND Lc->LambdaBar + pi
 589           4  Lc->Lambda + pi
 590           5  Lc->K0S + p AND Lc->Lambda + pi
 591           6  Lc->LambdaBar + pi AND Lc->Lambda + pi
 592           7  Lc->K0S + p AND Lc->LambdaBar + pi AND Lc->Lambda + pi
 593   */
 594
 595   Int_t returnvaluePID = 7;
 596
 597   // selection on candidate
 598   if (selectionLevel==AliRDHFCuts::kAll ||
 599       selectionLevel==AliRDHFCuts::kCandidate ||
 600       selectionLevel==AliRDHFCuts::kPID )
 601     returnvaluePID = IsSelectedPID(d);
 602
 603   //if (fUsePID && returnvaluePID==0) return 0;
 604
 605   Int_t returnvalueTot = 0;
 606   if ( fUsePID )
 607     returnvalueTot = CombinePIDCuts(returnvalue,returnvaluePID);
 608   else
 609     returnvalueTot = returnvalue;
 610
 611   return returnvalueTot;
 612
 613 }
 614 //---------------------------------------------------------------------------
 615 Int_t AliRDHFCutsLctoV0::IsSelectedPID(AliAODRecoDecayHF* obj) {
 616
 617   // fPidHF -> PID object for bachelor
 618   // fPidHFV0pos -> PID object for positive V0 daughter
 619   // fPidHFV0neg -> PID object for negative V0 daughter
 620
 621   if (!fUsePID || !obj) return 7; // all hypothesis are valid
 622
 623   if (fPidHF->GetPidResponse()==0x0 ||
 624       fPidHFV0pos->GetPidResponse()==0x0 ||
 625       fPidHFV0neg->GetPidResponse()==0x0) {
 626     AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
 627     AliInputEventHandler *inputHandler=(AliInputEventHandler*)mgr->GetInputEventHandler();
 628     AliPIDResponse *pidResp=inputHandler->GetPIDResponse();
 629     fPidHF->SetPidResponse(pidResp);
 630     fPidHFV0pos->SetPidResponse(pidResp);
 631     fPidHFV0neg->SetPidResponse(pidResp);
 632     fPidHF->SetOldPid(kFALSE);
 633     fPidHFV0pos->SetOldPid(kFALSE);
 634     fPidHFV0neg->SetOldPid(kFALSE);
 635   }
 636
 637   AliAODRecoCascadeHF *objD = (AliAODRecoCascadeHF*)obj;
 638
 639   Bool_t isPeriodd = fPidHF->GetOnePad();
 640   Bool_t isMC = fPidHF->GetMC();
 641
 642   if (isPeriodd) {
 643     fPidHFV0pos->SetOnePad(kTRUE);
 644     fPidHFV0neg->SetOnePad(kTRUE);
 645   }
 646   if (isMC) {
 647     fPidHFV0neg->SetMC(kTRUE);
 648     fPidHFV0pos->SetMC(kTRUE);
 649   }
 650
 651   AliAODTrack *bachelor = (AliAODTrack*)objD->GetBachelor();
 652   AliAODTrack *v0Pos = (AliAODTrack*)objD->Getv0PositiveTrack();
 653   AliAODTrack *v0Neg = (AliAODTrack*)objD->Getv0NegativeTrack();
 654
 655   if (!bachelor || !v0Pos || !v0Neg) return 0;
 656
 657   Bool_t okLcK0Sp = kTRUE; // K0S case
 658   Bool_t okLcLambdaBarPi = kTRUE; // LambdaBar case
 659   Bool_t okLcLambdaPi = kTRUE; // Lambda case
 660
 661   CheckPID(bachelor,v0Neg,v0Pos,okLcK0Sp,okLcLambdaBarPi,okLcLambdaPi);
 662
 663   Int_t returnvalue = okLcK0Sp+2*okLcLambdaBarPi+4*okLcLambdaPi;
 664
 665   return returnvalue;
 666 }
 667 //-----------------------
 668 void AliRDHFCutsLctoV0::CheckPID(AliAODTrack *bachelor, AliAODTrack *v0Neg, AliAODTrack *v0Pos,
 669                                  Bool_t &isBachelorID1, Bool_t &isV0NegID2, Bool_t &isV0PosID4) {
 670   // identification strategy
 671
 672   Int_t trackIDtof = -1;
 673   Int_t trackIDtpc = -1;
 674
 675   Bool_t dummy1 = kFALSE;
 676   Bool_t dummy2 = kFALSE;
 677   Bool_t dummy4 = kFALSE;
 678
 679   Int_t tpcID = -1;
 680   Int_t tofID = -1;
 681   Double_t nTPCsigmasPr = -999;
 682   Double_t nTOFsigmasPr = -999;
 683
 684   Bool_t trackIDtofB = -1;
 685   Bool_t trackIDtpcB = -1;
 686
 687   switch (fPidSelectionFlag) {
 688
 689   case 0:
 690
 691     // identify bachelor
 692     trackIDtof = fPidHF->ApplyPidTOFRaw(bachelor,4);
 693     trackIDtpc = fPidHF->ApplyPidTPCRaw(bachelor,4);
 694     AliDebug(1,Form(" fPidHF->ApplyPidTOFRaw(bachelor,4)=%d fPidHF->ApplyPidTPCRaw(bachelor,4)=%d",trackIDtof,trackIDtpc));
 695     isBachelorID1 = (trackIDtof==4) && (trackIDtpc==4); // K0S case
 696     //isBachelorID2 = (fPidHF->ApplyPidTPCRaw(bachelor,2)==2) && (fPidHF->ApplyPidTOFRaw(bachelor,2)==2); // LambdaBar case
 697     //isBachelorID4 = isBachelorID2; // Lambda case
 698
 699     // identify V0neg
 700     trackIDtof = fPidHFV0neg->ApplyPidTOFRaw(v0Neg,4);
 701     trackIDtpc = fPidHFV0neg->ApplyPidTPCRaw(v0Neg,4);
 702     AliDebug(1,Form(" fPidHFV0neg->ApplyPidTOFRaw(v0Neg,4)=%d fPidHFV0neg->ApplyPidTPCRaw(v0Neg,4)=%d",trackIDtof,trackIDtpc));
 703     //isV0NegID1 = (fPidHFV0neg->ApplyPidTPCRaw(v0Neg,2)==2) && (fPidHFV0neg->ApplyPidTOFRaw(v0Neg,2)==2); // K0S case
 704     isV0NegID2 = (trackIDtof==4) && (trackIDtpc==4); // LambdaBar case
 705     //isV0NegID4 = isV0NegID1; // Lambda case
 706
 707     // identify V0pos
 708     trackIDtof = fPidHFV0pos->ApplyPidTOFRaw(v0Pos,4);
 709     trackIDtpc = fPidHFV0pos->ApplyPidTPCRaw(v0Pos,4);
 710     AliDebug(1,Form(" fPidHFV0pos->ApplyPidTOFRaw(v0Pos,4)=%d fPidHFV0pos->ApplyPidTPCRaw(v0POS,4)=%d",trackIDtof,trackIDtpc));
 711     //isV0PosID1 = (fPidHFV0pos->ApplyPidTPCRaw(v0Pos,2)==2) && (fPidHFV0pos->ApplyPidTOFRaw(v0Pos,2)==2); // K0S case
 712     //isV0PosID2 = isV0PosID1; // LambdaBar case
 713     isV0PosID4 = (trackIDtof==4) && (trackIDtpc==4); // Lambda case
 714
 715     break;
 716   case 1:
 717
 718     // identify bachelor
 719     trackIDtof = fPidHF->ApplyPidTOFRaw(bachelor,4);
 720     trackIDtpc = fPidHF->ApplyPidTPCRaw(bachelor,4);
 721     AliDebug(1,Form(" fPidHF->ApplyPidTOFRaw(bachelor,4)=%d fPidHFV0->ApplyPidTPCRaw(bachelor,4)=%d",trackIDtof,trackIDtpc));
 722     isBachelorID1 = ( trackIDtof==4 );
 723     dummy1 = ( !(fPidHF->CheckTOFPIDStatus(bachelor)) && (trackIDtpc==4) &&
 724                fPidHF->IsExcluded(bachelor,2,2.,"TPC") && fPidHF->IsExcluded(bachelor,3,2.,"TPC") ); // K0S case
 725     isBachelorID1 = isBachelorID1 || dummy1;
 726
 727
 728     // identify V0neg
 729     trackIDtof = fPidHFV0neg->ApplyPidTOFRaw(v0Neg,4);
 730     trackIDtpc = fPidHFV0neg->ApplyPidTPCRaw(v0Neg,4);
 731     AliDebug(1,Form(" fPidHFV0neg->ApplyPidTOFRaw(v0Neg,4)=%d fPidHFV0neg->ApplyPidTPCRaw(v0Neg,4)=%d",trackIDtof,trackIDtpc));
 732     isV0NegID2    = ( trackIDtof==4 );
 733     dummy2 = ( !(fPidHFV0neg->CheckTOFPIDStatus(v0Neg)) && (trackIDtpc==4) &&
 734                fPidHFV0neg->IsExcluded(v0Neg,2,2.,"TPC") && fPidHFV0neg->IsExcluded(v0Neg,3,2.,"TPC") ); // LambdaBar case
 735     isV0NegID2 = isV0NegID2 || dummy2;
 736
 737
 738     // identify V0pos
 739     trackIDtof = fPidHFV0pos->ApplyPidTOFRaw(v0Pos,4);
 740     trackIDtpc = fPidHFV0pos->ApplyPidTPCRaw(v0Pos,4);
 741     AliDebug(1,Form(" fPidHFV0pos->ApplyPidTOFRaw(v0Pos,4)=%d fPidHFV0pos->ApplyPidTPCRaw(v0Pos,4)=%d",trackIDtof,trackIDtpc));
 742     isV0PosID4    = ( trackIDtof==4 );
 743     dummy4 = ( !(fPidHFV0pos->CheckTOFPIDStatus(v0Pos)) && (trackIDtpc==4) &&
 744                fPidHFV0pos->IsExcluded(v0Pos,2,2.,"TPC") && fPidHFV0pos->IsExcluded(v0Pos,3,2.,"TPC") ); // Lambda case
 745     isV0PosID4 = isV0PosID4 || dummy4;
 746
 747
 748     break;
 749   case 2:
 750
 751     // identify bachelor
 752     nTOFsigmasPr = -999;
 753     tofID = fPidHF->GetnSigmaTOF(bachelor,4,nTOFsigmasPr);
 754     nTPCsigmasPr = -999;
 755     tpcID = fPidHF->GetnSigmaTPC(bachelor,4,nTPCsigmasPr);
 756     trackIDtofB = (tofID==1) && ( (bachelor->P()>=1.0 && bachelor->P()<2.5 && TMath::Abs(nTOFsigmasPr)<3) ||
 757                                   (bachelor->P()>=2.5 && nTOFsigmasPr>-2 && nTOFsigmasPr<3) );
 758     trackIDtpcB = (tpcID==1) && ( (bachelor->P()<1.0 && TMath::Abs(nTPCsigmasPr)<2) ||
 759                                   (bachelor->P()>=1.0 && TMath::Abs(nTPCsigmasPr)<3) );
 760     AliDebug(1,Form(" trackIDtofB=%d trackIDtpcB=%d",trackIDtofB,trackIDtpcB));
 761     isBachelorID1 = (bachelor->P()<1 && trackIDtpcB) || (bachelor->P()>=1 && trackIDtpcB && trackIDtofB); // K0S case
 762
 763     // identify V0neg
 764     nTOFsigmasPr = -999;
 765     tofID = fPidHFV0neg->GetnSigmaTOF(v0Neg,4,nTOFsigmasPr);
 766     nTPCsigmasPr = -999;
 767     tpcID = fPidHFV0neg->GetnSigmaTPC(v0Neg,4,nTPCsigmasPr);
 768     trackIDtofB = (tofID==1) && ( (v0Neg->P()>=1.0 && v0Neg->P()<2.5 && TMath::Abs(nTOFsigmasPr)<3) ||
 769                                   (v0Neg->P()>=2.5 && nTOFsigmasPr>-2 && nTOFsigmasPr<3) );
 770     trackIDtpcB = (tpcID==1) && ( (v0Neg->P()<1.0 && TMath::Abs(nTPCsigmasPr)<2) ||
 771                                   (v0Neg->P()>=1.0 && TMath::Abs(nTPCsigmasPr)<3) );
 772     AliDebug(1,Form(" trackIDtofB=%d trackIDtpcB=%d",trackIDtofB,trackIDtpcB));
 773     isV0NegID2 = (v0Neg->P()<1 && trackIDtpcB) || (v0Neg->P()>=1 && trackIDtpcB && trackIDtofB); // LambdaBar case
 774
 775     // identify V0pos
 776     nTOFsigmasPr = -999;
 777     tofID = fPidHFV0pos->GetnSigmaTOF(v0Pos,4,nTOFsigmasPr);
 778     nTPCsigmasPr = -999;
 779     tpcID = fPidHFV0pos->GetnSigmaTPC(v0Pos,4,nTPCsigmasPr);
 780     trackIDtofB = (tofID==1) && ( (v0Pos->P()>=1.0 && v0Pos->P()<2.5 && TMath::Abs(nTOFsigmasPr)<3) ||
 781                                   (v0Pos->P()>=2.5 && nTOFsigmasPr>-2 && nTOFsigmasPr<3) );
 782     trackIDtpcB = (tpcID==1) && ( (v0Pos->P()<1.0 && TMath::Abs(nTPCsigmasPr)<2) ||
 783                                   (v0Pos->P()>=1.0 && TMath::Abs(nTPCsigmasPr)<3) );
 784     AliDebug(1,Form(" trackIDtofB=%d trackIDtpcB=%d",trackIDtofB,trackIDtpcB));
 785     isV0PosID4 = (v0Pos->P()<1 && trackIDtpcB) || (v0Pos->P()>=1 && trackIDtpcB && trackIDtofB); // Lambda case
 786
 787     break;
 788   case 3:
 789
 790     // identify bachelor
 791     nTOFsigmasPr = -999;
 792     tofID = fPidHF->GetnSigmaTOF(bachelor,4,nTOFsigmasPr);
 793     nTPCsigmasPr = -999;
 794     tpcID = fPidHF->GetnSigmaTPC(bachelor,4,nTPCsigmasPr);
 795     trackIDtofB = (tofID==1) && ( (bachelor->P()>=1.0 && bachelor->P()<2.5 && TMath::Abs(nTOFsigmasPr)<3) ||
 796                                   (bachelor->P()>=2.5 && nTOFsigmasPr>-2 && nTOFsigmasPr<3) );
 797     trackIDtpcB = (tpcID==1) && ( (bachelor->P()<1.0 && TMath::Abs(nTPCsigmasPr)<2) ||
 798                                   (bachelor->P()>=1.0 && bachelor->P()<3.0 && TMath::Abs(nTPCsigmasPr)<3) ||
 799                                   (bachelor->P()>=3.0 && nTPCsigmasPr>-3 && nTPCsigmasPr<2) );
 800     AliDebug(1,Form(" trackIDtofB=%d trackIDtpcB=%d",trackIDtofB,trackIDtpcB));
 801     isBachelorID1 = (bachelor->P()<1 && trackIDtpcB) || (bachelor->P()>=1 && trackIDtpcB && trackIDtofB); // K0S case
 802
 803     // identify V0neg
 804     nTOFsigmasPr = -999;
 805     tofID = fPidHFV0neg->GetnSigmaTOF(v0Neg,4,nTOFsigmasPr);
 806     nTPCsigmasPr = -999;
 807     tpcID = fPidHFV0neg->GetnSigmaTPC(v0Neg,4,nTPCsigmasPr);
 808     trackIDtofB = (tofID==1) && ( (v0Neg->P()>=1.0 && v0Neg->P()<2.5 && TMath::Abs(nTOFsigmasPr)<3) ||
 809                                   (v0Neg->P()>=2.5 && nTOFsigmasPr>-2 && nTOFsigmasPr<3) );
 810     trackIDtpcB = (tpcID==1) && ( (v0Neg->P()<1.0 && TMath::Abs(nTPCsigmasPr)<2) ||
 811                                   (v0Neg->P()>=1.0 && v0Neg->P()<3.0 && TMath::Abs(nTPCsigmasPr)<3) ||
 812                                   (v0Neg->P()>=3.0 && nTPCsigmasPr>-3 && nTPCsigmasPr<2) );
 813     AliDebug(1,Form(" trackIDtofB=%d trackIDtpcB=%d",trackIDtofB,trackIDtpcB));
 814     isV0NegID2 = (v0Neg->P()<1 && trackIDtpcB) || (v0Neg->P()>=1 && trackIDtpcB && trackIDtofB); // LambdaBar case
 815
 816     // identify V0pos
 817     nTOFsigmasPr = -999;
 818     tofID = fPidHFV0pos->GetnSigmaTOF(v0Pos,4,nTOFsigmasPr);
 819     nTPCsigmasPr = -999;
 820     tpcID = fPidHFV0pos->GetnSigmaTPC(v0Pos,4,nTPCsigmasPr);
 821     trackIDtofB = (tofID==1) && ( (v0Pos->P()>=1.0 && v0Pos->P()<2.5 && TMath::Abs(nTOFsigmasPr)<3) ||
 822                                   (v0Pos->P()>=2.5 && nTOFsigmasPr>-2 && nTOFsigmasPr<3) );
 823     trackIDtpcB = (tpcID==1) && ( (v0Pos->P()<1.0 && TMath::Abs(nTPCsigmasPr)<2) ||
 824                                   (v0Pos->P()>=1.0 && v0Pos->P()<3.0 && TMath::Abs(nTPCsigmasPr)<3) ||
 825                                   (v0Pos->P()>=3.0 && nTPCsigmasPr>-3 && nTPCsigmasPr<2) );
 826     AliDebug(1,Form(" trackIDtofB=%d trackIDtpcB=%d",trackIDtofB,trackIDtpcB));
 827     isV0PosID4 = (v0Pos->P()<1 && trackIDtpcB) || (v0Pos->P()>=1 && trackIDtpcB && trackIDtofB); // Lambda case
 828
 829     break;
 830
 831   }
 832
 833 }
 834 //----------------
 835 Int_t AliRDHFCutsLctoV0::CombinePIDCuts(Int_t returnvalue, Int_t returnvaluePID) const {
 836   // combine PID with topological cuts
 837
 838  Int_t returnvalueTot=returnvalue&returnvaluePID;
 839
 840  return returnvalueTot;
 841 }
 842
 843 //----------------------------------
 844 Int_t AliRDHFCutsLctoV0::IsSelectedSingleCut(TObject* obj, Int_t selectionLevel, Int_t cutIndex) {
 845   //
 846   // Apply selection on single cut
 847   //
 848
 849   if (!fCutsRD) {
 850     cout<<"Cut matrice not inizialized. Exit..."<<endl;
 851     return 0;
 852   }
 853
 854   AliAODRecoCascadeHF* d = (AliAODRecoCascadeHF*)obj;
 855   if (!d) {
 856     AliDebug(2,"AliAODRecoCascadeHF null");
 857     return 0;
 858   }
 859
 860   if (!d->GetSecondaryVtx()) {
 861     AliDebug(2,"No secondary vertex for cascade");
 862     return 0;
 863   }
 864
 865   if (d->GetNDaughters()!=2) {
 866     AliDebug(2,Form("No 2 daughters for current cascade (nDaughters=%d)",d->GetNDaughters()));
 867     return 0;
 868   }
 869
 870   AliVTrack *cascTrk0 = dynamic_cast<AliVTrack*>(d->GetDaughter(0));
 871   AliVTrack *cascTrk1 = dynamic_cast<AliVTrack*>(d->GetDaughter(1));
 872   if (!cascTrk0 || !cascTrk1) {
 873     AliDebug(2,"At least one of V0daughters doesn't exist");
 874     return 0;
 875   }
 876
 877   AliAODv0 * v0 = dynamic_cast<AliAODv0*>(d->Getv0());
 878   AliAODTrack * bachelorTrack = dynamic_cast<AliAODTrack*>(d->GetBachelor());
 879   if (!v0 || !bachelorTrack) {
 880     AliDebug(2,"No V0 or no bachelor for current cascade");
 881     return 0;
 882   }
 883
 884   if (bachelorTrack->GetID()<0) {
 885     AliDebug(2,Form("Bachelor has negative ID %d",bachelorTrack->GetID()));
 886     return 0;
 887   }
 888
 889   if (!v0->GetSecondaryVtx()) {
 890     AliDebug(2,"No secondary vertex for V0 by cascade");
 891     return 0;
 892   }
 893
 894   if (v0->GetNDaughters()!=2) {
 895     AliDebug(2,Form("No 2 daughters for V0 of current cascade (onTheFly=%d, nDaughters=%d)",v0->GetOnFlyStatus(),v0->GetNDaughters()));
 896     return 0;
 897   }
 898
 899
 900   // Get the V0 daughter tracks
 901   AliAODTrack *v0positiveTrack = dynamic_cast<AliAODTrack*>(d->Getv0PositiveTrack());
 902   AliAODTrack *v0negativeTrack = dynamic_cast<AliAODTrack*>(d->Getv0NegativeTrack());
 903   //AliVTrack *trk0 = dynamic_cast<AliVTrack*>(v0->GetDaughter(0));
 904   //AliVTrack *trk1 = dynamic_cast<AliVTrack*>(v0->GetDaughter(1));
 905   if (!v0positiveTrack || !v0negativeTrack ) {
 906     AliDebug(2,"No V0 daughters' objects");
 907     return 0;
 908   }
 909
 910   if (v0positiveTrack->GetLabel()<0 || v0negativeTrack->GetLabel()<0) {
 911     AliDebug(2,Form("At least one of V0daughters has label negative (%d %d)",v0positiveTrack->GetLabel(),v0negativeTrack->GetLabel()));
 912     return 0;
 913   }
 914
 915   if (v0positiveTrack->GetID()<0 || v0negativeTrack->GetID()<0) {
 916     AliDebug(2,Form("At least one of V0 daughters has negative ID %d %d",v0positiveTrack->GetID(),v0negativeTrack->GetID()));
 917     return 0;
 918   }
 919
 920   //if(fUseTrackSelectionWithFilterBits && d->HasBadDaughters()) return 0;
 921   if ( fUseTrackSelectionWithFilterBits && !(bachelorTrack->TestFilterMask(BIT(4))) ) return 0;
 922
 923   // selection on daughter tracks
 924   if (selectionLevel==AliRDHFCuts::kAll ||
 925       selectionLevel==AliRDHFCuts::kTracks) {
 926
 927     if (fIsCandTrackSPDFirst && d->Pt()<fMaxPtCandTrackSPDFirst) {
 928       if (!bachelorTrack->HasPointOnITSLayer(0)) return 0;
 929     }
 930     if (fTrackCuts && fV0daughtersCuts) {
 931       AliAODVertex *vAOD = (AliAODVertex*)d->GetPrimaryVtx();
 932       Double_t pos[3]; vAOD->GetXYZ(pos);
 933       Double_t cov[6]; vAOD->GetCovarianceMatrix(cov);
 934       const AliESDVertex vESD(pos,cov,100.,100);
 935       if ( !(IsDaughterSelected(bachelorTrack,&vESD,fTrackCuts)) ||
 936            !(IsDaughterSelected(v0negativeTrack,&vESD,fV0daughtersCuts)) ||
 937            !(IsDaughterSelected(v0positiveTrack,&vESD,fV0daughtersCuts)) ) return 0;
 938     }
 939     //if (!AreDaughtersSelected(d)) return 0;
 940   }
 941
 942   Bool_t okLck0sp=kFALSE, okLcLpi=kFALSE, okLcLBarpi=kFALSE;
 943
 944   // selection on candidate
 945   if (selectionLevel==AliRDHFCuts::kAll ||
 946       selectionLevel==AliRDHFCuts::kCandidate) {
 947
 948     Double_t pt = d->Pt();
 949     Int_t ptbin = PtBin(pt);
 950
 951     Double_t mLcPDG  = TDatabasePDG::Instance()->GetParticle(4122)->Mass();
 952     Double_t mk0sPDG = TDatabasePDG::Instance()->GetParticle(310)->Mass();
 953     Double_t mLPDG   = TDatabasePDG::Instance()->GetParticle(3122)->Mass();
 954
 955     // K0S + p
 956     Double_t mk0s    = v0->MassK0Short();
 957     Double_t mLck0sp = d->InvMassLctoK0sP();
 958
 959     // Lambda + pi
 960     Double_t mlambda  = v0->MassLambda();
 961     Double_t malambda = v0->MassAntiLambda();
 962     Double_t mLcLpi   = d->InvMassLctoLambdaPi();
 963
 964     switch (cutIndex) {
 965     case 0:
 966       // cut on Lc mass with K0S+p hypothesis
 967       okLck0sp   = TMath::Abs(mLck0sp-mLcPDG)<=fCutsRD[GetGlobalIndex(0,ptbin)];
 968       okLcLpi    = kFALSE;
 969       okLcLBarpi = kFALSE;
 970       break;
 971     case 1:
 972       // cut on Lc mass with Lambda+pi hypothesis
 973       okLck0sp   = kFALSE;
 974       okLcLpi    = TMath::Abs(mLcLpi-mLcPDG)<=fCutsRD[GetGlobalIndex(1,ptbin)];
 975       okLcLBarpi = okLcLpi;
 976       break;
 977     case 2:
 978       // cuts on the V0 mass: K0S case
 979       okLck0sp   = TMath::Abs(mk0s-mk0sPDG)<=fCutsRD[GetGlobalIndex(2,ptbin)];
 980       okLcLpi    = kFALSE;
 981       okLcLBarpi = kFALSE;
 982       break;
 983     case 3:
 984       // cuts on the V0 mass: Lambda/LambdaBar case
 985       okLck0sp   = kFALSE;
 986       okLcLpi    = TMath::Abs(mlambda-mLPDG)<=fCutsRD[GetGlobalIndex(3,ptbin)];
 987       //okLcLpi    = okLcLpi && (bachelorTrack->Charge()==+1);
 988       okLcLBarpi = TMath::Abs(malambda-mLPDG)<=fCutsRD[GetGlobalIndex(3,ptbin)];
 989       //okLcLBarpi = okLcLBarpi && (bachelorTrack->Charge()==-1);
 990       break;
 991     case 4:
 992       // cuts on the minimum pt of bachelor
 993       okLck0sp   = TMath::Abs(bachelorTrack->Pt())>=fCutsRD[GetGlobalIndex(4,ptbin)];
 994       okLcLpi    = okLck0sp;
 995       okLcLBarpi = okLck0sp;
 996       break;
 997     case 5:
 998       // cuts on the minimum pt of positive V0daughter
 999       okLck0sp   = TMath::Abs(v0positiveTrack->Pt())>=fCutsRD[GetGlobalIndex(5,ptbin)];
1000       okLcLpi    = okLck0sp;
1001       okLcLBarpi = okLck0sp;
1002       break;
1003     case 6:
1004       // cuts on the minimum pt of negative V0daughter
1005       okLck0sp   = TMath::Abs(v0negativeTrack->Pt())>=fCutsRD[GetGlobalIndex(6,ptbin)];
1006       okLcLpi    = okLck0sp;
1007       okLcLBarpi = okLck0sp;
1008       break;
1009     case 7:
1010       // cut on cascade dca
1011       okLck0sp   = TMath::Abs(d->GetDCA())<=fCutsRD[GetGlobalIndex(7,ptbin)];
1012       okLcLpi    = okLck0sp;
1013       okLcLBarpi = okLck0sp;
1014       break;
1015     case 8:
1016       // cut on V0 dca
1017       okLck0sp   = TMath::Abs(v0->GetDCA())<=fCutsRD[GetGlobalIndex(8,ptbin)];
1018       okLcLpi    = okLck0sp;
1019       okLcLBarpi = okLck0sp;
1020       break;
1021     case 9:
1022       // cut on V0 cosine of pointing angle wrt PV
1023       okLck0sp   = d->CosV0PointingAngle()>=fCutsRD[GetGlobalIndex(9,ptbin)];
1024       okLcLpi    = okLck0sp;
1025       okLcLBarpi = okLck0sp;
1026       break;
1027     case 10:
1028       // cut on bachelor transverse impact parameter wrt PV
1029       okLck0sp   = TMath::Abs(d->Getd0Prong(0))<=fCutsRD[GetGlobalIndex(10,ptbin)];
1030       okLcLpi    = okLck0sp;
1031       okLcLBarpi = okLck0sp;
1032       break;
1033     case 11:
1034       // cut on V0 transverse impact parameter wrt PV
1035       okLck0sp   = TMath::Abs(d->Getd0Prong(1))<=fCutsRD[GetGlobalIndex(11,ptbin)];
1036       okLcLpi    = okLck0sp;
1037       okLcLBarpi = okLck0sp;
1038       break;
1039     case 12:
1040       // cut on K0S invariant mass veto
1041       okLcLpi    = TMath::Abs(mk0s-mk0sPDG)>=fCutsRD[GetGlobalIndex(12,ptbin)];
1042       okLcLBarpi = TMath::Abs(mk0s-mk0sPDG)>=fCutsRD[GetGlobalIndex(12,ptbin)];
1043       break;
1044     case 13:
1045       // cut on Lambda/LambdaBar invariant mass veto
1046       okLck0sp   = (TMath::Abs(mlambda-mLPDG)>=fCutsRD[GetGlobalIndex(13,ptbin)] &&
1047                     TMath::Abs(malambda-mLPDG)>=fCutsRD[GetGlobalIndex(13,ptbin)]);
1048       break;
1049     case 14:
1050       // cut on gamma invariant mass veto
1051       okLck0sp   = v0->InvMass2Prongs(0,1,11,11)>=fCutsRD[GetGlobalIndex(14,ptbin)];
1052       okLcLpi    = okLck0sp;
1053       okLcLBarpi = okLck0sp;
1054       break;
1055     case 15:
1056       // cut on V0 pT min
1057       okLck0sp   = v0->Pt()>=fCutsRD[GetGlobalIndex(15,ptbin)];
1058       okLcLpi    = okLck0sp;
1059       okLcLBarpi = okLck0sp;
1060       break;
1061     }
1062   }
1063
1064   Int_t returnvalue = okLck0sp+2*okLcLBarpi+4*okLcLpi;
1065   /*
1066     retvalue case
1067           1  Lc->K0S + p
1068           2  Lc->LambdaBar + pi
1069           3  Lc->K0S + p AND Lc->LambdaBar + pi
1070           4  Lc->Lambda + pi
1071           5  Lc->K0S + p AND Lc->Lambda + pi
1072           6  Lc->LambdaBar + pi AND Lc->Lambda + pi
1073           7  Lc->K0S + p AND Lc->LambdaBar + pi AND Lc->Lambda + pi
1074   */
1075
1076
1077   /*
1078   Int_t returnvaluePID = 7;
1079
1080   // selection on PID
1081   if (selectionLevel==AliRDHFCuts::kAll ||
1082       selectionLevel==AliRDHFCuts::kCandidate ||
1083       selectionLevel==AliRDHFCuts::kPID )
1084     returnvaluePID = IsSelectedPID(d);
1085   */
1086
1087   Int_t returnvalueTot = 0;
1088   //if ( fUsePID )
1089   //returnvalueTot = CombinePIDCuts(returnvalue,returnvaluePID);
1090   //else
1091   returnvalueTot = returnvalue;
1092
1093   return returnvalueTot;
1094
1095 }
1096 //----------------------------------
1097 void AliRDHFCutsLctoV0::SetStandardCutsPP2010() {
1098
1099  SetName("LctoV0ProductionCuts");
1100  SetTitle("Production cuts for Lc->V0+bachelor analysis");
1101
1102   AliESDtrackCuts* esdTrackCuts=new AliESDtrackCuts();
1103   esdTrackCuts->SetRequireSigmaToVertex(kFALSE);
1104   //default
1105   esdTrackCuts->SetRequireTPCRefit(kTRUE);
1106   esdTrackCuts->SetRequireITSRefit(kTRUE);
1107   esdTrackCuts->SetMinNClustersITS(0);//(4); // default is 5
1108   esdTrackCuts->SetMinNClustersTPC(70);
1109   //esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
1110   //                                       AliESDtrackCuts::kAny);
1111   // default is kBoth, otherwise kAny
1112   esdTrackCuts->SetMinDCAToVertexXY(0.);
1113   esdTrackCuts->SetPtRange(0.3,1.e10);
1114   //esdTrackCuts->SetEtaRange(-0.8,+0.8);
1115   esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
1116   AddTrackCuts(esdTrackCuts);
1117
1118
1119   AliESDtrackCuts* esdTrackCutsV0daughters=new AliESDtrackCuts();
1120   esdTrackCutsV0daughters->SetRequireSigmaToVertex(kFALSE);
1121   //default
1122   esdTrackCutsV0daughters->SetRequireTPCRefit(kTRUE);
1123   esdTrackCutsV0daughters->SetRequireITSRefit(kFALSE);//(kTRUE);
1124   esdTrackCutsV0daughters->SetMinNClustersITS(0);//(4); // default is 5
1125   esdTrackCutsV0daughters->SetMinNClustersTPC(70);
1126   //esdTrackCutsV0daughters->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
1127   //                                                  AliESDtrackCuts::kAny);
1128   // default is kBoth, otherwise kAny
1129   esdTrackCutsV0daughters->SetMinDCAToVertexXY(0.);
1130   esdTrackCutsV0daughters->SetPtRange(0.,1.e10);
1131   esdTrackCutsV0daughters->SetAcceptKinkDaughters(kFALSE);
1132   AddTrackCutsV0daughters(esdTrackCutsV0daughters);
1133
1134   const Int_t nptbins=1;
1135   Float_t* ptbins;
1136   ptbins=new Float_t[nptbins+1];
1137   ptbins[0]=0.;
1138   ptbins[1]=99999999.;
1139
1140   SetPtBins(nptbins+1,ptbins);
1141   SetPtBins(nptbins+1,ptbins);
1142
1143   const Int_t nvars=17;
1144
1145   Float_t** prodcutsval;
1146   prodcutsval=new Float_t*[nvars];
1147   for(Int_t ic=0;ic<nvars;ic++){prodcutsval[ic]=new Float_t[nptbins];}
1148   for(Int_t ipt2=0;ipt2<nptbins;ipt2++){
1149    prodcutsval[0][ipt2]=1.;    // inv. mass if K0S [GeV/c2]
1150    prodcutsval[1][ipt2]=1.;    // inv. mass if Lambda [GeV/c2]
1151    prodcutsval[2][ipt2]=0.05;  // inv. mass V0 if K0S [GeV/c2]
1152    prodcutsval[3][ipt2]=0.05;  // inv. mass V0 if Lambda [GeV/c2]
1153    prodcutsval[4][ipt2]=0.3;   // pT min bachelor track [GeV/c] // AOD by construction
1154    prodcutsval[5][ipt2]=0.;    // pT min V0-positive track [GeV/c]
1155    prodcutsval[6][ipt2]=0.;    // pT min V0-negative track [GeV/c]
1156    prodcutsval[7][ipt2]=1000.; // dca cascade cut [cm]
1157    prodcutsval[8][ipt2]=1.5;   // dca V0 cut [nSigma] // it's 1.5 x offline V0s
1158    prodcutsval[9][ipt2]=-1.;   // cosPA V0 cut // it's 0.90 x offline V0s at reconstruction level, 0.99 at filtering level
1159    prodcutsval[10][ipt2]=3.;   // d0 max bachelor wrt PV [cm]
1160    prodcutsval[11][ipt2]=1000.;// d0 max V0 wrt PV [cm]
1161    prodcutsval[12][ipt2]=0.;   // mass K0S veto [GeV/c2]
1162    prodcutsval[13][ipt2]=0.;   // mass Lambda/LambdaBar veto [GeV/c2]
1163    prodcutsval[14][ipt2]=0.;   // mass Gamma veto [GeV/c2]
1164    prodcutsval[15][ipt2]=0.;   // pT min V0 track [GeV/c]
1165    prodcutsval[16][ipt2]=0.;   // V0 type cut
1166   }
1167   SetCuts(nvars,nptbins,prodcutsval);
1168
1169   SetGlobalIndex(nvars,nptbins);
1170   SetPtBins(nptbins+1,ptbins);
1171
1172
1173   //pid settings
1174   //1. bachelor: default one
1175   AliAODPidHF* pidObjBachelor = new AliAODPidHF();
1176   Double_t sigmasBac[5]={3.,1.,1.,3.,3.}; // 0, 1(A), 2(A) -> TPC; 3 -> TOF; 4 -> ITS
1177   pidObjBachelor->SetSigma(sigmasBac);
1178   pidObjBachelor->SetAsym(kFALSE);
1179   pidObjBachelor->SetMatch(1);
1180   pidObjBachelor->SetTPC(kTRUE);
1181   pidObjBachelor->SetTOF(kTRUE);
1182   pidObjBachelor->SetTOFdecide(kFALSE);
1183   SetPidHF(pidObjBachelor);
1184
1185   //2. V0pos
1186   AliAODPidHF* pidObjV0pos = new AliAODPidHF();
1187   Double_t sigmasV0pos[5]={3.,1.,1.,3.,3.}; // 0, 1(A), 2(A) -> TPC; 3 -> TOF; 4 -> ITS
1188   pidObjV0pos->SetSigma(sigmasV0pos);
1189   pidObjV0pos->SetAsym(kFALSE);
1190   pidObjV0pos->SetMatch(1);
1191   pidObjV0pos->SetTPC(kTRUE);
1192   pidObjV0pos->SetTOF(kTRUE);
1193   pidObjV0pos->SetTOFdecide(kFALSE);
1194   SetPidV0pos(pidObjV0pos);
1195
1196   //2. V0neg
1197   AliAODPidHF* pidObjV0neg = new AliAODPidHF();
1198   Double_t sigmasV0neg[5]={3.,1.,1.,3.,3.}; // 0, 1(A), 2(A) -> TPC; 3 -> TOF; 4 -> ITS
1199   pidObjV0neg->SetSigma(sigmasV0neg);
1200   pidObjV0neg->SetAsym(kFALSE);
1201   pidObjV0neg->SetMatch(1);
1202   pidObjV0neg->SetTPC(kTRUE);
1203   pidObjV0neg->SetTOF(kTRUE);
1204   pidObjV0neg->SetTOFdecide(kFALSE);
1205   SetPidV0neg(pidObjV0neg);
1206
1207   SetUsePID(kFALSE);//(kTRUE);
1208
1209   //PrintAll();
1210
1211  for(Int_t iiv=0;iiv<nvars;iiv++){
1212   delete [] prodcutsval[iiv];
1213  }
1214  delete [] prodcutsval;
1215  prodcutsval=NULL;
1216  delete [] ptbins;
1217  ptbins=NULL;
1218
1219
1220  delete pidObjBachelor;
1221  pidObjBachelor=NULL;
1222
1223  delete pidObjV0pos;
1224  pidObjV0pos=NULL;
1225
1226  delete pidObjV0neg;
1227  pidObjV0neg=NULL;
1228
1229  return;
1230 }
1231 //------------------
1232 void AliRDHFCutsLctoV0::SetStandardCutsPbPb2010() {
1233
1234  SetName("LctoV0ProductionCuts");
1235  SetTitle("Production cuts for Lc->V0+bachelor analysis");
1236
1237  SetStandardCutsPP2010();
1238
1239  return;
1240 }
1241 //------------------
1242 void AliRDHFCutsLctoV0::SetStandardCutsPbPb2011() {
1243
1244   // Default 2010 PbPb cut object
1245   SetStandardCutsPbPb2010();
1246
1247   //
1248   // Enable all 2011 PbPb run triggers
1249   //
1250   SetTriggerClass("");
1251   ResetMaskAndEnableMBTrigger();
1252   EnableCentralTrigger();
1253   EnableSemiCentralTrigger();
1254
1255 }
1256 //-----------------------
1257 Int_t AliRDHFCutsLctoV0::GetV0Type(){
1258
1259   const Int_t nvars = this->GetNVars() ;
1260   //Float_t *vArray =GetCuts();
1261   //fV0Type = vArray[nvars-1];
1262   fV0Type = (this->GetCuts())[nvars-1];
1263   //this->GetCuts(vArray);
1264   TString *sVarNames=GetVarNames();
1265
1266   if(sVarNames[nvars-1].Contains("V0 type")) return (Int_t)fV0Type;
1267   else {AliInfo("AliRDHFCutsLctoV0 Last variable is not the V0 type!!!"); return -999;}
1268 }
1269
1270 //---------------------------------------------------------------------------
1271 void AliRDHFCutsLctoV0::PrintAll() const {
1272   //
1273   // print all cuts values
1274   //
1275
1276   printf("Minimum vtx type %d\n",fMinVtxType);
1277   printf("Minimum vtx contr %d\n",fMinVtxContr);
1278   printf("Max vtx red chi2 %f\n",fMaxVtxRedChi2);
1279   printf("Min SPD mult %d\n",fMinSPDMultiplicity);
1280   printf("Use PID %d (PID selection flag = %d) OldPid=%d\n",(Int_t)fUsePID,(Int_t)fPidSelectionFlag,fPidHF ? fPidHF->GetOldPid() : -1);
1281   printf("Remove daughters from vtx %d\n",(Int_t)fRemoveDaughtersFromPrimary);
1282   printf("Recompute primary vertex %d\n",(Int_t)fRecomputePrimVertex);
1283   printf("Physics selection: %s\n",fUsePhysicsSelection ? "Yes" : "No");
1284   printf("Pileup rejection: %s\n",(fOptPileup > 0) ? "Yes" : "No");
1285   printf("UseTrackSelectionWithFilterBits: %s\n",fUseTrackSelectionWithFilterBits ? "Yes" : "No");
1286   if(fOptPileup==1) printf(" -- Reject pileup event");
1287   if(fOptPileup==2) printf(" -- Reject tracks from pileup vtx");
1288   if(fUseCentrality>0) {
1289     TString estimator="";
1290     if(fUseCentrality==1) estimator = "V0";
1291     if(fUseCentrality==2) estimator = "Tracks";
1292     if(fUseCentrality==3) estimator = "Tracklets";
1293     if(fUseCentrality==4) estimator = "SPD clusters outer";
1294     printf("Centrality class considered: %.1f-%.1f, estimated with %s",fMinCentrality,fMaxCentrality,estimator.Data());
1295   }
1296   if(fIsCandTrackSPDFirst) printf("Check for candidates with pt < %2.2f, that daughters fullfill kFirst criteria\n",fMaxPtCandTrackSPDFirst);
1297   /*
1298   if(fVarNames){
1299     cout<<"Array of variables"<<endl;
1300     for(Int_t iv=0;iv<fnVars;iv++){
1301       cout<<fVarNames[iv]<<"\t";
1302     }
1303     cout<<endl;
1304   }
1305   if(fVarsForOpt){
1306     cout<<"Array of optimization"<<endl;
1307     for(Int_t iv=0;iv<fnVars;iv++){
1308       cout<<fVarsForOpt[iv]<<"\t";
1309     }
1310     cout<<endl;
1311   }
1312   if(fIsUpperCut){
1313     cout<<"Array of upper/lower cut"<<endl;
1314    for(Int_t iv=0;iv<fnVars;iv++){
1315      cout<<fIsUpperCut[iv]<<"\t";
1316    }
1317    cout<<endl;
1318   }
1319   if(fPtBinLimits){
1320     cout<<"Array of ptbin limits"<<endl;
1321     for(Int_t ib=0;ib<fnPtBinLimits;ib++){
1322       cout<<fPtBinLimits[ib]<<"\t";
1323     }
1324     cout<<endl;
1325   }
1326   if(fCutsRD){
1327     cout<<"Matrix of cuts"<<endl;
1328    for(Int_t iv=0;iv<fnVars;iv++){
1329      for(Int_t ib=0;ib<fnPtBins;ib++){
1330        cout<<"fCutsRD["<<iv<<"]["<<ib<<"] = "<<fCutsRD[GetGlobalIndex(iv,ib)]<<"\t";
1331      }
1332      cout<<endl;
1333    }
1334    cout<<endl;
1335   }
1336   */
1337
1338   return;
1339
1340 }
1341
1342 //-------------------------
1343 Bool_t AliRDHFCutsLctoV0::IsInFiducialAcceptance(Double_t pt, Double_t y) const
1344 {
1345   //
1346   //
1347   // Checking if Lc is in fiducial acceptance region
1348   //
1349   //
1350
1351   if(fMaxRapidityCand>-998.){
1352     if(TMath::Abs(y) > fMaxRapidityCand) return kFALSE;
1353     else return kTRUE;
1354   }
1355
1356   if(pt > 5.) {
1357     // applying cut for pt > 5 GeV
1358     AliDebug(2,Form("pt of Lambda_c = %f (> 5), cutting at |y| < 0.8",pt));
1359     if (TMath::Abs(y) > 0.8) return kFALSE;
1360
1361   } else {
1362     // appliying smooth cut for pt < 5 GeV
1363     Double_t maxFiducialY = -0.2/15*pt*pt+1.9/15*pt+0.5;
1364     Double_t minFiducialY = 0.2/15*pt*pt-1.9/15*pt-0.5;
1365     AliDebug(2,Form("pt of Lambda_c = %f (< 5), cutting  according to the fiducial zone [%f, %f]\n",pt,minFiducialY,maxFiducialY));
1366     if (y < minFiducialY || y > maxFiducialY) return kFALSE;
1367   }
1368   //
1369   return kTRUE;
1370 }