InDetAlignFillTrack.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETALIGNGENTOOLS_INDETALIGNFILLTRACK_H
#define INDETALIGNGENTOOLS_INDETALIGNFILLTRACK_H
// ================================================
// InDetAlignFillTrack
// ================================================
//
// InDetAlignFillTrack.h
// Headerfile for InDetAlignFillTrack
//
// Carlos Escobar, started 27/12/2007
//
// AthAlgTool to create store Trk::Track and MCTrack information in a ntuple
 
#include <string>
 
#include "TrkTrack/TrackCollection.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/NTuple.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "InDetAlignGenTools/IInDetAlignFillTrack.h"
 
 
namespace Trk {
  class Track;
  class ITruthToTrack; 
  class IExtrapolator;
  class ITrackParticleCreatorTool;
}
 
class INTupleSvc;
 
 
class InDetAlignFillTrack: virtual public IInDetAlignFillTrack, public AthAlgTool {
 public:
  InDetAlignFillTrack(const std::string& type, const std::string& name,
              const IInterface* parent);
  virtual ~InDetAlignFillTrack();
  
  virtual StatusCode initialize();
  virtual StatusCode finalize();
  
  virtual StatusCode FillTrack();
 
  virtual int GetTrks() const { return m_totaltrks; }
  virtual int GetTrkHits() const { return m_totalhits; }
  virtual int GetTrkPixHits() const { return m_totalPixhits; }
  virtual int GetTrkSCTHits() const { return m_totalSCThits; }
  virtual int GetTrkTRTHits() const { return m_totalTRThits; }
 
 private:
  INTupleSvc*                           m_ntupleSvc; 
 
  // member functions
  void bookNtuple();
  void bookUpNtuple();
  void bookLowNtuple();
  void bookMatchingNtuple();
  int dumpTrackCol(const TrackCollection*);
  int dumpTrackCol(const TrackCollection*, const std::string&);
  void dumpTrack(int, const Trk::Track*, const std::string&);  
  void dumpPerigee(const Trk::TrackParameters*, int);
  StatusCode dumpMatching(const TrackCollection*,const TrackCollection*);
 
  // variables
  bool                                   m_doMatching;    
  bool                                   m_doTruth;       
 
  std::string                            m_inputCol;
  std::string                            m_inputUpCol;
  std::string                            m_inputLowCol;
  std::string                            m_TruthTrkCol;
  std::string                            m_ntupleName;    
 
  int m_totaltrks;
  int m_totalhits;
  int m_totalPixhits;
  int m_totalSCThits;
  int m_totalTRThits;
  int m_totalUptrks;
  int m_totalUphits;
  int m_totalUpPixhits;
  int m_totalUpSCThits;
  int m_totalUpTRThits;
  int m_totalLowtrks;
  int m_totalLowhits;
  int m_totalLowPixhits;
  int m_totalLowSCThits;
  int m_totalLowTRThits;
  int m_events;
 
  float m_matchedRcut;
  float m_mindR;
 
  ToolHandle < Trk::ITruthToTrack >      m_truthToTrack;  
  ToolHandle < Trk::IExtrapolator >      m_extrapolator;  
  ToolHandle <Trk::ITrackParticleCreatorTool> m_particleCreator; 
 
  NTuple::Item<long> m_nt_ntracks;            
  NTuple::Item<long> m_nt_nUptracks;          
  NTuple::Item<long> m_nt_nLowtracks;         
  NTuple::Item<long> m_nt_nmctracks;          
  NTuple::Item<long> m_nt_matchingTrk;        
  // ----------------------------------------------------------------------
  // Trk::Track parameters
  NTuple::Array<float> m_nt_Trk_d0;      
  NTuple::Array<float> m_nt_Trk_z0;      
  NTuple::Array<float> m_nt_Trk_phi0;    
  NTuple::Array<float> m_nt_Trk_theta0;  
  NTuple::Array<float> m_nt_Trk_qoverp;  
  NTuple::Array<float> m_nt_Trk_pt;      
  // ----------------------------------------------------------------------
 
  // ----------------------------------------------------------------------
  NTuple::Array<long> m_nt_Trk_nHits;      
  NTuple::Array<long> m_nt_Trk_nhitspix;   
  NTuple::Array<long> m_nt_Trk_nhitssct;   
  NTuple::Array<long> m_nt_Trk_nhitstrt;   
 
  NTuple::Array<long> m_nt_Trk_nsharedPixels; 
  NTuple::Array<long> m_nt_Trk_nsharedSCT;    
  NTuple::Array<long> m_nt_Trk_nshared;       
 
  NTuple::Array<long> m_nt_Trk_nholesPixels;  
  NTuple::Array<long> m_nt_Trk_nholesSCT;     
  NTuple::Array<long> m_nt_Trk_nholes;        
 
  NTuple::Array<float> m_nt_Trk_chi2;      
  NTuple::Array<long> m_nt_Trk_ndof;       
  NTuple::Array<float> m_nt_Trk_chi2Prob;  
 // ----------------------------------------------------------------------
 
  // ----------------------------------------------------------------------
  // Trk::Track parameters
  NTuple::Array<float> m_nt_Trk_d0_Up;      
  NTuple::Array<float> m_nt_Trk_z0_Up;      
  NTuple::Array<float> m_nt_Trk_phi0_Up;    
  NTuple::Array<float> m_nt_Trk_theta0_Up;  
  NTuple::Array<float> m_nt_Trk_qoverp_Up;  
  NTuple::Array<float> m_nt_Trk_pt_Up;      
  // ----------------------------------------------------------------------
 
