hwtransfer: implement support for hw->hw format conversion

Historically, we have not attempted to support hw->hw format conversion
in the autoconvert logic. If a user needed to do these kinds of
conversions, they needed to manually insert the hw format's conversion
filter manually (eg: scale_vaapi).

This was usually fine because the general rule is that any format
supported by the hardware can be used as well as any other. ie: You
would only need to do conversion if you have a specific goal in mind.

However, we now have two situations where we can find ourselves with a
hardware format being produced by a decoder that cannot be accepted by
a VO via hwdec-interop:
 * dmabuf-wayland can only accept formats that the Wayland compositor
   accepts. In the case of GNOME, it can only accept a handful of RGB
   formats.
 * When decoding via VAAPI on Intel hardware, some of the more unusual
   video encodings (4:2:2, 10bit 4:4:4) lead to packed frame formats
   which gpu-next cannot handle, causing rendering to fail.

In both these cases (at least when using VAAPI with dmabuf-wayland), if
we could detect the failure case and insert a `scale_vaapi` filter, we
could get successful output automatically. For `hwdec=drm`, there is
currently no conversion filter, so dmabuf-wayland is still out of luck
there.

The basic approach to implementing this is to detect the case where we
are evaluating a hardware format where the VO can accept the hardware
format itself, but may not accept the underlying sw format. In the
current code, we bypass autoconvert as soon as we see the hardware
format is compatible.

My first observation was that we actually have logic in autoconvert to
detect when the input sw format is not in the list of allowed sw
formats passed into the autoconverter. Unfortunately, we never populate
this list, and the way you would expect to do that is vo-query-format
returning sw format information, which it does not. We could define an
extended vo-query-format-2, but we'd still need to implement the
probing logic to fill it in.

On the other hand, we already have the probing logic in the hwupload
filter - and most recently I used that logic to implement conversion
on upload. So if we could leverage that, we could both detect when
hw->hw conversion is required, and pick the best target format.

This exercise is then primarily one of detecting when we are in this
case and letting that code run in a useful way. The hwupload filter is
a bit awkward to work with today, and so I refactored a bunch of the
set up code to actually make it more encapsulated. Now, instead of the
caller instantiating it and then triggering the probe, we probe on
creation and instantiate the correct underlying filter (hwupload vs
conversion) automatically.

1 files changed, 75 insertions, 35 deletions

diff --git a/filters/f_hwtransfer.c b/filters/f_hwtransfer.c
index b281a15838..a8f01e4054 100644
--- a/filters/f_hwtransfer.c
+++ b/filters/f_hwtransfer.c
@@ -1,6 +1,7 @@
 #include <libavutil/buffer.h>
 #include <libavutil/hwcontext.h>
 #include <libavutil/mem.h>
+#include <libavutil/pixdesc.h>
 
 #include "video/fmt-conversion.h"
 #include "video/hwdec.h"
@@ -8,7 +9,10 @@
 #include "video/mp_image_pool.h"
 
 #include "f_hwtransfer.h"
+#include "f_output_chain.h"
+#include "f_utils.h"
 #include "filter_internal.h"
