Source code for cv2ext.cli.resize_video._resize_video

# Copyright (c) 2024 Justin Davis (davisjustin302@gmail.com)
#
# MIT License
from __future__ import annotations

import argparse

import cv2
from tqdm import tqdm

from cv2ext import Fourcc, IterableVideo, VideoWriter




def resize_video_cli() -> None:
    parser = argparse.ArgumentParser(description="Resize a video file.")
    parser.add_argument(
        "--video",
        type=str,
        required=True,
        help="The video file to resize.",
    )
    parser.add_argument("--output", type=str, required=True, help="The output file.")
    parser.add_argument(
        "--size",
        type=str,
        required=True,
        help="The size to resize the video to. Example: [640, 480]",
    )
    parser.add_argument(
        "--fast",
        type=bool,
        default=False,
        required=False,
        help="Use a faster resize method.",
    )
    args = parser.parse_args()

    # parse size
    size = tuple(map(int, args.size.strip("[]").split(",")))
    tuple_size = 2
    if len(size) != tuple_size:
        err_msg = f"Invalid size: {size}"
        raise ValueError(err_msg)
    frame_size: tuple[int, int] = size  # type: ignore[assignment]

    # open video
    video = IterableVideo(args.video, use_thread=True)
    extension = args.output.split(".")[-1]
    fourcc = None
    if extension == "mp4":
        fourcc = Fourcc.MP4V
    elif extension == "avi":
        fourcc = Fourcc.XVID
    if fourcc is None:
        err_msg = f"Unsupported file extension: {extension}"
        raise ValueError(err_msg)

    with VideoWriter(
        args.output,
        fourcc=fourcc,
        fps=video.fps,
        frame_size=frame_size,
    ) as writer:
        for _, frame in tqdm(video):
            if args.fast:
                resized_frame = cv2.resize(frame, size, interpolation=cv2.INTER_LINEAR)
            else:
                resized_frame = cv2.resize(frame, size, interpolation=cv2.INTER_CUBIC)
            writer.write(resized_frame)