IHGTD_TrackTimeExtensionTool.h

Go to the documentation of this file.

 
#ifndef IHGTD_TRACKTIMEEXTENSIONTOOL_H
#define IHGTD_TRACKTIMEEXTENSIONTOOL_H
 
#include "GaudiKernel/IAlgTool.h"
#include "HGTD_PrepRawData/HGTD_ClusterContainer.h"
#include "HGTD_RecToolInterfaces/IHGTD_ClusterTruthTool.h"
#include "TrkTrack/Track.h"
#include "TrkTrack/TrackStateOnSurface.h"
#include "xAODTracking/TrackParticle.h"
#include "AtlasHepMC/GenEvent.h"
#include <memory>
 
namespace HGTD {
 
struct ExtensionObject {
  // track states after extrapolation, can be of hole type
  std::array<std::unique_ptr<const Trk::TrackStateOnSurface>, 4> m_hits;
  // truth info about primary hits left by the truth particles
  // is kept independent of them being found during the track extension
  std::array<const HGTD_Cluster*, 4> m_truth_primary_hits = {nullptr, nullptr, nullptr, nullptr};
  std::array<HGTD::ClusterTruthInfo, 4> m_truth_primary_info;
  // point of extrapolation on HGTD layer closest to IP
  float m_extrap_x = 0;
  float m_extrap_y = 0;
  // holes on track in HGTD
  std::vector<char> m_holes_hgtd = {0, 0, 0, 0};
};
 
} // namespace HGTD
 
class IHGTD_TrackTimeExtensionTool : virtual public IAlgTool {
 
public:
  DeclareInterfaceID(IHGTD_TrackTimeExtensionTool, 1, 0);

  virtual HGTD::ExtensionObject
  extendTrackToHGTD(const EventContext& ctx,
                    const xAOD::TrackParticle& track,
                    const HGTD_ClusterContainer* container,
                    const HepMC::GenEvent* hs_event = nullptr,
                    const InDetSimDataCollection* sim_data = nullptr) const = 0;
};
 
#endif // IHGTD_TRACKTIMEEXTENSIONTOOL_H