HITrackQualityAugmentationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
 
#ifndef DERIVATIONFRAMEWORK_HITrackQualityAugmentationTool_H
#define DERIVATIONFRAMEWORK_HITrackQualityAugmentationTool_H
 
#include <string>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "xAODTracking/TrackParticle.h"
#include "InDetTrackSelectionTool/IInDetTrackSelectionTool.h"
 
 
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
      
   };
      HITrackQualityAugmentationTool(const std::string& t, const std::string& n, const IInterface* p);
      virtual StatusCode addBranches() const;
    private:
      unsigned short GetTrackQuality   (const xAOD::TrackParticle* track,float z_vtx           ) const;
      unsigned short GetTrackQualityNew(const xAOD::TrackParticle* track,const xAOD::Vertex* pv) const;
      ToolHandle<InDet::IInDetTrackSelectionTool> m_trkSelTool_pp      ;
      ToolHandle<InDet::IInDetTrackSelectionTool> m_trkSelTool_hi_loose;
      ToolHandle<InDet::IInDetTrackSelectionTool> m_trkSelTool_hi_tight;
  };
}
 
#endif // DERIVATIONFRAMEWORK_HITrackQualityAugmentationTool_H