JetTrackSumMomentsTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// JetTrackSumMomentsTool.h 
 
#ifndef JETMOMENTTOOLS_JETTRACKSUMMOMENTSTOOL_H
#define JETMOMENTTOOLS_JETTRACKSUMMOMENTSTOOL_H
 
 
 
 
#include "AsgTools/ToolHandle.h"
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
#include "InDetTrackSelectionTool/IInDetTrackSelectionTool.h"
#include "JetInterface/IJetDecorator.h"
#include "JetEDM/TrackVertexAssociation.h"
#include "AsgDataHandles/ReadDecorHandleKey.h"
#include "AsgDataHandles/WriteDecorHandleKey.h"
 
#include "xAODJet/JetContainer.h"
#include "xAODTracking/TrackParticle.h" 
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/Vertex.h" 
#include "xAODTracking/VertexContainer.h" 
 
#include <vector>
#include <string>
 
 
class JetTrackSumMomentsTool : public asg::AsgTool,
                               virtual public IJetDecorator {
  ASG_TOOL_CLASS(JetTrackSumMomentsTool,IJetDecorator)
    
public:
 
  // Constructor from tool name
  JetTrackSumMomentsTool(const std::string& name);
 
  // Initialization.
  virtual StatusCode initialize() override;
 
  // Inherited methods to modify a jet
  // Calls moment and puts the results in the jet
  virtual StatusCode decorate(const xAOD::JetContainer& jets) const override;
 
  // Local method to return the vector track sums
  std::pair<float,float>
  getJetTrackSums(const xAOD::Vertex*, 
                        const std::vector<const xAOD::TrackParticle*>&, 
                        const jet::TrackVertexAssociation*) const;
 
  // Local method to return the HS vertex - that of type PriVtx
  const xAOD::Vertex* findHSVertex(const xAOD::VertexContainer*&) const;
 
private:
 
  Gaudi::Property<std::string> m_assocTracksName{this, "AssociatedTracks", "", "SG key for associated tracks container"};
  Gaudi::Property<bool> m_requireTrackPV{this, "RequireTrackPV", true, "Require track to be from the primary vertex?"};
  Gaudi::Property<std::string> m_jetContainerName{this, "JetContainer", "", "SG key for the input jet container"};
  ToolHandle<InDet::IInDetTrackSelectionTool> m_htsel{this, "TrackSelector", "", "Track selector"};
  Gaudi::Property<bool> m_useOriginVertex = {this, "UseOriginVertex", false, "use origin vertex for each jet"};
 
  SG::ReadHandleKey< xAOD::VertexContainer> m_vertexContainer_key{this, "VertexContainer", "", "Vertex container key"};
  SG::ReadHandleKey<jet::TrackVertexAssociation> m_tva_key{this, "TrackVertexAssociation", "", "Track vertex association key"};
  
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_trackSumPtKey{this, "SumPtName", "TrackSumPt", "SG key for output track SumPt decoration"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_trackSumMassKey{this, "SumMassName", "TrackSumMass", "SG key for output track SumMass decoration"};
};
 
#endif