test.py

import time

import cv2
import uiautomator2 as u2

from recognition import img2arr,img2arr_dev
from klotski import klotski99

# d = u2.connect() # connect to device
# d = u2.connect_wifi('192.168.1.3') # connect to device
# print(d.info)
# print(d.serial) #BEWOOZNBYLFYQWHA
# # d.screenshot("home.jpg")
# d.click(0.178, 0.49)

# 读取图像
# image = cv2.imread('gameimg/55/55.jpg')
# image = cv2.imread('game2.jpg')
image = cv2.imread('errgameimg/number_a8efaa8d-e86a-47f8-b976-98b475580cd1.jpg')

# image = cv2.imread('gameimg/99/99_wu_doren.jpg')
# image = cv2.imread('gameimg/99/99_xiandai_shengdan.jpg') # 卡在二分
# image = cv2.imread('gameimg/99/99lan.jpg') # 卡50
# image = cv2.imread('gameimg/99/99_huang_jianke.jpg')

recognition_start_time = time.time()
# numbers = img2arr_dev.img2arr(image, 9)
numbers = img2arr.img2arr(image, 9)
recognition_end_time = time.time()
print(f"图片识别耗时{recognition_end_time - recognition_start_time}秒")

# numbers = [0, 1, 2, 4, 13, 5, 6, 7, 8, 18, 9, 11, 3, 22, 25, 33, 16, 17, 10, 20, 28, 12, 24, 14, 15, 35, 34, 39, 27, 21,
#            32, 30, 51, 23, 26, 43, 19, 29, 47, 46, 31, 41, 50, 59, 44, 36, 45, 37, 38, 68, 61, 42, 77, 53, 54, 55, 74,
#            56, 49, 40, 66, 58, 62, 63, 64, 75, 48, 67, 78, 57, 70, 79, 72, 73, 99, 65, 76, 69, 60, 52, 71]

print(numbers)
sorted_numbers = sorted(numbers)
print(sorted_numbers)

split_count = 9
given_array = list(range(0, split_count * split_count - 1))
given_array.append(99)

# 检查排序后的数组是否等于给定数组
if sorted_numbers != given_array:
    print("识别的图像不正确")
    exit()
else:
    print("识别的图像正确")

paths = klotski99.get_path_warp(numbers)
print(f"移动步数:{len(paths)}")
print(paths)

print("打印移动")
current_state = numbers
for path in paths:
    index = path
    # print(index)

    zero_index = current_state.index(99)
    current_state[zero_index] = current_state[index]
    current_state[index] = 99
    # print(current_state)
print(current_state)

expect_result = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27,
                 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53,
                 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79,
                 99]
if current_state == expect_result:
    print("移动正确")
else:
    print("移动错误")