SDWrapper.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#include "SDWrapper.h"
 
// Geant4 includes
#include "G4SDManager.hh"
 
// Framework utilities
 
// Project includes
#include "ISF_FastCaloSimEvent/FCS_StepInfoCollection.h"
 
// Local includes
#include "FCS_StepInfoSD.h"
 
namespace FCS_Param {
 
namespace detail {
 
//-------------------------------------------------------------------------
// Construct with a hit collection name
//-------------------------------------------------------------------------
template <class SDType, class HitContainerType>
SDWrapper<SDType, HitContainerType>::SDWrapper(
    const std::string& name, const std::string& hitCollectionName)
    : G4VSensitiveDetector(name),
      m_hitCollName(hitCollectionName),
      m_hitColl(hitCollectionName) {}
 
//-------------------------------------------------------------------------
// Add a unique SD to the list
//-------------------------------------------------------------------------
template <class SDType, class HitContainerType>
void SDWrapper<SDType, HitContainerType>::addSD(std::unique_ptr<SDType> sd) {
  m_sdList.push_back(std::move(sd));
}
 
//-------------------------------------------------------------------------
// Initialize the hit collection at the beginning of the G4 event
//-------------------------------------------------------------------------
template <class SDType, class HitContainerType>
void SDWrapper<SDType, HitContainerType>::Initialize(G4HCofThisEvent*) {
  if (!m_hitColl.isValid()) {
    if (verboseLevel >= 5) {
      G4cout << GetName() << " \tDEBUG\t"
             << "Initializing hit container: " << m_hitCollName << G4endl;
    }
    m_hitColl = std::make_unique<HitContainerType>();
  }
}
 
//-------------------------------------------------------------------------
// This should not be called
//-------------------------------------------------------------------------
template <class SDType, class HitContainerType>
bool SDWrapper<SDType, HitContainerType>::ProcessHits(G4Step*,
                                                      G4TouchableHistory*) {
  G4ExceptionDescription description;
  description << "ProcessHits: this SD shouldn't be assigned to volumes!";
  G4Exception(GetName(), "SDError", FatalException, description);
  return false;
}
 
//-------------------------------------------------------------------------
// Gather the hits into the WriteHandle from all the SDs
//-------------------------------------------------------------------------
template <class SDType, class HitContainerType>
void SDWrapper<SDType, HitContainerType>::EndOfAthenaEvent() {
  if (!m_hitColl.isValid()) {
    G4cerr << GetName() << " \tERROR\t" << "Hit collection WriteHandle is "
           << "invalid!" << G4endl;
    throw std::runtime_error("Invalid hit container WriteHandle: " +
                             m_hitColl.name());
  }
  // Loop over each SD and fill the container
  for (auto& sd : m_sdList) {
    sd->EndOfAthenaEvent(&*m_hitColl);
  }
}
 
//-------------------------------------------------------------------------
// Explit template instantiations
//-------------------------------------------------------------------------
template class SDWrapper<FCS_StepInfoSD,
                         ISF_FCS_Parametrization::FCS_StepInfoCollection>;
 
}  // namespace detail
 
}  // namespace FCS_Param