11
11
#define _LIBCPP___FORMAT_FORMAT_FUNCTIONS
12
12
13
13
#include < __algorithm/clamp.h>
14
+ #include < __algorithm/ranges_find_first_of.h>
15
+ #include < __chrono/statically_widen.h>
14
16
#include < __concepts/convertible_to.h>
15
17
#include < __concepts/same_as.h>
16
18
#include < __config>
@@ -447,10 +449,47 @@ format_to(_OutIt __out_it, wformat_string<_Args...> __fmt, _Args&&... __args) {
447
449
}
448
450
# endif
449
451
452
+ // Try constant folding the format string instead of going through the whole formatting machinery. If there is no
453
+ // constant folding no extra code should be emitted (with optimizations enabled) and the function returns nullopt. When
454
+ // constant folding is successful, the formatting is performed and the resulting string is returned.
455
+ namespace __format {
456
+ template <class _CharT >
457
+ [[nodiscard]] _LIBCPP_HIDE_FROM_ABI optional<basic_string<_CharT>> __try_constant_folding (
458
+ basic_string_view<_CharT> __fmt,
459
+ basic_format_args<basic_format_context<back_insert_iterator<__format::__output_buffer<_CharT>>, _CharT>> __args) {
460
+ // Fold strings not containing '{' or '}' to just return the string
461
+ if (bool __is_identity = [&] [[__gnu__::__pure__]] // Make sure the compiler knows this call can be eliminated
462
+ { return std::ranges::find_first_of (__fmt, array{' {' , ' }' }) == __fmt.end (); }();
463
+ __builtin_constant_p (__is_identity) && __is_identity)
464
+ return basic_string<_CharT>{__fmt};
465
+
466
+ // Fold '{}' to the appropriate conversion function
467
+ if (auto __only_first_arg = __fmt == _LIBCPP_STATICALLY_WIDEN (_CharT, " {}" );
468
+ __builtin_constant_p (__only_first_arg) && __only_first_arg) {
469
+ if (auto __arg = __args.get (0 ); __builtin_constant_p (__arg.__type_ )) {
470
+ return std::__visit_format_arg (
471
+ []<class _Tp >(_Tp&& __argument) -> optional<basic_string<_CharT>> {
472
+ if constexpr (is_same_v<remove_cvref_t <_Tp>, basic_string_view<_CharT>>) {
473
+ return basic_string<_CharT>{__argument};
474
+ } else {
475
+ return nullopt;
476
+ }
477
+ },
478
+ __arg);
479
+ }
480
+ }
481
+
482
+ return nullopt;
483
+ }
484
+ } // namespace __format
485
+
450
486
// TODO FMT This needs to be a template or std::to_chars(floating-point) availability markup
451
487
// fires too eagerly, see http://llvm.org/PR61563.
452
488
template <class = void >
453
489
[[nodiscard]] _LIBCPP_ALWAYS_INLINE inline _LIBCPP_HIDE_FROM_ABI string vformat (string_view __fmt, format_args __args) {
490
+ auto __result = __format::__try_constant_folding (__fmt, __args);
491
+ if (__result.has_value ())
492
+ return *std::move (__result);
454
493
__format::__allocating_buffer<char > __buffer;
455
494
std::vformat_to (__buffer.__make_output_iterator (), __fmt, __args);
456
495
return string{__buffer.__view ()};
@@ -462,6 +501,9 @@ template <class = void>
462
501
template <class = void >
463
502
[[nodiscard]] _LIBCPP_ALWAYS_INLINE inline _LIBCPP_HIDE_FROM_ABI wstring
464
503
vformat (wstring_view __fmt, wformat_args __args) {
504
+ auto __result = __format::__try_constant_folding (__fmt, __args);
505
+ if (__result.has_value ())
506
+ return *std::move (__result);
465
507
__format::__allocating_buffer<wchar_t > __buffer;
466
508
std::vformat_to (__buffer.__make_output_iterator (), __fmt, __args);
467
509
return wstring{__buffer.__view ()};
0 commit comments