android_external_ffmpeg/libswscale/tests/sws_ops.c
Niklas Haas d5174f9e5b swscale/format: add source format info to ff_sws_encode_colors()
Specifically, I need access to this for generating a better dither matrix.
2025-12-15 14:31:58 +00:00

105 lines
3.1 KiB
C

/*
* Copyright (C) 2025 Niklas Haas
*
* This file is part of FFmpeg.
*
* FFmpeg 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.
*
* FFmpeg 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 FFmpeg; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "libavutil/pixdesc.h"
#include "libswscale/ops.h"
#include "libswscale/format.h"
#ifdef _WIN32
#include <io.h>
#include <fcntl.h>
#endif
static int run_test(SwsContext *const ctx, AVFrame *frame,
const AVPixFmtDescriptor *const src_desc,
const AVPixFmtDescriptor *const dst_desc)
{
/* Reuse ff_fmt_from_frame() to ensure correctly sanitized metadata */
frame->format = av_pix_fmt_desc_get_id(src_desc);
SwsFormat src = ff_fmt_from_frame(frame, 0);
frame->format = av_pix_fmt_desc_get_id(dst_desc);
SwsFormat dst = ff_fmt_from_frame(frame, 0);
bool incomplete = ff_infer_colors(&src.color, &dst.color);
SwsOpList *ops = ff_sws_op_list_alloc();
if (!ops)
return AVERROR(ENOMEM);
ops->src = src;
ops->dst = dst;
if (ff_sws_decode_pixfmt(ops, src.format) < 0)
goto fail;
if (ff_sws_decode_colors(ctx, SWS_PIXEL_F32, ops, src, &incomplete) < 0)
goto fail;
if (ff_sws_encode_colors(ctx, SWS_PIXEL_F32, ops, src, dst, &incomplete) < 0)
goto fail;
if (ff_sws_encode_pixfmt(ops, dst.format) < 0)
goto fail;
av_log(NULL, AV_LOG_INFO, "%s -> %s:\n",
av_get_pix_fmt_name(src.format), av_get_pix_fmt_name(dst.format));
ff_sws_op_list_optimize(ops);
ff_sws_op_list_print(NULL, AV_LOG_INFO, ops);
fail:
/* silently skip unsupported formats */
ff_sws_op_list_free(&ops);
return 0;
}
static void log_stdout(void *avcl, int level, const char *fmt, va_list vl)
{
if (level != AV_LOG_INFO) {
av_log_default_callback(avcl, level, fmt, vl);
} else {
vfprintf(stdout, fmt, vl);
}
}
int main(int argc, char **argv)
{
int ret = 1;
#ifdef _WIN32
_setmode(_fileno(stdout), _O_BINARY);
#endif
SwsContext *ctx = sws_alloc_context();
AVFrame *frame = av_frame_alloc();
if (!ctx || !frame)
goto fail;
frame->width = frame->height = 16;
av_log_set_callback(log_stdout);
for (const AVPixFmtDescriptor *src = NULL; (src = av_pix_fmt_desc_next(src));) {
for (const AVPixFmtDescriptor *dst = NULL; (dst = av_pix_fmt_desc_next(dst));) {
int err = run_test(ctx, frame, src, dst);
if (err < 0)
goto fail;
}
}
ret = 0;
fail:
av_frame_free(&frame);
sws_free_context(&ctx);
return ret;
}