port QRDataMaskTest

src/qrcode/cpp_port/test/QRDataMaskTest.rs

@@ -0,0 +1,135 @@
+/*
+* Copyright 2017 Huy Cuong Nguyen
+* Copyright 2007 ZXing authors
+*/
+// SPDX-License-Identifier: Apache-2.0
+
+use crate::{common::BitMatrix, qrcode::cpp_port::data_mask::GetMaskedBit};
+
+#[test]
+fn Mask0() {
+    TestMaskAcrossDimensions(0, |i, j| {
+        return (i + j) % 2 == 0;
+    });
+}
+
+#[test]
+fn Mask1() {
+    TestMaskAcrossDimensions(1, |i, _| {
+        return i % 2 == 0;
+    });
+}
+
+#[test]
+fn Mask2() {
+    TestMaskAcrossDimensions(2, |_, j| {
+        return j % 3 == 0;
+    });
+}
+
+#[test]
+fn Mask3() {
+    TestMaskAcrossDimensions(3, |i, j| {
+        return (i + j) % 3 == 0;
+    });
+}
+
+#[test]
+fn Mask4() {
+    TestMaskAcrossDimensions(4, |i, j| {
+        return (i / 2 + j / 3) % 2 == 0;
+    });
+}
+
+#[test]
+fn Mask5() {
+    TestMaskAcrossDimensions(5, |i, j| {
+        return (i * j) % 2 + (i * j) % 3 == 0;
+    });
+}
+
+#[test]
+fn Mask6() {
+    TestMaskAcrossDimensions(6, |i, j| {
+        return ((i * j) % 2 + (i * j) % 3) % 2 == 0;
+    });
+}
+
+#[test]
+fn Mask7() {
+    TestMaskAcrossDimensions(7, |i, j| {
+        return ((i + j) % 2 + (i * j) % 3) % 2 == 0;
+    });
+}
+
+#[test]
+fn MicroMask0() {
+    TestMicroMaskAcrossDimensions(0, |i, _| {
+        return i % 2 == 0;
+    });
+}
+
+#[test]
+fn MicroMask1() {
+    TestMicroMaskAcrossDimensions(1, |i, j| {
+        return (i / 2 + j / 3) % 2 == 0;
+    });
+}
+
+#[test]
+fn MicroMask2() {
+    TestMicroMaskAcrossDimensions(2, |i, j| {
+        return ((i * j) % 2 + (i * j) % 3) % 2 == 0;
+    });
+}
+
+#[test]
+fn MicroMask3() {
+    TestMicroMaskAcrossDimensions(3, |i, j| {
+        return ((i + j) % 2 + (i * j) % 3) % 2 == 0;
+    });
+}
+
+fn TestMaskAcrossDimensionsImpl<F>(
+    maskIndex: u32,
+    isMicro: bool,
+    versionMax: u32,
+    dimensionStart: u32,
+    dimensionStep: u32,
+    condition: F,
+) where
+    F: Fn(u32, u32) -> bool,
+{
+    for version in 1..=versionMax {
+        // for (int version = 1; version <= versionMax; version++) {
+        let dimension = dimensionStart + dimensionStep * version;
+        let bits = BitMatrix::with_single_dimension(dimension).expect("couldn't create bitmatrix");
+
+        for i in 0..dimension {
+            // for (int i = 0; i < dimension; i++)
+            for j in 0..dimension {
+                // for (int j = 0; j < dimension; j++)
+                assert_eq!(
+                    GetMaskedBit(&bits, j, i, maskIndex, Some(isMicro))
+                        .expect("could not get mask bit"),
+                    condition(i, j),
+                    "({i},{j})"
+                );
+            }
+        }
+    }
+}
+
+fn TestMaskAcrossDimensions<F>(maskIndex: u32, condition: F)
+where
+    F: Fn(u32, u32) -> bool,
+{
+    TestMaskAcrossDimensionsImpl(maskIndex, false, 40, 17, 4, condition);
+}
+
+fn TestMicroMaskAcrossDimensions<F>(maskIndex: u32, condition: F)
+where
+    F: Fn(u32, u32) -> bool,
+{
+    TestMaskAcrossDimensionsImpl(maskIndex, true, 4, 9, 2, condition);
+}