# DistanceSensorRed.py

import matplotlib.pyplot as plt
import matplotlib.animation as animation
from gpiozero import DistanceSensor
import time

# 거리 센서 설정 (GPIO 핀 번호를 적절히 변경하세요)
sensor = DistanceSensor(echo=17, trigger=4)

# 그래프 설정
fig, ax = plt.subplots()

xdata_red, ydata_red = [], []
line_red, = ax.plot([], [], 'o', color='red', ms=3)

xdata, ydata = [], []
line, = ax.plot([], [], 'o', color='blue', ms=3)


ax.set_ylim(0, 110)  # 거리 범위 (센티미터)
ax.set_xlim(0, 10)   # 시간 범위 (초)

# 축, 레이블 제목 설정
ax.set_title( "Distance Sensor" )
ax.set_xlabel('Time (seconds)')
ax.set_ylabel('Distance (cm)')

ax.grid( linestyle='--' )

# 시작 시간
start_time = time.time()

# 업데이트 함수: 이 함수는 새로운 데이터를 가져와서 그래프를 업데이트합니다.
def animate( frame ):
    current_time = time.time() - start_time
    distance = sensor.distance * 100  # meters to centimeters
    
    print( distance, "cm" )
    
    if distance > 20 :
        xdata.append( current_time )
        ydata.append( distance )
    else :
        xdata_red.append( current_time )
        ydata_red.append( distance )
    pass
    
    # 축 업데이트
    if current_time > ax.get_xlim()[1]:
        ax.set_xlim(ax.get_xlim()[0], current_time + 1)
        ax.figure.canvas.draw()
    pass

    line_red.set_data(xdata_red, ydata_red)
    line.set_data(xdata, ydata)
    
    return line, line_red
pass

# 애니메이션 설정
ani = animation.FuncAnimation(fig, animate, interval=10, blit=True)

# 그래프 표시
plt.show()