卫戍协议

mower/utils/graph/__init__.py

@@ -10,6 +10,7 @@ from . import (
     rogue,
     shop,
     sss,
+    stronghold_protocol,
     terminal,
 )
 from .utils import DG, SceneGraphSolver, edge
@@ -30,4 +31,5 @@ __all__ = [
     "sss",
     "terminal",
     "rogue",
+    "stronghold_protocol",
 ]

mower/utils/graph/navbar.py

@@ -52,6 +52,7 @@ from .utils import edge
 @edge(Scene.OPERATOR_MANAGEMENT, Scene.NAVIGATION_BAR)
 @edge(Scene.OPERATOR_DETAILS, Scene.NAVIGATION_BAR)
 @edge(Scene.ROGUE_INDEX, Scene.NAVIGATION_BAR)
+@edge(Scene.SP_MAIN, Scene.NAVIGATION_BAR)
 def index_nav(solver: BaseSolver):
     solver.tap("nav_button")
 

mower/utils/graph/stronghold_protocol.py

@@ -0,0 +1,26 @@
+from mower.utils.scene import Scene
+from mower.utils.solver import BaseSolver
+
+from .utils import edge
+
+# 卫戍协议
+
+
+@edge(Scene.SP_MAIN, Scene.TERMINAL_MAIN)
+@edge(Scene.SP_DEFENCE, Scene.SP_MAIN)
+@edge(Scene.SP_ACTION, Scene.SP_MAIN)
+@edge(Scene.SP_AUTO, Scene.SP_BACK_CONFIRM)
+@edge(Scene.SP_COMPLETE, Scene.SP_BACK_CONFIRM)
+def sp_back(solver: BaseSolver):
+    solver.back()
+
+
+@edge(Scene.SP_GIVEUP_CONFIRM, Scene.SP_MAIN)
+@edge(Scene.SP_BACK_CONFIRM, Scene.SP_MAIN)
+def dialog_confirm(solver: BaseSolver):
+    solver.tap("double_confirm/main", x_rate=1)
+
+
+@edge(Scene.SP_ANIMATION, Scene.SP_ACTION)
+def wait(solver: BaseSolver):
+    solver.sleep()

mower/utils/recognize/__init__.py

@@ -256,6 +256,10 @@ class Recognizer:
                 self.scene = Scene.ROGUE_REFRESH_SHOP
             elif self.find("double_confirm/headhunting"):
                 self.scene = Scene.HEADHUNTING_FREE_CONFIRM
+            elif self.find("double_confirm/stronghold_protocol_back"):
+                self.scene = Scene.SP_BACK_CONFIRM
+            elif self.find("double_confirm/stronghold_protocol_giveup"):
+                self.scene = Scene.SP_GIVEUP_CONFIRM
             else:
                 self.scene = Scene.DOUBLE_CONFIRM
         elif self.find("mission_trainee_on"):
@@ -341,6 +345,17 @@ class Recognizer:
             self.scene = Scene.SIGN_IN_ORUNDUM
         elif self.find("start_story"):
             self.scene = Scene.STORY_STAGE
+        elif self.find("stronghold_protocol/main"):
+            self.scene = Scene.SP_MAIN
+        elif self.find("stronghold_protocol/defence"):
+            self.scene = Scene.SP_DEFENCE
+        elif self.find("stronghold_protocol/auto"):
+            self.scene = Scene.SP_AUTO
+        elif self.find("stronghold_protocol/complete"):
+            self.scene = Scene.SP_COMPLETE
+        elif self.find("stronghold_protocol/animation"):
+            self.scene = Scene.SP_ANIMATION
+
         elif self.is_black():
             self.scene = Scene.LOADING
 
@@ -434,6 +449,8 @@ class Recognizer:
             self.scene = Scene.STORY_SKIP
         elif self.find("story_skip"):
             self.scene = Scene.STORY
+        elif self.find("stronghold_protocol/action"):
+            self.scene = Scene.SP_ACTION
 
         # 没弄完的
         # elif self.find("ope_elimi_finished"):

mower/utils/recognize/data.py

