DNNCaloSimSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// DNNCaloSimSvc.h, (c) ATLAS Detector software
 
#ifndef ISF_DNNCALOSIMSVC_H
#define ISF_DNNCALOSIMSVC_H 1
 
// ISF includes
#include "ISF_Interfaces/BaseSimulationSvc.h"
 
// Framework includes
#include "GaudiKernel/IChronoStatSvc.h"
 
// FastCaloSim includes
#include "ISF_FastCaloSimParametrization/IFastCaloSimCaloExtrapolation.h"
#include "TrkExInterfaces/ITimedExtrapolator.h" 
 
#include "CaloInterface/ICaloCellMakerTool.h"
 
#include "AthenaKernel/IAtRndmGenSvc.h"
 
#include "AtlasDetDescr/AtlasDetectorID.h"
#include "CaloIdentifier/LArEM_ID.h"
#include "CaloIdentifier/LArHEC_ID.h"
#include "CaloIdentifier/LArFCAL_ID.h"
#include "CaloIdentifier/TileID.h"
#include "CaloDetDescr/CaloDetDescrManager.h"
 
#include "lwtnn/LightweightGraph.hh"
#include "CaloEvent/CaloCellContainer.h"
#include "CLHEP/Units/PhysicalConstants.h"
 
#include "CxxUtils/checker_macros.h"
 
namespace CLHEP
{
 class HepRandomEngine;
}
 
//forward declarations
class CaloCellContainer;
class CaloGeometryFromCaloDDM;
class TFCSParametrizationBase;
 
 
namespace ISF {
  
  class ATLAS_NOT_THREAD_SAFE DNNCaloSimSvc final : public BaseSimulationSvc
  {
    public:
      DNNCaloSimSvc(const std::string& name, ISvcLocator* pSvcLocator);
       
      virtual ~DNNCaloSimSvc() final;
      
      virtual StatusCode  initialize() override final;
      virtual StatusCode  finalize() override final;
      
      StatusCode initializeNetwork();

    virtual StatusCode simulate(ISFParticle& isp, McEventCollection*) override final;
      // type of input requested by lwtnn
      typedef std::map<std::string, std::map<std::string, double> >  NetworkInputs ;
      typedef std::map<std::string, double> NetworkOutputs;
      StatusCode fillNetworkInputs(const ISF::ISFParticle& isfp, NetworkInputs  & inputs, double & trueEnergy);
      StatusCode fillWindowCells(const double etaExtrap,const double phiExtrap,const CaloDetDescrElement* & impactCellDDE);

      virtual StatusCode setupEvent() override final;
      
      virtual StatusCode releaseEvent() override final;
      
      std::string m_paramsFilename;
      std::string m_paramsInputArchitecture;
 
      std::unique_ptr<lwt::LightweightGraph> m_graph;
      
      ToolHandleArray<ICaloCellMakerTool> m_caloCellMakerToolsSetup ;
      ToolHandleArray<ICaloCellMakerTool> m_caloCellMakerToolsRelease ;
 
      ToolHandle<IFastCaloSimCaloExtrapolation> m_FastCaloSimCaloExtrapolation;
      ToolHandle<Trk::ITimedExtrapolator> m_extrapolator;
      
      CaloCellContainer *       m_theContainer;
 
      ServiceHandle<IAtRndmGenSvc>    m_rndGenSvc;
      CLHEP::HepRandomEngine*         m_randomEngine;
      std::string                     m_randomEngineName;
      
 
      const CaloDetDescrManager* m_caloDetDescrManager;
      std::unique_ptr<CaloGeometryFromCaloDDM> m_caloGeo;
      const LArEM_ID* m_emID = nullptr;
      std::vector<CaloCell*> m_windowCells;
 
      // specific to architecture
      // preprocessing of input
      int m_GANLatentSize = 0;
      double m_logTrueEnergyMean = 0.;
      double m_logTrueEnergyScale = 0.;
      double m_riImpactEtaMean = 0.;
      double m_riImpactEtaScale = 0.;
      double m_riImpactPhiMean = 0.;
      double m_riImpactPhiScale = 0.;
 
      // building of the 266 cells cluster
      const int m_numberOfCellsForDNN = 266;
      const double m_middleCellWidthEta = 0.025;
      const double m_middleCellWidthPhi = CLHEP::pi / std::pow(2,7);
      const double m_etaRawMiddleCut = m_middleCellWidthEta * 3.5;
      const double m_etaRawBackCut = m_middleCellWidthEta * 4.;
      const double m_phiRawMiddleCut = m_middleCellWidthPhi * 3.5;
      const double m_phiRawStripCut = m_middleCellWidthPhi * 6.0;
 
      
      std::string  m_caloCellsOutputName;
  };
 
}
 
#endif //> !ISF_DNNCALOSIMSVC_H