xref: /aosp_15_r20/external/cronet/base/parameter_pack.h (revision 6777b5387eb2ff775bb5750e3f5d96f37fb7352b)
1*6777b538SAndroid Build Coastguard Worker // Copyright 2019 The Chromium Authors
2*6777b538SAndroid Build Coastguard Worker // Use of this source code is governed by a BSD-style license that can be
3*6777b538SAndroid Build Coastguard Worker // found in the LICENSE file.
4*6777b538SAndroid Build Coastguard Worker 
5*6777b538SAndroid Build Coastguard Worker #ifndef BASE_PARAMETER_PACK_H_
6*6777b538SAndroid Build Coastguard Worker #define BASE_PARAMETER_PACK_H_
7*6777b538SAndroid Build Coastguard Worker 
8*6777b538SAndroid Build Coastguard Worker #include <stddef.h>
9*6777b538SAndroid Build Coastguard Worker 
10*6777b538SAndroid Build Coastguard Worker #include <initializer_list>
11*6777b538SAndroid Build Coastguard Worker #include <tuple>
12*6777b538SAndroid Build Coastguard Worker #include <type_traits>
13*6777b538SAndroid Build Coastguard Worker 
14*6777b538SAndroid Build Coastguard Worker #include "base/containers/contains.h"
15*6777b538SAndroid Build Coastguard Worker 
16*6777b538SAndroid Build Coastguard Worker namespace base {
17*6777b538SAndroid Build Coastguard Worker 
18*6777b538SAndroid Build Coastguard Worker // Checks if any of the elements in |ilist| is true.
any_of(std::initializer_list<bool> ilist)19*6777b538SAndroid Build Coastguard Worker inline constexpr bool any_of(std::initializer_list<bool> ilist) {
20*6777b538SAndroid Build Coastguard Worker   return base::Contains(ilist, true);
21*6777b538SAndroid Build Coastguard Worker }
22*6777b538SAndroid Build Coastguard Worker 
23*6777b538SAndroid Build Coastguard Worker // Checks if all of the elements in |ilist| are true.
all_of(std::initializer_list<bool> ilist)24*6777b538SAndroid Build Coastguard Worker inline constexpr bool all_of(std::initializer_list<bool> ilist) {
25*6777b538SAndroid Build Coastguard Worker   return !base::Contains(ilist, false);
26*6777b538SAndroid Build Coastguard Worker }
27*6777b538SAndroid Build Coastguard Worker 
28*6777b538SAndroid Build Coastguard Worker // Counts the elements in |ilist| that are equal to |value|.
29*6777b538SAndroid Build Coastguard Worker // Similar to std::count for the case of constexpr initializer_list.
30*6777b538SAndroid Build Coastguard Worker template <class T>
count(std::initializer_list<T> ilist,T value)31*6777b538SAndroid Build Coastguard Worker inline constexpr size_t count(std::initializer_list<T> ilist, T value) {
32*6777b538SAndroid Build Coastguard Worker   size_t c = 0;
33*6777b538SAndroid Build Coastguard Worker   for (const auto& v : ilist) {
34*6777b538SAndroid Build Coastguard Worker     c += (v == value);
35*6777b538SAndroid Build Coastguard Worker   }
36*6777b538SAndroid Build Coastguard Worker   return c;
37*6777b538SAndroid Build Coastguard Worker }
38*6777b538SAndroid Build Coastguard Worker 
39*6777b538SAndroid Build Coastguard Worker constexpr size_t pack_npos = static_cast<size_t>(-1);
40*6777b538SAndroid Build Coastguard Worker 
41*6777b538SAndroid Build Coastguard Worker template <typename... Ts>
42*6777b538SAndroid Build Coastguard Worker struct ParameterPack {
43*6777b538SAndroid Build Coastguard Worker   // Checks if |Type| occurs in the parameter pack.
44*6777b538SAndroid Build Coastguard Worker   template <typename Type>
45*6777b538SAndroid Build Coastguard Worker   using HasType = std::bool_constant<any_of({std::is_same_v<Type, Ts>...})>;
46*6777b538SAndroid Build Coastguard Worker 
47*6777b538SAndroid Build Coastguard Worker   // Checks if the parameter pack only contains |Type|.
48*6777b538SAndroid Build Coastguard Worker   template <typename Type>
49*6777b538SAndroid Build Coastguard Worker   using OnlyHasType = std::bool_constant<all_of({std::is_same_v<Type, Ts>...})>;
50*6777b538SAndroid Build Coastguard Worker 
51*6777b538SAndroid Build Coastguard Worker   // Checks if |Type| occurs only once in the parameter pack.
52*6777b538SAndroid Build Coastguard Worker   template <typename Type>
53*6777b538SAndroid Build Coastguard Worker   using IsUniqueInPack =
54*6777b538SAndroid Build Coastguard Worker       std::bool_constant<count({std::is_same_v<Type, Ts>...}, true) == 1>;
55*6777b538SAndroid Build Coastguard Worker 
56*6777b538SAndroid Build Coastguard Worker   // Returns the zero-based index of |Type| within |Pack...| or |pack_npos| if
57*6777b538SAndroid Build Coastguard Worker   // it's not within the pack.
58*6777b538SAndroid Build Coastguard Worker   template <typename Type>
IndexInPackParameterPack59*6777b538SAndroid Build Coastguard Worker   static constexpr size_t IndexInPack() {
60*6777b538SAndroid Build Coastguard Worker     size_t index = 0;
61*6777b538SAndroid Build Coastguard Worker     for (bool value : {std::is_same_v<Type, Ts>...}) {
62*6777b538SAndroid Build Coastguard Worker       if (value)
63*6777b538SAndroid Build Coastguard Worker         return index;
64*6777b538SAndroid Build Coastguard Worker       index++;
65*6777b538SAndroid Build Coastguard Worker     }
66*6777b538SAndroid Build Coastguard Worker     return pack_npos;
67*6777b538SAndroid Build Coastguard Worker   }
68*6777b538SAndroid Build Coastguard Worker 
69*6777b538SAndroid Build Coastguard Worker   // Helper for extracting the Nth type from a parameter pack.
70*6777b538SAndroid Build Coastguard Worker   template <size_t N>
71*6777b538SAndroid Build Coastguard Worker   using NthType = std::tuple_element_t<N, std::tuple<Ts...>>;
72*6777b538SAndroid Build Coastguard Worker 
73*6777b538SAndroid Build Coastguard Worker   // Checks if every type in the parameter pack is the same.
74*6777b538SAndroid Build Coastguard Worker   using IsAllSameType =
75*6777b538SAndroid Build Coastguard Worker       std::bool_constant<all_of({std::is_same_v<NthType<0>, Ts>...})>;
76*6777b538SAndroid Build Coastguard Worker };
77*6777b538SAndroid Build Coastguard Worker 
78*6777b538SAndroid Build Coastguard Worker }  // namespace base
79*6777b538SAndroid Build Coastguard Worker 
80*6777b538SAndroid Build Coastguard Worker #endif  // BASE_PARAMETER_PACK_H_
81