#!/usr/bin/env python3

import os
import sys
import time
import cv2

# 添加父目录到路径，以便能够导入utils
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from utils.image_raw import ImageProcessor
from utils.decode_arrow import detect_arrow_direction, visualize_arrow_detection

class ArrowDetector:
    def __init__(self, image_processor=None):
        """
        初始化箭头检测器
        
        参数:
            image_processor: 可选的ImageProcessor实例，如果不提供则会创建一个新的
        """
        # 如果提供了图像处理器，则使用它，否则创建新的
        if image_processor is not None:
            self.image_processor = image_processor
            self.should_destroy = False  # 不应该在destroy方法中销毁外部传入的实例
        else:
            self.image_processor = ImageProcessor()
            self.image_processor.run()
            self.should_destroy = True  # 应该在destroy方法中销毁自己创建的实例
            
        print("箭头检测器已初始化")
    
    def get_arrow_direction(self):
        """
        获取当前图像中箭头的方向
        
        返回:
            direction: 字符串，"left"表示左箭头，"right"表示右箭头，"unknown"表示无法确定
        """
        # 获取当前图像
        image = self.image_processor.get_current_image()
        
        if image is None:
            print("警告: 无法获取图像")
            return "unknown"
        
        # 检测箭头方向
        direction = detect_arrow_direction(image)
        return direction
    
    def visualize_current_detection(self, save_path=None):
        """
        可视化当前图像的箭头检测过程
        
        参数:
            save_path: 保存结果图像的路径（可选）
        """
        # 获取当前图像
        image = self.image_processor.get_current_image()
        
        if image is None:
            print("警告: 无法获取图像")
            return
        
        # 可视化箭头检测
        visualize_arrow_detection(image, save_path)
    
    def destroy(self):
        """
        清理资源
        """
        if self.should_destroy:
            self.image_processor.destroy()
        print("箭头检测器已销毁")

def main():
    """
    演示箭头检测器的用法
    """
    try:
        # 初始化箭头检测器
        detector = ArrowDetector()
        
        # 等待一段时间，确保图像已经接收
        print("等待接收图像...")
        time.sleep(3)
        
        # 持续检测箭头方向
        for i in range(10):  # 检测10次
            direction = detector.get_arrow_direction()
            print(f"检测到的箭头方向 ({i+1}/10): {direction}")
            
            # 可选: 保存第一次检测的可视化结果
            if i == 0:
                detector.visualize_current_detection("arrow_detection_result.jpg")
            
            time.sleep(1)  # 每秒检测一次
    
    except KeyboardInterrupt:
        print("\n程序被用户中断")
    except Exception as e:
        print(f"发生错误: {e}")
    finally:
        # 清理资源
        print("正在清理资源...")
        if 'detector' in locals():
            detector.destroy()
        print("程序已退出")

if __name__ == "__main__":
    main()