AliMCHFParticleSelector.h

Go to the documentation of this file.

#ifndef ALIMCHFPARTICLESELECTOR_H
#define ALIMCHFPARTICLESELECTOR_H

#include "AliEmcalMCTrackSelector.h"

class AliMCHFParticleSelector : public AliEmcalMCTrackSelector {
 public:
  AliMCHFParticleSelector();
  AliMCHFParticleSelector(const char *name);
  virtual ~AliMCHFParticleSelector();

  void   SetSpecialPDG(Int_t pdg)          { fSpecialPDG = pdg            ; }

  void   SetRejectQuarkNotFound(Bool_t c)  { fRejectQuarkNotFound = c     ; }
  Bool_t GetRejectQuarkNotFound() const    { return fRejectQuarkNotFound  ; }

  void   SetRejectDfromB(Bool_t c)         { fRejectDfromB = c            ; }
  Bool_t GetRejectDfromB() const           { return fRejectDfromB         ; }

  void   SetKeepOnlyDfromB(Bool_t c)       { fKeepOnlyDfromB = c          ; }
  Bool_t GetKeepOnlyDfromB() const         { return fKeepOnlyDfromB       ; }

  void   SetKeepOnlyD0toKpi(Bool_t c)      { fKeepOnlyD0toKpi = c         ; }
  Bool_t GetKeepOnlyD0toKpi() const        { return fKeepOnlyD0toKpi      ; }

  void   SetKeepOnlyDStartoKpipi(Bool_t c) { fKeepOnlyDStartoKpipi = c    ; }
  Bool_t GetKeepOnlyDStartoKpipi() const   { return fKeepOnlyDStartoKpipi ; }

  void   SelectCharmtoD0toKpi();
  void   SelectCharmtoDStartoKpipi();
  
 protected:
  Bool_t          AcceptParticle(AliAODMCParticle* part) const;

  Bool_t          IsSpecialPDGDaughter(AliAODMCParticle* part) const;
  Bool_t          IsSpecialPDGDaughter(Int_t iPart) const;
 
  Int_t           fSpecialPDG;             //  include particles with this PDG code even if they are not primary particles (and exclude their daughters)
  Bool_t          fRejectQuarkNotFound;    //  reject D mesons for which the original charm or bottom quark could not be found (MC)
  Bool_t          fRejectDfromB;           //  reject D mesons coming from a B meson decay (MC)
  Bool_t          fKeepOnlyDfromB;         //  only accept D mesons coming from a B meson decay (MC)
  Bool_t          fKeepOnlyD0toKpi;        //  only accept D0 mesons decaying to K pi
  Bool_t          fKeepOnlyDStartoKpipi;   //  only accept D* -> D0 pi -> K pi pi
  
 private:
  AliMCHFParticleSelector(const AliMCHFParticleSelector&);            // not implemented
  AliMCHFParticleSelector &operator=(const AliMCHFParticleSelector&); // not implemented

  ClassDef(AliMCHFParticleSelector, 2); // Task to select particle in MC events
};
#endif