AlasGG

campaign/event_20231026_cn/campaign_base.py

@@ -0,0 +1,75 @@
+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.55, -0.2, 0.45, 0.2), shape=(50, 20))
+        image = color_similarity_2d(image, color=(255, 150, 24))
+        if image[image > 221].shape[0] > 200:
+            if TEMPLATE_ENEMY_BOSS.match(image, similarity=0.6):
+                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_):
+    STAGE_INCREASE = [
+        """
+        T1 > T2 > T3 > T4 > T5 > T6
+        """
+    ]
+
+    def campaign_set_chapter_event(self, chapter, mode='normal'):
+        self.ui_goto_event()
+        self.campaign_ensure_chapter(index=chapter)
+        return True
+
+    def _campaign_get_chapter_index(self, name):
+        """
+        Args:
+            name (str, int):
+
+        Returns:
+            int
+        """
+        if name == 't1':
+            return 1
+        if name == 't2':
+            return 2
+        if name == 'ex_sp':
+            return 3
+        if name == 'ex_ex':
+            return 4
+
+        return super(CampaignBase, CampaignBase)._campaign_get_chapter_index(name)
+
+    @staticmethod
+    def _campaign_separate_name(name):
+        """
+        Args:
+            name (str): Stage name in lowercase, such as 7-2, d3, sp3.
+
+        Returns:
+            tuple[str]: Campaign_name and stage index in lowercase, Such as ['7', '2'], ['d', '3'], ['sp', '3'].
+        """
+        if name in ['t1', 't2', 't3']:
+            return 't1', name[-1]
+        if name in ['t4', 't5', 't6']:
+            return 't2', name[-1]
+        if 'esp' in name:
+            return ['ex_sp', '1']
+        if 'ex' in name:
+            return ['ex_ex', '1']
+
+        return super(CampaignBase, CampaignBase)._campaign_separate_name(name)