Add: Island UI base

Author: guoh064

assets/cn/island/ISLAND_CLICK_SAFE_AREA.png

@@ -0,0 +1,8 @@
+from module.base.button import Button
+from module.base.template import Template
+
+# This file was automatically generated by dev_tools/button_extract.py.
+# Don't modify it manually.
+
+ISLAND_CLICK_SAFE_AREA = Button(area={'cn': (596, 687, 680, 720), 'en': (596, 687, 680, 720), 'jp': (596, 687, 680, 720), 'tw': (596, 687, 680, 720)}, color={'cn': (255, 255, 255), 'en': (255, 255, 255), 'jp': (255, 255, 255), 'tw': (255, 255, 255)}, button={'cn': (596, 687, 680, 720), 'en': (596, 687, 680, 720), 'jp': (596, 687, 680, 720), 'tw': (596, 687, 680, 720)}, file={'cn': './assets/cn/island/ISLAND_CLICK_SAFE_AREA.png', 'en': './assets/cn/island/ISLAND_CLICK_SAFE_AREA.png', 'jp': './assets/cn/island/ISLAND_CLICK_SAFE_AREA.png', 'tw': './assets/cn/island/ISLAND_CLICK_SAFE_AREA.png'})
+ISLAND_LEVEL_UP = Button(area={'cn': (616, 339, 662, 377), 'en': (616, 339, 662, 377), 'jp': (616, 339, 662, 377), 'tw': (616, 339, 662, 377)}, color={'cn': (186, 213, 226), 'en': (186, 213, 226), 'jp': (186, 213, 226), 'tw': (186, 213, 226)}, button={'cn': (616, 339, 662, 377), 'en': (616, 339, 662, 377), 'jp': (616, 339, 662, 377), 'tw': (616, 339, 662, 377)}, file={'cn': './assets/cn/island/ISLAND_LEVEL_UP.png', 'en': './assets/cn/island/ISLAND_LEVEL_UP.png', 'jp': './assets/cn/island/ISLAND_LEVEL_UP.png', 'tw': './assets/cn/island/ISLAND_LEVEL_UP.png'})

assets/cn/island/ISLAND_LEVEL_UP.png

@@ -0,0 +1,105 @@
+import cv2
+import numpy as np
+
+from module.base.button import ButtonGrid
+from module.base.decorator import cached_property
+from module.base.utils import color_similarity_2d, rgb2luma
+from module.combat.assets import GET_ITEMS_1
+from module.island.assets import *
+from module.ui.navbar import Navbar
+from module.ui.ui import UI
+
+
+class IslandUI(UI):
+    def ui_additional(self, get_ship=True):
+        return super().ui_additional(get_ship=False)
+
+    def has_white_band(self, threshold=1000):
+        min_y = 405
+        max_y = 560
+        image = self.image_crop((0, min_y, 1280, max_y), copy=False)
+        luma = rgb2luma(image)
+        white_mask = cv2.inRange(luma, 200, 255)
+        kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (31, 3))
+        white_mask = cv2.morphologyEx(white_mask, cv2.MORPH_CLOSE, kernel)
+        row_white_width = (white_mask > 0).sum(axis=1)
+        candidate_rows = row_white_width > threshold
+        min_continuous_height = 50
+        count = 0
+        for ok in candidate_rows:
+            if ok:
+                count += 1
+                if count >= min_continuous_height:
+                    return True
+            else:
+                count = 0
+        return False
+
+    def handle_island_get_items(self):
+        if self.appear(GET_ITEMS_1, offset=(20, 20), interval=3):
+            self.device.click(ISLAND_CLICK_SAFE_AREA)
+            return True
+        if self.has_white_band():
+            self.device.click(ISLAND_CLICK_SAFE_AREA)
+            return True
+        return False
+
+    def handle_island_level_up(self):
+        if self.appear(ISLAND_LEVEL_UP, offset=(20, 20), interval=3):
+            self.device.click(ISLAND_CLICK_SAFE_AREA)
+            return True
+        return False
+
+    def handle_island_additional(self):
+        if self.handle_island_get_items():
+            return True
+        if self.handle_island_level_up():
+            return True
+        return False
+
+    def is_button_selected(self, button, color=(57, 189, 255), threshold=221, count=100):
+        """
+        Detects if the button is surrounded by a blue border,
+        which indicates that the button is chosen.
+
+        Args:
+            button (Button, tuple): Button instance or area.
+
+        Returns:
+            bool: True if the button is chosen, False otherwise.
+        """
+        if isinstance(button, np.ndarray):
+            image = button
+        else:
+            image = self.image_crop(button, copy=False)
+        mask = color_similarity_2d(image, color)
+        cv2.inRange(mask, threshold, 255, dst=mask)
+        mask[2:-2, 2:-2] = 0
+        sum_ = cv2.countNonZero(mask)
+        return sum_ > count
+
+    @cached_property
+    def _island_manage_side_navbar(self):
+        island_manage_side_navbar = ButtonGrid(
+            origin=(13, 107), delta=(0, 196/3),
+            button_shape=(128, 43), grid_shape=(1, 3)
+        )
+        return Navbar(grids=island_manage_side_navbar,
+                      active_color=(57, 189, 255),
+                      inactive_color=(50, 52, 55),
+                      active_count=500,
+                      inactive_count=500)
+
+    def island_manage_side_navbar_ensure(self, upper=1, skip_first_screenshot=True):
+        """
+        Args: 
+            upper (int):
+                1 for production,
+                2 for restaurant,
+                3 for collect
+            bottom (int):
+                1 for collect,
+                2 for restaurant,
+                3 for production
+        """
+        return self._island_manage_side_navbar.set(self, upper=upper, skip_first_screenshot=skip_first_screenshot)