MultivariateTagManager.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BTAGTOOL_MULTIVARIATETAGMANAGER_C
#define BTAGTOOL_MULTIVARIATETAGMANAGER_C
 
/******************************************************
    @class MultivariateTagManager
    Package : JetTagTools
    Created : Oct 2015
    DESCRIPTION:
    General interface for MV based classifiers.
 
    @authors Dan Guest, Marie Lanfermann
 
********************************************************/
#include "AthenaBaseComps/AthAlgTool.h"
#include "JetTagTools/ITagTool.h"
#include "JetTagTools/IMultivariateJetTagger.h"
 
#include <vector>
 
namespace Analysis {
  class IMultivariateJetTagger;
 
  class MultivariateTagManager: public extends<AthAlgTool, ITagTool>
  {
  public:
 
    MultivariateTagManager(const std::string&,
                           const std::string&,
                           const IInterface*);
 
    virtual ~MultivariateTagManager() = default;
 
    virtual StatusCode initialize() override;
    virtual void finalizeHistos() override {};
 
    virtual StatusCode tagJet(const xAOD::Vertex& priVtx,
                              const xAOD::Jet& jetToTag,
                              xAOD::BTagging& BTag,
                              const std::string &jetName) const override;
 
 
  private:
 
    // internal typdefs
    typedef std::map<std::string, double> var_map;
 
    // load input variables from xAOD
    void fill_ip2d(var_map& inputs, xAOD::BTagging& BTag) const;
    void fill_ip3d(var_map& inputs, xAOD::BTagging& BTag) const;
    void fill_sv0(const xAOD::Vertex& priVtx, var_map& inputs, xAOD::BTagging& BTag) const;
    void fill_sv1(var_map& inputs, xAOD::BTagging& BTag) const;
    void fill_jetfitter(var_map& inputs, xAOD::BTagging& BTag) const;
    void fill_mvb(var_map& inputs, xAOD::Jet& jet, xAOD::BTagging& BTag) const;
    void fill_trkSum(var_map& inputs, xAOD::BTagging& BTag) const;
    void fill_softmuon(var_map& inputs, xAOD::BTagging& BTag) const;
 
    void fill_arbitrary_aux_data(var_map& inputs, xAOD::BTagging& BTag) const;
 
    // container information
    std::string m_ip2d_infosource;
    std::string m_ip3d_infosource;
    std::string m_sv0_infosource;
    std::string m_sv1_infosource;
    std::string m_jftNN_infosource;
    std::string m_softmuon_infosource;
 
    // any other arbitrary aux data
    std::vector<std::string> m_arbitrary_aux_data;
    std::map<std::string, std::string> m_aux_data_name_map;
 
    ToolHandleArray< IMultivariateJetTagger > m_MultivariateTaggerHandleArray;
  }; // end class
 
} //end Analysis namespace
 
#endif // BTAGTOOL_MULTIVARIATETAGMANAGER_C