diff options
| -rw-r--r-- | libavformat/hlsenc.c | 12 | ||||
| -rw-r--r-- | libavformat/segment.c | 12 |
2 files changed, 14 insertions, 10 deletions
diff --git a/libavformat/hlsenc.c b/libavformat/hlsenc.c index 89980d7162..caf878ffd4 100644 --- a/libavformat/hlsenc.c +++ b/libavformat/hlsenc.c @@ -250,18 +250,20 @@ static int hls_write_packet(AVFormatContext *s, AVPacket *pkt) AVFormatContext *oc = hls->avf; AVStream *st = s->streams[pkt->stream_index]; int64_t end_pts = hls->recording_time * hls->number; - int ret; + int ret, can_split = 1; if (hls->start_pts == AV_NOPTS_VALUE) { hls->start_pts = pkt->pts; hls->end_pts = pkt->pts; } - if ((hls->has_video && st->codec->codec_type == AVMEDIA_TYPE_VIDEO) && - av_compare_ts(pkt->pts - hls->start_pts, st->time_base, - end_pts, AV_TIME_BASE_Q) >= 0 && - pkt->flags & AV_PKT_FLAG_KEY) { + if (hls->has_video) { + can_split = st->codec->codec_type == AVMEDIA_TYPE_VIDEO && + pkt->flags & AV_PKT_FLAG_KEY; + } + if (can_split && av_compare_ts(pkt->pts - hls->start_pts, st->time_base, + end_pts, AV_TIME_BASE_Q) >= 0) { ret = append_entry(hls, av_rescale(pkt->pts - hls->end_pts, st->time_base.num, st->time_base.den)); diff --git a/libavformat/segment.c b/libavformat/segment.c index 8afb41f93d..d79a32798e 100644 --- a/libavformat/segment.c +++ b/libavformat/segment.c @@ -272,13 +272,15 @@ static int seg_write_packet(AVFormatContext *s, AVPacket *pkt) AVFormatContext *oc = seg->avf; AVStream *st = s->streams[pkt->stream_index]; int64_t end_pts = seg->recording_time * seg->number; - int ret; + int ret, can_split = 1; - if ((seg->has_video && st->codec->codec_type == AVMEDIA_TYPE_VIDEO) && - av_compare_ts(pkt->pts, st->time_base, - end_pts, AV_TIME_BASE_Q) >= 0 && - pkt->flags & AV_PKT_FLAG_KEY) { + if (seg->has_video) { + can_split = st->codec->codec_type == AVMEDIA_TYPE_VIDEO && + pkt->flags & AV_PKT_FLAG_KEY; + } + if (can_split && av_compare_ts(pkt->pts, st->time_base, end_pts, + AV_TIME_BASE_Q) >= 0) { av_log(s, AV_LOG_DEBUG, "Next segment starts at %d %"PRId64"\n", pkt->stream_index, pkt->pts); |