AsgPtEtaSelectionTool.h

Go to the documentation of this file.

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

 
 
 
#ifndef ASG_ANALYSIS_ALGORITHMS__ASG_PT_ETA_SELECTION_TOOL_H
#define ASG_ANALYSIS_ALGORITHMS__ASG_PT_ETA_SELECTION_TOOL_H
 
#include <AsgTools/AsgTool.h>
#include <AthContainers/AuxElement.h>
#include <PATCore/IAsgSelectionTool.h>
#include "AsgTools/PropertyWrapper.h"
#include <atomic>
 
namespace CP
{
 
  class AsgPtEtaSelectionTool final
    : public asg::AsgTool, virtual public IAsgSelectionTool
  {
    //
    // public interface
    //
 
    // Create a proper constructor for Athena
    ASG_TOOL_CLASS( AsgPtEtaSelectionTool, IAsgSelectionTool )
 
 
    
  public:
    using asg::AsgTool::AsgTool;
 
 
    //
    // inherited interface
    //
 
    virtual StatusCode initialize () override;
 
    virtual const asg::AcceptInfo& getAcceptInfo( ) const override;
 
    virtual asg::AcceptData accept( const xAOD::IParticle *particle ) const override;
 
 
 
    //
    // private interface
    //

 
  private:
    Gaudi::Property<float> m_minPt {this, "minPt", 0, "minimum pt to require (or 0 for no pt cut)"};
    Gaudi::Property<float> m_maxPt {this, "maxPt", 0, "maximum pt to require (or 0 for no pt cut)"};
    Gaudi::Property<float> m_minEta {this, "minEta", 0, "minimum abs(eta) to allow (or 0 for no eta cut)"};
    Gaudi::Property<float> m_maxEta {this, "maxEta", 0, "maximum abs(eta) to allow (or 0 for no eta cut)"};
    Gaudi::Property<float> m_etaGapLow {this, "etaGapLow", 0, "low end of the eta gap"};
    Gaudi::Property<float> m_etaGapHigh {this, "etaGapHigh", 0, "high end of the eta gap (or 0 for no eta gap)"};
    Gaudi::Property<float> m_minRapidity {this, "minRapidity", 0, "minimum abs(rapidity) to allow (or 0 for no eta cut)"};
    Gaudi::Property<float> m_maxRapidity {this, "maxRapidity", 0, "maximum abs(rapidity) to allow (or 0 for no eta cut)"};
    Gaudi::Property<bool> m_useClusterEta {this, "useClusterEta", false, "whether to use the cluster eta (for electrons only)"};
    Gaudi::Property<bool> m_useDressedProperties {this, "useDressedProperties", false, "whether to use the dressed kinematic properties (for truth particles only)"};
    Gaudi::Property<bool> m_useConstituentMomentum {this, "useConstituentMomentum", false, "whether to use the constituent momentum (for jets only) (not for pt)"};
    Gaudi::Property<bool> m_printCastWarning {this, "printCastWarning", true, "whether to print a warning/error when the cast fails"};
    Gaudi::Property<bool> m_printClusterWarning {this, "printClusterWarning", true, "whether to print a warning/error when the cluster is missing"};


    int m_minPtCutIndex{ -1 };
    int m_maxPtCutIndex{ -1 };
    int m_minEtaCutIndex{ -1 };
    int m_maxEtaCutIndex{ -1 };
    int m_etaGapCutIndex{ -1 };
    int m_minRapidityCutIndex{ -1 };
    int m_maxRapidityCutIndex{ -1 };
    int m_egammaCastCutIndex{ -1 };
    int m_egammaClusterCutIndex{ -1 };
    int m_dressedPropertiesIndex{ -1 };
    int m_jetCastCutIndex{ -1 };

  private:
    mutable std::atomic<bool> m_shouldPrintCastWarning {true};

  private:
    mutable std::atomic<bool> m_shouldPrintClusterWarning {true};

  private:
    asg::AcceptInfo m_accept;

  private:
    std::unique_ptr<SG::ConstAccessor<float>> m_dressedPtAccessor{};
    std::unique_ptr<SG::ConstAccessor<float>> m_dressedEtaAccessor{};
  };
}
 
#endif