103 lines
3.0 KiB
C++
103 lines
3.0 KiB
C++
//===- llvm/unittest/ADT/PointerIntPairTest.cpp - Unit tests --------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/PointerIntPair.h"
|
|
#include "gtest/gtest.h"
|
|
#include <limits>
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
TEST(PointerIntPairTest, GetSet) {
|
|
struct S {
|
|
int i;
|
|
};
|
|
S s;
|
|
|
|
PointerIntPair<S *, 2> Pair(&s, 1U);
|
|
EXPECT_EQ(&s, Pair.getPointer());
|
|
EXPECT_EQ(1U, Pair.getInt());
|
|
|
|
Pair.setInt(2);
|
|
EXPECT_EQ(&s, Pair.getPointer());
|
|
EXPECT_EQ(2U, Pair.getInt());
|
|
|
|
Pair.setPointer(nullptr);
|
|
EXPECT_EQ(nullptr, Pair.getPointer());
|
|
EXPECT_EQ(2U, Pair.getInt());
|
|
|
|
Pair.setPointerAndInt(&s, 3U);
|
|
EXPECT_EQ(&s, Pair.getPointer());
|
|
EXPECT_EQ(3U, Pair.getInt());
|
|
|
|
// Make sure that we can perform all of our operations on enum classes.
|
|
//
|
|
// The concern is that enum classes are only explicitly convertible to
|
|
// integers. This means that if we assume in PointerIntPair this, a
|
|
// compilation error will result. This group of tests exercises the enum class
|
|
// code to make sure that we do not run into such issues in the future.
|
|
enum class E : unsigned {
|
|
Case1,
|
|
Case2,
|
|
Case3,
|
|
};
|
|
PointerIntPair<S *, 2, E> Pair2(&s, E::Case1);
|
|
EXPECT_EQ(&s, Pair2.getPointer());
|
|
EXPECT_EQ(E::Case1, Pair2.getInt());
|
|
|
|
Pair2.setInt(E::Case2);
|
|
EXPECT_EQ(&s, Pair2.getPointer());
|
|
EXPECT_EQ(E::Case2, Pair2.getInt());
|
|
|
|
Pair2.setPointer(nullptr);
|
|
EXPECT_EQ(nullptr, Pair2.getPointer());
|
|
EXPECT_EQ(E::Case2, Pair2.getInt());
|
|
|
|
Pair2.setPointerAndInt(&s, E::Case3);
|
|
EXPECT_EQ(&s, Pair2.getPointer());
|
|
EXPECT_EQ(E::Case3, Pair2.getInt());
|
|
}
|
|
|
|
TEST(PointerIntPairTest, DefaultInitialize) {
|
|
PointerIntPair<float *, 2> Pair;
|
|
EXPECT_EQ(nullptr, Pair.getPointer());
|
|
EXPECT_EQ(0U, Pair.getInt());
|
|
}
|
|
|
|
TEST(PointerIntPairTest, ManyUnusedBits) {
|
|
// In real code this would be a word-sized integer limited to 31 bits.
|
|
struct Fixnum31 {
|
|
uintptr_t Value;
|
|
};
|
|
class FixnumPointerTraits {
|
|
public:
|
|
static inline void *getAsVoidPointer(Fixnum31 Num) {
|
|
return reinterpret_cast<void *>(Num.Value << NumLowBitsAvailable);
|
|
}
|
|
static inline Fixnum31 getFromVoidPointer(void *P) {
|
|
// In real code this would assert that the value is in range.
|
|
return { reinterpret_cast<uintptr_t>(P) >> NumLowBitsAvailable };
|
|
}
|
|
enum { NumLowBitsAvailable = std::numeric_limits<uintptr_t>::digits - 31 };
|
|
};
|
|
|
|
PointerIntPair<Fixnum31, 1, bool, FixnumPointerTraits> pair;
|
|
EXPECT_EQ((uintptr_t)0, pair.getPointer().Value);
|
|
EXPECT_FALSE(pair.getInt());
|
|
|
|
pair.setPointerAndInt({ 0x7FFFFFFF }, true );
|
|
EXPECT_EQ((uintptr_t)0x7FFFFFFF, pair.getPointer().Value);
|
|
EXPECT_TRUE(pair.getInt());
|
|
|
|
EXPECT_EQ(FixnumPointerTraits::NumLowBitsAvailable - 1,
|
|
PointerLikeTypeTraits<decltype(pair)>::NumLowBitsAvailable);
|
|
}
|
|
|
|
} // end anonymous namespace
|