SiSpacePointsSeedMaker.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSTRKSEEDINGTOOL_SISPACEPOINTSSEEDMAKER_H
#define ACTSTRKSEEDINGTOOL_SISPACEPOINTSSEEDMAKER_H
 
#include "InDetRecToolInterfaces/ISiSpacePointsSeedMaker.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "GaudiKernel/ToolHandle.h"
#include "ActsToolInterfaces/ISeedingTool.h"
 
#include "BeamSpotConditionsData/BeamSpotData.h"
#include "MagFieldConditions/AtlasFieldCacheCondObj.h"
 
 
#include "TrkSpacePoint/SpacePointContainer.h" 
#include "TrkSpacePoint/SpacePointOverlapCollection.h"
 
#include "xAODInDetMeasurement/PixelClusterContainer.h"
#include "xAODInDetMeasurement/StripClusterContainer.h"
 
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "xAODInDetMeasurement/ContainerAccessor.h"
 
#include "TrkEventUtils/PRDtoTrackMap.h"
#include "TrkSpacePoint/SpacePoint.h" //used in an inline function in the header
#include <span>
//for validation
 
 
class TTree;
class ITHistSvc;
 
namespace InDet{
  class SiSpacePointsSeedMakerEventData;
}
 
namespace ActsTrk {
  class SpacePoint;
 
  static constexpr float s_toTesla = 0.299785832;
 
