DeltaRMatcher.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ASSOCIATIONUTILS_DELTARMATCHER_H
#define ASSOCIATIONUTILS_DELTARMATCHER_H
 
// System includes
#include <limits>
 
// EDM includes
#include "xAODBase/IParticle.h"
 
// Columnar includes
#include "ColumnarCore/ParticleDef.h"
#include "ColumnarCore/MomentumAccessors.h"
 
// Local includes
#include "AssociationUtils/IObjectAssociator.h"
 
namespace ADD_CMODE(ORUtils)
{

  class DeltaRMatcher final : public IParticleAssociator
  {
 
    public:

      DeltaRMatcher(double dR, bool useRapidity=true);

      virtual StatusCode setObjectTypes (xAODType::ObjectType type1,
                                   xAODType::ObjectType type2) override;

      virtual bool objectsMatch
      (columnar::Particle1Id p1, columnar::Particle2Id p2, bool swapArgs = false) const override;
      using IParticleAssociator::objectsMatch;
 
    private:
 
      double m_dR;
      bool m_useRapidity;

      columnar::MomentumAccessors<columnar::Particle1Def> m_momAcc1;
      columnar::MomentumAccessors<columnar::Particle2Def> m_momAcc2;
 
  }; // class DeltaRMatcher
 

  class SlidingDeltaRMatcher final : public IParticleAssociator
  {
 
    public:

      SlidingDeltaRMatcher(double c1, double c2,
                           double maxCone = std::numeric_limits<double>::max(),
                           bool useRapidity=true);

      virtual StatusCode setObjectTypes (xAODType::ObjectType type1,
                                   xAODType::ObjectType type2) override;

      virtual bool objectsMatch
      (columnar::Particle1Id p1, columnar::Particle2Id p2, bool swapArgs = false) const override;
 
    private:

      double m_c1;
      double m_c2;
      double m_maxCone;

      bool m_useRapidity;

      columnar::MomentumAccessors<columnar::Particle1Def> m_momAcc1;
      columnar::MomentumAccessors<columnar::Particle2Def> m_momAcc2;
 
  }; // class SlidingDeltaRMatcher
 
} // namespace ADD_CMODE(ORUtils)
 
#endif