xtaodada/StarRailCopilot
1
0
Fork 0
mirror of https://github.com/LmeSzinc/StarRailCopilot.git synced 2024-11-16 06:25:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add: [ALAS] Utility func image_paste() and get_bbox_reversed()

Browse Source
This commit is contained in:
LmeSzinc 2023-07-30 19:35:05 +08:00
parent c71dc5296b
commit 9f0528299f
1 changed files with 37 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
module/base/utils/utils.py
View File

@ -317,9 +317,9 @@ def area_in_area(area1, area2, threshold=5):
bool:
"""
return area2[0] - threshold <= area1[0] \
and area2[1] - threshold <= area1[1] \
and area1[2] <= area2[2] + threshold \
and area1[3] <= area2[3] + threshold
and area2[1] - threshold <= area1[1] \
and area1[2] <= area2[2] + threshold \
and area1[3] <= area2[3] + threshold
def area_cross_area(area1, area2, threshold=5):
@ -337,7 +337,7 @@ def area_cross_area(area1, area2, threshold=5):
xa1, ya1, xa2, ya2 = area1
xb1, yb1, xb2, yb2 = area2
return abs(xb2 + xb1 - xa2 - xa1) <= xa2 - xa1 + xb2 - xb1 + threshold * 2 \
and abs(yb2 + yb1 - ya2 - ya1) <= ya2 - ya1 + yb2 - yb1 + threshold * 2
and abs(yb2 + yb1 - ya2 - ya1) <= ya2 - ya1 + yb2 - yb1 + threshold * 2
def float2str(n, decimal=3):
@ -594,6 +594,21 @@ def image_size(image):
return shape[1], shape[0]
def image_paste(image, background, origin):
"""
Paste an image on background.
This method does not return a value, but instead updates the array "background".
Args:
image:
background:
origin: Upper-left corner, (x, y)
"""
x, y = origin
w, h = image_size(image)
background[y:y + h, x:x + w] = image
def rgb2gray(image):
"""
Args:
@ -687,6 +702,24 @@ def get_bbox(image, threshold=0):
return x[0], y[0], x[-1] + 1, y[-1] + 1
def get_bbox_reversed(image, threshold=0):
"""
Similar to `get_bbox` but for black contents on white background.
Args:
image (np.ndarray): Screenshot.
threshold (int): Color >= threshold will be considered white
Returns:
tuple: (upper_left_x, upper_left_y, bottom_right_x, bottom_right_y)
"""
if image_channel(image) == 3:
image = np.min(image, axis=2)
x = np.where(np.min(image, axis=0) < threshold)[0]
y = np.where(np.min(image, axis=1) < threshold)[0]
return x[0], y[0], x[-1] + 1, y[-1] + 1
def color_similarity(color1, color2):
"""
Args: