JetPtAssociationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// JetPtAssociationTool.h
 
#ifndef JetMomentTools_JetPtAssociationTool_H
#define JetMomentTools_JetPtAssociationTool_H

 
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
#include "JetInterface/IJetDecorator.h"
#include "AsgDataHandles/ReadHandleKey.h"
#include "AsgDataHandles/WriteDecorHandleKey.h"
#include "xAODJet/JetContainer.h"
 
class JetPtAssociationTool : public asg::AsgTool,
                             virtual public IJetDecorator {
  ASG_TOOL_CLASS(JetPtAssociationTool, IJetDecorator)
    
public:
 
  typedef std::vector<const xAOD::IParticle*> APVector;
  typedef std::vector<unsigned> IndexVector;
  typedef std::vector<float> FloatVector;
 
public:

  JetPtAssociationTool(const std::string& myname);

  virtual StatusCode initialize() override;

  virtual StatusCode decorate(const xAOD::JetContainer& jets) const override;
 
private:  // data

  int ptfrac(const APVector& aps, const xAOD::JetContainer& jets, FloatVector& ptsums) const;

  int match(const APVector& aps, const xAOD::Jet& jet, APVector& apvs, double& ptsum_constituents) const;

  Gaudi::Property<std::string> m_aname{this, "AssociationName", "", "Key for association vector"};
  Gaudi::Property<std::string> m_jetContainerName{this, "JetContainer", "", "Input jet container"};
  SG::ReadHandleKey<xAOD::JetContainer> m_matchingContainerName{this, "MatchingJetContainer", "", "Jet container to match to"};
 
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_assocFracKey{this, "AssociationFractionName", "AssociationFraction", "SG key for output AssociationFraction decoration"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_assocLinkKey{this, "AssociationLinkName", "AssociationLink", "SG key for output AssociationLink decoration"};
};
 
#endif