diff options
| author | AlexSm <[email protected]> | 2024-01-04 15:09:05 +0100 |
|---|---|---|
| committer | GitHub <[email protected]> | 2024-01-04 15:09:05 +0100 |
| commit | dab291146f6cd7d35684e3a1150e5bb1c412982c (patch) | |
| tree | 36ef35f6cacb6432845a4a33f940c95871036b32 /contrib/clickhouse/src/Functions/bitShiftRight.cpp | |
| parent | 63660ad5e7512029fd0218e7a636580695a24e1f (diff) | |
Library import 5, delete go dependencies (#832)
* Library import 5, delete go dependencies
* Fix yt client
Diffstat (limited to 'contrib/clickhouse/src/Functions/bitShiftRight.cpp')
| -rw-r--r-- | contrib/clickhouse/src/Functions/bitShiftRight.cpp | 146 |
1 files changed, 0 insertions, 146 deletions
diff --git a/contrib/clickhouse/src/Functions/bitShiftRight.cpp b/contrib/clickhouse/src/Functions/bitShiftRight.cpp deleted file mode 100644 index 21a0f7584aa..00000000000 --- a/contrib/clickhouse/src/Functions/bitShiftRight.cpp +++ /dev/null @@ -1,146 +0,0 @@ -#include <Functions/FunctionFactory.h> -#include <Functions/FunctionBinaryArithmetic.h> -#include <base/hex.h> - -namespace DB -{ -namespace ErrorCodes -{ - extern const int NOT_IMPLEMENTED; - extern const int LOGICAL_ERROR; -} - -namespace -{ - -template <typename A, typename B> -struct BitShiftRightImpl -{ - using ResultType = typename NumberTraits::ResultOfBit<A, B>::Type; - static const constexpr bool allow_fixed_string = false; - static const constexpr bool allow_string_integer = true; - - template <typename Result = ResultType> - static inline NO_SANITIZE_UNDEFINED Result apply(A a [[maybe_unused]], B b [[maybe_unused]]) - { - if constexpr (is_big_int_v<B>) - throw Exception(ErrorCodes::NOT_IMPLEMENTED, "BitShiftRight is not implemented for big integers as second argument"); - else if constexpr (is_big_int_v<A>) - return static_cast<Result>(a) >> static_cast<UInt32>(b); - else - return static_cast<Result>(a) >> static_cast<Result>(b); - } - - static inline NO_SANITIZE_UNDEFINED void bitShiftRightForBytes(const UInt8 * op_pointer, const UInt8 * begin, UInt8 * out, const size_t shift_right_bits) - { - while (op_pointer > begin) - { - op_pointer--; - out--; - *out = *op_pointer >> shift_right_bits; - if (op_pointer - 1 >= begin) - { - /// The right b bit of the left byte is moved to the left b bit of this byte - *out = static_cast<UInt8>(static_cast<UInt8>(*(op_pointer - 1) << (8 - shift_right_bits)) | *out); - } - } - } - - /// For String - static ALWAYS_INLINE NO_SANITIZE_UNDEFINED void apply(const UInt8 * pos [[maybe_unused]], const UInt8 * end [[maybe_unused]], const B & b [[maybe_unused]], ColumnString::Chars & out_vec, ColumnString::Offsets & out_offsets) - { - if constexpr (is_big_int_v<B>) - throw Exception(ErrorCodes::NOT_IMPLEMENTED, "BitShiftRight is not implemented for big integers as second argument"); - else - { - UInt8 word_size = 8; - /// To prevent overflow - if (static_cast<double>(b) >= (static_cast<double>(end - pos) * word_size) || b < 0) - { - /// insert default value - out_vec.push_back(0); - out_offsets.push_back(out_offsets.back() + 1); - return; - } - - size_t shift_right_bytes = b / word_size; - size_t shift_right_bits = b % word_size; - - const UInt8 * begin = pos; - const UInt8 * shift_right_end = end - shift_right_bytes; - - const size_t old_size = out_vec.size(); - size_t length = shift_right_end - begin; - const size_t new_size = old_size + length + 1; - out_vec.resize(new_size); - out_vec[old_size + length] = 0; - - /// We start from the byte on the right and shift right shift_right_bits bit by byte - UInt8 * op_pointer = const_cast<UInt8 *>(shift_right_end); - UInt8 * out = out_vec.data() + old_size + length; - bitShiftRightForBytes(op_pointer, begin, out, shift_right_bits); - out_offsets.push_back(new_size); - } - } - - /// For FixedString - static ALWAYS_INLINE NO_SANITIZE_UNDEFINED void apply(const UInt8 * pos [[maybe_unused]], const UInt8 * end [[maybe_unused]], const B & b [[maybe_unused]], ColumnFixedString::Chars & out_vec) - { - if constexpr (is_big_int_v<B>) - throw Exception(ErrorCodes::NOT_IMPLEMENTED, "BitShiftRight is not implemented for big integers as second argument"); - else - { - UInt8 word_size = 8; - size_t n = end - pos; - /// To prevent overflow - if (static_cast<double>(b) >= (static_cast<double>(n) * word_size) || b < 0) - { - // insert default value - out_vec.resize_fill(out_vec.size() + n); - return; - } - - size_t shift_right_bytes = b / word_size; - size_t shift_right_bits = b % word_size; - - const UInt8 * begin = pos; - const UInt8 * shift_right_end = end - shift_right_bytes; - - const size_t old_size = out_vec.size(); - const size_t new_size = old_size + n; - - /// Fill 0 to the left - out_vec.resize_fill(out_vec.size() + old_size + shift_right_bytes); - out_vec.resize(new_size); - - /// We start from the byte on the right and shift right shift_right_bits bit by byte - UInt8 * op_pointer = const_cast<UInt8 *>(shift_right_end); - UInt8 * out = out_vec.data() + new_size; - bitShiftRightForBytes(op_pointer, begin, out, shift_right_bits); - } - } - -#if USE_EMBEDDED_COMPILER - static constexpr bool compilable = true; - - static inline llvm::Value * compile(llvm::IRBuilder<> & b, llvm::Value * left, llvm::Value * right, bool is_signed) - { - if (!left->getType()->isIntegerTy()) - throw Exception(ErrorCodes::LOGICAL_ERROR, "BitShiftRightImpl expected an integral type"); - return is_signed ? b.CreateAShr(left, right) : b.CreateLShr(left, right); - } -#endif -}; - - -struct NameBitShiftRight { static constexpr auto name = "bitShiftRight"; }; -using FunctionBitShiftRight = BinaryArithmeticOverloadResolver<BitShiftRightImpl, NameBitShiftRight, true, false>; - -} - -REGISTER_FUNCTION(BitShiftRight) -{ - factory.registerFunction<FunctionBitShiftRight>(); -} - -} |
