# -*- coding: utf-8 -*-
import cv2
import numpy as np

def draw_detection_results(frame, results):
    """
    在图像上绘制检测结果
    
    Args:
        frame: BGR格式的图像
        results: 检测结果列表，每个元素为(cx, cy, w, h, area, circularity)
        
    Returns:
        frame: 绘制后的图像
    """
    for result in results:
        cx, cy, w, h, area, circularity = result
        
        # 计算矩形框坐标
        x = cx - w // 2
        y = cy - h // 2
        
        # 绘制绿色矩形框
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
        
        # 绘制十字准星
        cv2.line(frame, (cx - 10, cy), (cx + 10, cy), (0, 255, 0), 2)
        cv2.line(frame, (cx, cy - 10), (cx, cy + 10), (0, 255, 0), 2)
        
        # 绘制坐标文字
        text = f"({cx}, {cy})"
        cv2.putText(frame, text, (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
    
    return frame

def draw_status_bar(frame, exposure_time, gain, detected_count, fps):
    """
    在图像上绘制状态栏信息
    
    Args:
        frame: BGR格式的图像
        exposure_time: 当前曝光时间（微秒）
        gain: 当前增益（dB）
        detected_count: 检测到的目标数
        fps: 当前帧率
        
    Returns:
        frame: 绘制后的图像
    """
    # 在图像顶部添加状态栏
    status_bar = np.zeros((30, frame.shape[1], 3), dtype=np.uint8)
    
    # 绘制信息文字
    text = f"Exposure: {exposure_time:.0f}μs | Gain: {gain:.1f}dB | Detected: {detected_count} | FPS: {fps:.1f}"
    cv2.putText(status_bar, text, (10, 20), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 1)
    
    # 将状态栏添加到图像顶部
    frame_with_status = np.vstack((status_bar, frame))
    
    return frame_with_status