TauAntiTauJetOverlapTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ASSOCIATIONUTILS_TAU_ANTITAU_JET_OVERLAPTOOL_H
#define ASSOCIATIONUTILS_TAU_ANTITAU_JET_OVERLAPTOOL_H
 
// Framework includes
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
 
// EDM includes
#include "AsgDataHandles/ReadHandleKey.h"
#include <xAODEventInfo/EventInfo.h>
#include "xAODTau/TauJetContainer.h"
#include "xAODJet/JetContainer.h"
 
// Columnar includes
#include "ColumnarCore/ObjectColumn.h"
#include "ColumnarEventInfo/EventInfoDef.h"
#include "ColumnarTau/TauJetDef.h"
#include "ColumnarJet/JetDef.h"
 
// Local includes
#include "AssociationUtils/IOverlapTool.h"
#include "AssociationUtils/BaseOverlapTool.h"
#include "AssociationUtils/IObjectAssociator.h"
 
namespace ORUtils
{

  class TauAntiTauJetOverlapTool : public virtual IOverlapTool,
                                   public BaseOverlapTool
  {

      ASG_TOOL_CLASS(TauAntiTauJetOverlapTool, IOverlapTool)
 
    public:

      TauAntiTauJetOverlapTool(const std::string& name);

      virtual StatusCode
      findOverlaps(columnar::Particle1Range cont1,
                   columnar::Particle2Range cont2,
                   columnar::EventContextId eventContext) const override;

      virtual StatusCode
      internalFindOverlaps(columnar::Particle1Range jets,
                           columnar::Particle2Range taus,
                           columnar::EventContextId eventContext) const;
 
    protected:
 
    protected:

      virtual StatusCode initializeDerived() override;

      bool isBJet(columnar::Particle1Id jet) const;

      bool isSurvivingTau(columnar::Particle2Id tau) const;

      bool isSurvivingAntiTau(columnar::Particle2Id tau) const;
 
    private:

      std::string m_bJetLabel;

      std::string m_tauLabel;

      std::string m_antiTauLabel;

      float m_dR;
      bool m_useRapidity;

 
    private:

      struct Accessors final : columnar::ColumnarTool<>
      {
        columnar::EventInfoAccessor<std::uint64_t> m_eventNumberAcc {*this, "eventNumber"};
        columnar::Particle2Accessor<int> m_categoryAcc;

        columnar::EventInfoAccessor<columnar::ObjectColumn> m_evtAcc;

        columnar::Particle1Accessor<char> m_bJetAcc;
 
        using ColumnarTool::ColumnarTool;
      };
      std::unique_ptr<Accessors> m_accessors {std::make_unique<Accessors> (this)};

      std::unique_ptr<IParticleAssociator> m_dRMatcher;

      std::unique_ptr<OverlapDecorationHelper<columnar::ContainerId::particle2>> m_tauDecHelper;

      std::unique_ptr<OverlapDecorationHelper<columnar::ContainerId::particle2>> m_antiTauDecHelper;
 
      std::string m_antiTauEventCategoryDecorName;
 
      Gaudi::Property<std::string> m_evtKeyName{this, "EventInfoKey", "EventInfo", "xAOD::EventInfo ReadHandleKey"};

 
  }; // class TauAntiTauJetOverlapTool
 
} // namespace ORUtils
 
#endif