GeantinoRecording.cpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file GeantinoRecording.cpp

// This file is part of the Acts project.
//
// Copyright (C) 2017-2018 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include "ACTFW/MaterialMapping/GeantinoRecording.hpp"

#include <FTFP_BERT.hh>
#include <iostream>
#include <stdexcept>

#include "ACTFW/Framework/WhiteBoard.hpp"
#include "ACTFW/Plugins/Geant4/MMDetectorConstruction.hpp"
#include "ACTFW/Plugins/Geant4/MMEventAction.hpp"
#include "ACTFW/Plugins/Geant4/MMPrimaryGeneratorAction.hpp"
#include "ACTFW/Plugins/Geant4/MMRunAction.hpp"
#include "ACTFW/Plugins/Geant4/MMSteppingAction.hpp"

FW::GeantinoRecording::GeantinoRecording(
    const FW::GeantinoRecording::Config& cnf, Acts::Logging::Level level)
    : FW::BareAlgorithm("GeantinoRecording", level),
      m_cfg(cnf),
      m_runManager(std::make_unique<G4RunManager>()) {
  if (m_cfg.geant4Service) {
    m_runManager->SetUserInitialization(m_cfg.geant4Service->geant4Geometry());
  } else if (!m_cfg.gdmlFile.empty()) {
    ACTS_INFO(
        "received Geant4 geometry from GDML file: " << m_cfg.gdmlFile.c_str());
    FW::Geant4::MMDetectorConstruction* detConstruction =
        new FW::Geant4::MMDetectorConstruction();
    detConstruction->setGdmlInput(m_cfg.gdmlFile.c_str());
    m_runManager->SetUserInitialization(
        detConstruction);  // constructs detector (calls Construct in
                           // Geant4DetectorConstruction)
  } else {
    throw std::invalid_argument("Missing geometry input for Geant4");
  }

  m_runManager->SetUserInitialization(new FTFP_BERT);
  m_runManager->SetUserAction(new FW::Geant4::MMPrimaryGeneratorAction(
      "geantino", 1000., m_cfg.seed1, m_cfg.seed2));
  FW::Geant4::MMRunAction* runaction = new FW::Geant4::MMRunAction();
  m_runManager->SetUserAction(runaction);
  m_runManager->SetUserAction(new FW::Geant4::MMEventAction());
  m_runManager->SetUserAction(new FW::Geant4::MMSteppingAction());
  m_runManager->Initialize();
}

FW::ProcessCode FW::GeantinoRecording::execute(
    const FW::AlgorithmContext& context) const {
  // Begin with the simulation
  m_runManager->BeamOn(m_cfg.tracksPerEvent);
  // Retrieve the track material tracks from Geant4
  auto recordedMaterial =
      FW::Geant4::MMEventAction::Instance()->MaterialTracks();
  ACTS_INFO("Received " << recordedMaterial.size()
                        << " MaterialTracks. Writing them now onto file...");

  // Write the recorded material to the event store
  context.eventStore.add(m_cfg.geantMaterialCollection,
                         std::move(recordedMaterial));

  // // Retrieve the sim hit track steps from Geant4
  // auto trackSteps = FW::Geant4::MMEventAction::Instance()->TrackSteps();
  // ACTS_INFO("Received " << trackSteps.size()
  //                       << " steps per track. Writing them now into
  //                       file...");
  //
  // // Write the sim hit track steps info to the event store
  // context.eventStore.add(m_cfg.geantTrackStepCollection,
  // std::move(trackSteps));
  return FW::ProcessCode::SUCCESS;
}