HITrackQualityAugmentationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef DERIVATIONFRAMEWORK_HITrackQualityAugmentationTool_H
#define DERIVATIONFRAMEWORK_HITrackQualityAugmentationTool_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "InDetTrackSelectionTool/IInDetTrackSelectionTool.h"
#include "StoreGate/ReadHandle.h"
#include "StoreGate/WriteDecorHandle.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODEventInfo/EventInfo.h"
#include "TrkToLeptonPVTool/ITrkToLeptonPV.h"
 
#include <string>
 
namespace DerivationFramework {
 
  class HITrackQualityAugmentationTool : public extends<AthAlgTool, IAugmentationTool> {
    public:
      enum{
        PP_MIN_BIAS=1<<1, //2
        
        HI_LOOSE=1<<2, // 4
        HI_LOOSE_7SCT_HITS    =1<<5, // 32
        HI_LOOSE_TIGHT_D0_Z0  =1<<7,  // 128 
        HI_LOOSE_TIGHTER_D0_Z0=1<<8,  // 256
        
        HI_TIGHT=1<<3,  //8
        HI_TIGHT_TIGHTER_D0_Z0=1<<4, //16
        HI_TIGHT_LOOSE_D0_Z0  =1<<6, //64
      };
 
    using base_class::base_class;
 
    virtual StatusCode initialize() override final;
    virtual StatusCode addBranches(const EventContext& ctx) const override final;
 
    private:
      unsigned short GetTrackQuality   (const xAOD::TrackParticle* track,float z_vtx           ) const;
      unsigned short GetTrackQualityNew(const xAOD::TrackParticle* track,const xAOD::Vertex* pv) const;
 
      SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trackParticlesName{this, "TrackParticlesName", "InDetTrackParticles", ""};
      SG::ReadHandleKey<xAOD::VertexContainer> m_vertexContainerName {this, "VertexContainerName", "PrimaryVertices", ""};
      SG::ReadHandleKey<xAOD::EventInfo> m_eventInfoKey{this, "EventInfoKey", "EventInfo", ""};
 
      SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_decorator{ this, "TrackQuality", "TrackQuality", ""};
      SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_chi2Decorator{ this, "TrackChi2ToPV", "Chi2ToPV", ""};
      SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_vertexIndexDecorator{ this, "VertexIndex", "VertexIndex", ""};
      
      SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_covD0Decorator{ this, "CovD0", "CovD0", ""};
      SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_covZ0Decorator{ this, "CovZ0", "CovZ0", ""};
      SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_covThetaDecorator{ this, "CovTheta", "CovTheta", ""};
 
      ToolHandle<InDet::IInDetTrackSelectionTool> m_trkSelTool_pp {this, "TrackSelectionTool_pp", "", ""};
      ToolHandle<InDet::IInDetTrackSelectionTool> m_trkSelTool_hi_loose {this, "TrackSelectionTool_hi_loose", "", ""};
      ToolHandle<InDet::IInDetTrackSelectionTool> m_trkSelTool_hi_tight {this, "TrackSelectionTool_hi_tight", "", ""};
      
      ToolHandle<ITrkToLeptonPV> m_trkToLeptonPVTool {this, "TrkToLeptonPVTool", "", "Tool for matching tracks to PV"};
  };
}
 
 
#endif // DERIVATIONFRAMEWORK_HITrackQualityAugmentationTool_H