OLD | NEW |
1 /* | 1 /* |
2 * ADTS muxer. | 2 * ADTS muxer. |
3 * Copyright (c) 2006 Baptiste Coudurier <baptiste.coudurier@smartjog.com> | 3 * Copyright (c) 2006 Baptiste Coudurier <baptiste.coudurier@smartjog.com> |
4 * Mans Rullgard <mans@mansr.com> | 4 * Mans Rullgard <mans@mansr.com> |
5 * | 5 * |
6 * This file is part of FFmpeg. | 6 * This file is part of FFmpeg. |
7 * | 7 * |
8 * FFmpeg is free software; you can redistribute it and/or | 8 * FFmpeg is free software; you can redistribute it and/or |
9 * modify it under the terms of the GNU Lesser General Public | 9 * modify it under the terms of the GNU Lesser General Public |
10 * License as published by the Free Software Foundation; either | 10 * License as published by the Free Software Foundation; either |
11 * version 2.1 of the License, or (at your option) any later version. | 11 * version 2.1 of the License, or (at your option) any later version. |
12 * | 12 * |
13 * FFmpeg is distributed in the hope that it will be useful, | 13 * FFmpeg is distributed in the hope that it will be useful, |
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of | 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
16 * Lesser General Public License for more details. | 16 * Lesser General Public License for more details. |
17 * | 17 * |
18 * You should have received a copy of the GNU Lesser General Public | 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 | 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 | 20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
21 */ | 21 */ |
22 | 22 |
23 #include "libavcodec/get_bits.h" | 23 #include "libavcodec/get_bits.h" |
24 #include "libavcodec/put_bits.h" | 24 #include "libavcodec/put_bits.h" |
25 #include "libavcodec/avcodec.h" | 25 #include "libavcodec/avcodec.h" |
26 #include "libavcodec/mpeg4audio.h" | |
27 #include "avformat.h" | 26 #include "avformat.h" |
| 27 #include "adts.h" |
28 | 28 |
29 #define ADTS_HEADER_SIZE 7 | 29 int ff_adts_decode_extradata(AVFormatContext *s, ADTSContext *adts, uint8_t *buf
, int size) |
30 | |
31 typedef struct { | |
32 int write_adts; | |
33 int objecttype; | |
34 int sample_rate_index; | |
35 int channel_conf; | |
36 int pce_size; | |
37 uint8_t pce_data[MAX_PCE_SIZE]; | |
38 } ADTSContext; | |
39 | |
40 static int decode_extradata(AVFormatContext *s, ADTSContext *adts, uint8_t *buf,
int size) | |
41 { | 30 { |
42 GetBitContext gb; | 31 GetBitContext gb; |
43 PutBitContext pb; | 32 PutBitContext pb; |
44 | 33 |
45 init_get_bits(&gb, buf, size * 8); | 34 init_get_bits(&gb, buf, size * 8); |
46 adts->objecttype = get_bits(&gb, 5) - 1; | 35 adts->objecttype = get_bits(&gb, 5) - 1; |
47 adts->sample_rate_index = get_bits(&gb, 4); | 36 adts->sample_rate_index = get_bits(&gb, 4); |
48 adts->channel_conf = get_bits(&gb, 4); | 37 adts->channel_conf = get_bits(&gb, 4); |
49 | 38 |
50 if (adts->objecttype > 3U) { | 39 if (adts->objecttype > 3U) { |
(...skipping 28 matching lines...) Expand all Loading... |
79 | 68 |
80 return 0; | 69 return 0; |
81 } | 70 } |
82 | 71 |
83 static int adts_write_header(AVFormatContext *s) | 72 static int adts_write_header(AVFormatContext *s) |
84 { | 73 { |
85 ADTSContext *adts = s->priv_data; | 74 ADTSContext *adts = s->priv_data; |
86 AVCodecContext *avc = s->streams[0]->codec; | 75 AVCodecContext *avc = s->streams[0]->codec; |
87 | 76 |
88 if(avc->extradata_size > 0 && | 77 if(avc->extradata_size > 0 && |
89 decode_extradata(s, adts, avc->extradata, avc->extradata_size) < 0) | 78 ff_adts_decode_extradata(s, adts, avc->extradata, avc->extradata_siz
e) < 0) |
90 return -1; | 79 return -1; |
91 | 80 |
92 return 0; | 81 return 0; |
93 } | 82 } |
94 | 83 |
95 static int adts_write_frame_header(AVFormatContext *s, int size) | 84 int ff_adts_write_frame_header(ADTSContext *ctx, |
| 85 uint8_t *buf, int size, int pce_size) |
96 { | 86 { |
97 ADTSContext *ctx = s->priv_data; | |
98 PutBitContext pb; | 87 PutBitContext pb; |
99 uint8_t buf[ADTS_HEADER_SIZE]; | |
100 | 88 |
101 init_put_bits(&pb, buf, ADTS_HEADER_SIZE); | 89 init_put_bits(&pb, buf, ADTS_HEADER_SIZE); |
102 | 90 |
103 /* adts_fixed_header */ | 91 /* adts_fixed_header */ |
104 put_bits(&pb, 12, 0xfff); /* syncword */ | 92 put_bits(&pb, 12, 0xfff); /* syncword */ |
105 put_bits(&pb, 1, 0); /* ID */ | 93 put_bits(&pb, 1, 0); /* ID */ |
106 put_bits(&pb, 2, 0); /* layer */ | 94 put_bits(&pb, 2, 0); /* layer */ |
107 put_bits(&pb, 1, 1); /* protection_absent */ | 95 put_bits(&pb, 1, 1); /* protection_absent */ |
108 put_bits(&pb, 2, ctx->objecttype); /* profile_objecttype */ | 96 put_bits(&pb, 2, ctx->objecttype); /* profile_objecttype */ |
109 put_bits(&pb, 4, ctx->sample_rate_index); | 97 put_bits(&pb, 4, ctx->sample_rate_index); |
110 put_bits(&pb, 1, 0); /* private_bit */ | 98 put_bits(&pb, 1, 0); /* private_bit */ |
111 put_bits(&pb, 3, ctx->channel_conf); /* channel_configuration */ | 99 put_bits(&pb, 3, ctx->channel_conf); /* channel_configuration */ |
112 put_bits(&pb, 1, 0); /* original_copy */ | 100 put_bits(&pb, 1, 0); /* original_copy */ |
113 put_bits(&pb, 1, 0); /* home */ | 101 put_bits(&pb, 1, 0); /* home */ |
114 | 102 |
115 /* adts_variable_header */ | 103 /* adts_variable_header */ |
116 put_bits(&pb, 1, 0); /* copyright_identification_bit */ | 104 put_bits(&pb, 1, 0); /* copyright_identification_bit */ |
117 put_bits(&pb, 1, 0); /* copyright_identification_start */ | 105 put_bits(&pb, 1, 0); /* copyright_identification_start */ |
118 put_bits(&pb, 13, ADTS_HEADER_SIZE + size + ctx->pce_size); /* aac_frame_len
gth */ | 106 put_bits(&pb, 13, ADTS_HEADER_SIZE + size + pce_size); /* aac_frame_length *
/ |
119 put_bits(&pb, 11, 0x7ff); /* adts_buffer_fullness */ | 107 put_bits(&pb, 11, 0x7ff); /* adts_buffer_fullness */ |
120 put_bits(&pb, 2, 0); /* number_of_raw_data_blocks_in_frame */ | 108 put_bits(&pb, 2, 0); /* number_of_raw_data_blocks_in_frame */ |
121 | 109 |
122 flush_put_bits(&pb); | 110 flush_put_bits(&pb); |
123 put_buffer(s->pb, buf, ADTS_HEADER_SIZE); | |
124 if (ctx->pce_size) { | |
125 put_buffer(s->pb, ctx->pce_data, ctx->pce_size); | |
126 ctx->pce_size = 0; | |
127 } | |
128 | 111 |
129 return 0; | 112 return 0; |
130 } | 113 } |
131 | 114 |
132 static int adts_write_packet(AVFormatContext *s, AVPacket *pkt) | 115 static int adts_write_packet(AVFormatContext *s, AVPacket *pkt) |
133 { | 116 { |
134 ADTSContext *adts = s->priv_data; | 117 ADTSContext *adts = s->priv_data; |
135 ByteIOContext *pb = s->pb; | 118 ByteIOContext *pb = s->pb; |
| 119 uint8_t buf[ADTS_HEADER_SIZE]; |
136 | 120 |
137 if (!pkt->size) | 121 if (!pkt->size) |
138 return 0; | 122 return 0; |
139 if(adts->write_adts) | 123 if(adts->write_adts) { |
140 adts_write_frame_header(s, pkt->size); | 124 ff_adts_write_frame_header(adts, buf, pkt->size, adts->pce_size); |
| 125 put_buffer(pb, buf, ADTS_HEADER_SIZE); |
| 126 if(adts->pce_size) { |
| 127 put_buffer(pb, adts->pce_data, adts->pce_size); |
| 128 adts->pce_size = 0; |
| 129 } |
| 130 } |
141 put_buffer(pb, pkt->data, pkt->size); | 131 put_buffer(pb, pkt->data, pkt->size); |
142 put_flush_packet(pb); | 132 put_flush_packet(pb); |
143 | 133 |
144 return 0; | 134 return 0; |
145 } | 135 } |
146 | 136 |
147 AVOutputFormat adts_muxer = { | 137 AVOutputFormat adts_muxer = { |
148 "adts", | 138 "adts", |
149 NULL_IF_CONFIG_SMALL("ADTS AAC"), | 139 NULL_IF_CONFIG_SMALL("ADTS AAC"), |
150 "audio/aac", | 140 "audio/aac", |
151 "aac,adts", | 141 "aac,adts", |
152 sizeof(ADTSContext), | 142 sizeof(ADTSContext), |
153 CODEC_ID_AAC, | 143 CODEC_ID_AAC, |
154 CODEC_ID_NONE, | 144 CODEC_ID_NONE, |
155 adts_write_header, | 145 adts_write_header, |
156 adts_write_packet, | 146 adts_write_packet, |
157 }; | 147 }; |
OLD | NEW |