# ===================================================
# KEPCO DEMO 설정 파일
# 이 파일을 수정하여 프로그램 동작을 변경할 수 있습니다.
# ===================================================

# --- 모델 설정 ---
pt_type: "dev"          # "dev" 또는 "demo"
use_hpe_person: true    # Human Pose Estimation 사용 여부
use_helmet_model: true  # 헬멧 전용 모델 사용 여부

# --- 라벨 설정 ---
view_conf_score: false        # confidence score 표시 여부
show_gloves: true             # glove on/off 라벨링 여부
label_all_white: true         # bounding box 색상 전부 흰색 여부
use_hpe_frame_check: false    # n개 프레임 체크 후 HPE 위험 판단

# --- 디버그 ---
add_cross_arm: false    # cross arm 디버그 모드
img_snap_shot: false    # 이미지 스냅샷 전송 여부

# --- 소스 설정 ---
# RTSP 카메라 프리셋 (source에서 프리셋 이름으로 사용 가능)
cameras:
  MG: "rtsp://admin:admin1263!@10.20.10.99:28554/onvif/media?profile=Profile1"
  LF: "rtsp://admin:!1q2w3e4r@10.20.10.100:50554/profile1/media.smp"
  FC: "rtsp://daool:Ekdnfeldpsdptm1@192.168.200.101/axis-media/media.amp"
  MG_74_LTE: "rtsp://admin:admin1263!@223.171.48.74:28554/trackID=1"
  FERMAT: "rtsp://192.168.200.233:50399/wd"

source: "FC"                    # 카메라 프리셋 이름 또는 파일/RTSP 경로
save_path: "./250630_result"

# --- 모델 가중치 파일 ---
weights:
  pose: "yolov8l-pose.pt"
  helmet: "yolov8_dev1_97.pt"
  # pt_type별 OD 모델
  od_dev: "yolov11m_dev1_8.pt"
  od_demo: "yolov8_dev1_66.pt"

# --- 모델 파라미터 ---
model_confidence: 0.5
model_image_size: 640
kpt_min_confidence: 0.5

# --- HPE 임계값 ---
hpe:
  falldown_tilt_ratio: 0.80
  falldown_tilt_angle: 15
  body_tilt_ratio: 0.30
  cross_arm_ratio_threshold: 0.10
  cross_arm_angle_threshold: [15.0, 165.0]
  arm_angle_threshold_cross: [80.0, 135.0]     # add_cross_arm: true 일 때
  arm_angle_threshold_default: [150.0, 195.0]   # add_cross_arm: false 일 때
  elbow_angle_threshold: [150.0, 185.0]
  hpe_frame_check_max_count: 3

# --- 디스플레이 설정 ---
display:
  turn_on: false # 화면 표시 여부
  border_thickness: 40
  border_thickness_half: 20
  normal_thickness: 2
  warning_thickness: 4
  hpe_thickness_ratio: 1
  text_size: 1
  text_thickness: 0
  ppe_union_min_percent: 0.9
  loadstreams_img_buffer: 5
  fhd_resolution: [1920, 1080]

# --- CCTV 관제 설정 ---
cctv:
  switch_interval: 5.0
  sources:
    # - {src: 0, name: "Realtek_Webcam_0"}
    # - {src: "rtsp://daool:Ekdnfeldpsdptm1@192.168.200.101/axis-media/media.amp", name: "ipcam_1"}
    # - {src: "rtsp://daool:Ekdnfeldpsdptm1@192.168.200.102/axis-media/media.amp", name: "ipcam_2"}
    # - {src: "rtsp://192.168.200.231:50199/wd", name: "dev1_stream"}
    # - {src: "rtsp://192.168.200.232:50299/wd", name: "dev2_stream"}
    # - {src: "rtsp://192.168.200.236:8554/videodevice", name: "boss_webcam"}
    # - {src: "rtsp://192.168.200.236:8554/1.mp4", name: "boss_stream1"}
    # - {src: "rtsp://192.168.200.214/1.mp4", name: "sgm_stream1"}
    # - {src: "rtsp://192.168.200.214/2.mp4", name: "sgm_stream2"}
    # - {src: "rtsp://192.168.200.111/1.mp4", name: "hp_stream1"}
    # - {src: "rtsp://192.168.200.111/2.mp4", name: "hp_stream2"}
    # - {src: "rtsp://192.168.200.215/1.mp4", name: "msk_stream1"}
    # - {src: "rtsp://192.168.200.215/2.mp4", name: "msk_stream2"}

    - {src: "rtsp://192.168.200.233:50399/h1", name: "dev3_stream_hospital_1"}
    - {src: "rtsp://192.168.200.233:50399/h2", name: "dev3_stream_hospital_2"}
    - {src: "rtsp://192.168.200.233:50399/h3", name: "dev3_stream_hospital_3"}
    - {src: "rtsp://192.168.200.233:50399/h4", name: "dev3_stream_hospital_4"}
    - {src: "rtsp://192.168.200.233:50399/h5", name: "dev3_stream_hospital_5"}
    - {src: "rtsp://192.168.200.233:50399/h6", name: "dev3_stream_hospital_6"}

    # - {src: "rtsp://192.168.200.233:50399/k1", name: "dev3_stream_kepco_1"}
    # - {src: "rtsp://192.168.200.233:50399/k2", name: "dev3_stream_kepco_2"}
    # - {src: "rtsp://192.168.200.233:50399/k3", name: "dev3_stream_kepco_3"}
    # - {src: "rtsp://192.168.200.233:50399/k4", name: "dev3_stream_kepco_4"}
    # - {src: "rtsp://192.168.200.233:50399/k5", name: "dev3_stream_kepco_5"}
    # - {src: "rtsp://192.168.200.233:50399/k6", name: "dev3_stream_kepco_6"}