FatrasEvgenBase.cpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2019 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 "FatrasEvgenBase.hpp"

#include "ACTFW/Framework/RandomNumbers.hpp"
#include "ACTFW/Framework/Sequencer.hpp"
#include "ACTFW/Generators/EventGenerator.hpp"
#include "ACTFW/Generators/FlattenEvent.hpp"
#include "ACTFW/Io/Csv/CsvParticleWriter.hpp"
#include "ACTFW/Io/Root/RootParticleWriter.hpp"
#include "ACTFW/Options/CommonOptions.hpp"
#include "ACTFW/Options/ParticleGunOptions.hpp"
#include "ACTFW/Options/Pythia8Options.hpp"
#include "ACTFW/Utilities/Paths.hpp"

void FW::setupEvgenInput(
    const FW::Options::Variables& vm, FW::Sequencer& sequencer,
    std::shared_ptr<const FW::RandomNumbers> randomNumberSvc) {
  // Read the standard options
  auto logLevel = FW::Options::readLogLevel(vm);

  // Add requested event generator
  auto evgenInput = vm["evg-input-type"].template as<std::string>();
  if (evgenInput == "gun") {
    auto evgCfg = FW::Options::readParticleGunOptions(vm);
    evgCfg.output = "event_generated";
    evgCfg.randomNumbers = randomNumberSvc;
    sequencer.addReader(std::make_shared<FW::EventGenerator>(evgCfg, logLevel));

  } else if (evgenInput == "pythia8") {
    auto evgCfg = FW::Options::readPythia8Options(vm, logLevel);
    evgCfg.output = "event_generated";
    evgCfg.randomNumbers = randomNumberSvc;
    sequencer.addReader(std::make_shared<FW::EventGenerator>(evgCfg, logLevel));

  } else {
    throw std::runtime_error("unknown event generator input: " + evgenInput);
  }

  // Convert to particles for the writers and subsequent algorithms
  FW::FlattenEvent::Config flatten;
  flatten.inputEvent = "event_generated";
  flatten.outputParticles = "particles_generated";
  sequencer.addAlgorithm(std::make_shared<FW::FlattenEvent>(flatten, logLevel));

  // Output directory
  std::string outputDir = vm["output-dir"].template as<std::string>();

  // Write generated particles as CSV files
  if (vm["output-csv"].template as<bool>()) {
    FW::CsvParticleWriter::Config pWriterCsvConfig;
    pWriterCsvConfig.inputParticles = flatten.outputParticles;
    pWriterCsvConfig.outputDir = outputDir;
    pWriterCsvConfig.outputStem = "particles_generated";
    sequencer.addWriter(
        std::make_shared<FW::CsvParticleWriter>(pWriterCsvConfig, logLevel));
  }

  // Write generated particles as ROOT file
  if (vm["output-root"].template as<bool>()) {
    // Write particles as ROOT TTree
    FW::RootParticleWriter::Config pWriterRootConfig;
    pWriterRootConfig.inputParticles = flatten.outputParticles;
    pWriterRootConfig.filePath = FW::joinPaths(outputDir, "particles.root");
    sequencer.addWriter(
        std::make_shared<FW::RootParticleWriter>(pWriterRootConfig, logLevel));
  }
}