dsh/dual_camera_system/paired_image_saver.py

"""
配对图片保存管理器
将全景检测图片和对应的球机聚焦图片保存到同一目录
支持 OSS 上传和 batch_info.json 格式
"""

import os
import cv2
import time
import json
import logging
import threading
from pathlib import Path
from datetime import datetime
from typing import Optional, List, Dict, Tuple, Callable
from dataclasses import dataclass, field, asdict

logger = logging.getLogger(__name__)


@dataclass
class PersonInfo:
    """人员信息（轨迹追踪已禁用）"""
    person_index: int  # 人员序号（0-based）
    position: Tuple[float, float]  # (x_ratio, y_ratio)
    bbox: Tuple[int, int, int, int]  # (x1, y1, x2, y2)
    confidence: float
    ptz_position: Optional[Tuple[float, float, int]] = None  # (pan, tilt, zoom)
    ptz_bbox: Optional[Tuple[int, int, int, int]] = None  # 球机图中检测到的bbox (x1, y1, x2, y2)
    ptz_image_saved: bool = False
    ptz_image_path: Optional[str] = None
    ptz_oss_url: Optional[str] = None  # 球机图 OSS URL


@dataclass
class DetectionBatch:
    """一批检测记录"""
    batch_id: str
    timestamp: float
    panorama_image: Optional[object] = None  # numpy array
    panorama_path: Optional[str] = None
    panorama_oss_url: Optional[str] = None  # 全景图 OSS URL
    persons: List[PersonInfo] = field(default_factory=list)
    total_persons: int = 0
    ptz_images_count: int = 0
    completed: bool = False
    device_id: str = ''  # 设备编号
    project_id: str = ''  # 项目编号


