[block-freq] Add the APInt method extractBit.

[oota-llvm.git] / lib / Support / APInt.cpp

diff --git a/lib/Support/APInt.cpp b/lib/Support/APInt.cpp
index 89f96bd5774002b3956d409e244bf4467c795e73..731c8cc9cae20e78e08059969e6196b5d3d7e6d3 100644 (file)
--- a/lib/Support/APInt.cpp
+++ b/lib/Support/APInt.cpp
@@ -607,6 +607,14 @@ void APInt::flipBit(unsigned bitPosition) {
   else setBit(bitPosition);
 }
 
+bool APInt::extractBit(unsigned bitPosition) const {
+  assert(bitPosition < BitWidth && "Out of the bit-width range!");
+  if (isSingleWord())
+    return VAL & maskBit(bitPosition);
+  else
+    return pVal[whichWord(bitPosition)] & maskBit(bitPosition);
+}
+
 unsigned APInt::getBitsNeeded(StringRef str, uint8_t radix) {
   assert(!str.empty() && "Invalid string length");
   assert((radix == 10 || radix == 8 || radix == 16 || radix == 2 ||