1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
/*
* 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 */
|
