MCTruthSteppingActionTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MCTruthSteppingActionTool.h"
 
 
namespace G4UA
{
 
  //---------------------------------------------------------------------------
  // Constructor
  //---------------------------------------------------------------------------
  MCTruthSteppingActionTool::
  MCTruthSteppingActionTool(const std::string& type, const std::string& name,
                            const IInterface* parent)
    : UserActionToolBase<MCTruthSteppingAction>(type, name, parent),
      m_secondarySavingLevel(2),
      m_subDetVolLevel(1)
  {
    declareProperty("VolumeCollectionMap", m_volumeCollectionMap,
                    "Map of volume name to output collection name");
    declareProperty("SecondarySavingLevel", m_secondarySavingLevel,
                    "Three valid options: 1 - Primaries; "
                    "2 - StoredSecondaries(default); 3 - All");
    declareProperty("SubDetVolumeLevel", m_subDetVolLevel,
                    "The level in the G4 volume hierarchy at which can we find "
                    "the sub-detector name");
  }
 
  //---------------------------------------------------------------------------
  // Initialize the tool
  //---------------------------------------------------------------------------
  StatusCode MCTruthSteppingActionTool::initialize()
  {
    ATH_MSG_DEBUG( "Initializing " << name() );
    return StatusCode::SUCCESS;
  }
 
  //---------------------------------------------------------------------------
  // Create an MCTruthSteppingAction
  //---------------------------------------------------------------------------
  std::unique_ptr<MCTruthSteppingAction>
  MCTruthSteppingActionTool::makeAndFillAction(G4AtlasUserActions& actionLists)
  {
    ATH_MSG_DEBUG("Constructing an MCTruthSteppingAction");
    auto action = std::make_unique<MCTruthSteppingAction> (
        m_volumeCollectionMap, m_secondarySavingLevel, m_subDetVolLevel,
        msgSvc(), msg().level() );
    actionLists.eventActions.push_back( action.get() );
    actionLists.steppingActions.push_back( action.get() );
    return action;
  }

  StatusCode MCTruthSteppingActionTool::BeginOfAthenaEvent(HitCollectionMap& hitCollections)
  {
    for(const auto& volCollPair : m_volumeCollectionMap) {
      hitCollections.Emplace<TrackRecordCollection>(volCollPair.second, volCollPair.second);
    }
    return StatusCode::SUCCESS;
  }

  StatusCode MCTruthSteppingActionTool::EndOfAthenaEvent(HitCollectionMap& hitCollections)
  {
    for(const auto& volCollPair : m_volumeCollectionMap) {
      CHECK(hitCollections.Record<TrackRecordCollection>(volCollPair.second));
    }
    return StatusCode::SUCCESS;
  }
 
} // namespace G4UA