feat: verify classic klotski serial number

verify/classic.py

@@ -171,19 +171,19 @@ class SoluGroup:
 
         for num in range(len(z_pairs)):
             if z_pairs[num][0] == z_pairs[num][1]:
-                z_pairs[num][0].name = f'{self.name}-{num:02}Z'
+                z_pairs[num][0].name = f'{self.name}-{num:03}Z'
             else:
-                z_pairs[num][0].name = f'{self.name}-{num:02}ZL'
-                z_pairs[num][1].name = f'{self.name}-{num:02}ZR'
+                z_pairs[num][0].name = f'{self.name}-{num:03}ZL'
+                z_pairs[num][1].name = f'{self.name}-{num:03}ZR'
 
     def build_lx_index(self) -> dict[str, str]:
         assert self.__name[-1] == 'L'
         right_data = {}
         self.__x_solus.sort(key=lambda x: (-x.ethnic_size, -x.ethnic_deep, x.layout))
         for num in range(len(self.__x_solus)):
-            self.__x_solus[num].name = f'{self.name}-{num:01}X'
+            self.__x_solus[num].name = f'{self.name}-{num:03}X'
             mirror = str(self.__x_solus[num].layout.to_horizontal_mirror())[:7]
-            right_data[mirror] = f'{num:01}Y'
+            right_data[mirror] = f'{num:03}Y'
         return right_data
 
     def build_rx_index(self) -> dict[str, str]:
@@ -191,9 +191,9 @@ class SoluGroup:
         left_data = {}
         self.__x_solus.sort(key=lambda x: (-x.ethnic_size, -x.ethnic_deep, x.layout))
         for num in range(len(self.__x_solus)):
-            self.__x_solus[num].name = f'{self.name}-{num:01}X'
+            self.__x_solus[num].name = f'{self.name}-{num:03}X'
             mirror = str(self.__x_solus[num].layout.to_horizontal_mirror())[:7]
-            left_data[mirror] = f'{num:01}Y'
+            left_data[mirror] = f'{num:03}Y'
         return left_data
 
     def apply_ly_index(self, data: dict[str, str]) -> None:
@@ -211,9 +211,9 @@ class SoluGroup:
         data = {}
         self.__z_solus.sort(key=lambda x: (-x.ethnic_size, -x.ethnic_deep, x.layout))
         for num in range(len(self.__z_solus)):
-            self.__z_solus[num].name = f'{self.name}-{num:01}Z'
+            self.__z_solus[num].name = f'{self.name}-{num:03}Z'
             mirror = str(self.__z_solus[num].layout.to_horizontal_mirror())[:7]
-            data[mirror] = f'{num:01}Z'
+            data[mirror] = f'{num:03}Z'
         return data
 
     def apply_rz_index(self, data: dict[str, str]) -> None:
@@ -381,9 +381,11 @@ def main() -> None:
     for num in range(6):  # 0/1/2/3/4/5
         for solu_pattern in split_patterns(load_valid_solutions(num)):
             solus.extend(solu_pattern.solus)
-            print(solu_pattern, end='\n\n')
-        print('-' * 238)
-    pprint(solus)
+            # print(solu_pattern, end='\n\n')
+        # print('-' * 238)
+
+    data = sorted([(x.code, x.name) for x in solus], key=lambda x: x[1])
+    print(json.dumps(dict(data)))
 
 
 if __name__ == "__main__":

verify/verify.py

