def create_video_queue_vs(project: Project, split_locations: List[int]) -> List[Chunk]: """ Create a list of chunks using vspipe and ffms2 for frame accurate seeking :param project: the Project :param split_locations: a list of frames to split on :param script: source filter script to use with vspipe (ignored with vs input) :return: A list of chunks """ # add first frame and last frame last_frame = project.get_frames() split_locs_fl = [0] + split_locations + [last_frame] # pair up adjacent members of this list ex: [0, 10, 20, 30] -> [(0, 10), (10, 20), (20, 30)] chunk_boundaries = zip(split_locs_fl, split_locs_fl[1:]) source_file = project.input.absolute().as_posix() if project.is_vs: vs_script = project.input else: vs_script = create_vs_file(project.temp, source_file, project.chunk_method) chunk_queue = [ create_vs_chunk(project, index, vs_script, *cb) for index, cb in enumerate(chunk_boundaries) ] return chunk_queue
def create_video_queue_vs(project: Project, split_locations: List[int], script: str) -> List[Chunk]: """ Create a list of chunks using vspipe and ffms2 for frame accurate seeking :param project: the Project :param split_locations: a list of frames to split on :param script: source filter script to use with vspipe (ignored with vs input) :return: A list of chunks """ # add first frame and last frame last_frame = project.get_frames() split_locs_fl = [0] + split_locations + [last_frame] # pair up adjacent members of this list ex: [0, 10, 20, 30] -> [(0, 10), (10, 20), (20, 30)] chunk_boundaries = zip(split_locs_fl, split_locs_fl[1:]) source_file = project.input.absolute().as_posix() vs_script = project.input if not project.is_vs: # create a vapoursynth script that will load the source with ffms2 load_script = project.temp / 'split' / 'loadscript.vpy' cache_file = (project.temp / 'split' / 'ffms2cache.ffindex').absolute().as_posix() with open(load_script, 'w+') as file: file.write(script.format(source_file, cache_file)) vs_script = load_script chunk_queue = [ create_vs_chunk(project, index, vs_script, *cb) for index, cb in enumerate(chunk_boundaries) ] return chunk_queue
def create_chunk_from_segment(project: Project, index: int, file: Path) -> Chunk: """ Creates a Chunk object from a segment file generated by ffmpeg :param project: the Project :param index: the index of the chunk :param file: the segmented file :return: A Chunk """ ffmpeg_gen_cmd = [ "ffmpeg", "-y", "-hide_banner", "-loglevel", "error", "-i", file.as_posix(), *project.pix_format, "-f", "yuv4mpegpipe", "-", ] file_size = file.stat().st_size frames = project.get_frames() extension = ENCODERS[project.encoder].output_extension chunk = Chunk(project.temp, index, ffmpeg_gen_cmd, extension, file_size, frames) return chunk
def create_video_queue_select(project: Project, split_locations: List[int]) -> List[Chunk]: """ Create a list of chunks using the select filter :param project: the Project :param split_locations: a list of frames to split on :return: A list of chunks """ # add first frame and last frame last_frame = project.get_frames() split_locs_fl = [0] + split_locations + [last_frame] # pair up adjacent members of this list ex: [0, 10, 20, 30] -> [(0, 10), (10, 20), (20, 30)] chunk_boundaries = zip(split_locs_fl, split_locs_fl[1:]) chunk_queue = [ create_select_chunk(project, index, project.input, *cb) for index, cb in enumerate(chunk_boundaries) ] return chunk_queue
def create_video_queue_hybrid(project: Project, split_locations: List[int]) -> List[Chunk]: """ Create list of chunks using hybrid segment-select approach :param project: the Project :param split_locations: a list of frames to split on :return: A list of chunks """ keyframes = get_keyframes(str(project.input.resolve())) end = [project.get_frames()] splits = [0] + split_locations + end segments_list = list(zip(splits, splits[1:])) to_split = [x for x in keyframes if x in splits] segments = [] # Make segments log("Segmenting Video") segment(str(project.input.resolve()), str(project.temp.resolve()), to_split[1:]) log("Segment Done") source_path = project.temp / "split" queue_files = [x for x in source_path.iterdir() if x.suffix == ".mkv"] queue_files.sort(key=lambda p: p.stem) kf_list = list(zip(to_split, to_split[1:] + end)) for f, (x, y) in zip(queue_files, kf_list): to_add = [(f, [s[0] - x, s[1] - x]) for s in segments_list if s[0] >= x and s[1] <= y and s[0] - x < s[1] - x] segments.extend(to_add) chunk_queue = [ create_select_chunk(project, index, file, *cb) for index, (file, cb) in enumerate(segments) ] return chunk_queue
def create_chunk_from_segment(project: Project, index: int, file: Path) -> Chunk: """ Creates a Chunk object from a segment file generated by ffmpeg :param project: the Project :param index: the index of the chunk :param file: the segmented file :return: A Chunk """ ffmpeg_gen_cmd = [ 'ffmpeg', '-y', '-hide_banner', '-loglevel', 'error', '-i', file.as_posix(), *project.pix_format, '-color_range', '0', '-f', 'yuv4mpegpipe', '-' ] file_size = file.stat().st_size frames = project.get_frames() extension = ENCODERS[project.encoder].output_extension chunk = Chunk(project.temp, index, ffmpeg_gen_cmd, extension, file_size, frames) return chunk