CsvParticleWriter.cpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CsvParticleWriter.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/Io/Csv/CsvParticleWriter.hpp"

#include "ActsExamples/Framework/WhiteBoard.hpp"
#include "ActsExamples/Utilities/Paths.hpp"
#include <Acts/Utilities/Units.hpp>

#include <map>
#include <stdexcept>

#include <dfe/dfe_io_dsv.hpp>

#include "TrackMlData.hpp"

ActsExamples::CsvParticleWriter::CsvParticleWriter(
    const ActsExamples::CsvParticleWriter::Config& cfg,
    Acts::Logging::Level lvl)
    : WriterT(cfg.inputParticles, "CsvParticleWriter", lvl), m_cfg(cfg) {
  // inputParticles is already checked by base constructor
  if (m_cfg.outputStem.empty()) {
    throw std::invalid_argument("Missing ouput filename stem");
  }
}

ActsExamples::ProcessCode ActsExamples::CsvParticleWriter::writeT(
    const ActsExamples::AlgorithmContext& ctx,
    const SimParticleContainer& particles) {
  auto pathParticles = perEventFilepath(
      m_cfg.outputDir, m_cfg.outputStem + ".csv", ctx.eventNumber);
  dfe::NamedTupleCsvWriter<ParticleData> writer(pathParticles,
                                                m_cfg.outputPrecision);

  ParticleData data;
  for (const auto& particle : particles) {
    data.particle_id = particle.particleId().value();
    data.particle_type = particle.pdg();
    data.process = static_cast<decltype(data.process)>(particle.process());
    data.vx = particle.position().x() / Acts::UnitConstants::mm;
    data.vy = particle.position().y() / Acts::UnitConstants::mm;
    data.vz = particle.position().z() / Acts::UnitConstants::mm;
    data.vt = particle.time() / Acts::UnitConstants::ns;
    const auto p = particle.absMomentum() / Acts::UnitConstants::GeV;
    data.px = p * particle.unitDirection().x();
    data.py = p * particle.unitDirection().y();
    data.pz = p * particle.unitDirection().z();
    data.m = particle.mass() / Acts::UnitConstants::GeV;
    data.q = particle.charge() / Acts::UnitConstants::e;
    writer.append(data);
  }

  return ProcessCode::SUCCESS;
}