Blame view

3rdparty/boost_1_81_0/boost/hana/count_if.hpp 3.08 KB
63e88f80   Hu Chunming   提交三方库
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
  /*!
  @file
  Defines `boost::hana::count_if`.
  
  Copyright Louis Dionne 2013-2022
  Distributed under the Boost Software License, Version 1.0.
  (See accompanying file LICENSE.md or copy at http://boost.org/LICENSE_1_0.txt)
   */
  
  #ifndef BOOST_HANA_COUNT_IF_HPP
  #define BOOST_HANA_COUNT_IF_HPP
  
  #include <boost/hana/fwd/count_if.hpp>
  
  #include <boost/hana/concept/foldable.hpp>
  #include <boost/hana/config.hpp>
  #include <boost/hana/core/dispatch.hpp>
  #include <boost/hana/detail/algorithm.hpp>
  #include <boost/hana/detail/fast_and.hpp>
  #include <boost/hana/integral_constant.hpp>
  #include <boost/hana/unpack.hpp>
  
  #include <cstddef>
  #include <type_traits>
  #include <utility>
  
  
  namespace boost { namespace hana {
      //! @cond
      template <typename Xs, typename Pred>
      constexpr auto count_if_t::operator()(Xs&& xs, Pred&& pred) const {
          using S = typename hana::tag_of<Xs>::type;
          using CountIf = BOOST_HANA_DISPATCH_IF(count_if_impl<S>,
              hana::Foldable<S>::value
          );
  
      #ifndef BOOST_HANA_CONFIG_DISABLE_CONCEPT_CHECKS
          static_assert(hana::Foldable<S>::value,
          "hana::count_if(xs, pred) requires 'xs' to be Foldable");
      #endif
  
          return CountIf::apply(static_cast<Xs&&>(xs),
                                static_cast<Pred&&>(pred));
      }
      //! @endcond
  
      namespace detail {
          template <typename Pred>
          struct count_pred {
              Pred pred;
              template <typename ...Xs, typename = typename std::enable_if<
                  detail::fast_and<
                      Constant<decltype((*pred)(std::declval<Xs&&>()))>::value...
                  >::value
              >::type>
              constexpr auto operator()(Xs&& ...xs) const {
                  constexpr bool results[] = {false, // <-- avoid empty array
                      static_cast<bool>(hana::value<decltype((*pred)(static_cast<Xs&&>(xs)))>())...
                  };
                  constexpr std::size_t total = detail::count(
                      results, results + sizeof(results), true
                  );
                  return hana::size_c<total>;
              }
  
              template <typename ...Xs, typename = void, typename = typename std::enable_if<
                  !detail::fast_and<
                      Constant<decltype((*pred)(std::declval<Xs&&>()))>::value...
                  >::value
              >::type>
              constexpr auto operator()(Xs&& ...xs) const {
                  std::size_t total = 0;
                  using Swallow = std::size_t[];
                  (void)Swallow{0, ((*pred)(static_cast<Xs&&>(xs)) ? ++total : 0)...};
                  return total;
              }
          };
      }
  
      template <typename T, bool condition>
      struct count_if_impl<T, when<condition>> : default_ {
          template <typename Xs, typename Pred>
          static constexpr decltype(auto) apply(Xs&& xs, Pred&& pred) {
              // We use a pointer instead of a reference to avoid a Clang ICE.
              return hana::unpack(static_cast<Xs&&>(xs),
                  detail::count_pred<decltype(&pred)>{&pred}
              );
          }
      };
  }} // end namespace boost::hana
  
  #endif // !BOOST_HANA_COUNT_IF_HPP