RemoteROMAssessment/jointSelectorView.py at master · MinhasKamal/RemoteROMAssessment · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
# Minhas Kamal (minhaskamal024@gmail.com)
# 08 Mar 24

import tkinter as tk
import movement
import skeleton

class JointSelectorView:

    normal_color: str = "#fff"
    tracking_color: str = "#f00"
    observing_color: str = "#00f"

    joint_selection_button_properties: list[list] = [
            [skeleton.Skeleton.joints[0][0], 315, 85],
            [skeleton.Skeleton.joints[1][0], 185, 85],
            [skeleton.Skeleton.joints[2][0], 395, 130],
            [skeleton.Skeleton.joints[3][0], 105, 130],
            [skeleton.Skeleton.joints[4][0], 405, 210],
            [skeleton.Skeleton.joints[5][0], 100, 210],
            [skeleton.Skeleton.joints[6][0], 315, 200],
            [skeleton.Skeleton.joints[7][0], 190, 200],
            [skeleton.Skeleton.joints[8][0], 345, 270],
            [skeleton.Skeleton.joints[9][0], 165, 270],
            [skeleton.Skeleton.joints[10][0], 327, 350],
            [skeleton.Skeleton.joints[11][0], 178, 350]
        ]

    @classmethod
    def show(cls, view_frame: tk.Frame, doctor_movement: movement.Movement) -> None:
        view_frame.name = cls.__name__

        prompt_frame: tk.Frame = tk.Frame(view_frame)
        prompt_frame.pack(pady=10)
        tk.Label(
            prompt_frame,
            text="Select joints").pack()
        tk.Label(
            prompt_frame,
            width=1,
            height=1,
            text="",
            bg=cls.tracking_color).pack(
                side=tk.LEFT)
        tk.Label(
            prompt_frame,
            text="Tracking",
            fg=cls.tracking_color).pack(
                side=tk.LEFT,
                padx=(0, 20))
        tk.Label(
            prompt_frame,
            width=1,
            height=1,
            bg=cls.observing_color).pack(
                side=tk.LEFT)
        tk.Label(
            prompt_frame,
            text="Observing",
            fg=cls.observing_color).pack(
                side=tk.LEFT)

        model_frame: tk.Frame = tk.Frame(
            view_frame,
            width=500,
            height=400)
        model_frame.img = tk.PhotoImage(file="res/model.png").subsample(2, 2)
        model_frame.pack()
        tk.Label(
            model_frame,
            image=model_frame.img).place(
                x=0,
                y=0,
                width=500,
                height=400)
        for property in cls.joint_selection_button_properties:
            cls._create_joint_selection_button(
                model_frame,
                doctor_movement,
                text=property[0],
                x=property[1],
                y=property[2])

        return

    @classmethod
    def _create_joint_selection_button(
            cls, model_frame: tk.Frame, doctor_movement: movement.Movement,
            text: str, x: int, y: int) -> None:

        button = tk.Button(model_frame, text=text, font=('Arial', 10))
        if text in doctor_movement.tracking_joint_list:
            button.config(bg=cls.tracking_color)
        elif text in doctor_movement.observing_joint_list:
            button.config(bg=cls.observing_color)
        button.config(
            command=lambda:
                cls._joint_selection_button_command(button, doctor_movement))
        button.place(x=x, y=y, anchor="center")

        return

    @classmethod
    def _joint_selection_button_command(
            cls, button: tk.Button, doctor_movement: movement.Movement) -> None:

        joint_name: str = button.cget('text')

        if button.cget('bg') == cls.normal_color:
            button.config(bg=cls.tracking_color)
            doctor_movement.tracking_joint_list.append(joint_name)
        elif button.cget('bg') == cls.tracking_color:
            button.config(bg=cls.observing_color)
            doctor_movement.tracking_joint_list.remove(joint_name)
            doctor_movement.observing_joint_list.append(joint_name)
        else:
            button.config(bg=cls.normal_color)
            doctor_movement.observing_joint_list.remove(joint_name)

        return


# test
if __name__ == "__main__":
    ui = tk.Tk()
    ui.option_add("*Font", ('Arial', 12))
    ui.option_add("*Background", "#fff")
    view_frame = tk.Frame(ui)
    view_frame.pack(fill="both")
    # doctor_movement = movement.Movement()
    doctor_movement = movement.Movement.from_file("res/test")
    JointSelectorView.show(view_frame, doctor_movement)
    ui.mainloop()
    print(doctor_movement)