From fa97161484a098a06d433391006d012611e8ff6a Mon Sep 17 00:00:00 2001 From: Gustavo Buzogany Eboli Date: Wed, 14 Jun 2023 13:31:47 +0200 Subject: [PATCH] Progress adding vp8 support --- src/PyNvCodec/__init__.pyi | 2 ++ src/PyNvCodec/src/PyNvCodec.cpp | 1 + 2 files changed, 3 insertions(+) diff --git a/src/PyNvCodec/__init__.pyi b/src/PyNvCodec/__init__.pyi index eba52e25..4dfbd54e 100644 --- a/src/PyNvCodec/__init__.pyi +++ b/src/PyNvCodec/__init__.pyi @@ -83,6 +83,7 @@ SUPPORT_YUV444_ENCODE: NV_ENC_CAPS UDEF: ColorRange UNDEFINED: PixelFormat UNSPEC: ColorSpace +VP8: CudaVideoCodec VP9: CudaVideoCodec WIDTH_MAX: NV_ENC_CAPS WIDTH_MIN: NV_ENC_CAPS @@ -158,6 +159,7 @@ class CudaVideoCodec: __members__: ClassVar[dict] = ... # read-only H264: ClassVar[CudaVideoCodec] = ... HEVC: ClassVar[CudaVideoCodec] = ... + VP8: ClassVar[CudaVideoCodec] = ... VP9: ClassVar[CudaVideoCodec] = ... __entries: ClassVar[dict] = ... def __init__(self, value: int) -> None: ... diff --git a/src/PyNvCodec/src/PyNvCodec.cpp b/src/PyNvCodec/src/PyNvCodec.cpp index f89b5857..a2ac0fc0 100644 --- a/src/PyNvCodec/src/PyNvCodec.cpp +++ b/src/PyNvCodec/src/PyNvCodec.cpp @@ -253,6 +253,7 @@ PYBIND11_MODULE(_PyNvCodec, m) .value("H264", cudaVideoCodec::cudaVideoCodec_H264) .value("HEVC", cudaVideoCodec::cudaVideoCodec_HEVC) .value("VP9", cudaVideoCodec::cudaVideoCodec_VP9) + .value("VP8", cudaVideoCodec::cudaVideoCodec_VP8) .export_values(); py::enum_(m, "SeekMode")