TRT_SeededSpacePointFinder_ATL.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
/********************************************************************************
  Header file for class TRT_SeededSpacePointFinder_ATL
  (c) ATLAS Detector software
  Class for Si seed candidates generation using space point information
  Version 1.0: 04/15/2006
  Authors: Thomas Koffas
  email  : Thomas.Koffas@cern.ch
********************************************************************************/
 
#ifndef TRT_SeededSpacePointFinder_ATL_H
#define TRT_SeededSpacePointFinder_ATL_H
 
 
 
//Tool Handle
//
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
 
#include "TrkSpacePoint/SpacePointContainer.h"
#include "TrkSpacePoint/SpacePointOverlapCollection.h"
 
#include "InDetRecToolInterfaces/ITRT_SeededSpacePointFinder.h"
#include "TrkEventUtils/EventDataBase.h"
 
#include "TrkGeometry/MagneticFieldProperties.h"
 
#include "TrkEventUtils/PRDtoTrackMap.h"
 
// MagField cache
#include "MagFieldConditions/AtlasFieldCacheCondObj.h"
#include "MagFieldElements/AtlasFieldCache.h"
 
#include <list>
#include <vector>
#include <iosfwd>
 
class MsgStream   ;
class SCT_ID      ;
class SiSpacePointsSeed;
 
namespace InDet{
 
  class MyNewDataSortPredicate {
    public:
    bool operator()(std::pair<const Trk::SpacePoint*,int> sp1, std::pair<const Trk::SpacePoint*,int> sp2) const
      {return (sp1.first)->r() > (sp2.first)->r();}
  };
}
 
namespace InDet{
 
