CC:[email protected]
---
 libavformat/hlsenc.c  |   14 +++++++++-----
 libavformat/segment.c |   14 +++++++++-----
 2 files changed, 18 insertions(+), 10 deletions(-)

diff --git a/libavformat/hlsenc.c b/libavformat/hlsenc.c
index 89980d7..ae11e34 100644
--- a/libavformat/hlsenc.c
+++ b/libavformat/hlsenc.c
@@ -250,18 +250,22 @@ 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 = 0;
 
     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;
+    } else {
+        can_split = 1;
+    }
 
+    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 8afb41f..44bc3c5 100644
--- a/libavformat/segment.c
+++ b/libavformat/segment.c
@@ -272,13 +272,17 @@ 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 = 0;
 
-    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;
+    } else {
+        can_split = 1;
+    }
 
+    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);
 
-- 
1.7.10.4

_______________________________________________
libav-devel mailing list
[email protected]
https://lists.libav.org/mailman/listinfo/libav-devel

Reply via email to