test_parallel.py

"""Test to process input streams into an output stream in parallel"""

from stream_input import InputStream, InputQueue
from stream_output import OutputStream


def finished_in(filename: str):
    """dummy callback for the input stream finished event"""
    print(f"\n\n\nfinished input {filename}\n\n\n")


# call backs should also provide a frame number?
def finished_out(filename: str):
    """dummy callback for the output stream finished event"""
    print(f"\n\n\nfinished output {filename}\n\n\n")


input_queue = InputQueue("raw-video/dock*")
input1 = InputStream(input_queue, finished_in)
input2 = InputStream(input_queue, finished_in)
output_stream = OutputStream(
    {
        "output_template": "video/test.mp4",
        "output_seconds": 60,
    },
    input1.attributes,  # We need to make sure both streams match!
    finished_out,
)
while True:
    raw_bytes = input1.read()
    if len(raw_bytes) == 0:
        break
    output_stream.write(raw_bytes)

    raw_bytes = input2.read()
    if len(raw_bytes) == 0:
        break
    output_stream.write(raw_bytes)

output_stream.close()
input2.close()
input1.close()