feat: add AreAnyBitsSet() and AreAllBitsSet() (#823)

Add AreAnyBitsSet and AreAllBitsSet
philips-software · Feb 13, 2025 · f83c37b · f83c37b
1 parent 60a0b42
commit f83c37b
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/infra/util/BitLogic.hpp b/infra/util/BitLogic.hpp
@@ -49,6 +49,18 @@ namespace infra
         return (dataRegister & Bit<T>(index)) != 0;
     }
 
+    template<class T, class U>
+    bool AreAnyBitsSet(T dataRegister, const U& mask)
+    {
+        return (dataRegister & mask) != 0;
+    }
+
+    template<class T, class U>
+    bool AreAllBitsSet(T dataRegister, const U& mask)
+    {
+        return (dataRegister & mask) == mask;
+    }
+
     template<class T>
     T GetBits(T dataRegister, uint8_t size, uint8_t position)
     {

diff --git a/infra/util/test/TestBitLogic.cpp b/infra/util/test/TestBitLogic.cpp
@@ -1,4 +1,5 @@
 #include "infra/util/BitLogic.hpp"
+#include "gmock/gmock.h"
 #include "gtest/gtest.h"
 
 TEST(BitLogicTest, Bit)
@@ -52,3 +53,23 @@ TEST(BitLogicTest, GetBits)
     uint32_t x = 2;
     EXPECT_EQ(2, infra::GetBits(x, 1, 1));
 }
+
+TEST(BitLogicTest, AreAnyBitsSet)
+{
+    const uint32_t x = 0b10000001;
+    const uint32_t y = 0b10000000;
+    const uint32_t mask = 0b1001;
+
+    EXPECT_THAT(infra::AreAnyBitsSet(x, mask), testing::IsTrue());
+    EXPECT_THAT(infra::AreAnyBitsSet(y, mask), testing::IsFalse());
+}
+
+TEST(BitLogicTest, AreAllBitsSet)
+{
+    const uint32_t x = 0b10001001;
+    const uint32_t y = 0b10001000;
+    const uint32_t mask = 0b1001;
+
+    EXPECT_THAT(infra::AreAllBitsSet(x, mask), testing::IsTrue());
+    EXPECT_THAT(infra::AreAllBitsSet(y, mask), testing::IsFalse());
+}