From b942a71eaa09fa2d54f08b2d3c3e9c01b615bf56 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Tue, 20 Jan 2015 23:55:09 +0100 Subject: [PATCH] avformat/flvenc: accept AVMEDIA_TYPE_SUBTITLE instead of DATA for subtitles Reviewed-by: Paul B Mahol Signed-off-by: Michael Niedermayer --- libavformat/flvenc.c | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/libavformat/flvenc.c b/libavformat/flvenc.c index 5e27ed0912..7d3a0ca99f 100644 --- a/libavformat/flvenc.c +++ b/libavformat/flvenc.c @@ -389,6 +389,14 @@ static int flv_write_header(AVFormatContext *s) } flv->data_enc = enc; break; + case AVMEDIA_TYPE_SUBTITLE: + if (enc->codec_id != AV_CODEC_ID_TEXT) { + av_log(s, AV_LOG_ERROR, "Subtitle codec '%s' for stream %d is not compatible with FLV\n", + avcodec_get_name(enc->codec_id), i); + return AVERROR_INVALIDDATA; + } + flv->data_enc = enc; + break; default: av_log(s, AV_LOG_ERROR, "Codec type '%s' for stream %d is not compatible with FLV\n", av_get_media_type_string(enc->codec_type), i); @@ -545,6 +553,7 @@ static int flv_write_packet(AVFormatContext *s, AVPacket *pkt) avio_w8(pb, FLV_TAG_TYPE_AUDIO); break; + case AVMEDIA_TYPE_SUBTITLE: case AVMEDIA_TYPE_DATA: avio_w8(pb, FLV_TAG_TYPE_META); break; @@ -588,7 +597,8 @@ static int flv_write_packet(AVFormatContext *s, AVPacket *pkt) avio_w8(pb, (ts >> 24) & 0x7F); // timestamps are 32 bits _signed_ avio_wb24(pb, flv->reserved); - if (enc->codec_type == AVMEDIA_TYPE_DATA) { + if (enc->codec_type == AVMEDIA_TYPE_DATA || + enc->codec_type == AVMEDIA_TYPE_SUBTITLE ) { int data_size; int64_t metadata_size_pos = avio_tell(pb); if (enc->codec_id == AV_CODEC_ID_TEXT) {