6df9a13e40
This is more consistent with the rest of the LLVM project, and the resulting names are closer to the types defined in each of the headers.
144 lines
2.9 KiB
C++
144 lines
2.9 KiB
C++
//===- BitmaskEnumTest.cpp ------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Test bitmask-enum support APIs.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "orc-rt/BitmaskEnum.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
using namespace orc_rt;
|
|
|
|
namespace {
|
|
|
|
enum Flags { F0 = 0, F1 = 1, F2 = 2, F3 = 4, F4 = 8 };
|
|
|
|
} // namespace
|
|
|
|
namespace orc_rt {
|
|
ORC_RT_DECLARE_ENUM_AS_BITMASK(Flags, F4);
|
|
} // namespace orc_rt
|
|
|
|
static_assert(is_bitmask_enum<Flags>::value != 0);
|
|
static_assert(largest_bitmask_enum_bit<Flags>::value == Flags::F4);
|
|
|
|
namespace {
|
|
|
|
static_assert(is_bitmask_enum<Flags>::value != 0);
|
|
static_assert(largest_bitmask_enum_bit<Flags>::value == Flags::F4);
|
|
|
|
TEST(BitmaskEnumTest, BitwiseOr) {
|
|
Flags f = F1 | F2;
|
|
EXPECT_EQ(3, f);
|
|
|
|
f = f | F3;
|
|
EXPECT_EQ(7, f);
|
|
}
|
|
|
|
TEST(BitmaskEnumTest, BitwiseOrEquals) {
|
|
Flags f = F1;
|
|
f |= F3;
|
|
EXPECT_EQ(5, f);
|
|
|
|
// |= should return a reference to the LHS.
|
|
f = F2;
|
|
(f |= F3) = F1;
|
|
EXPECT_EQ(F1, f);
|
|
}
|
|
|
|
TEST(BitmaskEnumTest, BitwiseAnd) {
|
|
Flags f = static_cast<Flags>(3) & F2;
|
|
EXPECT_EQ(F2, f);
|
|
|
|
f = (f | F3) & (F1 | F2 | F3);
|
|
EXPECT_EQ(6, f);
|
|
}
|
|
|
|
TEST(BitmaskEnumTest, BitwiseAndEquals) {
|
|
Flags f = F1 | F2 | F3;
|
|
f &= F1 | F2;
|
|
EXPECT_EQ(3, f);
|
|
|
|
// &= should return a reference to the LHS.
|
|
(f &= F1) = F3;
|
|
EXPECT_EQ(F3, f);
|
|
}
|
|
|
|
TEST(BitmaskEnumTest, BitwiseXor) {
|
|
Flags f = (F1 | F2) ^ (F2 | F3);
|
|
EXPECT_EQ(5, f);
|
|
|
|
f = f ^ F1;
|
|
EXPECT_EQ(4, f);
|
|
}
|
|
|
|
TEST(BitmaskEnumTest, BitwiseXorEquals) {
|
|
Flags f = (F1 | F2);
|
|
f ^= (F2 | F4);
|
|
EXPECT_EQ(9, f);
|
|
|
|
// ^= should return a reference to the LHS.
|
|
(f ^= F4) = F3;
|
|
EXPECT_EQ(F3, f);
|
|
}
|
|
|
|
TEST(BitmaskEnumTest, ConstantExpression) {
|
|
constexpr Flags f1 = ~F1;
|
|
constexpr Flags f2 = F1 | F2;
|
|
constexpr Flags f3 = F1 & F2;
|
|
constexpr Flags f4 = F1 ^ F2;
|
|
EXPECT_EQ(f1, ~F1);
|
|
EXPECT_EQ(f2, F1 | F2);
|
|
EXPECT_EQ(f3, F1 & F2);
|
|
EXPECT_EQ(f4, F1 ^ F2);
|
|
}
|
|
|
|
TEST(BitmaskEnumTest, BitwiseNot) {
|
|
Flags f = ~F1;
|
|
EXPECT_EQ(14, f); // Largest value for f is 15.
|
|
EXPECT_EQ(15, ~F0);
|
|
}
|
|
|
|
enum class FlagsClass {
|
|
F0 = 0,
|
|
F1 = 1,
|
|
F2 = 2,
|
|
F3 = 4,
|
|
ORC_RT_MARK_AS_BITMASK_ENUM(F3)
|
|
};
|
|
|
|
TEST(BitmaskEnumTest, ScopedEnum) {
|
|
FlagsClass f = (FlagsClass::F1 & ~FlagsClass::F0) | FlagsClass::F2;
|
|
f |= FlagsClass::F3;
|
|
EXPECT_EQ(7, static_cast<int>(f));
|
|
}
|
|
|
|
struct Container {
|
|
enum Flags {
|
|
F0 = 0,
|
|
F1 = 1,
|
|
F2 = 2,
|
|
F3 = 4,
|
|
ORC_RT_MARK_AS_BITMASK_ENUM(F3)
|
|
};
|
|
|
|
static Flags getFlags() {
|
|
Flags f = F0 | F1;
|
|
f |= F2;
|
|
return f;
|
|
}
|
|
};
|
|
|
|
TEST(BitmaskEnumTest, EnumInStruct) { EXPECT_EQ(3, Container::getFlags()); }
|
|
|
|
} // namespace
|