TrigInDetAccelerationSvc.h

Go to the documentation of this file.

/*   
     Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRIGINDETACCELERATIONSERVICE_TRIGINDETACCELERATIONSVC_H
#define TRIGINDETACCELERATIONSERVICE_TRIGINDETACCELERATIONSVC_H
 
#include "GaudiKernel/IIncidentListener.h" 
#include "GaudiKernel/IIncidentSvc.h" 
 
#include "TrigInDetAccelerationService/ITrigInDetAccelerationSvc.h" 
 
#include "AthenaBaseComps/AthService.h" 
 
#include "GaudiKernel/ServiceHandle.h" 
 
#include "StoreGate/StoreGateSvc.h" 
 
#include "TrigAccelEvent/WorkFactory.h"
#include "TrigAccelEvent/Module.h"
 
#include "AthCUDAInterfaces/IGPUSystemInfoSvc.h"
 
#include<vector>
#include<map>
#include<cstring>
#include<mutex>

 
class TrigInDetAccelerationSvc : public extends<AthService, ITrigInDetAccelerationSvc, IIncidentListener> {
 
 public:   
  
  TrigInDetAccelerationSvc( const std::string&, ISvcLocator*);
   
  virtual ~TrigInDetAccelerationSvc() override {};
  
  virtual StatusCode initialize() override;   
  virtual StatusCode finalize() override;
  virtual void handle(const Incident&) override;
 
  virtual bool isReady() const override {
    return m_factoryConfigured;
  }
  
  virtual TrigAccel::Work* createWork(unsigned int, std::shared_ptr<TrigAccel::OffloadBuffer>) const override;
  virtual const std::vector<short>& getLayerInformation(int) const override;
  virtual size_t getMiddleLayersSize() const override {return m_middleSpacePointLayers.size();};
 
 private:   
  StatusCode initializeWorkFactory(); 
 
  struct PhiEtaHash {
 
    struct compare {
    public:
      bool operator()(const struct PhiEtaHash& p1, const struct PhiEtaHash& p2) {
    if(p1.m_phiIndex == p2.m_phiIndex) {
      return p1.m_etaIndex < p2.m_etaIndex;
    }
    else {
      return p1.m_phiIndex < p2.m_phiIndex;
    }
      }
    };
    
  public:
  PhiEtaHash(short phi, short eta, int hash) : m_phiIndex(phi), m_etaIndex(eta), m_hash(hash) {};
    short m_phiIndex, m_etaIndex;
    int m_hash;
  };
 
  int m_nDCs;
  std::string m_moduleName;
  bool m_useITkGeometry;
  std::vector<int> m_middleSpacePointLayers;
  void* m_libHandle; //for OffloadFactory
  TrigAccel::WorkFactory* m_pWF;
  TrigAccel::Module* m_module;
  ServiceHandle<StoreGateSvc> m_detStore; 
  ServiceHandle<StoreGateSvc> m_evtStore;   
  ServiceHandle<AthCUDA::IGPUSystemInfoSvc> m_cudaCheckSvc;
 
  bool m_factoryConfigured;
 
  bool exportGeometryInformation(const std::map<std::tuple<short,short,short>, std::vector<PhiEtaHash> >&) const;
  bool extractGeometryInformation(std::map<std::tuple<short,short,short>, std::vector<PhiEtaHash> >&);
 
  bool extractITkGeometryInformation(std::map<std::tuple<short, short, int, int>, std::vector<PhiEtaHash> >&);
  bool exportITkGeometryInformation(const std::map<std::tuple<short,short, int, int>,std::vector<PhiEtaHash> >& hashMap) const;
 
  // Index 0 stores mapping from a position (barrel_ec, subdetid, new_vol, new_lay) index to it's position in barrel/endcap
  // Index 1 and 2 stores mapping from subdetector's PhiEtaHash to the position index
  std::vector<short> m_layerInfo[3];
 
  mutable std::mutex m_workMutex;
         
};
 
 
 
#endif