DetailedTrackGradeFactory.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETTAGTOOLS_DetailedTrackGradeFactory_H
#define JETTAGTOOLS_DetailedTrackGradeFactory_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "JetTagTools/ITrackGradeFactory.h"
#include <string>
#include <vector>
 
#include "GaudiKernel/ToolHandle.h"
#include "xAODTracking/TrackParticle.h"
 
#include "JetTagInfo/TrackGradesDefinition.h"
 
namespace Analysis
{
 
  class TrackGrade;
 
class DetailedTrackGradeFactory : public AthAlgTool, virtual public ITrackGradeFactory
{
 public:
 
  DetailedTrackGradeFactory(const std::string&,const std::string&,const IInterface*);
  virtual ~DetailedTrackGradeFactory() = default;
 
  virtual StatusCode initialize() override;
 
  virtual TrackGrade* getGrade(const xAOD::TrackParticle & track,
                       const xAOD::IParticle::FourMom_t & jetMomentum) const override;
 
  virtual const TrackGradesDefinition & getTrackGradesDefinition() const override;
 
private:
 
  TrackGradesDefinition m_trackGradesDefinition;
 
  bool m_hitBLayerGrade;   // grade for tracks without hit in First Layers
  bool m_useSharedHitInfo; 
 
  bool m_useRun2TrackGrading; // enabling run2 grading
  bool m_useITkTrackGrading; // enabling ITk grading
  bool m_useInnerLayers0HitInfo; //grading categories with no hits in one of the first 2 innermost pixel layers
  bool m_useDetailSharedHitInfo; //grading categories according to shared hit content
  bool m_useDetailSplitHitInfo;  //grading categories according to Split hit content
 
  int m_nSharedBLayer;
  int m_nSharedPix;   
  int m_nSharedSct;   
  int m_nSharedSi;    
  int m_nSharedInnermostPixelLayer;
  int m_nSharedNextToInnermostPixelLayer;
  int m_nOutlierPix;  
  int m_nSpoiltPix;   
 
  bool m_ptFracGrade;   // grade for tracks with ptFrac < m_ptFracCut
  double m_ptFracCut;    
 
  bool m_ptEtaGrades;    
  std::vector<double> m_ptLowerCuts;
  std::vector<double> m_etaLowerCuts;
 
  std::vector<std::vector<double>> m_gammaLowerCuts;
 
};
 
}
#endif