PMGDecayProductsSelectionTool.h

Go to the documentation of this file.

/*
 Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
 */

 
 
 
#ifndef PMG_TOOLS__PMG_DECAY_PRODUCTS_SELECTION_TOOL_H
#define PMG_TOOLS__PMG_DECAY_PRODUCTS_SELECTION_TOOL_H
 
#include <AsgTools/AsgTool.h>
#include <PATCore/IAsgSelectionTool.h>
#include <xAODTruth/TruthParticle.h>
#include <xAODTruth/TruthParticleContainer.h>
 
 
namespace PMGTools
{
 
  class PMGDecayProductsSelectionTool final
    : public asg::AsgTool, virtual public IAsgSelectionTool
  {
    // Create a proper constructor for Athena
    ASG_TOOL_CLASS (PMGDecayProductsSelectionTool, IAsgSelectionTool)
 
 
    
  public:
    PMGDecayProductsSelectionTool (const std::string& name);
 
 
 
 
    //
    // inherited interface
    //
 
    virtual StatusCode initialize () override;
 
    virtual const asg::AcceptInfo& getAcceptInfo () const override;
 
    virtual asg::AcceptData accept (const xAOD::IParticle* /*part*/) const override;
 
 
 
    //
    // private interface
    //

  private:
    asg::AcceptData hasRequiredInitialParent (const xAOD::TruthParticle *truthParticle, asg::AcceptData& acceptData) const;

  private:
    size_t getNParents (const xAOD::TruthParticle *truthParticle) const;

  private:
    const xAOD::TruthParticle* getParent (const xAOD::TruthParticle *truthParticle,
                                          size_t index) const;

 
  private:
    std::vector<int> m_requiredParentPDGIDs;
    std::vector<int> m_allowedIntermediatePDGIDs;

 
  private:
    int m_truthParticleIndex{ -1 };
    int m_requiredParentIndex{ -1 };

  private:
    asg::AcceptInfo m_accept;

  private:
    std::unique_ptr<const SG::AuxElement::Accessor<std::vector<ElementLink<xAOD::TruthParticleContainer>>>> m_parentsAccessor{};
  };
}
 
#endif