VertexTrackParticleAssociationTool.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
*/
 
#ifndef TRACKD3PDMAKER_VERTEXTRACKPARTICLEASSOCIATIONTOOL_H
#define TRACKD3PDMAKER_VERTEXTRACKPARTICLEASSOCIATIONTOOL_H
 
#include "D3PDMakerUtils/MultiAssociationTool.h"
#include "Particle/TrackParticle.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/Vertex.h"
 
#include <vector>
 
namespace Trk{
class VxCandidate;
class VxTrackAtVertex;
}
 
namespace D3PD {
 
 
class VertexTrackParticleAssociationTool
  : public MultiAssociationTool<Types<Trk::VxCandidate, xAOD::Vertex>,
                                Types<Rec::TrackParticle, xAOD::TrackParticle> >
{
public:
  typedef MultiAssociationTool<Types<Trk::VxCandidate, xAOD::Vertex>,
                               Types<Rec::TrackParticle, xAOD::TrackParticle> > Base;
  using Base::next;
 
  VertexTrackParticleAssociationTool (const std::string& type,
                                      const std::string& name,
                                      const IInterface* parent);
 
 
  virtual StatusCode book() final;
 
 
  virtual StatusCode reset (const Trk::VxCandidate& p) override;
 
 
  virtual StatusCode reset (const xAOD::Vertex& p) override;
 
 
  virtual const Rec::TrackParticle* next (const Trk::VxCandidate* dummy) override;
 
 
  virtual const xAOD::TrackParticle* next (const xAOD::Vertex* dummy) override;
 
 
private:
  float* m_weight;
 
  const Trk::VxCandidate* m_aod_vx;
  const xAOD::Vertex* m_xaod_vx;
  size_t m_itrack;
};
 
 
} // namespace D3PD
 
 
#endif // not TRACKD3PDMAKER_VERTEXTRACKPARTICLEASSOCIATIONTOOL_H