Warning: This file is not a C or C++ file. It does not have highlighting.
| 1 | // -*- C++ -*- |
|---|---|
| 2 | //===----------------------------------------------------------------------===// |
| 3 | // |
| 4 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 5 | // See https://llvm.org/LICENSE.txt for license information. |
| 6 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
| 10 | #ifndef _LIBCPP___FORMAT_FORMAT_CONTEXT_H |
| 11 | #define _LIBCPP___FORMAT_FORMAT_CONTEXT_H |
| 12 | |
| 13 | #include <__concepts/same_as.h> |
| 14 | #include <__config> |
| 15 | #include <__format/buffer.h> |
| 16 | #include <__format/format_arg.h> |
| 17 | #include <__format/format_arg_store.h> |
| 18 | #include <__format/format_args.h> |
| 19 | #include <__format/format_error.h> |
| 20 | #include <__fwd/format.h> |
| 21 | #include <__iterator/back_insert_iterator.h> |
| 22 | #include <__iterator/concepts.h> |
| 23 | #include <__memory/addressof.h> |
| 24 | #include <__utility/move.h> |
| 25 | #include <__variant/monostate.h> |
| 26 | |
| 27 | #if _LIBCPP_HAS_LOCALIZATION |
| 28 | # include <__locale> |
| 29 | # include <optional> |
| 30 | #endif |
| 31 | |
| 32 | #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER) |
| 33 | # pragma GCC system_header |
| 34 | #endif |
| 35 | |
| 36 | _LIBCPP_PUSH_MACROS |
| 37 | #include <__undef_macros> |
| 38 | |
| 39 | _LIBCPP_BEGIN_NAMESPACE_STD |
| 40 | |
| 41 | #if _LIBCPP_STD_VER >= 20 |
| 42 | |
| 43 | template <class _OutIt, class _CharT> |
| 44 | requires output_iterator<_OutIt, const _CharT&> |
| 45 | class basic_format_context; |
| 46 | |
| 47 | # if _LIBCPP_HAS_LOCALIZATION |
| 48 | /** |
| 49 | * Helper to create a basic_format_context. |
| 50 | * |
| 51 | * This is needed since the constructor is private. |
| 52 | */ |
| 53 | template <class _OutIt, class _CharT> |
| 54 | _LIBCPP_HIDE_FROM_ABI basic_format_context<_OutIt, _CharT> |
| 55 | __format_context_create(_OutIt __out_it, |
| 56 | basic_format_args<basic_format_context<_OutIt, _CharT>> __args, |
| 57 | optional<std::locale>&& __loc = nullopt) { |
| 58 | return std::basic_format_context(std::move(__out_it), __args, std::move(__loc)); |
| 59 | } |
| 60 | # else |
| 61 | template <class _OutIt, class _CharT> |
| 62 | _LIBCPP_HIDE_FROM_ABI basic_format_context<_OutIt, _CharT> |
| 63 | __format_context_create(_OutIt __out_it, basic_format_args<basic_format_context<_OutIt, _CharT>> __args) { |
| 64 | return std::basic_format_context(std::move(__out_it), __args); |
| 65 | } |
| 66 | # endif |
| 67 | |
| 68 | using format_context = basic_format_context<back_insert_iterator<__format::__output_buffer<char>>, char>; |
| 69 | # if _LIBCPP_HAS_WIDE_CHARACTERS |
| 70 | using wformat_context = basic_format_context< back_insert_iterator<__format::__output_buffer<wchar_t>>, wchar_t>; |
| 71 | # endif |
| 72 | |
| 73 | template <class _OutIt, class _CharT> |
| 74 | requires output_iterator<_OutIt, const _CharT&> |
| 75 | class _LIBCPP_PREFERRED_NAME(format_context) |
| 76 | _LIBCPP_IF_WIDE_CHARACTERS(_LIBCPP_PREFERRED_NAME(wformat_context)) basic_format_context { |
| 77 | public: |
| 78 | using iterator = _OutIt; |
| 79 | using char_type = _CharT; |
| 80 | template <class _Tp> |
| 81 | using formatter_type = formatter<_Tp, _CharT>; |
| 82 | |
| 83 | _LIBCPP_HIDE_FROM_ABI basic_format_arg<basic_format_context> arg(size_t __id) const noexcept { |
| 84 | return __args_.get(__id); |
| 85 | } |
| 86 | # if _LIBCPP_HAS_LOCALIZATION |
| 87 | _LIBCPP_HIDE_FROM_ABI std::locale locale() { |
| 88 | if (!__loc_) |
| 89 | __loc_ = std::locale{}; |
| 90 | return *__loc_; |
| 91 | } |
| 92 | # endif |
| 93 | _LIBCPP_HIDE_FROM_ABI iterator out() { return std::move(__out_it_); } |
| 94 | _LIBCPP_HIDE_FROM_ABI void advance_to(iterator __it) { __out_it_ = std::move(__it); } |
| 95 | |
| 96 | private: |
| 97 | iterator __out_it_; |
| 98 | basic_format_args<basic_format_context> __args_; |
| 99 | # if _LIBCPP_HAS_LOCALIZATION |
| 100 | |
| 101 | // The Standard doesn't specify how the locale is stored. |
| 102 | // [format.context]/6 |
| 103 | // std::locale locale(); |
| 104 | // Returns: The locale passed to the formatting function if the latter |
| 105 | // takes one, and std::locale() otherwise. |
| 106 | // This is done by storing the locale of the constructor in this optional. If |
| 107 | // locale() is called and the optional has no value the value will be created. |
| 108 | // This allows the implementation to lazily create the locale. |
| 109 | // TODO FMT Validate whether lazy creation is the best solution. |
| 110 | optional<std::locale> __loc_; |
| 111 | |
| 112 | template <class _OtherOutIt, class _OtherCharT> |
| 113 | friend _LIBCPP_HIDE_FROM_ABI basic_format_context<_OtherOutIt, _OtherCharT> __format_context_create( |
| 114 | _OtherOutIt, basic_format_args<basic_format_context<_OtherOutIt, _OtherCharT>>, optional<std::locale>&&); |
| 115 | |
| 116 | // Note: the Standard doesn't specify the required constructors. |
| 117 | _LIBCPP_HIDE_FROM_ABI explicit basic_format_context( |
| 118 | _OutIt __out_it, basic_format_args<basic_format_context> __args, optional<std::locale>&& __loc) |
| 119 | : __out_it_(std::move(__out_it)), __args_(__args), __loc_(std::move(__loc)) {} |
| 120 | # else |
| 121 | template <class _OtherOutIt, class _OtherCharT> |
| 122 | friend _LIBCPP_HIDE_FROM_ABI basic_format_context<_OtherOutIt, _OtherCharT> |
| 123 | __format_context_create(_OtherOutIt, basic_format_args<basic_format_context<_OtherOutIt, _OtherCharT>>); |
| 124 | |
| 125 | _LIBCPP_HIDE_FROM_ABI explicit basic_format_context(_OutIt __out_it, basic_format_args<basic_format_context> __args) |
| 126 | : __out_it_(std::move(__out_it)), __args_(__args) {} |
| 127 | # endif |
| 128 | |
| 129 | public: |
| 130 | basic_format_context(const basic_format_context&) = delete; |
| 131 | basic_format_context& operator=(const basic_format_context&) = delete; |
| 132 | }; |
| 133 | |
| 134 | // A specialization for __retarget_buffer |
| 135 | // |
| 136 | // See __retarget_buffer for the motivation for this specialization. |
| 137 | // |
| 138 | // This context holds a reference to the instance of the basic_format_context |
| 139 | // that is retargeted. It converts a formatting argument when it is requested |
| 140 | // during formatting. It is expected that the usage of the arguments is rare so |
| 141 | // the lookups are not expected to be used often. An alternative would be to |
| 142 | // convert all elements during construction. |
| 143 | // |
| 144 | // The elements of the retargets context are only used when an underlying |
| 145 | // formatter uses a locale specific formatting or an formatting argument is |
| 146 | // part for the format spec. For example |
| 147 | // format("{:256:{}}", input, 8); |
| 148 | // Here the width of an element in input is determined dynamically. |
| 149 | // Note when the top-level element has no width the retargeting is not needed. |
| 150 | template <class _CharT> |
| 151 | class basic_format_context<typename __format::__retarget_buffer<_CharT>::__iterator, _CharT> { |
| 152 | public: |
| 153 | using iterator = typename __format::__retarget_buffer<_CharT>::__iterator; |
| 154 | using char_type = _CharT; |
| 155 | template <class _Tp> |
| 156 | using formatter_type = formatter<_Tp, _CharT>; |
| 157 | |
| 158 | template <class _Context> |
| 159 | _LIBCPP_HIDE_FROM_ABI explicit basic_format_context(iterator __out_it, _Context& __ctx) |
| 160 | : __out_it_(std::move(__out_it)), |
| 161 | # if _LIBCPP_HAS_LOCALIZATION |
| 162 | __loc_([](void* __c) { return static_cast<_Context*>(__c)->locale(); }), |
| 163 | # endif |
| 164 | __ctx_(std::addressof(__ctx)), |
| 165 | __arg_([](void* __c, size_t __id) { |
| 166 | auto __visitor = [&](auto __arg) -> basic_format_arg<basic_format_context> { |
| 167 | if constexpr (same_as<decltype(__arg), monostate>) |
| 168 | return {}; |
| 169 | else if constexpr (same_as<decltype(__arg), typename basic_format_arg<_Context>::handle>) |
| 170 | // At the moment it's not possible for formatting to use a re-targeted handle. |
| 171 | // TODO FMT add this when support is needed. |
| 172 | std::__throw_format_error("Re-targeting handle not supported"); |
| 173 | else |
| 174 | return basic_format_arg<basic_format_context>{ |
| 175 | __format::__determine_arg_t<basic_format_context, decltype(__arg)>(), |
| 176 | __basic_format_arg_value<basic_format_context>(__arg)}; |
| 177 | }; |
| 178 | # if _LIBCPP_STD_VER >= 26 && _LIBCPP_HAS_EXPLICIT_THIS_PARAMETER |
| 179 | return static_cast<_Context*>(__c)->arg(__id).visit(std::move(__visitor)); |
| 180 | # else |
| 181 | _LIBCPP_SUPPRESS_DEPRECATED_PUSH |
| 182 | return std::visit_format_arg(std::move(__visitor), static_cast<_Context*>(__c)->arg(__id)); |
| 183 | _LIBCPP_SUPPRESS_DEPRECATED_POP |
| 184 | # endif // _LIBCPP_STD_VER >= 26 && _LIBCPP_HAS_EXPLICIT_THIS_PARAMETER |
| 185 | }) { |
| 186 | } |
| 187 | |
| 188 | _LIBCPP_HIDE_FROM_ABI basic_format_arg<basic_format_context> arg(size_t __id) const noexcept { |
| 189 | return __arg_(__ctx_, __id); |
| 190 | } |
| 191 | # if _LIBCPP_HAS_LOCALIZATION |
| 192 | _LIBCPP_HIDE_FROM_ABI std::locale locale() { return __loc_(__ctx_); } |
| 193 | # endif |
| 194 | _LIBCPP_HIDE_FROM_ABI iterator out() { return std::move(__out_it_); } |
| 195 | _LIBCPP_HIDE_FROM_ABI void advance_to(iterator __it) { __out_it_ = std::move(__it); } |
| 196 | |
| 197 | private: |
| 198 | iterator __out_it_; |
| 199 | |
| 200 | # if _LIBCPP_HAS_LOCALIZATION |
| 201 | std::locale (*__loc_)(void* __ctx); |
| 202 | # endif |
| 203 | |
| 204 | void* __ctx_; |
| 205 | basic_format_arg<basic_format_context> (*__arg_)(void* __ctx, size_t __id); |
| 206 | }; |
| 207 | |
| 208 | _LIBCPP_CTAD_SUPPORTED_FOR_TYPE(basic_format_context); |
| 209 | #endif // _LIBCPP_STD_VER >= 20 |
| 210 | |
| 211 | _LIBCPP_END_NAMESPACE_STD |
| 212 | |
| 213 | _LIBCPP_POP_MACROS |
| 214 | |
| 215 | #endif // _LIBCPP___FORMAT_FORMAT_CONTEXT_H |
| 216 |
Warning: This file is not a C or C++ file. It does not have highlighting.
