IEMExtrapolationTools.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMAINTERFACES_IEMEXTRAPOLATIONTOOLS_H
#define EGAMMAINTERFACES_IEMEXTRAPOLATIONTOOLS_H
 
#include "GaudiKernel/AlgTool.h"
#include "GaudiKernel/EventContext.h"
#include "TrkCaloExtension/CaloExtension.h"
#include "TrkEventPrimitives/PropDirection.h"
#include "TrkNeutralParameters/NeutralParameters.h"
#include "TrkParameters/TrackParameters.h"
#include "xAODCaloEvent/CaloCluster.h"
#include "xAODTracking/TrackParticleFwd.h"
#include "xAODTracking/VertexFwd.h"
 
#include <array>
#include <memory>
#include <unordered_map>
 
class CaloDetDescrManager;
static const InterfaceID IID_IEMExtrapolationTools("IEMExtrapolationTools",
                                                   1,
                                                   0);
 
class IEMExtrapolationTools : virtual public IAlgTool
{
 
public:
  enum TrkExtrapDef
  {
    fromLastMeasurement,
    fromPerigee,
    fromPerigeeRescaled,
    fromCaloToPerigee
  };
 
  virtual ~IEMExtrapolationTools(){};

  static const InterfaceID& interfaceID();

  virtual StatusCode initialize() = 0;
 
  virtual std::pair<std::vector<CaloSampling::CaloSample>,
                    std::vector<std::unique_ptr<Trk::Surface>>>
  getClusterLayerSurfaces(const xAOD::CaloCluster& cluster,
                          const CaloDetDescrManager& caloDD) const = 0;

  virtual StatusCode getMatchAtCalo(
    const EventContext& ctx,
    const xAOD::CaloCluster& cluster,
    const xAOD::TrackParticle& trkPB,
    const std::vector<CaloSampling::CaloSample>& samples,
    const std::vector<std::unique_ptr<Trk::Surface>>& surfaces,
    std::array<double, 4>& eta,
    std::array<double, 4>& phi,
    std::array<double, 4>& deltaEta,
    std::array<double, 4>& deltaPhi,
    unsigned int extrapFrom = fromPerigee) const = 0;

  virtual bool matchesAtCalo(const xAOD::CaloCluster* cluster,
                             const xAOD::Vertex* vertex,
                             float etaAtCalo,
                             float phiAtCalo) const = 0;

  virtual bool getEtaPhiAtCalo(const EventContext& ctx,
                               const xAOD::Vertex* vertex,
                               float* etaAtCalo,
                               float* phiAtCalo) const = 0;

  virtual bool getEtaPhiAtCalo(const EventContext& ctx,
                               const Trk::TrackParameters* trkPar,
                               float* etaAtCalo,
                               float* phiAtCalo) const = 0;

  virtual Amg::Vector3D getMomentumAtVertex(const EventContext& ctx,
                                            const xAOD::Vertex&,
                                            unsigned int) const = 0;

  virtual Amg::Vector3D getMomentumAtVertex(const EventContext& ctx,
                                            const xAOD::Vertex&,
                                            bool reuse = true) const = 0;
};
 
inline const InterfaceID&
IEMExtrapolationTools::interfaceID()
{
  return IID_IEMExtrapolationTools;
}
 
#endif