mirror of
https://github.com/LmeSzinc/AzurLaneAutoScript.git
synced 2025-01-08 13:27:45 +08:00
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
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()
|
|
|
|
def predict_current_fleet(self):
|
|
count = self.relative_hsv_count(area=(-0.5, -3.5, 0.5, -2.5), h=(141 - 3, 141 + 10), shape=(50, 50))
|
|
if count < 200:
|
|
return False
|
|
|
|
# No TEMPLATE_FLEET_CURRENT check as giant boss on B1 covers fleet
|
|
return True
|
|
|
|
|
|
class CampaignBase(CampaignBase_):
|
|
"""
|
|
Not all event maps use this, some are the same like the classics
|
|
"""
|
|
grid_class = EventGrid
|