  class TRT_SeededSpacePointFinder_ATL :
    virtual public ITRT_SeededSpacePointFinder, public AthAlgTool
    {
      // Public methods:
 
    public:
 
 
 
      TRT_SeededSpacePointFinder_ATL
    (const std::string&,const std::string&,const IInterface*);
      virtual ~TRT_SeededSpacePointFinder_ATL();
      virtual StatusCode               initialize();
      virtual StatusCode               finalize  ();
 
 
 
      std::unique_ptr<InDet::ITRT_SeededSpacePointFinder::IEventData> newEvent () const;
      std::unique_ptr<InDet::ITRT_SeededSpacePointFinder::IEventData> newRegion
    (const std::vector<IdentifierHash>&,
         const std::vector<IdentifierHash>&) const;
 
 
 
      std::list<std::pair<const Trk::SpacePoint*,const Trk::SpacePoint*> > find2Sp (const EventContext& ctx,
                                                                                    const Trk::TrackParameters&,
                                                                                    ITRT_SeededSpacePointFinder::IEventData &event_data) const;
 
 
 
      const SiSpacePointsSeed* next(ITRT_SeededSpacePointFinder::IEventData &event_data) const;
 
 
 
      MsgStream&    dump          (MsgStream   & out) const;
      std::ostream& dump          (std::ostream& out) const;
 
    protected:
 
 
 
      StringProperty m_fieldmode{this, "MagneticFieldMode", "MapSolenoid",
      "Magnetic field mode"};
 
      Trk::MagneticFieldProperties           m_fieldprop             ;  
      const SCT_ID* m_sctId = nullptr;
 
 
 
      DoubleProperty m_ptmin{this, "pTmin", 500., "Minimum pT cut"};
 
      static constexpr double m_r_rmin = 0.;  
      static constexpr double m_r_rmax = 600.;  
      static constexpr double m_r_rstep = 10.;  
      static constexpr double m_r1max = 560.;  
      static constexpr double m_r12min = 400.;  
      static constexpr double m_r2min = 340.;  
      static constexpr double m_dzdrmin = -4.;  
      static constexpr double m_dzdrmax = 4.;  
      DoubleProperty m_xiC{this, "Xi2C", 100., "qOverP based chi2 cut"};
      DoubleProperty m_xiTC{this, "Xi2TC", 100., "theta based chi2 cut"};
      DoubleProperty m_xiFC{this, "Xi2FC", 100., "phi based chi2 cut"};
      BooleanProperty m_search{this, "NeighborSearch", true, "Do full neighbor search"};
      BooleanProperty m_loadFull{this, "LoadFull", true,
    "Load full Si space point container"};
      BooleanProperty m_doCosmics{this, "DoCosmics", false,
    "Disable seed selection cuts during reconstruction of cosmics tracks"};
 
 
 
       class EventData;
       class EventData : public Trk::EventDataBase<EventData,InDet::ITRT_SeededSpacePointFinder::IEventData>
       {
         friend class TRT_SeededSpacePointFinder_ATL;
      public:
         EventData()
            : m_r_size(0),
              m_ns(0),
              m_fNmax(0),
              m_nr(0),
              m_nrf(0),
              m_sF(0.)
         {
            m_r_index     = 0     ;
            m_r_map       = 0     ;
         }
         ~EventData() {
              if(m_r_index)  delete [] m_r_index;
              if(m_r_map  )  delete [] m_r_map  ;
         }
 
      protected:
         void buildFrameWork(double r_rmax, double r_rstep, double ptmin);
         void erase(); // unused?
 
         int m_r_size{}                                                   ;
         std::list<std::pair<const Trk::SpacePoint*,int> > m_rf_Sorted[530]   ;
         std::list<std::pair<const Trk::SpacePoint*,int> > m_newRfi_Sorted    ;
 
         int m_ns=0                                                     ;
         int m_fNmax=0                                                  ;
         int  m_nr=0     ; int* m_r_index=nullptr; int* m_r_map=nullptr ;
         int  m_nrf=0   , m_rf_index   [ 530]{}, m_rf_map   [ 530]{}        ;
         double m_sF{};
 
      };
 
 
 
      SG::ReadHandleKey<SpacePointContainer>         m_spacepointsPixname {this,"SpacePointsPixelName","PixelSpacePoints","RHK to retrieve Pixel SpacePointContainer"}            ;
      SG::ReadHandleKey<SpacePointContainer>         m_spacepointsSCTname {this,"SpacePointsSCTName","SCT_SpacePoints","RHK to retrieve SCT SpacePointContainer"}           ;
      SG::ReadHandleKey<SpacePointOverlapCollection> m_spacepointsOverlapname {this,"SpacePointsOverlapName","OverlapSpacePoints","RHK to retrieve OverlapCollection"}        ;
      SG::ReadHandleKey<Trk::PRDtoTrackMap>          m_prdToTrackMap
         {this,"PRDtoTrackMap",""};
 
      // Read handle for conditions object to get the field cache
      SG::ReadCondHandleKey<AtlasFieldCacheCondObj> m_fieldCondObjInputKey {this, "AtlasFieldCacheCondObj",
        "fieldCondObj", "Name of the Magnetic Field conditions object key"};
 
 
 
      MsgStream&    dumpConditions(MsgStream   & out) const;
      MsgStream&    dumpEvent     (MsgStream   & out,InDet::TRT_SeededSpacePointFinder_ATL::EventData &event_data) const;
 
 
      void fillLists     (std::vector< std::vector<const Trk::SpacePoint*> > &r_Sorted,
                          InDet::TRT_SeededSpacePointFinder_ATL::EventData &event_data) const;
 
 
      // // // // // // // // // // // // // // // // //
      void production2Spb (const EventContext& ctx, const Trk::TrackParameters&,
                           int,
                           std::list<std::pair<const Trk::SpacePoint*,const Trk::SpacePoint*> > &outputListBuffer,
                           InDet::TRT_SeededSpacePointFinder_ATL::EventData &event_data) const;
      // // // // // // // // // // // // // // // // //
 
      void geoInfo(const Trk::SpacePoint*, int&, int&) const          ;
 
 
      void magneticFieldInit();
 
      // ptr to SP-specific data to be stored after calculated only once
      // instead of repeating the same computations many times in the
      // nested loop (memory allocated and deallocated in production2Spb)
      struct bypass_struct {
        double X{}, Y{}, Z{};
        double R{}, invR{};
        double a{}, b{};
      };
 
      // place to keep scalar values needed for cuts, common to all seeds
      // updated only once per production2Sp call instead or each cutTPb
      struct invar_bypass_struct {
        double min_theta{}, max_theta{}, min_phi{}, max_phi{}, invp_min{}, invp_max{},
        invp_min2{}, invp_max2{};
      };
 
      // // // // // // // // // // // // // // // // //
      bool cutTPb(const invar_bypass_struct &invar_bypass, const std::vector<bypass_struct> &prod_bypass,long, long, double) const;
      // // // // // // // // // // // // // // // // //
 
    };
 
 
 
  inline const SiSpacePointsSeed* TRT_SeededSpacePointFinder_ATL::next(ITRT_SeededSpacePointFinder::IEventData &) const
    {
      return 0;
    }
 
} // end of name space
 
#endif // TRT_SeededSpacePointFinder_ATL_H