CsvMultiTrajectoryWriter.hpp

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

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

#pragma once

#include "Acts/Definitions/Units.hpp"
#include "Acts/EventData/MultiTrajectoryHelpers.hpp"
#include "Acts/Utilities/Logger.hpp"
#include "ActsExamples/EventData/Index.hpp"
#include "ActsExamples/EventData/Track.hpp"
#include "ActsExamples/EventData/Trajectories.hpp"
#include "ActsExamples/Framework/DataHandle.hpp"
#include "ActsExamples/Framework/ProcessCode.hpp"
#include "ActsExamples/Framework/WriterT.hpp"
#include "ActsFatras/EventData/Barcode.hpp"

#include <cstddef>
#include <cstdint>
#include <fstream>
#include <string>
#include <vector>

namespace ActsExamples {
struct AlgorithmContext;
}  // namespace ActsExamples

using namespace Acts::UnitLiterals;

namespace ActsExamples {

class CsvMultiTrajectoryWriter : public WriterT<TrajectoriesContainer> {
 public:
  using HitParticlesMap = ActsExamples::IndexMultimap<ActsFatras::Barcode>;

  struct Config {
    std::string inputTrajectories;           
    std::string outputDir;                   
    std::string fileName = "CKFtracks.csv";  
    std::string
        inputMeasurementParticlesMap;  
    size_t outputPrecision = 6;        
    size_t nMeasurementsMin = 7;       
    bool onlyTruthMatched = false;     
    double truthMatchProbMin = 0.5;  
    double ptMin = 1_GeV;            
  };

  CsvMultiTrajectoryWriter(const Config& config,
                           Acts::Logging::Level level = Acts::Logging::INFO);

  const Config& config() const { return m_cfg; }

 protected:
  ProcessCode writeT(const AlgorithmContext& context,
                     const TrajectoriesContainer& trajectories) override;

 private:
  Config m_cfg;  

  ReadDataHandle<HitParticlesMap> m_inputMeasurementParticlesMap{
      this, "InputMeasurementParticlesMap"};

  struct trackInfo : public Acts::MultiTrajectoryHelpers::TrajectoryState {
    size_t trackId = 0;
    ActsFatras::Barcode particleId;
    size_t nMajorityHits = 0;
    std::string trackType;
    double truthMatchProb = 0;
    const TrackParameters* fittedParameters = nullptr;
    std::vector<uint64_t> measurementsID;
  };  // trackInfo struct
};

}  // namespace ActsExamples