"""
快速测试单个标注文件格式
用于验证从CVAT导出的多边形标注是否正确
"""
import sys
from verify_seg_labels import verify_yolo_seg_format, visualize_seg_label

def test_label_file(label_path, img_path=None):
    """测试单个标注文件"""
    print(f"验证标注文件: {label_path}")
    print("-" * 60)
    
    result = verify_yolo_seg_format(label_path, img_path, show_visualization=(img_path is not None))
    
    if result["valid"]:
        print("✓ 格式正确！")
        print(f"  类别ID: {result['class_id']}")
        print(f"  多边形顶点数: {result['num_points']}")
        print(f"  坐标值数量: {result['coords_count']}")
    else:
        print("✗ 格式错误！")
    
    if result["errors"]:
        print("错误信息:")
        for error in result["errors"]:
            print(f"  - {error}")
    
    if result["warnings"]:
        print("警告信息:")
        for warning in result["warnings"]:
            print(f"  - {warning}")
    
    if img_path and result["valid"]:
        print("\n显示可视化结果...")
        visualize_seg_label(img_path, label_path)

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("使用方法: python test_seg_label.py <label_file> [image_file]")
        print("\n示例:")
        print("  python test_seg_label.py datasets/搭电设备/train/labels/your_file.txt")
        print("  python test_seg_label.py datasets/搭电设备/train/labels/your_file.txt datasets/搭电设备/train/images/your_file.jpg")
        sys.exit(1)
    
    label_path = sys.argv[1]
    img_path = sys.argv[2] if len(sys.argv) > 2 else None
    
    test_label_file(label_path, img_path)