from sophon_chakcy.utils.task_manager import TaskManager, TaskArgs
import time
from sophon_chakcy.sail import MultiEngine

def main():
    task_manager = TaskManager(16)
    
    for i in range(16):
        task_name = f"yolo_test_{i}"
        output_path = f"rtsp://192.168.137.1:8554/stream{i}"
        input_path = f"./resources/datasets/test_car_person_1080P_{i}.mp4"
        args = TaskArgs(
            task_name=task_name,
            algorithms=["yolo_example", "yolo_example", "yolo_example"],
            input=input_path,
            output=output_path,
            dev_id=0
        )
        task_manager.submit(args)
    start_time = time.time()
    task_manager.wait_for_all()
    print(time.time() - start_time)
    task_manager.shutdown()

if __name__ == "__main__":
    main()