libcruft-util/cast.hpp

224 lines
6.7 KiB
C++
Raw Normal View History

/*
2018-08-04 15:14:06 +10:00
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*
2018-08-20 15:08:15 +10:00
* Copyright 2011-2018 Danny Robson <danny@nerdcruft.net>
*/
2018-08-20 15:08:15 +10:00
#pragma once
#include "debug.hpp"
2018-08-20 15:08:15 +10:00
#include "platform.hpp"
#include <type_traits>
#include <limits>
namespace cruft::cast {
///////////////////////////////////////////////////////////////////////////
/// Safely cast a numeric type to its (un)signed counterpart, aborting if
/// the dynamically checked result is not representable. May be optimised
/// out if NDEBUG is defined.
///
/// The signed/unsigned and unsigned/signed cases are split so we can
/// simplify the out of range tests.
///
/// The same-type case is not provided because we want to error out on
/// unnecessary casts.
template <
typename T,
typename U
>
std::enable_if_t<
sizeof(T) == sizeof(U) &&
std::is_unsigned<T>::value &&
std::is_signed<U>::value,
T
>
sign (const U u)
{
CHECK_GE (u, 0);
return static_cast<T> (u);
}
//-------------------------------------------------------------------------
template <
typename T,
typename U
>
std::enable_if_t<
sizeof(T) == sizeof (U) &&
std::is_signed<T>::value &&
std::is_unsigned<U>::value,
T
>
sign (const U u)
{
CHECK_LT (u, std::numeric_limits<U>::max () / 2);
return static_cast<T> (u);
}
2012-06-08 16:46:03 +10:00
///////////////////////////////////////////////////////////////////////////
/// Cast to a smaller type of the same signedness and realness and assert
/// that both values are still equal.
///
/// Any runtime checks will be compiled out if NDEBUG is defined.
///
/// Identity casts are allowed so as to simplify the use of this routine
/// in template code.
template <
typename NarrowT,
typename WideT,
typename = std::enable_if_t<
std::is_arithmetic_v<NarrowT> &&
std::is_arithmetic_v<WideT> &&
std::is_signed_v<NarrowT> == std::is_signed_v<WideT> &&
std::is_floating_point_v<NarrowT> == std::is_floating_point_v<WideT> &&
sizeof (NarrowT) <= sizeof (WideT),
void
>
>
constexpr NarrowT
narrow (const WideT &val)
{
static_assert (sizeof (NarrowT) <= sizeof (WideT));
#ifndef NDEBUG
auto narrow = static_cast<NarrowT> (val);
CHECK_EQ (narrow, val);
return narrow;
#else
return static_cast<NarrowT> (val);
#endif
}
///////////////////////////////////////////////////////////////////////////
/// Cast between types checking that exact equality holds if the result is
/// casted back to the original type.
///
/// Runtime checks will be compiled out if NDEBUG is defined.
template <typename DstT, typename SrcT>
constexpr DstT
lossless (const SrcT &src)
{
#ifndef NDEBUG
auto dst = static_cast<DstT> (src);
if constexpr (std::is_floating_point_v<SrcT>) {
if (std::isnan (src)) {
// NaNs must remaing as NaN's. They're important.
CHECK (std::is_floating_point_v<DstT>);
CHECK (std::isnan (dst));
}
}
// Cast dst back to src to check round-trip conversion
// is lossless.
CHECK_EQ (static_cast<SrcT> (dst), src);
return dst;
#else
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wshorten-64-to-32"
if constexpr (std::is_enum_v<DstT>) {
auto const val = static_cast<std::underlying_type_t<DstT>> (src);
return static_cast<DstT> (val);
} else {
return static_cast<DstT> (src);
}
#pragma GCC diagnostic pop
#endif
}
2012-06-20 16:48:40 +10:00
///////////////////////////////////////////////////////////////////////////
/// Assert if the value is not a pointer to a subclass of T, else return
/// the converted value. Note: this is only a debug-time check and is
/// compiled out in optimised builds.
template <
typename T,
typename V
>
std::enable_if_t<std::is_pointer_v<T>,T>
known (V *const v)
{
CHECK (dynamic_cast<T> (v));
return static_cast<T> (v);
}
//-------------------------------------------------------------------------
template <
typename T,
typename V
>
std::enable_if_t<std::is_reference_v<T>, T>
known (V &v)
{
CHECK_NOTHROW (dynamic_cast<T> (v));
return reinterpret_cast<T> (v);
}
2018-05-09 17:47:47 +10:00
///////////////////////////////////////////////////////////////////////////
/// Cast a pointer from one type to another, asserting that the required
2018-05-09 17:47:47 +10:00
/// alignment of the destination type has been satisfied.
///
/// Runtime checks will be compiled out if NDEBUG is defined.
template <
typename DstT,
typename SrcT,
typename = std::enable_if_t<std::is_pointer_v<DstT> && std::is_pointer_v<SrcT>>
>
DstT
alignment (SrcT src)
2018-05-09 17:47:47 +10:00
{
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wcast-align"
CHECK_MOD (reinterpret_cast<uintptr_t> (src), alignof (std::remove_pointer_t<DstT>));
return reinterpret_cast<DstT> (src);
2018-05-09 17:47:47 +10:00
#pragma GCC diagnostic pop
}
2018-05-10 13:52:49 +10:00
///////////////////////////////////////////////////////////////////////////
/// Cast from SrcT to DstT and damn any consequences; just make it compile.
2018-05-10 13:52:49 +10:00
template <typename DstT, typename SrcT>
DstT ffs (SrcT src)
{
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wcast-align"
2018-08-20 15:08:15 +10:00
#if defined(COMPILER_GCC)
#pragma GCC diagnostic ignored "-Wcast-function-type"
#endif
// If we're casting a pointer then we may need to first use a
// const_cast if we want to avoid any warnings about casting, eg in
// the case of `const ValueT*` to `ValueT*`
if constexpr (std::is_pointer_v<SrcT>) {
using value_type = std::remove_pointer_t<SrcT>;
return reinterpret_cast<DstT> (
const_cast<std::remove_cv_t<value_type>*> (src)
);
} else {
return reinterpret_cast<DstT> (src);
}
2018-05-10 13:52:49 +10:00
#pragma GCC diagnostic pop
}
2018-12-03 15:29:04 +11:00
///////////////////////////////////////////////////////////////////////////
/// Cast from SrcT to DstT, performing sanity checks on the src and dst
/// values before returning the result.
template <typename DstT, typename SrcT>
DstT sanity (SrcT src)
{
cruft::debug::sanity (src);
DstT dst = static_cast<DstT> (src);
cruft::debug::sanity (dst);
return dst;
}
2018-05-10 13:52:49 +10:00
}