@@ -0,0 +1,198 @@
+#!/usr/bin/env python3
+
+import re
+import json
+import functools
+from klotski import Block, Layout
+
+type Info = tuple[int, int, str, int, str]
+
+
+@functools.cache
+def solu_data() -> dict[str, dict]:
+    return json.loads(open('./data/data.json').read())
+
+
+def solu_ethnic_info(code: str) -> tuple[int, int]:  # (ethnic_size, ethnic_deep)
+    cases = [x for x, y in solu_data().items() if code in y['solutions']]
+    steps = [solu_data()[x]['step'] for x in cases]
+    return len(cases), max(steps)
+
+
+def is_ethnic_case(code: Layout) -> bool:
+    seq = code.dump_seq()
+    if seq[13] != Block.B_2x2:
+        return True
+    if seq[12] == Block.SPACE and seq[16] == Block.SPACE:
+        return True
+    if seq[15] == Block.SPACE and seq[19] == Block.SPACE:
+        return True
+    if seq[9] == Block.SPACE and seq[10] == Block.SPACE:
+        return True
+    return False
+
+
+def split_name(code: str, name: str) -> Info:
+    match = re.match(r'^([0-5])-(\d{2})([MLR])-(\d{3})(X|Y|Z|ZL|ZR)$', name)
+    type_id, solu_type = int(match[1]), match[5]
+    assert Layout(code).n_1x2 == type_id
+
+    layout_seq = Layout(code).dump_seq()
+    assert solu_type[0] in ['X', 'Y', 'Z']
+    if solu_type[0] == 'X':
+        assert layout_seq[12] == Block.SPACE and layout_seq[16] == Block.SPACE
+    elif solu_type[0] == 'Y':
+        assert layout_seq[15] == Block.SPACE and layout_seq[19] == Block.SPACE
+    elif solu_type[0] == 'Z':
+        assert layout_seq[9] == Block.SPACE and layout_seq[10] == Block.SPACE
+
+    return type_id, int(match[2]), match[3], int(match[4]), solu_type
+
+
+def split_into_groups(data: dict[str, Info]) -> dict[tuple[int, int, str], dict[str, Info]]:
+    groups = {}
+    for code, info in data.items():
+        name = info[0], info[1], info[2]
+        if name not in groups:
+            groups[name] = {}
+        groups[name][code] = info
+
+    for name, codes in groups.items():
+        group = Layout(list(codes)[0]).group
+        cases = list(group.cases())
+        for code in codes:
+            assert Layout(code) in cases
+
+        if name[2] == 'M':
+            assert group.is_horizontal_mirror
+        else:  # L / R
+            assert not group.is_horizontal_mirror
+
+        if name[2] == 'L':
+            min_solu_l = min(codes.keys())
+            min_solu_r = min(groups[(name[0], name[1], 'R')].keys())
+            assert min_solu_l < min_solu_r
+
+    return groups
+
+
+def verify_m_group(data: dict[str, Info]) -> tuple[int, int, int, str]:
+    for code, info in data.items():
+        assert info[4] in ['X', 'Y', 'Z', 'ZL', 'ZR']
+        if info[4] == 'X':
+            y_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert data[y_code] == (info[0], info[1], info[2], info[3], 'Y')
+        elif info[4] == 'Y':
+            x_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert data[x_code] == (info[0], info[1], info[2], info[3], 'X')
+        elif info[4] == 'Z':
+            assert Layout(code).is_horizontal_mirror
+        elif info[4] == 'ZL':
+            zr_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert code < zr_code
+            assert data[zr_code] == (info[0], info[1], info[2], info[3], 'ZR')
+        elif info[4] == 'ZR':
+            zl_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert zl_code < code
+            assert data[zl_code] == (info[0], info[1], info[2], info[3], 'ZL')
+
+    x_solus = sorted([(x, y[3]) for x, y in data.items() if y[4] == 'X'], key=lambda x: x[1])
+    assert [x[1] for x in x_solus] == list(range(len(x_solus)))
+    x_data = [(*solu_ethnic_info(x[0]), x[0]) for x in x_solus]
+    assert len(set(x_data)) == len(x_data) and x_data == sorted(x_data, key=lambda x: (-x[0], -x[1], x[2]))
+
+    z_solus = sorted([(x, y[3]) for x, y in data.items() if y[4] in {'Z', 'ZL'}], key=lambda x: x[1])
+    assert [x[1] for x in z_solus] == list(range(len(z_solus)))
+    z_data = [(*solu_ethnic_info(x[0]), x[0]) for x in z_solus]
+    assert len(set(z_data)) == len(z_data) and z_data == sorted(z_data, key=lambda x: (-x[0], -x[1], x[2]))
+
+    group = Layout(list(data)[0]).group
+    valid_size = len([x for x in list(group.cases()) if is_ethnic_case(x)])
+    max_ethnic_size = max(x_data[0][0] if x_data else -1, z_data[0][0] if z_data else -1)
+    min_solu = min(list(data))
+    return valid_size, group.size, max_ethnic_size, min_solu
+
+
+def verify_lr_group(l_data: dict[str, Info], r_data: dict[str, Info]) -> tuple[int, int, int, str]:
+    assert len(l_data) == len(r_data)
+    for code, info in l_data.items():
+        assert info[4] in ['X', 'Y', 'Z']
+        if info[4] == 'X':
+            y_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert r_data[y_code] == (info[0], info[1], 'R', info[3], 'Y')
+        elif info[4] == 'Y':
+            x_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert r_data[x_code] == (info[0], info[1], 'R', info[3], 'X')
+        elif info[4] == 'Z':
+            assert not Layout(code).is_horizontal_mirror
+            z_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert r_data[z_code] == (info[0], info[1], 'R', info[3], 'Z')
+    for code, info in r_data.items():
+        assert info[4] in ['X', 'Y', 'Z']
+        if info[4] == 'X':
+            y_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert l_data[y_code] == (info[0], info[1], 'L', info[3], 'Y')
+        elif info[4] == 'Y':
+            x_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert l_data[x_code] == (info[0], info[1], 'L', info[3], 'X')
+        elif info[4] == 'Z':
+            assert not Layout(code).is_horizontal_mirror
+            z_code = str(Layout(code).to_horizontal_mirror())[:7]
+            assert l_data[z_code] == (info[0], info[1], 'L', info[3], 'Z')
+
+    lx_solus = sorted([(x, y[3]) for x, y in l_data.items() if y[4] == 'X'], key=lambda x: x[1])
+    assert [x[1] for x in lx_solus] == list(range(len(lx_solus)))
+    lx_data = [(*solu_ethnic_info(x[0]), x[0]) for x in lx_solus]
+    assert len(set(lx_data)) == len(lx_data) and lx_data == sorted(lx_data, key=lambda x: (-x[0], -x[1], x[2]))
+
+    rx_solus = sorted([(x, y[3]) for x, y in r_data.items() if y[4] == 'X'], key=lambda x: x[1])
+    assert [x[1] for x in rx_solus] == list(range(len(rx_solus)))
+    rx_data = [(*solu_ethnic_info(x[0]), x[0]) for x in rx_solus]
+    assert len(set(rx_data)) == len(rx_data) and rx_data == sorted(rx_data, key=lambda x: (-x[0], -x[1], x[2]))
+
+    lz_solus = sorted([(x, y[3]) for x, y in l_data.items() if y[4] == 'Z'], key=lambda x: x[1])
+    assert [x[1] for x in lz_solus] == list(range(len(lz_solus)))
+    lz_data = [(*solu_ethnic_info(x[0]), x[0]) for x in lz_solus]
+    assert len(set(lz_data)) == len(lz_data) and lz_data == sorted(lz_data, key=lambda x: (-x[0], -x[1], x[2]))
+
+    l_group = Layout(list(l_data)[0]).group
+    r_group = Layout(list(r_data)[0]).group
+    assert l_group.size == r_group.size
+
+    l_valid_size = len([x for x in list(l_group.cases()) if is_ethnic_case(x)])
+    r_valid_size = len([x for x in list(r_group.cases()) if is_ethnic_case(x)])
+    assert l_valid_size == r_valid_size
+
+    max_ethnic_size = max(lx_data[0][0] if lx_data else -1, rx_data[0][0] if rx_data else -1, lz_data[0][0] if lz_data else -1)
+    min_solu = min(list(l_data) + list(r_data))
+    return l_valid_size * 2, l_group.size * 2, max_ethnic_size, min_solu
+
+
+def verify_pattern(info: dict[int, tuple[int, int, int, str]]) -> None:
+    assert list(info.keys()) == list(range(len(info)))
+    info = list(info.values())
+    assert info == sorted(info, key=lambda x: (-x[0], -x[1], -x[2], x[3]))
+
+
+def main() -> None:
+    data = json.loads(open('classic.json').read())
+    data = {x: split_name(x, y) for x, y in data.items()}
+    groups = split_into_groups(data)
+
+    pattern_info = {0: {}, 1: {}, 2: {}, 3: {}, 4: {}, 5: {}}
+    for name, group in groups.items():
+        if name[2] == 'M':
+            pattern_info[name[0]][name[1]] = verify_m_group(group)
+        elif name[2] == 'L':
+            pattern_info[name[0]][name[1]] = verify_lr_group(group, groups[(name[0], name[1], 'R')])
+
+    verify_pattern(pattern_info[0])
+    verify_pattern(pattern_info[1])
+    verify_pattern(pattern_info[2])
+    verify_pattern(pattern_info[3])
+    verify_pattern(pattern_info[4])
+    verify_pattern(pattern_info[5])
+
+
+if __name__ == '__main__':
+    main()