SimpleIDNtupleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 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;
class TRT_ID;
 
namespace Trk
{
  class AlignTrack;
  class ITrackParticleCreatorTool;
  class IAlignModuleTool;
}
 
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();
 
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;

    virtual void dumpTrack(int itrk, const Trk::AlignTrack * alignTrack) override;

    void fillHits(const Trk::AlignTrack * alignTrack);
 
    virtual void storeHitmap() override;
    virtual void fillHitmap() override;
    virtual void fillSummary() override;
    virtual void showStatistics() override;
 
    virtual void setNtuple(TFile* ntuple)  override { m_file=ntuple; }
    virtual StatusCode fillNtuple() override;
 
  private:
    void initializeNtuple();
 
    ToolHandle <Trk::ITrackParticleCreatorTool> m_particleCreator; 
    ToolHandle <Trk::IAlignModuleTool>          m_alignModuleTool; 
 
    const TRT_ID * m_idHelper;
 
    TFile * m_file;
    TTree * m_tree;
 
    // 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