EIC code displayed by LXR master/​acts/​Core/​include/​Acts/​Utilities/​BinningData.hpp	Source navigation Diff markup Identifier search General search
	Version: master test Revert   Change

File indexing completed on 2025-07-01 07:52:39

 // This file is part of the ACTS project.
 //
 // Copyright (C) 2016 CERN for the benefit of the ACTS project
 //
 // This Source Code Form is subject to the terms of the Mozilla Public
 // License, v. 2.0. If a copy of the MPL was not distributed with this
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
 #pragma once
 
 #include "Acts/Definitions/Algebra.hpp"
 #include "Acts/Utilities/AxisDefinitions.hpp"
 #include "Acts/Utilities/BinningType.hpp"
 #include "Acts/Utilities/Helpers.hpp"
 #include "Acts/Utilities/ProtoAxis.hpp"
 #include "Acts/Utilities/ThrowAssert.hpp"
 #include "Acts/Utilities/VectorHelpers.hpp"
 
 #include <algorithm>
 #include <cmath>
 #include <memory>
 #include <sstream>
 #include <string>
 #include <utility>
 #include <vector>
 
 namespace Acts {
 
 /// @class BinningData
 ///
 ///   This class holds all the data necessary for the bin calculation
 ///
 ///   phi has a very particular behaviour:
 ///   - there's the change around +/- PI
 ///
 ///   - it can be multiplicative or additive
 ///   multiplicative : each major bin has the same sub structure
 ///                    i.e. first binnning
 ///
 /// structure is equidistant
 ///   additive : sub structure replaces one bin (and one bin only)
 ///
 ///
 class BinningData {
  public:
   BinningType type{};        ///< binning type: equidistant, arbitrary
   BinningOption option{};    ///< binning option: open, closed
   AxisDirection binvalue{};  ///< axis direction: AxisX, AxisY, AxisZ, ...
   float min{};               ///< minimum value
   float max{};               ///< maximum value
   float step{};              ///< binning step
   bool zdim{};               ///< zero dimensional binning : direct access
 
   /// sub structure: describe some sub binning
   std::unique_ptr<const BinningData> subBinningData;
   /// sub structure: additive or multipicative
   bool subBinningAdditive{};
 
   /// Constructor for 0D binning
   ///
   /// @param bValue is the axis direction AxisX, AxisY, etc.
   /// @param bMin is the minimum value
   /// @param bMax is the maximum value
   BinningData(AxisDirection bValue, float bMin, float bMax)
       : type(equidistant),
         option(open),
         binvalue(bValue),
         min(bMin),
         max(bMax),
         step((bMax - bMin)),
         zdim(true),
         subBinningData(nullptr),
         m_bins(1),
         m_boundaries({{min, max}}),
         m_totalBins(1),
         m_totalBoundaries(std::vector<float>()),
         m_functionPtr(&searchEquidistantWithBoundary) {}
 
   /// Constructor for equidistant binning
   /// and optional sub structure can be
   /// multiplicative or additive
   ///
   /// @param bOption is the binning option : open, closed
   /// @param bValue is the axis direction: Axis, AxisY, etc.
   /// @param bBins is number of equidistant bins
   /// @param bMin is the minimum value
   /// @param bMax is the maximum value
   /// @param sBinData is (optional) sub structure
   /// @param sBinAdditive is the prescription for the sub structure
   BinningData(BinningOption bOption, AxisDirection bValue, std::size_t bBins,
               float bMin, float bMax,
               std::unique_ptr<const BinningData> sBinData = nullptr,
               bool sBinAdditive = false)
       : type(equidistant),
         option(bOption),
         binvalue(bValue),
         min(bMin),
         max(bMax),
         step((bMax - bMin) / bBins),
         zdim(bBins == 1 ? true : false),
         subBinningData(std::move(sBinData)),
         subBinningAdditive(sBinAdditive),
         m_bins(bBins),
         m_boundaries(std::vector<float>()),
         m_totalBins(bBins),
         m_totalBoundaries(std::vector<float>()) {
     // set to equidistant search
     m_functionPtr = &searchEquidistantWithBoundary;
     // fill the boundary vector for fast access to center & boundaries
     m_boundaries.reserve(m_bins + 1);
     for (std::size_t ib = 0; ib < m_bins + 1; ++ib) {
       m_boundaries.push_back(min + ib * step);
     }
     // the binning data has sub structure - multiplicative or additive
     checkSubStructure();
   }
 
   /// Constructor for non-equidistant binning
   ///
   /// @param bOption is the binning option : open / closed
   /// @param bValue is the axis direction : AxisX, AxisY, etc.
   /// @param bBoundaries are the bin boundaries
   /// @param sBinData is (optional) sub structure
   BinningData(BinningOption bOption, AxisDirection bValue,
               const std::vector<float>& bBoundaries,
               std::unique_ptr<const BinningData> sBinData = nullptr)
       : type(arbitrary),
         option(bOption),
         binvalue(bValue),
         zdim(bBoundaries.size() == 2 ? true : false),
         subBinningData(std::move(sBinData)),
         subBinningAdditive(true),
         m_bins(bBoundaries.size() - 1),
         m_boundaries(bBoundaries),
         m_totalBins(bBoundaries.size() - 1),
         m_totalBoundaries(bBoundaries) {
     // assert a no-size case
     throw_assert(m_boundaries.size() > 1, "Must have more than one boundary");
     min = m_boundaries[0];
     max = m_boundaries[m_boundaries.size() - 1];
     // set to equidistant search
     m_functionPtr = &searchInVectorWithBoundary;
     // the binning data has sub structure - multiplicative
     checkSubStructure();
   }
 
   /// Copy constructor
   ///
   /// @param bdata is the source object
   BinningData(const BinningData& bdata)
       : type(bdata.type),
         option(bdata.option),
         binvalue(bdata.binvalue),
         min(bdata.min),
         max(bdata.max),
         step(bdata.step),
         zdim(bdata.zdim),
         subBinningData(nullptr),
         subBinningAdditive(bdata.subBinningAdditive),
         m_bins(bdata.m_bins),
         m_boundaries(bdata.m_boundaries),
         m_totalBins(bdata.m_totalBins),
         m_totalBoundaries(bdata.m_totalBoundaries) {
     // get the binning data
     subBinningData =
         bdata.subBinningData
             ? std::make_unique<const BinningData>(*bdata.subBinningData)
             : nullptr;
     // set the pointer depending on the type
     // set the correct function pointer
     if (type == equidistant) {
       m_functionPtr = &searchEquidistantWithBoundary;
     } else {
       m_functionPtr = &searchInVectorWithBoundary;
     }
   }
 
   /// Constructor from DirectedProtoAxis
   ///
   /// @param dpAxis is the ProtoAxis object
   ///
   explicit BinningData(const DirectedProtoAxis& dpAxis)
       : binvalue(dpAxis.getAxisDirection()), subBinningData(nullptr) {
     const auto& axis = dpAxis.getAxis();
     type = axis.getType() == AxisType::Equidistant ? equidistant : arbitrary;
     option = axis.getBoundaryType() == AxisBoundaryType::Closed ? closed : open;
     min = static_cast<float>(axis.getMin());
     max = static_cast<float>(axis.getMax());
     m_bins = axis.getNBins();
     step = (max - min) / static_cast<float>(m_bins);
     zdim = (m_bins == 1);
     m_boundaries.reserve(axis.getBinEdges().size());
     for (const auto& edge : axis.getBinEdges()) {
       m_boundaries.push_back(static_cast<float>(edge));
     }
     m_totalBins = m_bins;
     m_totalBoundaries = m_boundaries;
     // Set the search function pointer based on axis type
     m_functionPtr = (type == equidistant) ? &searchEquidistantWithBoundary
                                           : &searchInVectorWithBoundary;
   }
 
   /// Assignment operator
   ///
   /// @param bdata is the source object
   BinningData& operator=(const BinningData& bdata) {
     if (this != &bdata) {
       type = bdata.type;
       option = bdata.option;
       binvalue = bdata.binvalue;
       min = bdata.min;
       max = bdata.max;
       step = bdata.step;
       zdim = bdata.zdim;
       subBinningAdditive = bdata.subBinningAdditive;
       subBinningData =
           bdata.subBinningData
               ? std::make_unique<const BinningData>(*bdata.subBinningData)
               : nullptr;
       m_bins = bdata.m_bins;
       m_boundaries = bdata.m_boundaries;
       m_totalBins = bdata.m_totalBins;
       m_totalBoundaries = bdata.m_totalBoundaries;
       // set the correct function pointer
       if (type == equidistant) {
         m_functionPtr = &searchEquidistantWithBoundary;
       } else {
         m_functionPtr = &searchInVectorWithBoundary;
       }
     }
     return (*this);
   }
 
   BinningData() = default;
   ~BinningData() = default;
 
   /// Equality operator
   ///
   /// @param bData is the binning data to be checked against
   ///
   /// @return a boolean indicating if they are the same
   bool operator==(const BinningData& bData) const {
     return (type == bData.type && option == bData.option &&
             binvalue == bData.binvalue && min == bData.min &&
             max == bData.max && step == bData.step && zdim == bData.zdim &&
             ((subBinningData == nullptr && bData.subBinningData == nullptr) ||
              (subBinningData != nullptr && bData.subBinningData != nullptr &&
               (*subBinningData == *bData.subBinningData))) &&
             subBinningAdditive == bData.subBinningAdditive);
   }
 
   /// Return the number of bins - including sub bins
   std::size_t bins() const { return m_totalBins; }
 
   /// Return the boundaries  - including sub boundaries
   /// @return vector of floats indicating the boundary values
   const std::vector<float>& boundaries() const {
     if (subBinningData) {
       return m_totalBoundaries;
     }
     return m_boundaries;
   }
 
   /// Take the right float value
   ///
   /// @param lposition assumes the correct local position expression
   ///
   /// @return float value according to the binning setup
   float value(const Vector2& lposition) const {
     // ordered after occurrence
     if (binvalue == AxisDirection::AxisR ||
         binvalue == AxisDirection::AxisRPhi ||
         binvalue == AxisDirection::AxisX ||
         binvalue == AxisDirection::AxisTheta) {
       return lposition[0];
     }
 
     return lposition[1];
   }
 
   /// Take the right float value
   ///
   /// @param position is the global position
   ///
   /// @return float value according to the binning setup
   float value(const Vector3& position) const {
     using VectorHelpers::eta;
     using VectorHelpers::perp;
     using VectorHelpers::phi;
     // ordered after occurrence
     if (binvalue == AxisDirection::AxisR ||
         binvalue == AxisDirection::AxisTheta) {
       return (perp(position));
     }
     if (binvalue == AxisDirection::AxisRPhi) {
       return (perp(position) * phi(position));
     }
     if (binvalue == AxisDirection::AxisEta) {
       return (eta(position));
     }
     if (toUnderlying(binvalue) < 3) {
       return static_cast<float>(position[toUnderlying(binvalue)]);
     }
     // phi gauging
     return phi(position);
   }
 
   /// Get the center value of a bin
   ///
   /// @param bin is the bin for which the center value is requested
   ///
   /// @return float value according to the bin center
   float center(std::size_t bin) const {
     const std::vector<float>& bvals = boundaries();
     // take the center between bin boundaries
     float value =
         bin < (bvals.size() - 1) ? 0.5 * (bvals[bin] + bvals[bin + 1]) : 0.;
     return value;
   }
 
   /// Get the width of a bin
   ///
   /// @param bin is the bin for which the width is requested
   ///
   /// @return float value of width
   float width(std::size_t bin) const {
     const std::vector<float>& bvals = boundaries();
     // take the center between bin boundaries
     float value = bin < (bvals.size() - 1) ? bvals[bin + 1] - bvals[bin] : 0.;
     return value;
   }
 
   /// Check if bin is inside from Vector3
   ///
   /// @param position is the search position in global coordinated
   ///
   /// @return boolean if this is inside() method is true
   bool inside(const Vector3& position) const {
     // closed one is always inside
     if (option == closed) {
       return true;
     }
     // all other options
     // @todo remove hard-coded tolerance parameters
     float val = value(position);
     return (val > min - 0.001 && val < max + 0.001);
   }
 
   /// Check if bin is inside from Vector2
   ///
   /// @param lposition is the search position in global coordinated
   ///
   /// @return boolean if this is inside() method is true
   bool inside(const Vector2& lposition) const {
     // closed one is always inside
     if (option == closed) {
       return true;
     }
     // all other options
     // @todo remove hard-coded tolerance parameters
     float val = value(lposition);
     return (val > min - 0.001 && val < max + 0.001);
   }
 
   /// Generic search from a 2D position
   /// -- corresponds to local coordinate schema
   /// @param lposition is the search position in local coordinated
   ///
   /// @return bin according tot this
   std::size_t searchLocal(const Vector2& lposition) const {
     if (zdim) {
       return 0;
     }
     return search(value(lposition));
   }
 
   /// Generic search from a 3D position
   /// -- corresponds to global coordinate schema
   /// @param position is the search position in global coordinated
   ///
   /// @return bin according tot this
   std::size_t searchGlobal(const Vector3& position) const {
     if (zdim) {
       return 0;
     }
     return search(value(position));
   }
 
   /// Generic search - forwards to correct function pointer
   ///
   /// @param value is the searchvalue as float
   ///
   /// @return bin according tot this
   std::size_t search(float value) const {
     if (zdim) {
       return 0;
     }
     assert(m_functionPtr != nullptr);
     return (!subBinningData) ? (*m_functionPtr)(value, *this)
                              : searchWithSubStructure(value);
   }
 
   ///  Generic search with sub structure
   /// - forwards to correct function pointer
   ///
   /// @param value is the searchvalue as float
   ///
   /// @return bin according tot this
   std::size_t searchWithSubStructure(float value) const {
     // find the masterbin with the correct function pointer
     std::size_t masterbin = (*m_functionPtr)(value, *this);
     // additive sub binning -
     if (subBinningAdditive) {
       // no gauging done, for additive sub structure
       return masterbin + subBinningData->search(value);
     }
     // gauge the value to the subBinData
     float gvalue =
         value - masterbin * (subBinningData->max - subBinningData->min);
     // now go / additive or multiplicative
     std::size_t subbin = subBinningData->search(gvalue);
     // now return
     return masterbin * subBinningData->bins() + subbin;
   }
 
   /// Layer next direction is needed
   ///
   /// @param position is the start search position
   /// @param dir is the direction
   /// @todo check if this can be changed
   ///
   /// @return integer that indicates which direction to move
   int nextDirection(const Vector3& position, const Vector3& dir) const {
     if (zdim) {
       return 0;
     }
     float val = value(position);
     Vector3 probe = position + dir.normalized();
     float nextval = value(probe);
     return (nextval > val) ? 1 : -1;
   }
 
   /// access to the center value
   /// this uses the bin boundary vector, it also works with sub structure
   ///
   /// @param bin is the bin for which the value is requested, if bin > nbins
   /// it is set to max
   ///
   /// @return the center value of the bin is given
   float centerValue(std::size_t bin) const {
     if (zdim) {
       return 0.5 * (min + max);
     }
     float bmin = m_boundaries[bin];
     float bmax = bin < m_boundaries.size() ? m_boundaries[bin + 1] : max;
     return 0.5 * (bmin + bmax);
   }
 
  private:
   std::size_t m_bins{};             ///< number of bins
   std::vector<float> m_boundaries;  ///< vector of holding the bin boundaries
   std::size_t m_totalBins{};        ///< including potential substructure
   std::vector<float> m_totalBoundaries;  ///< including potential substructure
 
   std::size_t (*m_functionPtr)(float,
                                const BinningData&){};  /// function pointer
 
   /// helper method to set the sub structure
   void checkSubStructure() {
     // sub structure is only checked when sBinData is defined
     if (subBinningData) {
       m_totalBoundaries.clear();
       // (A) additive sub structure
       if (subBinningAdditive) {
         // one bin is replaced by the sub bins
         m_totalBins = m_bins + subBinningData->bins() - 1;
         // the tricky one - exchange one bin by many others
         m_totalBoundaries.reserve(m_totalBins + 1);
         // get the sub bin boundaries
         const std::vector<float>& subBinBoundaries =
             subBinningData->boundaries();
         float sBinMin = subBinBoundaries[0];
         // get the min value of the sub bin boundaries
         std::vector<float>::const_iterator mbvalue = m_boundaries.begin();
         for (; mbvalue != m_boundaries.end(); ++mbvalue) {
           // should define numerically stable
           if (std::abs((*mbvalue) - sBinMin) < 10e-10) {
             // copy the sub bin boundaries into the vector
             m_totalBoundaries.insert(m_totalBoundaries.begin(),
                                      subBinBoundaries.begin(),
                                      subBinBoundaries.end());
             ++mbvalue;
           } else {
             m_totalBoundaries.push_back(*mbvalue);
           }
         }
       } else {  // (B) multiplicative sub structure
         // every bin is just replaced by the sub binning structure
         m_totalBins = m_bins * subBinningData->bins();
         m_totalBoundaries.reserve(m_totalBins + 1);
         // get the sub bin boundaries if there are any
         const std::vector<float>& subBinBoundaries =
             subBinningData->boundaries();
         // create the boundary vector
         m_totalBoundaries.push_back(min);
         for (std::size_t ib = 0; ib < m_bins; ++ib) {
           float offset = ib * step;
           for (std::size_t isb = 1; isb < subBinBoundaries.size(); ++isb) {
             m_totalBoundaries.push_back(offset + subBinBoundaries[isb]);
           }
         }
       }
       // sort the total boundary vector
       std::ranges::sort(m_totalBoundaries);
     }
   }
 
   // Equidistant search
   // - fastest method
   static std::size_t searchEquidistantWithBoundary(float value,
                                                    const BinningData& bData) {
     // vanilla
 
     int bin = static_cast<int>((value - bData.min) / bData.step);
     // special treatment of the 0 bin for closed
     if (bData.option == closed) {
       if (value < bData.min) {
         return (bData.m_bins - 1);
       }
       if (value > bData.max) {
         return 0;
       }
     }
     // if outside boundary : return boundary for open, opposite bin for closed
     bin = bin < 0 ? ((bData.option == open) ? 0 : (bData.m_bins - 1)) : bin;
     return static_cast<std::size_t>(
         (bin <= static_cast<int>(bData.m_bins - 1))
             ? bin
             : ((bData.option == open) ? (bData.m_bins - 1) : 0));
   }
 
   // Search in arbitrary boundary
   static std::size_t searchInVectorWithBoundary(float value,
                                                 const BinningData& bData) {
     // lower boundary
     if (value <= bData.m_boundaries[0]) {
       return (bData.option == closed) ? (bData.m_bins - 1) : 0;
     }
     // higher boundary
     if (value >= bData.max) {
       return (bData.option == closed) ? 0 : (bData.m_bins - 1);
     }
 
     auto lb = std::lower_bound(bData.m_boundaries.begin(),
                                bData.m_boundaries.end(), value);
     return static_cast<std::size_t>(
         std::distance(bData.m_boundaries.begin(), lb) - 1);
   }
 
  public:
   /// String screen output method
   /// @param indent the current indentation
   /// @return a string containing the screen information
   std::string toString(const std::string& indent = "") const {
     std::stringstream sl;
     sl << indent << "BinningData object:" << '\n';
     sl << indent << "  - type       : " << static_cast<std::size_t>(type)
        << '\n';
     sl << indent << "  - option     : " << static_cast<std::size_t>(option)
        << '\n';
     sl << indent << "  - value      : " << static_cast<std::size_t>(binvalue)
        << '\n';
     sl << indent << "  - bins       : " << bins() << '\n';
     sl << indent << "  - min/max    : " << min << " / " << max << '\n';
     if (type == equidistant) {
       sl << indent << "  - step       : " << step << '\n';
     }
     sl << indent << "  - boundaries : | ";
     for (const auto& b : boundaries()) {
       sl << b << " | ";
     }
     sl << '\n';
     return sl.str();
   }
 };
 }  // namespace Acts

This page was automatically generated by the 2.3.7 LXR engine. The LXR team