AliHFAODMCParticleContainer.h

Go to the documentation of this file.

#ifndef ALIHFAODMCPARTICLECONTAINER_H
#define ALIHFAODMCPARTICLECONTAINER_H

/**************************************************************************
* Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/

#include "AliMCParticleContainer.h"
#include "AliAnalysisTaskDmesonJets.h"

class AliHFAODMCParticleContainer : public AliMCParticleContainer {
 public:
  AliHFAODMCParticleContainer();
  AliHFAODMCParticleContainer(const char *name);

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

  void   SetRejectQuarkNotFound(Bool_t c)  { fRejectedOrigin = c ?  fRejectedOrigin | AliAnalysisTaskDmesonJets::kUnknownQuark : fRejectedOrigin & ~AliAnalysisTaskDmesonJets::kUnknownQuark; }
  Bool_t GetRejectQuarkNotFound() const    { return (fRejectedOrigin & AliAnalysisTaskDmesonJets::kUnknownQuark) != 0 ; }

  void   SetRejectDfromB(Bool_t c)         { fRejectedOrigin = c ?  fRejectedOrigin | AliAnalysisTaskDmesonJets::kFromBottom : fRejectedOrigin & ~AliAnalysisTaskDmesonJets::kFromBottom; }
  Bool_t GetRejectDfromB() const           { return (fRejectedOrigin & AliAnalysisTaskDmesonJets::kFromBottom) != 0 ; }

  void   SetKeepOnlyDfromB(Bool_t c)       { fRejectedOrigin = c ?  fRejectedOrigin | AliAnalysisTaskDmesonJets::kFromCharm : fRejectedOrigin & ~AliAnalysisTaskDmesonJets::kFromCharm; }
  Bool_t GetKeepOnlyDfromB() const         { return (fRejectedOrigin & AliAnalysisTaskDmesonJets::kFromCharm) != 0 ; }

  void   SetKeepOnlyD0toKpi()              { fAcceptedDecay = AliAnalysisTaskDmesonJets::kDecayD0toKpi     ; }
  void   SetKeepOnlyDStartoKpipi()         { fAcceptedDecay = AliAnalysisTaskDmesonJets::kDecayDStartoKpipi; }

  void   SetRejectedOriginMap(UInt_t m)    { fRejectedOrigin = m; }
  void   SetAcceptedDecayMap(UInt_t m)     { fAcceptedDecay  = m; }

  void   SelectCharmtoD0toKpi();
  void   SelectCharmtoDStartoKpipi();
  
  Bool_t          AcceptMCParticle(AliAODMCParticle* vp);
  Bool_t          AcceptMCParticle(Int_t i);

 protected:
  Bool_t          IsSpecialPDGDaughter(AliAODMCParticle* part) const;
  Bool_t          IsSpecialPDGDaughter(Int_t iPart) const;
 
  Int_t           fSpecialPDG;             
  UInt_t          fRejectedOrigin;         
  UInt_t          fAcceptedDecay;          
  
 private:
  AliHFAODMCParticleContainer(const AliHFAODMCParticleContainer&);            // not implemented
  AliHFAODMCParticleContainer &operator=(const AliHFAODMCParticleContainer&); // not implemented

  ClassDef(AliHFAODMCParticleContainer, 2);
};
#endif