Add: W16 preparations

campaign/campaign_main/campaign_support_fleet.py

@@ -0,0 +1,28 @@
+from module.base.mask import Mask
+from module.campaign.campaign_base import CampaignBase as CampaignBase_
+from module.logger import logger
+from module.map.assets import FLEET_SUPPORT_EMPTY
+from module.map_detection.utils_assets import ASSETS
+
+MASK_MAP_UI_SUPPORT = Mask(file='./assets/mask/MASK_MAP_UI_SUPPORT.png')
+
+
+class CampaignBase(CampaignBase_):
+    use_support_fleet = True
+
+    def fleet_preparation(self):
+        if self.appear(FLEET_SUPPORT_EMPTY, offset=(5, 5)):
+            self.use_support_fleet = False
+        logger.attr("use_support_fleet", self.use_support_fleet)
+        super().fleet_preparation()
+
+    def _map_swipe(self, vector, box=(239, 159, 1175, 628)):
+        # Left border to 239, avoid swiping on support fleet
+        return super()._map_swipe(vector, box=box)
+
+    def map_data_init(self, map_):
+        super().map_data_init(map_)
+        if self.use_support_fleet:
+            # Patch ui_mask, get rid of supporting fleet
+            _ = ASSETS.ui_mask
+            ASSETS.ui_mask = MASK_MAP_UI_SUPPORT.image