FFmpeg
adtsenc.c
Go to the documentation of this file.
1 /*
2  * ADTS muxer.
3  * Copyright (c) 2006 Baptiste Coudurier <baptiste.coudurier@smartjog.com>
4  * Mans Rullgard <mans@mansr.com>
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #include "libavcodec/get_bits.h"
24 #include "libavcodec/put_bits.h"
25 #include "libavcodec/avcodec.h"
26 #include "libavcodec/mpeg4audio.h"
27 #include "libavutil/opt.h"
28 #include "avformat.h"
29 #include "apetag.h"
30 #include "id3v2.h"
31 
32 #define ADTS_HEADER_SIZE 7
33 
34 typedef struct ADTSContext {
35  AVClass *class;
40  int pce_size;
41  int apetag;
42  int id3v2tag;
43  int mpeg_id;
45 } ADTSContext;
46 
47 #define ADTS_MAX_FRAME_BYTES ((1 << 13) - 1)
48 
49 static int adts_decode_extradata(AVFormatContext *s, ADTSContext *adts, const uint8_t *buf, int size)
50 {
51  GetBitContext gb;
52  PutBitContext pb;
53  MPEG4AudioConfig m4ac;
54  int off;
55 
56  init_get_bits(&gb, buf, size * 8);
57  off = avpriv_mpeg4audio_get_config2(&m4ac, buf, size, 1, s);
58  if (off < 0)
59  return off;
60  skip_bits_long(&gb, off);
61  adts->objecttype = m4ac.object_type - 1;
62  adts->sample_rate_index = m4ac.sampling_index;
63  adts->channel_conf = m4ac.chan_config;
64 
65  if (adts->objecttype > 3U) {
66  av_log(s, AV_LOG_ERROR, "MPEG-4 AOT %d is not allowed in ADTS\n", adts->objecttype+1);
67  return AVERROR_INVALIDDATA;
68  }
69  if (adts->sample_rate_index == 15) {
70  av_log(s, AV_LOG_ERROR, "Escape sample rate index illegal in ADTS\n");
71  return AVERROR_INVALIDDATA;
72  }
73  if (get_bits(&gb, 1)) {
74  av_log(s, AV_LOG_ERROR, "960/120 MDCT window is not allowed in ADTS\n");
75  return AVERROR_INVALIDDATA;
76  }
77  if (get_bits(&gb, 1)) {
78  av_log(s, AV_LOG_ERROR, "Scalable configurations are not allowed in ADTS\n");
79  return AVERROR_INVALIDDATA;
80  }
81  if (get_bits(&gb, 1)) {
82  av_log(s, AV_LOG_ERROR, "Extension flag is not allowed in ADTS\n");
83  return AVERROR_INVALIDDATA;
84  }
85  if (!adts->channel_conf) {
86  init_put_bits(&pb, adts->pce_data, MAX_PCE_SIZE);
87 
88  put_bits(&pb, 3, 5); //ID_PCE
89  adts->pce_size = (ff_copy_pce_data(&pb, &gb) + 3) / 8;
90  flush_put_bits(&pb);
91  }
92 
93  adts->write_adts = 1;
94 
95  return 0;
96 }
97 
99 {
100  ADTSContext *adts = s->priv_data;
101  AVCodecParameters *par = s->streams[0]->codecpar;
102 
103  if (par->codec_id != AV_CODEC_ID_AAC) {
104  av_log(s, AV_LOG_ERROR, "Only AAC streams can be muxed by the ADTS muxer\n");
105  return AVERROR(EINVAL);
106  }
107  if (par->extradata_size > 0)
108  return adts_decode_extradata(s, adts, par->extradata,
109  par->extradata_size);
110 
111  return 0;
112 }
113 
115 {
116  ADTSContext *adts = s->priv_data;
117 
118  if (adts->id3v2tag)
120 
121  return 0;
122 }
123 
125  uint8_t *buf, int size, int pce_size)
126 {
127  PutBitContext pb;
128 
129  unsigned full_frame_size = (unsigned)ADTS_HEADER_SIZE + size + pce_size;
130  if (full_frame_size > ADTS_MAX_FRAME_BYTES) {
131  av_log(NULL, AV_LOG_ERROR, "ADTS frame size too large: %u (max %d)\n",
132  full_frame_size, ADTS_MAX_FRAME_BYTES);
133  return AVERROR_INVALIDDATA;
134  }
135 
136  init_put_bits(&pb, buf, ADTS_HEADER_SIZE);
137 
138  /* adts_fixed_header */
139  put_bits(&pb, 12, 0xfff); /* syncword */
140  put_bits(&pb, 1, ctx->mpeg_id); /* ID */
141  put_bits(&pb, 2, 0); /* layer */
142  put_bits(&pb, 1, 1); /* protection_absent */
143  put_bits(&pb, 2, ctx->objecttype); /* profile_objecttype */
144  put_bits(&pb, 4, ctx->sample_rate_index);
145  put_bits(&pb, 1, 0); /* private_bit */
146  put_bits(&pb, 3, ctx->channel_conf); /* channel_configuration */
147  put_bits(&pb, 1, 0); /* original_copy */
148  put_bits(&pb, 1, 0); /* home */
149 
150  /* adts_variable_header */
151  put_bits(&pb, 1, 0); /* copyright_identification_bit */
152  put_bits(&pb, 1, 0); /* copyright_identification_start */
153  put_bits(&pb, 13, full_frame_size); /* aac_frame_length */
154  put_bits(&pb, 11, 0x7ff); /* adts_buffer_fullness */
155  put_bits(&pb, 2, 0); /* number_of_raw_data_blocks_in_frame */
156 
157  flush_put_bits(&pb);
158 
159  return 0;
160 }
161 
163 {
164  ADTSContext *adts = s->priv_data;
165  AVCodecParameters *par = s->streams[0]->codecpar;
166  AVIOContext *pb = s->pb;
168 
169  if (!pkt->size)
170  return 0;
171  if (!par->extradata_size) {
172  uint8_t *side_data;
173  int side_data_size, ret;
174 
176  &side_data_size);
177  if (side_data_size) {
178  ret = adts_decode_extradata(s, adts, side_data, side_data_size);
179  if (ret < 0)
180  return ret;
181  ret = ff_alloc_extradata(par, side_data_size);
182  if (ret < 0)
183  return ret;
184  memcpy(par->extradata, side_data, side_data_size);
185  }
186  }
187  if (adts->write_adts) {
188  int err = adts_write_frame_header(adts, buf, pkt->size,
189  adts->pce_size);
190  if (err < 0)
191  return err;
192  avio_write(pb, buf, ADTS_HEADER_SIZE);
193  if (adts->pce_size) {
194  avio_write(pb, adts->pce_data, adts->pce_size);
195  adts->pce_size = 0;
196  }
197  }
198  avio_write(pb, pkt->data, pkt->size);
199 
200  return 0;
201 }
202 
204 {
205  ADTSContext *adts = s->priv_data;
206 
207  if (adts->apetag)
208  ff_ape_write_tag(s);
209 
210  return 0;
211 }
212 
213 #define ENC AV_OPT_FLAG_ENCODING_PARAM
214 #define OFFSET(obj) offsetof(ADTSContext, obj)
215 static const AVOption options[] = {
216  { "write_id3v2", "Enable ID3v2 tag writing", OFFSET(id3v2tag), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, ENC},
217  { "write_apetag", "Enable APE tag writing", OFFSET(apetag), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, ENC},
218  { "write_mpeg2", "Use MPE2 ID when writing", OFFSET(mpeg_id), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, ENC, "mpeg_id"},
219  { NULL },
220 };
221 
222 static const AVClass adts_muxer_class = {
223  .class_name = "ADTS muxer",
224  .item_name = av_default_item_name,
225  .option = options,
226  .version = LIBAVUTIL_VERSION_INT,
227 };
228 
230  .name = "adts",
231  .long_name = NULL_IF_CONFIG_SMALL("ADTS AAC (Advanced Audio Coding)"),
232  .mime_type = "audio/aac",
233  .extensions = "aac,adts",
234  .priv_data_size = sizeof(ADTSContext),
235  .audio_codec = AV_CODEC_ID_AAC,
236  .video_codec = AV_CODEC_ID_NONE,
237  .init = adts_init,
241  .priv_class = &adts_muxer_class,
243 };
uint8_t pce_data[MAX_PCE_SIZE]
Definition: adtsenc.c:44
static void write_packet(OutputFile *of, AVPacket *pkt, OutputStream *ost, int unqueue)
Definition: ffmpeg.c:702
#define NULL
Definition: coverity.c:32
int id3v2tag
Definition: adtsenc.c:42
Bytestream IO Context.
Definition: avio.h:161
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
AVOption.
Definition: opt.h:248
int ff_id3v2_write_simple(struct AVFormatContext *s, int id3v2_version, const char *magic)
Write an ID3v2 tag containing all global metadata from s.
Definition: id3v2enc.c:444
static void put_bits(Jpeg2000EncoderContext *s, int val, int n)
put n times val bit
Definition: j2kenc.c:218
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:379
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
static void skip_bits_long(GetBitContext *s, int n)
Skips the specified number of bits.
Definition: get_bits.h:291
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
static int adts_init(AVFormatContext *s)
Definition: adtsenc.c:98
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:60
int size
Definition: packet.h:364
#define ID3v2_DEFAULT_MAGIC
Default magic bytes for ID3v2 header: "ID3".
Definition: id3v2.h:35
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:235
static int adts_decode_extradata(AVFormatContext *s, ADTSContext *adts, const uint8_t *buf, int size)
Definition: adtsenc.c:49
int objecttype
Definition: adtsenc.c:37
#define ADTS_MAX_FRAME_BYTES
Definition: adtsenc.c:47
static AVPacket pkt
This struct describes the properties of an encoded stream.
Definition: codec_par.h:52
static int adts_write_packet(AVFormatContext *s, AVPacket *pkt)
Definition: adtsenc.c:162
Format I/O context.
Definition: avformat.h:1351
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
uint8_t
int avpriv_mpeg4audio_get_config2(MPEG4AudioConfig *c, const uint8_t *buf, int size, int sync_extension, void *logctx)
Parse MPEG-4 systems extradata from a raw buffer to retrieve audio configuration. ...
Definition: mpeg4audio.c:190
static const AVOption options[]
Definition: adtsenc.c:215
AVOptions.
static int adts_write_header(AVFormatContext *s)
Definition: adtsenc.c:114
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1419
uint8_t * data
Definition: packet.h:363
#define ADTS_HEADER_SIZE
Definition: adtsenc.c:32
bitstream reader API header.
ptrdiff_t size
Definition: opengl_enc.c:100
void avio_write(AVIOContext *s, const unsigned char *buf, int size)
Definition: aviobuf.c:225
#define av_log(a,...)
#define U(x)
Definition: vp56_arith.h:37
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
uint8_t * av_packet_get_side_data(const AVPacket *pkt, enum AVPacketSideDataType type, int *size)
Get side information from packet.
Definition: avpacket.c:353
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:153
int channel_conf
Definition: adtsenc.c:39
int write_adts
Definition: adtsenc.c:36
int sample_rate_index
Definition: adtsenc.c:38
int extradata_size
Size of the extradata content in bytes.
Definition: codec_par.h:78
#define ENC
Definition: adtsenc.c:213
int ff_alloc_extradata(AVCodecParameters *par, int size)
Allocate extradata with additional AV_INPUT_BUFFER_PADDING_SIZE at end which is always set to 0...
Definition: utils.c:3259
int mpeg_id
Definition: adtsenc.c:43
static int write_trailer(AVFormatContext *s1)
Definition: v4l2enc.c:98
const char * name
Definition: avformat.h:500
AVFormatContext * ctx
Definition: movenc.c:48
#define s(width, name)
Definition: cbs_vp9.c:257
int pce_size
Definition: adtsenc.c:40
#define AVFMT_NOTIMESTAMPS
Format does not need / have any timestamps.
Definition: avformat.h:462
The AV_PKT_DATA_NEW_EXTRADATA is used to notify the codec or the format that the extradata buffer was...
Definition: packet.h:55
Libavcodec external API header.
AVIOContext * pb
I/O context.
Definition: avformat.h:1393
int ff_ape_write_tag(AVFormatContext *s)
Write an APE tag into a file.
Definition: apetag.c:185
Describe the class of an AVClass context structure.
Definition: log.h:67
static int ff_copy_pce_data(PutBitContext *pb, GetBitContext *gb)
Definition: mpeg4audio.h:146
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:659
int apetag
Definition: adtsenc.c:41
#define flags(name, subs,...)
Definition: cbs_av1.c:560
static const AVClass adts_muxer_class
Definition: adtsenc.c:222
Main libavformat public API header.
static void flush_put_bits(PutBitContext *s)
Pad the end of the output stream with zeros.
Definition: put_bits.h:115
static int adts_write_frame_header(ADTSContext *ctx, uint8_t *buf, int size, int pce_size)
Definition: adtsenc.c:124
static void init_put_bits(PutBitContext *s, uint8_t *buffer, int buffer_size)
Initialize the PutBitContext s.
Definition: put_bits.h:62
#define MAX_PCE_SIZE
Maximum size of a PCE including the 3-bit ID_PCE.
Definition: mpeg4audio.h:134
void * priv_data
Format private data.
Definition: avformat.h:1379
static void write_header(FFV1Context *f)
Definition: ffv1enc.c:346
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: codec_par.h:74
#define OFFSET(obj)
Definition: adtsenc.c:214
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1023
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later.That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another.Frame references ownership and permissions
This structure stores compressed data.
Definition: packet.h:340
AVOutputFormat ff_adts_muxer
Definition: adtsenc.c:229
static int adts_write_trailer(AVFormatContext *s)
Definition: adtsenc.c:203
bitstream writer API