Something went wrong. Try again.
🎥 Command line media player
Something went wrong. Try again.
C
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430/* * This file is part of mpv. * * mpv is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * mpv is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with mpv. If not, see <http://www.gnu.org/licenses/>. */
#include <stdio.h>#include <stdlib.h>#include <string.h>
#include <libavcodec/avcodec.h>#include <libavutil/mem.h>#include <libavutil/opt.h>
#include "config.h"
#if HAVE_JPEG#include <setjmp.h>#include <jpeglib.h>#endif
#include "osdep/io.h"
#include "image_writer.h"#include "mpv_talloc.h"#include "video/img_format.h"#include "video/mp_image.h"#include "video/fmt-conversion.h"#include "video/sws_utils.h"
#include "options/m_option.h"
const struct image_writer_opts image_writer_opts_defaults = { .format = AV_CODEC_ID_MJPEG, .high_bit_depth = 1, .png_compression = 7, .png_filter = 5, .jpeg_quality = 90, .jpeg_source_chroma = 1, .webp_lossless = 0, .webp_quality = 75, .webp_compression = 4, .tag_csp = 0,};
const struct m_opt_choice_alternatives mp_image_writer_formats[] = { {"jpg", AV_CODEC_ID_MJPEG}, {"jpeg", AV_CODEC_ID_MJPEG}, {"png", AV_CODEC_ID_PNG}, {"webp", AV_CODEC_ID_WEBP}, {0}};
#define OPT_BASE_STRUCT struct image_writer_opts
const struct m_option image_writer_opts[] = { OPT_CHOICE_C("format", format, 0, mp_image_writer_formats), OPT_INTRANGE("jpeg-quality", jpeg_quality, 0, 0, 100), OPT_FLAG("jpeg-source-chroma", jpeg_source_chroma, 0), OPT_INTRANGE("png-compression", png_compression, 0, 0, 9), OPT_INTRANGE("png-filter", png_filter, 0, 0, 5), OPT_FLAG("webp-lossless", webp_lossless, 0), OPT_INTRANGE("webp-quality", webp_quality, 0, 0, 100), OPT_INTRANGE("webp-compression", webp_compression, 0, 0, 6), OPT_FLAG("high-bit-depth", high_bit_depth, 0), OPT_FLAG("tag-colorspace", tag_csp, 0), {0},};
struct image_writer_ctx { struct mp_log *log; const struct image_writer_opts *opts; struct mp_imgfmt_desc original_format;};
static enum AVPixelFormat replace_j_format(enum AVPixelFormat fmt){ switch (fmt) { case AV_PIX_FMT_YUV420P: return AV_PIX_FMT_YUVJ420P; case AV_PIX_FMT_YUV422P: return AV_PIX_FMT_YUVJ422P; case AV_PIX_FMT_YUV444P: return AV_PIX_FMT_YUVJ444P; } return fmt;}
static bool write_lavc(struct image_writer_ctx *ctx, mp_image_t *image, FILE *fp){ bool success = 0; AVFrame *pic = NULL; AVPacket pkt = {0}; int got_output = 0;
av_init_packet(&pkt);
struct AVCodec *codec; if (ctx->opts->format == AV_CODEC_ID_WEBP) { codec = avcodec_find_encoder_by_name("libwebp"); // non-animated encoder } else { codec = avcodec_find_encoder(ctx->opts->format); }
AVCodecContext *avctx = NULL; if (!codec) goto print_open_fail; avctx = avcodec_alloc_context3(codec); if (!avctx) goto print_open_fail;
avctx->time_base = AV_TIME_BASE_Q; avctx->width = image->w; avctx->height = image->h; avctx->color_range = mp_csp_levels_to_avcol_range(image->params.color.levels); avctx->pix_fmt = imgfmt2pixfmt(image->imgfmt); if (codec->id == AV_CODEC_ID_MJPEG) { // Annoying deprecated garbage for the jpg encoder. if (image->params.color.levels == MP_CSP_LEVELS_PC) avctx->pix_fmt = replace_j_format(avctx->pix_fmt); } if (avctx->pix_fmt == AV_PIX_FMT_NONE) { MP_ERR(ctx, "Image format %s not supported by lavc.\n", mp_imgfmt_to_name(image->imgfmt)); goto error_exit; } if (codec->id == AV_CODEC_ID_PNG) { avctx->compression_level = ctx->opts->png_compression; av_opt_set_int(avctx, "pred", ctx->opts->png_filter, AV_OPT_SEARCH_CHILDREN); } else if (codec->id == AV_CODEC_ID_WEBP) { avctx->compression_level = ctx->opts->webp_compression; av_opt_set_int(avctx, "lossless", ctx->opts->webp_lossless, AV_OPT_SEARCH_CHILDREN); av_opt_set_int(avctx, "quality", ctx->opts->webp_quality, AV_OPT_SEARCH_CHILDREN); }
if (avcodec_open2(avctx, codec, NULL) < 0) { print_open_fail: MP_ERR(ctx, "Could not open libavcodec encoder for saving images\n"); goto error_exit; }
pic = av_frame_alloc(); if (!pic) goto error_exit; for (int n = 0; n < 4; n++) { pic->data[n] = image->planes[n]; pic->linesize[n] = image->stride[n]; } pic->format = avctx->pix_fmt; pic->width = avctx->width; pic->height = avctx->height; pic->color_range = avctx->color_range; if (ctx->opts->tag_csp) { pic->color_primaries = mp_csp_prim_to_avcol_pri(image->params.color.primaries); pic->color_trc = mp_csp_trc_to_avcol_trc(image->params.color.gamma); }
int ret = avcodec_send_frame(avctx, pic); if (ret < 0) goto error_exit; ret = avcodec_send_frame(avctx, NULL); // send EOF if (ret < 0) goto error_exit; ret = avcodec_receive_packet(avctx, &pkt); if (ret < 0) goto error_exit; got_output = 1;
fwrite(pkt.data, pkt.size, 1, fp);
success = !!got_output;error_exit: avcodec_free_context(&avctx); av_frame_free(&pic); av_packet_unref(&pkt); return success;}
#if HAVE_JPEG
static void write_jpeg_error_exit(j_common_ptr cinfo){ // NOTE: do not write error message, too much effort to connect the libjpeg // log callbacks with mplayer's log function mp_msp()
// Return control to the setjmp point longjmp(*(jmp_buf*)cinfo->client_data, 1);}
static bool write_jpeg(struct image_writer_ctx *ctx, mp_image_t *image, FILE *fp){ struct jpeg_compress_struct cinfo; struct jpeg_error_mgr jerr;
cinfo.err = jpeg_std_error(&jerr); jerr.error_exit = write_jpeg_error_exit;
jmp_buf error_return_jmpbuf; cinfo.client_data = &error_return_jmpbuf; if (setjmp(cinfo.client_data)) { jpeg_destroy_compress(&cinfo); return false; }
jpeg_create_compress(&cinfo); jpeg_stdio_dest(&cinfo, fp);
cinfo.image_width = image->w; cinfo.image_height = image->h; cinfo.input_components = 3; cinfo.in_color_space = JCS_RGB;
cinfo.write_JFIF_header = TRUE; cinfo.JFIF_major_version = 1; cinfo.JFIF_minor_version = 2;
jpeg_set_defaults(&cinfo); jpeg_set_quality(&cinfo, ctx->opts->jpeg_quality, 0);
if (ctx->opts->jpeg_source_chroma) { cinfo.comp_info[0].h_samp_factor = 1 << ctx->original_format.chroma_xs; cinfo.comp_info[0].v_samp_factor = 1 << ctx->original_format.chroma_ys; }
jpeg_start_compress(&cinfo, TRUE);
while (cinfo.next_scanline < cinfo.image_height) { JSAMPROW row_pointer[1]; row_pointer[0] = image->planes[0] + (ptrdiff_t)cinfo.next_scanline * image->stride[0]; jpeg_write_scanlines(&cinfo, row_pointer,1); }
jpeg_finish_compress(&cinfo);
jpeg_destroy_compress(&cinfo);
return true;}
#endif
static int get_encoder_format(struct AVCodec *codec, int srcfmt, bool highdepth){ const enum AVPixelFormat *pix_fmts = codec->pix_fmts; int current = 0; for (int n = 0; pix_fmts && pix_fmts[n] != AV_PIX_FMT_NONE; n++) { int fmt = pixfmt2imgfmt(pix_fmts[n]); if (!fmt) continue; // Ignore formats larger than 8 bit per pixel. if (!highdepth && IMGFMT_RGB_DEPTH(fmt) > 32) continue; current = current ? mp_imgfmt_select_best(current, fmt, srcfmt) : fmt; } return current;}
static int get_target_format(struct image_writer_ctx *ctx){ struct AVCodec *codec = avcodec_find_encoder(ctx->opts->format); if (!codec) goto unknown;
int srcfmt = ctx->original_format.id;
int target = get_encoder_format(codec, srcfmt, ctx->opts->high_bit_depth); if (!target) target = get_encoder_format(codec, srcfmt, true);
if (!target) goto unknown;
return target;
unknown: return IMGFMT_RGB0;}
const char *image_writer_file_ext(const struct image_writer_opts *opts){ struct image_writer_opts defs = image_writer_opts_defaults;
if (!opts) opts = &defs;
return m_opt_choice_str(mp_image_writer_formats, opts->format);}
bool image_writer_high_depth(const struct image_writer_opts *opts){ return opts->format == AV_CODEC_ID_PNG;}
int image_writer_format_from_ext(const char *ext){ for (int n = 0; mp_image_writer_formats[n].name; n++) { if (ext && strcmp(mp_image_writer_formats[n].name, ext) == 0) return mp_image_writer_formats[n].value; } return 0;}
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; mp_image_params_get_dsize(&image->params, &d_w, &d_h);
struct mp_image_params p = { .imgfmt = destfmt, .w = d_w, .h = d_h, .p_w = 1, .p_h = 1, }; mp_image_params_guess_csp(&p);
// If RGB, just assume everything is correct. if (p.color.space != MP_CSP_RGB) { // Currently, assume what FFmpeg's jpg encoder or libwebp needs. // Of course this works only for non-HDR (no HDR support in libswscale). p.color.levels = yuv_levels; p.color.space = MP_CSP_BT_601; p.chroma_location = MP_CHROMA_CENTER; mp_image_params_guess_csp(&p); }
if (mp_image_params_equal(&p, &image->params)) return mp_image_new_ref(image);
struct mp_image *dst = mp_image_alloc(p.imgfmt, p.w, p.h); if (!dst) { mp_err(log, "Out of memory.\n"); return NULL; } mp_image_copy_attributes(dst, image);
dst->params = p;
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; }
return dst;}
bool write_image(struct mp_image *image, const struct image_writer_opts *opts, const char *filename, struct mpv_global *global, struct mp_log *log){ struct image_writer_opts defs = image_writer_opts_defaults; if (!opts) opts = &defs;
struct image_writer_ctx ctx = { log, opts, image->fmt }; bool (*write)(struct image_writer_ctx *, mp_image_t *, FILE *) = write_lavc; int destfmt = 0;
#if HAVE_JPEG if (opts->format == AV_CODEC_ID_MJPEG) { write = write_jpeg; destfmt = IMGFMT_RGB24; }#endif if (opts->format == AV_CODEC_ID_WEBP && !opts->webp_lossless) { // For lossy images, libwebp has its own RGB->YUV conversion. // We don't want that, so force YUV/YUVA here. int alpha = image->fmt.flags & MP_IMGFLAG_ALPHA; destfmt = alpha ? pixfmt2imgfmt(AV_PIX_FMT_YUVA420P) : IMGFMT_420P; }
if (!destfmt) destfmt = get_target_format(&ctx);
enum mp_csp_levels levels; // Ignored if destfmt is a RGB format if (opts->format == AV_CODEC_ID_WEBP) { levels = MP_CSP_LEVELS_TV; } else { levels = MP_CSP_LEVELS_PC; }
struct mp_image *dst = convert_image(image, destfmt, levels, global, log); if (!dst) return false;
FILE *fp = fopen(filename, "wb"); bool success = false; if (fp == NULL) { mp_err(log, "Error opening '%s' for writing!\n", filename); } else { success = write(&ctx, dst, fp); success = !fclose(fp) && success; if (!success) mp_err(log, "Error writing file '%s'!\n", filename); }
talloc_free(dst); return success;}
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, NULL, log);}