"""
测试修改后的处理器
"""
from pipeline.handler.advanced_heatmap_processor import AdvancedHeatmapProcessor
from pipeline.handler.draw_heatmap_box_processor import DrawHeatMapBoxProcessor
from pipeline.handler.draw_id_box_processor import DrawIdBoxProcessor
from pipeline.handler.draw_motion_vector_processor import DrawMotionVectorProcessor


def test_modifications():
    print("测试修改后的处理器...")
    
    # 测试所有处理器是否都能正确创建
    proc1 = AdvancedHeatmapProcessor()
    proc2 = DrawHeatMapBoxProcessor()
    proc3 = DrawIdBoxProcessor()
    proc4 = DrawMotionVectorProcessor()
    
    print('所有处理器创建成功')
    print(f'高级热力图处理器: {proc1.name}')
    print(f'热力图框绘制处理器: {proc2.name}')
    print(f'ID框绘制处理器: {proc3.name}')
    print(f'运动向量绘制处理器: {proc4.name}')
    
    print("修改验证完成!")


if __name__ == "__main__":
    test_modifications()