#include <GNNVertexFitterTool.h>

Inheritance diagram for Rec::GNNVertexFitterTool:

Collaboration diagram for Rec::GNNVertexFitterTool:

Classes
struct	WrkVrt

Public Member Functions
	GNNVertexFitterTool (const std::string &type, const std::string &name, const IInterface *parent)

virtual	~GNNVertexFitterTool ()

StatusCode	initialize ()

StatusCode	finalize ()

virtual StatusCode	fitAllVertices (const xAOD::JetContainer , xAOD::VertexContainer , const xAOD::Vertex &primaryVertex, const EventContext &) const

ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	evtStore () const
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard `StoreGateSvc/DetectorStore` Returns (kind of) a pointer to the `StoreGateSvc`. More...

virtual StatusCode	sysInitialize () override
	Perform system initialization for an algorithm. More...

virtual StatusCode	sysStart () override
	Handle START transition. More...

virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles. More...

virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles. More...

Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T > &t)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKey &hndl, const std::string &doc, const SG::VarHandleKeyType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleBase &hndl, const std::string &doc, const SG::VarHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKeyArray &hndArr, const std::string &doc, const SG::VarHandleKeyArrayType &)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc, const SG::NotHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc="none")
	Declare a new Gaudi property. More...

void	updateVHKA (Gaudi::Details::PropertyBase &)

MsgStream &	msg () const

MsgStream &	msg (const MSG::Level lvl) const

bool	msgLvl (const MSG::Level lvl) const

Static Public Member Functions
static const InterfaceID &	interfaceID ()

Public Attributes
ToolHandle< Trk::TrkVKalVrtFitter >	m_vertexFitterTool

SG::ReadDecorHandleKey< xAOD::JetContainer >	m_trackLinksKey

SG::ReadDecorHandleKey< xAOD::JetContainer >	m_trackOriginsKey

SG::ReadDecorHandleKey< xAOD::JetContainer >	m_vertexLinksKey

SG::ReadHandleKey< xAOD::EventInfo >	m_eventInfoKey {this, "eventInfoKey", "EventInfo", "EventInfo container to use"}

SG::WriteDecorHandleKey< xAOD::JetContainer >	m_jetWriteDecorKeyVertexLink

SG::WriteDecorHandleKey< xAOD::JetContainer >	m_jetWriteDecorKeyVertexNumber

SG::ReadCondHandleKey< InDet::BeamSpotData >	m_beamSpotKey {this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot"}

const xAOD::Vertex *	m_thePV {}

Protected Member Functions
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution More...

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void >	renounce (T &h)

void	extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
	Add StoreName to extra input/output deps as needed. More...

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
TLorentzVector	TotalMom (const std::vector< const xAOD::TrackParticle * > &selTrk) const

double	vrtVrtDist (const xAOD::Vertex &primVrt, const Amg::Vector3D &secVrt, const std::vector< double > &vrtErr, double &signif) const

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyArrayType &)
	specialization for handling Gaudi::Property<SG::VarHandleKeyArray> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleType &)
	specialization for handling Gaudi::Property<SG::VarHandleBase> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &t, const SG::NotHandleType &)
	specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray> More...

Private Attributes
SG::AuxElement::Decorator< float >	m_deco_mass

SG::AuxElement::Decorator< float >	m_deco_pt

SG::AuxElement::Decorator< float >	m_deco_charge

SG::AuxElement::Decorator< float >	m_deco_vPos

SG::AuxElement::Decorator< float >	m_deco_lxy

SG::AuxElement::Decorator< float >	m_deco_sig3D

SG::AuxElement::Decorator< float >	m_deco_deltaR

SG::AuxElement::Decorator< float >	m_deco_ntrk

SG::AuxElement::Decorator< float >	m_deco_lxyz

SG::AuxElement::Decorator< float >	m_deco_eFrac

SG::AuxElement::Decorator< float >	m_deco_nHFTracks

StringProperty	m_gnnModel {this, "GNNModel", "GN2v01", "GNN model being used" }

StringProperty	m_jetCollection {this, "JetCollection", "AntiKt4EMPFlowJets", "Jet Collection being used" }

BooleanProperty	m_includePrimaryVertex {this, "includePrimaryVertex", false, "Include Primary Vertices"}

BooleanProperty	m_removeNonHFVertices {this, "removeNonHFVertices", true, "Remove vertices with no heavy flavour tracks"}

BooleanProperty	m_doInclusiveVertexing {this, "doInclusiveVertexing", false, "Merge all vertices so that there is at most one vertex per jet"}

DoubleProperty	m_maxChi2 {this, "maxChi2", 20, "Maximum Chi Squared"}

BooleanProperty	m_applyCuts {this, "applyCuts", false, "Cut on vertex properties"}

DoubleProperty	m_minLxy {this, "minLxy", 0.0, "Minimum radial distance from the PV"}