  // ----------------------------------------------------------------------
  NTuple::Array<long> m_nt_Trk_nHits_Up;      
  NTuple::Array<long> m_nt_Trk_nhitspix_Up;   
  NTuple::Array<long> m_nt_Trk_nhitssct_Up;   
  NTuple::Array<long> m_nt_Trk_nhitstrt_Up;   
 
  NTuple::Array<long> m_nt_Trk_nsharedPixels_Up; 
  NTuple::Array<long> m_nt_Trk_nsharedSCT_Up;    
  NTuple::Array<long> m_nt_Trk_nshared_Up;       
 
  NTuple::Array<long> m_nt_Trk_nholesPixels_Up;  
  NTuple::Array<long> m_nt_Trk_nholesSCT_Up;     
  NTuple::Array<long> m_nt_Trk_nholes_Up;        
 
  NTuple::Array<float> m_nt_Trk_chi2_Up;      
  NTuple::Array<long> m_nt_Trk_ndof_Up;       
  NTuple::Array<float> m_nt_Trk_chi2Prob_Up;  
 // ----------------------------------------------------------------------
 
  // ----------------------------------------------------------------------
  // Trk::Track parameters
  NTuple::Array<float> m_nt_Trk_d0_Low;      
  NTuple::Array<float> m_nt_Trk_z0_Low;      
  NTuple::Array<float> m_nt_Trk_phi0_Low;    
  NTuple::Array<float> m_nt_Trk_theta0_Low;  
  NTuple::Array<float> m_nt_Trk_qoverp_Low;  
  NTuple::Array<float> m_nt_Trk_pt_Low;      
  // ----------------------------------------------------------------------
 
  // ----------------------------------------------------------------------
  NTuple::Array<long> m_nt_Trk_nHits_Low;      
  NTuple::Array<long> m_nt_Trk_nhitspix_Low;   
  NTuple::Array<long> m_nt_Trk_nhitssct_Low;   
  NTuple::Array<long> m_nt_Trk_nhitstrt_Low;   
 
  NTuple::Array<long> m_nt_Trk_nsharedPixels_Low; 
  NTuple::Array<long> m_nt_Trk_nsharedSCT_Low;    
  NTuple::Array<long> m_nt_Trk_nshared_Low;       
 
  NTuple::Array<long> m_nt_Trk_nholesPixels_Low;  
  NTuple::Array<long> m_nt_Trk_nholesSCT_Low;     
  NTuple::Array<long> m_nt_Trk_nholes_Low;        
  
  NTuple::Array<float> m_nt_Trk_chi2_Low;      
  NTuple::Array<long> m_nt_Trk_ndof_Low;       
  NTuple::Array<float> m_nt_Trk_chi2Prob_Low;  
  // ----------------------------------------------------------------------
  
  // ----------------------------------------------------------------------
  NTuple::Array<float> m_nt_mc_trkistruth;    
  
  // generated particle parameters
  NTuple::Array<float> m_nt_mc_Trk_genParticlePt;  
  NTuple::Array<float> m_nt_mc_Trk_genParticleEta; 
  NTuple::Array<float> m_nt_mc_Trk_genParticlePhi; 
  
  // MonteCarlo Track parameters
  NTuple::Array<float> m_nt_mc_Trk_d0;        
  NTuple::Array<float> m_nt_mc_Trk_z0;        
  NTuple::Array<float> m_nt_mc_Trk_phi0;      
  NTuple::Array<float> m_nt_mc_Trk_theta0;    
  NTuple::Array<float> m_nt_mc_Trk_eta;       
  NTuple::Array<float> m_nt_mc_Trk_qoverp;    
  NTuple::Array<float> m_nt_mc_Trk_qoverpt;   
    
  NTuple::Array<float> m_nt_mc_Trk_pt;        
  NTuple::Array<float> m_nt_mc_Trk_charge;    
  NTuple::Array<float> m_nt_mc_Trk_prob;      
  NTuple::Array<float> m_nt_mc_Trk_pdg;       
  
  NTuple::Array<float> m_nt_mc_Trk_vtxX;      
  NTuple::Array<float> m_nt_mc_Trk_vtxY;      
  NTuple::Array<float> m_nt_mc_Trk_vtxZ;      
  // ----------------------------------------------------------------------
 
 
  // ----------------------------------------------------------------------
  // Matching Trk::Track_Up and Trk::Track_Low
  NTuple::Array<float> m_nt_Trk_delta_d0;      
  NTuple::Array<float> m_nt_Trk_delta_z0;      
  NTuple::Array<float> m_nt_Trk_delta_phi0;    
  //**
  NTuple::Array<float> m_nt_Trk_delta_theta0;  
  NTuple::Array<float> m_nt_Trk_delta_eta;     
  NTuple::Array<float> m_nt_Trk_delta_qoverpt; 
  NTuple::Array<float> m_nt_Trk_delta_pt;      
  NTuple::Array<float> m_nt_Trk_delta_charge;  
  // ----------------------------------------------------------------------
 
};
 
#endif // INDETALIGNGENTOOLS_INDETALIGNFILLTRACK_H