refactor: refactor enum helper as flag enum.

- refactor enum helper. - add testbench for it.
2025-08-04 22:31:37 +08:00
parent 54134b342e
commit b9f81c16a0
4 changed files with 274 additions and 0 deletions
--- a/testbench/yycc/flag_enum.cpp
+++ b/testbench/yycc/flag_enum.cpp
@ -0,0 +1,77 @@
+#include <gtest/gtest.h>
+#include <yycc.hpp>
+#include <yycc/flag_enum.hpp>
+
+#include <yycc/rust/prelude.hpp>
+
+#define FLAG_ENUM ::yycc::flag_enum
+
+namespace yycctest::flag_enum {
+
+    enum class TestEnum : u8 {
+        Bit1 = 0b00000001,
+        Bit2 = 0b00000010,
+        Bit3 = 0b00000100,
+        Bit4 = 0b00001000,
+        Bit5 = 0b00010000,
+        Bit6 = 0b00100000,
+        Bit7 = 0b01000000,
+        Bit8 = 0b10000000,
+        Empty = 0b00000000,
+        InvBit8 = 0b01111111,
+        MergedBit247 = Bit2 + Bit4 + Bit7,
+    };
+
+    TEST(FlagEnum, Merge) {
+        EXPECT_EQ(FLAG_ENUM::merge(TestEnum::Bit2, TestEnum::Bit4, TestEnum::Bit7), TestEnum::MergedBit247);
+    }
+
+    TEST(FlagEnum, Invert) {
+        EXPECT_EQ(FLAG_ENUM::invert(TestEnum::Bit8), TestEnum::InvBit8);
+    }
+
+    TEST(FlagEnum, Mask) {
+        TestEnum src = FLAG_ENUM::merge(TestEnum::Bit2, TestEnum::Bit4);
+        TestEnum val;
+
+        val = src;
+        FLAG_ENUM::mask(val, TestEnum::Bit2);
+        EXPECT_EQ(val, TestEnum::Bit2);
+
+        val = src;
+        FLAG_ENUM::mask(val, TestEnum::Bit4);
+        EXPECT_EQ(val, TestEnum::Bit4);
+
+        val = src;
+        FLAG_ENUM::mask(val, TestEnum::Bit3);
+        EXPECT_EQ(val, TestEnum::Empty);
+    }
+
+    TEST(FlagEnum, Add) {
+        TestEnum val = TestEnum::Bit2;
+        FLAG_ENUM::add(val, TestEnum::Bit4, TestEnum::Bit7);
+        EXPECT_EQ(val, TestEnum::MergedBit247);
+    }
+
+    TEST(FlagEnum, Remove) {
+        TestEnum val = TestEnum::MergedBit247;
+        FLAG_ENUM::remove(val, TestEnum::Bit2, TestEnum::Bit7);
+        EXPECT_EQ(val, TestEnum::Bit4);
+    }
+
+    TEST(FlagEnum, Has) {
+        TestEnum val = TestEnum::MergedBit247;
+        EXPECT_TRUE(FLAG_ENUM::has(val, TestEnum::Bit2));
+        EXPECT_FALSE(FLAG_ENUM::has(val, TestEnum::Bit3));
+        EXPECT_TRUE(FLAG_ENUM::has(val, TestEnum::Bit4));
+        EXPECT_TRUE(FLAG_ENUM::has(val, TestEnum::Bit7));
+    }
+
+    TEST(FlagEnum, Boolean) {
+        EXPECT_FALSE(FLAG_ENUM::boolean(TestEnum::Empty));
+        EXPECT_TRUE(FLAG_ENUM::boolean(TestEnum::Bit1));
+        EXPECT_TRUE(FLAG_ENUM::boolean(TestEnum::InvBit8));
+        EXPECT_TRUE(FLAG_ENUM::boolean(TestEnum::MergedBit247));
+    }
+
+}