RootMaterialTrackReader.hpp

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

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

#pragma once

#include "ActsExamples/Framework/IReader.hpp"
#include "ActsExamples/Framework/IService.hpp"
#include "ActsExamples/Framework/ProcessCode.hpp"
#include <Acts/Propagator/MaterialInteractor.hpp>
#include <Acts/Utilities/Definitions.hpp>
#include <Acts/Utilities/Logger.hpp>

#include <mutex>
#include <vector>

class TChain;

namespace ActsExamples {

class RootMaterialTrackReader : public IReader {
 public:
  struct Config {
    std::string collection =
        "material-tracks";                     
    std::string filePath = "";                 
    std::string treeName = "material-tracks";  
    std::vector<std::string> fileList;         

    unsigned int batchSize = 1;  

    std::shared_ptr<const Acts::Logger> logger;

    std::string name;

    Config(const std::string& lname = "MaterialReader",
           Acts::Logging::Level lvl = Acts::Logging::INFO)
        : logger(Acts::getDefaultLogger(lname, lvl)), name(lname) {}
  };

  RootMaterialTrackReader(const Config& cfg);

  ~RootMaterialTrackReader();

  std::string name() const final override;

  std::pair<size_t, size_t> availableEvents() const final override;

  ProcessCode read(
      const ActsExamples::AlgorithmContext& context) final override;

 private:
  const Acts::Logger& logger() const { return *m_cfg.logger; }

  Config m_cfg;

  std::mutex m_read_mutex;

  size_t m_events = 0;

  TChain* m_inputChain = nullptr;

  float m_v_x;    
  float m_v_y;    
  float m_v_z;    
  float m_v_px;   
  float m_v_py;   
  float m_v_pz;   
  float m_v_phi;  
  float m_v_eta;  
  float m_tX0;    
  float m_tL0;    

  std::vector<float>* m_step_x = new std::vector<float>;   
  std::vector<float>* m_step_y = new std::vector<float>;   
  std::vector<float>* m_step_z = new std::vector<float>;   
  std::vector<float>* m_step_dx = new std::vector<float>;  
  std::vector<float>* m_step_dy = new std::vector<float>;  
  std::vector<float>* m_step_dz = new std::vector<float>;  
  std::vector<float>* m_step_length = new std::vector<float>;  
  std::vector<float>* m_step_X0 = new std::vector<float>;  
  std::vector<float>* m_step_L0 = new std::vector<float>;  
  std::vector<float>* m_step_A = new std::vector<float>;   
  std::vector<float>* m_step_Z = new std::vector<float>;   
  std::vector<float>* m_step_rho =
      new std::vector<float>;  
};

}  // namespace ActsExamples