utils.py

from pyglet.text import Label
from math import sqrt


def dualdigit(num: int) -> str:
    return (str(num) if num > 9 else "0" + str(num)) if num < 100 else str(num % 100)


def format_time(time: list) -> str:
    return f"{str(time[0])}:{dualdigit(time[1])}:{dualdigit(time[2])}"


def center_image(image):
    # sets an image's anchor point to its center
    # works with pyglet
    image.anchor_x = image.width // 2
    image.anchor_y = image.height // 2
    return image


def calc_lanes_x(window_width: int, lane_width: int, lanes: int) -> tuple:
    # calculate x axis of the center of each lane
    start = int((window_width - lane_width * lanes) / 2 + lane_width / 2)
    end = int(start + lane_width * lanes) - 1
    return tuple(range(start, end, lane_width))


def calc_lane_separators_x(window_width: int, lane_width: int, lanes: int) -> tuple:
    lanes_x = calc_lanes_x(window_width, lane_width, lanes)
    return [int(x - lane_width / 2) for x in lanes_x + (lanes_x[-1] + lane_width,)]


def calc_selecs_x(
    window_width: int, padding: int, labelwidth: int, len_choices: int
) -> tuple:
    if len_choices == 2:
        return (
            window_width / 2 - padding / 2 - labelwidth / 2,
            window_width / 2 + padding / 2 + labelwidth / 2,
        )
    elif len_choices == 3:
        return (
            window_width / 2 - padding - labelwidth,
            window_width / 2,
            window_width / 2 + padding + labelwidth,
        )


def make_label(text="", size=24, x=0, y=0, color=(0, 0, 0, 255), *args, **kwargs):
    return Label(
        text=text,
        x=x,
        y=y,
        color=color,
        font_name="Noto Sans CJK SC",
        font_size=size,
        *args,
        **kwargs,
    )


def move_cursor(sc: list, di: str, sz: tuple) -> list:
    # sc: selection cursor
    # di: direction (vim keybinding!)
    if di == "h" and sc[0] > 0:  # left
        sc[0] -= 1
    elif di == "j" and sc[1] < sz[1] - 1:  # down
        sc[1] += 1
    elif di == "k" and sc[1] > 0:  # up
        sc[1] -= 1
    elif di == "l" and sc[0] < sz[0] - 1:  # right
        sc[0] += 1


AVATAR_LABEL_AXES = (
    {"x": 750, "y": 800},
    {"x": 950, "y": 800},
    {"x": 1150, "y": 800},
    {"x": 750, "y": 600},
    {"x": 950, "y": 600},
    {"x": 1150, "y": 600},
)

classroom_positions = []
for x in range(760, 1161, 200):
    for y in range(840, 440, -40):
        classroom_positions.append({"x": x, "y": y})

CLASSROOM_LABEL_AXES = tuple(classroom_positions)


def modulus_of_vector(vector: tuple) -> float:
    return sqrt(vector[0] ** 2 + vector[1] ** 2)


def time_spent(time: list):
    h, m, s = tuple(time)
    return (
        (f"{h-12} 小时 " if h > 12 else "")
        + (f"{m} 分 " if m else "")
        + f"{s if s > 9 else '0'+str(s)} 秒"
    )


def seconds_spent(time: list):
    h, m, s = tuple(time)
    return (h - 12) * 3600 + m * 60 + s