BTagTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
/* **************************************************************************
                           BTagTool.h  -  Description
                             -------------------
    begin   : 10.03.04
    authors : Andreas Wildauer (CERN PH-ATC), Fredrik Akesson (CERN PH-ATC)
    email   : andreas.wildauer@cern.ch, fredrik.akesson@cern.ch
    comments: prototype implementation of a general jet tag strategy for the case of b-tagging
   ***************************************************************************/
 
#ifndef BTAGGING_BTAGTOOL_H
#define BTAGGING_BTAGTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "BTagging/IBTagTool.h"
#include "JetTagTools/ITagTool.h"
#include "StoreGate/ReadHandleKey.h"
 
#include <string>
#include <vector>
#include <map>
#include <atomic>
 
 
namespace Analysis
{
 
  class BTagTool : public extends<AthAlgTool, IBTagTool>
  {
    public:
 
      BTagTool(const std::string&,const std::string&,const IInterface*);
      virtual ~BTagTool() = default;
 
      virtual StatusCode initialize() override;
      virtual StatusCode finalize() override;
      virtual void finalizeHistos() override;
 
      virtual
      StatusCode tagJet(const xAOD::Jet*, xAOD::BTagging*, const std::string &jetName, const xAOD::Vertex* vtx = 0) const override;
      virtual
      StatusCode tagJet(const xAOD::JetContainer * jetContainer, xAOD::BTaggingContainer * btaggingContainer, const std::string &jetName) const override;
 
    private:
 
      mutable std::atomic<unsigned int> m_nBeamSpotPvx{};
      mutable std::atomic<unsigned int> m_nAllJets{};
 
      ToolHandleArray< ITagTool > m_bTagToolHandleArray;
      std::map<std::string, ITagTool*> m_bTagTool; 
 
      SG::ReadHandleKey<xAOD::VertexContainer> m_VertexCollectionName {this, "vxPrimaryCollectionName", "", "Input primary vertex container"};
 
  }; // End class
} // End namespace
#endif