1
0
mirror of https://github.com/TheFunny/ArisuAutoSweeper synced 2025-12-16 17:45:12 +00:00

feat(sweep): support finding max sweepable index

This commit is contained in:
YoursFunny 2023-11-29 14:55:55 +08:00
parent 6ae785634c
commit d684fd79f6
Signed by: YoursFunny
GPG Key ID: 207EDC3CD5B40F9C

View File

@ -131,9 +131,63 @@ class StageList:
timeout=Timer(1.5, 5) timeout=Timer(1.5, 5)
) )
def insight_max_sweepable_index(self, main: ModuleBase, skip_first_screenshot=True) -> int:
"""
Args:
main:
skip_first_screenshot:
Returns:
Index of max sweepable stage
"""
logger.info('Insight sweepable index')
max_sweepable_index = 0
last_max_sweepable_index = 0
while 1:
if skip_first_screenshot:
skip_first_screenshot = False
else:
main.device.screenshot()
self.load_stage_indexes(main=main)
sweepable_index = next(
filter(
lambda x: not self.is_sweepable(main, self.search_box(x[-1][:2])),
self.current_indexes
), None
)
# all sweepable
if sweepable_index is None:
logger.info('All sweepable')
max_sweepable_index = self.current_index_max
self.swipe_page(self.swipe_direction, main)
if max_sweepable_index == last_max_sweepable_index:
logger.info(f'Max sweepable index: {max_sweepable_index}')
return max_sweepable_index
last_max_sweepable_index = max_sweepable_index
# all not sweepable
elif int(sweepable_index[0]) == self.current_index_min:
logger.info('All not sweepable')
if int(sweepable_index[0]) == 1:
logger.warning('No sweepable index')
return 0
self.swipe_page(self.swipe_direction, main, reverse=True)
else:
logger.info(f'Sweepable index: {int(sweepable_index[0]) - 1}')
return int(sweepable_index[0]) - 1
main.wait_until_stable(
self.stage.button,
timer=Timer(0, 0),
timeout=Timer(1.5, 5)
)
def is_sweepable(self, main: ModuleBase, search_box) -> bool: def is_sweepable(self, main: ModuleBase, search_box) -> bool:
self.sweepable.load_search(search_box) self.sweepable.load_search(search_box)
return main.appear(self.sweepable) return main.appear(self.sweepable, similarity=0.8)
def search_box( def search_box(
self, self,