diff --git a/include/llvm/ADT/APInt.h b/include/llvm/ADT/APInt.h index 0e127b8c44d..9043227f0b7 100644 --- a/include/llvm/ADT/APInt.h +++ b/include/llvm/ADT/APInt.h @@ -423,6 +423,16 @@ public: /// @brief Zero extend to a new width. APInt &zext(uint32_t width); + /// Make this APInt have the bit width given by \p width. The value is sign + /// extended, truncated, or left alone to make it that width. + /// @brief Sign extend or truncate to width + APInt &sextOrTrunc(uint32_t width); + + /// Make this APInt have the bit width given by \p width. The value is zero + /// extended, truncated, or left alone to make it that width. + /// @brief Zero extend or truncate to width + APInt &zextOrTrunc(uint32_t width); + /// @brief Set every bit to 1. APInt& set(); diff --git a/lib/Support/APInt.cpp b/lib/Support/APInt.cpp index 93442f344c1..50b0dc34fea 100644 --- a/lib/Support/APInt.cpp +++ b/lib/Support/APInt.cpp @@ -985,6 +985,22 @@ APInt &APInt::zext(uint32_t width) { return *this; } +APInt &APInt::zextOrTrunc(uint32_t width) { + if (BitWidth < width) + return zext(width); + if (BitWidth > width) + return trunc(width); + return *this; +} + +APInt &APInt::sextOrTrunc(uint32_t width) { + if (BitWidth < width) + return sext(width); + if (BitWidth > width) + return trunc(width); + return *this; +} + /// Arithmetic right-shift this APInt by shiftAmt. /// @brief Arithmetic right-shift function. APInt APInt::ashr(uint32_t shiftAmt) const {