aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/yt/string/format_arg-inl.h
blob: e2f08e540dc3bd0becde8bd9f9e060ab36a7d117 (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
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
#ifndef FORMAT_ARG_INL_H_
#error "Direct inclusion of this file is not allowed, include format_arg.h"
// For the sake of sane code completion.
#include "format_arg.h"
#endif

namespace NYT {

////////////////////////////////////////////////////////////////////////////////

namespace NDetail {

template <class T>
constexpr std::string_view QualidName()
{
    constexpr size_t openBracketOffset = 5;
    constexpr size_t closeBracketOffset = 1;
    constexpr std::string_view func = __PRETTY_FUNCTION__;
    constexpr auto left = std::find(std::begin(func), std::end(func), '[') + openBracketOffset;
    return std::string_view(left, std::prev(std::end(func), closeBracketOffset));
}

template <class T>
constexpr bool IsNYTName()
{
    constexpr auto qualidName = QualidName<T>();
    return qualidName.find("NYT::") == 0;
}

template <class T>
constexpr bool IsStdName()
{
    constexpr auto qualidName = QualidName<T>();
    return qualidName.find("std::") == 0;
}

} // namespace NDetail

////////////////////////////////////////////////////////////////////////////////

template <bool Hot, size_t N, std::array<char, N> List, class TFrom>
consteval auto TFormatArgBase::ExtendConversion()
{
    static_assert(std::same_as<TFrom, TFormatArgBase> || CFormattable<TFrom>);
    return AppendArrays<Hot, N, List, &TFormatArg<TFrom>::ConversionSpecifiers>();
}

template <bool Hot, size_t N, std::array<char, N> List, class TFrom>
consteval auto TFormatArgBase::ExtendFlags()
{
    static_assert(std::same_as<TFrom, TFormatArgBase> || CFormattable<TFrom>);
    return AppendArrays<Hot, N, List, &TFormatArg<TFrom>::FlagSpecifiers>();
}

template <bool Hot, size_t N, std::array<char, N> List, auto* From>
consteval auto TFormatArgBase::AppendArrays()
{
    auto& from = *From;
    return [] <size_t... I, size_t... J> (
        std::index_sequence<I...>,
        std::index_sequence<J...>) {
            if constexpr (Hot) {
                return std::array{List[J]..., from[I]...};
            } else {
                return std::array{from[I]..., List[J]...};
            }
        } (
            std::make_index_sequence<std::size(from)>(),
            std::make_index_sequence<N>());
}

////////////////////////////////////////////////////////////////////////////////

} // namespace NYT