PropagatorStepWriter.h

#ifndef TRACKING_SIM_PROPAGATORSTEPWRITER_H
#define TRACKING_SIM_PROPAGATORSTEPWRITER_H
 
/* This class is a strip down version of the ActsExample::PropagatorStepWriter
 * It's used to dump in a root file all the steps information of the
 * Acts::Propagator for a complete validation of a tracking geometry.
 */
 
#include <mutex>
 
//--- Framework ---//
#include "Framework/Event.h"
 
//--- ACTS ---//
#include "Acts/Propagator/detail/SteppingLogger.hpp"
 
//--- Tracking ---//
#include "Tracking/Event/Measurement.h"
 
//--- ROOT ---//
#include "TFile.h"
#include "TTree.h"
 
using PropagationSteps = std::vector<Acts::detail::Step>;
 
namespace tracking {
namespace sim {
 
class PropagatorStepWriter {
 public:
  struct Config {
    // std::string collection =
    //     "propagation_steps";            ///< particle collection to write
 
    std::string file_path_ = "";                   
    std::string file_mode_ = "RECREATE";           
    std::string tree_name_ = "propagation_steps";  
    TFile* root_file_ = nullptr;                   
  };
 
  PropagatorStepWriter(const Config& cfg);
 
  ~PropagatorStepWriter();
 
  bool writeSteps(framework::Event& event,
                  const std::vector<PropagationSteps>& stepCollection,
                  const std::vector<ldmx::Measurement>& measurements,
                  const Acts::Vector3& start_pos,
                  const Acts::Vector3& start_mom);
 
 protected:
  Config m_cfg_;              
  std::mutex m_write_mutex_;  
  TFile* m_output_file_;      
  TTree* m_output_tree_;      
  int m_event_nr_;            
 
  //  std::vector<int> m_volumeID;     ///< volume identifier
  std::vector<int> m_boundary_id_;   
  std::vector<int> m_layer_id_;      
  std::vector<int> m_approach_id_;   
  std::vector<int> m_sensitive_id_;  
  std::vector<float> m_x_;           
  std::vector<float> m_y_;           
  std::vector<float> m_z_;           
  std::vector<float> m_dx_;          
  std::vector<float> m_dy_;          
  std::vector<float> m_dz_;          
  std::vector<int> m_step_type_;     
  std::vector<float> m_step_acc_;    
  std::vector<float> m_step_act_;    
  std::vector<float> m_step_abt_;    
  std::vector<float> m_step_usr_;    
  std::vector<float> m_hit_x_;       
  std::vector<float> m_hit_y_;       
  std::vector<float> m_hit_z_;       
  std::vector<float>
      m_start_pos_;  
  std::vector<float>
      m_start_mom_;  
};
}  // namespace sim
}  // namespace tracking
 
#endif