mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-18 13:42:02 +00:00
56 lines
1.7 KiB
C++
56 lines
1.7 KiB
C++
#include <ext/bit_cast.h>
|
|
#include <Functions/FunctionFactory.h>
|
|
#include <Functions/FunctionUnaryArithmetic.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
template <typename A>
|
|
struct BitCountImpl
|
|
{
|
|
using ResultType = UInt8;
|
|
static constexpr bool allow_fixed_string = false;
|
|
|
|
static inline ResultType apply(A a)
|
|
{
|
|
/// We count bits in the value representation in memory. For example, we support floats.
|
|
/// We need to avoid sign-extension when converting signed numbers to larger type. So, uint8_t(-1) has 8 bits.
|
|
|
|
if constexpr (std::is_same_v<A, UInt64> || std::is_same_v<A, Int64>)
|
|
return __builtin_popcountll(a);
|
|
if constexpr (std::is_same_v<A, UInt32> || std::is_same_v<A, Int32> || std::is_unsigned_v<A>)
|
|
return __builtin_popcount(a);
|
|
if constexpr (std::is_same_v<A, Int16>)
|
|
return __builtin_popcount(static_cast<UInt16>(a));
|
|
if constexpr (std::is_same_v<A, Int8>)
|
|
return __builtin_popcount(static_cast<UInt8>(a));
|
|
else
|
|
return __builtin_popcountll(ext::bit_cast<unsigned long long>(a));
|
|
}
|
|
|
|
#if USE_EMBEDDED_COMPILER
|
|
static constexpr bool compilable = false;
|
|
#endif
|
|
};
|
|
|
|
struct NameBitCount { static constexpr auto name = "bitCount"; };
|
|
using FunctionBitCount = FunctionUnaryArithmetic<BitCountImpl, NameBitCount, false /* is injective */>;
|
|
|
|
/// The function has no ranges of monotonicity.
|
|
template <> struct FunctionUnaryArithmeticMonotonicity<NameBitCount>
|
|
{
|
|
static bool has() { return false; }
|
|
static IFunction::Monotonicity get(const Field &, const Field &)
|
|
{
|
|
return {};
|
|
}
|
|
};
|
|
|
|
void registerFunctionBitCount(FunctionFactory & factory)
|
|
{
|
|
factory.registerFunction<FunctionBitCount>();
|
|
}
|
|
|
|
}
|