AliHFAODMCParticleContainer.cxx

Go to the documentation of this file.

/*************************************************************************
* 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 <TClonesArray.h>
#include <AliAODMCParticle.h>
#include <AliLog.h>

#include "AliHFAODMCParticleContainer.h"

ClassImp(AliHFAODMCParticleContainer)

AliHFAODMCParticleContainer::AliHFAODMCParticleContainer() :
  AliMCParticleContainer(),
  fSpecialPDG(0),
  fRejectedOrigin(0),
  fAcceptedDecay(0)
{
  // Constructor.
}

AliHFAODMCParticleContainer::AliHFAODMCParticleContainer(const char *name) :
  AliMCParticleContainer(name),
  fSpecialPDG(0),
  fRejectedOrigin(AliAnalysisTaskDmesonJets::kUnknownQuark | AliAnalysisTaskDmesonJets::kFromBottom),
  fAcceptedDecay(AliAnalysisTaskDmesonJets::kAnyDecay)
{
  // Constructor.
}


void AliHFAODMCParticleContainer::SelectCharmtoD0toKpi()
{
  SetSpecialPDG(421);
  SetKeepOnlyD0toKpi();
  SetRejectDfromB(kTRUE);
  SetRejectQuarkNotFound(kTRUE);
  SetKeepOnlyDfromB(kFALSE);
}

void AliHFAODMCParticleContainer::SelectCharmtoDStartoKpipi()
{
  SetSpecialPDG(413);
  SetKeepOnlyDStartoKpipi();
  SetRejectDfromB(kTRUE);
  SetRejectQuarkNotFound(kTRUE);
  SetKeepOnlyDfromB(kFALSE);
}

Bool_t AliHFAODMCParticleContainer::ApplyMCParticleCuts(const AliAODMCParticle* part, UInt_t &rejectionReason) const
{
  Int_t partPdgCode = TMath::Abs(part->PdgCode());

  Bool_t isSpecialPdg = (fSpecialPDG != 0 && partPdgCode == fSpecialPDG && part->IsPrimary());

  if (isSpecialPdg) {
    AliAnalysisTaskDmesonJets::EMesonOrigin_t origin = AliAnalysisTaskDmesonJets::AnalysisEngine::CheckOrigin(part, fClArray);

    if ((origin & fRejectedOrigin) != 0) isSpecialPdg = kFALSE;

    AliAnalysisTaskDmesonJets::EMesonDecayChannel_t decayChannel = AliAnalysisTaskDmesonJets::AnalysisEngine::CheckDecayChannel(part, fClArray);

    if ((decayChannel & fAcceptedDecay) == 0) isSpecialPdg = kFALSE;

    AliDebug(2, Form("Including particle %d (PDG = %d, pT = %.3f, eta = %.3f, phi = %.3f)",
                     part->Label(), partPdgCode, part->Pt(), part->Eta(), part->Phi()));

    // Special PDG particle, skip regular MC particle cuts and apply particle cuts.
    return ApplyParticleCuts(part, rejectionReason);
  }

  if (IsSpecialPDGDaughter(part)) {
    rejectionReason = kHFCut;
    return kFALSE;  // daughter of a special PDG particle, reject it without any other check.
  }

  // Not a special PDG particle, and not a daughter of a special PDG particle. Apply regular MC particle cuts.
  return AliMCParticleContainer::ApplyMCParticleCuts(part, rejectionReason);
}

Bool_t AliHFAODMCParticleContainer::IsSpecialPDGDaughter(const AliAODMCParticle* part) const
{
  if (fSpecialPDG == 0) return kTRUE;
  
  const AliAODMCParticle* pm = part;
  Int_t imo = -1;
  while (pm != 0) {
    imo = pm->GetMother();
    if (imo < 0) break;
    pm = static_cast<const AliAODMCParticle*>(fClArray->At(imo));
    if (TMath::Abs(pm->GetPdgCode()) == fSpecialPDG && pm->IsPrimary()) {
      AliDebug(2, Form("Rejecting particle (PDG = %d, pT = %.3f, eta = %.3f, phi = %.3f) daughter of %d (PDG = %d, pT = %.3f, eta = %.3f, phi = %.3f)",
                       part->PdgCode(), part->Pt(), part->Eta(), part->Phi(), imo, pm->PdgCode(), pm->Pt(), pm->Eta(), pm->Phi()));
      return kTRUE;
    }
  }
  return kFALSE;
}

Bool_t AliHFAODMCParticleContainer::IsSpecialPDGFound() const
{
  for (auto part : accepted()) {
    if (TMath::Abs(part->GetPdgCode()) == fSpecialPDG) return kTRUE;
  }
  return kFALSE;
}