static void MCDec_RegisterParameterSet(MCDec *ctx, char *data, u32 size, Bool is_sps)
{
Bool add = GF_TRUE;
u32 i, count;
s32 ps_id;
GF_List *dest = is_sps ? ctx->SPSs : ctx->PPSs;
if (is_sps) {
ps_id = gf_media_avc_read_sps(data, size, &ctx->avc, 0, NULL);
if (ps_id<0) return;
}
else {
ps_id = gf_media_avc_read_pps(data, size, &ctx->avc);
if (ps_id<0) return;
}
count = gf_list_count(dest);
for (i = 0; i<count; i++) {
GF_AVCConfigSlot *a_slc = gf_list_get(dest, i);
if (a_slc->id != ps_id) continue;
//not same size or different content but same ID, remove old xPS
if ((a_slc->size != size) || memcmp(a_slc->data, data, size)) {
gf_free(a_slc->data);
gf_free(a_slc);
gf_list_rem(dest, i);
break;
}
else {
add = GF_FALSE;
}
break;
}
if (add) {
GF_AVCConfigSlot *slc;
GF_SAFEALLOC(slc, GF_AVCConfigSlot);
slc->data = gf_malloc(size);
memcpy(slc->data, data, size);
slc->size = size;
slc->id = ps_id;
gf_list_add(dest, slc);
//force re-activation of sps/pps
if (is_sps) ctx->active_sps = -1;
else ctx->active_pps = -1;
}
}
/**
* A function which takes FFmpeg H264 extradata (SPS/PPS) and bring them ready to be pushed to the MP4 muxer.
* @param extradata
* @param extradata_size
* @param dstcfg
* @returns GF_OK is the extradata was parsed and is valid, other values otherwise.
*/
static GF_Err avc_import_ffextradata(const u8 *extradata, const u64 extradata_size, GF_AVCConfig *dstcfg)
{
#ifdef GPAC_DISABLE_AV_PARSERS
return GF_OK;
#else
u8 nal_size;
AVCState avc;
GF_BitStream *bs;
if (!extradata || (extradata_size < sizeof(u32)))
return GF_BAD_PARAM;
bs = gf_bs_new(extradata, extradata_size, GF_BITSTREAM_READ);
if (!bs)
return GF_BAD_PARAM;
if (gf_bs_read_u32(bs) != 0x00000001) {
gf_bs_del(bs);
return GF_BAD_PARAM;
}
//SPS
{
s32 idx;
char *buffer = NULL;
const u64 nal_start = 4;
nal_size = gf_media_nalu_next_start_code_bs(bs);
if (nal_start + nal_size > extradata_size) {
gf_bs_del(bs);
return GF_BAD_PARAM;
}
buffer = (char*)gf_malloc(nal_size);
gf_bs_read_data(bs, buffer, nal_size);
gf_bs_seek(bs, nal_start);
if ((gf_bs_read_u8(bs) & 0x1F) != GF_AVC_NALU_SEQ_PARAM) {
gf_bs_del(bs);
gf_free(buffer);
return GF_BAD_PARAM;
}
idx = gf_media_avc_read_sps(buffer, nal_size, &avc, 0, NULL);
if (idx < 0) {
gf_bs_del(bs);
gf_free(buffer);
return GF_BAD_PARAM;
}
dstcfg->configurationVersion = 1;
dstcfg->profile_compatibility = avc.sps[idx].prof_compat;
dstcfg->AVCProfileIndication = avc.sps[idx].profile_idc;
dstcfg->AVCLevelIndication = avc.sps[idx].level_idc;
dstcfg->chroma_format = avc.sps[idx].chroma_format;
dstcfg->luma_bit_depth = 8 + avc.sps[idx].luma_bit_depth_m8;
dstcfg->chroma_bit_depth = 8 + avc.sps[idx].chroma_bit_depth_m8;
{
GF_AVCConfigSlot *slc = (GF_AVCConfigSlot*)gf_malloc(sizeof(GF_AVCConfigSlot));
slc->size = nal_size;
slc->id = idx;
slc->data = buffer;
gf_list_add(dstcfg->sequenceParameterSets, slc);
}
}
//PPS
{
s32 idx;
char *buffer = NULL;
const u64 nal_start = 4 + nal_size + 4;
gf_bs_seek(bs, nal_start);
nal_size = gf_media_nalu_next_start_code_bs(bs);
if (nal_start + nal_size > extradata_size) {
gf_bs_del(bs);
return GF_BAD_PARAM;
}
buffer = (char*)gf_malloc(nal_size);
gf_bs_read_data(bs, buffer, nal_size);
gf_bs_seek(bs, nal_start);
if ((gf_bs_read_u8(bs) & 0x1F) != GF_AVC_NALU_PIC_PARAM) {
gf_bs_del(bs);
gf_free(buffer);
return GF_BAD_PARAM;
}
idx = gf_media_avc_read_pps(buffer, nal_size, &avc);
if (idx < 0) {
gf_bs_del(bs);
gf_free(buffer);
return GF_BAD_PARAM;
}
{
GF_AVCConfigSlot *slc = (GF_AVCConfigSlot*)gf_malloc(sizeof(GF_AVCConfigSlot));
slc->size = nal_size;
slc->id = idx;
slc->data = buffer;
gf_list_add(dstcfg->pictureParameterSets, slc);
}
}
gf_bs_del(bs);
return GF_OK;
#endif
}