[FFmpeg-devel] [PATCH v3 2/2] dxva: add a ID3D11DeviceContext to the old D3D11VA API

[Steve Lhomme]

This way it's possible to use the wait_copy feature with the old API as well.

If it's NULL, wait_copies remains NULL and is not used.
---
 doc/APIchanges       |  3 +++
 libavcodec/d3d11va.h |  6 ++++++
 libavcodec/dxva2.c   | 38 +++++++++++++++++++++++++++++++-------
 libavcodec/version.h |  2 +-
 4 files changed, 41 insertions(+), 8 deletions(-)

diff --git a/doc/APIchanges b/doc/APIchanges
index a9bf1afd4c..11a10ddb16 100644
--- a/doc/APIchanges
+++ b/doc/APIchanges
@@ -15,6 +15,9 @@ libavutil:     2017-10-21
 
 API changes, most recent first:
 
+2020-08-05 - xxxxxxxxxx - lavu 56.100.100 - d3d11va.h
+  Add a ID3D11VideoContext to AVD3D11VAContext matching the ID3D11VideoContext.
+
 2020-08-04 - xxxxxxxxxx - lavu 56.58.100 - channel_layout.h
   Add AV_CH_LAYOUT_22POINT2 together with its newly required pieces:
   AV_CH_TOP_SIDE_LEFT, AV_CH_TOP_SIDE_RIGHT, AV_CH_BOTTOM_FRONT_CENTER,
diff --git a/libavcodec/d3d11va.h b/libavcodec/d3d11va.h
index 6816b6c1e6..134dee388c 100644
--- a/libavcodec/d3d11va.h
+++ b/libavcodec/d3d11va.h
@@ -96,6 +96,12 @@ typedef struct AVD3D11VAContext {
       * Mutex to access video_context
       */
     HANDLE  context_mutex;
+
+    /**
+      * Device Context on which to send the D3D11VA commands.
+      * It should be the same used to get the ID3D11VideoDecoder or NULL.
+      */
+    ID3D11DeviceContext *device_context;
 } AVD3D11VAContext;
 
 /**
diff --git a/libavcodec/dxva2.c b/libavcodec/dxva2.c
index 1a0e5b69b2..fd4685305e 100644
--- a/libavcodec/dxva2.c
+++ b/libavcodec/dxva2.c
@@ -656,7 +656,27 @@ int ff_dxva2_decode_init(AVCodecContext *avctx)
 
     // Old API.
     if (avctx->hwaccel_context)
+    {
+#if CONFIG_D3D11VA
+        if (avctx->hwaccel->pix_fmt == AV_PIX_FMT_D3D11VA_VLD) {
+            AVDXVAContext *ctx = DXVA_CONTEXT(avctx);
+
+            if (D3D11VA_CONTEXT(ctx)->device_context) {
+                ID3D11Device *d3ddevice;
+                ID3D11DeviceContext_GetDevice(D3D11VA_CONTEXT(ctx)->device_context, &d3ddevice);
+
+                D3D11_QUERY_DESC query = { 0 };
+                query.Query = D3D11_QUERY_EVENT;
+                if (FAILED(ID3D11Device_CreateQuery(d3ddevice, &query,
+                                                    (ID3D11Query**)&sctx->wait_copies)))
+                    sctx->wait_copies = NULL;
+
+                ID3D11Device_Release(d3ddevice);
+            }
+        }
+#endif
         return 0;
+    }
 
     // (avctx->pix_fmt is not updated yet at this point)
     sctx->pix_fmt = avctx->hwaccel->pix_fmt;
@@ -896,6 +916,7 @@ int ff_dxva2_common_end_frame(AVCodecContext *avctx, AVFrame *frame,
     unsigned               buffer_count = 0;
 #if CONFIG_D3D11VA
     D3D11_VIDEO_DECODER_BUFFER_DESC buffer11[4];
+    ID3D11DeviceContext *device_context;
 #endif
 #if CONFIG_DXVA2
     DXVA2_DecodeBufferDesc          buffer2[4];
@@ -941,9 +962,14 @@ int ff_dxva2_common_end_frame(AVCodecContext *avctx, AVFrame *frame,
 #if CONFIG_D3D11VA
     if (ff_dxva2_is_d3d11(avctx)) {
         if (sctx->wait_copies) {
-            AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
-            AVD3D11VADeviceContext *device_hwctx = frames_ctx->device_ctx->hwctx;
-            ID3D11DeviceContext_Begin(device_hwctx->device_context, sctx->wait_copies);
+            if (avctx->hwaccel_context)
+                device_context = D3D11VA_CONTEXT(ctx)->device_context;
+            else {
+                AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
+                AVD3D11VADeviceContext *device_hwctx = frames_ctx->device_ctx->hwctx;
+                device_context = device_hwctx->device_context;
+            }
+            ID3D11DeviceContext_Begin(device_context, sctx->wait_copies);
         }
 
         buffer = &buffer11[buffer_count];
@@ -1024,12 +1050,10 @@ int ff_dxva2_common_end_frame(AVCodecContext *avctx, AVFrame *frame,
         /* wait until all the buffer release is done copying data to the GPU
          * before doing the submit command */
         if (sctx->wait_copies) {
-            AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
-            AVD3D11VADeviceContext *device_hwctx = frames_ctx->device_ctx->hwctx;
-            ID3D11DeviceContext_End(device_hwctx->device_context, sctx->wait_copies);
+            ID3D11DeviceContext_End(device_context, sctx->wait_copies);
 
             while (maxWait-- && S_FALSE ==
-                   ID3D11DeviceContext_GetData(device_hwctx->device_context,
+                   ID3D11DeviceContext_GetData(device_context,
                                                sctx->wait_copies, NULL, 0, 0)) {
                 ff_dxva2_unlock(avctx);
                 SleepEx(2, TRUE);
diff --git a/libavcodec/version.h b/libavcodec/version.h
index f66919617a..a3f9f828ee 100644
--- a/libavcodec/version.h
+++ b/libavcodec/version.h
@@ -28,7 +28,7 @@
 #include "libavutil/version.h"
 
 #define LIBAVCODEC_VERSION_MAJOR  58
-#define LIBAVCODEC_VERSION_MINOR  99
+#define LIBAVCODEC_VERSION_MINOR 100
 #define LIBAVCODEC_VERSION_MICRO 100
 
 #define LIBAVCODEC_VERSION_INT  AV_VERSION_INT(LIBAVCODEC_VERSION_MAJOR, \
-- 
2.26.2