@@ -37,6 +37,8 @@ color = {
     "double_confirm/refresh_shop": (901, 435),
     "double_confirm/sss": (901, 465),
     "double_confirm/sss_abandon_drop": (901, 463),
+    "double_confirm/stronghold_protocol_back": (1056, 437),
+    "double_confirm/stronghold_protocol_giveup": (921, 434),
     "double_confirm/voice": (745, 435),
     "drone": (274, 437),
     "factory_collect": (1542, 886),
@@ -160,6 +162,13 @@ color = {
     "sss/switch_to_ex": (1255, 942),
     "sss/switch_to_normal": (1255, 934),
     "start_story": (1392, 623),
+    "stronghold_protocol/animation": (803, 391),
+    "stronghold_protocol/auto": (898, 56),
+    "stronghold_protocol/complete": (231, 101),
+    "stronghold_protocol/defence": (29, 14),
+    "stronghold_protocol/main": (1456, 472),
+    "stronghold_protocol/start": (1709, 879),
+    "stronghold_protocol/stop": (1272, 801),
     "terminal_main": (1658, 734),
 }
 
@@ -307,6 +316,8 @@ template_matching = {
     "stone_fragment": None,
     "story_skip": (1718, 58),
     "story_skip_confirm_dialog": ((685, 504), (1234, 730)),
+    "stronghold_protocol/action": (849, 60),
+    "stronghold_protocol/medic": (37, 167),
     "switch_order/check": None,
     "switch_order/lmb": (1442, 891),
     "switch_order/oru": (1442, 891),

mower/utils/scene.py

@@ -353,6 +353,22 @@ class Scene:
     "放弃本次肉鸽探索"
     ROGUE_REFRESH_SHOP = 1704
     "肉鸽商店刷新确认"
+    SP_MAIN = 1801
+    "卫戍协议首页"
+    SP_DEFENCE = 1802
+    "卫戍协议防守方策略"
+    SP_ACTION = 1803
+    "卫戍协议玩家行动"
+    SP_AUTO = 1804
+    "卫戍协议玩家发呆"
+    SP_GIVEUP_CONFIRM = 1805
+    "卫戍协议放弃模拟确认"
+    SP_COMPLETE = 1806
+    "卫戍协议结算"
+    SP_BACK_CONFIRM = 1807
+    "卫戍协议回到主页确认"
+    SP_ANIMATION = 1808
+    "卫戍协议动画"
     LOADING = 9998
     "场景跳转时的等待界面"
     CONFIRM = 9999
@@ -537,6 +553,14 @@ SceneComment = {
     1702: "肉鸽首页",
     1703: "放弃本次肉鸽探索",
     1704: "肉鸽商店刷新确认",
+    1801: "卫戍协议首页",
+    1802: "卫戍协议防守方策略",
+    1803: "卫戍协议玩家行动",
+    1804: "卫戍协议玩家发呆",
+    1805: "卫戍协议放弃模拟确认",
+    1806: "卫戍协议结算",
+    1807: "卫戍协议回到主页确认",
+    1808: "卫戍协议动画",
     9998: "场景跳转时的等待界面",
     9999: "确认对话框",
 }

mower/utils/solver.py

@@ -14,8 +14,9 @@ from mower.utils.csleep import MowerExit, csleep
 from mower.utils.device.adb_client.const import KeyCode
 from mower.utils.device.adb_client.session import Session
 from mower.utils.device.device import Device
-from mower.utils.image import cropimg, thres2
+from mower.utils.image import cropimg, loadres, thres2
 from mower.utils.log import logger
+from mower.utils.matcher import GOOD_DISTANCE_LIMIT, flann, keypoints_scale_invariant
 from mower.utils.recognize import RecognizeError, Recognizer, Scene
 from mower.utils.simulator import restart_simulator
 from mower.utils.traceback import caller_info
@@ -642,3 +643,19 @@ class BaseSolver:
             csleep(3)
             self.check_current_focus()
             return False
+
+    def terminal_entry(self, res: tp.Res):
+        img = loadres(res, True)
+        kp1, des1 = keypoints_scale_invariant(img)
+        kp2, des2 = config.recog.matcher.kp, config.recog.matcher.des
+        matches = flann.knnMatch(des1, des2, k=2)
+        good = []
+        for pair in matches:
+            if (len_pair := len(pair)) == 2:
+                x, y = pair
+                if x.distance < GOOD_DISTANCE_LIMIT * y.distance:
+                    good.append(x)
+            elif len_pair == 1:
+                good.append(pair[0])
+        good = sorted(good, key=lambda x: x.distance)
+        self.tap(kp2[good[0].trainIdx].pt, interval=4)

mower/utils/typealias/res.py

@@ -87,6 +87,8 @@ Res = Literal[
     "double_confirm/refresh_shop",
     "double_confirm/sss",
     "double_confirm/sss_abandon_drop",
+    "double_confirm/stronghold_protocol_back",
+    "double_confirm/stronghold_protocol_giveup",
     "double_confirm/voice",
     "drone",
     "episode",
@@ -175,6 +177,7 @@ Res = Literal[
     "navigation/act/2",
     "navigation/activity/banner",
     "navigation/activity/entry",
+    "navigation/activity/terminal.jpg",
     "navigation/biography/OF_banner",
     "navigation/biography/OF_entry",
     "navigation/collection/AP-1",
@@ -504,6 +507,16 @@ Res = Literal[
     "story_skip",
     "story_skip_confirm_dialog",
     "story_stage",
+    "stronghold_protocol/action",
+    "stronghold_protocol/animation",
+    "stronghold_protocol/auto",
+    "stronghold_protocol/complete",
+    "stronghold_protocol/defence",
+    "stronghold_protocol/entry.jpg",
+    "stronghold_protocol/main",
+    "stronghold_protocol/medic",
+    "stronghold_protocol/start",
+    "stronghold_protocol/stop",
     "switch_order/check",
     "switch_order/lmb",
     "switch_order/oru",