+#include "user_filters.h"
 
 struct priv {
     AVBufferRef *av_device_ctx;
@@ -39,7 +43,9 @@ struct priv {
     int *map_fmts;
     int num_map_fmts;
 
-    struct mp_hwupload public;
+    // If the selected hwdec has a conversion filter available for converting
+    // between sw formats in hardware, the name will be set. NULL otherwise.
+    const char *conversion_filter_name;
 };
 
 struct hwmap_pairs {
@@ -66,6 +72,11 @@ static const struct hwmap_pairs hwmap_pairs[] = {
 /**
  * @brief Find the closest supported format when hw uploading
  *
+ * Return the best format suited for upload that is supported for a given input
+ * imgfmt. This returns the same as imgfmt if the format is natively supported,
+ * and otherwise a format that likely results in the least loss.
+ * Returns 0 if completely unsupported.
+ *
  * Some hardware types support implicit format conversion on upload. For these
  * types, it is possible for the set of formats that are accepts as inputs to
  * the upload process to differ from the set of formats that can be outputs of
@@ -122,23 +133,6 @@ static bool select_format(struct priv *p, int input_fmt,
     return true;
 }
 
-int mp_hwupload_find_upload_format(struct mp_hwupload *u, int imgfmt)
-{
-    struct priv *p = u->f->priv;
-
-    int sw = 0, up = 0;
-    select_format(p, imgfmt, &sw, &up);
-    // In th case where the imgfmt is not natively supported, it must be
-    // converted, either before or during upload. If the imgfmt is supported as
-    // an hw input format, then prefer that, and if the upload has to do implicit
-    // conversion, that's fine. On the other hand, if the imgfmt is not a
-    // supported input format, then pick the output format as the conversion
-    // target to avoid doing two conversions (one before upload, and one during
-    // upload). Note that for most hardware types, there is no ability to convert
-    // during upload, and the two formats will always be the same.
-    return imgfmt == sw ? sw : up;
-}
-
 static void process(struct mp_filter *f)
 {
     struct priv *p = f->priv;
@@ -266,17 +260,17 @@ static bool vo_supports(struct mp_hwdec_ctx *ctx, int hw_fmt, int sw_fmt)
     return false;
 }
 
-static bool probe_formats(struct mp_hwupload *u, int hw_imgfmt)
+static bool probe_formats(struct mp_filter *f, int hw_imgfmt)
 {
-    struct priv *p = u->f->priv;
+    struct priv *p = f->priv;
 
     p->hw_imgfmt = hw_imgfmt;
     p->num_fmts = 0;
     p->num_upload_fmts = 0;
 
-    struct mp_stream_info *info = mp_filter_find_stream_info(u->f);
+    struct mp_stream_info *info = mp_filter_find_stream_info(f);
     if (!info || !info->hwdec_devs) {
-        MP_ERR(u->f, "no hw context\n");
+        MP_ERR(f, "no hw context\n");
         return false;
     }
 
@@ -312,7 +306,7 @@ static bool probe_formats(struct mp_hwupload *u, int hw_imgfmt)
     }
 
     if (!ctx) {
-        MP_INFO(u->f, "no support for this hw format\n");
+        MP_INFO(f, "no support for this hw format\n");
         return false;
     }
 
@@ -339,14 +333,14 @@ static bool probe_formats(struct mp_hwupload *u, int hw_imgfmt)
         if (!imgfmt)
             continue;
 
-        MP_VERBOSE(u->f, "looking at format %s/%s\n",
+        MP_VERBOSE(f, "looking at format %s/%s\n",
                    mp_imgfmt_to_name(hw_imgfmt),
                    mp_imgfmt_to_name(imgfmt));
 
         if (IMGFMT_IS_HWACCEL(imgfmt)) {
             // If the enumerated format is a hardware format, we don't need to
             // do any further probing. It will be supported.
-            MP_VERBOSE(u->f, "  supports %s (a hardware format)\n",
+            MP_VERBOSE(f, "  supports %s (a hardware format)\n",
                        mp_imgfmt_to_name(imgfmt));
             continue;
         }
@@ -356,7 +350,7 @@ static bool probe_formats(struct mp_hwupload *u, int hw_imgfmt)
         if (!mp_update_av_hw_frames_pool(&frames, ctx->av_device_ref,
                                          hw_imgfmt, imgfmt, 128, 128, false))
         {
-            MP_WARN(u->f, "failed to allocate pool\n");
+            MP_WARN(f, "failed to allocate pool\n");
             continue;
         }
 
@@ -375,9 +369,9 @@ static bool probe_formats(struct mp_hwupload *u, int hw_imgfmt)
                 int fmt = pixfmt2imgfmt(fmts[i]);
                 if (!fmt)
                     continue;
-                MP_VERBOSE(u->f, "  supports %s\n", mp_imgfmt_to_name(fmt));
+                MP_VERBOSE(f, "  supports %s\n", mp_imgfmt_to_name(fmt));
                 if (!vo_supports(ctx, hw_imgfmt, fmt)) {
-                    MP_VERBOSE(u->f, "  ... not supported by VO\n");
+                    MP_VERBOSE(f, "  ... not supported by VO\n");
                     continue;
                 }
                 MP_TARRAY_APPEND(p, p->upload_fmts, p->num_upload_fmts, fmt);
@@ -396,32 +390,78 @@ static bool probe_formats(struct mp_hwupload *u, int hw_imgfmt)
     p->av_device_ctx = av_buffer_ref(ctx->av_device_ref);
     if (!p->av_device_ctx)
         return false;
+    p->conversion_filter_name = ctx->conversion_filter_name;
 
     return p->num_upload_fmts > 0;
 }
 
-struct mp_hwupload *mp_hwupload_create(struct mp_filter *parent, int hw_imgfmt)
+struct mp_hwupload mp_hwupload_create(struct mp_filter *parent, int hw_imgfmt,
+                                       int sw_imgfmt, bool src_is_same_hw)
 {
+    struct mp_hwupload u = {0,};
     struct mp_filter *f = mp_filter_create(parent, &hwupload_filter);
     if (!f)
-        return NULL;
+        return u;
 
     struct priv *p = f->priv;
-    struct mp_hwupload *u = &p->public;
-    u->f = f;
-
     mp_filter_add_pin(f, MP_PIN_IN, "in");
     mp_filter_add_pin(f, MP_PIN_OUT, "out");
 
-    if (!probe_formats(u, hw_imgfmt)) {
+    if (!probe_formats(f, hw_imgfmt)) {
         MP_INFO(f, "hardware format not supported\n");
         goto fail;
     }
 
+    int hw_input_fmt = 0, hw_output_fmt = 0;
+    if (!select_format(p, sw_imgfmt, &hw_input_fmt, &hw_output_fmt)) {
+        MP_ERR(f, "Unable to find a compatible upload format for %s\n",
+               mp_imgfmt_to_name(sw_imgfmt));
+        goto fail;
+    }
+
+    if (src_is_same_hw) {
+        if (p->conversion_filter_name) {
+            /*
+            * If we are converting from one sw format to another within the same
+            * hw format, we will use that hw format's conversion filter rather
+            * than the actual hwupload filter.
+            */
+            u.selected_sw_imgfmt = hw_output_fmt;
+            if (sw_imgfmt != u.selected_sw_imgfmt) {
+                enum AVPixelFormat pixfmt = imgfmt2pixfmt(u.selected_sw_imgfmt);
+                const char *avfmt_name = av_get_pix_fmt_name(pixfmt);
+                char *args[] = {"format", (char *)avfmt_name, NULL};
+                MP_VERBOSE(f, "Hardware conversion: %s -> %s\n",
+                           p->conversion_filter_name, avfmt_name);
+                struct mp_filter *sv =
+                    mp_create_user_filter(parent, MP_OUTPUT_CHAIN_VIDEO,
+                                        p->conversion_filter_name, args);
+                u.f = sv;
+                talloc_free(f);
+            }
+        }
+    } else {
+        u.f = f;
+        /*
+         * In the case where the imgfmt is not natively supported, it must be
+         * converted, either before or during upload. If the imgfmt is supported
+         * as a hw input format, then prefer that, and if the upload has to do
+         * implicit conversion, that's fine. On the other hand, if the imgfmt is
+         * not a supported input format, then pick the output format as the
+         * conversion target to avoid doing two conversions (one before upload,
+         * and one during upload). Note that for most hardware types, there is
+         * no ability to convert during upload, and the two formats will always
+         * be the same.
+         */
+        u.selected_sw_imgfmt =
+            sw_imgfmt == hw_input_fmt ? hw_input_fmt : hw_output_fmt;
+    }
+
+    u.successful_init = true;
     return u;
 fail:
     talloc_free(f);
-    return NULL;
+    return u;
 }
 
 static void hwdownload_process(struct mp_filter *f)