TrackReaderVertexingPerformanceWriterExample.cpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2020-2021 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 "Acts/Definitions/Units.hpp"
#include "ActsExamples/Framework/Sequencer.hpp"
#include "ActsExamples/Io/Performance/VertexPerformanceWriter.hpp"
#include "ActsExamples/Io/Root/RootParticleReader.hpp"
#include "ActsExamples/Io/Root/RootTrajectorySummaryReader.hpp"
#include "ActsExamples/Options/CommonOptions.hpp"
#include "ActsExamples/Options/MagneticFieldOptions.hpp"
#include "ActsExamples/Options/ParticleSelectorOptions.hpp"
#include "ActsExamples/Options/VertexingOptions.hpp"
#include "ActsExamples/Printers/TrackParametersPrinter.hpp"
#include "ActsExamples/TruthTracking/ParticleSelector.hpp"
#include "ActsExamples/TruthTracking/TrackParameterSelector.hpp"
#include "ActsExamples/Utilities/Paths.hpp"
#include "ActsExamples/Vertexing/AdaptiveMultiVertexFinderAlgorithm.hpp"

#include <memory>

using namespace Acts::UnitLiterals;
using namespace ActsExamples;

int main(int argc, char* argv[]) {
  // setup and parse options
  auto desc = Options::makeDefaultOptions();
  Options::addSequencerOptions(desc);
  Options::addRandomNumbersOptions(desc);
  Options::addVertexingOptions(desc);
  Options::addInputOptions(desc);
  Options::addMagneticFieldOptions(desc);
  Options::addOutputOptions(desc, OutputFormat::DirectoryOnly);
  Options::addParticleSelectorOptions(desc);
  auto vars = Options::parse(desc, argc, argv);
  if (vars.empty()) {
    return EXIT_FAILURE;
  }

  // basic setup
  auto logLevel = Options::readLogLevel(vars);
  auto rnd =
      std::make_shared<RandomNumbers>(Options::readRandomNumbersConfig(vars));
  Sequencer sequencer(Options::readSequencerConfig(vars));

  auto inputDir = vars["input-dir"].as<std::string>();
  auto outputDir =
      ensureWritableDirectory(vars["output-dir"].as<std::string>());

  // Setup the magnetic field
  auto magneticField = Options::readMagneticField(vars);

  RootParticleReader::Config particleReaderConfig;
  particleReaderConfig.particleCollection = "allTruthParticles";
  particleReaderConfig.filePath = inputDir + "/particles_initial.root";
  particleReaderConfig.treeName = "particles";
  sequencer.addReader(std::make_shared<RootParticleReader>(
      particleReaderConfig, Acts::Logging::INFO));

  // add additional particle selection
  auto select = Options::readParticleSelectorConfig(vars);
  select.inputParticles = particleReaderConfig.particleCollection;
  select.outputParticles = "detectorAcceptanceSelectedTruthParticles";
  sequencer.addAlgorithm(
      std::make_shared<ActsExamples::ParticleSelector>(select, logLevel));

  RootTrajectorySummaryReader::Config trackSummaryReader;
  trackSummaryReader.outputTracks = "fittedTrackParameters";
  trackSummaryReader.outputParticles = "associatedTruthParticles";
  trackSummaryReader.filePath = inputDir + "/tracksummary_fitter.root";
  sequencer.addReader(std::make_shared<RootTrajectorySummaryReader>(
      trackSummaryReader, logLevel));

  // Apply some primary vertexing selection cuts
  TrackParameterSelector::Config trackSelectorConfig;
  trackSelectorConfig.inputTrackParameters = trackSummaryReader.outputTracks;
  trackSelectorConfig.outputTrackParameters = "selectedTracks";
  trackSelectorConfig.absEtaMax = vars["vertexing-eta-max"].as<double>();
  trackSelectorConfig.loc0Max = vars["vertexing-rho-max"].as<double>() * 1_mm;
  trackSelectorConfig.ptMin = vars["vertexing-pt-min"].as<double>() * 1_MeV;
  sequencer.addAlgorithm(
      std::make_shared<TrackParameterSelector>(trackSelectorConfig, logLevel));

  // find vertices
  AdaptiveMultiVertexFinderAlgorithm::Config findVertices;
  findVertices.bField = magneticField;
  findVertices.inputTrackParameters = trackSelectorConfig.outputTrackParameters;
  findVertices.outputProtoVertices = "fittedProtoVertices";
  findVertices.outputVertices = "fittedVertices";
  sequencer.addAlgorithm(std::make_shared<AdaptiveMultiVertexFinderAlgorithm>(
      findVertices, logLevel));

  // write track parameters from fitting
  VertexPerformanceWriter::Config vertexWriterConfig;
  vertexWriterConfig.inputAllTruthParticles =
      particleReaderConfig.particleCollection;
  vertexWriterConfig.inputSelectedTruthParticles = select.outputParticles;
  vertexWriterConfig.inputAssociatedTruthParticles =
      trackSummaryReader.outputParticles;
  vertexWriterConfig.inputTrackParameters = trackSummaryReader.outputTracks;
  vertexWriterConfig.inputVertices = findVertices.outputVertices;
  vertexWriterConfig.filePath = outputDir + "/vertexperformance_AMVF.root";
  vertexWriterConfig.treeName = "amvf";
  sequencer.addWriter(
      std::make_shared<VertexPerformanceWriter>(vertexWriterConfig, logLevel));

  return sequencer.run();
}