| /** |
| * 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 |
| */ |
| |
| #include "film_grain_params.h" |
| #include "mem.h" |
| #include "pixdesc.h" |
| |
| AVFilmGrainParams *av_film_grain_params_alloc(size_t *size) |
| { |
| AVFilmGrainParams *params = av_mallocz(sizeof(AVFilmGrainParams)); |
| |
| if (size) |
| *size = sizeof(*params); |
| |
| return params; |
| } |
| |
| AVFilmGrainParams *av_film_grain_params_create_side_data(AVFrame *frame) |
| { |
| AVFilmGrainParams *fgp; |
| AVFrameSideData *side_data = av_frame_new_side_data(frame, |
| AV_FRAME_DATA_FILM_GRAIN_PARAMS, |
| sizeof(AVFilmGrainParams)); |
| if (!side_data) |
| return NULL; |
| |
| fgp = (AVFilmGrainParams *) side_data->data; |
| *fgp = (AVFilmGrainParams) { |
| .color_range = AVCOL_RANGE_UNSPECIFIED, |
| .color_primaries = AVCOL_PRI_UNSPECIFIED, |
| .color_trc = AVCOL_TRC_UNSPECIFIED, |
| .color_space = AVCOL_SPC_UNSPECIFIED, |
| }; |
| |
| return fgp; |
| } |
| |
| const AVFilmGrainParams *av_film_grain_params_select(const AVFrame *frame) |
| { |
| const AVFilmGrainParams *fgp, *best = NULL; |
| const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format); |
| int bit_depth_luma, bit_depth_chroma; |
| if (!desc) |
| return NULL; |
| |
| /* There are no YUV formats with different bit depth per component, |
| * so just check both against the first component for simplicity */ |
| bit_depth_luma = bit_depth_chroma = desc->comp[0].depth; |
| |
| for (int i = 0; i < frame->nb_side_data; i++) { |
| if (frame->side_data[i]->type != AV_FRAME_DATA_FILM_GRAIN_PARAMS) |
| continue; |
| fgp = (const AVFilmGrainParams*)frame->side_data[i]->data; |
| if (fgp->width && fgp->width > frame->width || |
| fgp->height && fgp->height > frame->height) |
| continue; |
| |
| #define CHECK(a, b, unspec) \ |
| if ((a) != (unspec) && (b) != (unspec) && (a) != (b)) \ |
| continue |
| |
| CHECK(fgp->bit_depth_luma, bit_depth_luma, 0); |
| CHECK(fgp->bit_depth_chroma, bit_depth_chroma, 0); |
| CHECK(fgp->color_range, frame->color_range, AVCOL_RANGE_UNSPECIFIED); |
| CHECK(fgp->color_primaries, frame->color_primaries, AVCOL_PRI_UNSPECIFIED); |
| CHECK(fgp->color_trc, frame->color_trc, AVCOL_TRC_UNSPECIFIED); |
| CHECK(fgp->color_space, frame->colorspace, AVCOL_SPC_UNSPECIFIED); |
| |
| switch (fgp->type) { |
| case AV_FILM_GRAIN_PARAMS_NONE: |
| continue; |
| case AV_FILM_GRAIN_PARAMS_AV1: |
| /* AOM FGS needs an exact match for the chroma resolution */ |
| if (fgp->subsampling_x != desc->log2_chroma_w || |
| fgp->subsampling_y != desc->log2_chroma_h) |
| continue; |
| break; |
| case AV_FILM_GRAIN_PARAMS_H274: |
| /* H.274 FGS can be adapted to any lower chroma resolution */ |
| if (fgp->subsampling_x > desc->log2_chroma_w || |
| fgp->subsampling_y > desc->log2_chroma_h) |
| continue; |
| break; |
| } |
| |
| if (!best || best->width < fgp->width || best->height < fgp->height) |
| best = fgp; |
| } |
| |
| return best; |
| } |