PhysicsAnalysis/JetTagging/JetTagTools/JetTagTools/TrackSelector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETTAGTOOLS_TRACKSELECTOR_H
#define JETTAGTOOLS_TRACKSELECTOR_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "GeoPrimitives/GeoPrimitives.h"
#include "InDetRecToolInterfaces/IInDetEtaDependentCutsSvc.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "ITrackToVertex/ITrackToVertex.h"
 
#include <string>
#include <bitset>
 
namespace Reco { class ITrackToVertex; }
 
namespace Analysis {
 
class TrackSelector : public AthAlgTool {
 
public:
  TrackSelector (const std::string& type,
            const std::string& name, const IInterface* parent);
  virtual ~TrackSelector();
 
  static const InterfaceID& interfaceID();
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
 
  //  inline void setSumTrkPt(double sum) { m_SumTrkPt = sum; }
 
  bool selectTrack(const Amg::Vector3D& pv,
                   const xAOD::TrackParticle* track,
                   double refPt = 0) const;
 
  bool selectTrack(const Amg::Vector3D& pv,
                   const xAOD::TrackParticle* track,
                   std::bitset<17>& failedCuts,
                   double refPt = 0) const;
 
  inline const std::bitset<17> currentTrackpassedCuts() const { return m_passedCuts; }
 
private:
  bool m_usePerigeeParameters; 
  double m_pTMin;     
  bool m_usepTDepTrackSel;
  double m_pTMinOffset;
  double m_pTMinSlope;
  double m_d0Max;     
  double m_z0Max;     
  double m_sigd0Max; // max sig d0:
  double m_sigz0Max; // max sig z0:
  double m_etaMax;    
  bool m_useTrackSummaryInfo; 
  bool m_useBLayerHitPrediction; // use b-layer tool tied to conddb
  int m_nHitBLayer;   
  int m_nHitPix;      
  int m_nHitSct;      
  int m_nHitSi;       
  int m_nHitTrt;      
  int m_nHitTrtHighE; 
  bool m_useDeadPixInfo;  
  bool m_useDeadSctInfo;  
  bool m_useTrackQualityInfo; 
  double m_fitChi2;   
  double m_fitProb;   
  double m_fitChi2OnNdfMax; //max. fitchi2/ndf
  std::string m_inputTrackCollection; 
  std::string m_outputTrackCollection; 
  std::string m_sharedHitMapLocation; 
  bool m_useAntiPileUpCuts; 
  double m_antiPileUpSigD0Cut;
  double m_antiPileUpSigZ0Cut;
  int m_antiPileUpNHitSiCut;
  int m_antiPileUpNHolePixCut;
  bool m_useTrackingTightDefinition;
  // for debugging:
  enum m_Cuts { pTMin, d0Max, z0Max, sigd0Max, sigz0Max, etaMax,
                nHitBLayer, deadBLayer, nHitPix, nHitSct, nHitSi, nHitTrt, nHitTrtHighE,
                fitChi2, fitProb,fitChi2OnNdfMax, trackingTightDef,
                numCuts };
 
  int m_ntri;
  int m_ntrf;
  mutable std::atomic<int> m_ntrc[numCuts];
 
  ToolHandle< Reco::ITrackToVertex > m_trackToVertexTool{this, "trackToVertexTool", "Reco::TrackToVertex"};
 
  std::bitset<numCuts> m_passedCuts;
 
  bool m_useEtaDependentCuts;
  ServiceHandle<InDet::IInDetEtaDependentCutsSvc> m_etaDependentCutsSvc{this, "InDetEtaDependentCutsSvc", ""};
 
};
 
 
}
#endif // JETTAGTOOLS_TRACKSELECTOR_H