Index: include/math.h =================================================================== --- include/math.h +++ include/math.h @@ -1553,6 +1553,41 @@ typename std::enable_if::value, double>::type trunc(_A1 __lcpp_x) _NOEXCEPT {return ::trunc((double)__lcpp_x);} +_LIBCPP_BEGIN_NAMESPACE_STD +template ::digits > numeric_limits<_IntT>::digits)> +struct __max_representable_int_for_float; + +template +struct __max_representable_int_for_float<_IntT, _FloatT, true> { + static_assert(is_floating_point<_FloatT>::value, "must be a floating point type"); + static_assert(is_integral<_IntT>::value, "must be an integral type"); + static const _IntT value = numeric_limits<_IntT>::max(); +}; + +template +struct __max_representable_int_for_float<_IntT, _FloatT, false> { + static_assert(is_floating_point<_FloatT>::value, "must be a floating point type"); + static_assert(is_integral<_IntT>::value, "must be an integral type"); + enum { _Bits = numeric_limits<_IntT>::digits - numeric_limits<_FloatT>::digits }; + static const _IntT value = numeric_limits<_IntT>::max() >> _Bits << _Bits; +}; + +template +_LIBCPP_INLINE_VISIBILITY +_IntT __truncating_cast(_RealT __r) _NOEXCEPT { + using _Lim = std::numeric_limits<_IntT>; + const _IntT _MaxVal = __max_representable_int_for_float<_IntT, _RealT>::value; + const _RealT __trunc_r = __builtin_trunc(__r); + if (__trunc_r >= ::nextafter(static_cast<_RealT>(_MaxVal), INFINITY)) { + return _Lim::max(); + } else if (__trunc_r <= _Lim::lowest()) { + return _Lim::min(); + } + return static_cast<_IntT>(__trunc_r); +} +_LIBCPP_END_NAMESPACE_STD + } // extern "C++" #endif // __cplusplus Index: test/libcxx/numerics/truncating_cast.pass.cpp =================================================================== --- /dev/null +++ test/libcxx/numerics/truncating_cast.pass.cpp @@ -0,0 +1,56 @@ +//===----------------------------------------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +// __truncating_cast(RealT) + +// Test the conversion function that truncates floating point types to the +// closes representable value for the specified integer type, or +// numeric_limits::max()/min() if the value isn't representable. + +#include +#include +#include + +template +void test() { + using Lim = std::numeric_limits; + constexpr bool MaxIsRepresentable = sizeof(IntT) < 8; + constexpr bool IsSigned = std::is_signed::value; + struct { + double Input; + IntT Expect; + bool IsRepresentable; + } TestCases[] = { + {0, 0, true}, + {1, 1, true}, + {IsSigned ? static_cast(-1) : 0, + IsSigned ? static_cast(-1) : 0, true}, + {Lim::lowest(), Lim::lowest(), true}, + {static_cast(Lim::max()), Lim::max(), MaxIsRepresentable}, + {static_cast(Lim::max()) + 1, Lim::max(), false}, + {static_cast(Lim::max()) + 1024, Lim::max(), false}, + }; + for (auto TC : TestCases) { + auto res = std::__truncating_cast(TC.Input); + assert(res == TC.Expect); + if (TC.IsRepresentable) { + auto other = static_cast(std::trunc(TC.Input)); + assert(res == other); + } else + assert(res == Lim::min() || res == Lim::max()); + } +} + +int main() { + test(); + test(); + test(); + test(); + test(); + test(); +}