da/dc2/QuantileFactory_8h_source.html

/*

 * Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration

*/

#ifndef CP_QUANTILEFACTORY_H

#define CP_QUANTILEFACTORY_H


#include "AthContainers/AuxElement.h"

#include <string>

#include <nlohmann/json.hpp>


class QuantileFactory {


  public:

  using QuantileFunc = std::pair<size_t, std::function<int(const SG::AuxElement&)>>;

  static QuantileFunc quantileFactory(const nlohmann::json& cfg);


  static std::vector<float> parseEdges(const nlohmann::json& cfg);


  private:

  static QuantileFunc makeEnumerate(const nlohmann::json& cfg);


  static QuantileFunc makeCategory(const nlohmann::json& cfg);


  static QuantileFunc makeNodes(const nlohmann::json& cfg);


  static QuantileFunc makeDense(const nlohmann::json& cfg);


};


#endif // CP_QUANTILEFACTORY_H

AuxElement.h
Base class for elements of a container that can have aux data.

QuantileFactory
Builds binning and quantisation functions from a JSON configuration.
Definition QuantileFactory.h:16

QuantileFactory::QuantileFunc
std::pair< size_t, std::function< int(const SG::AuxElement &)> > QuantileFunc
Definition QuantileFactory.h:19

QuantileFactory::makeEnumerate
static QuantileFunc makeEnumerate(const nlohmann::json &cfg)
Maps discrete values to category indices.
Definition QuantileFactory.cxx:66

QuantileFactory::makeNodes
static QuantileFunc makeNodes(const nlohmann::json &cfg)
Builds a tree-based lookup structure for node-based binning, primarily used for pseudo-continuous 2D ...
Definition QuantileFactory.cxx:94

QuantileFactory::makeDense
static QuantileFunc makeDense(const nlohmann::json &cfg)
Combines multiple binning dimensions into a single flattened global index used for scale factor looku...
Definition QuantileFactory.cxx:141

QuantileFactory::quantileFactory
static QuantileFunc quantileFactory(const nlohmann::json &cfg)
Dispatch to the appropriate binning scheme based on the JSON config.
Definition QuantileFactory.cxx:174

QuantileFactory::parseEdges
static std::vector< float > parseEdges(const nlohmann::json &cfg)
Parse an ordered list of bin edges from the JSON config.
Definition QuantileFactory.cxx:9

QuantileFactory::makeCategory
static QuantileFunc makeCategory(const nlohmann::json &cfg)
Converts a continuous variable into a bin index using an ordered list of bin edges.
Definition QuantileFactory.cxx:43

SG::AuxElement
AuxElement(SG::AuxVectorData *container, size_t index)
Base class for elements of a container that can have aux data.