TrackCounting.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETTAGTOOLS_TRACKCOUNTING_H
#define JETTAGTOOLS_TRACKCOUNTING_H
 
/******************************************************
    @class TrackCounting
    
    b-tagging AlgTool based on counting tracks with an 
    impact parameter significence exceeding a given
    threshold
 
    @author Thomas Goepfert <thomas.goepfert@cern.ch>
********************************************************/
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "JetTagTools/ITagTool.h"
 
#include <vector>
 
namespace Reco { class ITrackToVertex; }
namespace Trk  { class VxCandidate; class ITrackToVertexIPEstimator; }
//class Jet;
 
namespace Analysis { 
 
  class TrackCountingInfo;
  class TrackSelector;
  class TrackGrade;
  class TrackGradePartition;
  class ITrackGradeFactory;
  class GradedTrack;
  class SVForIPTool;
          
  class TrackCounting : public AthAlgTool , virtual public ITagTool
  {
    public:
      TrackCounting(const std::string&,const std::string&,const IInterface*);
      
      virtual ~TrackCounting();
      StatusCode initialize();
      StatusCode finalize();
      void finalizeHistos(){};      
 
      void setOrigin(const Trk::VxCandidate* priVtx);
      
      void tagJet(xAOD::Jet& jetToTag);    
 
    private:      
 
      ToolHandle< Reco::ITrackToVertex > m_trackToVertexTool;
      
      ToolHandle< TrackSelector > m_trackSelectorTool;
     
      std::vector<GradedTrack> m_tracksInJet;
       
      // this pointer does not need to be deleted in the destructor (because it
      // points to something in storegate)
      const Trk::VxCandidate* m_priVtx = 0;
 
      std::vector<std::string> m_trackGradePartitionsDefinition;
      std::vector<TrackGradePartition*> m_trackGradePartitions;
 
      bool m_RejectBadTracks;
      bool m_SignWithSvx;
 
      bool m_flipIP;
 
      bool m_writeInfo;
 
      std::string m_truthMatchingName; // name of truthMatchingTool instance to get TruthInfo
      double m_purificationDeltaR;     // skip light jets with heavy flavor in this cone
 
      std::vector<std::string> m_jetCollectionList;
 
      //GP: use Tool for getting the secondary vertex information
      std::string m_secVxFinderNameForV0Removal;
      std::string m_secVxFinderNameForIPSign;
      ToolHandle< SVForIPTool > m_SVForIPTool;
      
      ToolHandle< ITrackGradeFactory > m_trackGradeFactory;
 
      ToolHandle< Trk::ITrackToVertexIPEstimator > m_trackToVertexIPEstimator;
      bool m_unbiasIPEstimation;
      
      template <class T> static bool m_greater(T a, T b){
    return a > b;
      }
 
  }; // End class
  inline void TrackCounting::setOrigin(const Trk::VxCandidate* priVtx) { m_priVtx=priVtx; }
  
} // End namespace 
 
#endif