HW/gpio_controller.py at main · salguy/HW · 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
import os
import threading
import time

try:
    import RPi.GPIO as GPIO
    RPI_AVAILABLE = True
except (ImportError, RuntimeError):
    RPI_AVAILABLE = False
    class GPIO:
        BCM = OUT = IN = HIGH = LOW = PUD_UP = None
        @staticmethod
        def setmode(*args, **kwargs): pass
        @staticmethod
        def setup(*args, **kwargs): pass
        @staticmethod
        def output(*args, **kwargs): pass
        @staticmethod
        def input(*args): return 1
        @staticmethod
        def cleanup(): pass

# 핀 번호 설정
RED_LED = 17
BLUE_LED = 27
GREEN_LED = 22
SCEDULE_SWITCH = 23
SKIP_SWITCH = 24
RESET_SWITCH = 25

class GPIOController:
    def __init__(self, refresh_callback, skip_callback):
        self.refresh_callback = refresh_callback
        self.skip_callback = skip_callback
        self.initialized = False
        self.last_reset_time = 0
        self.initialized = self._setup_gpio()
        if self.initialized:
            self._start_switch_monitor()

    def _setup_gpio(self):
        try:
            GPIO.setmode(GPIO.BCM)
            GPIO.setup(RED_LED, GPIO.OUT)
            GPIO.setup(BLUE_LED, GPIO.OUT)
            GPIO.setup(GREEN_LED, GPIO.OUT)
            GPIO.setup(SCEDULE_SWITCH, GPIO.IN, pull_up_down=GPIO.PUD_UP)
            GPIO.setup(RESET_SWITCH, GPIO.IN, pull_up_down=GPIO.PUD_UP)
            GPIO.setup(SKIP_SWITCH, GPIO.IN, pull_up_down=GPIO.PUD_UP)
            self.set_mode("default")
            return True
        except Exception as e:
            print(f"GPIO 설정 중 오류 발생: {e}")
            return False

    def _start_switch_monitor(self):
        def monitor():
            while True:
                try:
                    if GPIO.input(SCEDULE_SWITCH) == GPIO.LOW:
                        self.refresh_callback()
                        time.sleep(1)

                    if GPIO.input(SKIP_SWITCH) == GPIO.LOW:
                        self.skip_callback()
                        time.sleep(1)

                    if GPIO.input(RESET_SWITCH) == GPIO.LOW:
                        now = time.time()
                        if now - self.last_reset_time > 5:
                            self.last_reset_time = now
                            restart_program()

                except Exception as e:
                    print(f"스위치 모니터링 오류: {e}")
                    self.set_mode("error")
                    time.sleep(1)

                time.sleep(0.1)

        threading.Thread(target=monitor, daemon=True).start()

    def set_mode(self, mode):
        if not self.initialized:
            return
        GPIO.output(RED_LED, GPIO.LOW)
        GPIO.output(BLUE_LED, GPIO.LOW)
        GPIO.output(GREEN_LED, GPIO.LOW)
        if mode == "default":
            GPIO.output(BLUE_LED, GPIO.LOW)
            GPIO.output(RED_LED, GPIO.LOW)
            GPIO.output(GREEN_LED, GPIO.LOW)
        elif mode == "wakeword":
            GPIO.output(BLUE_LED, GPIO.HIGH)
        elif mode == "llmtts":
            GPIO.output(BLUE_LED, GPIO.HIGH)
        elif mode == "error":
            GPIO.output(RED_LED, GPIO.HIGH)
        elif mode == "thinking" :
            GPIO.output(GREEN_LED, GPIO.HIGH)

    def cleanup(self):
        if self.initialized:
            GPIO.cleanup()

def restart_program():
    print("main 재시작합니다...")
    os.execv("/home/pi/my_project/env/bin/python", [
        "/home/pi/my_project/env/bin/python",
        "/home/pi/my_project/main.py"
    ])