const int isom_write_avcc(DllAvUtil *av_util_ctx, DllAvFormat *av_format_ctx,
ByteIOContext *pb, const uint8_t *data, int len)
{
// extradata from bytestream h264, convert to avcC atom data for bitstream
if (len > 6)
{
/* check for h264 start code */
if (VDA_RB32(data) == 0x00000001 || VDA_RB24(data) == 0x000001)
{
uint8_t *buf=NULL, *end, *start;
uint32_t sps_size=0, pps_size=0;
uint8_t *sps=0, *pps=0;
int ret = avc_parse_nal_units_buf(av_util_ctx, av_format_ctx, data, &buf, &len);
if (ret < 0)
return ret;
start = buf;
end = buf + len;
/* look for sps and pps */
while (buf < end)
{
unsigned int size;
uint8_t nal_type;
size = VDA_RB32(buf);
nal_type = buf[4] & 0x1f;
if (nal_type == 7) /* SPS */
{
sps = buf + 4;
sps_size = size;
}
else if (nal_type == 8) /* PPS */
{
pps = buf + 4;
pps_size = size;
}
buf += size + 4;
}
assert(sps);
av_format_ctx->put_byte(pb, 1); /* version */
av_format_ctx->put_byte(pb, sps[1]); /* profile */
av_format_ctx->put_byte(pb, sps[2]); /* profile compat */
av_format_ctx->put_byte(pb, sps[3]); /* level */
av_format_ctx->put_byte(pb, 0xff); /* 6 bits reserved (111111) + 2 bits nal size length - 1 (11) */
av_format_ctx->put_byte(pb, 0xe1); /* 3 bits reserved (111) + 5 bits number of sps (00001) */
av_format_ctx->put_be16(pb, sps_size);
av_format_ctx->put_buffer(pb, sps, sps_size);
if (pps)
{
av_format_ctx->put_byte(pb, 1); /* number of pps */
av_format_ctx->put_be16(pb, pps_size);
av_format_ctx->put_buffer(pb, pps, pps_size);
}
av_util_ctx->av_free(start);
}
else
{
av_format_ctx->put_buffer(pb, data, len);
}
}
return 0;
}
const int isom_write_avcc(AVIOContext *pb, const uint8_t *data, int len)
{
// extradata from bytestream h264, convert to avcC atom data for bitstream
if (len > 6)
{
/* check for h264 start code */
if (VDA_RB32(data) == 0x00000001 || VDA_RB24(data) == 0x000001)
{
uint8_t *buf=NULL, *end, *start;
uint32_t sps_size=0, pps_size=0;
uint8_t *sps=0, *pps=0;
int ret = avc_parse_nal_units_buf(data, &buf, &len);
if (ret < 0)
return ret;
start = buf;
end = buf + len;
/* look for sps and pps */
while (buf < end)
{
unsigned int size;
uint8_t nal_type;
size = VDA_RB32(buf);
nal_type = buf[4] & 0x1f;
if (nal_type == 7) /* SPS */
{
sps = buf + 4;
sps_size = size;
//parse_sps(sps+1, sps_size-1);
}
else if (nal_type == 8) /* PPS */
{
pps = buf + 4;
pps_size = size;
}
buf += size + 4;
}
if (!sps)
{
LOG(VB_GENERAL, LOG_ERR, LOC + "Invalid data (sps)");
return -1;
}
avio_w8(pb, 1); /* version */
avio_w8(pb, sps[1]); /* profile */
avio_w8(pb, sps[2]); /* profile compat */
avio_w8(pb, sps[3]); /* level */
avio_w8(pb, 0xff); /* 6 bits reserved (111111) + 2 bits nal size length - 1 (11) */
avio_w8(pb, 0xe1); /* 3 bits reserved (111) + 5 bits number of sps (00001) */
avio_wb16(pb, sps_size);
avio_write(pb, sps, sps_size);
if (pps)
{
avio_w8(pb, 1); /* number of pps */
avio_wb16(pb, pps_size);
avio_write(pb, pps, pps_size);
}
av_free(start);
}
else
{
avio_write(pb, data, len);
}
}
return 0;
}
