Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
Michael Niedermayer authored on 2017/01/16 06:24:24... | ... |
@@ -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"); |