RootExCellWriter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
// This file was largely imported from the Acts testing framework
 
#pragma once
 
#include <TFile.h>
#include <TTree.h>
#include <mutex>
#include "Acts/Extrapolation/ExtrapolationCell.hpp"
#include "Acts/Utilities/Logger.hpp"
 
class TFile;
 
#ifndef MAXSTEPS
#define MAXSTEPS 100
#endif
 
  template <class T>
  class RootExCellWriter
  {
  public:
 
    struct ExtrapolationStep
    {
      float x{}, y{}, z{};     
      float px{}, py{}, pz{};  
      float type{};        
    };
 
    // @struct Config
    //
    // The nested config class
    struct Config
    {
    public:
      std::string filePath;               
      std::string fileMode = "RECREATE";  
      std::string treeName
          = "extrapolation_cells";  
      bool writeSensitive{};
      bool writeMaterial{};
      bool writePassive{};
      bool writeBoundary{};
    };
 
    RootExCellWriter(const Config&        cfg
                     //Acts::Logging::Level level = Acts::Logging::INFO
                     );
    
    RootExCellWriter() = delete;                 
    
    RootExCellWriter(const & RootExCellWriter) = delete;
    
    RootExCellWriter & operator =(const RootExCellWriter & )= delete;
 
    ~RootExCellWriter();
 
    void
    endRun();
    
    void
    write(
        const Acts::ExtrapolationCell<T>& ecell, int eventNum);
 
  protected:
 
    Config             m_cfg;           
    std::mutex         m_writeMutex;    
    TFile*             m_outputFile{};    
    TTree*             m_outputTree{};    
    float              m_eta{};           
    float              m_phi{};           
    float              m_materialX0{};    
    float              m_materialL0{};    
    int             m_eventNum{};
    std::vector<float> m_s_positionX;   
    std::vector<float> m_s_positionY;   
    std::vector<float> m_s_positionZ;   
    std::vector<float> m_s_positionR;   
    std::vector<float> m_s_materialX0;  
    std::vector<float> m_s_materialL0;  
    std::vector<int>   m_s_material;    
    std::vector<int>   m_s_boundary;    
    std::vector<int>   m_s_sensitive;   
    std::vector<int>   m_s_volumeID;    
    std::vector<int>   m_s_layerID;     
    std::vector<int>   m_s_surfaceID;   
    std::vector<float>
        m_s_localposition0;  
    std::vector<float>
        m_s_localposition1;  
    int m_hits{};              
  };
 
 
#include "RootExCellWriter.icc"