TrackParticleVertexAssociationTool.h

Go to the documentation of this file.

// This file's extension implies that it's C, but it's really -*- C++ -*-.
 
/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id$
#ifndef TRACKD3PDMAKER_TRACKPARTICLEVERTEXASSOCIATIONTOOL_H
#define TRACKD3PDMAKER_TRACKPARTICLEVERTEXASSOCIATIONTOOL_H
 
#include "D3PDMakerUtils/SingleAssociationTool.h"
#include "xAODTracking/TrackParticleContainerFwd.h"
#include "xAODTracking/VertexFwd.h"
#include "xAODTracking/VertexContainer.h"
 
namespace Trk {
  class VxCandidate;
}
namespace Rec {
  class TrackParticle;
}
 
namespace D3PD {
 
class TrackParticleVertexAssociationTool
  : public SingleAssociationTool<Types<Rec::TrackParticle, xAOD::TrackParticle>,
                                 Types<Trk::VxCandidate, xAOD::Vertex> >
{
public:
  typedef SingleAssociationTool<Types<Rec::TrackParticle, xAOD::TrackParticle>,
                                 Types<Trk::VxCandidate, xAOD::Vertex> > Base;
 
  TrackParticleVertexAssociationTool (const std::string& type,
                                      const std::string& name,
                                      const IInterface* parent);
 
  virtual StatusCode initialize() override;
  virtual StatusCode book() final;
 
 
  virtual const Trk::VxCandidate* get (const Rec::TrackParticle& p) override;
 
 
  virtual const xAOD::Vertex* get (const xAOD::TrackParticle& p) override;
 
 
private:
 
  std::string m_targetSGKey;
 
  bool m_fast;
 
  float *m_weight;
 
  SG::ReadHandleKey<xAOD::VertexContainer> m_vertexKey { this, "VertexContainer", "PrimaryVertices", "primary vertex container" };
 
}; // class TrackParticleVertexAssociationTool
 
} // namespace D3PD
 
#endif // not TRACKD3PDMAKER_TRACKPARTICLEVERTEXASSOCIATIONTOOL_H