TrackContainerHandlesHelper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSTRKEVENT_TRACKCONTAINERHANDLESHELPER_H
#define ACTSTRKEVENT_TRACKCONTAINERHANDLESHELPER_H
 
#include <memory>
 
#include "Acts/Geometry/TrackingGeometry.hpp"
#include "ActsEvent/MultiTrajectory.h"
#include "ActsEvent/TrackContainer.h"
#include "ActsEvent/TrackSummaryContainer.h"
#include "ActsGeometryInterfaces/ActsGeometryContext.h"
#include "GaudiKernel/StatusCode.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/ReadHandleKey.h"
#include "xAODTracking/TrackStateContainer.h"
#include "xAODTracking/TrackJacobianContainer.h"
#include "xAODTracking/TrackMeasurementContainer.h"
#include "xAODTracking/TrackParametersContainer.h"
#include "xAODTracking/TrackSurfaceContainer.h"
#include "xAODTracking/TrackSummaryContainer.h"
 
namespace ActsTrk {
 
std::string prefixFromTrackContainerName(const std::string& tracks);
 
class MutableTrackContainerHandlesHelper {
 public:
  StatusCode initialize(const std::string& prefix);
 
  std::unique_ptr<ActsTrk::TrackContainer> moveToConst(
      ActsTrk::MutableTrackContainer&& tc,
      const Acts::GeometryContext& geoContext,
      const EventContext& evtContext) const;
 
 private:
  // MTJ part
  SG::WriteHandleKey<xAOD::TrackStateContainer> m_statesKey;
  SG::WriteHandleKey<xAOD::TrackParametersContainer> m_parametersKey;
  SG::WriteHandleKey<xAOD::TrackJacobianContainer> m_jacobiansKey;
  SG::WriteHandleKey<xAOD::TrackMeasurementContainer> m_measurementsKey;
  SG::WriteHandleKey<xAOD::TrackSurfaceContainer> m_surfacesKey;
  SG::WriteHandleKey<ActsTrk::MultiTrajectory> m_mtjKey;
 
  std::unique_ptr<ActsTrk::MultiTrajectory> moveToConst(
      ActsTrk::MutableMultiTrajectory&& mmtj,
      const EventContext& context) const;
 
  // TrackContainer part
  SG::WriteHandleKey<xAOD::TrackSummaryContainer> m_xAODTrackSummaryKey;
  SG::WriteHandleKey<xAOD::TrackSurfaceContainer> m_trackSurfacesKey;
  SG::WriteHandleKey<ActsTrk::TrackSummaryContainer> m_trackSummaryKey;
};
 
class ConstTrackContainerHandlesHelper {
 public:
  StatusCode initialize(const std::string& prefix);
 
  std::unique_ptr<ActsTrk::TrackContainer> build(
      const Acts::TrackingGeometry* geo,
      const Acts::GeometryContext& geoContext,
      const EventContext& context) const;
 
 private:
  // restore the pointer to uncalibrated measurements from element links
  void restoreUncalibMeasurementPtr(xAOD::TrackStateAuxContainer &statesLink) const;
  // MTJ part
  SG::ReadHandleKey<xAOD::TrackStateContainer> m_statesKey;
  SG::ReadHandleKey<xAOD::TrackParametersContainer> m_parametersKey;
  SG::ReadHandleKey<xAOD::TrackJacobianContainer> m_jacobiansKey;
  SG::ReadHandleKey<xAOD::TrackMeasurementContainer> m_measurementsKey;
  SG::ReadHandleKey<xAOD::TrackSurfaceContainer> m_surfacesKey;
 
  SG::WriteHandleKey<ActsTrk::MultiTrajectory> m_mtjKey;
  // build for MTJ part
  std::unique_ptr<ActsTrk::MultiTrajectory> buildMtj(
      const Acts::TrackingGeometry* geo,
      const Acts::GeometryContext& geoContext,
      const EventContext& context) const;
  // TrackContainer part
 
  // track and its backend
  SG::ReadHandleKey<xAOD::TrackSummaryContainer> m_xAODTrackSummaryKey;
  SG::ReadHandleKey<xAOD::TrackSurfaceContainer> m_trackSurfacesKey;
  SG::WriteHandleKey<ActsTrk::TrackSummaryContainer> m_trackSummaryKey;
};
 
}  // namespace ActsTrk
 
#endif