class PairedImageSaver:
    """
    配对图片保存管理器
    
    功能：
    1. 为每次全景检测创建批次目录
    2. 保存全景标记图到批次目录
    3. 为每个人员保存对应的球机聚焦图到同一目录
    4. 支持时间窗口内的批量保存
    5. 支持 OSS 上传
    6. 生成 batch_info.json
    """
    
    def __init__(self, base_dir: str = '/home/admin/dsh/paired_images',
                 time_window: float = 5.0,  # 时间窗口（秒）
                 max_batches: int = 100,
                 enable_oss: bool = False,
                 oss_uploader = None,
                 device_config: Dict = None):
        """
        初始化
        
        Args:
            base_dir: 基础保存目录
            time_window: 批次时间窗口（秒），同一窗口内的检测归为一批
            max_batches: 最大保留批次数量
            enable_oss: 是否启用 OSS 上传
            oss_uploader: OSS 上传器实例
            device_config: 设备配置字典
        """
        self.base_dir = Path(base_dir)
        self.time_window = time_window
        self.max_batches = max_batches
        
        # 从配置模块读取 OSS 和设备配置（确保即使外部不传也能正确配置）
        try:
            from config import S3_COMPATIBLE_CONFIG, DEVICE_CONFIG

            # OSS 配置：优先使用传入参数，否则从配置模块读取
            oss_enabled_in_config = S3_COMPATIBLE_CONFIG.get('enabled', False)
            if enable_oss or (not enable_oss and oss_enabled_in_config):
                self.enable_oss = True
            else:
                self.enable_oss = enable_oss

            logger.info(f"[配对保存] OSS配置: enable_oss={enable_oss}, config_enabled={oss_enabled_in_config}, 最终启用={self.enable_oss}")

            # OSS 上传器：优先使用传入的实例，否则从全局获取
            if oss_uploader is not None:
                self.oss_uploader = oss_uploader
                logger.info("[配对保存] 使用传入的 OSS 上传器")
            elif self.enable_oss:
                try:
                    from oss_uploader import get_oss_uploader
                    self.oss_uploader = get_oss_uploader()
                    logger.info(f"[配对保存] 获取 OSS 上传器: enabled={self.oss_uploader.enabled}, running={getattr(self.oss_uploader, 'running', False)}")
                    if not self.oss_uploader.running:
                        self.oss_uploader.start()
                        logger.info("[配对保存] OSS 上传器已启动")
                except Exception as e:
                    logger.warning(f"[配对保存] OSS 上传器初始化失败: {e}")
                    self.oss_uploader = None
                    self.enable_oss = False
            else:
                self.oss_uploader = None

            # 设备配置：合并传入参数和配置模块
            self.device_config = DEVICE_CONFIG.copy()
            if device_config:
                self.device_config.update(device_config)

        except ImportError as e:
            # 配置模块不可用时使用传入参数
            logger.warning(f"[配对保存] 配置模块导入失败: {e}")
            self.enable_oss = enable_oss
            self.oss_uploader = oss_uploader
            self.device_config = device_config or {}
        
        self._current_batch: Optional[DetectionBatch] = None
        self._batch_lock = threading.Lock()
        self._last_batch_time = 0.0
        
        # 上传状态追踪
        self._upload_status: Dict[str, Dict] = {}  # batch_id -> {panorama: bool, ptz: Dict}
        self._upload_callback: Optional[Callable] = None
        
        # 统计信息
        self._stats = {
            'total_batches': 0,
            'total_persons': 0,
            'total_ptz_images': 0,
            'oss_upload_success': 0,
            'oss_upload_failed': 0,
        }
        self._stats_lock = threading.Lock()
        
        # 确保目录存在
        self._ensure_base_dir()
        
        logger.info(f"[配对保存] 初始化完成: 目录={base_dir}, 时间窗口={time_window}s, OSS={enable_oss}")
    
    def _ensure_base_dir(self):
        """确保基础目录存在"""
        try:
            self.base_dir.mkdir(parents=True, exist_ok=True)
        except Exception as e:
            logger.error(f"[配对保存] 创建目录失败: {e}")
    
    def _generate_batch_id(self) -> str:
        """生成批次ID"""
        return datetime.now().strftime("%Y%m%d_%H%M%S_%f")[:-3]
    
    def _create_batch_dir(self, batch_id: str) -> Path:
        """创建批次目录"""
        batch_dir = self.base_dir / f"batch_{batch_id}"
        try:
            batch_dir.mkdir(parents=True, exist_ok=True)
            return batch_dir
        except Exception as e:
            logger.error(f"[配对保存] 创建批次目录失败: {e}")
            return self.base_dir
    
    def start_new_batch(self, panorama_frame, persons: List[Dict]) -> Optional[str]:
        """
        开始新批次
        
        Args:
            panorama_frame: 全景帧图像
            persons: 人员列表，每项包含 track_id, position, bbox, confidence
            
        Returns:
            batch_id: 批次ID，失败返回 None
        """
        with self._batch_lock:
            current_time = time.time()
            
            # 完成上一批次（如果有）
            # 注意：每次检测都创建独立批次，不复用，确保 batch_info 与实际检测一致
            if self._current_batch is not None:
                self._finalize_batch(self._current_batch)
            
            # 创建新批次
            batch_id = self._generate_batch_id()
            batch_dir = self._create_batch_dir(batch_id)
            
            # 保存全景图片
            panorama_path = None
            if panorama_frame is not None:
                panorama_path = self._save_panorama_image(
                    batch_dir, batch_id, panorama_frame, persons
                )
            
            # 创建人员信息（轨迹追踪已禁用，使用序号代替track_id）
            person_infos = []
            for i, p in enumerate(persons):
                info = PersonInfo(
                    person_index=i,
                    position=p.get('position', (0, 0)),
                    bbox=p.get('bbox', (0, 0, 0, 0)),
                    confidence=p.get('confidence', 0.0)
                )
                person_infos.append(info)
            
            # 获取设备信息
            device_id = self.device_config.get('device_id', 'UNKNOWN')
            project_id = self.device_config.get('project_id', 'UNKNOWN')
            
            # 创建批次记录
            self._current_batch = DetectionBatch(
                batch_id=batch_id,
                timestamp=current_time,
                panorama_image=panorama_frame,
                panorama_path=panorama_path,
                persons=person_infos,
                total_persons=len(persons),
                device_id=device_id,
                project_id=project_id
            )
            
            # 初始化上传状态
            self._upload_status[batch_id] = {
                'panorama': False,
                'panorama_url': None,
                'ptz': {},
                'completed': False
            }
            
            self._last_batch_time = current_time
            
            with self._stats_lock:
                self._stats['total_batches'] += 1
                self._stats['total_persons'] += len(persons)
            
            logger.info(
                f"[配对保存] 新批次创建: {batch_id}, "
                f"人员={len(persons)}, 目录={batch_dir}"
            )
            
            # 上传全景图到 OSS
            logger.info(f"[配对保存] 开始新批次: enable_oss={self.enable_oss}, uploader={self.oss_uploader}, path={panorama_path}")
            if self.enable_oss and panorama_path and self.oss_uploader:
                logger.info(f"[配对保存] 准备上传全景图到 OSS: {panorama_path}")
                self._upload_panorama_to_oss(batch_id, panorama_path)
            else:
                logger.warning(f"[配对保存] OSS未启用或上传器不可用: enable_oss={self.enable_oss}, uploader={self.oss_uploader}")
            
            return batch_id
    
    def _save_panorama_image(self, batch_dir: Path, batch_id: str,
                              frame, persons: List[Dict]) -> Optional[str]:
        """
        保存全景标记图片
        
        Args:
            batch_dir: 批次目录
            batch_id: 批次ID
            frame: 全景帧
            persons: 人员列表（已由调用方过滤，此处不再过滤）
            
        Returns:
            保存路径或 None
        """
        try:
            # 复制图像避免修改原图
            marked_frame = frame.copy()
            
            # 绘制每个人员的标记（使用连续的序号）
            # 注意：persons 已由调用方（coordinator）过滤，置信度均 >= 阈值
            for i, person in enumerate(persons):
                bbox = person.get('bbox', (0, 0, 0, 0))
                x1, y1, x2, y2 = bbox
                conf = person.get('confidence', 0.0)
                
                # 绘制边界框（绿色）
                cv2.rectangle(marked_frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
                
                # 绘制序号标签（带置信度）
                label = f"person_{i}({conf:.2f})"
                (label_w, label_h), baseline = cv2.getTextSize(
                    label, cv2.FONT_HERSHEY_SIMPLEX, 0.8, 2
                )
                
                # 标签背景
                cv2.rectangle(
                    marked_frame,
                    (x1, y1 - label_h - 8),
                    (x1 + label_w, y1),
                    (0, 255, 0),
                    -1
                )
                
                # 标签文字（黑色）
                cv2.putText(
                    marked_frame, label,
                    (x1, y1 - 4),
                    cv2.FONT_HERSHEY_SIMPLEX, 0.8,
                    (0, 0, 0), 2
                )
            
            # 保存图片（使用人员数量）
            filename = f"00_panorama_n{len(persons)}.png"
            filepath = batch_dir / filename
            # 保存全景图（PNG无损格式，不压缩）
            cv2.imwrite(str(filepath), marked_frame)
            
            logger.info(f"[配对保存] 全景图已保存: {filepath}，人员数量 {len(persons)}")
            return str(filepath)
            
        except Exception as e:
            logger.error(f"[配对保存] 保存全景图失败: {e}")
            return None
    
    def save_ptz_image(self, batch_id: str, person_index: int,
                       ptz_frame, ptz_position: Tuple[float, float, int],
                       ptz_bbox: Tuple[int, int, int, int] = None,
                       person_info: Dict = None) -> Optional[str]:
        """
        保存球机聚焦图片

        Args:
            batch_id: 批次ID
            person_index: 人员序号（0-based）
            ptz_frame: 球机帧
            ptz_position: PTZ位置 (pan, tilt, zoom)
            ptz_bbox: 球机图中检测到的bbox (x1, y1, x2, y2)
            person_info: 额外人员信息

        Returns:
            保存路径或 None
        """
        # 【调试】记录传入的参数
        logger.info(f"[配对保存] save_ptz_image: batch={batch_id}, person={person_index}, "
                   f"PTZ=({ptz_position[0]:.1f}°, {ptz_position[1]:.1f}°, zoom={ptz_position[2]})")

        with self._batch_lock:
            if self._current_batch is None or self._current_batch.batch_id != batch_id:
                logger.warning(f"[配对保存] 批次不存在或已过期: {batch_id}, current={self._current_batch.batch_id if self._current_batch else None}")
                return None

            batch_dir = self.base_dir / f"batch_{batch_id}"
            
            try:
                # 复制图像
                marked_frame = ptz_frame.copy() if ptz_frame is not None else None
                
                if marked_frame is not None:
                    # 在球机图上添加标记（如果有检测框）
                    h, w = marked_frame.shape[:2]
                    
                    # 添加PTZ位置信息到图片
                    pan, tilt, zoom = ptz_position
                    info_text = f"PTZ: P={pan:.1f} T={tilt:.1f} Z={zoom}"
                    cv2.putText(
                        marked_frame, info_text,
                        (10, 30),
                        cv2.FONT_HERSHEY_SIMPLEX, 0.7,
                        (0, 255, 0), 2
                    )
                    
                    # 添加人员序号
                    person_text = f"person_{person_index}"
                    cv2.putText(
                        marked_frame, person_text,
                        (10, 60),
                        cv2.FONT_HERSHEY_SIMPLEX, 0.7,
                        (0, 255, 0), 2
                    )
                    
                    # 绘制PTZ检测到的bbox（红色）
                    if ptz_bbox is not None:
                        x1, y1, x2, y2 = ptz_bbox
                        cv2.rectangle(marked_frame, (x1, y1), (x2, y2), (0, 0, 255), 2)
                        bbox_text = f"PTZ_BBox: ({x1},{y1},{x2},{y2})"
                        cv2.putText(
                            marked_frame, bbox_text,
                            (10, 90),
                            cv2.FONT_HERSHEY_SIMPLEX, 0.6,
                            (0, 0, 255), 2
                        )
                
                # 保存图片
                filename = f"01_ptz_person{person_index}_p{int(ptz_position[0])}_t{int(ptz_position[1])}_z{int(ptz_position[2])}.jpg"
                filepath = batch_dir / filename
                
                if marked_frame is not None:
                    cv2.imwrite(str(filepath), marked_frame, [cv2.IMWRITE_JPEG_QUALITY, 90])
                
                # 更新批次信息
                # 【关键修复】只有当人员索引有效时才更新和计数
                if person_index < len(self._current_batch.persons):
                    self._current_batch.persons[person_index].ptz_position = ptz_position
                    self._current_batch.persons[person_index].ptz_bbox = ptz_bbox
                    self._current_batch.persons[person_index].ptz_image_saved = True
                    self._current_batch.persons[person_index].ptz_image_path = str(filepath)
                    
                    self._current_batch.ptz_images_count += 1
                    
                    with self._stats_lock:
                        self._stats['total_ptz_images'] += 1
                    
                    logger.info(f"[配对保存] 球机图已保存: {filepath}, BBox={ptz_bbox}")

                    # 上传球机图到 OSS
                    logger.info(f"[配对保存] 准备上传球机图到 OSS: enable_oss={self.enable_oss}, uploader={self.oss_uploader}")
                    if self.enable_oss and self.oss_uploader:
                        self._upload_ptz_to_oss(batch_id, person_index, str(filepath))
                else:
                    # 人员索引超出范围，说明批次信息不一致，跳过保存
                    logger.warning(f"[配对保存] 人员索引 {person_index} 超出批次范围 {len(self._current_batch.persons)}，跳过计数")
                    
                return str(filepath)
                
            except Exception as e:
                logger.error(f"[配对保存] 保存球机图失败: {e}")
                return None
    
    def _upload_panorama_to_oss(self, batch_id: str, panorama_path: str):
        """上传全景图到 OSS"""
        logger.info(f"[OSS] _upload_panorama_to_oss 被调用: batch_id={batch_id}, path={panorama_path}")

        def on_upload_complete(result):
            logger.info(f"[OSS] 全景图上传完成回调: success={result.success}, url={result.oss_url}")
            # 更新上传状态（即使 _current_batch 已切换也能正确记录）
            self._upload_status[batch_id]['panorama'] = True
            self._upload_status[batch_id]['panorama_url'] = result.oss_url
            if self._current_batch and self._current_batch.batch_id == batch_id:
                self._current_batch.panorama_oss_url = result.oss_url
            with self._stats_lock:
                self._stats['oss_upload_success'] += 1
            logger.info(f"[OSS] 全景图上传成功: {result.oss_url}")

        def on_upload_error(result):
            logger.error(f"[OSS] 全景图上传失败回调: {result.error}")
            self._upload_status[batch_id]['panorama'] = False
            self._upload_status[batch_id]['panorama_url'] = None
            with self._stats_lock:
                self._stats['oss_upload_failed'] += 1
            logger.error(f"[OSS] 全景图上传失败: {result.error}")

        # 包装回调，同时处理成功和失败
        def on_upload_done(result):
            logger.info(f"[OSS] 全景图上传结果: success={result.success}")
            if result.success:
                on_upload_complete(result)
            else:
                on_upload_error(result)

        try:
            oss_key = self.oss_uploader.upload_image(
                local_path=panorama_path,
                batch_id=batch_id,
                image_type='panorama',
                callback=on_upload_done
            )
            logger.info(f"[OSS] 全景图已加入上传队列: oss_key={oss_key}")
        except Exception as e:
            logger.error(f"[OSS] 全景图上传异常: {e}")
    
    def _upload_ptz_to_oss(self, batch_id: str, person_index: int, ptz_path: str):
        """上传球机图到 OSS"""
        logger.info(f"[OSS] _upload_ptz_to_oss 被调用: batch_id={batch_id}, person={person_index}, path={ptz_path}")

        def on_upload_complete(result):
            logger.info(f"[OSS] 球机图上传完成回调: person={person_index}, success={result.success}, url={result.oss_url}")
            # 更新上传状态（即使 _current_batch 已切换也能正确记录）
            self._upload_status[batch_id]['ptz'][person_index] = result.oss_url
            if self._current_batch and self._current_batch.batch_id == batch_id:
                if person_index < len(self._current_batch.persons):
                    self._current_batch.persons[person_index].ptz_oss_url = result.oss_url
            with self._stats_lock:
                self._stats['oss_upload_success'] += 1
            logger.info(f"[OSS] 球机图上传成功 (person_{person_index}): {result.oss_url}")

        def on_upload_error(result):
            logger.error(f"[OSS] 球机图上传失败回调: person={person_index}, error={result.error}")
            self._upload_status[batch_id]['ptz'][person_index] = None
            with self._stats_lock:
                self._stats['oss_upload_failed'] += 1
            logger.error(f"[OSS] 球机图上传失败 (person_{person_index}): {result.error}")

        # 包装回调，同时处理成功和失败
        def on_upload_done(result):
            logger.info(f"[OSS] 球机图上传结果: person={person_index}, success={result.success}")
            if result.success:
                on_upload_complete(result)
            else:
                on_upload_error(result)

        try:
            oss_key = self.oss_uploader.upload_image(
                local_path=ptz_path,
                batch_id=batch_id,
                image_type='ptz',
                person_index=person_index,
                callback=on_upload_done
            )
            logger.info(f"[OSS] 球机图已加入上传队列: person={person_index}, oss_key={oss_key}")
        except Exception as e:
            logger.error(f"[OSS] 球机图上传异常: {e}")
    
    def _finalize_batch(self, batch: DetectionBatch):
        """完成批次处理"""
        batch.completed = True

        # 等待 OSS 上传完成（最多等待10秒）
        if self.enable_oss and batch.batch_id in self._upload_status:
            wait_start = time.time()
            max_wait = 10.0  # 增加等待时间
            while time.time() - wait_start < max_wait:
                status = self._upload_status[batch.batch_id]
                # 检查全景图是否上传完成（优先检查 _upload_status 中的状态）
                panorama_url = status.get('panorama_url')
                panorama_done = panorama_url is not None or not batch.panorama_path
                # 检查所有球机图是否上传完成
                ptz_status = status.get('ptz', {})
                ptz_done = all(
                    ptz_status.get(idx) is not None
                    for idx, person in enumerate(batch.persons)
                    if person.ptz_image_saved
                )
                if panorama_done and ptz_done:
                    break
                time.sleep(0.2)
        
        # 创建 batch_info.json 文件
        try:
            batch_dir = self.base_dir / f"batch_{batch.batch_id}"
            
            # 构建 JSON 数据
            batch_info = self._build_batch_info_json(batch)
            
            # 保存为 JSON 文件
            json_path = batch_dir / "batch_info.json"
            with open(json_path, 'w', encoding='utf-8') as f:
                json.dump(batch_info, f, ensure_ascii=False, indent=2)
            
            logger.info(f"[配对保存] batch_info.json 已保存: {json_path}")
            
            # 同时保留 txt 格式用于兼容（可选）
            txt_path = batch_dir / "batch_info.txt"
            self._save_batch_info_txt(batch, txt_path)
            
            # 标记上传完成
            if batch.batch_id in self._upload_status:
                self._upload_status[batch.batch_id]['completed'] = True
            
            # 触发回调
            if self._upload_callback:
                try:
                    self._upload_callback(batch_info)
                except Exception as e:
                    logger.error(f"[配对保存] 回调执行错误: {e}")
            
            logger.info(f"[配对保存] 批次完成: {batch.batch_id}, "
                       f"人员={batch.total_persons}, 球机图={batch.ptz_images_count}")
            
        except Exception as e:
            logger.error(f"[配对保存] 保存批次信息失败: {e}")
        
        # 清理旧批次
        self._cleanup_old_batches()
    
    def _build_batch_info_json(self, batch: DetectionBatch) -> Dict:
        """
        构建 batch_info.json 数据结构

        Returns:
            Dict: 批次信息字典
        """
        # 从上传状态获取最新的 OSS URL（异步回调可能已更新）
        upload_status = self._upload_status.get(batch.batch_id, {})

        # 获取全景图 OSS URL（优先使用回调更新的状态）
        panorama_oss_url = upload_status.get('panorama_url', batch.panorama_oss_url)

        # 人员信息列表
        persons_list = []
        for person in batch.persons:
            # 获取球机图 OSS URL（优先使用回调更新的状态）
            ptz_oss_url = upload_status.get('ptz', {}).get(person.person_index, person.ptz_oss_url)

            person_data = {
                'person_index': person.person_index,
                'position': {
                    'x': round(person.position[0], 4),
                    'y': round(person.position[1], 4)
                },
                'bbox': {
                    'x1': person.bbox[0],
                    'y1': person.bbox[1],
                    'x2': person.bbox[2],
                    'y2': person.bbox[3]
                },
                'confidence': round(person.confidence, 4),
                'ptz_position': {
                    'pan': round(person.ptz_position[0], 2) if person.ptz_position else None,
                    'tilt': round(person.ptz_position[1], 2) if person.ptz_position else None,
                    'zoom': person.ptz_position[2] if person.ptz_position else None
                } if person.ptz_position else None,
                'ptz_bbox': {
                    'x1': person.ptz_bbox[0],
                    'y1': person.ptz_bbox[1],
                    'x2': person.ptz_bbox[2],
                    'y2': person.ptz_bbox[3]
                } if person.ptz_bbox else None,
                'ptz_image_saved': person.ptz_image_saved,
                'ptz_image_path': person.ptz_image_path,
                'ptz_oss_url': ptz_oss_url
            }
            persons_list.append(person_data)

        # 构建完整批次信息
        batch_info = {
            'batch_id': batch.batch_id,
            'device_id': batch.device_id,
            'project_id': batch.project_id,
            'timestamp': batch.timestamp,
            'datetime': datetime.fromtimestamp(batch.timestamp).isoformat(),
            'total_persons': batch.total_persons,
            'ptz_images_count': batch.ptz_images_count,
            'panorama': {
                'local_path': batch.panorama_path,
                'oss_url': panorama_oss_url
            },
            'persons': persons_list,
            'upload_status': {
                'panorama_uploaded': panorama_oss_url is not None,
                'all_ptz_uploaded': all(
                    upload_status.get('ptz', {}).get(p.person_index) is not None
                    for p in batch.persons if p.ptz_image_saved
                )
            }
        }

        return batch_info
    
    def _save_batch_info_txt(self, batch: DetectionBatch, txt_path: Path):
        """保存批次信息为 TXT 格式（兼容旧版本）"""
        try:
            with open(txt_path, 'w', encoding='utf-8') as f:
                f.write(f"批次ID: {batch.batch_id}\n")
                f.write(f"设备ID: {batch.device_id}\n")
                f.write(f"项目ID: {batch.project_id}\n")
                f.write(f"时间戳: {datetime.fromtimestamp(batch.timestamp)}\n")
                f.write(f"总人数: {batch.total_persons}\n")
                f.write(f"球机图数量: {batch.ptz_images_count}\n")
                f.write(f"全景图: {batch.panorama_path}\n")
                f.write(f"全景图OSS: {batch.panorama_oss_url}\n")
                f.write("\n人员详情:\n")
                
                for i, person in enumerate(batch.persons):
                    f.write(f"\n  Person {i}:\n")
                    f.write(f"    Person Index: {person.person_index}\n")
                    f.write(f"    Position: ({person.position[0]:.3f}, {person.position[1]:.3f})\n")
                    f.write(f"    BBox: ({person.bbox[0]}, {person.bbox[1]}, {person.bbox[2]}, {person.bbox[3]})\n")
                    f.write(f"    Confidence: {person.confidence:.2f}\n")
                    f.write(f"    PTZ Position: {person.ptz_position}\n")
                    if person.ptz_bbox:
                        f.write(f"    PTZ BBox: ({person.ptz_bbox[0]}, {person.ptz_bbox[1]}, {person.ptz_bbox[2]}, {person.ptz_bbox[3]})\n")
                    else:
                        f.write(f"    PTZ BBox: None\n")
                    f.write(f"    PTZ Image: {person.ptz_image_path}\n")
                    f.write(f"    PTZ OSS URL: {person.ptz_oss_url}\n")
        except Exception as e:
            logger.error(f"[配对保存] 保存 TXT 批次信息失败: {e}")
    
    def _cleanup_old_batches(self):
        """清理旧批次目录"""
        try:
            batch_dirs = sorted(
                [d for d in self.base_dir.iterdir() if d.is_dir() and d.name.startswith('batch_')],
                key=lambda x: x.stat().st_mtime
            )
            
            if len(batch_dirs) > self.max_batches:
                to_delete = batch_dirs[:len(batch_dirs) - self.max_batches]
                for d in to_delete:
                    import shutil
                    shutil.rmtree(d)
                    logger.info(f"[配对保存] 清理旧批次: {d.name}")
                    
        except Exception as e:
            logger.error(f"[配对保存] 清理旧批次失败: {e}")
    
    def get_current_batch_id(self) -> Optional[str]:
        """获取当前批次ID"""
        with self._batch_lock:
            return self._current_batch.batch_id if self._current_batch else None
    
    def get_stats(self) -> Dict:
        """获取统计信息"""
        with self._stats_lock:
            return self._stats.copy()
    
    def set_upload_callback(self, callback: Callable):
        """
        设置批次完成回调函数
        
        Args:
            callback: 回调函数，接收 batch_info_dict 参数
        """
        self._upload_callback = callback
    
    def get_batch_info(self, batch_id: str) -> Optional[Dict]:
        """
        获取指定批次的 batch_info.json 数据
        
        Args:
            batch_id: 批次ID
            
        Returns:
            Dict 或 None
        """
        try:
            batch_dir = self.base_dir / f"batch_{batch_id}"
            json_path = batch_dir / "batch_info.json"
            
            if json_path.exists():
                with open(json_path, 'r', encoding='utf-8') as f:
                    return json.load(f)
        except Exception as e:
            logger.error(f"[配对保存] 读取 batch_info.json 失败: {e}")
        
        return None
    
    def close(self):
        """关闭管理器，完成当前批次"""
        with self._batch_lock:
            if self._current_batch is not None:
                self._finalize_batch(self._current_batch)
                self._current_batch = None
        
        logger.info("[配对保存] 管理器已关闭")


# 全局单例实例
_paired_saver_instance: Optional[PairedImageSaver] = None


def get_paired_saver(base_dir: str = None, time_window: float = 5.0,
                     enable_oss: bool = False, oss_uploader = None,
                     device_config: Dict = None) -> PairedImageSaver:
    """
    获取配对保存管理器实例（单例模式）
    
    如果实例已存在但缺少 OSS/设备配置，会自动从配置模块更新
    
    Args:
        base_dir: 基础保存目录
        time_window: 时间窗口
        enable_oss: 是否启用 OSS 上传
        oss_uploader: OSS 上传器实例
        device_config: 设备配置字典
        
    Returns:
        PairedImageSaver 实例
    """
    global _paired_saver_instance
    
    if _paired_saver_instance is None:
        _paired_saver_instance = PairedImageSaver(
            base_dir=base_dir or '/home/admin/dsh/paired_images',
            time_window=time_window,
            enable_oss=enable_oss,
            oss_uploader=oss_uploader,
            device_config=device_config
        )
    else:
        # 单例已存在，检查是否需要更新配置
        # PairedImageSaver.__init__ 已从配置模块自动读取，
        # 这里只处理外部显式传入的参数覆盖
        if oss_uploader is not None and _paired_saver_instance.oss_uploader is None:
            _paired_saver_instance.oss_uploader = oss_uploader
            _paired_saver_instance.enable_oss = True
            logger.info("[配对保存] 更新 OSS 上传器配置")
        if device_config is not None:
            _paired_saver_instance.device_config.update(device_config)
            logger.info(f"[配对保存] 更新设备配置: {device_config}")
    
    return _paired_saver_instance


def reset_paired_saver():
    """重置单例实例（用于测试）"""
    global _paired_saver_instance
    if _paired_saver_instance is not None:
        _paired_saver_instance.close()
    _paired_saver_instance = None