diff options
| -rw-r--r-- | include/llvm/ADT/APInt.h | 22 | ||||
| -rw-r--r-- | lib/Support/APInt.cpp | 27 |
2 files changed, 49 insertions, 0 deletions
diff --git a/include/llvm/ADT/APInt.h b/include/llvm/ADT/APInt.h index bf3f80c55b..737c5e620f 100644 --- a/include/llvm/ADT/APInt.h +++ b/include/llvm/ADT/APInt.h @@ -584,6 +584,10 @@ public: return shl(Bits); } + APInt operator<<(const APInt &Bits) const { + return shl(Bits); + } + /// Arithmetic right-shift this APInt by shiftAmt. /// @brief Arithmetic right-shift function. APInt ashr(uint32_t shiftAmt) const; @@ -602,6 +606,24 @@ public: /// @brief Rotate right by rotateAmt. APInt rotr(uint32_t rotateAmt) const; + /// Arithmetic right-shift this APInt by shiftAmt. + /// @brief Arithmetic right-shift function. + APInt ashr(const APInt &shiftAmt) const; + + /// Logical right-shift this APInt by shiftAmt. + /// @brief Logical right-shift function. + APInt lshr(const APInt &shiftAmt) const; + + /// Left-shift this APInt by shiftAmt. + /// @brief Left-shift function. + APInt shl(const APInt &shiftAmt) const; + + /// @brief Rotate left by rotateAmt. + APInt rotl(const APInt &rotateAmt) const; + + /// @brief Rotate right by rotateAmt. + APInt rotr(const APInt &rotateAmt) const; + /// Perform an unsigned divide operation on this APInt by RHS. Both this and /// RHS are treated as unsigned quantities for purposes of this division. /// @returns a new APInt value containing the division result diff --git a/lib/Support/APInt.cpp b/lib/Support/APInt.cpp index bc5df9cf47..d5ce5035ac 100644 --- a/lib/Support/APInt.cpp +++ b/lib/Support/APInt.cpp @@ -1085,6 +1085,12 @@ APInt &APInt::sextOrTrunc(uint32_t width) { /// Arithmetic right-shift this APInt by shiftAmt. /// @brief Arithmetic right-shift function. +APInt APInt::ashr(const APInt &shiftAmt) const { + return ashr(shiftAmt.getLimitedValue(BitWidth)); +} + +/// Arithmetic right-shift this APInt by shiftAmt. +/// @brief Arithmetic right-shift function. APInt APInt::ashr(uint32_t shiftAmt) const { assert(shiftAmt <= BitWidth && "Invalid shift amount"); // Handle a degenerate case @@ -1168,6 +1174,12 @@ APInt APInt::ashr(uint32_t shiftAmt) const { /// Logical right-shift this APInt by shiftAmt. /// @brief Logical right-shift function. +APInt APInt::lshr(const APInt &shiftAmt) const { + return ashr(shiftAmt.getLimitedValue(BitWidth)); +} + +/// Logical right-shift this APInt by shiftAmt. +/// @brief Logical right-shift function. APInt APInt::lshr(uint32_t shiftAmt) const { if (isSingleWord()) { if (shiftAmt == BitWidth) @@ -1230,6 +1242,13 @@ APInt APInt::lshr(uint32_t shiftAmt) const { /// Left-shift this APInt by shiftAmt. /// @brief Left-shift function. +APInt APInt::shl(const APInt &shiftAmt) const { + // It's undefined behavior in C to shift by BitWidth or greater, but + return shl(shiftAmt.getLimitedValue(BitWidth)); +} + +/// Left-shift this APInt by shiftAmt. +/// @brief Left-shift function. APInt APInt::shl(uint32_t shiftAmt) const { assert(shiftAmt <= BitWidth && "Invalid shift amount"); if (isSingleWord()) { @@ -1287,6 +1306,10 @@ APInt APInt::shl(uint32_t shiftAmt) const { return APInt(val, BitWidth).clearUnusedBits(); } +APInt APInt::rotl(const APInt &rotateAmt) const { + return rotl(rotateAmt.getLimitedValue(BitWidth)); +} + APInt APInt::rotl(uint32_t rotateAmt) const { if (rotateAmt == 0) return *this; @@ -1298,6 +1321,10 @@ APInt APInt::rotl(uint32_t rotateAmt) const { return hi | lo; } +APInt APInt::rotr(const APInt &rotateAmt) const { + return rotr(rotateAmt.getLimitedValue(BitWidth)); +} + APInt APInt::rotr(uint32_t rotateAmt) const { if (rotateAmt == 0) return *this; |