Csv/src/CsvMuonSimHitReader.cpp

0001 // This file is part of the Acts project.
0002 //
0003 // Copyright (C) 2023 CERN for the benefit of the Acts project
0004 //
0005 // This Source Code Form is subject to the terms of the Mozilla Public
0006 // License, v. 2.0. If a copy of the MPL was not distributed with this
0007 // file, You can obtain one at http://mozilla.org/MPL/2.0/.
0008
0009 #include "ActsExamples/Io/Csv/CsvMuonSimHitReader.hpp"
0010
0011 #include "Acts/Definitions/Units.hpp"
0012 #include "Acts/Geometry/GeometryIdentifier.hpp"
0013 #include "ActsExamples/EventData/MuonSimHit.hpp"
0014 #include "ActsExamples/Framework/AlgorithmContext.hpp"
0015 #include "ActsExamples/Utilities/Paths.hpp"
0016 #include "ActsFatras/EventData/Barcode.hpp"
0017 #include "ActsFatras/EventData/Hit.hpp"
0018
0019 #include <array>
0020 #include <stdexcept>
0021
0022 #include <dfe/dfe_io_dsv.hpp>
0023
0024 #include "CsvOutputData.hpp"
0025
0026 ActsExamples::CsvMuonSimHitReader::CsvMuonSimHitReader(
0027     const ActsExamples::CsvMuonSimHitReader::Config& config,
0028     Acts::Logging::Level level)
0029     : m_cfg(config),
0030       // TODO check that all files (hits,cells,truth) exists
0031       m_eventsRange(
0032           determineEventFilesRange(m_cfg.inputDir, m_cfg.inputStem + ".csv")),
0033       m_logger(Acts::getDefaultLogger("CsvMuonSimHitReader", level)) {
0034   if (m_cfg.inputStem.empty()) {
0035     throw std::invalid_argument("Missing input filename stem");
0036   }
0037   if (m_cfg.outputSimHits.empty()) {
0038     throw std::invalid_argument("Missing simulated hits output collection");
0039   }
0040
0041   m_outputSimHits.initialize(m_cfg.outputSimHits);
0042 }
0043
0044 std::string ActsExamples::CsvMuonSimHitReader::CsvMuonSimHitReader::name()
0045     const {
0046   return "CsvMuonSimHitReader";
0047 }
0048
0049 std::pair<std::size_t, std::size_t>
0050 ActsExamples::CsvMuonSimHitReader::availableEvents() const {
0051   return m_eventsRange;
0052 }
0053
0054 ActsExamples::ProcessCode ActsExamples::CsvMuonSimHitReader::read(
0055     const ActsExamples::AlgorithmContext& ctx) {
0056   auto path = perEventFilepath(m_cfg.inputDir, m_cfg.inputStem + ".csv",
0057                                ctx.eventNumber);
0058
0059   dfe::NamedTupleCsvReader<MuonSimHitData> reader(path);
0060
0061   MuonSimHitData data;
0062
0063   SimHitContainer::sequence_type unordered;
0064   while (reader.read(data)) {
0065     ActsFatras::Hit::Vector4 pos{
0066         data.LocalPositionExtrx * Acts::UnitConstants::mm,
0067         data.LocalPositionExtry * Acts::UnitConstants::mm,
0068         data.LocalPositionExtrz * Acts::UnitConstants::mm, 0};
0069     ActsFatras::Hit::Vector4 mom{
0070         data.LocalDirectionx * Acts::UnitConstants::GeV,
0071         data.LocalDirectiony * Acts::UnitConstants::GeV,
0072         data.LocalDirectionz * Acts::UnitConstants::GeV,
0073         std::sqrt(data.LocalDirectionx * data.LocalDirectionx +
0074                   data.LocalDirectiony * data.LocalDirectiony +
0075                   data.LocalDirectionz * data.LocalDirectionz) *
0076             Acts::UnitConstants::GeV};
0077     muonMdtIdentifierFields f;
0078     f.multilayer = 0;
0079     f.tube = 0;
0080     f.tubeLayer = 0;
0081     f.stationEta = data.StationEta;
0082     f.stationPhi = data.StationPhi;
0083     f.stationName = data.StationName;
0084
0085     unordered.push_back(SimHit(compressId(f), data.pdgId, pos, mom, mom, -1));
0086   }
0087   SimHitContainer simHits;
0088   simHits.insert(unordered.begin(), unordered.end());
0089
0090   // write the ordered data to the EventStore (according to geometry_id).
0091   m_outputSimHits(ctx, std::move(simHits));
0092
0093   return ActsExamples::ProcessCode::SUCCESS;
0094 }