46 lines
1.9 KiB
Python
46 lines
1.9 KiB
Python
"""Identify an open game page or plan a route, without clicking anything."""
|
|
import argparse
|
|
import json
|
|
from pathlib import Path
|
|
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
from automation.offers import MacGameDesktop
|
|
from automation.pages import PageMap
|
|
|
|
ROOT = Path(__file__).resolve().parent
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='只读识别当前游戏页面或规划导航路径')
|
|
parser.add_argument('--image', type=Path, help='已裁剪的游戏内容截图；不能包含标题栏或桌面')
|
|
parser.add_argument('--from-page', dest='start', help='导航起点页面 ID')
|
|
parser.add_argument('--to-page', dest='target', help='导航终点页面 ID')
|
|
parser.add_argument('--previous-page', help='已确认的进入前页面；用于区分外观相同的弹框')
|
|
args = parser.parse_args()
|
|
page_map = PageMap(ROOT)
|
|
if args.start or args.target:
|
|
if not (args.start and args.target):
|
|
parser.error('--from-page 与 --to-page 必须同时提供')
|
|
print(json.dumps(page_map.route(args.start, args.target), ensure_ascii=False, indent=2))
|
|
return 0
|
|
if args.image:
|
|
with Image.open(args.image) as image:
|
|
if abs(image.width / image.height - 562 / 1002) > .035:
|
|
parser.error('图片必须仅包含竖屏游戏内容。')
|
|
frame = np.asarray(image.convert('RGB').resize((562, 1002), Image.Resampling.LANCZOS))
|
|
else:
|
|
desktop = MacGameDesktop()
|
|
shot = desktop.screenshot()
|
|
frame = page_map.matcher.frame(shot, desktop.region_for(shot))
|
|
if args.previous_page and args.previous_page not in page_map.pages:
|
|
parser.error('未知的进入前页面 ID。')
|
|
result = page_map.identify(frame, previous_page=args.previous_page)
|
|
print(json.dumps(result, ensure_ascii=False, indent=2))
|
|
return 0 if result['status'] == 'identified' else 1
|
|
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main())
|