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 filePath = "";                   
    std::string fileMode = "RECREATE";           
    std::string treeName = "propagation_steps";  
    TFile* rootFile = 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_writeMutex;  
  TFile* m_outputFile;      
  TTree* m_outputTree;      
  int m_eventNr;            
 
  //  std::vector<int> m_volumeID;     ///< volume identifier
  std::vector<int> m_boundaryID;   
  std::vector<int> m_layerID;      
  std::vector<int> m_approachID;   
  std::vector<int> m_sensitiveID;  
  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