summaryrefslogtreecommitdiffstats
path: root/yql/essentials/utils/arity.h
blob: c923187c85c47654c553e5947d832e19a507e48c (plain) (blame)
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
#pragma once

#include <cstddef>

namespace NYql {

template <typename T>
struct TArity;

template <typename Ret, typename... Args>
struct TArity<Ret(Args...)> {
    static constexpr std::size_t value = sizeof...(Args);
};

template <typename Ret, typename... Args>
struct TArity<Ret (*)(Args...)>: TArity<Ret(Args...)> {};

#define METHOD_ARITY(TAIL)                                        \
    template <typename Ret, typename Class, typename... Args>     \
    struct TArity<Ret (Class::*)(Args...) TAIL> {                 \
        static constexpr std::size_t value = 1 + sizeof...(Args); \
    };

METHOD_ARITY()
METHOD_ARITY(const)
METHOD_ARITY(volatile)
METHOD_ARITY(const volatile)
METHOD_ARITY(&)
METHOD_ARITY(const&)
METHOD_ARITY(&&)
METHOD_ARITY(const&&)
METHOD_ARITY(noexcept)
METHOD_ARITY(const noexcept)
METHOD_ARITY(volatile noexcept)
METHOD_ARITY(const volatile noexcept)

#undef METHOD_ARITY

}; // namespace NYql

template <typename T>
inline constexpr std::size_t arity_v = NYql::TArity<T>::value;