Add extend and extOrTrunc methods that do sign or zero extension depending on whether...

[oota-llvm.git] / include / llvm / ADT / APSInt.h

diff --git a/include/llvm/ADT/APSInt.h b/include/llvm/ADT/APSInt.h
index 4eb4be231765d948cd29b90ffdd19a2eeaa43204..a2f0321c8bbef95ad24fc28b194126aee573c5a3 100644 (file)
--- 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);
   }