TBExtrapolTrackToCaloTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
// ***************************************************************************
// Liquid Argon H8 Combined TestBeam Geometry 
// -----------------------------------------
//
// 25.10.2004 Creation of the class by claire.bourdarios@cern.ch
//
// ***************************************************************************
 
#ifndef TBREC_TBEXTRAPOLTRACKTOCALO_H
#define TBREC_TBEXTRAPOLTRACKTOCALO_H
 
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/NeutralParticle.h"
 
#include "GeoPrimitives/GeoPrimitives.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "CaloIdentifier/CaloCell_ID.h"
#include "CaloGeoHelpers/CaloPhiRange.h"
 
#include "CaloDetDescr/CaloDetDescrManager.h"
#include "StoreGate/ReadCondHandleKey.h"
 
class ICaloCoordinateTool;
class IMessageSvc;
class CaloDepthTool;
class ICaloSurfaceBuilder;
class ImpactInCalo;
 
#include "TrkParameters/TrackParameters.h"
#include "TrkDetDescrUtils/Intersection.h"
 
namespace Trk {
  class Track;
  class IExtrapolator;
  class TrackParticleBase;
}
 
static const InterfaceID IID_TBExtrapolTrackToCaloTool("TBExtrapolTrackToCaloTool", 1 , 0); 
 
//<<<<<< INCLUDES >>>>>>

 
//class TBExtrapolTrackToCaloTool : public AthAlgTool, 
//       virtual public IExtrapolateToCaloTool
class TBExtrapolTrackToCaloTool : public AthAlgTool
{
public:
  
  // Retrieve interface ID
  static const InterfaceID& interfaceID() { return IID_TBExtrapolTrackToCaloTool; } 
  // Constructor:
 
  TBExtrapolTrackToCaloTool(const std::string& type, 
          const std::string& name, 
          const IInterface* parent); 
 
  virtual ~TBExtrapolTrackToCaloTool();
 
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
 

  void CaloVariables (const Trk::TrackParameters* parm, 
               double& etaCaloLocal, double& phiCaloLocal);
  void CaloLocalPoint (const Trk::TrackParameters* parm,
               Amg::Vector3D* pt_ctb,
                       Amg::Vector3D* pt_local);

  bool TrackSeenByCalo(const Trk::Track* trk,
                       const CaloCell_ID::CaloSample sample,
                       const double offset,
                       Amg::Vector3D* pt_ctb,
                       Amg::Vector3D* pt_local);
 
  bool TrackSeenByCalo(const Trk::TrackParameters* parm,
                       const CaloCell_ID::CaloSample sample,
                       const double offset,
                       Amg::Vector3D* pt_ctb,
                       Amg::Vector3D* pt_local);
 
  bool TrackSeenByCalo (const Trk::Track* trk, 
            const bool barrel,
            const CaloCell_ID::SUBCALO subcalo, 
            const int sampling_or_module, 
            const double offset, 
            Amg::Vector3D* pt_ctb,
                        Amg::Vector3D* pt_local);
 
  bool TrackSeenByCalo (const Trk::TrackParameters* parm, 
            const bool barrel,
            const CaloCell_ID::SUBCALO subcalo, 
            const int sampling_or_module, 
            const double offset, 
            Amg::Vector3D* pt_ctb,
                        Amg::Vector3D* pt_local);
 
  bool TrackSeenByCalo (const Trk::Track* trk, 
            const CaloCell_ID::CaloSample sample,
            const double offset, 
            Amg::Vector3D* pt_ctb,
                        Amg::Vector3D* pt_local,
            double& trketa_atcalo, double& trkphi_atcalo);
 
  /* new methods in base class, not implemented yet */
  const Trk::TrackParameters* extrapolate (const Trk::Track& /*trk*/,
    const CaloCell_ID::CaloSample /*sample*/,
    const double /*offset*/,
    const Trk::PropDirection /*dir*/=Trk::alongMomentum,
    const Trk::ParticleHypothesis /*particle*/=Trk::undefined) const{std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}  
 
