ActsMaterialTrackWriterSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSGEOMETRY_ACTSMATERIALTRACKWRITERSVC_H
#define ACTSGEOMETRY_ACTSMATERIALTRACKWRITERSVC_H
 
#include "ActsGeometryInterfaces/IActsMaterialTrackWriterSvc.h"
#include "ActsGeometryInterfaces/IActsTrackingGeometrySvc.h"
 
#include "AthenaBaseComps/AthService.h"
#include "GaudiKernel/IInterface.h"
#include "Gaudi/Property.h"  /*no forward decl: typedef*/
 
#include <vector>
#include <deque>
#include <mutex>
#include <thread>
#include <atomic>
 
#include "TTree.h"
#include "TFile.h"
 
class ActsMaterialTrackWriterSvc : public extends<AthService, IActsMaterialTrackWriterSvc> {
public:
 
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
 
  ActsMaterialTrackWriterSvc( const std::string& name, ISvcLocator* svc );
 
  virtual void
  write(const Acts::RecordedMaterialTrack& mTrack) override;
 
private:
 
  std::deque<Acts::RecordedMaterialTrack> m_mTracks;
  std::mutex m_writeMutex;
  std::thread m_writeThread;
  std::atomic<bool> m_doEnd;
  TFile* p_tFile{};
  TTree* p_tree{};
 
  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_sx;      
  std::vector<float> m_step_sy;      
  std::vector<float> m_step_sz;      
  std::vector<float> m_step_x;       
  std::vector<float> m_step_y;       
  std::vector<float> m_step_z;       
  std::vector<float> m_step_ex;      
  std::vector<float> m_step_ey;      
  std::vector<float> m_step_ez;      
  std::vector<float> m_step_dx;      
  std::vector<float> m_step_dy;      
  std::vector<float> m_step_dz;      
  std::vector<float> m_step_length;  
  std::vector<float> m_step_X0;      
  std::vector<float> m_step_L0;      
  std::vector<float> m_step_A;       
  std::vector<float> m_step_Z;       
  std::vector<float> m_step_rho;     
 
  std::vector<std::uint64_t>
      m_sur_id;  
  std::vector<int32_t>
      m_sur_type;              
  std::vector<float> m_sur_x;  
  std::vector<float> m_sur_y;  
  std::vector<float> m_sur_z;  
 
  std::vector<float>
      m_sur_range_min;  
  std::vector<float>
      m_sur_range_max;  
 
  std::vector<std::uint64_t>
      m_vol_id;  
 
  void writerThread();
  void doWrite(const Acts::RecordedMaterialTrack &mTrack);
 
  ServiceHandle<IActsTrackingGeometrySvc> m_trackingGeometrySvc;
 
  // jobOptions properties
  Gaudi::Property<std::string> m_filePath{this, "FilePath", "MaterialTracks.root", "Output root file for charged particle"};
  Gaudi::Property<std::string> m_treeName{this, "TreeName", "material-tracks", ""};
  Gaudi::Property<bool> m_storeSurface{this, "StoreSurface", true, "Store the surface info in the root file"};
  Gaudi::Property<bool> m_storeVolume{this, "StoreVolume", true, "Store the volume info in the root file"};
  Gaudi::Property<size_t> m_maxQueueSize{this, "MaxQueueSize", 5000, "Limit the write queue to this size"};
 
};
 
 
#endif