  class SiSpacePointsSeedMaker :
    public extends<AthAlgTool, InDet::ISiSpacePointsSeedMaker> {
  private:
    enum SeedStrategy : short {PPP, SSS};
 
  public:
    SiSpacePointsSeedMaker(const std::string &t, const std::string &n, const IInterface *p);
    virtual ~SiSpacePointsSeedMaker() override = default;
 
    virtual StatusCode initialize() override;
 
    // Interface
    virtual void 
      newEvent(const EventContext& ctx, 
           InDet::SiSpacePointsSeedMakerEventData& data, 
           int iteration=-1) const override;
 
    virtual void 
      find3Sp(const EventContext& ctx, 
          InDet::SiSpacePointsSeedMakerEventData& data,
          const std::list<Trk::Vertex>& lv) const override;
 
    virtual const InDet::SiSpacePointsSeed* 
      next(const EventContext& ctx, 
       InDet::SiSpacePointsSeedMakerEventData& data) const override;
 
    virtual void 
      writeNtuple(const InDet::SiSpacePointsSeed* seed, 
          const Trk::Track* track, 
          int seedType, 
          long eventNumber) const override;
    
    virtual bool 
      getWriteNtupleBoolProperty() const override;
 
    virtual MsgStream& 
      dump(InDet::SiSpacePointsSeedMakerEventData& data, 
       MsgStream& out) const override;
 
    // ===================================================================== //
    // WE DO NOT NEED THESE FOR NOW
    // ===================================================================== //
 
    virtual void 
      newRegion(const EventContext&,
        InDet::SiSpacePointsSeedMakerEventData&,
        const std::vector<IdentifierHash>&,
        const std::vector<IdentifierHash>&) const override;
    
    virtual void 
      newRegion(const EventContext&, 
        InDet::SiSpacePointsSeedMakerEventData&,
        const std::vector<IdentifierHash>&,
        const std::vector<IdentifierHash>&,
        const IRoiDescriptor&) const override 
    {
      ATH_MSG_ERROR("Method `newRegion` is not implemented.");
    }
 
    virtual void 
      find2Sp(InDet::SiSpacePointsSeedMakerEventData&,
          const std::list<Trk::Vertex>&) const override
    {
      ATH_MSG_ERROR("Method `findfind2Sp` is not implemented.");
    }

    virtual void 
      find3Sp(const EventContext&, 
          InDet::SiSpacePointsSeedMakerEventData&,
          const std::list<Trk::Vertex>&, 
          const double*) const override 
    {
      ATH_MSG_ERROR("Method `find3Sp` is not implemented.");
    }

    virtual void 
      findVSp(const EventContext&, 
          InDet::SiSpacePointsSeedMakerEventData&,
          const std::list<Trk::Vertex>&) const override 
    { 
      ATH_MSG_ERROR("Method `findVSp` is not implemented.");
    }
 
    // ===================================================================== //
    // ===================================================================== //
 
  private:
    MsgStream& 
      dumpConditions(InDet::SiSpacePointsSeedMakerEventData& data, 
             MsgStream& out) const;
 
    static MsgStream& 
      dumpEvent(InDet::SiSpacePointsSeedMakerEventData& data, 
        MsgStream& out);
    
 
  private:
    void buildBeamFrameWork(const EventContext& ctx,
                InDet::SiSpacePointsSeedMakerEventData& data) const;
    void
      newSpacePoint(InDet::SiSpacePointsSeedMakerEventData& data,
            const xAOD::SpacePoint* sp) const;
    
    static void pixInform(const Trk::SpacePoint* sp,
              float* r) ;
    static void stripInform(InDet::SiSpacePointsSeedMakerEventData& data,
                const Trk::SpacePoint* sp,
                std::span<float,15> r) ;
 
    bool convertPixelSeed(const EventContext& ctx,
              InDet::SiSpacePointsSeedMakerEventData& data,
              const ActsTrk::SeedContainer& seedPtrs) const;
 
    bool convertStripSeed(const EventContext& ctx,
              InDet::SiSpacePointsSeedMakerEventData& data,
              const ActsTrk::SeedContainer& seedPtrs) const;
 
    // Retrieve
    StatusCode retrievePixel(const EventContext& ctx,
                 InDet::SiSpacePointsSeedMakerEventData& data,
                 const Trk::PRDtoTrackMap* prd_to_track_map_cptr) const;
    StatusCode retrieveStrip(const EventContext& ctx,
                             InDet::SiSpacePointsSeedMakerEventData& data,
                             const Trk::PRDtoTrackMap* prd_to_track_map_cptr) const;
    StatusCode retrieveOverlap(const EventContext& ctx,
                   InDet::SiSpacePointsSeedMakerEventData& data,
                   const Trk::PRDtoTrackMap* prd_to_track_map_cptr) const;
 
    StatusCode retrievePixel(const EventContext& ctx,
                 InDet::SiSpacePointsSeedMakerEventData& data,
                 const std::vector<IdentifierHash>& ids,
                 const Trk::PRDtoTrackMap* prd_to_track_map_cptr = nullptr) const;
 
    StatusCode retrieveStrip(const EventContext& ctx,
                             InDet::SiSpacePointsSeedMakerEventData& data,
                             const std::vector<IdentifierHash>& ids,
                             const Trk::PRDtoTrackMap* prd_to_track_map_cptr = nullptr) const;
    StatusCode retrieveOverlap(const EventContext& ctx,
                   InDet::SiSpacePointsSeedMakerEventData& data,
                   const std::vector<IdentifierHash>& ids,
                   const Trk::PRDtoTrackMap* prd_to_track_map_cptr) const;
 
    // Validation
    StatusCode InitTree();
 
  private:
    bool isUsed(const Trk::SpacePoint*, const Trk::PRDtoTrackMap &prd_to_track_map) const;
    bool isUsed(const xAOD::SpacePoint*, const Trk::PRDtoTrackMap &prd_to_track_map) const;
 
  private:
    ToolHandle< ActsTrk::ISeedingTool > m_seedsToolPixel {this, "SeedToolPixel", "","Seed Tool for Pixel detector"};
    ToolHandle< ActsTrk::ISeedingTool > m_seedsToolStrip {this, "SeedToolStrip", "","Seed Tool for Strip detector"};
 
    SG::ReadHandleKey< Trk::PRDtoTrackMap > m_prdToTrackMap {this, "PRDtoTrackMap", "", "option PRD-to-track association"};
 
    SG::ReadHandleKey< xAOD::SpacePointContainer > m_actsSpacepointsPixel {this, "ActsSpacePointsPixelName", "", "Pixel space points container"};
    SG::ReadHandleKey< xAOD::SpacePointContainer > m_actsSpacepointsStrip {this, "ActsSpacePointsStripName", "", "Strip space points container"};
    SG::ReadHandleKey< xAOD::SpacePointContainer > m_actsSpacepointsOverlap {this, "ActsSpacePointsOverlapName", "", "Strip overlap space points container"};
 
    SG::ReadCondHandleKey< InDet::BeamSpotData > m_beamSpotKey{this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot"};
    SG::ReadCondHandleKey< AtlasFieldCacheCondObj > m_fieldCondObjInputKey {this, "AtlasFieldCacheCondObj", "fieldCondObj",
        "Name of the Magnetic Field conditions object key"};
 
    SG::ReadCondHandleKey< InDetDD::SiDetectorElementCollection > m_pixelDetEleCollKey {this, "PixelDetectorElements", "ITkPixelDetectorElementCollection",
      "Key of input SiDetectorElementCollection"};
    SG::ReadCondHandleKey< InDetDD::SiDetectorElementCollection > m_stripDetEleCollKey {this, "StripDetectorElements", "ITkStripDetectorElementCollection",
      "Key of input SiDetectorElementCollection"};
    
    Gaudi::Property< bool > m_pixel {this, "usePixel", true};
    Gaudi::Property< bool > m_strip {this, "useStrip", true };
    Gaudi::Property< bool > m_useOverlap {this, "useOverlapSpCollection", true};
    Gaudi::Property< bool > m_fastTracking {this, "useFastTracking", false};
    Gaudi::Property< bool > m_doSeedConversion {this, "doSeedConversion", true, "Convert ActsTrk::Seed into ITk::SiSpacePointsProSeed"};
    Gaudi::Property< bool > m_useClusters {this, "useClustersForSeedConversion", false};
    
  private:
    // Validation
    Gaudi::Property< bool > m_writeNtuple {this, "WriteNtuple", false};
 
    mutable std::mutex m_mutex;
 
    ServiceHandle<ITHistSvc> m_thistSvc;
    TTree* m_outputTree = nullptr;
 
    mutable std::string m_treeName ATLAS_THREAD_SAFE = "";
    mutable std::string m_treeFolder ATLAS_THREAD_SAFE = "/valNtuples/";
 
    mutable float m_d0 ATLAS_THREAD_SAFE = 0;
    mutable float m_z0 ATLAS_THREAD_SAFE = 0;
    mutable float m_pt ATLAS_THREAD_SAFE = 0;
    mutable float m_eta ATLAS_THREAD_SAFE = 0;
    mutable double m_x1 ATLAS_THREAD_SAFE = 0;
    mutable double m_x2 ATLAS_THREAD_SAFE = 0;
    mutable double m_x3 ATLAS_THREAD_SAFE = 0;
    mutable double m_y1 ATLAS_THREAD_SAFE = 0;
    mutable double m_y2 ATLAS_THREAD_SAFE = 0;
    mutable double m_y3 ATLAS_THREAD_SAFE = 0;
    mutable double m_z1 ATLAS_THREAD_SAFE = 0;
    mutable double m_z2 ATLAS_THREAD_SAFE = 0;
    mutable double m_z3 ATLAS_THREAD_SAFE = 0;
    mutable double m_r1 ATLAS_THREAD_SAFE = 0;
    mutable double m_r2 ATLAS_THREAD_SAFE = 0;
    mutable double m_r3 ATLAS_THREAD_SAFE = 0;
    mutable float m_quality ATLAS_THREAD_SAFE = 0;
    mutable int m_type ATLAS_THREAD_SAFE = 0;
    mutable double m_dzdr_t ATLAS_THREAD_SAFE = 0;
    mutable double m_dzdr_b ATLAS_THREAD_SAFE = 0;
    mutable bool m_givesTrack ATLAS_THREAD_SAFE = false;
    mutable float m_trackPt ATLAS_THREAD_SAFE = 0;
    mutable float m_trackEta ATLAS_THREAD_SAFE = 0;
    mutable long m_eventNumber ATLAS_THREAD_SAFE = 0;
  };

  // Inline Method
 
  inline bool
  SiSpacePointsSeedMaker::isUsed(const Trk::SpacePoint* sp,
                 const Trk::PRDtoTrackMap &prd_to_track_map) const
  {
    const Trk::PrepRawData* d = sp->clusterList().first;
    if (!d || !prd_to_track_map.isUsed(*d)) return false;
    
    d = sp->clusterList().second;
    if (!d || prd_to_track_map.isUsed(*d)) return true;
    
    return false;
  }
 
  inline bool
  SiSpacePointsSeedMaker::isUsed(const xAOD::SpacePoint* sp, const Trk::PRDtoTrackMap &prd_to_track_map) const
  {
    static const SG::AuxElement::ConstAccessor< ElementLink< ::SpacePointCollection > > pixelLinkAcc("pixelSpacePointLink");
    static const SG::AuxElement::ConstAccessor< ElementLink< ::SpacePointCollection > > stripLinkAcc("sctSpacePointLink");
    static const SG::AuxElement::ConstAccessor< ElementLink< ::SpacePointOverlapCollection > > stripOverlapLinkacc("stripOverlapSpacePointLink");
 
    const Trk::SpacePoint *indetSp = nullptr;
    if ( pixelLinkAcc.isAvailable(*sp) ) {
      const ElementLink< ::SpacePointCollection > link = pixelLinkAcc( *sp );
      indetSp = *link;
    }
    else if ( stripLinkAcc.isAvailable(*sp) ) {
      const ElementLink< ::SpacePointCollection > link = stripLinkAcc( *sp );
      indetSp = *link;
    } else if ( stripOverlapLinkacc.isAvailable(*sp) ) {
      const ElementLink< ::SpacePointOverlapCollection > link = stripOverlapLinkacc( *sp );
      indetSp = *link;
    } else {
      return false;
    }
 
    return isUsed(indetSp, prd_to_track_map);
  }
  
}
 
#endif