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