  const Trk::TrackParameters* extrapolate (const Trk::TrackParticleBase& /*trk*/,
    const CaloCell_ID::CaloSample /*sample*/,
    const double /*offset*/,
    const Trk::PropDirection /*dir*/=Trk::alongMomentum,
    const Trk::ParticleHypothesis /*particle*/=Trk::undefined) const {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  const Trk::TrackParameters* extrapolate (const Trk::TrackParameters& /*trkPar*/,
    const CaloCell_ID::CaloSample /*sample*/,
    const double /*offset*/,
    Trk::ParticleHypothesis /*particle*/,
    const Trk::PropDirection /*dir*/=Trk::alongMomentum) const  {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  const Trk::NeutralParameters* extrapolate (const Trk::NeutralParameters& /*trkPar*/,
    const CaloCell_ID::CaloSample /*sample*/,
    const double /*offset*/,
    Trk::ParticleHypothesis /*particle*/,
    const Trk::PropDirection /*dir*/=Trk::alongMomentum) const  {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
    virtual const Trk::TrackParameters* extrapolate (const  xAOD::TrackParticle& /*trk*/,
            const CaloCell_ID::CaloSample /*sample*/,
            const double /*offset*/,
            const Trk::PropDirection /*dir*/=Trk::alongMomentum,
            const Trk::ParticleHypothesis /*particle*/=Trk::undefined) const
      {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}

    virtual const Trk::NeutralParameters* extrapolate (const xAOD::NeutralParticle& /*trk*/,
            const CaloCell_ID::CaloSample /*sample*/,
            const double /*offset*/,
            const Trk::PropDirection /*dir*/=Trk::alongMomentum ) const
    {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  virtual const DataVector< const Trk::TrackParameters >* getParametersInCalo (const Trk::TrackParameters& /*parameters*/,
    Trk::ParticleHypothesis /*particle*/,
    const Trk::PropDirection /*dir*/=Trk::alongMomentum) const  {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  virtual const DataVector< const Trk::NeutralParameters >* getParametersInCalo (const Trk::NeutralParameters& /*parameters*/,
    Trk::ParticleHypothesis /*particle*/,
    const Trk::PropDirection /*dir*/=Trk::alongMomentum) const  {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  virtual const DataVector< const Trk::TrackParameters >* getParametersInCalo (const Trk::TrackParticleBase& /*trk*/,
    const Trk::ParticleHypothesis /*particle*/=Trk::undefined,
    const Trk::PropDirection /*dir*/=Trk::alongMomentum) const  {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  virtual const DataVector< const Trk::TrackParameters >* getParametersInCalo (const Trk::Track& /*trk*/,
    const Trk::ParticleHypothesis /*particle*/=Trk::undefined,
    const Trk::PropDirection /*dir*/=Trk::alongMomentum) const  {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  virtual const DataVector< const Trk::TrackParameters >* getParametersInCalo (const xAOD::TrackParticle& /*trk*/,
               const Trk::ParticleHypothesis /*particle*/=Trk::undefined,
                                const Trk::PropDirection /*dir*/=Trk::alongMomentum) const {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
 
  ImpactInCalo* getImpactInCalo ( const Trk::Track& /*trk*/,
                                  const Trk::ParticleHypothesis /*particle*/=Trk::undefined,
                                  const Trk::PropDirection /*dir*/=Trk::alongMomentum) const  {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  ImpactInCalo* getImpactInCalo ( const Trk::TrackParticleBase& /*trk*/,
                                  const Trk::ParticleHypothesis /*particle*/=Trk::undefined,
                                  const Trk::PropDirection /*dir*/=Trk::alongMomentum) const  {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  virtual ImpactInCalo* getImpactInCalo (const Trk::TrackParameters& /*parameters*/,
            Trk::ParticleHypothesis /*particleHypo*/,
            const Trk::PropDirection /*dir*/=Trk::alongMomentum) const {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}
 
  virtual ImpactInCalo* getImpactInCalo (const Trk::NeutralParameters& /*parameters*/,
            Trk::ParticleHypothesis /*particleHypo*/,
            const Trk::PropDirection /*dir*/=Trk::alongMomentum) const {std::cout<<"Not Implemented !!!!"<<std::endl; return 0;}

  CaloDepthTool* getCaloDepth();
 

  const Trk::TrackParameters* extrapolate(const Trk::TrackParameters* parm,
                      const Trk::Surface* surf);

  const Trk::TrackParameters* extrapolate(const Trk::Track* trk,
                      const Trk::Surface* surf);
 
 
  virtual Amg::Vector3D getMomentumAtVertex(const xAOD::Vertex& vertex, bool /*reuse*/) const;
 
#if 0
  virtual Trk::Intersection getIntersectionInCalo(const Amg::Vector3D& position, const Amg::Vector3D& momentum, const CaloCell_ID::CaloSample sample) const;
#endif
 
private:
 
 
 
  // CaloDetDescr usual stuff
  SG::ReadCondHandleKey<CaloDetDescrManager> m_caloMgrKey { this
      , "CaloDetDescrManager"
      , "CaloDetDescrManager"
      , "SG Key for CaloDetDescrManager in the Condition Store" };
 
  const CaloCell_ID* m_calo_id{nullptr};
  ICaloCoordinateTool* m_calo_tb_coord{nullptr};
 
  CaloPhiRange m_range;
 
  // Defines the surface  for extrapolation : 
  ToolHandle<ICaloSurfaceBuilder> m_calosurf;
 
 
  // Pre-configured extrapolator : will be choosen via jobOpt
  ToolHandle< Trk::IExtrapolator > m_extrapolator;
};
 
 
#endif // TRACKTOCALO_EXTRAPOLTRACKTOCALO_H