fix(team-race): wrong template size

auto_derby/__main__.py

@@ -5,7 +5,6 @@
 
 from auto_derby import templates
 import ctypes
-from os import name
 import time
 import webbrowser
 
@@ -24,7 +23,6 @@ def main():
         "team_race": jobs.team_race,
         "daily_race_money": lambda : jobs.daily_race(templates.MOONLIGHT_PRIZE),
         "daily_race_sp": lambda : jobs.daily_race(templates.JUPITER_CUP),
-        "create_pos_mask": jobs.create_pos_mask,
     }
     parser = argparse.ArgumentParser()
     parser.add_argument("job")

auto_derby/jobs/__init__.py

@@ -1,4 +1,3 @@
 from .nurturing import nurturing
 from .team_race import team_race
 from .daily_race import daily_race
-from .create_pos_mask import create_pos_mask

auto_derby/jobs/team_race.py

@@ -14,7 +14,8 @@ def team_race():
             templates.GREEN_NEXT_BUTTON,
             templates.TEAM_RACE_CHOOSE_COMPETITOR,
             templates.RACE_START_BUTTON,
-            templates.TEAM_RACE_SEE_RESULT_BUTTON,
+            templates.TEAM_RACE_RESULT_BUTTON,
+            templates.RACE_AGAIN_BUTTON,
             templates.TEAM_RACE_WIN,
             templates.TEAM_RACE_LOSE,
             templates.TEAM_RACE_NEXT_BUTTON,

auto_derby/templates/__init__.py

@@ -1,6 +1,7 @@
 # Code Generated from [__init__.py.gotmpl], DO NOT EDIT.
 
 CANCEL_BUTTON = "cancel_button.png"
+CIRCLE_ITEM_REQUEST = "circle_item_request.png"
 CONNECTING = "connecting.png"
 DAILY_RACE = "daily_race.png"
 DAILY_RACE_HARD = "daily_race_hard.png"
@@ -25,5 +26,5 @@
 TEAM_RACE_GUARANTEED_WIN_REWARD = "team_race_guaranteed_win_reward.png"
 TEAM_RACE_LOSE = "team_race_lose.png"
 TEAM_RACE_NEXT_BUTTON = "team_race_next_button.png"
-TEAM_RACE_SEE_RESULT_BUTTON = "team_race_see_result_button.png"
+TEAM_RACE_RESULT_BUTTON = "team_race_result_button.png"
 TEAM_RACE_WIN = "team_race_win.png"

auto_derby/templates/__init__.py.gotmpl

@@ -7,7 +7,7 @@
 {{- end }}
 {{ "" }}
 {{- range .Files }}
-    {{- if . | hassuffix ".pos.png" }}
+    {{- if . | hasSuffix ".pos.png" }}
     {{- else }}
 {{ template "stem" . | snakecase | upper }} = "{{.}}"
     {{- end }}

scripts/capture_template_position.py

@@ -0,0 +1,71 @@
+# -*- coding=UTF-8 -*-
+# pyright: strict
+
+if True:
+    import sys
+    import os
+    sys.path.insert(0, os.path.join(__file__, "../.."), )
+
+
+from typing import Text
+import cv2
+from auto_derby import window, template, templates
+import pathlib
+import PIL.Image
+import numpy as np
+import argparse
+
+_TEMPLATES_PATH = pathlib.Path(templates.__file__).parent
+
+
+def _all_templates():
+    for i in _TEMPLATES_PATH.glob("*.png"):
+        if i.is_dir():
+            continue
+        if str(i).endswith(".pos.png"):
+            continue
+        yield i
+
+
+def _latest_file():
+    return str((sorted(_all_templates(), key=lambda x: x.stat().st_mtime, reverse=True))[0].name)
+
+
+def create_pos_mask(name: Text):
+
+    game_img = template.screenshot(window.get_game())
+    match_img = template.load(name)
+    match = template.match(game_img, name)
+    last_match = template.DEBUG_DATA["last_match"]
+    if last_match is None:
+        raise ValueError("missing debug data")
+
+    out_img = np.zeros((game_img.height, game_img.width), dtype=float)
+
+    if match:
+        _, pos = match
+        x, y = pos
+        cv2.rectangle(out_img, pos, (x+match_img.width,
+                                     y+match_img.height), (255,), -1)
+
+    img = PIL.Image.fromarray(out_img).convert("1")
+    dest = str(_TEMPLATES_PATH / template.add_middle_ext(name, "pos"))
+    img.save(dest)
+    cv2.imshow("out", out_img)
+    cv2.waitKey()
+
+
+def main():
+
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument("--name", "-n", dest="name")
+    args = parser.parse_args()
+    name = args.name
+    if not name:
+        name = _latest_file()
+    create_pos_mask(name)
+
+
+if __name__ == '__main__':
+    main()