SimpleIDNtupleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETALIGNNTUPLETOOLS_SIMPLEIDNTUPLETOOL_H
#define INDETALIGNNTUPLETOOLS_SIMPLEIDNTUPLETOOL_H
 
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "TrkAlignInterfaces/IFillNtupleTool.h"
 
class TFile;
class TTree;
 
namespace Trk
{
    class AlignTrack;
    class ITrackParticleCreatorTool;
    class IAlignModuleTool;
}
 
namespace InDetDD
{
    class TRT_DetectorManager;
}
 
class TRT_ID;
 
namespace InDet
{
    class SimpleIDNtupleTool : virtual public Trk::IFillNtupleTool, public AthAlgTool
    {
 
        public:
            SimpleIDNtupleTool(const std::string & type, const std::string & name, const IInterface * parent);
            virtual ~SimpleIDNtupleTool();
 
            StatusCode initialize();
            StatusCode finalize();
 
            void dumpTrack(int itrk, const Trk::AlignTrack * alignTrack);
 
            void fillHits(const Trk::AlignTrack * alignTrack);
 
            void storeHitmap();
            void fillHitmap();
            void fillSummary();
            void showStatistics();
 
            void setNtuple(TFile* ntuple) { m_file=ntuple; }
            StatusCode fillNtuple();
 
        private:
 
            void initializeNtuple();
 
            ToolHandle <Trk::ITrackParticleCreatorTool> m_particleCreator; 
            ToolHandle <Trk::IAlignModuleTool>          m_alignModuleTool; 
 
            const TRT_ID * m_idHelper;
 
            TFile * m_file;
            TTree * m_tree;
 
            //std::string m_filename; //!< name of ntuple file
            //std::string m_filepath;  //!< path to ntuple file
 
            // Reconstructed variables
            int      m_runNumber;
            int      m_evtNumber;
 
            // Track information
 
            double m_xvtx;
 
            double m_yvtx;
 
            double m_zvtx;
 
            double m_d0;
 
            double m_z0;
 
            double m_phi0;
 
            double m_theta;
 
            double m_qoverp;
 
            double m_pt;
 
            double m_eta;
 
            double m_chi2;
 
            int m_ndof;
 
            double m_chi2prob;
 
            // Hit information
 
            int m_nhits;
 
            int m_nhitstrt;
 
            int m_nhitssct;
 
            int m_nhitspix;
 
            int m_nshared;
 
            int m_nshsct;
 
            int m_nshpix;
 
            int m_nholes;
 
            int m_nhsct;
 
            int m_nhpix;
 
            double * m_hit_rho;
 
            double * m_hit_phi;
 
            double * m_hit_z;
 
            double * m_hit_cotth;
 
            double * m_hit_eta;
 
            double * m_hit_xloc;
 
            double * m_hit_yloc;
 
            double * m_hit_zloc;
 
            double * m_hit_drd;
 
            double * m_hit_sig;
 
            double * m_hit_resx;     
            double * m_hit_resx_err; 
            double * m_hit_resy;     
            double * m_hit_resy_err; 
 
            double * m_hit_tpar_x;   
            double * m_hit_tpar_y;   
 
            int * m_hit_det;         
            int * m_hit_type;        
 
            double * m_hit_derivx_transx;  
            double * m_hit_derivx_transy;  
            double * m_hit_derivx_transz;  
            double * m_hit_derivx_rotx;    
            double * m_hit_derivx_roty;    
            double * m_hit_derivx_rotz;    
            double * m_hit_derivx_bowx;    
            double * m_hit_derivx_bowy;    
            double * m_hit_derivx_bowz;    
            double * m_hit_derivy_transx;  
            double * m_hit_derivy_transy;  
            double * m_hit_derivy_transz;  
            double * m_hit_derivy_rotx;    
            double * m_hit_derivy_roty;    
            double * m_hit_derivy_rotz;    
            double * m_hit_derivy_bowx;    
            double * m_hit_derivy_bowy;    
            double * m_hit_derivy_bowz;    
 
            int m_max_hits; 
 
            bool m_storeDerivatives;
    }; // end class
  
} // end namespace
 
#endif // INDETALIGNNTUPLETOOLS_SIMPLEIDNTUPLETOOL_H