| 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 |