调整项目结构。修改工具类逻辑
This commit is contained in:
zw
149
Utils/AiUtils.py
149
Utils/AiUtils.py
@@ -3,6 +3,9 @@ import os
|
||||
import re
|
||||
import cv2
|
||||
import numpy as np
|
||||
import wda
|
||||
|
||||
from Utils.LogManager import LogManager
|
||||
|
||||
|
||||
# 工具类
|
||||
@@ -10,42 +13,47 @@ class AiUtils(object):
|
||||
|
||||
# 在屏幕中找到对应的图片
|
||||
@classmethod
|
||||
def findImageInScreen(cls, target):
|
||||
# 加载原始图像和模板图像
|
||||
image_path = AiUtils.imagePathWithName("bgv") # 替换为你的图像路径
|
||||
template_path = AiUtils.imagePathWithName(target) # 替换为你的模板路径
|
||||
def findImageInScreen(cls, target, udid):
|
||||
try:
|
||||
# 加载原始图像和模板图像
|
||||
image_path = AiUtils.imagePathWithName(udid,"bgv") # 替换为你的图像路径
|
||||
template_path = AiUtils.imagePathWithName("", target) # 替换为你的模板路径
|
||||
|
||||
# 读取图像和模板，确保它们都是单通道灰度图
|
||||
image = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
|
||||
template = cv2.imread(template_path, cv2.IMREAD_GRAYSCALE)
|
||||
# 读取图像和模板，确保它们都是单通道灰度图
|
||||
image = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
|
||||
template = cv2.imread(template_path, cv2.IMREAD_GRAYSCALE)
|
||||
|
||||
if image is None:
|
||||
raise ValueError("背景图无法加载")
|
||||
if image is None:
|
||||
LogManager.error("加载背景图失败")
|
||||
return -1, -1
|
||||
|
||||
if template is None:
|
||||
raise ValueError("模板图无法加载")
|
||||
if template is None:
|
||||
LogManager.error("加载模板图失败")
|
||||
return -1, -1
|
||||
|
||||
# 获取模板的宽度和高度
|
||||
w, h = template.shape[::-1]
|
||||
|
||||
# 获取模板的宽度和高度
|
||||
w, h = template.shape[::-1]
|
||||
|
||||
# 使用模板匹配方法
|
||||
res = cv2.matchTemplate(image, template, cv2.TM_CCOEFF_NORMED)
|
||||
threshold = 0.7 # 匹配度阈值，可以根据需要调整
|
||||
loc = np.where(res >= threshold)
|
||||
|
||||
# 检查是否有匹配结果
|
||||
if loc[0].size > 0:
|
||||
# 取第一个匹配位置
|
||||
pt = zip(*loc[::-1]).__next__() # 获取第一个匹配点的坐标
|
||||
center_x = int(pt[0] + w // 2)
|
||||
center_y = int(pt[1] + h // 2)
|
||||
# print(f"第一个匹配到的小心心中心坐标: ({center_x}, {center_y})")
|
||||
return center_x, center_y
|
||||
else:
|
||||
print("未找到匹配的目标")
|
||||
return -1, -1
|
||||
# 使用模板匹配方法
|
||||
res = cv2.matchTemplate(image, template, cv2.TM_CCOEFF_NORMED)
|
||||
threshold = 0.7 # 匹配度阈值，可以根据需要调整
|
||||
loc = np.where(res >= threshold)
|
||||
|
||||
# 检查是否有匹配结果
|
||||
if loc[0].size > 0:
|
||||
print("匹配到目标了")
|
||||
# 取第一个匹配位置
|
||||
pt = zip(*loc[::-1]).__next__() # 获取第一个匹配点的坐标
|
||||
center_x = int(pt[0] + w // 2)
|
||||
center_y = int(pt[1] + h // 2)
|
||||
# print(f"第一个匹配到的小心心中心坐标: ({center_x}, {center_y})")
|
||||
return center_x, center_y
|
||||
else:
|
||||
print("未找到匹配的目标")
|
||||
return -1, -1
|
||||
except Exception as e:
|
||||
LogManager.error(f"加载素材失败：{e}", udid)
|
||||
print(e)
|
||||
|
||||
# 使用正则查找字符串中的数字
|
||||
@classmethod
|
||||
@@ -60,7 +68,11 @@ class AiUtils(object):
|
||||
# 选择截图
|
||||
@classmethod
|
||||
def screenshot(cls):
|
||||
image_path = AiUtils.imagePathWithName("bgv") # 替换为你的图像路径
|
||||
client = wda.USBClient("eca000fcb6f55d7ed9b4c524055214c26a7de7aa")
|
||||
session = client.session()
|
||||
image = session.screenshot()
|
||||
image_path = "screenshot.png"
|
||||
image.save(image_path)
|
||||
image = cv2.imread(image_path)
|
||||
|
||||
# 如果图像过大，缩小显示
|
||||
@@ -133,15 +145,84 @@ class AiUtils(object):
|
||||
|
||||
# 根据名称获取图片完整地址
|
||||
@classmethod
|
||||
def imagePathWithName(cls, name):
|
||||
def imagePathWithName(cls, udid, name):
|
||||
current_file_path = os.path.abspath(__file__)
|
||||
# 获取当前文件所在的目录（即script目录）
|
||||
current_dir = os.path.dirname(current_file_path)
|
||||
# 由于script目录位于项目根目录下一级，因此需要向上一级目录移动两次
|
||||
project_root = os.path.abspath(os.path.join(current_dir, '..'))
|
||||
# 构建资源文件的完整路径，向上两级目录，然后进入 resources 目录
|
||||
resource_path = os.path.abspath(os.path.join(project_root, 'resources', name + ".png")).replace('/', '\\')
|
||||
resource_path = os.path.abspath(os.path.join(project_root, "resources", udid, name + ".png")).replace('/', '\\')
|
||||
return resource_path
|
||||
|
||||
# 获取根目录
|
||||
@classmethod
|
||||
def getRootDir(cls):
|
||||
current_file = os.path.abspath(__file__)
|
||||
# 获取当前文件所在的目录
|
||||
current_dir = os.path.dirname(current_file)
|
||||
# 获取项目根目录（假设根目录是当前文件的父目录的父目录）
|
||||
project_root = os.path.dirname(current_dir)
|
||||
# 返回根目录
|
||||
return project_root
|
||||
|
||||
|
||||
# 创建一个以udid命名的目录
|
||||
@classmethod
|
||||
def makeUdidDir(cls, udid):
|
||||
# 获取项目根目录
|
||||
home = cls.getRootDir()
|
||||
# 拼接 resources 目录的路径
|
||||
resources_dir = os.path.join(home, "resources")
|
||||
# 拼接 udid 目录的路径
|
||||
udid_dir = os.path.join(resources_dir, udid)
|
||||
# 检查 udid 目录是否存在，如果不存在则创建
|
||||
if not os.path.exists(udid_dir):
|
||||
try:
|
||||
os.makedirs(udid_dir)
|
||||
LogManager.info(f"目录 {udid_dir} 创建成功", udid)
|
||||
print(f"目录 {udid_dir} 创建成功")
|
||||
except Exception as e:
|
||||
print(f"创建目录时出错: {e}")
|
||||
LogManager.error(f"创建目录时出错: {e}", udid)
|
||||
else:
|
||||
LogManager.info(f"目录 {udid_dir} 已存在，跳过创建", udid)
|
||||
print(f"目录 {udid_dir} 已存在，跳过创建")
|
||||
|
||||
# 查找首页按钮
|
||||
# uuid 设备id
|
||||
# click 是否点击该按钮
|
||||
@classmethod
|
||||
def findHomeButton(cls, udid="eca000fcb6f55d7ed9b4c524055214c26a7de7aa"):
|
||||
client = wda.USBClient(udid)
|
||||
session = client.session()
|
||||
session.appium_settings({"snapshotMaxDepth": 10})
|
||||
homeButton = session.xpath("//*[@label='首页']")
|
||||
try:
|
||||
if homeButton.label == "首页":
|
||||
print("找到了")
|
||||
return homeButton
|
||||
else:
|
||||
print("没找到")
|
||||
return None
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return None
|
||||
|
||||
|
||||
# 查找关闭按钮
|
||||
@classmethod
|
||||
def findCloseButton(cls,udid="eca000fcb6f55d7ed9b4c524055214c26a7de7aa"):
|
||||
client = wda.USBClient(udid)
|
||||
session = client.session()
|
||||
session.appium_settings({"snapshotMaxDepth": 30})
|
||||
r = session.xpath("//XCUIElementTypeButton[@name='关闭屏幕']")
|
||||
try:
|
||||
if r.label == "关闭屏幕":
|
||||
return r
|
||||
else:
|
||||
return None
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return None
|
||||
|
||||
# AiUtils.screenshot()
|
||||
Reference in New Issue
Block a user