GNNVertexWeightCalculator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKVERTEXWEIGHTCALCULATORS_GNNVERTEXWEIGHTCALCULATOR_H
#define TRKVERTEXWEIGHTCALCULATORS_GNNVERTEXWEIGHTCALCULATOR_H
 
#include "StoreGate/ReadDecorHandleKey.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include <TrkVertexFitterInterfaces/IVertexWeightCalculator.h>
#include <xAODTracking/VertexContainer.h>
 
#include <memory>
#include <string>

class GNNVertexWeightCalculator final
    : public AthAlgTool,
      virtual public Trk::IVertexWeightCalculator {
public:
  GNNVertexWeightCalculator(const std::string& t,
                            const std::string& n,
                            const IInterface* p);
 
  virtual ~GNNVertexWeightCalculator() override;
  virtual StatusCode initialize() override;
  virtual double
  estimateSignalCompatibility(const xAOD::Vertex &vertex) const override;
 
private:
  SG::ReadDecorHandleKey<xAOD::VertexContainer> m_gnnScoreKey{
      this, "GNNScoreKey", "PrimaryVertices_initial.HSGN2_phsvertex",
      "Name of the hard-scatter GNN decoration"};
};
 
#endif // TRKVERTEXWEIGHTCALCULATORS_GNNVERTEXWEIGHTCALCULATOR_H