FFmpeg
mpeg2_metadata_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 "libavutil/avstring.h"
20 #include "libavutil/common.h"
21 #include "libavutil/opt.h"
22 
23 #include "bsf.h"
24 #include "cbs.h"
25 #include "cbs_bsf.h"
26 #include "cbs_mpeg2.h"
27 #include "mpeg12.h"
28 
29 typedef struct MPEG2MetadataContext {
31 
33 
35 
37 
42 
45 
46 
48  AVPacket *pkt,
50 {
55  int i, se_pos;
56 
57  for (i = 0; i < frag->nb_units; i++) {
58  if (frag->units[i].type == MPEG2_START_SEQUENCE_HEADER) {
59  sh = frag->units[i].content;
60  } else if (frag->units[i].type == MPEG2_START_EXTENSION) {
61  MPEG2RawExtensionData *ext = frag->units[i].content;
64  se = &ext->data.sequence;
65  se_pos = i;
66  } else if (ext->extension_start_code_identifier ==
68  sde = &ext->data.sequence_display;
69  }
70  }
71  }
72 
73  if (!sh || !se) {
74  // No sequence header and sequence extension: not an MPEG-2 video
75  // sequence.
76  if (sh && !ctx->mpeg1_warned) {
77  av_log(bsf, AV_LOG_WARNING, "Stream contains a sequence "
78  "header but not a sequence extension: maybe it's "
79  "actually MPEG-1?\n");
80  ctx->mpeg1_warned = 1;
81  }
82  return 0;
83  }
84 
86  int num, den;
87 
88  av_reduce(&num, &den, ctx->display_aspect_ratio.num,
89  ctx->display_aspect_ratio.den, 65535);
90 
91  if (num == 4 && den == 3)
93  else if (num == 16 && den == 9)
95  else if (num == 221 && den == 100)
97  else
99  }
100 
101  if (ctx->frame_rate.num && ctx->frame_rate.den) {
102  int code, ext_n, ext_d;
103 
105  &code, &ext_n, &ext_d, 0);
106 
107  sh->frame_rate_code = code;
108  se->frame_rate_extension_n = ext_n;
109  se->frame_rate_extension_d = ext_d;
110  }
111 
112  if (ctx->video_format >= 0 ||
113  ctx->colour_primaries >= 0 ||
114  ctx->transfer_characteristics >= 0 ||
115  ctx->matrix_coefficients >= 0) {
116  if (!sde) {
117  int err;
123 
125  .video_format = 5,
126 
127  .colour_description = 0,
128  .colour_primaries = 2,
129  .transfer_characteristics = 2,
130  .matrix_coefficients = 2,
131 
132  .display_horizontal_size =
134  .display_vertical_size =
136  };
137 
138  err = ff_cbs_insert_unit_content(frag, se_pos + 1,
141  NULL);
142  if (err < 0) {
143  av_log(bsf, AV_LOG_ERROR, "Failed to insert new sequence "
144  "display extension.\n");
145  return err;
146  }
147  }
148 
149  if (ctx->video_format >= 0)
150  sde->video_format = ctx->video_format;
151 
152  if (ctx->colour_primaries >= 0 ||
153  ctx->transfer_characteristics >= 0 ||
154  ctx->matrix_coefficients >= 0) {
155  sde->colour_description = 1;
156 
157  if (ctx->colour_primaries >= 0)
159 
160  if (ctx->transfer_characteristics >= 0)
162 
163  if (ctx->matrix_coefficients >= 0)
165  }
166  }
167 
168  return 0;
169 }
170 
173  .fragment_name = "frame",
174  .unit_name = "start code",
175  .update_fragment = &mpeg2_metadata_update_fragment,
176 };
177 
179 {
181 
182 #define VALIDITY_CHECK(name) do { \
183  if (!ctx->name) { \
184  av_log(bsf, AV_LOG_ERROR, "The value 0 for %s is " \
185  "forbidden.\n", #name); \
186  return AVERROR(EINVAL); \
187  } \
188  } while (0)
192 #undef VALIDITY_CHECK
193 
194  return ff_cbs_bsf_generic_init(bsf, &mpeg2_metadata_type);
195 }
196 
197 #define OFFSET(x) offsetof(MPEG2MetadataContext, x)
198 #define FLAGS (AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_BSF_PARAM)
200  { "display_aspect_ratio", "Set display aspect ratio (table 6-3)",
202  { .dbl = 0.0 }, 0, 65535, FLAGS },
203 
204  { "frame_rate", "Set frame rate",
206  { .dbl = 0.0 }, 0, UINT_MAX, FLAGS },
207 
208  { "video_format", "Set video format (table 6-6)",
210  { .i64 = -1 }, -1, 7, FLAGS },
211  { "colour_primaries", "Set colour primaries (table 6-7)",
213  { .i64 = -1 }, -1, 255, FLAGS },
214  { "transfer_characteristics", "Set transfer characteristics (table 6-8)",
216  { .i64 = -1 }, -1, 255, FLAGS },
217  { "matrix_coefficients", "Set matrix coefficients (table 6-9)",
219  { .i64 = -1 }, -1, 255, FLAGS },
220 
221  { NULL }
222 };
223 
225  .class_name = "mpeg2_metadata_bsf",
226  .item_name = av_default_item_name,
227  .option = mpeg2_metadata_options,
228  .version = LIBAVUTIL_VERSION_INT,
229 };
230 
231 static const enum AVCodecID mpeg2_metadata_codec_ids[] = {
233 };
234 
236  .name = "mpeg2_metadata",
237  .priv_data_size = sizeof(MPEG2MetadataContext),
238  .priv_class = &mpeg2_metadata_class,
240  .close = &ff_cbs_bsf_generic_close,
243 };
#define NULL
Definition: coverity.c:32
int nb_units
Number of units in this fragment.
Definition: cbs.h:149
uint8_t extension_start_code_identifier
Definition: cbs_mpeg2.h:179
#define se(name, range_min, range_max)
Definition: cbs_h2645.c:275
AVOption.
Definition: opt.h:248
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:200
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
MPEG2RawSequenceExtension sequence
Definition: cbs_mpeg2.h:182
enum AVCodecID codec_id
Definition: cbs_bsf.h:26
CodedBitstreamUnitType type
Codec-specific type of this unit.
Definition: cbs.h:70
int num
Numerator.
Definition: rational.h:59
The bitstream filter state.
Definition: bsf.h:49
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:235
static AVPacket pkt
void * priv_data
Opaque filter-specific private data.
Definition: bsf.h:70
static const CBSBSFType mpeg2_metadata_type
int ff_cbs_bsf_generic_init(AVBSFContext *bsf, const CBSBSFType *type)
Initialise generic CBS BSF setup.
Definition: cbs_bsf.c:112
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:72
uint16_t vertical_size_value
Definition: cbs_mpeg2.h:63
AVOptions.
union MPEG2RawExtensionData::@37 data
#define VALIDITY_CHECK(name)
const char * name
Definition: bsf.h:99
int av_reduce(int *dst_num, int *dst_den, int64_t num, int64_t den, int64_t max)
Reduce a fraction.
Definition: rational.c:35
void * content
Pointer to the decomposed form of this unit.
Definition: cbs.h:103
CodedBitstreamUnit * units
Pointer to an array of units of length nb_units_allocated.
Definition: cbs.h:164
#define av_log(a,...)
AVCodecID
Identify the syntax and semantics of the bitstream.
Definition: codec_id.h:46
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
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
void ff_cbs_bsf_generic_close(AVBSFContext *bsf)
Close a generic CBS BSF instance.
Definition: cbs_bsf.c:152
void ff_mpeg12_find_best_frame_rate(AVRational frame_rate, int *code, int *ext_n, int *ext_d, int nonstandard)
const AVBitStreamFilter ff_mpeg2_metadata_bsf
static int mpeg2_metadata_update_fragment(AVBSFContext *bsf, AVPacket *pkt, CodedBitstreamFragment *frag)
MPEG2RawExtensionData sequence_display_extension
AVFormatContext * ctx
Definition: movenc.c:48
int ff_cbs_bsf_generic_filter(AVBSFContext *bsf, AVPacket *pkt)
Filter operation for CBS BSF.
Definition: cbs_bsf.c:63
uint8_t horizontal_size_extension
Definition: cbs_mpeg2.h:88
preferred ID for MPEG-1/2 video decoding
Definition: codec_id.h:51
MPEG2RawSequenceDisplayExtension sequence_display
Definition: cbs_mpeg2.h:183
Coded bitstream fragment structure, combining one or more units.
Definition: cbs.h:118
#define FLAGS
static const AVClass mpeg2_metadata_class
uint8_t frame_rate_extension_n
Definition: cbs_mpeg2.h:93
uint8_t frame_rate_extension_d
Definition: cbs_mpeg2.h:94
static enum AVCodecID mpeg2_metadata_codec_ids[]
Describe the class of an AVClass context structure.
Definition: log.h:67
Rational number (pair of numerator and denominator).
Definition: rational.h:58
uint8_t extension_start_code
Definition: cbs_mpeg2.h:178
uint16_t horizontal_size_value
Definition: cbs_mpeg2.h:62
int ff_cbs_insert_unit_content(CodedBitstreamFragment *frag, int position, CodedBitstreamUnitType type, void *content, AVBufferRef *content_buf)
Insert a new unit into a fragment with the given content.
Definition: cbs.c:737
uint8_t aspect_ratio_information
Definition: cbs_mpeg2.h:64
and forward the test the status of outputs and forward it to the corresponding return FFERROR_NOT_READY If the filters stores internally one or a few frame for some it can consider them to be part of the FIFO and delay acknowledging a status change accordingly Example code
common internal and external API header
static enum AVCodecID codec_ids[]
int den
Denominator.
Definition: rational.h:60
static const AVOption mpeg2_metadata_options[]
uint8_t vertical_size_extension
Definition: cbs_mpeg2.h:89
#define OFFSET(x)
static int mpeg2_metadata_init(AVBSFContext *bsf)
This structure stores compressed data.
Definition: packet.h:340
int i
Definition: input.c:407