mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-23 09:01:45 +00:00
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from module.base.utils import area_pad
|
||
|
|
from module.map.grid_info import GridInfo
|
||
|
|
from module.map.grid_predictor import GridPredictor
|
||
|
|
|
||
|
|
|
||
|
|
class Grid(GridPredictor, GridInfo):
|
||
|
|
def __init__(self, location, image, corner):
|
||
|
|
"""
|
||
|
|
|
||
|
|
Args:
|
||
|
|
location(tuple):
|
||
|
|
image:
|
||
|
|
corner: (x0, y0) +-------+ (x1, y1)
|
||
|
|
/ \
|
||
|
|
/ \
|
||
|
|
(x2, y2) +-------------+ (x3, y3)
|
||
|
|
"""
|
||
|
|
self.location = location
|
||
|
|
super().__init__(location, image, corner)
|
||
|
|
self.corner = corner.flatten()
|
||
|
|
|
||
|
|
@property
|
||
|
|
def button(self):
|
||
|
|
"""
|
||
|
|
Returns:
|
||
|
|
tuple: area
|
||
|
|
"""
|
||
|
|
x0, y0, x1, y1, x2, y2, x3, y3 = self.corner
|
||
|
|
area = tuple(np.rint((max(x0, x2), max(y0, y1), min(x1, x3), min(y2, y3))).astype(int))
|
||
|
|
return area_pad(area, pad=25)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def avoid_area(self):
|
||
|
|
x0, y0, x1, y1, x2, y2, x3, y3 = self.corner
|
||
|
|
area = tuple(np.rint((min(x0, x2), min(y0, y1), max(x1, x3), max(y2, y3))).astype(int))
|
||
|
|
return area_pad(area, pad=-25)
|