from biz.checkpoint.checkpoint_biz import FrameProcessorWorker as CheckpointWorker
from biz.prison.trajectory02_biz import FrameProcessorWorker as TrajectoryWorker
from biz.prison.supervision_room_biz import FrameProcessorWorker as SupervisionWorker
from biz.prison.ab_biz import FrameProcessorWorker as AbWorker
from biz.prison.prison_biz import FrameProcessorWorker as CorridorWorker
from biz.prison.indoor_biz import FrameProcessorWorker as IndoorWorker

# ... 其他导入

PROCESSOR_MAP = {
    "checkpoint": CheckpointWorker,
    "trajectory": TrajectoryWorker,
    "supervision_room": SupervisionWorker,
    "ab": AbWorker,
    "corridor": CorridorWorker,
    "indoor": IndoorWorker
}

def get_processor(processor_type: str):
    if processor_type not in PROCESSOR_MAP:
        raise ValueError(f"Unknown processor type: {processor_type}")
    return PROCESSOR_MAP[processor_type]