mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
3.1 KiB
98 lines
3.1 KiB
/* |
|
* 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 *ff_sws_lut3d_alloc(void); |
|
void ff_sws_lut3d_free(SwsLut3D **lut3d); |
|
|
|
/** |
|
* Test to see if a given format is supported by the 3DLUT input/output code. |
|
*/ |
|
bool ff_sws_lut3d_test_fmt(enum AVPixelFormat fmt, int output); |
|
|
|
/** |
|
* Pick the best compatible pixfmt for a given SwsFormat. |
|
*/ |
|
enum AVPixelFormat ff_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 |
|
* ff_sws_lut3d_update(). |
|
* |
|
* Returns 0 or a negative error code. |
|
*/ |
|
int ff_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 ff_sws_lut3d_update(SwsLut3D *lut3d, const SwsColor *new_src); |
|
|
|
/** |
|
* Applies a color transformation to a plane. The format must match the format |
|
* provided during ff_sws_lut3d_update(). |
|
*/ |
|
void ff_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 */
|
|
|