NNJvtBinning.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#ifndef JETMOMENTTOOLS_NNJVTBINNING_H
#define JETMOMENTTOOLS_NNJVTBINNING_H
 
#include <vector>
#include <istream>
#include <string>
 
namespace xAOD {
    class IParticle;
}
 
namespace JetPileupTag {
    struct NNJvtBinning {
        std::vector<float> ptEdges;
        std::vector<float> etaEdges;
 
        static NNJvtBinning fromJSON(std::istream &is);
        std::string toJSON() const;
        bool operator()(float pt, float eta, std::size_t &ptBin, std::size_t &etaBin) const;
        bool
        operator()(const xAOD::IParticle &particle, std::size_t &ptBin, std::size_t &etaBin) const;
    };
 
    struct NNJvtCutMap {
        NNJvtBinning edges;
        std::vector<std::vector<float>> cutMap;
 
        static NNJvtCutMap fromJSON(std::istream &is);
        std::string toJSON() const;
        float operator()(float pt, float eta) const;
        float operator()(const xAOD::IParticle &particle) const;
        float operator()(std::size_t ptBin, std::size_t etaBin) const;
    };
} // namespace JetPileupTag
 
#endif //> !JETMOMENTTOOLS_NNJVTBINNING_H