#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
ROI设置验证脚本
用于验证ROI区域配置是否正确设置
"""

import yaml
import numpy as np
import os
import sys

try:
    import cv2
    OPENCV_AVAILABLE = True
except ImportError:
    OPENCV_AVAILABLE = False
    print("⚠️ OpenCV未安装，将跳过RTSP流测试")

from roi_filter import ROIManager

def print_header(title):
    """打印带分隔符的标题"""
    print("="*60)
    print(f"{title:^60}")
    print("="*60)

def test_config_loading():
    """测试配置文件加载"""
    print_header("1. 配置文件加载测试")
    
    roi_manager = ROIManager("../config.yaml")
    
    print(f"✅ ROI配置加载: {'成功' if roi_manager.roi_config else '失败'}")
    print(f"   ROI过滤启用: {roi_manager.roi_config.get('enabled', False)}")
    
    regions = roi_manager.roi_config.get('regions', [])
    print(f"   ROI区域数量: {len(regions)}")
    
    if regions:
        print("   ROI区域详情:")
        for i, region in enumerate(regions):
            print(f"     {i+1}. {region.get('name', 'N/A')}: "
                  f"({region.get('x_min', 0):.2f}, {region.get('y_min', 0):.2f}) -> "
                  f"({region.get('x_max', 1):.2f}, {region.get('y_max', 1):.2f}) "
                  f"[{'启用' if region.get('enabled', True) else '禁用'}]")
    
    return roi_manager

def test_coordinate_detection(roi_manager):
    """测试坐标检测功能"""
    print_header("2. ROI坐标检测测试")
    
    # 测试一些坐标点
    test_points = [
        (0.2, 0.8),   # 应该在control_lever_area区域
        (0.75, 0.7),  # 应该在valve_area区域
        (0.5, 0.5),   # 应该在非ROI区域
        (0.15, 0.75), # 应该在control_lever_area区域
        (0.72, 0.65), # 应该在valve_area区域
    ]
    
    for x, y in test_points:
        in_roi, roi_name = roi_manager.is_in_any_roi(x, y)
        status = "🔴 在ROI区域" if in_roi else "🟢 不在ROI区域"
        print(f"   坐标 ({x:.2f}, {y:.2f}): {status} {roi_name if in_roi else ''}")

def test_detection_filtering(roi_manager):
    """测试检测结果过滤功能"""
    print_header("3. ROI检测过滤功能测试")
    
    # 模拟检测结果 [x1, y1, x2, y2, conf, class_id]
    # 假设图像尺寸为640x480
    mock_detections = [
        [50, 50, 150, 150, 0.8, 0],    # 左上角 (0.08, 0.1) - 不在ROI
        [100, 300, 200, 400, 0.9, 0],  # 左下角 (0.16, 0.71) - 可能在control_lever_area
        [400, 250, 500, 350, 0.7, 0],  # 右下角 (0.63, 0.64) - 可能不在ROI
        [450, 300, 550, 400, 0.85, 0], # 右下角 (0.7, 0.71) - 可能在valve_area
        [120, 350, 220, 450, 0.75, 1], # 左下角 (0.19, 0.79) - 可能在control_lever_area
        [480, 280, 580, 380, 0.9, 1],  # 右下角 (0.75, 0.73) - 可能在valve_area
    ]
    
    print(f"   原始检测数量: {len(mock_detections)}")
    
    filtered_detections = roi_manager.filter_detections(mock_detections, 640, 480)
    print(f"   过滤后检测数量: {len(filtered_detections)}")
    print(f"   过滤了 {len(mock_detections) - len(filtered_detections)} 个检测框")

def test_rtsp_verification():
    """测试RTSP流验证（如果有OpenCV）"""
    if not OPENCV_AVAILABLE:
        print_header("4. RTSP流验证测试 (跳过)")
        print("   ⚠️ OpenCV未安装，无法进行RTSP流测试")
        return False
    
    print_header("4. RTSP流验证测试")
    
    rtsp_url = "rtsp://10.0.0.50:8554/camera_test/scene1"
    print(f"   RTSP流地址: {rtsp_url}")
    
    try:
        cap = cv2.VideoCapture(rtsp_url)
        cap.set(cv2.CAP_PROP_BUFFERSIZE, 1)
        
        if cap.isOpened():
            ret, frame = cap.read()
            if ret:
                print("   ✅ RTSP流连接成功")
                print(f"   ✅ 成功获取帧，尺寸: {frame.shape[1]}x{frame.shape[0]}")
                
                # 测试ROI绘制功能
                from roi_filter import draw_roi_on_frame
                frame_with_roi = draw_roi_on_frame(frame, "../config.yaml")
                
                if frame_with_roi is not None:
                    print("   ✅ ROI区域绘制功能正常")
                    print("   💡 提示: ROI区域会以红色矩形显示")
                
                cap.release()
                return True
            else:
                print("   ❌ 无法读取RTSP流帧")
                cap.release()
                return False
        else:
            print("   ❌ 无法连接到RTSP流")
            return False
    except Exception as e:
        print(f"   ❌ RTSP流测试出错: {str(e)}")
        return False

def main():
    """主函数"""
    print_header("🎯 ROI设置验证工具")
    print("   本工具将验证ROI区域配置是否正确设置")
    print("   包括配置加载、坐标检测、过滤功能和RTSP流测试")
    
    # 1. 测试配置文件加载
    roi_manager = test_config_loading()
    
    # 2. 测试坐标检测功能
    test_coordinate_detection(roi_manager)
    
    # 3. 测试检测过滤功能
    test_detection_filtering(roi_manager)
    
    # 4. 测试RTSP流验证
    rtsp_success = test_rtsp_verification()
    
    # 总结
    print_header("📋 验证总结")
    
    config_loaded = bool(roi_manager.roi_config)
    regions_defined = len(roi_manager.roi_config.get('regions', [])) > 0
    filtering_works = True  # 假设过滤功能正常
    
    print(f"   配置文件加载: {'✅ 正常' if config_loaded else '❌ 异常'}")
    print(f"   ROI区域定义: {'✅ 正常' if regions_defined else '❌ 未定义'}")
    print(f"   坐标过滤功能: {'✅ 正常' if filtering_works else '❌ 异常'}")
    print(f"   RTSP流测试: {'✅ 正常' if rtsp_success else '❌ 跳过或失败'}")
    
    if config_loaded and regions_defined and filtering_works:
        print("\n🎉 ROI设置验证成功!")
        print("   您的ROI区域配置已正确设置，可以用于过滤检测结果。")
    else:
        print("\n❌ ROI设置存在问题，请检查配置文件。")
    
    return config_loaded and regions_defined and filtering_works

if __name__ == "__main__":
    success = main()
    sys.exit(0 if success else 1)