2020-11-12 13:44:30 +02:00
|
|
|
/*
|
|
|
|
* 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 AVUTIL_FILM_GRAIN_PARAMS_H
|
|
|
|
#define AVUTIL_FILM_GRAIN_PARAMS_H
|
|
|
|
|
|
|
|
#include "frame.h"
|
|
|
|
|
|
|
|
enum AVFilmGrainParamsType {
|
2020-11-27 03:12:10 +02:00
|
|
|
AV_FILM_GRAIN_PARAMS_NONE = 0,
|
2020-11-12 13:44:30 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* The union is valid when interpreted as AVFilmGrainAOMParams (codec.aom)
|
|
|
|
*/
|
2020-11-27 03:12:10 +02:00
|
|
|
AV_FILM_GRAIN_PARAMS_AV1,
|
2021-07-23 16:06:31 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* The union is valid when interpreted as AVFilmGrainH274Params (codec.h274)
|
|
|
|
*/
|
|
|
|
AV_FILM_GRAIN_PARAMS_H274,
|
2020-11-12 13:44:30 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
/**
|
|
|
|
* This structure describes how to handle film grain synthesis for AOM codecs.
|
|
|
|
*
|
|
|
|
* @note The struct must be allocated as part of AVFilmGrainParams using
|
|
|
|
* av_film_grain_params_alloc(). Its size is not a part of the public ABI.
|
|
|
|
*/
|
|
|
|
typedef struct AVFilmGrainAOMParams {
|
|
|
|
/**
|
|
|
|
* Number of points, and the scale and value for each point of the
|
|
|
|
* piecewise linear scaling function for the uma plane.
|
|
|
|
*/
|
|
|
|
int num_y_points;
|
|
|
|
uint8_t y_points[14][2 /* value, scaling */];
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Signals whether to derive the chroma scaling function from the luma.
|
|
|
|
* Not equivalent to copying the luma values and scales.
|
|
|
|
*/
|
|
|
|
int chroma_scaling_from_luma;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* If chroma_scaling_from_luma is set to 0, signals the chroma scaling
|
|
|
|
* function parameters.
|
|
|
|
*/
|
2020-11-26 22:29:35 +02:00
|
|
|
int num_uv_points[2 /* cb, cr */];
|
|
|
|
uint8_t uv_points[2 /* cb, cr */][10][2 /* value, scaling */];
|
2020-11-12 13:44:30 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the shift applied to the chroma components. For AV1, its within
|
|
|
|
* [8; 11] and determines the range and quantization of the film grain.
|
|
|
|
*/
|
|
|
|
int scaling_shift;
|
|
|
|
|
|
|
|
/**
|
2020-12-03 18:25:21 +02:00
|
|
|
* Specifies the auto-regression lag.
|
2020-11-12 13:44:30 +02:00
|
|
|
*/
|
|
|
|
int ar_coeff_lag;
|
|
|
|
|
|
|
|
/**
|
2020-12-03 18:25:21 +02:00
|
|
|
* Luma auto-regression coefficients. The number of coefficients is given by
|
|
|
|
* 2 * ar_coeff_lag * (ar_coeff_lag + 1).
|
2020-11-12 13:44:30 +02:00
|
|
|
*/
|
|
|
|
int8_t ar_coeffs_y[24];
|
|
|
|
|
|
|
|
/**
|
2020-12-03 18:25:21 +02:00
|
|
|
* Chroma auto-regression coefficients. The number of coefficients is given by
|
|
|
|
* 2 * ar_coeff_lag * (ar_coeff_lag + 1) + !!num_y_points.
|
2020-11-12 13:44:30 +02:00
|
|
|
*/
|
2020-11-26 22:29:35 +02:00
|
|
|
int8_t ar_coeffs_uv[2 /* cb, cr */][25];
|
2020-11-12 13:44:30 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the range of the auto-regressive coefficients. Values of 6,
|
|
|
|
* 7, 8 and so on represent a range of [-2, 2), [-1, 1), [-0.5, 0.5) and
|
|
|
|
* so on. For AV1 must be between 6 and 9.
|
|
|
|
*/
|
|
|
|
int ar_coeff_shift;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Signals the down shift applied to the generated gaussian numbers during
|
|
|
|
* synthesis.
|
|
|
|
*/
|
|
|
|
int grain_scale_shift;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the luma/chroma multipliers for the index to the component
|
|
|
|
* scaling function.
|
|
|
|
*/
|
2020-11-26 22:29:35 +02:00
|
|
|
int uv_mult[2 /* cb, cr */];
|
|
|
|
int uv_mult_luma[2 /* cb, cr */];
|
2020-11-12 13:44:30 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Offset used for component scaling function. For AV1 its a 9-bit value
|
|
|
|
* with a range [-256, 255]
|
|
|
|
*/
|
2020-11-26 22:29:35 +02:00
|
|
|
int uv_offset[2 /* cb, cr */];
|
2020-11-12 13:44:30 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Signals whether to overlap film grain blocks.
|
|
|
|
*/
|
|
|
|
int overlap_flag;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Signals to clip to limited color levels after film grain application.
|
|
|
|
*/
|
|
|
|
int limit_output_range;
|
|
|
|
} AVFilmGrainAOMParams;
|
|
|
|
|
2021-07-23 16:06:31 +02:00
|
|
|
/**
|
|
|
|
* This structure describes how to handle film grain synthesis for codecs using
|
|
|
|
* the ITU-T H.274 Versatile suplemental enhancement information message.
|
|
|
|
*
|
|
|
|
* @note The struct must be allocated as part of AVFilmGrainParams using
|
|
|
|
* av_film_grain_params_alloc(). Its size is not a part of the public ABI.
|
|
|
|
*/
|
|
|
|
typedef struct AVFilmGrainH274Params {
|
|
|
|
/**
|
|
|
|
* Specifies the film grain simulation mode.
|
|
|
|
* 0 = Frequency filtering, 1 = Auto-regression
|
|
|
|
*/
|
|
|
|
int model_id;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the bit depth used for the luma component.
|
|
|
|
*/
|
|
|
|
int bit_depth_luma;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the bit depth used for the chroma components.
|
|
|
|
*/
|
|
|
|
int bit_depth_chroma;
|
|
|
|
|
|
|
|
enum AVColorRange color_range;
|
|
|
|
enum AVColorPrimaries color_primaries;
|
|
|
|
enum AVColorTransferCharacteristic color_trc;
|
|
|
|
enum AVColorSpace color_space;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the blending mode used to blend the simulated film grain
|
|
|
|
* with the decoded images.
|
|
|
|
*
|
|
|
|
* 0 = Additive, 1 = Multiplicative
|
|
|
|
*/
|
|
|
|
int blending_mode_id;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies a scale factor used in the film grain characterization equations.
|
|
|
|
*/
|
|
|
|
int log2_scale_factor;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Indicates if the modelling of film grain for a given component is present.
|
|
|
|
*/
|
|
|
|
int component_model_present[3 /* y, cb, cr */];
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the number of intensity intervals for which a specific set of
|
|
|
|
* model values has been estimated, with a range of [1, 256].
|
|
|
|
*/
|
|
|
|
uint16_t num_intensity_intervals[3 /* y, cb, cr */];
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the number of model values present for each intensity interval
|
|
|
|
* in which the film grain has been modelled, with a range of [1, 6].
|
|
|
|
*/
|
|
|
|
uint8_t num_model_values[3 /* y, cb, cr */];
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the lower ounds of each intensity interval for whichthe set of
|
|
|
|
* model values applies for the component.
|
|
|
|
*/
|
|
|
|
uint8_t intensity_interval_lower_bound[3 /* y, cb, cr */][256 /* intensity interval */];
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the upper bound of each intensity interval for which the set of
|
|
|
|
* model values applies for the component.
|
|
|
|
*/
|
|
|
|
uint8_t intensity_interval_upper_bound[3 /* y, cb, cr */][256 /* intensity interval */];
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Specifies the model values for the component for each intensity interval.
|
|
|
|
* - When model_id == 0, the following applies:
|
|
|
|
* For comp_model_value[y], the range of values is [0, 2^bit_depth_luma - 1]
|
|
|
|
* For comp_model_value[cb..cr], the range of values is [0, 2^bit_depth_chroma - 1]
|
|
|
|
* - Otherwise, the following applies:
|
|
|
|
* For comp_model_value[y], the range of values is [-2^(bit_depth_luma - 1), 2^(bit_depth_luma - 1) - 1]
|
|
|
|
* For comp_model_value[cb..cr], the range of values is [-2^(bit_depth_chroma - 1), 2^(bit_depth_chroma - 1) - 1]
|
|
|
|
*/
|
|
|
|
int16_t comp_model_value[3 /* y, cb, cr */][256 /* intensity interval */][6 /* model value */];
|
|
|
|
} AVFilmGrainH274Params;
|
|
|
|
|
2020-11-12 13:44:30 +02:00
|
|
|
/**
|
|
|
|
* This structure describes how to handle film grain synthesis in video
|
|
|
|
* for specific codecs. Must be present on every frame where film grain is
|
|
|
|
* meant to be synthesised for correct presentation.
|
|
|
|
*
|
|
|
|
* @note The struct must be allocated with av_film_grain_params_alloc() and
|
|
|
|
* its size is not a part of the public ABI.
|
|
|
|
*/
|
|
|
|
typedef struct AVFilmGrainParams {
|
|
|
|
/**
|
|
|
|
* Specifies the codec for which this structure is valid.
|
|
|
|
*/
|
|
|
|
enum AVFilmGrainParamsType type;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Seed to use for the synthesis process, if the codec allows for it.
|
|
|
|
*/
|
|
|
|
uint64_t seed;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Additional fields may be added both here and in any structure included.
|
|
|
|
* If a codec's film grain structure differs slightly over another
|
|
|
|
* codec's, fields within may change meaning depending on the type.
|
|
|
|
*/
|
|
|
|
union {
|
|
|
|
AVFilmGrainAOMParams aom;
|
2021-07-23 16:06:31 +02:00
|
|
|
AVFilmGrainH274Params h274;
|
2020-11-12 13:44:30 +02:00
|
|
|
} codec;
|
|
|
|
} AVFilmGrainParams;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Allocate an AVFilmGrainParams structure and set its fields to
|
|
|
|
* default values. The resulting struct can be freed using av_freep().
|
|
|
|
* If size is not NULL it will be set to the number of bytes allocated.
|
|
|
|
*
|
|
|
|
* @return An AVFilmGrainParams filled with default values or NULL
|
|
|
|
* on failure.
|
|
|
|
*/
|
|
|
|
AVFilmGrainParams *av_film_grain_params_alloc(size_t *size);
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Allocate a complete AVFilmGrainParams and add it to the frame.
|
|
|
|
*
|
|
|
|
* @param frame The frame which side data is added to.
|
|
|
|
*
|
|
|
|
* @return The AVFilmGrainParams structure to be filled by caller.
|
|
|
|
*/
|
|
|
|
AVFilmGrainParams *av_film_grain_params_create_side_data(AVFrame *frame);
|
|
|
|
|
|
|
|
#endif /* AVUTIL_FILM_GRAIN_PARAMS_H */
|