2018-11-26 16:20:40 +00:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <type_traits>
|
|
|
|
#include <Common/Exception.h>
|
|
|
|
#include <DataTypes/NumberTraits.h>
|
|
|
|
|
2020-04-16 12:31:57 +00:00
|
|
|
#if !defined(ARCADIA_BUILD)
|
|
|
|
# include <Common/config.h>
|
|
|
|
#endif
|
2018-11-26 16:20:40 +00:00
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
|
|
|
extern const int ILLEGAL_DIVISION;
|
|
|
|
}
|
|
|
|
|
|
|
|
#pragma GCC diagnostic push
|
|
|
|
#pragma GCC diagnostic ignored "-Wsign-compare"
|
|
|
|
|
|
|
|
template <typename A, typename B>
|
|
|
|
inline void throwIfDivisionLeadsToFPE(A a, B b)
|
|
|
|
{
|
|
|
|
/// Is it better to use siglongjmp instead of checks?
|
|
|
|
|
|
|
|
if (unlikely(b == 0))
|
|
|
|
throw Exception("Division by zero", ErrorCodes::ILLEGAL_DIVISION);
|
|
|
|
|
|
|
|
/// http://avva.livejournal.com/2548306.html
|
2019-11-02 05:55:06 +00:00
|
|
|
if (unlikely(is_signed_v<A> && is_signed_v<B> && a == std::numeric_limits<A>::min() && b == -1))
|
2018-11-26 16:20:40 +00:00
|
|
|
throw Exception("Division of minimal signed number by minus one", ErrorCodes::ILLEGAL_DIVISION);
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename A, typename B>
|
|
|
|
inline bool divisionLeadsToFPE(A a, B b)
|
|
|
|
{
|
|
|
|
if (unlikely(b == 0))
|
|
|
|
return true;
|
|
|
|
|
2019-11-02 05:55:06 +00:00
|
|
|
if (unlikely(is_signed_v<A> && is_signed_v<B> && a == std::numeric_limits<A>::min() && b == -1))
|
2018-11-26 16:20:40 +00:00
|
|
|
return true;
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2020-06-27 22:32:43 +00:00
|
|
|
template <typename A, typename B>
|
|
|
|
inline auto checkedDivision(A a, B b)
|
|
|
|
{
|
|
|
|
throwIfDivisionLeadsToFPE(a, b);
|
|
|
|
return a / b;
|
|
|
|
}
|
|
|
|
|
2018-11-26 16:20:40 +00:00
|
|
|
|
|
|
|
#pragma GCC diagnostic pop
|
|
|
|
|
|
|
|
template <typename A, typename B>
|
|
|
|
struct DivideIntegralImpl
|
|
|
|
{
|
|
|
|
using ResultType = typename NumberTraits::ResultOfIntegerDivision<A, B>::Type;
|
2020-02-14 07:11:37 +00:00
|
|
|
static const constexpr bool allow_fixed_string = false;
|
2018-11-26 16:20:40 +00:00
|
|
|
|
|
|
|
template <typename Result = ResultType>
|
|
|
|
static inline Result apply(A a, B b)
|
|
|
|
{
|
2019-07-31 18:04:24 +00:00
|
|
|
/// Otherwise overflow may occur due to integer promotion. Example: int8_t(-1) / uint64_t(2).
|
|
|
|
/// NOTE: overflow is still possible when dividing large signed number to large unsigned number or vice-versa. But it's less harmful.
|
2019-11-02 05:55:06 +00:00
|
|
|
if constexpr (is_integral_v<A> && is_integral_v<B> && (is_signed_v<A> || is_signed_v<B>))
|
2020-06-27 22:32:43 +00:00
|
|
|
return checkedDivision(std::make_signed_t<A>(a), std::make_signed_t<B>(b));
|
2019-07-31 01:05:33 +00:00
|
|
|
else
|
2020-06-27 22:32:43 +00:00
|
|
|
return checkedDivision(a, b);
|
2018-11-26 16:20:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#if USE_EMBEDDED_COMPILER
|
|
|
|
static constexpr bool compilable = false; /// don't know how to throw from LLVM IR
|
|
|
|
#endif
|
|
|
|
};
|
|
|
|
|
2020-02-25 09:45:23 +00:00
|
|
|
template <typename A, typename B>
|
|
|
|
struct ModuloImpl
|
|
|
|
{
|
|
|
|
using ResultType = typename NumberTraits::ResultOfModulo<A, B>::Type;
|
|
|
|
static const constexpr bool allow_fixed_string = false;
|
|
|
|
|
|
|
|
template <typename Result = ResultType>
|
|
|
|
static inline Result apply(A a, B b)
|
|
|
|
{
|
|
|
|
throwIfDivisionLeadsToFPE(typename NumberTraits::ToInteger<A>::Type(a), typename NumberTraits::ToInteger<B>::Type(b));
|
|
|
|
return typename NumberTraits::ToInteger<A>::Type(a) % typename NumberTraits::ToInteger<B>::Type(b);
|
|
|
|
}
|
|
|
|
|
|
|
|
#if USE_EMBEDDED_COMPILER
|
|
|
|
static constexpr bool compilable = false; /// don't know how to throw from LLVM IR
|
|
|
|
#endif
|
|
|
|
};
|
|
|
|
|
2018-11-26 16:20:40 +00:00
|
|
|
}
|