TriggerMatchingTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORKTRIGGER_TRIGGERMATCHINGTOOL_H
#define DERIVATIONFRAMEWORKTRIGGER_TRIGGERMATCHINGTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "CxxUtils/checker_macros.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "TriggerMatchingTool/IIParticleRetrievalTool.h"
#include "TriggerMatchingTool/IMatchScoringTool.h"
#include "TrigDecisionTool/TrigDecisionTool.h"
#include "xAODBase/IParticle.h"
#include "xAODBase/IParticleContainer.h"
#include "xAODTrigger/TrigCompositeContainer.h"
#include "StoreGate/ReadHandleKey.h"
#include <vector>
 
namespace DerivationFramework
{
  class TriggerMatchingTool final : public AthAlgTool,
                                    virtual public IAugmentationTool
  {
  public:
    using particleVec_t = std::vector<const xAOD::IParticle *>;
    TriggerMatchingTool(
        const std::string &type,
        const std::string &name,
        const IInterface *pSvcLocator);
 
    virtual StatusCode initialize() override;
 
    virtual StatusCode addBranches() const override;
 
  private:
    // Properties
    mutable std::vector<std::string> m_chainNames ATLAS_THREAD_SAFE;
    // This being mutable isn't exactly the greatest thing but we can't check
    // whether the chains actually exist before the first event. A better
    // solution would be to filter the chains by what is available in the
    // input file but I don't know if that's possible.
    // It is thread-safe because the mutable operation is done as part of a
    // static const initialization on the first call of addBranches().
 
    ToolHandle<Trig::IIParticleRetrievalTool> m_trigParticleTool{"Trig::IParticleRetrievalTool/OnlineParticleTool"};
 
    std::map<xAOD::Type::ObjectType, SG::ReadHandleKey<xAOD::IParticleContainer>> m_offlineInputs;
 
    float m_drThreshold;
 
    bool m_rerun;
 
    std::string m_outputPrefix;
 
    bool m_checkEmptyChainGroups;
 
    bool m_inputDependentConfig;
 
    ToolHandle<Trig::TrigDecisionTool> m_tdt{"Trig::TrigDecisionTool/TrigDecisionTool"};
 
    ToolHandle<Trig::IMatchScoringTool> m_scoreTool{
        this, "ScoringTool", "Trig::DRScoringTool", "The pair scoring tool"};
 
    // Internal functions
    StatusCode createOutputContainer(
        xAOD::TrigCompositeContainer *&container,
        const std::string &chain) const;
 
    const particleVec_t &getCandidateMatchesFor(
        const xAOD::IParticle *part,
        std::map<xAOD::Type::ObjectType, particleVec_t> &offlineParticles,
        std::map<const xAOD::IParticle *, particleVec_t> &cache) const;
 
    bool matchParticles(
        const xAOD::IParticle *lhs,
        const xAOD::IParticle *rhs) const;
  }; //> end class TriggerMatchingTool
} // namespace DerivationFramework
 
#endif //> ! DERIVATIONFRAMEWORKTRIGGER_TRIGGERMATCHINGTOOL_H