| """Startup mode menu.""" |
|
|
| from __future__ import annotations |
|
|
| import cv2 |
| import numpy as np |
|
|
| import ui |
|
|
| WINDOW = "Hand Tracking Drawing" |
| ITEMS = ( |
| ("2d", "2D Drawing"), |
| ("3d", "3D Drawing"), |
| ("poses", "Poses"), |
| ("translater", "Translater"), |
| ) |
| W, H = 720, 480 |
| TOP = 150 |
| ROW_H = 56 |
| PAD = 60 |
|
|
|
|
| class Menu: |
| """Mode selection screen.""" |
|
|
| def __init__(self) -> None: |
| self.index = 0 |
| self.click = -1 |
| self.note = "" |
|
|
| def on_mouse(self, event: int, x: int, y: int, flags: int, param) -> None: |
| """Track hover and clicks.""" |
| row = (y - TOP) // ROW_H |
| if not 0 <= row < len(ITEMS) or not PAD <= x <= W - PAD: |
| return |
| self.index = int(row) |
| if event == cv2.EVENT_LBUTTONDOWN: |
| self.click = int(row) |
|
|
| def frame(self) -> np.ndarray: |
| """Render menu image.""" |
| img = np.zeros((H, W, 3), dtype=np.uint8) |
| ui.put_text(img, "HAND TRACKING DRAWING", (PAD, 54), 26, (255, 255, 255)) |
| ui.put_text(img, "Select a mode", (PAD, 96), 17, (150, 150, 150)) |
| for i, (_, label) in enumerate(ITEMS): |
| y = TOP + i * ROW_H |
| box = (PAD, y, W - PAD, y + ROW_H - 12) |
| if i == self.index: |
| cv2.rectangle(img, box[:2], box[2:], (45, 45, 45), -1) |
| cv2.rectangle(img, box[:2], box[2:], (200, 200, 200), 1, cv2.LINE_AA) |
| color = (255, 255, 255) |
| else: |
| cv2.rectangle(img, box[:2], box[2:], (70, 70, 70), 1, cv2.LINE_AA) |
| color = (185, 185, 185) |
| ui.put_text(img, f"{i + 1}) {label}", (PAD + 20, y + 12), 19, color) |
| ui.put_text(img, "1..4 select W S move ENTER start Q quit", |
| (PAD, H - 70), 16, (140, 140, 140)) |
| if self.note: |
| ui.put_text(img, self.note, (PAD, H - 42), 16, (120, 120, 230)) |
| return img |
|
|
| def run(self) -> str | None: |
| """Show menu and return mode.""" |
| cv2.namedWindow(WINDOW, cv2.WINDOW_NORMAL) |
| cv2.resizeWindow(WINDOW, W, H) |
| cv2.setMouseCallback(WINDOW, self.on_mouse) |
| while True: |
| cv2.imshow(WINDOW, self.frame()) |
| key = cv2.waitKey(20) & 0xFF |
| if cv2.getWindowProperty(WINDOW, cv2.WND_PROP_VISIBLE) < 1: |
| return None |
|
|
| chosen = -1 |
| if self.click >= 0: |
| chosen, self.click = self.click, -1 |
| elif key in (ord("q"), 27): |
| cv2.destroyWindow(WINDOW) |
| cv2.waitKey(1) |
| return None |
| elif key in (ord("w"), 82): |
| self.index = (self.index - 1) % len(ITEMS) |
| elif key in (ord("s"), 84): |
| self.index = (self.index + 1) % len(ITEMS) |
| elif key in (13, 10, 32): |
| chosen = self.index |
| elif ord("1") <= key <= ord("0") + len(ITEMS): |
| chosen = key - ord("1") |
|
|
| if chosen < 0: |
| continue |
| self.index = chosen |
| name = ITEMS[chosen][0] |
| if name in ("2d", "3d", "poses"): |
| cv2.destroyWindow(WINDOW) |
| cv2.waitKey(1) |
| return name |
| self.note = f"{ITEMS[chosen][1]} is not available yet" |
|
|
