HistogramWrapper.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2024 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#pragma once
 
#include <Gaudi/Accumulators/StaticHistogram.h>
 
namespace Gaudi::Accumulators {
 
  template <typename HistogramType,
            typename Seq =
                std::make_integer_sequence<unsigned int, std::tuple_size_v<typename HistogramType::AxisTupleType>>>
  class HistogramWrapperInternal;
  template <typename HistogramType, unsigned int... ND>
  class HistogramWrapperInternal<HistogramType, std::integer_sequence<unsigned int, ND...>> {
  public:
    using AxisTupleType      = typename HistogramType::AxisTupleType;
    using AxisArithmeticType = typename HistogramType::AxisArithmeticType;
    template <unsigned int I>
    using AxisType = std::tuple_element_t<I, AxisTupleType>;
 
    template <typename OWNER>
    HistogramWrapperInternal( OWNER* owner, std::string const& name, std::string const& title = "",
                              typename HistogramType::AxisTupleType axis = {}, bool doNotInitialize = false )
        : m_name{ name }, m_title{ title }, m_axis{ axis } {
      // Create associated properties
      owner->declareProperty( titlePropertyName(), m_title, fmt::format( "Title of histogram {}", name ) )
          ->template setOwnerType<OWNER>();
      ( owner
            ->declareProperty( axisPropertyName<ND>(), std::get<ND>( m_axis ),
                               fmt::format( "Axis {} of histogram {}", ND, name ) )
            ->template setOwnerType<OWNER>(),
        ... );
      // register creation of the Histogram at initialization time
      if ( !doNotInitialize ) {
        if ( owner->FSMState() >= Gaudi::StateMachine::INITIALIZED ) {
          // if the owner is already initialized (e.g. the histogram is being created during `start()`)
          // it is too late to register the callback
          createHistogram( *owner );
        } else {
          owner->registerCallBack( StateMachine::INITIALIZE, [this, owner]() { createHistogram( *owner ); } );
        }
      }
    }
    template <typename OWNER>
    HistogramWrapperInternal( OWNER const* owner, std::string const& name, std::string const& title = "",
                              typename HistogramType::AxisTupleType axis = {} )
        : m_name{ name }, m_title{ title }, m_axis{ axis } {
      createHistogram( *owner );
    }
    template <typename OWNER>
    HistogramWrapperInternal( OWNER* owner, std::string const& name, std::string const& title, AxisType<ND>... allAxis )
        : HistogramWrapperInternal( owner, name, title, std::make_tuple( allAxis... ) ) {}
 
    [[nodiscard]] auto operator[]( typename HistogramType::AxisTupleArithmeticType&& v ) {
      if ( !m_histo ) {
        throw std::logic_error( fmt::format( "Histogram {} is used before being initialized", m_name ) );
      }
      return m_histo.value()[std::forward<typename HistogramType::AxisTupleArithmeticType>( v )];
    }
 
    template <typename OWNER>
    void createHistogram( OWNER& owner ) {
      m_histo.emplace( &owner, m_name, m_title, m_axis );
    }
 
    friend void to_json( nlohmann::json& j, HistogramWrapperInternal const& h ) {
      if ( !h.m_histo ) {
        throw std::logic_error( fmt::format( "Histogram {} is converted to json before being initialized", h.m_name ) );
      }
      j = h.m_histo.value();
    }
 
    // set directly some properties, only if histogram was not yet created
    void setTitle( std::string const& title ) {
      if ( m_histo )
        throw std::logic_error(
            fmt::format( "Cannot modify title of histogram {} after it has been initialized", m_name ) );
      m_title = title;
    }
    template <unsigned int N>
    void setAxis( std::tuple_element_t<N, typename HistogramType::AxisTupleType> const& axis ) {
      if ( m_histo )
        throw std::logic_error(
            fmt::format( "Cannot modify axis {} of histogram {} after it has been initialized", N, m_name ) );
      std::get<N>( m_axis ) = axis;
    }
 
    void reset() { m_histo.reset(); }
 
    // wrapping some methods of the underlyoing histogram
    auto buffer() {
      if ( !m_histo )
        throw std::logic_error( fmt::format( "`buffer()` called on histogram {} before being initialized", m_name ) );
      return m_histo->buffer();
    }
 
  private:
    std::string basePropertyName() const {
      // properties are used as identifiers in python and thus cannot anything alse than _, letters and numbers
      // we thus replace anything else with '_' in the property names
      std::string name = m_name;
      std::replace_if(
          begin( name ), end( name ), []( auto& c ) { return !std::isalnum( c ); }, '_' );
      return name;
    }
    std::string titlePropertyName() const { return fmt::format( "{}_Title", basePropertyName() ); }
    template <unsigned int N>
    std::string axisPropertyName() const {
      return fmt::format( "{}_Axis{}", basePropertyName(), N );
    }
 
    // Members of the custom histogrem
    std::string                           m_name{};
    std::string                           m_title{};
    typename HistogramType::AxisTupleType m_axis{};
    std::optional<HistogramType>          m_histo{};
  };
 
  template <typename HistogramType>
  using HistogramWrapper = HistogramWrapperInternal<HistogramType>;
 
} // namespace Gaudi::Accumulators