static int
isom_write_avcc(sbuf_t *sb, const uint8_t *data, int len)
{
if (len > 6) {
/* check for h264 start code */
if (RB32(data) == 0x00000001 ||
RB24(data) == 0x000001) {
uint8_t *buf=NULL, *end, *start;
uint32_t *sps_size_array=0, *pps_size_array=0;
uint32_t pps_count=0,sps_count=0;
uint8_t **sps_array=0, **pps_array=0;
int i;
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 = RB32(buf);
nal_type = buf[4] & 0x1f;
if (nal_type == 7) { /* SPS */
sps_array = realloc(sps_array,sizeof(uint8_t*)*(sps_count+1));
sps_size_array = realloc(sps_size_array,sizeof(uint32_t)*(sps_count+1));
sps_array[sps_count] = buf + 4;
sps_size_array[sps_count] = size;
sps_count++;
} else if (nal_type == 8) { /* PPS */
pps_size_array = realloc(pps_size_array,sizeof(uint32_t)*(pps_count+1));
pps_array = realloc(pps_array,sizeof (uint8_t*)*(pps_count+1));
pps_array[pps_count] = buf + 4;
pps_size_array[pps_count] = size;
pps_count++;
}
buf += size + 4;
}
if(!sps_count || !pps_count) {
free(start);
if (sps_count)
free(sps_array);
if (pps_count)
free(pps_array);
return -1;
}
sbuf_put_byte(sb, 1); /* version */
sbuf_put_byte(sb, sps_array[0][1]); /* profile */
sbuf_put_byte(sb, sps_array[0][2]); /* profile compat */
sbuf_put_byte(sb, sps_array[0][3]); /* level */
sbuf_put_byte(sb, 0xff); /* 6 bits reserved (111111) + 2 bits nal size length - 1 (11) */
sbuf_put_byte(sb, 0xe0+sps_count); /* 3 bits reserved (111) + 5 bits number of sps (00001) */
for (i=0;i<sps_count;i++) {
sbuf_put_be16(sb, sps_size_array[i]);
sbuf_append(sb, sps_array[i], sps_size_array[i]);
}
sbuf_put_byte(sb, pps_count); /* number of pps */
for (i=0;i<pps_count;i++) {
sbuf_put_be16(sb, pps_size_array[i]);
sbuf_append(sb, pps_array[i], pps_size_array[i]);
}
free(start);
if (sps_count)
free(sps_array);
if (pps_count)
free(pps_array);
} else {
sbuf_append(sb, data, len);
}
}
return 0;
}
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;
}