[FFmpeg-cvslog] hwcontext_cuda: implement frames_get_constraints

wm4 git at videolan.org
Mon Jan 23 17:22:02 EET 2017


ffmpeg | branch: master | wm4 <nfxjfg at googlemail.com> | Mon Jan 16 16:43:13 2017 +0100| [c16fe1432d88f87a96be9e943e0f1229543ad61d] | committer: wm4

hwcontext_cuda: implement frames_get_constraints

Copied and modified from hwcontext_qsv.c.

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=c16fe1432d88f87a96be9e943e0f1229543ad61d
---

 libavutil/hwcontext_cuda.c | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/libavutil/hwcontext_cuda.c b/libavutil/hwcontext_cuda.c
index 5dd0d99..ed595c3 100644
--- a/libavutil/hwcontext_cuda.c
+++ b/libavutil/hwcontext_cuda.c
@@ -39,6 +39,31 @@ static const enum AVPixelFormat supported_formats[] = {
     AV_PIX_FMT_P016,
 };
 
+static int cuda_frames_get_constraints(AVHWDeviceContext *ctx,
+                                       const void *hwconfig,
+                                       AVHWFramesConstraints *constraints)
+{
+    int i;
+
+    constraints->valid_sw_formats = av_malloc_array(FF_ARRAY_ELEMS(supported_formats) + 1,
+                                                    sizeof(*constraints->valid_sw_formats));
+    if (!constraints->valid_sw_formats)
+        return AVERROR(ENOMEM);
+
+    for (i = 0; i < FF_ARRAY_ELEMS(supported_formats); i++)
+        constraints->valid_sw_formats[i] = supported_formats[i];
+    constraints->valid_sw_formats[FF_ARRAY_ELEMS(supported_formats)] = AV_PIX_FMT_NONE;
+
+    constraints->valid_hw_formats = av_malloc_array(2, sizeof(*constraints->valid_hw_formats));
+    if (!constraints->valid_hw_formats)
+        return AVERROR(ENOMEM);
+
+    constraints->valid_hw_formats[0] = AV_PIX_FMT_CUDA;
+    constraints->valid_hw_formats[1] = AV_PIX_FMT_NONE;
+
+    return 0;
+}
+
 static void cuda_buffer_free(void *opaque, uint8_t *data)
 {
     AVHWFramesContext *ctx = opaque;
@@ -376,6 +401,7 @@ const HWContextType ff_hwcontext_type_cuda = {
     .device_create        = cuda_device_create,
     .device_init          = cuda_device_init,
     .device_uninit        = cuda_device_uninit,
+    .frames_get_constraints = cuda_frames_get_constraints,
     .frames_init          = cuda_frames_init,
     .frames_get_buffer    = cuda_get_buffer,
     .transfer_get_formats = cuda_transfer_get_formats,



More information about the ffmpeg-cvslog mailing list