FFmpeg
film_grain_params.c
Go to the documentation of this file.
1 /**
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 #include "film_grain_params.h"
20 #include "mem.h"
21 #include "pixdesc.h"
22 
24 {
26 
27  if (size)
28  *size = sizeof(*params);
29 
30  return params;
31 }
32 
34 {
35  AVFilmGrainParams *fgp;
38  sizeof(AVFilmGrainParams));
39  if (!side_data)
40  return NULL;
41 
42  fgp = (AVFilmGrainParams *) side_data->data;
43  *fgp = (AVFilmGrainParams) {
45  .color_primaries = AVCOL_PRI_UNSPECIFIED,
46  .color_trc = AVCOL_TRC_UNSPECIFIED,
47  .color_space = AVCOL_SPC_UNSPECIFIED,
48  };
49 
50  return fgp;
51 }
52 
54 {
55  const AVFilmGrainParams *fgp, *best = NULL;
57  int bit_depth_luma, bit_depth_chroma;
58  if (!desc)
59  return NULL;
60 
61  /* There are no YUV formats with different bit depth per component,
62  * so just check both against the first component for simplicity */
63  bit_depth_luma = bit_depth_chroma = desc->comp[0].depth;
64 
65  for (int i = 0; i < frame->nb_side_data; i++) {
66  if (frame->side_data[i]->type != AV_FRAME_DATA_FILM_GRAIN_PARAMS)
67  continue;
68  fgp = (const AVFilmGrainParams*)frame->side_data[i]->data;
69  if (fgp->width && fgp->width > frame->width ||
70  fgp->height && fgp->height > frame->height)
71  continue;
72 
73 #define CHECK(a, b, unspec) \
74  if ((a) != (unspec) && (b) != (unspec) && (a) != (b)) \
75  continue
76 
77  CHECK(fgp->bit_depth_luma, bit_depth_luma, 0);
78  CHECK(fgp->bit_depth_chroma, bit_depth_chroma, 0);
79  CHECK(fgp->color_range, frame->color_range, AVCOL_RANGE_UNSPECIFIED);
80  CHECK(fgp->color_primaries, frame->color_primaries, AVCOL_PRI_UNSPECIFIED);
81  CHECK(fgp->color_trc, frame->color_trc, AVCOL_TRC_UNSPECIFIED);
82  CHECK(fgp->color_space, frame->colorspace, AVCOL_SPC_UNSPECIFIED);
83 
84  switch (fgp->type) {
86  continue;
88  /* AOM FGS needs an exact match for the chroma resolution */
89  if (fgp->subsampling_x != desc->log2_chroma_w ||
90  fgp->subsampling_y != desc->log2_chroma_h)
91  continue;
92  break;
94  /* H.274 FGS can be adapted to any lower chroma resolution */
95  if (fgp->subsampling_x > desc->log2_chroma_w ||
96  fgp->subsampling_y > desc->log2_chroma_h)
97  continue;
98  break;
99  }
100 
101  if (!best || best->width < fgp->width || best->height < fgp->height)
102  best = fgp;
103  }
104 
105  return best;
106 }
AVFilmGrainParams::bit_depth_luma
int bit_depth_luma
Intended bit depth, or 0 for unknown/unspecified.
Definition: film_grain_params.h:287
av_frame_new_side_data
AVFrameSideData * av_frame_new_side_data(AVFrame *frame, enum AVFrameSideDataType type, size_t size)
Add a new side data to a frame.
Definition: frame.c:811
av_pix_fmt_desc_get
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:3170
av_film_grain_params_alloc
AVFilmGrainParams * av_film_grain_params_alloc(size_t *size)
This file is part of FFmpeg.
Definition: film_grain_params.c:23
AV_FRAME_DATA_FILM_GRAIN_PARAMS
@ AV_FRAME_DATA_FILM_GRAIN_PARAMS
Film grain parameters for a frame, described by AVFilmGrainParams.
Definition: frame.h:188
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:389
pixdesc.h
AVCOL_TRC_UNSPECIFIED
@ AVCOL_TRC_UNSPECIFIED
Definition: pixfmt.h:614
AVFilmGrainParams::color_space
enum AVColorSpace color_space
Definition: film_grain_params.h:282
AVFilmGrainParams::color_trc
enum AVColorTransferCharacteristic color_trc
Definition: film_grain_params.h:281
av_film_grain_params_select
const AVFilmGrainParams * av_film_grain_params_select(const AVFrame *frame)
Select the most appropriate film grain parameters set for the frame, taking into account the frame's ...
Definition: film_grain_params.c:53
AVFilmGrainParams::bit_depth_chroma
int bit_depth_chroma
Definition: film_grain_params.h:288
film_grain_params.h
AVFilmGrainParams::width
int width
Intended display resolution.
Definition: film_grain_params.h:269
av_film_grain_params_create_side_data
AVFilmGrainParams * av_film_grain_params_create_side_data(AVFrame *frame)
Allocate a complete AVFilmGrainParams and add it to the frame.
Definition: film_grain_params.c:33
AVCOL_PRI_UNSPECIFIED
@ AVCOL_PRI_UNSPECIFIED
Definition: pixfmt.h:589
AV_FILM_GRAIN_PARAMS_NONE
@ AV_FILM_GRAIN_PARAMS_NONE
Definition: film_grain_params.h:25
if
if(ret)
Definition: filter_design.txt:179
NULL
#define NULL
Definition: coverity.c:32
AVFilmGrainParams::subsampling_x
int subsampling_x
Intended subsampling ratio, or 0 for luma-only streams.
Definition: film_grain_params.h:274
AVCOL_RANGE_UNSPECIFIED
@ AVCOL_RANGE_UNSPECIFIED
Definition: pixfmt.h:683
size
int size
Definition: twinvq_data.h:10344
CHECK
#define CHECK(a, b, unspec)
AVFrameSideData::data
uint8_t * data
Definition: frame.h:267
AVFilmGrainParams
This structure describes how to handle film grain synthesis in video for specific codecs.
Definition: film_grain_params.h:238
AVFilmGrainParams::color_primaries
enum AVColorPrimaries color_primaries
Definition: film_grain_params.h:280
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
AVFilmGrainParams::subsampling_y
int subsampling_y
Definition: film_grain_params.h:274
av_mallocz
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:256
AVCOL_SPC_UNSPECIFIED
@ AVCOL_SPC_UNSPECIFIED
Definition: pixfmt.h:643
AVFilmGrainParams::height
int height
Definition: film_grain_params.h:269
frame
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several the filter must be ready for frames arriving randomly on any input any filter with several inputs will most likely require some kind of queuing mechanism It is perfectly acceptable to have a limited queue and to drop frames when the inputs are too unbalanced request_frame For filters that do not use the this method is called when a frame is wanted on an output For a it should directly call filter_frame on the corresponding output For a if there are queued frames already one of these frames should be pushed If the filter should request a frame on one of its repeatedly until at least one frame has been pushed Return or at least make progress towards producing a frame
Definition: filter_design.txt:264
AV_FILM_GRAIN_PARAMS_H274
@ AV_FILM_GRAIN_PARAMS_H274
The union is valid when interpreted as AVFilmGrainH274Params (codec.h274)
Definition: film_grain_params.h:35
AVFilmGrainParams::color_range
enum AVColorRange color_range
Intended video signal characteristics.
Definition: film_grain_params.h:279
desc
const char * desc
Definition: libsvtav1.c:79
mem.h
AVFrameSideData
Structure to hold side data for an AVFrame.
Definition: frame.h:265
AVPixFmtDescriptor
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:69
AV_FILM_GRAIN_PARAMS_AV1
@ AV_FILM_GRAIN_PARAMS_AV1
The union is valid when interpreted as AVFilmGrainAOMParams (codec.aom)
Definition: film_grain_params.h:30
AVFilmGrainParams::type
enum AVFilmGrainParamsType type
Specifies the codec for which this structure is valid.
Definition: film_grain_params.h:242