import torch


def get_default_config():
    return {
        'rtmp': {
            # 'url': "rtmp://123.132.248.154:6009/live/14",
            'url': "rtmp://localhost:1935/live/14",
            'max_reconnect_attempts': 20,  # 增加重连次数
            'reconnect_delay': 1,  # 减少初始延迟
            'buffer_size': 1,
            'timeout_ms': 5000,
            'gpu_decode': True  # 启用硬件解码
        },
        'push': {
            'enable_push': True,
            # 'url': 'rtmp://123.132.248.154:6009/live/11',
            'url': 'rtmp://localhost:1935/live/13',
            'format': 'flv',
            'video_codec': 'h264_nvenc' if torch.cuda.is_available() else 'libx264',  # 使用硬件编码
            'pixel_format': 'bgr24',
            'preset': 'p1' if torch.cuda.is_available() else 'ultrafast',  # NVIDIA专用预设
            'framerate': 30,
            'gpu_acceleration': True,  # 启用硬件加速
            'tune': 'll',  # 低延迟模式
            'zerolatency': 1,  # 零延迟
            'delay': 0,  # 无延迟
            'rc': 'cbr_ld_hq',  # 恒定码率低延迟高质量
            'bufsize': '500k'  # 减少缓冲区大小
        },
        'model': {
            'path': 'yolo11x.pt',  # 保留大模型
            'download_url': 'https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8x.pt'
        },
        'predict': {
            'conf_thres': 0.25,  # 提高置信度阈值
            'iou_thres': 0.45,
            'imgsz': 1280,  # 初始推理尺寸
            'line_width': 1,
            'font_size': 18,
            'font':"E:\\tsgz\\fonts\\PingFangSC-Medium.ttf",
            'device': 'cuda:0' if torch.cuda.is_available() else 'cpu',
            'frame_skip': 1,  # 初始跳帧值
            'half': True  # 启用半精度推理
        },
        'task': {
            'taskname': '',  # 文件夹名称
            'taskid': '',
            'tag': {},
            'aiid': '',
            'res_api': 'http://123.132.248.154:6033/api/DaHuaAi/AddImg',
            'api': 'http://123.132.248.154:6033/'
        },
        'mqtt': {
            'enable': True,  # 是否启用MQTT
            'broker': '175.27.168.120',  # MQTT代理地址
            'port': 6011,  # MQTT端口
            'topic': 'thing/product/1581F8HGX254V00A0BUY/osd',  # 订阅的主题
            'client_id': 'yolo_detection_client',  # 客户端ID------自己生成个
            'username': 'sdhc',  # 用户名（可选）
            'password': None,  # 密码（可选）
            'keepalive': 60  # 保活时间
        },
        'minio': {
            "UseSSL": False,
            "Endpoint": "175.27.168.120:6013",
            "AccessKey": "minioadmin",
            "SecretKey": "minioadmin",
            "BucketName": "test"
        }
    }