DoubleProperty	m_maxLxy {this, "maxLxy", 1e5, "Maximum radial distance from the PV"}

DoubleProperty	m_minSig3D {this, "minSig3D", 0, "Minimum 3D significance from the PV"}

DoubleProperty	m_HFRatioThres {this, "HFRatio", 0.0, "The threshold for the Ratio between HF tracks and all track for a vertex"}

DoubleProperty	m_minNTrack {this, "minNTrk", 2, "Minimum number of tracks in a vertex"}

double	m_massPi {}

StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default) More...

StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default) More...

std::vector< SG::VarHandleKeyArray * >	m_vhka

bool	m_varHandleArraysDeclared

Detailed Description

Definition at line 49 of file GNNVertexFitterTool.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< AlgTool > >::StoreGateSvc_t

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ GNNVertexFitterTool()

Rec::GNNVertexFitterTool::GNNVertexFitterTool	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent
	)

Definition at line 18 of file GNNVertexFitterTool.cxx.

     : AthAlgTool(type, name, parent), m_vertexFitterTool("Trk::TrkVKalVrtFitter/VertexFitterTool", this),
       m_deco_mass("mass"),
       m_deco_pt("pt"),
       m_deco_charge("charge"),
       m_deco_vPos("vPos"),
       m_deco_lxy("Lxy"),
       m_deco_sig3D("significance3D"),
       m_deco_deltaR("deltaR"),
       m_deco_ntrk("ntrk"),
       m_deco_lxyz("Lxyz"),
       m_deco_eFrac("efracsv"),
       m_deco_nHFTracks("nHFtrk"){
   declareInterface<IGNNVertexFitterInterface>(this);
   declareProperty("JetTrackLinks", m_trackLinksKey = "AntiKt4EMPFlow."+m_gnnModel+"_TrackLinks");
   declareProperty("JetTrackOrigins", m_trackOriginsKey = "AntiKt4EMPFlow."+m_gnnModel+"_TrackOrigin");
   declareProperty("JetVertexLinks", m_vertexLinksKey = "AntiKt4EMPFlow."+m_gnnModel+"_VertexIndex");
   declareProperty("VertexFitterTool", m_vertexFitterTool, "Vertex fitting tool");
   m_massPi = 139.5702 * Gaudi::Units::MeV;
 }

◆ ~GNNVertexFitterTool()

Rec::GNNVertexFitterTool::~GNNVertexFitterTool ( )

virtual

Definition at line 41 of file GNNVertexFitterTool.cxx.

41 { ATH_MSG_DEBUG("GNNVertexFitterTool destructor called"); }

Member Function Documentation

◆ declareGaudiProperty() [1/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleKeyArrayType &
	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKeyArray>

Definition at line 170 of file AthCommonDataStore.h.

   {
     return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                        hndl.value(), 
                                                        hndl.documentation());
  
   }

◆ declareGaudiProperty() [2/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleKeyType &
	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

   {
     return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                        hndl.value(), 
                                                        hndl.documentation());
  
   }

◆ declareGaudiProperty() [3/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleType &
	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleBase>

Definition at line 184 of file AthCommonDataStore.h.

   {
     return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                        hndl.value(), 
                                                        hndl.documentation());
   }

◆ declareGaudiProperty() [4/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	t,
		const SG::NotHandleType &
	)

inlineprivateinherited

specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray>

Definition at line 199 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(t);
   }

