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();
  
  virtual Bool_t AcceptMCParticle(const AliAODMCParticle *vp, UInt_t &rejectionReason) const;
  virtual Bool_t AcceptMCParticle(Int_t i, UInt_t &rejectionReason) const;

  Bool_t IsSpecialPDGFound() const;

  void SetHistOrigin(TH1* h) { fHistOrigin = h; }

 protected:
  Bool_t          IsSpecialPDGDaughter(const AliAODMCParticle* part) const;
  Bool_t          IsSpecialPDG(const AliAODMCParticle* part, TH1* histOrigin = 0) const;

  Int_t           fSpecialPDG;             
  UInt_t          fRejectedOrigin;         
  UInt_t          fAcceptedDecay;          
  TH1*            fHistOrigin;             

 private:
  AliHFAODMCParticleContainer(const AliHFAODMCParticleContainer&);            // not implemented
  AliHFAODMCParticleContainer &operator=(const AliHFAODMCParticleContainer&); // not implemented

  ClassDef(AliHFAODMCParticleContainer, 2);
};
#endif