InDetPRD_Provider.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/

// InDetPRD_Provider.h, (c) ATLAS Detector software
 
#ifndef INDET_INDETASSOCIATIONTOOLS_INDETPRDPROVIDER_H
#define INDET_INDETASSOCIATIONTOOLS_INDETPRDPROVIDER_H 1
 
// Gaudi
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
// InDet includes; these are needed (cannot be fwd declared) for template type deduction
#include "InDetPrepRawData/PixelClusterContainer.h"
#include "InDetPrepRawData/SCT_ClusterContainer.h"
#include "InDetPrepRawData/TRT_DriftCircleContainer.h"
// Trk includes
#include "TrkToolInterfaces/IPRD_Provider.h"
#include "TrkPrepRawData/PrepRawDataContainer.h"
#include "TrkPrepRawData/PrepRawDataCollection.h"
 
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/ReadCondHandleKey.h"
 
class AtlasDetectorID;
class PixelID;
class SCT_ID;
class TRT_ID; 
class Identifier;
class IdentifierHash;
namespace InDet {
 
     
  class InDetPRD_Provider : public AthAlgTool, virtual public Trk::IPRD_Provider {
 
     public:     
        //** Constructor with parameters */
        InDetPRD_Provider( const std::string& t, const std::string& n, const IInterface* p );
 
       StatusCode  initialize();
       
       StatusCode  finalize();

       const Trk::PrepRawData* prdFromIdentifier(const Identifier& ide, size_t& ndof ) const;
       
     private:
        template < class PrdT > const Trk::PrepRawData* prdFromIdentifierContainer(
                            const Trk::PrepRawDataContainer< Trk::PrepRawDataCollection< PrdT > >& cont,
                            const Identifier& ideh, const IdentifierHash& ideHash ) const 
        {
          // find the collection
          const Trk::PrepRawDataCollection< PrdT >* prdCollection = cont.indexFindPtr(ideHash);
          if ( prdCollection == nullptr ){
              ATH_MSG_VERBOSE("PRD Collection to IdentifierHash could not be found. Return 0.");
              return 0;
          }
          // search for the PRD in the collection --- do a loop, can be done better with std::find probably
          const PrdT* prd = 0;
          // iterate through the collections
          typename Trk::PrepRawDataCollection< PrdT >::const_iterator prdIter  = prdCollection->begin();
          typename Trk::PrepRawDataCollection< PrdT >::const_iterator prdIterE = prdCollection->end();
          for ( ; prdIter != prdIterE; ++prdIter ){
              if ( (*prdIter)->identify() == ideh ){
                  prd = (*prdIter);
                  break;
              }
          }
          // return what you have
          return prd;
        }
         
        const AtlasDetectorID*                      m_idHelper;                     
                                                    
        const PixelID*                              m_pixIdHelper;
        SG::ReadHandleKey<PixelClusterContainer>    m_pixClusterContainerKey
      {this, "PixelClusterContainer", ""};
        // For P->T converter of Pixel Clusters
        SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_pixelDetEleCollKey{this, "PixelDetEleCollKey", "PixelDetectorElementCollection", "Key of SiDetectorElementCollection for Pixel"};
        
        const SCT_ID*                               m_sctIdHelper;
        SG::ReadHandleKey<SCT_ClusterContainer>     m_sctClusterContainerKey
      {this, "SCT_ClusterContainer", ""};
        // For P->T converter of SCT_Clusters
        SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_SCTDetEleCollKey{this, "SCTDetEleCollKey", "SCT_DetectorElementCollection", "Key of SiDetectorElementCollection for SCT"};
        
        const TRT_ID*                               m_trtIdHelper;
        SG::ReadHandleKey<TRT_DriftCircleContainer> m_trtDriftCircleContainerKey
      {this, "TRT_DriftCircleContainer", ""};
 
  };
 
 
} // end of namespace
 
#endif // INDET_INDETASSOCIATIONTOOLS_INDETPRDPROVIDER_H