EleEleOverlapTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ASSOCIATIONUTILS_ELEELEOVERLAPTOOL_H
#define ASSOCIATIONUTILS_ELEELEOVERLAPTOOL_H
 
// EDM includes
#include "xAODEgamma/ElectronContainer.h"
#include "ColumnarEgamma/EgammaDef.h"
 
// Columnar includes
#include "ColumnarCore/ColumnAccessor.h"
#include "ColumnarCore/LinkColumn.h"
#include "ColumnarCore/VectorColumn.h"
#include "ColumnarCluster/ClusterHelpers.h"
#include "ColumnarTracking/TrackDef.h"
 
// Local includes
#include "AssociationUtils/IOverlapTool.h"
#include "AssociationUtils/BaseOverlapTool.h"
 
namespace ORUtils
{

  class EleEleOverlapTool : public virtual IOverlapTool,
                            public BaseOverlapTool
  {

      ASG_TOOL_CLASS(EleEleOverlapTool, IOverlapTool)
 
    public:

      EleEleOverlapTool(const std::string& name);

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

      virtual StatusCode
      internalFindOverlaps(columnar::Particle1Range electrons) const;
 
    protected:

      virtual StatusCode initializeDerived() override;
 
    private:

      bool electronsMatch(columnar::Particle1Id el1, columnar::Particle1Id el2) const;

      bool rejectFirst(columnar::Particle1Id el1, columnar::Particle1Id el2) const;


      bool m_useTrackMatch;

      bool m_useClusterMatch;

      double m_clusterDeltaEta;

      double m_clusterDeltaPhi;


      using MyTrackDef = columnar::VariantContainerId<columnar::Track0Def,columnar::Track0Def>;
      struct Accessors final : columnar::ColumnarTool<>
      {
        columnar::ClusterAccessor<columnar::ObjectColumn> m_clusterContainerAcc;
        columnar::Track0Accessor<columnar::ObjectColumn> m_track0Acc;
        columnar::Particle1Accessor<float> m_ptAcc {*this, "pt"};
        columnar::Particle1Accessor<std::uint16_t> m_authorAcc {*this, "author"};
        columnar::Particle1Accessor<std::vector<columnar::OptClusterId>> m_caloClusterAcc;
        columnar::Particle1Accessor<std::vector<columnar::OptTrackId>> m_trackAcc;
        std::optional<columnar::ClusterHelpers::EtaBEAccessor<>> m_etaBEAcc;
        std::optional<columnar::ClusterHelpers::PhiBEAccessor<>> m_phiBEAcc;
        using ColumnarTool::ColumnarTool;
      };
      std::unique_ptr<Accessors> m_accessors {std::make_unique<Accessors> (this)};
 
  }; // class EleEleOverlapTool
 
} // namespace ORUtils
 
#endif