Browse code

libavformat/mpegtsenc: support hevc with missing in stream headers like h.264

Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>

Michael Niedermayer authored on 2017/01/16 06:24:24
Showing 1 changed files
... ...
@@ -1619,9 +1619,39 @@ static int mpegts_write_packet_internal(AVFormatContext *s, AVPacket *pkt)
1619 1619
             }
1620 1620
         }
1621 1621
     } else if (st->codecpar->codec_id == AV_CODEC_ID_HEVC) {
1622
+        const uint8_t *p = buf, *buf_end = p + size;
1623
+        uint32_t state = -1;
1624
+        int extradd = (pkt->flags & AV_PKT_FLAG_KEY) ? st->codecpar->extradata_size : 0;
1622 1625
         int ret = check_hevc_startcode(s, st, pkt);
1623 1626
         if (ret < 0)
1624 1627
             return ret;
1628
+
1629
+        if (extradd && AV_RB24(st->codecpar->extradata) > 1)
1630
+            extradd = 0;
1631
+
1632
+        do {
1633
+            p = avpriv_find_start_code(p, buf_end, &state);
1634
+            av_log(s, AV_LOG_TRACE, "nal %d\n", (state & 0x7e)>>1);
1635
+            if ((state & 0x7e) == 2*32)
1636
+                extradd = 0;
1637
+        } while (p < buf_end && (state & 0x7e) != 2*35 &&
1638
+                 (state & 0x7e) >= 2*32);
1639
+
1640
+        if ((state & 0x7e) < 2*16 && (state & 0x7e) >= 2*24)
1641
+            extradd = 0;
1642
+        if ((state & 0x7e) != 2*35) { // AUD NAL
1643
+            data = av_malloc(pkt->size + 7 + extradd);
1644
+            if (!data)
1645
+                return AVERROR(ENOMEM);
1646
+            memcpy(data + 7, st->codecpar->extradata, extradd);
1647
+            memcpy(data + 7 + extradd, pkt->data, pkt->size);
1648
+            AV_WB32(data, 0x00000001);
1649
+            data[4] = 2*35;
1650
+            data[5] = 1;
1651
+            data[6] = 0x50; // any slice type (0x4) + rbsp stop one bit
1652
+            buf     = data;
1653
+            size    = pkt->size + 7 + extradd;
1654
+        }
1625 1655
     } else if (st->codecpar->codec_id == AV_CODEC_ID_OPUS) {
1626 1656
         if (pkt->size < 2) {
1627 1657
             av_log(s, AV_LOG_ERROR, "Opus packet too short\n");