123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- // Copyright 2019 Hans Dembinski
- //
- // Distributed under the Boost Software License, Version 1.0.
- // (See accompanying file LICENSE_1_0.txt
- // or copy at http://www.boost.org/LICENSE_1_0.txt)
- #ifndef BOOST_HISTOGRAM_DETAIL_LINEARIZE_HPP
- #define BOOST_HISTOGRAM_DETAIL_LINEARIZE_HPP
- #include <boost/assert.hpp>
- #include <boost/config.hpp>
- #include <boost/histogram/axis/option.hpp>
- #include <boost/histogram/axis/traits.hpp>
- #include <boost/histogram/axis/variant.hpp>
- #include <boost/histogram/detail/optional_index.hpp>
- #include <boost/histogram/fwd.hpp>
- namespace boost {
- namespace histogram {
- namespace detail {
- // initial offset to out must be set
- template <class Index, class Opts>
- std::size_t linearize(Opts, Index& out, const std::size_t stride,
- const axis::index_type size, const axis::index_type idx) {
- constexpr bool u = Opts::test(axis::option::underflow);
- constexpr bool o = Opts::test(axis::option::overflow);
- #ifdef BOOST_NO_CXX17_IF_CONSTEXPR
- if
- #else
- if constexpr
- #endif
- (std::is_same<Index, std::size_t>::value || (u && o)) {
- BOOST_ASSERT(idx >= (u ? -1 : 0));
- BOOST_ASSERT(idx < (o ? size + 1 : size));
- BOOST_ASSERT(idx >= 0 || static_cast<std::size_t>(-idx * stride) <= out);
- out += idx * stride;
- } else {
- BOOST_ASSERT(idx >= -1);
- BOOST_ASSERT(idx < size + 1);
- if ((u || idx >= 0) && (o || idx < size))
- out += idx * stride;
- else
- out = invalid_index;
- }
- return size + u + o;
- }
- template <class Index, class Axis, class Value>
- std::size_t linearize(Index& out, const std::size_t stride, const Axis& ax,
- const Value& v) {
- // mask options to reduce no. of template instantiations
- constexpr auto opts = axis::traits::static_options<Axis>{} &
- (axis::option::underflow | axis::option::overflow);
- return linearize(opts, out, stride, ax.size(), axis::traits::index(ax, v));
- }
- // initial offset of out must be zero
- template <class Index, class Axis, class Value>
- std::size_t linearize_growth(Index& out, axis::index_type& shift,
- const std::size_t stride, Axis& a, const Value& v) {
- axis::index_type idx;
- std::tie(idx, shift) = axis::traits::update(a, v);
- constexpr bool u = axis::traits::static_options<Axis>::test(axis::option::underflow);
- if (u) ++idx;
- if (std::is_same<Index, std::size_t>::value) {
- BOOST_ASSERT(idx < axis::traits::extent(a));
- out += idx * stride;
- } else {
- if (0 <= idx && idx < axis::traits::extent(a))
- out += idx * stride;
- else
- out = invalid_index;
- }
- return axis::traits::extent(a);
- }
- // initial offset of out must be zero
- template <class A>
- std::size_t linearize_index(optional_index& out, const std::size_t stride, const A& ax,
- const axis::index_type idx) {
- // cannot use static_options here, since A may be variant
- const auto opt = axis::traits::options(ax);
- const axis::index_type begin = opt & axis::option::underflow ? -1 : 0;
- const axis::index_type end = opt & axis::option::overflow ? ax.size() + 1 : ax.size();
- const axis::index_type extent = end - begin;
- // i may be arbitrarily out of range
- if (begin <= idx && idx < end)
- out += (idx - begin) * stride;
- else
- out = invalid_index;
- return extent;
- }
- template <class Index, class... Ts, class Value>
- std::size_t linearize(Index& o, const std::size_t s, const axis::variant<Ts...>& a,
- const Value& v) {
- return axis::visit([&o, &s, &v](const auto& a) { return linearize(o, s, a, v); }, a);
- }
- template <class Index, class... Ts, class Value>
- std::size_t linearize_growth(Index& o, axis::index_type& sh, const std::size_t st,
- axis::variant<Ts...>& a, const Value& v) {
- return axis::visit([&](auto& a) { return linearize_growth(o, sh, st, a, v); }, a);
- }
- } // namespace detail
- } // namespace histogram
- } // namespace boost
- #endif // BOOST_HISTOGRAM_DETAIL_LINEARIZE_HPP
|