GenericReadCsvExample.cpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file GenericReadCsvExample.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 "ActsExamples/Framework/Sequencer.hpp"
#include "ActsExamples/Framework/WhiteBoard.hpp"
#include "ActsExamples/GenericDetector/GenericDetector.hpp"
#include "ActsExamples/Geometry/CommonGeometry.hpp"
#include "ActsExamples/Io/Csv/CsvOptionsReader.hpp"
#include "ActsExamples/Io/Csv/CsvParticleReader.hpp"
#include "ActsExamples/Io/Csv/CsvPlanarClusterReader.hpp"
#include "ActsExamples/Io/Csv/CsvPlanarClusterWriter.hpp"
#include "ActsExamples/Options/CommonOptions.hpp"
#include "ActsExamples/Printers/PrintHits.hpp"
#include "ActsExamples/Utilities/Options.hpp"

#include <memory>

int main(int argc, char* argv[]) {
  GenericDetector detector;

  // setup and parse options
  auto desc = ActsExamples::Options::makeDefaultOptions();
  ActsExamples::Options::addSequencerOptions(desc);
  ActsExamples::Options::addGeometryOptions(desc);
  ActsExamples::Options::addMaterialOptions(desc);
  ActsExamples::Options::addInputOptions(desc);
  detector.addOptions(desc);

  auto vm = ActsExamples::Options::parse(desc, argc, argv);
  if (vm.empty()) {
    return EXIT_FAILURE;
  }

  ActsExamples::Sequencer sequencer(
      ActsExamples::Options::readSequencerConfig(vm));

  // Read some standard options
  auto logLevel = ActsExamples::Options::readLogLevel(vm);
  auto inputDir = vm["input-dir"].as<std::string>();

  // Setup detector geometry
  auto geometry = ActsExamples::Geometry::build(vm, detector);
  auto trackingGeometry = geometry.first;
  // Add context decorators
  for (auto cdr : geometry.second) {
    sequencer.addContextDecorator(cdr);
  }

  // Read particles from CSV files
  auto particleReaderCfg =
      ActsExamples::Options::readCsvParticleReaderConfig(vm);
  particleReaderCfg.outputParticles = "particles";
  sequencer.addReader(std::make_shared<ActsExamples::CsvParticleReader>(
      particleReaderCfg, logLevel));

  // Read clusters from CSV files
  auto clusterReaderCfg =
      ActsExamples::Options::readCsvPlanarClusterReaderConfig(vm);
  clusterReaderCfg.trackingGeometry = trackingGeometry;
  clusterReaderCfg.outputClusters = "clusters";
  clusterReaderCfg.outputHitParticlesMap = "hit_particle_map";
  clusterReaderCfg.outputHitIds = "hit_ids";
  sequencer.addReader(std::make_shared<ActsExamples::CsvPlanarClusterReader>(
      clusterReaderCfg, logLevel));

  // Print some information as crosscheck
  ActsExamples::PrintHits::Config printCfg;
  printCfg.inputClusters = clusterReaderCfg.outputClusters;
  printCfg.inputHitParticlesMap = clusterReaderCfg.outputHitParticlesMap;
  printCfg.inputHitIds = clusterReaderCfg.outputHitIds;
  // the following print selections work for the original author.
  // you probably need to adapt them to your data.
  printCfg.hitIdStart = 10224;
  printCfg.hitIdLength = 8;
  printCfg.volumeId = 13;
  printCfg.layerId = 4;
  printCfg.moduleId = 116;
  sequencer.addAlgorithm(
      std::make_shared<ActsExamples::PrintHits>(printCfg, logLevel));

  return sequencer.run();
}