From 4d99506b4c3c87fc4d34932cfeaa59e09d54bf12 Mon Sep 17 00:00:00 2001
From: Merry <git@mary.rs>
Date: Sun, 3 Jul 2022 17:48:14 +0100
Subject: [PATCH] UnitTests/MovI2R: Test all logical immediates

---
 .../Core/PowerPC/JitArm64/MovI2R.cpp          | 31 +++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/Source/UnitTests/Core/PowerPC/JitArm64/MovI2R.cpp b/Source/UnitTests/Core/PowerPC/JitArm64/MovI2R.cpp
index 125e3de963..889878d43a 100644
--- a/Source/UnitTests/Core/PowerPC/JitArm64/MovI2R.cpp
+++ b/Source/UnitTests/Core/PowerPC/JitArm64/MovI2R.cpp
@@ -79,6 +79,37 @@ TEST(JitArm64, MovI2R_Rand)
   }
 }
 
+// Construct and test every 64-bit logical immediate
+TEST(JitArm64, MovI2R_LogImm)
+{
+  TestMovI2R test;
+
+  for (unsigned size = 2; size <= 64; size *= 2)
+  {
+    for (unsigned length = 1; length < size; ++length)
+    {
+      u64 imm = ~u64{0} >> (64 - length);
+      for (unsigned e = size; e < 64; e *= 2)
+      {
+        imm |= imm << e;
+      }
+      for (unsigned rotation = 0; rotation < size; ++rotation)
+      {
+        test.Check64(imm);
+        EXPECT_EQ(static_cast<bool>(LogicalImm(imm, 64)), true);
+
+        if (size < 64)
+        {
+          test.Check32(imm);
+          EXPECT_EQ(static_cast<bool>(LogicalImm(static_cast<u32>(imm), 32)), true);
+        }
+
+        imm = (imm >> 63) | (imm << 1);
+      }
+    }
+  }
+}
+
 TEST(JitArm64, MovI2R_ADP)
 {
   TestMovI2R test;