PWGLF/RESONANCES/AliRsnCutSetDaughterParticle.cxx

   1 //
   2 // Author: Francesca Bellini (fbellini@cern.ch)
   3 //
   4 //
   5
   6 #include <Riostream.h>
   7
   8 #include "AliPID.h"
   9 #include "AliPIDResponse.h"
  10 #include "AliRsnCutSetDaughterParticle.h"
  11
  12 class AliRsnCutPIDNSigma;
  13 class AliRsnPIDRange;
  14
  15 ClassImp(AliRsnCutSetDaughterParticle)
  16
  17 //__________________________________________________________________________________________________
  18 AliRsnCutSetDaughterParticle::AliRsnCutSetDaughterParticle() :
  19    AliRsnCutSet("AliRsnCutSetDaughterParticle", AliRsnTarget::kDaughter),
  20    fPID(AliPID::kPion),
  21    fAppliedCutSetID(AliRsnCutSetDaughterParticle::kNDaughterCuts),
  22    fNsigmaTPC(1E20),
  23    fNsigmaTOF(1E20),
  24    fCutQuality(0x0),
  25    fAODTrkCutFilterBit(0)
  26 {
  27    //
  28    // Default constructor
  29 }
  30
  31 //__________________________________________________________________________________________________
  32 AliRsnCutSetDaughterParticle::AliRsnCutSetDaughterParticle(const char *name, AliRsnCutSetDaughterParticle::ERsnDaughterCutSet cutSetID, AliPID::EParticleType pid, Float_t nSigmaFast = -1.0, Int_t AODfilterBit = 0) :
  33    AliRsnCutSet(name, AliRsnTarget::kDaughter),
  34    fPID(pid),
  35    fAppliedCutSetID(cutSetID),
  36    fNsigmaTPC(1E20),
  37    fNsigmaTOF(1E20),
  38    fCutQuality(new AliRsnCutTrackQuality("CutQuality")),
  39    fAODTrkCutFilterBit(AODfilterBit)
  40 {
  41    //
  42    // Constructor
  43    //
  44    if ( (nSigmaFast<=0) &&
  45         ((cutSetID == AliRsnCutSetDaughterParticle::kFastTPCpidNsigma) || (cutSetID == AliRsnCutSetDaughterParticle::kFastTOFpidNsigma)) ) {
  46       AliError("Requested fast n-sigma PID with invalid value for n. Setting n = 1E20");
  47    } else {
  48       if (cutSetID == AliRsnCutSetDaughterParticle::kFastTPCpidNsigma) {
  49          fNsigmaTPC = nSigmaFast;
  50       }
  51       if ( (cutSetID == AliRsnCutSetDaughterParticle::kFastTOFpidNsigma) ||
  52            (cutSetID == AliRsnCutSetDaughterParticle::kTOFpidKstarPbPb2010) ) {
  53          fNsigmaTOF = nSigmaFast;
  54       }
  55    }
  56
  57    Init();
  58 }
  59
  60 //__________________________________________________________________________________________________
  61 AliRsnCutSetDaughterParticle::AliRsnCutSetDaughterParticle(const AliRsnCutSetDaughterParticle &copy) :
  62    AliRsnCutSet(copy),
  63    fPID(copy.fPID),
  64    fAppliedCutSetID(copy.fAppliedCutSetID),
  65    fNsigmaTPC(copy.fNsigmaTPC),
  66    fNsigmaTOF(copy.fNsigmaTOF),
  67    fCutQuality(copy.fCutQuality),
  68    fAODTrkCutFilterBit(copy.fAODTrkCutFilterBit)
  69 {
  70    //
  71    // copy constructor
  72 }
  73
  74 //__________________________________________________________________________________________________
  75 AliRsnCutSetDaughterParticle &AliRsnCutSetDaughterParticle::operator=(const AliRsnCutSetDaughterParticle &copy)
  76 {
  77    //
  78    // operator =
  79    //
  80    AliRsnCutSet::operator=(copy);
  81    if (this == &copy)
  82       return *this;
  83    fPID=copy.fPID;
  84    fAppliedCutSetID=copy.fAppliedCutSetID;
  85    fNsigmaTPC=copy.fNsigmaTPC;
  86    fNsigmaTOF=copy.fNsigmaTOF;
  87    fAODTrkCutFilterBit=copy.fAODTrkCutFilterBit;
  88    fCutQuality=copy.fCutQuality;
  89    return (*this);
  90 }
  91
  92 //----------------------------------------------------------------------------
  93 AliRsnCutSetDaughterParticle::~AliRsnCutSetDaughterParticle()
  94 {
  95    //
  96    //destructor
  97    //
  98    if (fCutQuality)
  99       delete fCutQuality;
 100 }
 101 //----------------------------------------------------------------------------
 102 void AliRsnCutSetDaughterParticle::Init()
 103 {
 104    //
 105    // init cut sets by setting variable params
 106    //
 107    fCutQuality->SetDefaults2010();
 108    fCutQuality->SetPtRange(0.15, 20.0);
 109    fCutQuality->SetEtaRange(-0.8, 0.8);
 110    fCutQuality->SetDCARPtFormula("0.0182+0.0350/pt^1.01");
 111    fCutQuality->SetDCAZmax(2.0);
 112    fCutQuality->SetSPDminNClusters(1);
 113    fCutQuality->SetITSminNClusters(0);
 114    fCutQuality->SetITSmaxChi2(36);
 115    fCutQuality->SetTPCminNClusters(70);
 116    fCutQuality->SetTPCmaxChi2(4.0);
 117    fCutQuality->SetRejectKinkDaughters();
 118    fCutQuality->SetAODTestFilterBit(fAODTrkCutFilterBit);
 119    //fCutQuality->SetITSmaxChi2(36);
 120    //fCutQuality->SetMaxChi2TPCConstrainedGlobal(36);
 121
 122    AliRsnCutTOFMatch  *iCutTOFMatch     = new AliRsnCutTOFMatch("CutTOFMatch");
 123    AliRsnCutPIDNSigma *iCutTPCNSigma    = new AliRsnCutPIDNSigma("CutTPCNSigma", fPID, AliRsnCutPIDNSigma::kTPC);//, AliRsnCutPIDNSigma::kTPCinnerP );
 124    AliRsnCutPIDNSigma *iCutTPCTOFNSigma = new AliRsnCutPIDNSigma("CutTPCTOFNSigma", fPID, AliRsnCutPIDNSigma::kTPC);//, AliRsnCutPIDNSigma::kTPCinnerP );
 125    AliRsnCutPIDNSigma *iCutTOFNSigma    = new AliRsnCutPIDNSigma("CutTOFNSigma", fPID, AliRsnCutPIDNSigma::kTOF);//, AliRsnCutPIDNSigma::kP );
 126
 127    switch (fAppliedCutSetID)
 128    {
 129       case AliRsnCutSetDaughterParticle::kNoCuts :
 130          AliInfo("No cuts applied to daughter particle");
 131          break;
 132
 133       case AliRsnCutSetDaughterParticle::kQualityStd2010 :
 134          AddCut(fCutQuality);
 135          SetCutScheme(fCutQuality->GetName());
 136          break;
 137
 138       case AliRsnCutSetDaughterParticle::kQualityStd2011:
 139          //fCutQuality->SetAODTestFilterBit(10);     //1024
 140          AddCut(fCutQuality);
 141          SetCutScheme(fCutQuality->GetName());
 142          break;
 143
 144       case AliRsnCutSetDaughterParticle::kTOFMatch :
 145          AddCut(fCutQuality);
 146          AddCut(iCutTOFMatch);
 147          SetCutScheme( Form("%s&(%s)",fCutQuality->GetName(), iCutTOFMatch->GetName()) );
 148          break;
 149
 150       case    AliRsnCutSetDaughterParticle::kFastTPCpidNsigma :
 151          if (fNsigmaTPC <= 0.0) {
 152             AliWarning(Form("Invalid number of sigmas required for %s. Setting default nSigma = 10",iCutTPCNSigma->GetName()));
 153             SetNsigmaForFastTPCpid(10.0);
 154          }
 155          iCutTPCNSigma->SinglePIDRange(fNsigmaTPC);
 156          AddCut(fCutQuality);
 157          AddCut(iCutTPCNSigma);
 158          SetCutScheme( Form("%s&%s",fCutQuality->GetName(), iCutTPCNSigma->GetName()) );
 159          break;
 160
 161       case    AliRsnCutSetDaughterParticle::kFastTOFpidNsigma :
 162          if (fNsigmaTOF <= 0.0) {
 163             AliWarning(Form("Invalid number of sigmas required for %s. Setting default nSigma = 10",iCutTOFNSigma->GetName()));
 164             SetNsigmaForFastTOFpid(10.0);
 165          }
 166          iCutTOFNSigma->SinglePIDRange(fNsigmaTOF);
 167          //iCutTOFNSigma->AddPIDRange(3.0, 0.35, 1E20);
 168          AddCut(fCutQuality);
 169          AddCut(iCutTOFNSigma);
 170          SetCutScheme( Form("%s&%s",fCutQuality->GetName(), iCutTOFNSigma->GetName()) );
 171          break;
 172
 173       case    AliRsnCutSetDaughterParticle::kTPCTOFpidKstarPP2010 :
 174          /* Set TPC  PID (if no TOF)*/
 175          // all   below   350         MeV: 5sigma
 176          // all   between 350 and 500 MeV: 3sigma
 177          // pios above   500         MeV: 2sigma
 178          // kaons between 500 and 700 MeV: 2sigma
 179          // kaons above   700         MeV: rejected
 180          iCutTPCNSigma->AddPIDRange(5.0, 0.0, 0.35);
 181          iCutTPCNSigma->AddPIDRange(3.0, 0.35, 0.5);
 182          if (fPID==AliPID::kPion)
 183             iCutTPCNSigma->AddPIDRange(2.0, 0.5, 1E20);
 184          if (fPID==AliPID::kKaon)
 185             iCutTPCNSigma->AddPIDRange(2.0, 0.5, 0.7);
 186
 187          AddCut(fCutQuality);
 188          AddCut(iCutTOFMatch);
 189          AddCut(iCutTPCNSigma);
 190
 191          /* set TPC+TOF PID*/
 192          iCutTPCTOFNSigma->SinglePIDRange(5.0);
 193          iCutTOFNSigma->AddPIDRange(3.0, 0.0, 1.5);
 194          iCutTOFNSigma->AddPIDRange(2.0, 1.5, 1E20);
 195
 196          AddCut(iCutTPCTOFNSigma);
 197          AddCut(iCutTOFNSigma);
 198
 199          // scheme:
 200          // quality & [ (TOF & TPCTOF) || (!TOFmatch & TPConly) ]
 201          SetCutScheme( Form("%s&((%s&%s)|((!%s)&%s))",fCutQuality->GetName(), iCutTPCTOFNSigma->GetName(), iCutTOFNSigma->GetName(), iCutTOFMatch->GetName(), iCutTPCNSigma->GetName()) ) ;
 202          break;
 203
 204       case AliRsnCutSetDaughterParticle::kTOFpidKstarPbPb2010:
 205          if (fNsigmaTOF <= 0.0) {
 206             AliWarning(Form("Invalid number of sigmas required for %s. Setting default nSigma = 10",iCutTOFNSigma->GetName()));
 207             SetNsigmaForFastTOFpid(10.0);
 208          }
 209          iCutTOFNSigma->SinglePIDRange(fNsigmaTOF);
 210          //iCutTOFNSigma->AddPIDRange(3.0, 0.35, 1E20);
 211          iCutTPCTOFNSigma->SinglePIDRange(5.0); //5-sigma veto on tpc signal
 212
 213          AddCut(fCutQuality);
 214          AddCut(iCutTOFNSigma);
 215          AddCut(iCutTPCTOFNSigma);
 216          SetCutScheme( Form("%s&%s&%s",fCutQuality->GetName(), iCutTOFNSigma->GetName(), iCutTPCTOFNSigma->GetName()) );
 217          break;
 218
 219       case AliRsnCutSetDaughterParticle::kTOFTPCmismatchKstarPbPb2010:
 220          iCutTPCTOFNSigma->SinglePIDRange(5.0); //5-sigma veto on tpc signal
 221          AddCut(fCutQuality);
 222          AddCut(iCutTOFMatch);
 223          AddCut(iCutTPCTOFNSigma);
 224          SetCutScheme( Form("%s&%s&%s",fCutQuality->GetName(), iCutTOFMatch->GetName(), iCutTPCTOFNSigma->GetName()) );
 225          break;
 226
 227       default :
 228          break;
 229    }
 230
 231 }
 232
 233
 234