mirror of
https://github.com/LmeSzinc/AzurLaneAutoScript.git
synced 2025-01-08 12:27:33 +08:00
31 lines
963 B
Python
31 lines
963 B
Python
from module.base.utils import color_similarity_2d
|
|
from module.campaign.campaign_base import CampaignBase as CampaignBase_
|
|
from module.map_detection.grid import Grid
|
|
from module.template.assets import TEMPLATE_ENEMY_BOSS
|
|
|
|
|
|
class EventGrid(Grid):
|
|
def predict_enemy_genre(self):
|
|
if self.enemy_scale:
|
|
return ''
|
|
|
|
image = self.relative_crop((-0, -0.2, 0.8, 0.2), shape=(40, 20))
|
|
image = color_similarity_2d(image, color=(255, 150, 24))
|
|
if image[image > 221].shape[0] > 30:
|
|
if TEMPLATE_ENEMY_BOSS.match(image, similarity=0.6, scaling=0.5):
|
|
return 'Siren_Siren'
|
|
|
|
return super().predict_enemy_genre()
|
|
|
|
def predict_boss(self):
|
|
if self.enemy_genre == 'Siren_Siren':
|
|
return False
|
|
return super().predict_boss()
|
|
|
|
|
|
class CampaignBase(CampaignBase_):
|
|
"""
|
|
Not all event maps use this, some are the same like the classics
|
|
"""
|
|
grid_class = EventGrid
|