InDetAlignFillTrack.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 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"
#include "TrkToolInterfaces/ITruthToTrack.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkToolInterfaces/ITrackParticleCreatorTool.h"
 
namespace Trk {
  class Track;
}
 
class INTupleSvc;
 
 
class InDetAlignFillTrack: public extends<AthAlgTool, IInDetAlignFillTrack> {
 public:
  InDetAlignFillTrack(const std::string& type, const std::string& name,
              const IInterface* parent);
  
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
  
  virtual StatusCode FillTrack() override;
 
  virtual int GetTrks() const override { return m_totaltrks; }
  virtual int GetTrkHits() const override { return m_totalhits; }
  virtual int GetTrkPixHits() const override { return m_totalPixhits; }
  virtual int GetTrkSCTHits() const override { return m_totalSCThits; }
  virtual int GetTrkTRTHits() const override { return m_totalTRThits; }
 
 private:
  ServiceHandle<INTupleSvc> m_ntupleSvc{this,"NTupleSvc","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*);
 
  // properties
 
  // Truth information
  BooleanProperty m_doTruth{this,"doTruth",false};       
  StringProperty m_TruthTrkCol{this,"TruthTrkCol","TrackTruthCollection"};
 
  StringProperty m_inputCol{this,"InputTrkCol","Tracks"};
  StringProperty m_inputUpCol{this,"InputUpTrkCol",""};
  StringProperty m_inputLowCol{this,"InputLowTrkCol",""};
 
  // cosmic segments matching
  BooleanProperty m_doMatching{this,"doMatching",true};  
  FloatProperty m_matchedRcut{this,"dRCut",100.};
  FloatProperty m_mindR{this,"minimumdR",10000.};
 
  // Ntuple
  StringProperty m_ntupleName{this,"NtupleName","/NTUPLES/GLOBFILE"};
 
  // variables
  int m_totaltrks{0};
  int m_totalhits{0};
  int m_totalPixhits{0};
  int m_totalSCThits{0};
  int m_totalTRThits{0};
  int m_totalUptrks{0};
  int m_totalUphits{0};
  int m_totalUpPixhits{0};
  int m_totalUpSCThits{0};
  int m_totalUpTRThits{0};
  int m_totalLowtrks{0};
  int m_totalLowhits{0};
  int m_totalLowPixhits{0};
  int m_totalLowSCThits{0};
  int m_totalLowTRThits{0};
  int m_events{0};
 
  // tool handles
  ToolHandle < Trk::ITruthToTrack > m_truthToTrack{this
    , "TruthToTrackTool"
    , "Trk::TruthToTrack"
    , "tool to produce perigee track parameters from generated parameters"};
 
  ToolHandle < Trk::IExtrapolator > m_extrapolator{this
    , "ExtrapolationTool"
    , "Trk::Extrapolator/CosmicsExtrapolator"
    , "tool to extrapolate tracks"};
 
  ToolHandle <Trk::ITrackParticleCreatorTool> m_particleCreator{this
    , "TrackParticleCreatorTool"
    , "Trk::TrackParticleCreatorTool/TrackParticleCreatorTool"
    , "tool to build TrackParticle"};
 
  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