◆ declareProperty() [1/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleBase &	hndl,
		const std::string &	doc,
		const SG::VarHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleBase. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 245 of file AthCommonDataStore.h.

   {
     this->declare(hndl.vhKey());
     hndl.vhKey().setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [2/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKey &	hndl,
		const std::string &	doc,
		const SG::VarHandleKeyType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleKey. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 221 of file AthCommonDataStore.h.

   {
     this->declare(hndl);
     hndl.setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [3/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKeyArray &	hndArr,
		const std::string &	doc,
		const SG::VarHandleKeyArrayType &
	)

inlineinherited

Definition at line 259 of file AthCommonDataStore.h.

   {
  
     // std::ostringstream ost;
     // ost << Algorithm::name() << " VHKA declareProp: " << name 
     //     << " size: " << hndArr.keys().size() 
     //     << " mode: " << hndArr.mode() 
     //     << "  vhka size: " << m_vhka.size()
     //     << "\n";
     // debug() << ost.str() << endmsg;
  
     hndArr.setOwner(this);
     m_vhka.push_back(&hndArr);
  
     Gaudi::Details::PropertyBase* p =  PBASE::declareProperty(name, hndArr, doc);
     if (p != 0) {
       p->declareUpdateHandler(&AthCommonDataStore<PBASE>::updateVHKA, this);
     } else {
       ATH_MSG_ERROR("unable to call declareProperty on VarHandleKeyArray " 
                     << name);
     }
  
     return p;
  
   }

◆ declareProperty() [4/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc,
		const SG::NotHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This is the generic version, for types that do not derive from SG::VarHandleKey. It just forwards to the base class version of declareProperty.

Definition at line 333 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(name, property, doc);
   }

◆ declareProperty() [5/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc = `"none"`
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This dispatches to either the generic declareProperty or the one for VarHandle/Key/KeyArray.

Definition at line 352 of file AthCommonDataStore.h.

   {
     typedef typename SG::HandleClassifier<T>::type htype;
     return declareProperty (name, property, doc, htype());
   }

◆ declareProperty() [6/6]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty ( Gaudi::Property< T > & t )

inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

                                                                  {
     typedef typename SG::HandleClassifier<T>::type htype;
     return AthCommonDataStore<PBASE>::declareGaudiProperty(t, htype());
   }

◆ detStore()

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< AlgTool > >::detStore ( ) const

inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

95 { return m_detStore; }

◆ evtStore() [1/2]

ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( )

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

85 { return m_evtStore; }

◆ evtStore() [2/2]

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( ) const

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 90 of file AthCommonDataStore.h.

90 { return m_evtStore; }

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps )

protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

◆ finalize()

StatusCode Rec::GNNVertexFitterTool::finalize ( )

Definition at line 358 of file GNNVertexFitterTool.cxx.

358 { return StatusCode::SUCCESS; }

◆ fitAllVertices()

StatusCode Rec::GNNVertexFitterTool::fitAllVertices	(	const xAOD::JetContainer *	inJetContainer,
		xAOD::VertexContainer *	outVertexContainer,
		const xAOD::Vertex &	primaryVertex,
		const EventContext &	ctx
	)		const

virtual

Implements Rec::IGNNVertexFitterInterface.

Definition at line 106 of file GNNVertexFitterTool.cxx.

                                                                                                                 {
  
   using TLC = std::vector<ElementLink<DataVector<xAOD::TrackParticle_v1>>>;
   using TL = ElementLink<DataVector<xAOD::TrackParticle_v1>>;
  
   // Read Decor Handle for Track links and Vertex links
   SG::ReadDecorHandle<xAOD::JetContainer, TLC> trackLinksHandle(m_trackLinksKey, ctx);
   SG::ReadDecorHandle<xAOD::JetContainer, std::vector<char>> trackOriginsHandle(m_trackOriginsKey, ctx);
   SG::ReadDecorHandle<xAOD::JetContainer, std::vector<char>> vertexLinksHandle(m_vertexLinksKey, ctx);
   SG::WriteDecorHandle<xAOD::JetContainer, std::vector<ElementLink<xAOD::VertexContainer>>>
       jetWriteDecorHandleVertexLink(m_jetWriteDecorKeyVertexLink, ctx);
  
   // Loop over the jets
   for (const auto &jet : *inJetContainer) {
  
     // Retrieve the Vertex and Track Collections
     auto vertexCollection = vertexLinksHandle(*jet);
     auto trackCollection = trackLinksHandle(*jet);
     auto trackOriginCollection = trackOriginsHandle(*jet);
  
     using indexList = std::vector<int>;
     using trackCountMap = std::map<char, std::set<TL>>;
  
     indexList iList(vertexCollection.size());
  
     trackCountMap allTracksMap;          // All the vertices and the corresponding track links
     trackCountMap heavyFlavourVertexMap; // All Heavy Flavour Tracks associated with a vertex
     trackCountMap primaryVertexMap; // All Primary Tracks associated with a vertex
     trackCountMap fittingMap;            // Map filled with vertices to be fitted
  
     fittingMap.clear();
     heavyFlavourVertexMap.clear();
     primaryVertexMap.clear();
     allTracksMap.clear();
  
     // Fill the map of predicted vertex indices -> tracks
     for (int index=0; index < int(vertexCollection.size()); index++){
     
       auto vertex = vertexCollection[index];
       auto trackOrigin = trackOriginCollection[index];
       auto trackLink   = trackCollection[index];
  
       allTracksMap[vertex].insert(trackLink);
  
       // Add heavy flavour tracks associated to each vertex to the map
       if (InDet::ExclusiveOrigin::FromB == trackOrigin || InDet::ExclusiveOrigin::FromBC == trackOrigin ||
           InDet::ExclusiveOrigin::FromC == trackOrigin) {
         heavyFlavourVertexMap[vertex].insert(trackLink);
       }
       // Add primary tracks associated to each vertex to the map
       else if (InDet::ExclusiveOrigin::Primary == trackOrigin) { 
         primaryVertexMap[vertex].insert(trackLink);
       }
     };
  
     // determine the vertex with the largest number of primary tracks
     auto pvCandidate = std::max_element(
         primaryVertexMap.begin(), primaryVertexMap.end(),
         [](const auto& a, const auto& b) { return a.second.size() < b.second.size(); }
     )->first;
  
     // filter the vertices according to the configurable options
     for (const auto &vertexTrackPair : allTracksMap) {
       // unpack vertexTrackPair
       const auto &[vertex, trackLinks] = vertexTrackPair;
       // remove the primary vertex unless requested
       if(!m_includePrimaryVertex) {
         if(vertex == pvCandidate) {
           continue;
         }
       }
       // remove vertices which do not contain any heavy flavour tracks if requested
       if(m_removeNonHFVertices) {
         if (heavyFlavourVertexMap.find(vertex) == heavyFlavourVertexMap.end()) {
           continue;
         }
       }
       // remove vertices which contain insufficient heavy flavour tracks if requested
       if(m_HFRatioThres > 0) {
         if (heavyFlavourVertexMap.find(vertex) != heavyFlavourVertexMap.end() &&
             (static_cast<float>(heavyFlavourVertexMap[vertex].size()) / trackLinks.size()) < m_HFRatioThres) {
           continue;
         }
       }
       // now we've passed all vertex filtering, add the vertex to the fittingMap
       fittingMap.insert(std::pair<char, std::set<TL>>(vertex, trackLinks));
     }
     // If inclusive vertexing is requested, merge the vertices
     if (m_doInclusiveVertexing) {
       // Define the key for the merged vertex
       const char mergedVertexKey = 0;
       std::set<TL> mergedSet;
       std::set<TL> combinedHFTracks;
  
       // Iterate over the fittingMap to merge all sets into a single set
       for (const auto &vertexTrackPair : fittingMap) {
         // Insert all elements from trackLinks into mergedSet
         mergedSet.insert(vertexTrackPair.second.begin(), vertexTrackPair.second.end());
         combinedHFTracks.insert(heavyFlavourVertexMap[vertexTrackPair.first].begin(), heavyFlavourVertexMap[vertexTrackPair.first].end());
       }
  
       // Clear fittingMap and insert the merged set under the mergedVertexKey
       fittingMap.clear();
       fittingMap[mergedVertexKey] = std::move(mergedSet);
       heavyFlavourVertexMap.clear();
       heavyFlavourVertexMap[mergedVertexKey] = std::move(combinedHFTracks);
     }
  
     // Working xAOD   
     std::unique_ptr<workVectorArrxAOD> xAODwrk (new workVectorArrxAOD());
     SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle{m_beamSpotKey, ctx};
  
     // Beam Conditions
     xAODwrk->beamX = beamSpotHandle->beamPos().x();
     xAODwrk->beamY = beamSpotHandle->beamPos().y();
     xAODwrk->beamZ = beamSpotHandle->beamPos().z();
     xAODwrk->tanBeamTiltX = tan(beamSpotHandle->beamTilt(0));
     xAODwrk->tanBeamTiltY = tan(beamSpotHandle->beamTilt(1));
  
     std::unique_ptr<std::vector<WrkVrt>> wrkVrtSet = std::make_unique<std::vector<WrkVrt>>();
     WrkVrt newvrt;
     newvrt.Good = true;
     std::unique_ptr<Trk::IVKalState> state = m_vertexFitterTool->makeState();
     std::vector<const xAOD::NeutralParticle *> neutralPartDummy(0);
     Amg::Vector3D IniVrt(0., 0., 0.);
  
     for (const auto &pair : fittingMap) {
       // Need at least 2 tracks to perform a fit
       if (pair.second.size() >= 2) {
         int NTRKS = pair.second.size();
         std::vector<double> InpMass(NTRKS, m_massPi);
         m_vertexFitterTool->setMassInputParticles(InpMass, *state);
  
         xAODwrk->listSelTracks.clear();
  
         for (auto TrackLink : pair.second) {
           xAODwrk->listSelTracks.push_back(*TrackLink);
         }
  
         Amg::Vector3D FitVertex, vDist;
         TLorentzVector jetDir(jet->p4()); // Jet Direction
  
         // Get Estimate
         StatusCode sc = m_vertexFitterTool->VKalVrtFitFast(xAODwrk->listSelTracks, FitVertex, *state);
  
         if (sc.isFailure() || FitVertex.perp() > m_maxLxy) { /* No initial estimation */
           IniVrt = primVrt.position();
           if (m_includePrimaryVertex)
             IniVrt.setZero();
         } else {
           vDist = FitVertex - primVrt.position();
           double JetVrtDir = jetDir.Px() * vDist.x() + jetDir.Py() * vDist.y() + jetDir.Pz() * vDist.z();
           if (m_includePrimaryVertex)
             JetVrtDir = fabs(JetVrtDir); /* Always positive when primary vertex is seeked for*/
           if (JetVrtDir > 0.)
             IniVrt = FitVertex; /* Good initial estimation */
           else
             IniVrt = primVrt.position();
         }
  
         m_vertexFitterTool->setApproximateVertex(IniVrt.x(), IniVrt.y(), IniVrt.z(), *state);
  
         // Perform the Vertex Fit
         sc = (m_vertexFitterTool->VKalVrtFit(xAODwrk->listSelTracks, neutralPartDummy, newvrt.vertex, newvrt.vertexMom,
                                              newvrt.vertexCharge, newvrt.vertexCov, newvrt.chi2PerTrk, newvrt.trkAtVrt,
                                              newvrt.chi2, *state, false));
         if (sc.isFailure())
           continue;
  
         // Chi2 Cut
         auto NDOF = 2 * (newvrt.trkAtVrt.size()) - 3.0; // From VrtSecInclusive
  
         if (newvrt.chi2 / NDOF >= m_maxChi2)
           continue;
         ATH_MSG_DEBUG("Found IniVertex=" << newvrt.vertex[0] << ", " << newvrt.vertex[1] << ", " << newvrt.vertex[2]
                                          << " trks " << newvrt.trkAtVrt.size());
  
         Amg::Vector3D vDir = newvrt.vertex - primVrt.position(); // Vertex Dirction in relation to Primary
  
         Amg::Vector3D jetVrtDir(jet->p4().Px(), jet->p4().Py(), jet->p4().Pz());
         
         double vPos =
             (vDir.x() * newvrt.vertexMom.Px() + vDir.y() * newvrt.vertexMom.Py() + vDir.z() * newvrt.vertexMom.Pz()) /
             newvrt.vertexMom.Rho();
  
         double Lxy = sqrt(vDir[0] * vDir[0] + vDir[1] * vDir[1]);
         double Lxyz = sqrt(vDir[0] * vDir[0] + vDir[1] * vDir[1] + vDir[2] * vDir[2]);
         ATH_MSG_DEBUG("Lxyz  " << Lxyz);
         
         double drJPVSV = Amg::deltaR(jetVrtDir, vDir); // DeltaR
         
         int ntrk = newvrt.trkAtVrt.size(); // # Tracks in Vertex
  
         TLorentzVector MomentumVtx = TotalMom(xAODwrk->listSelTracks);
  
         double eRatio = MomentumVtx.E() / jet->p4().E();
         double signif3D;
         [[maybe_unused]] double distToPV = vrtVrtDist(primVrt, newvrt.vertex, newvrt.vertexCov, signif3D);
         
         // apply quality cuts
         if (m_applyCuts) {
           // cut on minimum number of tracks in the vertex
           if (ntrk < m_minNTrack)
             continue;
           // cut on 3D significance
           if (signif3D < m_minSig3D)
             continue;
           // cut on minumum transverse displacement from the PV
           if (Lxy < m_minLxy ) 
             continue;
         }
         
         // Register Container
         auto* GNNvertex = outVertexContainer->emplace_back(new xAOD::Vertex);
  
         for (const auto *trk : xAODwrk->listSelTracks) {
           ElementLink<xAOD::TrackParticleContainer> link_trk(
               *(dynamic_cast<const xAOD::TrackParticleContainer *>(trk->container())),
               trk->index());
           GNNvertex->addTrackAtVertex(link_trk, 1.);
         }
  
         //Add Vertex Info into Container 
         GNNvertex->setVertexType(xAOD::VxType::SecVtx);
         GNNvertex->setPosition(newvrt.vertex);
         GNNvertex->setFitQuality(newvrt.chi2, NDOF);
  
         m_deco_mass(*GNNvertex)            = newvrt.vertexMom.M();
         m_deco_pt(*GNNvertex)              = newvrt.vertexMom.Perp();
         m_deco_charge(*GNNvertex)          = newvrt.vertexCharge;
         m_deco_vPos(*GNNvertex)            = vPos;
         m_deco_lxy(*GNNvertex)             = Lxy;
         m_deco_lxyz(*GNNvertex)            = Lxyz;
         m_deco_sig3D(*GNNvertex)           = signif3D; 
         m_deco_ntrk(*GNNvertex)            = ntrk;
         m_deco_deltaR(*GNNvertex)          = drJPVSV;
         m_deco_eFrac(*GNNvertex)           = eRatio;
         m_deco_nHFTracks(*GNNvertex)       = heavyFlavourVertexMap[pair.first].size();
         
         ElementLink<xAOD::VertexContainer> linkVertex;
         linkVertex.setElement(GNNvertex);
         linkVertex.setStorableObject(*outVertexContainer);
         jetWriteDecorHandleVertexLink(*jet).push_back(linkVertex);
  
       } // end of 2 Track requirement
     }
   } // end loop over jets
   return StatusCode::SUCCESS;
 } // end performVertexFit

◆ initialize()

StatusCode Rec::GNNVertexFitterTool::initialize ( )

Definition at line 43 of file GNNVertexFitterTool.cxx.

                                            {
  
   ATH_MSG_DEBUG("GNNVertexFitter Tool in initialize()");
  
   // Initialize keys
   ATH_CHECK(m_trackLinksKey.initialize());
   ATH_CHECK(m_trackOriginsKey.initialize());
   ATH_CHECK(m_vertexLinksKey.initialize());
  
   std::string linkNameMod = "";
   if (m_doInclusiveVertexing) linkNameMod = "Inclusive";
   m_jetWriteDecorKeyVertexLink = m_jetCollection + "."+linkNameMod+"GNNVerticesLink";
   ATH_CHECK(m_jetWriteDecorKeyVertexLink.initialize());
  
   // Retrieve tools
   ATH_CHECK(m_vertexFitterTool.retrieve());
   // Additional Info for Vertex Fit
   ATH_CHECK(m_beamSpotKey.initialize());
   ATH_CHECK(m_eventInfoKey.initialize());
  
   return StatusCode::SUCCESS;
 }

◆ inputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< AlgTool > >::inputHandles ( ) const

overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ interfaceID()

static const InterfaceID& Rec::IGNNVertexFitterInterface::interfaceID ( )

inlinestaticinherited

Definition at line 19 of file IGNNVertexFitterInterface.h.

19 { return IID_IGNNVertexFitterInterface; }

◆ msg() [1/2]

MsgStream& AthCommonMsg< AlgTool >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                 {
     return this->msgStream();
   }

◆ msg() [2/2]

MsgStream& AthCommonMsg< AlgTool >::msg ( const MSG::Level lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                   {
     return this->msgStream(lvl);
   }

◆ msgLvl()

bool AthCommonMsg< AlgTool >::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                                {
     return this->msgLevel(lvl);
   }

◆ outputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< AlgTool > >::outputHandles ( ) const

overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ renounce()

std::enable_if_t<std::is_void_v<std::result_of_t<decltype(&T::renounce)(T)> > && !std::is_base_of_v<SG::VarHandleKeyArray, T> && std::is_base_of_v<Gaudi::DataHandle, T>, void> AthCommonDataStore< AthCommonMsg< AlgTool > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

   {
     h.renounce();
     PBASE::renounce (h);
   }

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::renounceArray ( SG::VarHandleKeyArray & handlesArray )

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                           {
     handlesArray.renounce();
   }

◆ sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )

overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in DerivationFramework::CfAthAlgTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and asg::AsgMetadataTool.

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysStart ( )

overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

◆ TotalMom()

TLorentzVector Rec::GNNVertexFitterTool::TotalMom ( const std::vector< const xAOD::TrackParticle * > & selTrk ) const

private

Definition at line 67 of file GNNVertexFitterTool.cxx.

                                                                                                      {
   TLorentzVector sum(0., 0., 0., 0.);
   for (int i = 0; i < (int)selTrk.size(); ++i) {
     if (selTrk[i] == NULL)
       continue;
     sum += selTrk[i]->p4();
   }
   return sum;
 }

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::updateVHKA ( Gaudi::Details::PropertyBase & )

inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

                                                {
     // debug() << "updateVHKA for property " << p.name() << " " << p.toString() 
     //         << "  size: " << m_vhka.size() << endmsg;
     for (auto &a : m_vhka) {
       std::vector<SG::VarHandleKey*> keys = a->keys();
       for (auto k : keys) {
         k->setOwner(this);
       }
     }
   }

◆ vrtVrtDist()

double Rec::GNNVertexFitterTool::vrtVrtDist	(	const xAOD::Vertex &	primVrt,
		const Amg::Vector3D &	secVrt,
		const std::vector< double > &	vrtErr,
		double &	signif
	)		const

private

Definition at line 78 of file GNNVertexFitterTool.cxx.

                                                                                                       {
   double distx = primVrt.x() - secVrt.x();
   double disty = primVrt.y() - secVrt.y();
   double distz = primVrt.z() - secVrt.z();
  
   AmgSymMatrix(3) primCovMtx = primVrt.covariancePosition(); // Create
   primCovMtx(0, 0) += secVrtErr[0];
   primCovMtx(0, 1) += secVrtErr[1];
   primCovMtx(1, 0) += secVrtErr[1];
   primCovMtx(1, 1) += secVrtErr[2];
   primCovMtx(0, 2) += secVrtErr[3];
   primCovMtx(2, 0) += secVrtErr[3];
   primCovMtx(1, 2) += secVrtErr[4];
   primCovMtx(2, 1) += secVrtErr[4];
   primCovMtx(2, 2) += secVrtErr[5];
  
   AmgSymMatrix(3) wgtMtx = primCovMtx.inverse();
  
   signif = distx * wgtMtx(0, 0) * distx + disty * wgtMtx(1, 1) * disty + distz * wgtMtx(2, 2) * distz +
            2. * distx * wgtMtx(0, 1) * disty + 2. * distx * wgtMtx(0, 2) * distz + 2. * disty * wgtMtx(1, 2) * distz;
   signif = std::sqrt(std::abs(signif));
   if (signif != signif)
     signif = 0.;
   return std::sqrt(distx * distx + disty * disty + distz * distz);
 }

Member Data Documentation

◆ m_applyCuts

BooleanProperty Rec::GNNVertexFitterTool::m_applyCuts {this, "applyCuts", false, "Cut on vertex properties"}

private

Definition at line 122 of file GNNVertexFitterTool.h.

◆ m_beamSpotKey

SG::ReadCondHandleKey<InDet::BeamSpotData> Rec::GNNVertexFitterTool::m_beamSpotKey {this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot"}

Definition at line 80 of file GNNVertexFitterTool.h.

◆ m_deco_charge

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_charge

private

Definition at line 106 of file GNNVertexFitterTool.h.

◆ m_deco_deltaR

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_deltaR

private

Definition at line 110 of file GNNVertexFitterTool.h.

◆ m_deco_eFrac

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_eFrac

private

Definition at line 113 of file GNNVertexFitterTool.h.

◆ m_deco_lxy

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_lxy

private

Definition at line 108 of file GNNVertexFitterTool.h.

◆ m_deco_lxyz

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_lxyz

private

Definition at line 112 of file GNNVertexFitterTool.h.

◆ m_deco_mass

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_mass

private

Definition at line 104 of file GNNVertexFitterTool.h.

◆ m_deco_nHFTracks

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_nHFTracks

private

Definition at line 114 of file GNNVertexFitterTool.h.

◆ m_deco_ntrk

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_ntrk

private

Definition at line 111 of file GNNVertexFitterTool.h.

◆ m_deco_pt

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_pt

private

Definition at line 105 of file GNNVertexFitterTool.h.

◆ m_deco_sig3D

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_sig3D

private

Definition at line 109 of file GNNVertexFitterTool.h.

◆ m_deco_vPos

SG::AuxElement::Decorator<float> Rec::GNNVertexFitterTool::m_deco_vPos

private

Definition at line 107 of file GNNVertexFitterTool.h.

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_doInclusiveVertexing

BooleanProperty Rec::GNNVertexFitterTool::m_doInclusiveVertexing {this, "doInclusiveVertexing", false, "Merge all vertices so that there is at most one vertex per jet"}

private

Definition at line 120 of file GNNVertexFitterTool.h.

◆ m_eventInfoKey

SG::ReadHandleKey<xAOD::EventInfo> Rec::GNNVertexFitterTool::m_eventInfoKey {this, "eventInfoKey", "EventInfo", "EventInfo container to use"}

Definition at line 71 of file GNNVertexFitterTool.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_gnnModel

StringProperty Rec::GNNVertexFitterTool::m_gnnModel {this, "GNNModel", "GN2v01", "GNN model being used" }

private

Definition at line 116 of file GNNVertexFitterTool.h.

◆ m_HFRatioThres

DoubleProperty Rec::GNNVertexFitterTool::m_HFRatioThres {this, "HFRatio", 0.0, "The threshold for the Ratio between HF tracks and all track for a vertex"}

private

Definition at line 126 of file GNNVertexFitterTool.h.

◆ m_includePrimaryVertex

BooleanProperty Rec::GNNVertexFitterTool::m_includePrimaryVertex {this, "includePrimaryVertex", false, "Include Primary Vertices"}

private

Definition at line 118 of file GNNVertexFitterTool.h.

◆ m_jetCollection

StringProperty Rec::GNNVertexFitterTool::m_jetCollection {this, "JetCollection", "AntiKt4EMPFlowJets", "Jet Collection being used" }

private

Definition at line 117 of file GNNVertexFitterTool.h.

◆ m_jetWriteDecorKeyVertexLink

SG::WriteDecorHandleKey<xAOD::JetContainer> Rec::GNNVertexFitterTool::m_jetWriteDecorKeyVertexLink

Initial value:

{

this, "jetDecorKeyJetLink", "", "WriteDecorHandleKey for adding VertexLink to Jets"}

Definition at line 74 of file GNNVertexFitterTool.h.

◆ m_jetWriteDecorKeyVertexNumber

SG::WriteDecorHandleKey<xAOD::JetContainer> Rec::GNNVertexFitterTool::m_jetWriteDecorKeyVertexNumber

Initial value:

{

this, "jetDecorKeyVertexNumber", "", "WriteDecorHandleKey for adding number of vertices within a Jet"}

Definition at line 76 of file GNNVertexFitterTool.h.

◆ m_massPi

double Rec::GNNVertexFitterTool::m_massPi {}

private

Definition at line 128 of file GNNVertexFitterTool.h.

◆ m_maxChi2

DoubleProperty Rec::GNNVertexFitterTool::m_maxChi2 {this, "maxChi2", 20, "Maximum Chi Squared"}

private

Definition at line 121 of file GNNVertexFitterTool.h.

◆ m_maxLxy

DoubleProperty Rec::GNNVertexFitterTool::m_maxLxy {this, "maxLxy", 1e5, "Maximum radial distance from the PV"}

private

Definition at line 124 of file GNNVertexFitterTool.h.

◆ m_minLxy

DoubleProperty Rec::GNNVertexFitterTool::m_minLxy {this, "minLxy", 0.0, "Minimum radial distance from the PV"}

private

Definition at line 123 of file GNNVertexFitterTool.h.

◆ m_minNTrack

DoubleProperty Rec::GNNVertexFitterTool::m_minNTrack {this, "minNTrk", 2, "Minimum number of tracks in a vertex"}

private

Definition at line 127 of file GNNVertexFitterTool.h.

◆ m_minSig3D

DoubleProperty Rec::GNNVertexFitterTool::m_minSig3D {this, "minSig3D", 0, "Minimum 3D significance from the PV"}

private

Definition at line 125 of file GNNVertexFitterTool.h.

◆ m_removeNonHFVertices

BooleanProperty Rec::GNNVertexFitterTool::m_removeNonHFVertices {this, "removeNonHFVertices", true, "Remove vertices with no heavy flavour tracks"}

private

Definition at line 119 of file GNNVertexFitterTool.h.

◆ m_thePV

const xAOD::Vertex* Rec::GNNVertexFitterTool::m_thePV {}

Definition at line 83 of file GNNVertexFitterTool.h.

◆ m_trackLinksKey

SG::ReadDecorHandleKey<xAOD::JetContainer> Rec::GNNVertexFitterTool::m_trackLinksKey

Initial value:

{this, "trackLinksKey", "",

"Jet GNN Deco Read Key for track link"}

Definition at line 65 of file GNNVertexFitterTool.h.

◆ m_trackOriginsKey

SG::ReadDecorHandleKey<xAOD::JetContainer> Rec::GNNVertexFitterTool::m_trackOriginsKey

Initial value:

{this, "trackOriginsKey", "",

"Jet GNN Deco Read Key for track origin"}

Definition at line 67 of file GNNVertexFitterTool.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vertexFitterTool

ToolHandle<Trk::TrkVKalVrtFitter> Rec::GNNVertexFitterTool::m_vertexFitterTool

Definition at line 62 of file GNNVertexFitterTool.h.

◆ m_vertexLinksKey

SG::ReadDecorHandleKey<xAOD::JetContainer> Rec::GNNVertexFitterTool::m_vertexLinksKey

Initial value:

{this, "vertexLinksKey", "",

"Jet GNN Deco Read Key for vertex link"}

Definition at line 69 of file GNNVertexFitterTool.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< AlgTool > >::m_vhka

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

The documentation for this class was generated from the following files:

Classes

Public Member Functions

Static Public Member Functions

Public Attributes

Protected Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ StoreGateSvc_t

Constructor & Destructor Documentation

◆ GNNVertexFitterTool()

◆ ~GNNVertexFitterTool()

Member Function Documentation

◆ declareGaudiProperty() [1/4]

◆ declareGaudiProperty() [2/4]

◆ declareGaudiProperty() [3/4]

◆ declareGaudiProperty() [4/4]

◆ declareProperty() [1/6]

◆ declareProperty() [2/6]

◆ declareProperty() [3/6]

◆ declareProperty() [4/6]

◆ declareProperty() [5/6]

◆ declareProperty() [6/6]

◆ detStore()

◆ evtStore() [1/2]

◆ evtStore() [2/2]

◆ extraDeps_update_handler()

◆ finalize()

◆ fitAllVertices()

◆ initialize()

◆ inputHandles()

◆ interfaceID()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ outputHandles()

◆ renounce()

◆ renounceArray()

◆ sysInitialize()

◆ sysStart()

◆ TotalMom()

◆ updateVHKA()

◆ vrtVrtDist()

Member Data Documentation

◆ m_applyCuts

◆ m_beamSpotKey

◆ m_deco_charge

◆ m_deco_deltaR

◆ m_deco_eFrac

◆ m_deco_lxy

◆ m_deco_lxyz

◆ m_deco_mass

◆ m_deco_nHFTracks

◆ m_deco_ntrk

◆ m_deco_pt

◆ m_deco_sig3D

◆ m_deco_vPos

◆ m_detStore

◆ m_doInclusiveVertexing

◆ m_eventInfoKey

◆ m_evtStore

◆ m_gnnModel

◆ m_HFRatioThres

◆ m_includePrimaryVertex

◆ m_jetCollection

◆ m_jetWriteDecorKeyVertexLink

◆ m_jetWriteDecorKeyVertexNumber

◆ m_massPi

◆ m_maxChi2

◆ m_maxLxy

◆ m_minLxy

◆ m_minNTrack

◆ m_minSig3D

◆ m_removeNonHFVertices

◆ m_thePV

◆ m_trackLinksKey

◆ m_trackOriginsKey

◆ m_varHandleArraysDeclared