forked from FFmpeg/FFmpeg
a57fe519b6
This is a lightweight wrapper around the underlying color management system, whose job it is merely to manage the 3DLUT state and apply them to the frame data. This is where we might add platform-specific optimizations in the future. I also plan on adding support for more pixel formats in the future. In particular, we could support YUV or XYZ input formats directly using only negligible additional code in the 3DLUT setup functions. This would eliminate the major source of slowdown, which is currently the roundtrip to RGBA64.
98 lines
3.1 KiB
C
98 lines
3.1 KiB
C
/*
|
|
* Copyright (C) 2024 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
|
|
*/
|
|
|
|
#ifndef SWSCALE_LUT3D_H
|
|
#define SWSCALE_LUT3D_H
|
|
|
|
#include <stdint.h>
|
|
|
|
#include "libavutil/csp.h"
|
|
#include "libavutil/pixfmt.h"
|
|
|
|
#include "cms.h"
|
|
#include "csputils.h"
|
|
#include "utils.h"
|
|
|
|
enum {
|
|
/* Input LUT size. This is only calculated once. */
|
|
INPUT_LUT_BITS = 6,
|
|
INPUT_LUT_SIZE = (1 << INPUT_LUT_BITS) + 1, /* +1 to simplify interpolation */
|
|
|
|
/* Tone mapping LUT size. This is regenerated possibly per frame. */
|
|
TONE_LUT_BITS = 8,
|
|
TONE_LUT_SIZE = (1 << TONE_LUT_BITS) + 1,
|
|
|
|
/* Output LUT size (for dynamic tone mapping). This is only calculated once. */
|
|
OUTPUT_LUT_BITS_I = 6,
|
|
OUTPUT_LUT_BITS_PT = 7,
|
|
|
|
OUTPUT_LUT_SIZE_I = (1 << OUTPUT_LUT_BITS_I) + 1,
|
|
OUTPUT_LUT_SIZE_PT = (1 << OUTPUT_LUT_BITS_PT) + 1,
|
|
};
|
|
|
|
typedef struct SwsLut3D {
|
|
SwsColorMap map;
|
|
bool dynamic;
|
|
|
|
/* Gamut mapping 3DLUT(s) */
|
|
v3u16_t input[INPUT_LUT_SIZE][INPUT_LUT_SIZE][INPUT_LUT_SIZE];
|
|
v3u16_t output[OUTPUT_LUT_SIZE_PT][OUTPUT_LUT_SIZE_PT][OUTPUT_LUT_SIZE_I];
|
|
|
|
/* Split tone mapping LUT (for dynamic tone mapping) */
|
|
v2u16_t tone_map[TONE_LUT_SIZE]; /* new luma, desaturation */
|
|
} SwsLut3D;
|
|
|
|
SwsLut3D *sws_lut3d_alloc(void);
|
|
void sws_lut3d_free(SwsLut3D **lut3d);
|
|
|
|
/**
|
|
* Test to see if a given format is supported by the 3DLUT input/output code.
|
|
*/
|
|
bool sws_lut3d_test_fmt(enum AVPixelFormat fmt, int output);
|
|
|
|
/**
|
|
* Pick the best compatible pixfmt for a given SwsFormat.
|
|
*/
|
|
enum AVPixelFormat sws_lut3d_pick_pixfmt(SwsFormat fmt, int output);
|
|
|
|
/**
|
|
* Recalculate the (static) 3DLUT state with new settings. This will recompute
|
|
* everything. To only update per-frame tone mapping state, instead call
|
|
* sws_lut3d_update().
|
|
*
|
|
* Returns 0 or a negative error code.
|
|
*/
|
|
int sws_lut3d_generate(SwsLut3D *lut3d, enum AVPixelFormat fmt_in,
|
|
enum AVPixelFormat fmt_out, const SwsColorMap *map);
|
|
|
|
/**
|
|
* Update the tone mapping state. This will only use per-frame metadata. The
|
|
* static metadata is ignored.
|
|
*/
|
|
void sws_lut3d_update(SwsLut3D *lut3d, const SwsColor *new_src);
|
|
|
|
/**
|
|
* Applies a color transformation to a plane. The format must match the format
|
|
* provided during sws_lut3d_update().
|
|
*/
|
|
void sws_lut3d_apply(const SwsLut3D *lut3d, const uint8_t *in, int in_stride,
|
|
uint8_t *out, int out_stride, int w, int h);
|
|
|
|
#endif /* SWSCALE_LUT3D_H */
|