mirror of
https://github.com/MaaAssistantArknights/MaaAssistantArknights.git
synced 2026-07-19 18:47:55 +08:00
feat: 添加用于快速适配新主界面主题的小工具
This commit is contained in:
0
tools/GetImageFromROI/dst/.gitkeep
Normal file
0
tools/GetImageFromROI/dst/.gitkeep
Normal file
80
tools/GetImageFromROI/main.py
Normal file
80
tools/GetImageFromROI/main.py
Normal file
@@ -0,0 +1,80 @@
|
||||
import cv2
|
||||
import json
|
||||
from pathlib import Path
|
||||
|
||||
'''
|
||||
This is a tool for cropping one image into several images based on current roi.
|
||||
Use this tool when a raw image contains many templates.
|
||||
For example, when adding a new homepage theme.
|
||||
Put one picture in ./src and add the template's tasks to 'task_list',
|
||||
This tools will try to read roi (if not 'roi_crop_doc') from tasks in tasks.json and then crop the image.
|
||||
'''
|
||||
|
||||
std_width: int = 1280
|
||||
std_height: int = 720
|
||||
std_ratio = std_width / std_height
|
||||
|
||||
task_paths = {
|
||||
"CN": Path("../../resource/tasks.json"),
|
||||
"twxy": Path("../../resource/global/txwy/resource/tasks.json"),
|
||||
"YoStarEN": Path("../../resource/global/YoStarEN/resource/tasks.json"),
|
||||
"YoStarJP": Path("../../resource/global/YoStarJP/resource/tasks.json"),
|
||||
"YoStarKR": Path("../../resource/global/YoStarKR/resource/tasks.json"),
|
||||
}
|
||||
|
||||
# Change this to your client
|
||||
task_file = task_paths["CN"]
|
||||
|
||||
# Add the tasks you want to get its image
|
||||
task_list = []
|
||||
# You should get these names from tasks.json
|
||||
#
|
||||
# For example, when adding a new homepage theme:
|
||||
task_list = [
|
||||
"Award",
|
||||
"GachaEnter",
|
||||
"Home",
|
||||
"Infrast",
|
||||
"OperBoxEnter",
|
||||
"Recruit",
|
||||
"Visit"
|
||||
]
|
||||
|
||||
with task_file.open("r", encoding="utf-8") as f:
|
||||
tasks = json.load(f)
|
||||
|
||||
src_path = Path("./src")
|
||||
dst_path = Path("./dst")
|
||||
|
||||
for raw_image in src_path.glob("*.png"):
|
||||
|
||||
print("Processing file:", str(raw_image))
|
||||
|
||||
image = cv2.imread(str(raw_image))
|
||||
cur_ratio = image.shape[1] / image.shape[0]
|
||||
if cur_ratio >= std_ratio: # 说明是宽屏或默认16:9,按照高度计算缩放
|
||||
dsize_width: int = (int)(cur_ratio * std_height)
|
||||
dsize_height: int = std_height
|
||||
else: # 否则可能是偏正方形的屏幕,按宽度计算
|
||||
dsize_width: int = std_width
|
||||
dsize_height: int = std_width / cur_ratio
|
||||
dsize = (dsize_width, dsize_height)
|
||||
image = cv2.resize(image, dsize, interpolation=cv2.INTER_AREA)
|
||||
|
||||
for i in task_list:
|
||||
if type(tasks[i]["template"]) == str:
|
||||
filename = tasks[i]["template"]
|
||||
elif type(tasks[i]["template"]) == list:
|
||||
# this is for multi-template:
|
||||
filename = tasks[i]["template"][0].split(".")[0] + raw_image.stem + ".png"
|
||||
try:
|
||||
roi = tasks[i]["roi_crop_doc"]
|
||||
except KeyError:
|
||||
roi = tasks[i]["roi"]
|
||||
cropped = image[roi[1]:roi[1]+roi[3], roi[0]:roi[0]+roi[2]]
|
||||
|
||||
print("Saving", dst_path / filename)
|
||||
cv2.imwrite(str(dst_path / filename), cropped)
|
||||
|
||||
print("Finished processing file:", str(raw_image))
|
||||
|
||||
1
tools/GetImageFromROI/requirements.txt
Normal file
1
tools/GetImageFromROI/requirements.txt
Normal file
@@ -0,0 +1 @@
|
||||
opencv-python~=4.5.3
|
||||
0
tools/GetImageFromROI/src/.gitkeep
Normal file
0
tools/GetImageFromROI/src/.gitkeep
Normal file
2
tools/GetImageFromROI/start.bat
Normal file
2
tools/GetImageFromROI/start.bat
Normal file
@@ -0,0 +1,2 @@
|
||||
python main.py
|
||||
pause
|
||||
Reference in New Issue
Block a user