From 2c43d2b75a88b8e0e8f0a715f993ffc1c8977d13 Mon Sep 17 00:00:00 2001
From: wm4 <wm4@nowhere>
Date: Thu, 31 Oct 2019 15:44:09 +0100
Subject: screenshot, vo_image: use global swscale/zimg parameters

Lots of dumb crap to do... something. Instead of adding yet another dumb
helper, just use the main" sws_utils API in both callers. (Which,
unfortunately, has been duplicated for glorious webp screenshots,
despite the fact that webp is crap.)

Good part: can enable zimg for screenshots (as far as needed).
Bad part: uses "default" swscale parameters instead of HQ now.
---
 video/image_loader.c |  2 +-
 video/image_writer.c | 17 +++++++++++++----
 video/image_writer.h |  5 ++++-
 video/out/vo_image.c |  2 +-
 4 files changed, 19 insertions(+), 7 deletions(-)

(limited to 'video')

diff --git a/video/image_loader.c b/video/image_loader.c
index 77deea0f8d..ba4d62acab 100644
--- a/video/image_loader.c
+++ b/video/image_loader.c
@@ -38,7 +38,7 @@ struct mp_image *load_image_png_buf(void *buffer, size_t buffer_size, int imgfmt
     if (frame && avcodec_receive_frame(avctx, frame) >= 0) {
         struct mp_image *r = mp_image_from_av_frame(frame);
         if (r)
-            res = convert_image(r, imgfmt, mp_null_log);
+            res = convert_image(r, imgfmt, NULL, mp_null_log);
         talloc_free(r);
     }
     av_frame_free(&frame);
diff --git a/video/image_writer.c b/video/image_writer.c
index e66543a772..5438249c42 100644
--- a/video/image_writer.c
+++ b/video/image_writer.c
@@ -314,6 +314,7 @@ int image_writer_format_from_ext(const char *ext)
 
 static struct mp_image *convert_image(struct mp_image *image, int destfmt,
                                       enum mp_csp_levels yuv_levels,
+                                      struct mpv_global *global,
                                       struct mp_log *log)
 {
     int d_w, d_h;
@@ -350,7 +351,14 @@ static struct mp_image *convert_image(struct mp_image *image, int destfmt,
 
     dst->params = p;
 
-    if (mp_image_swscale(dst, image, mp_sws_hq_flags) < 0) {
+    struct mp_sws_context *sws = mp_sws_alloc(NULL);
+    sws->log = log;
+    if (global)
+        mp_sws_enable_cmdline_opts(sws, global);
+    bool ok = mp_sws_scale(sws, dst, image) >= 0;
+    talloc_free(sws);
+
+    if (!ok) {
         mp_err(log, "Error when converting image.\n");
         talloc_free(dst);
         return NULL;
@@ -360,7 +368,8 @@ static struct mp_image *convert_image(struct mp_image *image, int destfmt,
 }
 
 bool write_image(struct mp_image *image, const struct image_writer_opts *opts,
-                const char *filename, struct mp_log *log)
+                const char *filename, struct mpv_global *global,
+                 struct mp_log *log)
 {
     struct image_writer_opts defs = image_writer_opts_defaults;
     if (!opts)
@@ -393,7 +402,7 @@ bool write_image(struct mp_image *image, const struct image_writer_opts *opts,
         levels = MP_CSP_LEVELS_PC;
     }
 
-    struct mp_image *dst = convert_image(image, destfmt, levels, log);
+    struct mp_image *dst = convert_image(image, destfmt, levels, global, log);
     if (!dst)
         return false;
 
@@ -416,5 +425,5 @@ void dump_png(struct mp_image *image, const char *filename, struct mp_log *log)
 {
     struct image_writer_opts opts = image_writer_opts_defaults;
     opts.format = AV_CODEC_ID_PNG;
-    write_image(image, &opts, filename, log);
+    write_image(image, &opts, filename, NULL, log);
 }
diff --git a/video/image_writer.h b/video/image_writer.h
index d178d7398b..f6d3b58f87 100644
--- a/video/image_writer.h
+++ b/video/image_writer.h
@@ -57,12 +57,15 @@ int image_writer_format_from_ext(const char *ext);
  *
  * File format and compression settings are controlled via the opts parameter.
  *
+ * If global!=NULL, use command line scaler options etc.
+ *
  * NOTE: The fields w/h/width/height of the passed mp_image must be all set
  *       accordingly. Setting w and width or h and height to different values
  *       can be used to store snapshots of anamorphic video.
  */
 bool write_image(struct mp_image *image, const struct image_writer_opts *opts,
-                 const char *filename, struct mp_log *log);
+                const char *filename, struct mpv_global *global,
+                 struct mp_log *log);
 
 // Debugging helper.
 void dump_png(struct mp_image *image, const char *filename, struct mp_log *log);
diff --git a/video/out/vo_image.c b/video/out/vo_image.c
index 9d30d5b6d2..2a3b8fae87 100644
--- a/video/out/vo_image.c
+++ b/video/out/vo_image.c
@@ -120,7 +120,7 @@ static void flip_page(struct vo *vo)
         filename = mp_path_join(t, p->opts->outdir, filename);
 
     MP_INFO(vo, "Saving %s\n", filename);
-    write_image(p->current, p->opts->opts, filename, vo->log);
+    write_image(p->current, p->opts->opts, filename, vo->global, vo->log);
 
     talloc_free(t);
     mp_image_unrefp(&p->current);
-- 
cgit v1.2.3