mpegtsenc: convert to new channel layout API

Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
Anton Khirnov
2019-05-13 11:11:26 +02:00
committed by James Almer
parent 203a7bd785
commit 77869a3e43

View File

@ -466,7 +466,7 @@ static int get_m2ts_stream_type(AVFormatContext *s, AVStream *st)
stream_type = 0x81; stream_type = 0x81;
break; break;
case AV_CODEC_ID_DTS: case AV_CODEC_ID_DTS:
stream_type = (st->codecpar->channels > 6) ? 0x85 : 0x82; stream_type = (st->codecpar->ch_layout.nb_channels > 6) ? 0x85 : 0x82;
break; break;
case AV_CODEC_ID_TRUEHD: case AV_CODEC_ID_TRUEHD:
stream_type = 0x83; stream_type = 0x83;
@ -592,6 +592,8 @@ static int mpegts_write_pmt(AVFormatContext *s, MpegTSService *service)
if (codec_id == AV_CODEC_ID_S302M) if (codec_id == AV_CODEC_ID_S302M)
put_registration_descriptor(&q, MKTAG('B', 'S', 'S', 'D')); put_registration_descriptor(&q, MKTAG('B', 'S', 'S', 'D'));
if (codec_id == AV_CODEC_ID_OPUS) { if (codec_id == AV_CODEC_ID_OPUS) {
int ch = st->codecpar->ch_layout.nb_channels;
/* 6 bytes registration descriptor, 4 bytes Opus audio descriptor */ /* 6 bytes registration descriptor, 4 bytes Opus audio descriptor */
if (q - data > SECTION_LENGTH - 6 - 4) { if (q - data > SECTION_LENGTH - 6 - 4) {
err = 1; err = 1;
@ -605,11 +607,11 @@ static int mpegts_write_pmt(AVFormatContext *s, MpegTSService *service)
*q++ = 0x80; *q++ = 0x80;
if (st->codecpar->extradata && st->codecpar->extradata_size >= 19) { if (st->codecpar->extradata && st->codecpar->extradata_size >= 19) {
if (st->codecpar->extradata[18] == 0 && st->codecpar->channels <= 2) { if (st->codecpar->extradata[18] == 0 && ch <= 2) {
/* RTP mapping family */ /* RTP mapping family */
*q++ = st->codecpar->channels; *q++ = ch;
} else if (st->codecpar->extradata[18] == 1 && st->codecpar->channels <= 8 && } else if (st->codecpar->extradata[18] == 1 && ch <= 8 &&
st->codecpar->extradata_size >= 21 + st->codecpar->channels) { st->codecpar->extradata_size >= 21 + ch) {
static const uint8_t coupled_stream_counts[9] = { static const uint8_t coupled_stream_counts[9] = {
1, 0, 1, 1, 2, 2, 2, 3, 3 1, 0, 1, 1, 2, 2, 2, 3, 3
}; };
@ -635,14 +637,14 @@ static int mpegts_write_pmt(AVFormatContext *s, MpegTSService *service)
}; };
/* Vorbis mapping family */ /* Vorbis mapping family */
if (st->codecpar->extradata[19] == st->codecpar->channels - coupled_stream_counts[st->codecpar->channels] && if (st->codecpar->extradata[19] == ch - coupled_stream_counts[ch] &&
st->codecpar->extradata[20] == coupled_stream_counts[st->codecpar->channels] && st->codecpar->extradata[20] == coupled_stream_counts[ch] &&
memcmp(&st->codecpar->extradata[21], channel_map_a[st->codecpar->channels-1], st->codecpar->channels) == 0) { memcmp(&st->codecpar->extradata[21], channel_map_a[ch - 1], ch) == 0) {
*q++ = st->codecpar->channels; *q++ = ch;
} else if (st->codecpar->channels >= 2 && st->codecpar->extradata[19] == st->codecpar->channels && } else if (ch >= 2 && st->codecpar->extradata[19] == ch &&
st->codecpar->extradata[20] == 0 && st->codecpar->extradata[20] == 0 &&
memcmp(&st->codecpar->extradata[21], channel_map_b[st->codecpar->channels-1], st->codecpar->channels) == 0) { memcmp(&st->codecpar->extradata[21], channel_map_b[ch - 1], ch) == 0) {
*q++ = st->codecpar->channels | 0x80; *q++ = ch | 0x80;
} else { } else {
/* Unsupported, could write an extended descriptor here */ /* Unsupported, could write an extended descriptor here */
av_log(s, AV_LOG_ERROR, "Unsupported Opus Vorbis-style channel mapping"); av_log(s, AV_LOG_ERROR, "Unsupported Opus Vorbis-style channel mapping");
@ -653,9 +655,9 @@ static int mpegts_write_pmt(AVFormatContext *s, MpegTSService *service)
av_log(s, AV_LOG_ERROR, "Unsupported Opus channel mapping for family %d", st->codecpar->extradata[18]); av_log(s, AV_LOG_ERROR, "Unsupported Opus channel mapping for family %d", st->codecpar->extradata[18]);
*q++ = 0xff; *q++ = 0xff;
} }
} else if (st->codecpar->channels <= 2) { } else if (ch <= 2) {
/* Assume RTP mapping family */ /* Assume RTP mapping family */
*q++ = st->codecpar->channels; *q++ = ch;
} else { } else {
/* Unsupported */ /* Unsupported */
av_log(s, AV_LOG_ERROR, "Unsupported Opus channel mapping"); av_log(s, AV_LOG_ERROR, "Unsupported Opus channel mapping");