o /4i8@sXdZddlmZmZmZmZmZddlZddlZddl Z ddl Z GdddZ e Z dS)u9 检测器注册中心 管理所有已注册的检测器 )DictTypeOptionalListAnyNc seZdZdZdZdZfddZddZd2ded e fd d Z d e d e e fddZd3d e de d e e fddZd ee effddZd e d e efddZd e fddZddZd e d ee e ffddZ d4d e de dee efd ee effd d!Zd"d#Zd$d%Zd4d e d&e d ee effd'd(Zd2d e d)e d ee effd*d+Zd e dee efd e fd,d-Z  d5d e d&e dee efd e fd.d/Zd e d ee effd0d1ZZS)6DetectorRegistryu检测器注册中心NFcs"|jdurtt|||_|jS)u 单例模式N) _instancesuperr__new__)cls __class__;D:\pyccd\极耳翻折-拆分\detectors\detector_registry.pyr s zDetectorRegistry.__new__cCs<|jsi|_i|_tt|_i|_t |_ d|_dSdS)u初始化注册中心TN) _initialized detectorsdetector_instancesloggingZ getLogger__name__logger params_cache threadingLock cache_lockselfrrr__init__s   zDetectorRegistry.__init__detector_class overwritecCsnddlm}t||std||}|j}||jvr'|s'td|d||j|<|jd|dS)u 注册检测器类 Args: detector_class: 检测器类，必须继承BaseDetector overwrite: 是否覆盖已存在的检测器 ) BaseDetectoru&检测器类必须继承BaseDetector: u检测器已存在: u ，使用overwrite=True来覆盖u注册检测器: N) base_detectorr issubclass ValueErrornamerrinfo)rrrr Z temp_instance detector_namerrrregister&s   zDetectorRegistry.registerr&returnc Os~||jvr|jd|dSz|j|}||i|}|WSty>}z|jd|d|WYd}~dSd}~ww)u 创建检测器实例 Args: detector_name: 检测器名称 *args, **kwargs: 传递给检测器构造函数的参数 Returns: 检测器实例，如果不存在则返回None u未找到检测器: Nu创建检测器实例失败 : )rrerror Exception)rr&argskwargsrinstanceerrrcreate_detector=s  z DetectorRegistry.create_detectorTcreate_if_not_existscCs8||jvr |j|S|r||}|r||j|<|SdS)u 获取检测器实例（单例模式） Args: detector_name: 检测器名称 create_if_not_exists: 如果实例不存在是否创建 Returns: 检测器实例，如果不存在则返回None N)rr0)rr&r1r.rrr get_detectorTs   zDetectorRegistry.get_detectorcC |jS)u~ 获取所有已注册的检测器 Returns: 检测器名称到类的映射字典 rcopyrrrrlist_detectorsj zDetectorRegistry.list_detectorsc Cs|j|dd}|r |S||jvrAz||}|r|WSWdSty@}z|jd|d|WYd}~dSd}~wwdS)u 获取检测器信息 Args: detector_name: 检测器名称 Returns: 检测器信息字典，如果不存在则返回None F)r1u获取检测器信息失败 r)N)r2get_inforr0r+rr*)rr&ZdetectorZ temp_detectorr/rrrget_detector_infoss     "z"DetectorRegistry.get_detector_infocCsV||jvr |j|=||jvr |j|}t|dr||j|=|jd|dS)uc 注销检测器 Args: detector_name: 检测器名称 cleanupu注销检测器: N)rrhasattrr:rr%)rr&r.rrr unregisters    zDetectorRegistry.unregistercCr3)u 获取所有已注册的检测器 Returns: Dict[str, Type]: 检测器名称到检测器类的映射 r4rrrrget_registered_detectorsr7z)DetectorRegistry.get_registered_detectorscCs6||jvriS||}|rt|dr|SddiS)u 获取指定检测器的可用函数列表 Args: detector_name: 检测器名称 Returns: Dict[str, str]: 函数名称到描述的映射 get_available_functionsdetectu主检测函数)rr2r;r>)rr&detector_instancerrrget_detector_functionss  z'DetectorRegistry.get_detector_functions function_nameparamscCs||}|std|||}|r|}||n|}t|dr-||||S|dkr7|||Std|d|)uH 调用检测器的指定函数 Args: detector_name: 检测器名称 function_name: 函数名称 image: 输入图像 params: 外部传入的参数（可选），会与缓存参数合并 Returns: Dict: 检测结果 检测器不存在: call_functionr?u 检测器 u 不支持函数: )r2r#get_detector_paramsr5updater;rEr?)rr&rBimagerCr@Z cached_paramsZ final_paramsrrrcall_detector_functions     z'DetectorRegistry.call_detector_functioncCst|jS)u 获取所有已注册检测器的名称列表 Returns: List[str]: 检测器名称列表 )listrkeysrrrrlist_detector_namessz$DetectorRegistry.list_detector_namescCsv|jD] }t|dr|q|j|j|j |jWdn1s.wY|j ddS)u清空所有注册的检测器r:Nu清空所有检测器注册) rvaluesr;r:rclearrrrr%)rr.rrrrNs    zDetectorRegistry.clearconfig_file_pathc CsN||}|s|jd|iS|}|dur0t|dr$|}n|d}tjd|}| }tj |rz,t |ddd }t |}Wdn1sRwY|||jd |Wn%ty} z|jd | d WYd} ~ nd} ~ ww|jd ||j||j|<Wd|S1swY|S) u 加载检测器参数到缓存 Args: detector_name: 检测器名称 config_file_path: 配置文件路径，如果为None则使用默认路径 Returns: Dict[str, Any]: 合并后的参数字典 rDNget_config_filename _params.jsonconfigrutf-8encodingu已加载检测器配置: u加载检测器配置失败: u，使用默认参数u,配置文件不存在，使用默认参数: )r2rr*get_default_parametersr;rPospathjoinr5existsopenjsonloadrGr%r+Zwarningrr) rr&rOr@Zdefault_paramsconfig_filenameZ merged_paramsfZ json_paramsr/rrrload_detector_paramss:          z%DetectorRegistry.load_detector_paramsreloadcCsZ|j|s||jvr|j|WdSWdn1s#wY||S)u  获取检测器参数（从缓存或重新加载） Args: detector_name: 检测器名称 reload: 是否强制重新加载参数 Returns: Dict[str, Any]: 检测器参数字典 N)rrr5ra)rr&rbrrrrF"s   z$DetectorRegistry.get_detector_paramscCsj||jvr|jd|dS|j||j|<Wdn1s%wY|jd|dS)u 更新检测器参数缓存 Args: detector_name: 检测器名称 params: 新的参数字典 Returns: bool: 更新是否成功 rDFNu 已更新检测器参数缓存: T)rrr*rr5rr%)rr&rCrrrupdate_detector_params4s z'DetectorRegistry.update_detector_paramsc CsR|dur6|j'||jvr |jd|d WddS|j|}Wdn1s1wY|durU||}t|drI|}n|d}tj d|}tj tj |dd z,t |d d d }tj||d ddWdn1s|wY|jd|WdSty}z|jd|WYd}~dSd}~ww)uc 保存检测器参数到文件 Args: detector_name: 检测器名称 config_file_path: 配置文件路径，如果为None则使用默认路径 params: 要保存的参数字典，如果为None则使用缓存中的参数 Returns: bool: 保存是否成功 Nu参数缓存中没有检测器 u 的参数FrPrQrRT)exist_okwrTrU)indentZ ensure_asciiu已保存检测器参数: u保存检测器参数失败: )rrrr*r5r2r;rPrXrYrZmakedirsdirnamer\r]dumpr%r+)rr&rOrCr@r_r`r/rrrsave_detector_paramsIs4     z%DetectorRegistry.save_detector_paramscCsL|j||jvr|||j|iWdS1swYdS)u 获取用于TreeViewer的参数（返回缓存的引用，允许直接修改） Args: detector_name: 检测器名称 Returns: Dict[str, Any]: 参数字典的引用 N)rrraget)rr&rrrget_params_for_tree_viewerts    $z+DetectorRegistry.get_params_for_tree_viewer)F)T)N)NN) r __module__ __qualname____doc__rrr rrboolr'strrobjectr0r2rr6r9r<r=rArrIrLrNrarFrcrkrm __classcell__rrr rr sB       #  .  "+r) rptypingrrrrrrrr]rXrdetector_registryrrrrs }