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 "Acts/Material/MaterialInteraction.hpp"
#include "ActsExamples/Framework/DataHandle.hpp"
#include "ActsExamples/Framework/IReader.hpp"
#include "ActsExamples/Framework/ProcessCode.hpp"
#include <Acts/Definitions/Algebra.hpp>
#include <Acts/Propagator/MaterialInteractor.hpp>
#include <Acts/Utilities/Logger.hpp>

#include <cstddef>
#include <cstdint>
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
#include <utility>
#include <vector>

class TChain;

namespace ActsExamples {
struct AlgorithmContext;

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

    bool orderedEvents = true;
    // Read surface information for the root file
    bool readCachedSurfaceInformation = false;
  };

  RootMaterialTrackReader(const Config& config, Acts::Logging::Level level);

  ~RootMaterialTrackReader() override;

  std::string name() const override;

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

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

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

 private:
  std::unique_ptr<const Acts::Logger> m_logger;

  const Acts::Logger& logger() const { return *m_logger; }

  Config m_cfg;

  WriteDataHandle<std::unordered_map<size_t, Acts::RecordedMaterialTrack>>
      m_outputMaterialTracks{this, "OutputMaterialTracks"};

  std::mutex m_read_mutex;

  size_t m_events = 0;

  size_t m_batchSize = 0;

  TChain* m_inputChain = nullptr;

  uint32_t m_eventId = 0;

  std::vector<long long> m_entryNumbers = {};

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

  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>;  

  std::vector<std::uint64_t>* m_sur_id =
      new std::vector<std::uint64_t>;  

  std::vector<float>* m_sur_x =
      new std::vector<float>;  

  std::vector<float>* m_sur_y =
      new std::vector<float>;  

  std::vector<float>* m_sur_z =
      new std::vector<float>;  

  std::vector<float>* m_sur_pathCorrection =
      new std::vector<float>;  

};

}  // namespace ActsExamples