File size: 1,749 Bytes
aae3ba1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import os
import ffmpeg
# ------------------------------------------------------
# Concatenate TS batch files into a final MP4
# ------------------------------------------------------
def concatenate_ts_files(folder, video_name, batch_counts):
"""Concatenate batch TS files into final mp4."""
inputs_path = os.path.join(folder, 'inputs.txt')
# Create a file list for ffmpeg
with open(inputs_path, 'w') as f:
for i in range(batch_counts):
f.write(f"file '{video_name}_b{i:04d}.ts'\n")
# Merge using ffmpeg concat demuxer
ffmpeg.input(inputs_path, format='concat', safe=0).output(
os.path.join(folder, f'{video_name}.mp4'),
c='copy'
).run()
# Cleanup temporary TS files and list file
for i in range(batch_counts):
os.remove(os.path.join(folder, f"{video_name}_b{i:04d}.ts"))
os.remove(inputs_path)
# ------------------------------------------------------
# Create a new ffmpeg writer process
# ------------------------------------------------------
def create_ffmpeg_writer(output_path, width, height, fps, crf):
"""Spawn an ffmpeg async encoding process for writing raw frames."""
return (
ffmpeg.output(
ffmpeg.input(
'pipe:0',
format='rawvideo',
pix_fmt='rgb24',
s=f'{width}x{height}',
r=fps,
),
output_path,
**{
'preset': 'medium',
'pix_fmt': 'yuv420p',
'b:v': '0',
'c:v': 'libx264',
'crf': str(crf),
'r': fps,
}
)
.overwrite_output()
.run_async(quiet=True, pipe_stdin=True)
) |