Add extend and extOrTrunc methods that do sign or zero extension depending on whether the integer is signed or not

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@40724 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 16 insertions, 0 deletions

diff --git a/include/llvm/ADT/APSInt.h b/include/llvm/ADT/APSInt.h
index 4eb4be2..a2f0321 100644
--- a/include/llvm/ADT/APSInt.h
+++ b/include/llvm/ADT/APSInt.h
@@ -83,6 +83,22 @@ public:
     return *this;
   }
   
+  APSInt& extend(uint32_t width) {
+    if (IsUnsigned)
+      *this = zext(width);
+    else
+      *this = sext(width);
+    return *this;
+  }
+  
+  APSInt& extOrTrunc(uint32_t width) {
+      if (IsUnsigned)
+        *this = zextOrTrunc(width);
+      else
+        *this = sextOrTrunc(width);
+      return *this;
+  }
+  
   APSInt operator>>(unsigned Amt) const {
     return IsUnsigned ? lshr(Amt) : ashr(Amt);
   }