From 5f99ff97b99de176e812feaba8f544750764c228 Mon Sep 17 00:00:00 2001
From: cudawarped <mail@jamesbowley.co.uk>
Date: Mon, 7 Dec 2020 10:01:02 +0000
Subject: [PATCH] Update cudacodec to work with Nvidia Video Codec SDK 11.0

---
 modules/cudacodec/include/opencv2/cudacodec.hpp | 1 +
 modules/cudacodec/src/video_decoder.cpp         | 1 +
 2 files changed, 2 insertions(+)

diff --git a/modules/cudacodec/include/opencv2/cudacodec.hpp b/modules/cudacodec/include/opencv2/cudacodec.hpp
index 3fad950e102..401054eefcd 100644
--- a/modules/cudacodec/include/opencv2/cudacodec.hpp
+++ b/modules/cudacodec/include/opencv2/cudacodec.hpp
@@ -255,6 +255,7 @@ enum Codec
     HEVC,
     VP8,
     VP9,
+    AV1,
     NumCodecs,
 
     Uncompressed_YUV420 = (('I'<<24)|('Y'<<16)|('U'<<8)|('V')),   //!< Y,U,V (4:2:0)
diff --git a/modules/cudacodec/src/video_decoder.cpp b/modules/cudacodec/src/video_decoder.cpp
index a84f79c9b6f..1293055bff4 100644
--- a/modules/cudacodec/src/video_decoder.cpp
+++ b/modules/cudacodec/src/video_decoder.cpp
@@ -78,6 +78,7 @@ void cv::cudacodec::detail::VideoDecoder::create(const FormatInfo& videoFormat)
 #if  ((CUDART_VERSION == 7500) || (CUDART_VERSION >= 9000))
     codecSupported |=       cudaVideoCodec_VP8      == _codec ||
                             cudaVideoCodec_VP9      == _codec ||
+                            cudaVideoCodec_AV1      == _codec ||
                             cudaVideoCodec_YUV420   == _codec;
 #endif
 #endif