Adding macro to plot <Ncoll>

[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / AliRsnCutDaughterD0.h

#ifndef ALIRSNCUTDAUGHTERD0_H
#define ALIRSNCUTDAUGHTERD0_H

//
// Cuts for selecting good pion candidates for D0 analysis
// with the data samples from pp/PbPb/pPb runs in 2010/2011/2012/2013.
// Applies track quality selection plus PID selection,
// with different tolerance ranges depending on the momentum.
//

#include "AliVTrack.h"
#include "AliRsnCut.h"
#include "AliRsnCutTrackQuality.h"

class AliRsnCutDaughterD0 : public AliRsnCut {

 public:

  AliRsnCutDaughterD0(const char *name = "", AliPID::EParticleType pid = AliPID::kPion);
  AliRsnCutDaughterD0(const AliRsnCutDaughterD0 &copy);
  AliRsnCutDaughterD0 &operator=(const AliRsnCutDaughterD0 &copy);
  virtual ~AliRsnCutDaughterD0() { }

  void           SetNoPID(Bool_t yn = kTRUE)                  {fNoPID = yn;}
  void           SetPtDependentPIDCut(Bool_t yn = kTRUE)      {fPtDepPIDCut = yn;}
  void           SetPID(AliPID::EParticleType type)           {fPID = type;}
   
  void           SetTPCPionPIDCut(Double_t value)             {fPionTPCPIDCut = value;}
  void           SetTPCKaonPIDCut(Double_t value)             {fKaonTPCPIDCut = value;}
   
  void           SetTOFPionPIDCut(Double_t value)             {fPionTOFPIDCut = value;}
  void           SetTOFKaonPIDCut(Double_t value)             {fKaonTOFPIDCut = value;}
   
  AliRsnCutTrackQuality *CutQuality()                       {return &fCutQuality;}
  Bool_t                 MatchTOF(const AliVTrack *vtrack);
  virtual Bool_t         IsSelected(TObject *obj);
   
 private:
  Bool_t                fNoPID;            // flag to switch off PID check
  AliPID::EParticleType fPID;              // PID for track
  AliRsnCutTrackQuality fCutQuality;       // track quality cut 

 protected:
  Double_t         fPionTPCPIDCut;                // TPC nsigmas for pions
  Double_t         fKaonTPCPIDCut;                // TPC nsigmas for kaons
  Double_t         fPionTOFPIDCut;                // TOF nsigmas for pions
  Double_t         fKaonTOFPIDCut;                // TOF nsigmas for kaons
  Bool_t           fPtDepPIDCut;                  // flag for setting a pt dependent or independent PID cut

  ClassDef(AliRsnCutDaughterD0, 2)          // cut definitions for D0
    };

//__________________________________________________________________________________________________
inline Bool_t AliRsnCutDaughterD0::MatchTOF(const AliVTrack *vtrack)
{
  //
  // Checks if the track has matched the TOF detector
  //

  if (!vtrack) {
    AliWarning("NULL argument: impossible to check status");
    return kFALSE;
  }

  if (!(vtrack->GetStatus() & AliESDtrack::kTOFout)) return kFALSE;
  if (!(vtrack->GetStatus() & AliESDtrack::kTIME  )) return kFALSE;

  return kTRUE;
}

#endif