FFmpeg
h264_redundant_pps_bsf.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 <string.h>
20 
21 #include "libavutil/common.h"
22 #include "libavutil/mem.h"
23 
24 #include "bsf.h"
25 #include "cbs.h"
26 #include "cbs_h264.h"
27 #include "h264.h"
28 
29 
30 typedef struct H264RedundantPPSContext {
33 
35 
40 
41 
43  H264RawPPS *pps)
44 {
45  // Record the current value of pic_init_qp in order to fix up
46  // following slices, then overwrite with the global value.
48  pps->pic_init_qp_minus26 = ctx->global_pic_init_qp - 26;
49 
50  // Some PPSs have this set, so it must be set in all of them.
51  // (Slices which do not use such a PPS on input will still have
52  // *_weight_l*flag as zero and therefore write equivalently.)
53  pps->weighted_pred_flag = 1;
54 
55  return 0;
56 }
57 
59  H264RawSliceHeader *slice)
60 {
61  int qp;
62 
63  qp = ctx->current_pic_init_qp + slice->slice_qp_delta;
64  slice->slice_qp_delta = qp - ctx->global_pic_init_qp;
65 
66  return 0;
67 }
68 
70 {
72  AVPacket *in;
74  int au_has_sps;
75  int err, i;
76 
77  err = ff_bsf_get_packet(bsf, &in);
78  if (err < 0)
79  return err;
80 
81  err = ff_cbs_read_packet(ctx->input, au, in);
82  if (err < 0)
83  goto fail;
84 
85  au_has_sps = 0;
86  for (i = 0; i < au->nb_units; i++) {
87  CodedBitstreamUnit *nal = &au->units[i];
88 
89  if (nal->type == H264_NAL_SPS)
90  au_has_sps = 1;
91  if (nal->type == H264_NAL_PPS) {
92  err = h264_redundant_pps_fixup_pps(ctx, nal->content);
93  if (err < 0)
94  goto fail;
95  if (!au_has_sps) {
96  av_log(bsf, AV_LOG_VERBOSE, "Deleting redundant PPS "
97  "at %"PRId64".\n", in->pts);
98  err = ff_cbs_delete_unit(ctx->input, au, i);
99  if (err < 0)
100  goto fail;
101  }
102  }
103  if (nal->type == H264_NAL_SLICE ||
104  nal->type == H264_NAL_IDR_SLICE) {
105  H264RawSlice *slice = nal->content;
107  }
108  }
109 
110  err = ff_cbs_write_packet(ctx->output, out, au);
111  if (err < 0)
112  goto fail;
113 
114 
115  err = av_packet_copy_props(out, in);
116  if (err < 0)
117  goto fail;
118 
119  err = 0;
120 fail:
121  ff_cbs_fragment_reset(ctx->output, au);
122  av_packet_free(&in);
123  if (err < 0)
124  av_packet_unref(out);
125 
126  return err;
127 }
128 
130 {
133  int err, i;
134 
135  err = ff_cbs_init(&ctx->input, AV_CODEC_ID_H264, bsf);
136  if (err < 0)
137  return err;
138 
139  err = ff_cbs_init(&ctx->output, AV_CODEC_ID_H264, bsf);
140  if (err < 0)
141  return err;
142 
143  ctx->global_pic_init_qp = 26;
144 
145  if (bsf->par_in->extradata) {
146  err = ff_cbs_read_extradata(ctx->input, au, bsf->par_in);
147  if (err < 0) {
148  av_log(bsf, AV_LOG_ERROR, "Failed to read extradata.\n");
149  goto fail;
150  }
151 
152  for (i = 0; i < au->nb_units; i++) {
153  if (au->units[i].type == H264_NAL_PPS) {
154  err = h264_redundant_pps_fixup_pps(ctx, au->units[i].content);
155  if (err < 0)
156  goto fail;
157  }
158  }
159 
161  err = ff_cbs_write_extradata(ctx->output, bsf->par_out, au);
162  if (err < 0) {
163  av_log(bsf, AV_LOG_ERROR, "Failed to write extradata.\n");
164  goto fail;
165  }
166  }
167 
168  err = 0;
169 fail:
170  ff_cbs_fragment_reset(ctx->output, au);
171  return err;
172 }
173 
175 {
178 }
179 
181 {
183 
185  ff_cbs_close(&ctx->input);
186  ff_cbs_close(&ctx->output);
187 }
188 
191 };
192 
194  .name = "h264_redundant_pps",
195  .priv_data_size = sizeof(H264RedundantPPSContext),
198  .close = &h264_redundant_pps_close,
201 };
static int h264_redundant_pps_init(AVBSFContext *bsf)
int nb_units
Number of units in this fragment.
Definition: cbs.h:147
AVCodecParameters * par_out
Parameters of the output stream.
Definition: avcodec.h:5793
static void flush(AVCodecContext *avctx)
Memory handling functions.
int ff_cbs_write_packet(CodedBitstreamContext *ctx, AVPacket *pkt, CodedBitstreamFragment *frag)
Write the bitstream of a fragment to a packet.
Definition: cbs.c:345
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
int ff_cbs_init(CodedBitstreamContext **ctx_ptr, enum AVCodecID codec_id, void *log_ctx)
Create and initialise a new context for the given codec.
Definition: cbs.c:74
CodedBitstreamUnitType type
Codec-specific type of this unit.
Definition: cbs.h:68
The bitstream filter state.
Definition: avcodec.h:5759
void * priv_data
Opaque filter-specific private data.
Definition: avcodec.h:5780
const AVBitStreamFilter ff_h264_redundant_pps_bsf
CodedBitstreamContext * input
uint8_t weighted_pred_flag
Definition: cbs_h264.h:204
static int h264_redundant_pps_fixup_slice(H264RedundantPPSContext *ctx, H264RawSliceHeader *slice)
int8_t slice_qp_delta
Definition: cbs_h264.h:417
void av_packet_free(AVPacket **pkt)
Free the packet, if the packet is reference counted, it will be unreferenced first.
Definition: avpacket.c:62
int ff_cbs_read_packet(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag, const AVPacket *pkt)
Read the data bitstream from a packet into a fragment, then split into units and decompose.
Definition: cbs.c:239
const char * name
Definition: avcodec.h:5809
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:192
static int h264_redundant_pps_fixup_pps(H264RedundantPPSContext *ctx, H264RawPPS *pps)
Coded bitstream unit structure.
Definition: cbs.h:64
void * content
Pointer to the decomposed form of this unit.
Definition: cbs.h:101
CodedBitstreamUnit * units
Pointer to an array of units of length nb_units_allocated.
Definition: cbs.h:162
#define av_log(a,...)
void ff_cbs_fragment_free(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag)
Free the units array of a fragment in addition to what ff_cbs_fragment_reset does.
Definition: cbs.c:154
H.264 common definitions.
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:260
AVCodecID
Identify the syntax and semantics of the bitstream.
Definition: avcodec.h:215
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
filter_frame For filters that do not use the this method is called when a frame is pushed to the filter s input It can be called at any time except in a reentrant way If the input frame is enough to produce then the filter should push the output frames on the output link immediately As an exception to the previous rule if the input frame is enough to produce several output frames then the filter needs output only at least one per link The additional frames can be left buffered in the filter
#define fail()
Definition: checkasm.h:120
int av_packet_copy_props(AVPacket *dst, const AVPacket *src)
Copy only "properties" fields from src to dst.
Definition: avpacket.c:565
int ff_cbs_write_extradata(CodedBitstreamContext *ctx, AVCodecParameters *par, CodedBitstreamFragment *frag)
Write the bitstream of a fragment to the extradata in codec parameters.
Definition: cbs.c:320
AVFormatContext * ctx
Definition: movenc.c:48
CodedBitstreamContext * output
int ff_cbs_delete_unit(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag, int position)
Delete a unit from a fragment and free all memory it uses.
Definition: cbs.c:739
static void h264_redundant_pps_flush(AVBSFContext *bsf)
static void h264_redundant_pps_close(AVBSFContext *bsf)
Coded bitstream fragment structure, combining one or more units.
Definition: cbs.h:116
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:599
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi-0x80)*(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi-0x80)*(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t,*(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t,*(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t,*(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t,*(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31))))#define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac){}void ff_audio_convert_free(AudioConvert **ac){if(!*ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);}AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map){AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method!=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2){ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc){av_free(ac);return NULL;}return ac;}in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar){ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar?ac->channels:1;}else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;}int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in){int use_generic=1;int len=in->nb_samples;int p;if(ac->dc){av_log(ac->avr, AV_LOG_TRACE,"%d samples - audio_convert: %s to %s (dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> in
void ff_cbs_fragment_reset(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag)
Free the units contained in a fragment as well as the fragment&#39;s own data buffer, but not the units a...
Definition: cbs.c:139
static int FUNC() pps(CodedBitstreamContext *ctx, RWContext *rw, H264RawPPS *current)
static int h264_redundant_pps_filter(AVBSFContext *bsf, AVPacket *out)
Context structure for coded bitstream operations.
Definition: cbs.h:168
void ff_cbs_close(CodedBitstreamContext **ctx_ptr)
Close a context and free all internal state.
Definition: cbs.c:113
int ff_cbs_read_extradata(CodedBitstreamContext *ctx, CodedBitstreamFragment *frag, const AVCodecParameters *par)
Read the extradata bitstream found in codec parameters into a fragment, then split into units and dec...
Definition: cbs.c:221
CodedBitstreamFragment access_unit
common internal and external API header
static enum AVCodecID codec_ids[]
int ff_bsf_get_packet(AVBSFContext *ctx, AVPacket **pkt)
Called by the bitstream filters to get the next packet for filtering.
Definition: bsf.c:216
static enum AVCodecID h264_redundant_pps_codec_ids[]
H264RawSliceHeader header
Definition: cbs_h264.h:430
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: avcodec.h:3967
int8_t pic_init_qp_minus26
Definition: cbs_h264.h:207
FILE * out
Definition: movenc.c:54
This structure stores compressed data.
Definition: avcodec.h:1454
AVCodecParameters * par_in
Parameters of the input stream.
Definition: avcodec.h:5787
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1470