def main()

in tools/make_video.py [0:0]

• 37 lines of code
• 5 McCabe index (conditional complexity)

def main(args):
    COLOR_NAME = "color"

    args = augment_args(args)

    size = frame_size(pjoin(args.color_dir, args.frame_fmt))

    os.makedirs(args.out_dir, exist_ok=True)

    color_video_prefix = pjoin(args.out_dir, COLOR_NAME)
    make_video(
        args.ffmpeg,
        pjoin(args.color_dir, args.frame_fmt),
        color_video_prefix,
        ext=args.ext,
    )

    depth_video_prefixes = [pjoin(args.out_dir, name) for name in args.depth_names]
    for depth_dir, prefix in zip(args.depth_dirs, depth_video_prefixes):
        make_depth_videos(
            args.ffmpeg,
            pjoin(depth_dir, args.frame_fmt),
            pjoin(args.color_dir, args.frame_fmt),
            prefix,
            size=size,
            ext=args.ext,
        )
    if len(args.depth_dirs) > 0:
        stack_videos(
            args.ffmpeg,
            [color_video_prefix] + depth_video_prefixes,
            args.out_dir,
            size=size,
            ext=args.ext,
        )

        # merge overlay videos
        overlay_video_prefixes = []
        for pre in depth_video_prefixes:
            overlay_video_prefixes.extend([pre, pre + "-overlay"])
        stack_videos(
            args.ffmpeg, overlay_video_prefixes, args.out_dir, size=size, ext=args.ext
        )

    return 0