LArCalorimeter/LArG4/LArG4Code/LArG4Code/SDWrapper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef LARG4CODE_SDWRAPPER_H
#define LARG4CODE_SDWRAPPER_H
 
// System includes
#include <string>
#include <vector>
#include <memory>
 
// External includes
#include "G4VSensitiveDetector.hh"
 
// Framework includes
#include "StoreGate/WriteHandle.h"
 
 
// Forward declarations
class LArG4SimpleSD;
class LArHitContainer;
class LArG4CalibSD;
class CaloCalibrationHitContainer;
 
 
namespace LArG4
{
 
  namespace detail
  {
 
    template<class SDType, class HitContainerType>
    class SDWrapper : public G4VSensitiveDetector
    {
 
      public:
 
        using SDList_t = std::vector< std::unique_ptr<SDType> >;
 
        SDWrapper(const std::string& name, const std::string& hitCollectionName, const std::string& deadHitCollectionName="");
 
        void addSD(std::unique_ptr<SDType> sd);
 
        void addFastSimSD(const std::string& fastSimSDName);
 
        virtual void Initialize(G4HCofThisEvent*) override final;
 
        virtual bool ProcessHits(G4Step*, G4TouchableHistory*) override final;
 
        void EndOfAthenaEvent();
 
#ifdef SDWRAPPER_TEST//here use conditional compiling to solve the private members accessing problem from test code since adding-friend method can not work in this context(where there are two layer namespaces)
      public:
#else
      private:
#endif
 
        std::string m_hitCollName;
 
        SG::WriteHandle<HitContainerType> m_hitColl;
 
        std::string m_deadHitCollName;
 
        SG::WriteHandle<HitContainerType> m_deadHitColl;
 
        SDList_t m_sdList;
 
        std::string m_fastSimSDName;
 
    }; // class SDWrapper
 
    template<>
    void SDWrapper<LArG4CalibSD, CaloCalibrationHitContainer>::EndOfAthenaEvent();
 
  } // namespace detail
 
 
  using SimpleSDWrapper = detail::SDWrapper<LArG4SimpleSD, LArHitContainer>;
 
  using CalibSDWrapper = detail::SDWrapper<LArG4CalibSD, CaloCalibrationHitContainer>;
 
} // namespace LArG4
 
#endif