This source file includes following definitions.
- gf_isom_is_nalu_based_entry
- rewrite_nalus_list
- process_extractor
- sap_type_from_nal_type
- is_sample_idr
- nalu_merge_ps
- gf_isom_nalu_sample_rewrite
- HEVC_DuplicateConfig
- AVC_DuplicateConfig
- merge_avc_config
- merge_hevc_config
- merge_all_config
- AVC_RewriteESDescriptorEx
- AVC_RewriteESDescriptor
- HEVC_RewriteESDescriptorEx
- HEVC_RewriteESDescriptor
- AVC_HEVC_UpdateESD
- gf_isom_avc_config_new
- gf_isom_avc_config_update_ex
- gf_isom_avc_set_inband_config
- gf_isom_avc_config_update
- gf_isom_svc_config_update
- gf_isom_mvc_config_update
- gf_isom_svc_mvc_config_del
- gf_isom_svc_config_del
- gf_isom_mvc_config_del
- gf_isom_set_ipod_compatible
- gf_isom_svc_mvc_config_new
- gf_isom_svc_config_new
- gf_isom_mvc_config_new
- gf_isom_hevc_config_new
- hevc_cleanup_config
- gf_isom_hevc_config_update_ex
- gf_isom_hevc_config_update
- gf_isom_hevc_set_inband_config
- gf_isom_hevc_set_tile_config
- gf_isom_lhvc_config_update
- gf_isom_clone_config_box
- gf_isom_avc_config_get
- gf_isom_hevc_config_get
- gf_isom_svc_config_get
- gf_isom_mvc_config_get
- gf_isom_get_avc_svc_type
- gf_isom_get_hevc_lhvc_type
- gf_isom_lhvc_config_get
- btrt_del
- btrt_Read
- btrt_New
- btrt_Write
- btrt_Size
- m4ds_del
- m4ds_Read
- m4ds_New
- m4ds_Write
- m4ds_Size
- avcc_del
- avcc_Read
- avcc_New
- avcc_Write
- avcc_Size
- hvcc_del
- hvcc_Read
- hvcc_New
- hvcc_Write
- hvcc_Size
- gf_isom_oinf_new_entry
- gf_isom_oinf_del_entry
- gf_isom_oinf_read_entry
- gf_isom_oinf_write_entry
- gf_isom_oinf_size_entry
- gf_isom_linf_new_entry
- gf_isom_linf_del_entry
- gf_isom_linf_read_entry
- gf_isom_linf_write_entry
- gf_isom_linf_size_entry
#include <gpac/internal/isomedia_dev.h>
#include <gpac/constants.h>
#ifndef GPAC_DISABLE_AV_PARSERS
#include <gpac/internal/media_dev.h>
#endif
#ifndef GPAC_DISABLE_ISOM
Bool gf_isom_is_nalu_based_entry(GF_MediaBox *mdia, GF_SampleEntryBox *_entry)
{
GF_MPEGVisualSampleEntryBox *entry;
if (mdia->handler->handlerType != GF_ISOM_MEDIA_VISUAL) return GF_FALSE;
switch (_entry->type) {
case GF_4CC('a','v','c','1'):
case GF_4CC('a','v','c','2'):
case GF_4CC('a','v','c','3'):
case GF_4CC('a','v','c','4'):
case GF_4CC('s','v','c','1'):
case GF_4CC('s','v','c','2'):
case GF_4CC('m','v','c','1'):
case GF_4CC('m','v','c','2'):
case GF_4CC('h','v','c','1'):
case GF_4CC('h','e','v','1'):
case GF_4CC('h','v','c','2'):
case GF_4CC('h','e','v','2'):
case GF_4CC('l','h','v','1'):
case GF_4CC('l','h','e','1'):
case GF_4CC('m','h','v','1'):
case GF_4CC('m','h','c','1'):
case GF_4CC('h','v','t','1'):
case GF_4CC('l','h','t','1'):
return GF_TRUE;
case GF_ISOM_BOX_TYPE_GNRV:
case GF_ISOM_BOX_TYPE_GNRA:
case GF_ISOM_BOX_TYPE_GNRM:
return GF_FALSE;
default:
break;
}
entry = (GF_MPEGVisualSampleEntryBox*)_entry;
if (!entry) return GF_FALSE;
if (entry->avc_config || entry->svc_config || entry->mvc_config || entry->hevc_config || entry->lhvc_config) return GF_TRUE;
return GF_FALSE;
}
static void rewrite_nalus_list(GF_List *nalus, GF_BitStream *bs, Bool rewrite_start_codes, u32 nal_unit_size_field)
{
u32 i, count = gf_list_count(nalus);
for (i=0; i<count; i++) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot*)gf_list_get(nalus, i);
if (rewrite_start_codes) gf_bs_write_u32(bs, 1);
else gf_bs_write_int(bs, sl->size, 8*nal_unit_size_field);
gf_bs_write_data(bs, sl->data, sl->size);
}
}
static GF_Err process_extractor(GF_ISOFile *file, GF_MediaBox *mdia, u32 sampleNumber, u64 sampleDTS, u32 nal_size, u16 nal_hdr, u32 nal_unit_size_field, Bool is_hevc, Bool rewrite_ps, Bool rewrite_start_codes, GF_BitStream *src_bs, GF_BitStream *dst_bs, u32 extractor_mode)
{
GF_Err e;
u32 di, ref_track_index, ref_track_num, data_offset, data_length, cur_extract_mode, ref_extract_mode, ref_nalu_size, nb_bytes_nalh;
GF_TrackReferenceTypeBox *dpnd;
GF_ISOSample *ref_samp;
GF_BitStream *ref_bs;
GF_TrackBox *ref_trak;
s8 sample_offset;
char*buffer = NULL;
u32 max_size = 0;
u32 last_byte, ref_sample_num, prev_ref_sample_num;
nb_bytes_nalh = is_hevc ? 2 : 1;
switch (extractor_mode) {
case 0:
last_byte = (u32) gf_bs_get_position(src_bs) + nal_size - (is_hevc ? 2 : 1);
if (!is_hevc) gf_bs_read_int(src_bs, 24);
while (gf_bs_get_position(src_bs) < last_byte) {
u32 xmode = 0;
if (is_hevc) xmode = gf_bs_read_u8(src_bs);
if (xmode) {
u8 len = gf_bs_read_u8(src_bs);
while (len) {
u8 c = gf_bs_read_u8(src_bs);
gf_bs_write_u8(dst_bs, c);
len--;
}
continue;
}
ref_track_index = gf_bs_read_u8(src_bs);
sample_offset = (s8) gf_bs_read_int(src_bs, 8);
data_offset = gf_bs_read_int(src_bs, nal_unit_size_field*8);
data_length = gf_bs_read_int(src_bs, nal_unit_size_field*8);
Track_FindRef(mdia->mediaTrack, GF_4CC('s','c','a','l'), &dpnd);
ref_track_num = 0;
if (dpnd && ref_track_index && (ref_track_index<=dpnd->trackIDCount))
ref_track_num = gf_isom_get_track_by_id(file, dpnd->trackIDs[ref_track_index-1]);
if (!ref_track_num) {
GF_LOG(GF_LOG_INFO, GF_LOG_CONTAINER, ("ISOBMF: Extractor target track is not present in file - skipping.\n"));
return GF_OK;
}
cur_extract_mode = gf_isom_get_nalu_extract_mode(file, ref_track_num);
ref_extract_mode = GF_ISOM_NALU_EXTRACT_INSPECT;
gf_isom_set_nalu_extract_mode(file, ref_track_num, ref_extract_mode);
ref_trak = gf_isom_get_track_from_file(file, ref_track_num);
if (!ref_trak) return GF_ISOM_INVALID_FILE;
ref_samp = gf_isom_sample_new();
if (!ref_samp) return GF_IO_ERR;
e = stbl_findEntryForTime(ref_trak->Media->information->sampleTable, sampleDTS, 0, &ref_sample_num, &prev_ref_sample_num);
if (e) return e;
if (!ref_sample_num) ref_sample_num = prev_ref_sample_num;
if (!ref_sample_num) return GF_ISOM_INVALID_FILE;
if ((sample_offset<0) && (ref_sample_num > (u32) -sample_offset)) return GF_ISOM_INVALID_FILE;
ref_sample_num = (u32) ( (s32) ref_sample_num + sample_offset);
e = Media_GetSample(ref_trak->Media, ref_sample_num, &ref_samp, &di, GF_FALSE, NULL);
if (e) return e;
if (rewrite_start_codes) {
gf_bs_write_int(dst_bs, 1, 32);
if (is_hevc) {
gf_bs_write_int(dst_bs, 0, 1);
gf_bs_write_int(dst_bs, GF_HEVC_NALU_ACCESS_UNIT, 6);
gf_bs_write_int(dst_bs, 0, 9);
gf_bs_write_int(dst_bs, 2, 3);
gf_bs_write_int(dst_bs, 0, 5);
} else {
gf_bs_write_int(dst_bs, (ref_samp->data[0] & 0x60) | GF_AVC_NALU_ACCESS_UNIT, 8);
gf_bs_write_int(dst_bs, 0xF0 , 8); ;
}
}
ref_bs = gf_bs_new(ref_samp->data + data_offset, ref_samp->dataLength - data_offset, GF_BITSTREAM_READ);
if (!data_length)
data_length = ref_samp->dataLength - data_offset;
if (ref_samp->dataLength - data_offset >= data_length) {
while (data_length && gf_bs_available(ref_bs)) {
ref_nalu_size = gf_bs_read_int(ref_bs, 8*nal_unit_size_field);
assert(ref_nalu_size <= data_length);
if (ref_nalu_size > max_size) {
buffer = (char*) gf_realloc(buffer, sizeof(char) * ref_nalu_size );
max_size = ref_nalu_size;
}
gf_bs_read_data(ref_bs, buffer, ref_nalu_size);
if (rewrite_start_codes)
gf_bs_write_u32(dst_bs, 1);
else
gf_bs_write_int(dst_bs, ref_nalu_size, 8*nal_unit_size_field);
gf_bs_write_data(dst_bs, buffer, ref_nalu_size);
data_length -= ref_nalu_size + nal_unit_size_field;
}
} else {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("ISOBMF: Extractor size is larger than refered sample size - skipping.\n"));
}
gf_isom_sample_del(&ref_samp);
ref_samp = NULL;
gf_bs_del(ref_bs);
ref_bs = NULL;
if (buffer) gf_free(buffer);
buffer = NULL;
gf_isom_set_nalu_extract_mode(file, ref_track_num, cur_extract_mode);
if (!is_hevc) break;
}
break;
case 1:
gf_bs_skip_bytes(src_bs, nal_size - nb_bytes_nalh);
break;
case 2:
buffer = (char*) gf_malloc( sizeof(char) * (nal_size - nb_bytes_nalh));
gf_bs_read_data(src_bs, buffer, nal_size - nb_bytes_nalh);
if (rewrite_start_codes)
gf_bs_write_u32(dst_bs, 1);
else
gf_bs_write_int(dst_bs, nal_size, 8*nal_unit_size_field);
gf_bs_write_u8(dst_bs, nal_hdr);
gf_bs_write_data(dst_bs, buffer, nal_size - nb_bytes_nalh);
gf_free(buffer);
break;
}
return GF_OK;
}
#ifndef GPAC_DISABLE_HEVC
static SAPType sap_type_from_nal_type(u8 nal_type) {
switch (nal_type) {
case GF_HEVC_NALU_SLICE_CRA:
return SAP_TYPE_3;
case GF_HEVC_NALU_SLICE_IDR_N_LP:
case GF_HEVC_NALU_SLICE_BLA_N_LP:
return SAP_TYPE_1;
case GF_HEVC_NALU_SLICE_IDR_W_DLP:
case GF_HEVC_NALU_SLICE_BLA_W_DLP:
case GF_HEVC_NALU_SLICE_BLA_W_LP:
return SAP_TYPE_2;
default:
return RAP_NO;
}
}
#endif
static SAPType is_sample_idr(GF_ISOSample *sample, GF_MPEGVisualSampleEntryBox *entry)
{
Bool is_hevc = GF_FALSE;
u32 nalu_size_field = 0;
GF_BitStream *bs;
if (entry->avc_config && entry->avc_config->config) nalu_size_field = entry->avc_config->config->nal_unit_size;
else if (entry->svc_config && entry->svc_config->config) nalu_size_field = entry->svc_config->config->nal_unit_size;
else if (entry->mvc_config && entry->mvc_config->config) nalu_size_field = entry->mvc_config->config->nal_unit_size;
else if (entry->hevc_config && entry->hevc_config->config) {
nalu_size_field = entry->hevc_config->config->nal_unit_size;
is_hevc = GF_TRUE;
}
else if (entry->lhvc_config && entry->lhvc_config->config) {
nalu_size_field = entry->lhvc_config->config->nal_unit_size;
is_hevc = GF_TRUE;
}
if (!nalu_size_field) return RAP_NO;
bs = gf_bs_new(sample->data, sample->dataLength, GF_BITSTREAM_READ);
if (!bs) return RAP_NO;
while (gf_bs_available(bs)) {
u8 nal_type;
u32 size = gf_bs_read_int(bs, 8*nalu_size_field);
if (is_hevc) {
#ifndef GPAC_DISABLE_HEVC
u16 nal_hdr = gf_bs_read_u16(bs);
nal_type = (nal_hdr&0x7E00) >> 9;
switch (nal_type) {
case GF_HEVC_NALU_SLICE_CRA:
gf_bs_del(bs);
return SAP_TYPE_3;
case GF_HEVC_NALU_SLICE_IDR_N_LP:
case GF_HEVC_NALU_SLICE_BLA_N_LP:
gf_bs_del(bs);
return SAP_TYPE_1;
case GF_HEVC_NALU_SLICE_IDR_W_DLP:
case GF_HEVC_NALU_SLICE_BLA_W_DLP:
case GF_HEVC_NALU_SLICE_BLA_W_LP:
gf_bs_del(bs);
return SAP_TYPE_2;
case GF_HEVC_NALU_ACCESS_UNIT:
case GF_HEVC_NALU_FILLER_DATA:
case GF_HEVC_NALU_SEI_PREFIX:
case GF_HEVC_NALU_VID_PARAM:
case GF_HEVC_NALU_SEQ_PARAM:
case GF_HEVC_NALU_PIC_PARAM:
break;
default:
gf_bs_del(bs);
return RAP_NO;
}
gf_bs_skip_bytes(bs, size - 2);
#endif
} else {
u8 nal_hdr = gf_bs_read_u8(bs);
nal_type = nal_hdr & 0x1F;
switch (nal_type) {
case GF_AVC_NALU_IDR_SLICE:
gf_bs_del(bs);
return SAP_TYPE_1;
case GF_AVC_NALU_ACCESS_UNIT:
case GF_AVC_NALU_FILLER_DATA:
case GF_AVC_NALU_SEI:
break;
default:
gf_bs_del(bs);
return RAP_NO;
}
gf_bs_skip_bytes(bs, size - 1);
}
}
gf_bs_del(bs);
return RAP_NO;
}
static void nalu_merge_ps(GF_BitStream *ps_bs, Bool rewrite_start_codes, u32 nal_unit_size_field, GF_MPEGVisualSampleEntryBox *entry, Bool is_hevc)
{
u32 i, count;
if (is_hevc) {
if (entry->hevc_config) {
count = gf_list_count(entry->hevc_config->config->param_array);
for (i=0; i<count; i++) {
GF_HEVCParamArray *ar = (GF_HEVCParamArray*)gf_list_get(entry->hevc_config->config->param_array, i);
rewrite_nalus_list(ar->nalus, ps_bs, rewrite_start_codes, nal_unit_size_field);
}
}
if (entry->lhvc_config) {
count = gf_list_count(entry->lhvc_config->config->param_array);
for (i=0; i<count; i++) {
GF_HEVCParamArray *ar = (GF_HEVCParamArray*)gf_list_get(entry->lhvc_config->config->param_array, i);
rewrite_nalus_list(ar->nalus, ps_bs, rewrite_start_codes, nal_unit_size_field);
}
}
} else {
if (entry->avc_config) {
rewrite_nalus_list(entry->avc_config->config->sequenceParameterSets, ps_bs, rewrite_start_codes, nal_unit_size_field);
rewrite_nalus_list(entry->avc_config->config->sequenceParameterSetExtensions, ps_bs, rewrite_start_codes, nal_unit_size_field);
rewrite_nalus_list(entry->avc_config->config->pictureParameterSets, ps_bs, rewrite_start_codes, nal_unit_size_field);
}
if (entry->svc_config) {
rewrite_nalus_list(entry->svc_config->config->sequenceParameterSets, ps_bs, rewrite_start_codes, nal_unit_size_field);
rewrite_nalus_list(entry->svc_config->config->pictureParameterSets, ps_bs, rewrite_start_codes, nal_unit_size_field);
}
if (entry->mvc_config) {
rewrite_nalus_list(entry->mvc_config->config->sequenceParameterSets, ps_bs, rewrite_start_codes, nal_unit_size_field);
rewrite_nalus_list(entry->mvc_config->config->pictureParameterSets, ps_bs, rewrite_start_codes, nal_unit_size_field);
}
}
}
GF_Err gf_isom_nalu_sample_rewrite(GF_MediaBox *mdia, GF_ISOSample *sample, u32 sampleNumber, GF_MPEGVisualSampleEntryBox *entry)
{
Bool is_hevc = GF_FALSE;
Bool check_cra_bla = GF_TRUE;
Bool insert_nalu_delim = GF_TRUE;
GF_Err e = GF_OK;
GF_ISOSample *ref_samp;
GF_BitStream *src_bs, *ref_bs, *dst_bs, *ps_bs;
u32 nal_size, max_size, nal_unit_size_field, extractor_mode;
Bool rewrite_ps, rewrite_start_codes, insert_vdrd_code;
s8 nal_type;
u32 nal_hdr, sabt_ref, i, track_num;
u32 temporal_id = 0;
char *buffer;
GF_ISOFile *file = mdia->mediaTrack->moov->mov;
GF_TrackReferenceTypeBox *scal = NULL;
src_bs = ref_bs = dst_bs = ps_bs = NULL;
ref_samp = NULL;
buffer = NULL;
Track_FindRef(mdia->mediaTrack, GF_4CC('s','c','a','l'), &scal);
rewrite_ps = (mdia->mediaTrack->extractor_mode & GF_ISOM_NALU_EXTRACT_INBAND_PS_FLAG) ? GF_TRUE : GF_FALSE;
rewrite_start_codes = (mdia->mediaTrack->extractor_mode & GF_ISOM_NALU_EXTRACT_ANNEXB_FLAG) ? GF_TRUE : GF_FALSE;
insert_vdrd_code = (mdia->mediaTrack->extractor_mode & GF_ISOM_NALU_EXTRACT_VDRD_FLAG) ? GF_TRUE : GF_FALSE;
if (!entry->svc_config && !entry->mvc_config && !entry->lhvc_config) insert_vdrd_code = GF_FALSE;
extractor_mode = mdia->mediaTrack->extractor_mode&0x0000FFFF;
if (mdia->mediaTrack->extractor_mode & GF_ISOM_NALU_EXTRACT_TILE_ONLY) {
insert_nalu_delim = GF_FALSE;
}
track_num = 1 + gf_list_find(mdia->mediaTrack->moov->trackList, mdia->mediaTrack);
if ( (extractor_mode != GF_ISOM_NALU_EXTRACT_INSPECT) && !(mdia->mediaTrack->extractor_mode & GF_ISOM_NALU_EXTRACT_TILE_ONLY) ) {
u32 ref_track, di;
if (entry->lhvc_config && !entry->hevc_config) {
GF_ISOSample *base_samp;
if (gf_isom_get_reference_count(mdia->mediaTrack->moov->mov, track_num, GF_ISOM_REF_SCAL) <= 0) {
if ( gf_isom_get_reference_count(mdia->mediaTrack->moov->mov, track_num, GF_ISOM_REF_BASE) >= 1) {
gf_isom_get_reference(mdia->mediaTrack->moov->mov, track_num, GF_ISOM_REF_BASE, 1, &ref_track);
switch (gf_isom_get_media_subtype(mdia->mediaTrack->moov->mov , ref_track, 1)) {
case GF_ISOM_SUBTYPE_HVC1:
case GF_ISOM_SUBTYPE_HVC2:
case GF_ISOM_SUBTYPE_HEV1:
case GF_ISOM_SUBTYPE_HEV2:
base_samp = gf_isom_get_sample(mdia->mediaTrack->moov->mov, ref_track, sampleNumber + mdia->mediaTrack->sample_count_at_seg_start, &di);
if (base_samp && base_samp->data) {
sample->data = gf_realloc(sample->data, sample->dataLength+base_samp->dataLength);
memmove(sample->data + base_samp->dataLength, sample->data , sample->dataLength);
memcpy(sample->data, base_samp->data, base_samp->dataLength);
sample->dataLength += base_samp->dataLength;
}
if (base_samp) gf_isom_sample_del(&base_samp);
Track_FindRef(mdia->mediaTrack, GF_4CC('s','b','a','s'), &scal);
break;
}
}
}
}
sabt_ref = gf_isom_get_reference_count(mdia->mediaTrack->moov->mov, track_num, GF_ISOM_REF_SABT);
if ((s32) sabt_ref >= 0) {
for (i=0; i<sabt_ref; i++) {
GF_ISOSample *tile_samp;
gf_isom_get_reference(mdia->mediaTrack->moov->mov, track_num, GF_ISOM_REF_SABT, i+1, &ref_track);
tile_samp = gf_isom_get_sample(mdia->mediaTrack->moov->mov, ref_track, sampleNumber + mdia->mediaTrack->sample_count_at_seg_start, &di);
if (tile_samp && tile_samp ->data) {
sample->data = gf_realloc(sample->data, sample->dataLength+tile_samp->dataLength);
memcpy(sample->data + sample->dataLength, tile_samp->data, tile_samp->dataLength);
sample->dataLength += tile_samp->dataLength;
}
if (tile_samp) gf_isom_sample_del(&tile_samp);
}
}
}
if ( gf_isom_get_reference_count(mdia->mediaTrack->moov->mov, track_num, GF_ISOM_REF_TBAS) >= 1) {
u32 ref_track;
u32 idx = gf_list_find(mdia->information->sampleTable->SampleDescription->other_boxes, entry);
GF_TrackBox *tbas;
gf_isom_get_reference(mdia->mediaTrack->moov->mov, track_num, GF_ISOM_REF_TBAS, 1, &ref_track);
tbas = (GF_TrackBox *)gf_list_get(mdia->mediaTrack->moov->trackList, ref_track-1);
entry = gf_list_get(tbas->Media->information->sampleTable->SampleDescription->other_boxes, idx);
}
if (sample->IsRAP < SAP_TYPE_2) {
if (mdia->information->sampleTable->no_sync_found || (!sample->IsRAP && check_cra_bla) ) {
sample->IsRAP = is_sample_idr(sample, entry);
}
}
if (!sample->IsRAP)
rewrite_ps = GF_FALSE;
if (extractor_mode != GF_ISOM_NALU_EXTRACT_LAYER_ONLY)
insert_vdrd_code = GF_FALSE;
if (entry->lhvc_config && entry->hevc_config)
insert_vdrd_code = GF_FALSE;
if (extractor_mode == GF_ISOM_NALU_EXTRACT_INSPECT) {
if (!rewrite_ps && !rewrite_start_codes)
return GF_OK;
}
if (!entry) return GF_BAD_PARAM;
nal_unit_size_field = 0;
if (entry->svc_config && entry->svc_config->config)
nal_unit_size_field = entry->svc_config->config->nal_unit_size;
if (entry->mvc_config && entry->mvc_config->config)
nal_unit_size_field = entry->mvc_config->config->nal_unit_size;
else if (entry->lhvc_config && entry->lhvc_config->config) {
is_hevc = GF_TRUE;
nal_unit_size_field = entry->lhvc_config->config->nal_unit_size;
}
else if (!rewrite_ps && !rewrite_start_codes && !scal) {
return GF_OK;
}
if (!nal_unit_size_field) {
if (entry->avc_config) nal_unit_size_field = entry->avc_config->config->nal_unit_size;
else if (entry->hevc_config || entry->lhvc_config ) {
nal_unit_size_field = entry->lhvc_config ? entry->lhvc_config->config->nal_unit_size : entry->hevc_config->config->nal_unit_size;
is_hevc = GF_TRUE;
}
}
if (!nal_unit_size_field) return GF_ISOM_INVALID_FILE;
dst_bs = gf_bs_new(NULL, 0, GF_BITSTREAM_WRITE);
ps_bs = gf_bs_new(NULL, 0, GF_BITSTREAM_WRITE);
src_bs = gf_bs_new(sample->data, sample->dataLength, GF_BITSTREAM_READ);
if (!src_bs && sample->data) return GF_ISOM_INVALID_FILE;
max_size = 4096;
if (rewrite_start_codes) {
if (insert_vdrd_code) {
if (is_hevc) {
} else {
gf_bs_write_int(dst_bs, 1, 32);
gf_bs_write_int(dst_bs, GF_AVC_NALU_VDRD , 8);
insert_nalu_delim=0;
}
}
if (insert_nalu_delim) {
gf_bs_write_int(dst_bs, 1, 32);
if (is_hevc) {
#ifndef GPAC_DISABLE_HEVC
gf_bs_write_int(dst_bs, 0, 1);
gf_bs_write_int(dst_bs, GF_HEVC_NALU_ACCESS_UNIT, 6);
gf_bs_write_int(dst_bs, insert_vdrd_code ? 1 : 0, 6);
gf_bs_write_int(dst_bs, 1, 3);
gf_bs_write_int(dst_bs, 2, 3);
gf_bs_write_int(dst_bs, 0, 5);
#endif
} else {
gf_bs_write_int(dst_bs, (sample->data[0] & 0x60) | GF_AVC_NALU_ACCESS_UNIT, 8);
gf_bs_write_int(dst_bs, 0xF0 , 8); ;
}
}
}
if (rewrite_ps) {
if (extractor_mode == GF_ISOM_NALU_EXTRACT_DEFAULT) {
u32 i;
if (scal) {
for (i=0; i<scal->trackIDCount; i++) {
GF_TrackBox *a_track = GetTrackbyID(mdia->mediaTrack->moov, scal->trackIDs[i]);
GF_MPEGVisualSampleEntryBox *an_entry = NULL;
if (a_track && a_track->Media && a_track->Media->information && a_track->Media->information->sampleTable && a_track->Media->information->sampleTable->SampleDescription)
an_entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(a_track->Media->information->sampleTable->SampleDescription->other_boxes, 0);
if (an_entry)
nalu_merge_ps(ps_bs, rewrite_start_codes, nal_unit_size_field, an_entry, is_hevc);
}
}
}
nalu_merge_ps(ps_bs, rewrite_start_codes, nal_unit_size_field, entry, is_hevc);
if (is_hevc) {
if (!rewrite_start_codes && !entry->lhvc_config && !scal) {
if (ps_bs) {
u8 nal_type = (sample->data[nal_unit_size_field] & 0x7E) >> 1;
switch (nal_type) {
#ifndef GPAC_DISABLE_HEVC
case GF_HEVC_NALU_VID_PARAM:
case GF_HEVC_NALU_SEQ_PARAM:
case GF_HEVC_NALU_PIC_PARAM:
break;
#endif
default:
gf_bs_transfer(dst_bs, ps_bs);
break;
}
gf_bs_del(ps_bs);
ps_bs = NULL;
}
gf_bs_write_data(dst_bs, sample->data, sample->dataLength);
gf_free(sample->data);
sample->data = NULL;
gf_bs_get_content(dst_bs, &sample->data, &sample->dataLength);
gf_bs_del(src_bs);
gf_bs_del(dst_bs);
return GF_OK;
}
}
}
if (!scal && !rewrite_start_codes && !rewrite_ps) {
if (ps_bs)
{
gf_bs_transfer(dst_bs, ps_bs);
gf_bs_del(ps_bs);
ps_bs = NULL;
}
gf_bs_write_data(dst_bs, sample->data, sample->dataLength);
gf_free(sample->data);
sample->data = NULL;
gf_bs_get_content(dst_bs, &sample->data, &sample->dataLength);
gf_bs_del(src_bs);
gf_bs_del(dst_bs);
return GF_OK;
}
buffer = (char *)gf_malloc(sizeof(char)*max_size);
while (gf_bs_available(src_bs)) {
nal_size = gf_bs_read_int(src_bs, 8*nal_unit_size_field);
if (gf_bs_get_position(src_bs) + nal_size > sample->dataLength) {
GF_LOG(GF_LOG_WARNING, GF_LOG_CODING, ("Sample %u (size %u) rewrite: corrupted NAL Unit (size %u)\n", sampleNumber, sample->dataLength, nal_size));
goto exit;
}
if (nal_size>max_size) {
buffer = (char*) gf_realloc(buffer, sizeof(char)*nal_size);
max_size = nal_size;
}
if (is_hevc) {
nal_hdr = gf_bs_read_u16(src_bs);
nal_type = (nal_hdr&0x7E00) >> 9;
} else {
nal_hdr = gf_bs_read_u8(src_bs);
nal_type = nal_hdr & 0x1F;
}
if (is_hevc) {
if (ps_bs) {
gf_bs_transfer(dst_bs, ps_bs);
gf_bs_del(ps_bs);
ps_bs = NULL;
}
#ifndef GPAC_DISABLE_HEVC
if (nal_type==GF_HEVC_NALU_ACCESS_UNIT) {
gf_bs_skip_bytes(src_bs, nal_size-2);
continue;
}
switch (nal_type) {
case 49:
e = process_extractor(file, mdia, sampleNumber, sample->DTS, nal_size, nal_hdr, nal_unit_size_field, GF_TRUE, rewrite_ps, rewrite_start_codes, src_bs, dst_bs, extractor_mode);
if (e) goto exit;
break;
case GF_HEVC_NALU_SLICE_TSA_N:
case GF_HEVC_NALU_SLICE_STSA_N:
case GF_HEVC_NALU_SLICE_TSA_R:
case GF_HEVC_NALU_SLICE_STSA_R:
if (temporal_id < (nal_hdr & 0x7))
temporal_id = (nal_hdr & 0x7);
gf_bs_read_data(src_bs, buffer, nal_size-2);
if (rewrite_start_codes)
gf_bs_write_u32(dst_bs, 1);
else
gf_bs_write_int(dst_bs, nal_size, 8*nal_unit_size_field);
gf_bs_write_u16(dst_bs, nal_hdr);
gf_bs_write_data(dst_bs, buffer, nal_size-2);
break;
case GF_HEVC_NALU_SLICE_BLA_W_LP:
case GF_HEVC_NALU_SLICE_BLA_W_DLP:
case GF_HEVC_NALU_SLICE_BLA_N_LP:
case GF_HEVC_NALU_SLICE_IDR_W_DLP:
case GF_HEVC_NALU_SLICE_IDR_N_LP:
case GF_HEVC_NALU_SLICE_CRA:
if (check_cra_bla && !sample->IsRAP) {
if (ref_samp) gf_isom_sample_del(&ref_samp);
if (src_bs) gf_bs_del(src_bs);
if (ref_bs) gf_bs_del(ref_bs);
if (dst_bs) gf_bs_del(dst_bs);
if (buffer) gf_free(buffer);
sample->IsRAP = sap_type_from_nal_type(nal_type);
return gf_isom_nalu_sample_rewrite(mdia, sample, sampleNumber, entry);
}
default:
if (nal_size<2) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[iso file] Invalid nal size %d in sample %d\n", nal_type, sampleNumber));
e = GF_NON_COMPLIANT_BITSTREAM;
goto exit;
}
gf_bs_read_data(src_bs, buffer, nal_size-2);
if (rewrite_start_codes)
gf_bs_write_u32(dst_bs, 1);
else
gf_bs_write_int(dst_bs, nal_size, 8*nal_unit_size_field);
gf_bs_write_u16(dst_bs, nal_hdr);
gf_bs_write_data(dst_bs, buffer, nal_size-2);
}
#endif
continue;
}
switch(nal_type) {
case GF_AVC_NALU_ACCESS_UNIT:
gf_bs_skip_bytes(src_bs, nal_size-1);
continue;
case 31:
e = process_extractor(file, mdia, sampleNumber, sample->DTS, nal_size, nal_hdr, nal_unit_size_field, GF_FALSE, rewrite_ps, rewrite_start_codes, src_bs, dst_bs, extractor_mode);
if (e) goto exit;
break;
case GF_AVC_NALU_SEQ_PARAM:
case GF_AVC_NALU_PIC_PARAM:
case GF_AVC_NALU_SEQ_PARAM_EXT:
case GF_AVC_NALU_SVC_SUBSEQ_PARAM:
if (ps_bs) {
gf_bs_del(ps_bs);
ps_bs = NULL;
}
default:
if (ps_bs) {
gf_bs_transfer(dst_bs, ps_bs);
gf_bs_del(ps_bs);
ps_bs = NULL;
}
gf_bs_read_data(src_bs, buffer, nal_size-1);
if (rewrite_start_codes)
gf_bs_write_u32(dst_bs, 1);
else
gf_bs_write_int(dst_bs, nal_size, 8*nal_unit_size_field);
gf_bs_write_u8(dst_bs, nal_hdr);
gf_bs_write_data(dst_bs, buffer, nal_size-1);
}
}
gf_free(sample->data);
sample->data = NULL;
gf_bs_get_content(dst_bs, &sample->data, &sample->dataLength);
if (rewrite_start_codes && is_hevc && temporal_id) {
sample->data[6] = (sample->data[6] & 0xF8) | (temporal_id+1);
}
exit:
if (ref_samp) gf_isom_sample_del(&ref_samp);
if (src_bs) gf_bs_del(src_bs);
if (ref_bs) gf_bs_del(ref_bs);
if (dst_bs) gf_bs_del(dst_bs);
if (buffer) gf_free(buffer);
return e;
}
GF_HEVCConfig *HEVC_DuplicateConfig(GF_HEVCConfig *cfg)
{
char *data;
u32 data_size;
GF_HEVCConfig *new_cfg;
GF_BitStream *bs;
if (!cfg) return NULL;
bs = gf_bs_new(NULL, 0, GF_BITSTREAM_WRITE);
gf_odf_hevc_cfg_write_bs(cfg, bs);
gf_bs_get_content(bs, &data, &data_size);
gf_bs_del(bs);
bs = gf_bs_new(data, data_size, GF_BITSTREAM_READ);
new_cfg = gf_odf_hevc_cfg_read_bs(bs, cfg->is_lhvc);
new_cfg->is_lhvc = cfg->is_lhvc;
gf_bs_del(bs);
gf_free(data);
return new_cfg;
}
static GF_AVCConfig *AVC_DuplicateConfig(GF_AVCConfig *cfg)
{
u32 i, count;
GF_AVCConfigSlot *p1, *p2;
GF_AVCConfig *cfg_new = gf_odf_avc_cfg_new();
cfg_new->AVCLevelIndication = cfg->AVCLevelIndication;
cfg_new->AVCProfileIndication = cfg->AVCProfileIndication;
cfg_new->configurationVersion = cfg->configurationVersion;
cfg_new->nal_unit_size = cfg->nal_unit_size;
cfg_new->profile_compatibility = cfg->profile_compatibility;
cfg_new->complete_representation = cfg->complete_representation;
cfg_new->chroma_bit_depth = cfg->chroma_bit_depth;
cfg_new->luma_bit_depth = cfg->luma_bit_depth;
cfg_new->chroma_format = cfg->chroma_format;
count = gf_list_count(cfg->sequenceParameterSets);
for (i=0; i<count; i++) {
p1 = (GF_AVCConfigSlot*)gf_list_get(cfg->sequenceParameterSets, i);
p2 = (GF_AVCConfigSlot*)gf_malloc(sizeof(GF_AVCConfigSlot));
p2->size = p1->size;
p2->id = p1->id;
p2->data = (char *)gf_malloc(sizeof(char)*p1->size);
memcpy(p2->data, p1->data, sizeof(char)*p1->size);
gf_list_add(cfg_new->sequenceParameterSets, p2);
}
count = gf_list_count(cfg->pictureParameterSets);
for (i=0; i<count; i++) {
p1 = (GF_AVCConfigSlot*)gf_list_get(cfg->pictureParameterSets, i);
p2 = (GF_AVCConfigSlot*)gf_malloc(sizeof(GF_AVCConfigSlot));
p2->size = p1->size;
p2->id = p1->id;
p2->data = (char*)gf_malloc(sizeof(char)*p1->size);
memcpy(p2->data, p1->data, sizeof(char)*p1->size);
gf_list_add(cfg_new->pictureParameterSets, p2);
}
if (cfg->sequenceParameterSetExtensions) {
cfg_new->sequenceParameterSetExtensions = gf_list_new();
count = gf_list_count(cfg->sequenceParameterSetExtensions);
for (i=0; i<count; i++) {
p1 = (GF_AVCConfigSlot*)gf_list_get(cfg->sequenceParameterSetExtensions, i);
p2 = (GF_AVCConfigSlot*)gf_malloc(sizeof(GF_AVCConfigSlot));
p2->size = p1->size;
p2->id = p1->id;
p2->data = (char*)gf_malloc(sizeof(char)*p1->size);
memcpy(p2->data, p1->data, sizeof(char)*p1->size);
gf_list_add(cfg_new->sequenceParameterSetExtensions, p2);
}
}
return cfg_new;
}
static void merge_avc_config(GF_AVCConfig *dst_cfg, GF_AVCConfig *src_cfg)
{
GF_AVCConfig *cfg = AVC_DuplicateConfig(src_cfg);
if (!cfg || !dst_cfg) return;
while (gf_list_count(cfg->sequenceParameterSets)) {
GF_AVCConfigSlot *p = (GF_AVCConfigSlot*)gf_list_get(cfg->sequenceParameterSets, 0);
gf_list_rem(cfg->sequenceParameterSets, 0);
gf_list_insert(dst_cfg->sequenceParameterSets, p, 0);
}
while (gf_list_count(cfg->pictureParameterSets)) {
GF_AVCConfigSlot *p = (GF_AVCConfigSlot*)gf_list_get(cfg->pictureParameterSets, 0);
gf_list_rem(cfg->pictureParameterSets, 0);
gf_list_insert(dst_cfg->pictureParameterSets, p, 0);
}
gf_odf_avc_cfg_del(cfg);
}
void merge_hevc_config(GF_HEVCConfig *dst_cfg, GF_HEVCConfig *src_cfg, Bool force_insert)
{
GF_HEVCConfig *cfg = HEVC_DuplicateConfig(src_cfg);
u32 i, j, count = cfg->param_array ? gf_list_count(cfg->param_array) : 0;
for (i=0; i<count; i++) {
GF_HEVCParamArray *ar_h = NULL;
u32 count2 = dst_cfg->param_array ? gf_list_count(dst_cfg->param_array) : 0;
GF_HEVCParamArray *ar = (GF_HEVCParamArray*)gf_list_get(cfg->param_array, i);
for (j=0; j<count2; j++) {
ar_h = (GF_HEVCParamArray*)gf_list_get(dst_cfg->param_array, j);
if (ar_h->type==ar->type) {
break;
}
ar_h = NULL;
}
if (!ar_h) {
gf_list_add(dst_cfg->param_array, ar);
gf_list_rem(cfg->param_array, i);
count--;
i--;
} else {
while (gf_list_count(ar->nalus)) {
GF_AVCConfigSlot *p = (GF_AVCConfigSlot*)gf_list_get(ar->nalus, 0);
gf_list_rem(ar->nalus, 0);
if (force_insert)
gf_list_insert(ar_h->nalus, p, 0);
else
gf_list_add(ar_h->nalus, p);
}
}
}
gf_odf_hevc_cfg_del(cfg);
#define CHECK_CODE(__code) if (dst_cfg->__code < src_cfg->__code) dst_cfg->__code = src_cfg->__code;
CHECK_CODE(configurationVersion)
CHECK_CODE(profile_idc)
CHECK_CODE(profile_space)
CHECK_CODE(tier_flag)
CHECK_CODE(general_profile_compatibility_flags)
CHECK_CODE(progressive_source_flag)
CHECK_CODE(interlaced_source_flag)
CHECK_CODE(constraint_indicator_flags)
CHECK_CODE(level_idc)
CHECK_CODE(min_spatial_segmentation_idc)
}
void merge_all_config(GF_AVCConfig *avc_cfg, GF_HEVCConfig *hevc_cfg, GF_MediaBox *mdia)
{
u32 i;
GF_TrackReferenceTypeBox *scal = NULL;
Track_FindRef(mdia->mediaTrack, GF_4CC('s','c','a','l'), &scal);
if (!scal) return;
for (i=0; i<scal->trackIDCount; i++) {
GF_TrackBox *a_track = GetTrackbyID(mdia->mediaTrack->moov, scal->trackIDs[i]);
GF_MPEGVisualSampleEntryBox *an_entry = NULL;
if (a_track && a_track->Media && a_track->Media->information && a_track->Media->information->sampleTable && a_track->Media->information->sampleTable->SampleDescription)
an_entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(a_track->Media->information->sampleTable->SampleDescription->other_boxes, 0);
if (!an_entry) continue;
if (avc_cfg && an_entry->svc_config && an_entry->svc_config->config)
merge_avc_config(avc_cfg, an_entry->svc_config->config);
if (avc_cfg && an_entry->mvc_config && an_entry->mvc_config->config)
merge_avc_config(avc_cfg, an_entry->mvc_config->config);
if (avc_cfg && an_entry->avc_config && an_entry->avc_config->config)
merge_avc_config(avc_cfg, an_entry->avc_config->config);
if (hevc_cfg && an_entry->lhvc_config && an_entry->lhvc_config->config)
merge_hevc_config(hevc_cfg, an_entry->lhvc_config->config, GF_TRUE);
if (hevc_cfg && an_entry->hevc_config && an_entry->hevc_config->config)
merge_hevc_config(hevc_cfg, an_entry->hevc_config->config, GF_TRUE);
}
if (hevc_cfg) hevc_cfg->is_lhvc = GF_FALSE;
}
void AVC_RewriteESDescriptorEx(GF_MPEGVisualSampleEntryBox *avc, GF_MediaBox *mdia)
{
GF_AVCConfig *avcc, *svcc, *mvcc;
GF_BitRateBox *btrt = gf_isom_sample_entry_get_bitrate((GF_SampleEntryBox *)avc, GF_FALSE);
if (avc->emul_esd) gf_odf_desc_del((GF_Descriptor *)avc->emul_esd);
avc->emul_esd = gf_odf_desc_esd_new(2);
avc->emul_esd->decoderConfig->streamType = GF_STREAM_VISUAL;
if (avc->svc_config && !avc->avc_config)
avc->emul_esd->decoderConfig->objectTypeIndication = GPAC_OTI_VIDEO_SVC;
else if (avc->mvc_config && !avc->avc_config)
avc->emul_esd->decoderConfig->objectTypeIndication = GPAC_OTI_VIDEO_MVC;
else
avc->emul_esd->decoderConfig->objectTypeIndication = GPAC_OTI_VIDEO_AVC;
if (btrt) {
avc->emul_esd->decoderConfig->bufferSizeDB = btrt->bufferSizeDB;
avc->emul_esd->decoderConfig->avgBitrate = btrt->avgBitrate;
avc->emul_esd->decoderConfig->maxBitrate = btrt->maxBitrate;
}
if (avc->descr) {
u32 i=0;
GF_Descriptor *desc,*clone;
i=0;
while ((desc = (GF_Descriptor *)gf_list_enum(avc->descr->descriptors, &i))) {
clone = NULL;
gf_odf_desc_copy(desc, &clone);
if (gf_odf_desc_add_desc((GF_Descriptor *)avc->emul_esd, clone) != GF_OK)
gf_odf_desc_del(clone);
}
}
if (avc->avc_config) {
avcc = avc->avc_config->config ? AVC_DuplicateConfig(avc->avc_config->config) : NULL;
if (avc->svc_config) {
merge_avc_config(avcc, avc->svc_config->config);
}
if (avc->mvc_config) {
merge_avc_config(avcc, avc->mvc_config->config);
}
if (avcc) {
if (mdia) merge_all_config(avcc, NULL, mdia);
gf_odf_avc_cfg_write(avcc, &avc->emul_esd->decoderConfig->decoderSpecificInfo->data, &avc->emul_esd->decoderConfig->decoderSpecificInfo->dataLength);
gf_odf_avc_cfg_del(avcc);
}
} else if (avc->svc_config) {
svcc = AVC_DuplicateConfig(avc->svc_config->config);
if (mdia) merge_all_config(svcc, NULL, mdia);
gf_odf_avc_cfg_write(svcc, &avc->emul_esd->decoderConfig->decoderSpecificInfo->data, &avc->emul_esd->decoderConfig->decoderSpecificInfo->dataLength);
gf_odf_avc_cfg_del(svcc);
}
else if (avc->mvc_config) {
mvcc = AVC_DuplicateConfig(avc->mvc_config->config);
if (mdia) merge_all_config(mvcc, NULL, mdia);
gf_odf_avc_cfg_write(mvcc, &avc->emul_esd->decoderConfig->decoderSpecificInfo->data, &avc->emul_esd->decoderConfig->decoderSpecificInfo->dataLength);
gf_odf_avc_cfg_del(mvcc);
}
}
void AVC_RewriteESDescriptor(GF_MPEGVisualSampleEntryBox *avc)
{
AVC_RewriteESDescriptorEx(avc, NULL);
}
void HEVC_RewriteESDescriptorEx(GF_MPEGVisualSampleEntryBox *hevc, GF_MediaBox *mdia)
{
GF_BitRateBox *btrt = gf_isom_sample_entry_get_bitrate((GF_SampleEntryBox *)hevc, GF_FALSE);
if (hevc->emul_esd) gf_odf_desc_del((GF_Descriptor *)hevc->emul_esd);
hevc->emul_esd = gf_odf_desc_esd_new(2);
hevc->emul_esd->decoderConfig->streamType = GF_STREAM_VISUAL;
hevc->emul_esd->decoderConfig->objectTypeIndication = GPAC_OTI_VIDEO_HEVC;
if (hevc->lhvc_config )
hevc->emul_esd->decoderConfig->objectTypeIndication = GPAC_OTI_VIDEO_LHVC;
if (btrt) {
hevc->emul_esd->decoderConfig->bufferSizeDB = btrt->bufferSizeDB;
hevc->emul_esd->decoderConfig->avgBitrate = btrt->avgBitrate;
hevc->emul_esd->decoderConfig->maxBitrate = btrt->maxBitrate;
}
if (hevc->descr) {
u32 i=0;
GF_Descriptor *desc,*clone;
i=0;
while ((desc = (GF_Descriptor *)gf_list_enum(hevc->descr->descriptors, &i))) {
clone = NULL;
gf_odf_desc_copy(desc, &clone);
if (gf_odf_desc_add_desc((GF_Descriptor *)hevc->emul_esd, clone) != GF_OK)
gf_odf_desc_del(clone);
}
}
if (hevc->hevc_config || hevc->lhvc_config) {
GF_HEVCConfig *hcfg = HEVC_DuplicateConfig(hevc->hevc_config ? hevc->hevc_config->config : hevc->lhvc_config->config);
if (hevc->hevc_config && hevc->lhvc_config) {
merge_hevc_config(hcfg, hevc->lhvc_config->config, GF_FALSE);
}
if (mdia) merge_all_config(NULL, hcfg, mdia);
if (hcfg) {
if (mdia && ((mdia->mediaTrack->extractor_mode&0x0000FFFF) != GF_ISOM_NALU_EXTRACT_INSPECT)) {
hcfg->is_lhvc=GF_FALSE;
}
gf_odf_hevc_cfg_write(hcfg, &hevc->emul_esd->decoderConfig->decoderSpecificInfo->data, &hevc->emul_esd->decoderConfig->decoderSpecificInfo->dataLength);
gf_odf_hevc_cfg_del(hcfg);
}
}
}
void HEVC_RewriteESDescriptor(GF_MPEGVisualSampleEntryBox *hevc)
{
HEVC_RewriteESDescriptorEx(hevc, NULL);
}
GF_Err AVC_HEVC_UpdateESD(GF_MPEGVisualSampleEntryBox *avc, GF_ESD *esd)
{
GF_BitRateBox *btrt = gf_isom_sample_entry_get_bitrate((GF_SampleEntryBox *)avc, GF_TRUE);
if (avc->descr) gf_isom_box_del((GF_Box *) avc->descr);
avc->descr = NULL;
btrt->avgBitrate = esd->decoderConfig->avgBitrate;
btrt->maxBitrate = esd->decoderConfig->maxBitrate;
btrt->bufferSizeDB = esd->decoderConfig->bufferSizeDB;
if (gf_list_count(esd->IPIDataSet)
|| gf_list_count(esd->IPMPDescriptorPointers)
|| esd->langDesc
|| gf_list_count(esd->extensionDescriptors)
|| esd->ipiPtr || esd->qos || esd->RegDescriptor) {
avc->descr = (GF_MPEG4ExtensionDescriptorsBox *)gf_isom_box_new(GF_ISOM_BOX_TYPE_M4DS);
if (esd->RegDescriptor) {
gf_list_add(avc->descr->descriptors, esd->RegDescriptor);
esd->RegDescriptor = NULL;
}
if (esd->qos) {
gf_list_add(avc->descr->descriptors, esd->qos);
esd->qos = NULL;
}
if (esd->ipiPtr) {
gf_list_add(avc->descr->descriptors, esd->ipiPtr);
esd->ipiPtr= NULL;
}
while (gf_list_count(esd->IPIDataSet)) {
GF_Descriptor *desc = (GF_Descriptor *)gf_list_get(esd->IPIDataSet, 0);
gf_list_rem(esd->IPIDataSet, 0);
gf_list_add(avc->descr->descriptors, desc);
}
while (gf_list_count(esd->IPMPDescriptorPointers)) {
GF_Descriptor *desc = (GF_Descriptor *)gf_list_get(esd->IPMPDescriptorPointers, 0);
gf_list_rem(esd->IPMPDescriptorPointers, 0);
gf_list_add(avc->descr->descriptors, desc);
}
if (esd->langDesc) {
gf_list_add(avc->descr->descriptors, esd->langDesc);
esd->langDesc = NULL;
}
while (gf_list_count(esd->extensionDescriptors)) {
GF_Descriptor *desc = (GF_Descriptor *)gf_list_get(esd->extensionDescriptors, 0);
gf_list_rem(esd->extensionDescriptors, 0);
gf_list_add(avc->descr->descriptors, desc);
}
}
if (!avc->lhvc_config && (esd->decoderConfig->objectTypeIndication==GPAC_OTI_VIDEO_HEVC)) {
if (!avc->hevc_config) avc->hevc_config = (GF_HEVCConfigurationBox *)gf_isom_box_new(GF_ISOM_BOX_TYPE_HVCC);
if (esd->decoderConfig->decoderSpecificInfo && esd->decoderConfig->decoderSpecificInfo->data) {
if (avc->hevc_config->config) gf_odf_hevc_cfg_del(avc->hevc_config->config);
avc->hevc_config->config = gf_odf_hevc_cfg_read(esd->decoderConfig->decoderSpecificInfo->data, esd->decoderConfig->decoderSpecificInfo->dataLength, GF_FALSE);
}
}
else if (!avc->svc_config && !avc->mvc_config && (esd->decoderConfig->objectTypeIndication==GPAC_OTI_VIDEO_AVC)) {
if (!avc->avc_config) avc->avc_config = (GF_AVCConfigurationBox *)gf_isom_box_new(GF_ISOM_BOX_TYPE_AVCC);
if (esd->decoderConfig->decoderSpecificInfo && esd->decoderConfig->decoderSpecificInfo->data) {
if (avc->avc_config->config) gf_odf_avc_cfg_del(avc->avc_config->config);
avc->avc_config->config = gf_odf_avc_cfg_read(esd->decoderConfig->decoderSpecificInfo->data, esd->decoderConfig->decoderSpecificInfo->dataLength);
}
}
gf_odf_desc_del((GF_Descriptor *)esd);
if (avc->hevc_config) {
HEVC_RewriteESDescriptor(avc);
} else {
AVC_RewriteESDescriptor(avc);
}
return GF_OK;
}
#ifndef GPAC_DISABLE_ISOM_WRITE
GF_EXPORT
GF_Err gf_isom_avc_config_new(GF_ISOFile *the_file, u32 trackNumber, GF_AVCConfig *cfg, char *URLname, char *URNname, u32 *outDescriptionIndex)
{
GF_TrackBox *trak;
GF_Err e;
u32 dataRefIndex;
GF_MPEGVisualSampleEntryBox *entry;
e = CanAccessMovie(the_file, GF_ISOM_OPEN_WRITE);
if (e) return e;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !cfg) return GF_BAD_PARAM;
e = Media_FindDataRef(trak->Media->information->dataInformation->dref, URLname, URNname, &dataRefIndex);
if (e) return e;
if (!dataRefIndex) {
e = Media_CreateDataRef(trak->Media->information->dataInformation->dref, URLname, URNname, &dataRefIndex);
if (e) return e;
}
if (!the_file->keep_utc)
trak->Media->mediaHeader->modificationTime = gf_isom_get_mp4time();
entry = (GF_MPEGVisualSampleEntryBox *) gf_isom_box_new(GF_ISOM_BOX_TYPE_AVC1);
if (!entry) return GF_OUT_OF_MEM;
entry->avc_config = (GF_AVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_AVCC);
entry->avc_config->config = AVC_DuplicateConfig(cfg);
entry->dataReferenceIndex = dataRefIndex;
e = gf_list_add(trak->Media->information->sampleTable->SampleDescription->other_boxes, entry);
*outDescriptionIndex = gf_list_count(trak->Media->information->sampleTable->SampleDescription->other_boxes);
AVC_RewriteESDescriptor(entry);
return e;
}
static GF_Err gf_isom_avc_config_update_ex(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, GF_AVCConfig *cfg, u32 op_type)
{
GF_TrackBox *trak;
GF_Err e;
GF_MPEGVisualSampleEntryBox *entry;
e = CanAccessMovie(the_file, GF_ISOM_OPEN_WRITE);
if (e) return e;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return GF_BAD_PARAM;
entry = (GF_MPEGVisualSampleEntryBox *)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return GF_BAD_PARAM;
switch (entry->type) {
case GF_ISOM_BOX_TYPE_AVC1:
case GF_ISOM_BOX_TYPE_AVC2:
case GF_ISOM_BOX_TYPE_AVC3:
case GF_ISOM_BOX_TYPE_AVC4:
case GF_ISOM_BOX_TYPE_SVC1:
case GF_ISOM_BOX_TYPE_MVC1:
break;
default:
return GF_BAD_PARAM;
}
switch (op_type) {
case 0:
if (!cfg) return GF_BAD_PARAM;
if (!entry->avc_config) entry->avc_config = (GF_AVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_AVCC);
if (entry->avc_config->config) gf_odf_avc_cfg_del(entry->avc_config->config);
entry->avc_config->config = AVC_DuplicateConfig(cfg);
entry->type = GF_ISOM_BOX_TYPE_AVC1;
break;
case 1:
if (!cfg) return GF_BAD_PARAM;
if (!entry->svc_config) entry->svc_config = (GF_AVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_SVCC);
if (entry->svc_config->config) gf_odf_avc_cfg_del(entry->svc_config->config);
entry->svc_config->config = AVC_DuplicateConfig(cfg);
entry->type = GF_ISOM_BOX_TYPE_AVC1;
break;
case 2:
if (!cfg) return GF_BAD_PARAM;
if (entry->avc_config) {
gf_isom_box_del((GF_Box*)entry->avc_config);
entry->avc_config = NULL;
}
if (!entry->svc_config) entry->svc_config = (GF_AVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_SVCC);
if (entry->svc_config->config) gf_odf_avc_cfg_del(entry->svc_config->config);
entry->svc_config->config = AVC_DuplicateConfig(cfg);
entry->type = GF_ISOM_BOX_TYPE_SVC1;
break;
case 3:
if (!entry->avc_config || !entry->avc_config->config)
return GF_BAD_PARAM;
if (entry->svc_config) {
gf_isom_box_del((GF_Box*)entry->svc_config);
entry->svc_config = NULL;
}
if (entry->mvc_config) {
gf_isom_box_del((GF_Box*)entry->mvc_config);
entry->mvc_config = NULL;
}
while (gf_list_count(entry->avc_config->config->sequenceParameterSets)) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot*)gf_list_get(entry->avc_config->config->sequenceParameterSets, 0);
gf_list_rem(entry->avc_config->config->sequenceParameterSets, 0);
if (sl->data) gf_free(sl->data);
gf_free(sl);
}
while (gf_list_count(entry->avc_config->config->pictureParameterSets)) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot*)gf_list_get(entry->avc_config->config->pictureParameterSets, 0);
gf_list_rem(entry->avc_config->config->pictureParameterSets, 0);
if (sl->data) gf_free(sl->data);
gf_free(sl);
}
if (entry->type == GF_ISOM_BOX_TYPE_AVC1)
entry->type = GF_ISOM_BOX_TYPE_AVC3;
else if (entry->type == GF_ISOM_BOX_TYPE_AVC2)
entry->type = GF_ISOM_BOX_TYPE_AVC4;
break;
case 4:
if (!cfg) return GF_BAD_PARAM;
if (!entry->mvc_config) entry->mvc_config = (GF_AVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_MVCC);
if (entry->mvc_config->config) gf_odf_avc_cfg_del(entry->mvc_config->config);
entry->mvc_config->config = AVC_DuplicateConfig(cfg);
entry->type = GF_ISOM_BOX_TYPE_AVC1;
break;
case 5:
if (!cfg) return GF_BAD_PARAM;
if (entry->avc_config) {
gf_isom_box_del((GF_Box*)entry->avc_config);
entry->avc_config = NULL;
}
if (!entry->mvc_config) entry->mvc_config = (GF_AVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_MVCC);
if (entry->mvc_config->config) gf_odf_avc_cfg_del(entry->mvc_config->config);
entry->mvc_config->config = AVC_DuplicateConfig(cfg);
entry->type = GF_ISOM_BOX_TYPE_MVC1;
break;
}
AVC_RewriteESDescriptor(entry);
return GF_OK;
}
GF_EXPORT
GF_Err gf_isom_avc_set_inband_config(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
return gf_isom_avc_config_update_ex(the_file, trackNumber, DescriptionIndex, NULL, 3);
}
GF_EXPORT
GF_Err gf_isom_avc_config_update(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, GF_AVCConfig *cfg)
{
return gf_isom_avc_config_update_ex(the_file, trackNumber, DescriptionIndex, cfg, 0);
}
GF_Err gf_isom_svc_config_update(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, GF_AVCConfig *cfg, Bool is_add)
{
return gf_isom_avc_config_update_ex(the_file, trackNumber, DescriptionIndex, cfg, is_add ? 1 : 2);
}
GF_Err gf_isom_mvc_config_update(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, GF_AVCConfig *cfg, Bool is_add)
{
return gf_isom_avc_config_update_ex(the_file, trackNumber, DescriptionIndex, cfg, is_add ? 4 : 5);
}
static GF_Err gf_isom_svc_mvc_config_del(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, Bool is_mvc)
{
GF_TrackBox *trak;
GF_Err e;
GF_MPEGVisualSampleEntryBox *entry;
e = CanAccessMovie(the_file, GF_ISOM_OPEN_WRITE);
if (e) return e;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return GF_BAD_PARAM;
entry = (GF_MPEGVisualSampleEntryBox *)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return GF_BAD_PARAM;
switch (entry->type) {
case GF_ISOM_BOX_TYPE_AVC1:
case GF_ISOM_BOX_TYPE_AVC2:
case GF_ISOM_BOX_TYPE_AVC3:
case GF_ISOM_BOX_TYPE_AVC4:
case GF_ISOM_BOX_TYPE_SVC1:
case GF_ISOM_BOX_TYPE_MVC1:
break;
default:
return GF_BAD_PARAM;
}
if (is_mvc && entry->mvc_config) {
gf_isom_box_del((GF_Box*)entry->mvc_config);
entry->mvc_config = NULL;
}
else if (!is_mvc && entry->svc_config) {
gf_isom_box_del((GF_Box*)entry->svc_config);
entry->svc_config = NULL;
}
AVC_RewriteESDescriptor(entry);
return GF_OK;
}
GF_Err gf_isom_svc_config_del(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
return gf_isom_svc_mvc_config_del(the_file, trackNumber, DescriptionIndex, GF_FALSE);
}
GF_Err gf_isom_mvc_config_del(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
return gf_isom_svc_mvc_config_del(the_file, trackNumber, DescriptionIndex, GF_TRUE);
}
GF_EXPORT
GF_Err gf_isom_set_ipod_compatible(GF_ISOFile *the_file, u32 trackNumber)
{
GF_TrackBox *trak;
GF_Err e;
GF_MPEGVisualSampleEntryBox *entry;
e = CanAccessMovie(the_file, GF_ISOM_OPEN_WRITE);
if (e) return e;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media) return GF_BAD_PARAM;
entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, 0);
if (!entry) return GF_OK;
switch (entry->type) {
case GF_ISOM_BOX_TYPE_AVC1:
case GF_ISOM_BOX_TYPE_AVC2:
case GF_ISOM_BOX_TYPE_AVC3:
case GF_ISOM_BOX_TYPE_AVC4:
case GF_ISOM_BOX_TYPE_SVC1:
case GF_ISOM_BOX_TYPE_MVC1:
case GF_ISOM_BOX_TYPE_HVC1:
case GF_ISOM_BOX_TYPE_HEV1:
case GF_ISOM_BOX_TYPE_HVT1:
break;
default:
return GF_OK;
}
if (!entry->ipod_ext) entry->ipod_ext = (GF_UnknownUUIDBox *) gf_isom_box_new(GF_ISOM_BOX_TYPE_UUID);
memcpy(entry->ipod_ext->uuid, GF_ISOM_IPOD_EXT, sizeof(u8)*16);
entry->ipod_ext->dataSize = 0;
return GF_OK;
}
static GF_Err gf_isom_svc_mvc_config_new(GF_ISOFile *the_file, u32 trackNumber, GF_AVCConfig *cfg, Bool is_mvc, char *URLname, char *URNname, u32 *outDescriptionIndex)
{
GF_TrackBox *trak;
GF_Err e;
u32 dataRefIndex;
GF_MPEGVisualSampleEntryBox *entry;
e = CanAccessMovie(the_file, GF_ISOM_OPEN_WRITE);
if (e) return e;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !cfg) return GF_BAD_PARAM;
e = Media_FindDataRef(trak->Media->information->dataInformation->dref, URLname, URNname, &dataRefIndex);
if (e) return e;
if (!dataRefIndex) {
e = Media_CreateDataRef(trak->Media->information->dataInformation->dref, URLname, URNname, &dataRefIndex);
if (e) return e;
}
if (!the_file->keep_utc)
trak->Media->mediaHeader->modificationTime = gf_isom_get_mp4time();
if (is_mvc) {
entry = (GF_MPEGVisualSampleEntryBox *) gf_isom_box_new(GF_ISOM_BOX_TYPE_MVC1);
if (!entry) return GF_OUT_OF_MEM;
entry->mvc_config = (GF_AVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_MVCC);
entry->mvc_config->config = AVC_DuplicateConfig(cfg);
} else {
entry = (GF_MPEGVisualSampleEntryBox *) gf_isom_box_new(GF_ISOM_BOX_TYPE_SVC1);
if (!entry) return GF_OUT_OF_MEM;
entry->svc_config = (GF_AVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_SVCC);
entry->svc_config->config = AVC_DuplicateConfig(cfg);
}
entry->dataReferenceIndex = dataRefIndex;
e = gf_list_add(trak->Media->information->sampleTable->SampleDescription->other_boxes, entry);
*outDescriptionIndex = gf_list_count(trak->Media->information->sampleTable->SampleDescription->other_boxes);
AVC_RewriteESDescriptor(entry);
return e;
}
GF_EXPORT
GF_Err gf_isom_svc_config_new(GF_ISOFile *the_file, u32 trackNumber, GF_AVCConfig *cfg, char *URLname, char *URNname, u32 *outDescriptionIndex)
{
return gf_isom_svc_mvc_config_new(the_file, trackNumber, cfg, GF_FALSE, URLname, URNname,outDescriptionIndex);
}
GF_EXPORT
GF_Err gf_isom_mvc_config_new(GF_ISOFile *the_file, u32 trackNumber, GF_AVCConfig *cfg, char *URLname, char *URNname, u32 *outDescriptionIndex)
{
return gf_isom_svc_mvc_config_new(the_file, trackNumber, cfg, GF_TRUE, URLname, URNname,outDescriptionIndex);
}
GF_EXPORT
GF_Err gf_isom_hevc_config_new(GF_ISOFile *the_file, u32 trackNumber, GF_HEVCConfig *cfg, char *URLname, char *URNname, u32 *outDescriptionIndex)
{
GF_TrackBox *trak;
GF_Err e;
u32 dataRefIndex;
GF_MPEGVisualSampleEntryBox *entry;
e = CanAccessMovie(the_file, GF_ISOM_OPEN_WRITE);
if (e) return e;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !cfg) return GF_BAD_PARAM;
e = Media_FindDataRef(trak->Media->information->dataInformation->dref, URLname, URNname, &dataRefIndex);
if (e) return e;
if (!dataRefIndex) {
e = Media_CreateDataRef(trak->Media->information->dataInformation->dref, URLname, URNname, &dataRefIndex);
if (e) return e;
}
if (!the_file->keep_utc)
trak->Media->mediaHeader->modificationTime = gf_isom_get_mp4time();
entry = (GF_MPEGVisualSampleEntryBox *) gf_isom_box_new(GF_ISOM_BOX_TYPE_HVC1);
if (!entry) return GF_OUT_OF_MEM;
entry->hevc_config = (GF_HEVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_HVCC);
entry->hevc_config->config = HEVC_DuplicateConfig(cfg);
entry->dataReferenceIndex = dataRefIndex;
e = gf_list_add(trak->Media->information->sampleTable->SampleDescription->other_boxes, entry);
*outDescriptionIndex = gf_list_count(trak->Media->information->sampleTable->SampleDescription->other_boxes);
HEVC_RewriteESDescriptor(entry);
return e;
}
typedef enum
{
GF_ISOM_HVCC_UPDATE = 0,
GF_ISOM_HVCC_SET_INBAND,
GF_ISOM_HVCC_SET_TILE,
GF_ISOM_HVCC_SET_TILE_BASE_TRACK,
GF_ISOM_HVCC_SET_LHVC,
GF_ISOM_HVCC_SET_LHVC_WITH_BASE,
GF_ISOM_HVCC_SET_LHVC_WITH_BASE_BACKWARD,
} HevcConfigUpdateType;
static Bool hevc_cleanup_config(GF_HEVCConfig *cfg, HevcConfigUpdateType operand_type)
{
u32 i;
Bool array_incomplete = (operand_type==GF_ISOM_HVCC_SET_INBAND) ? 1 : 0;
if (!cfg) return 0;
for (i=0; i<gf_list_count(cfg->param_array); i++) {
GF_HEVCParamArray *ar = (GF_HEVCParamArray*)gf_list_get(cfg->param_array, i);
if (operand_type==GF_ISOM_HVCC_SET_INBAND)
ar->array_completeness = 0;
if (!ar->array_completeness) {
array_incomplete = 1;
while (gf_list_count(ar->nalus)) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot*)gf_list_get(ar->nalus, 0);
gf_list_rem(ar->nalus, 0);
if (sl->data) gf_free(sl->data);
gf_free(sl);
}
gf_list_del(ar->nalus);
gf_free(ar);
gf_list_rem(cfg->param_array, i);
i--;
}
}
return array_incomplete;
}
static
GF_Err gf_isom_hevc_config_update_ex(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, GF_HEVCConfig *cfg, u32 operand_type)
{
u32 array_incomplete;
GF_TrackBox *trak;
GF_Err e;
GF_MPEGVisualSampleEntryBox *entry;
e = CanAccessMovie(the_file, GF_ISOM_OPEN_WRITE);
if (e) return e;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return GF_BAD_PARAM;
entry = (GF_MPEGVisualSampleEntryBox *)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return GF_BAD_PARAM;
switch (entry->type) {
case GF_ISOM_BOX_TYPE_HVC1:
case GF_ISOM_BOX_TYPE_HEV1:
case GF_ISOM_BOX_TYPE_HVC2:
case GF_ISOM_BOX_TYPE_HEV2:
case GF_ISOM_BOX_TYPE_LHV1:
case GF_ISOM_BOX_TYPE_LHE1:
case GF_ISOM_BOX_TYPE_HVT1:
break;
default:
return GF_BAD_PARAM;
}
if (operand_type == GF_ISOM_HVCC_SET_TILE_BASE_TRACK) {
if (entry->type==GF_ISOM_BOX_TYPE_HVC1)
entry->type = GF_ISOM_BOX_TYPE_HVC2;
else if (entry->type==GF_ISOM_BOX_TYPE_HEV1)
entry->type = GF_ISOM_BOX_TYPE_HEV2;
} else if (operand_type == GF_ISOM_HVCC_SET_TILE) {
if (!entry->hevc_config) entry->hevc_config = (GF_HEVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_HVCC);
if (entry->hevc_config->config) gf_odf_hevc_cfg_del(entry->hevc_config->config);
entry->hevc_config->config = NULL;
entry->type = GF_ISOM_BOX_TYPE_HVT1;
} else if (operand_type < GF_ISOM_HVCC_SET_LHVC) {
if ((operand_type != GF_ISOM_HVCC_SET_INBAND) && !entry->hevc_config)
entry->hevc_config = (GF_HEVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_HVCC);
if (cfg) {
if (entry->hevc_config->config) gf_odf_hevc_cfg_del(entry->hevc_config->config);
entry->hevc_config->config = HEVC_DuplicateConfig(cfg);
} else if (operand_type != GF_ISOM_HVCC_SET_TILE) {
operand_type=GF_ISOM_HVCC_SET_INBAND;
}
array_incomplete = (operand_type==GF_ISOM_HVCC_SET_INBAND) ? 1 : 0;
if (entry->hevc_config && hevc_cleanup_config(entry->hevc_config->config, operand_type))
array_incomplete=1;
if (entry->lhvc_config && hevc_cleanup_config(entry->lhvc_config->config, operand_type))
array_incomplete=1;
switch (entry->type) {
case GF_ISOM_BOX_TYPE_HEV1:
case GF_ISOM_BOX_TYPE_HVC1:
entry->type = array_incomplete ? GF_ISOM_BOX_TYPE_HEV1 : GF_ISOM_BOX_TYPE_HVC1;
break;
case GF_ISOM_BOX_TYPE_HEV2:
case GF_ISOM_BOX_TYPE_HVC2:
entry->type = array_incomplete ? GF_ISOM_BOX_TYPE_HEV2 : GF_ISOM_BOX_TYPE_HVC2;
break;
case GF_ISOM_BOX_TYPE_LHE1:
case GF_ISOM_BOX_TYPE_LHV1:
entry->type = array_incomplete ? GF_ISOM_BOX_TYPE_LHE1 : GF_ISOM_BOX_TYPE_LHV1;
break;
}
} else {
if ((operand_type==GF_ISOM_HVCC_SET_LHVC_WITH_BASE_BACKWARD) || (operand_type==GF_ISOM_HVCC_SET_LHVC_WITH_BASE)) {
if (!entry->hevc_config) return GF_BAD_PARAM;
if (!cfg) {
if (entry->lhvc_config) {
gf_isom_box_del((GF_Box*)entry->lhvc_config);
entry->lhvc_config = NULL;
}
if (entry->type==GF_ISOM_BOX_TYPE_LHE1) entry->type = (operand_type==GF_ISOM_HVCC_SET_LHVC_WITH_BASE) ? GF_ISOM_BOX_TYPE_HEV2 : GF_ISOM_BOX_TYPE_HEV1;
else if (entry->type==GF_ISOM_BOX_TYPE_HEV1) entry->type = (operand_type==GF_ISOM_HVCC_SET_LHVC_WITH_BASE) ? GF_ISOM_BOX_TYPE_HEV2 : GF_ISOM_BOX_TYPE_HEV1;
else entry->type = (operand_type==GF_ISOM_HVCC_SET_LHVC_WITH_BASE) ? GF_ISOM_BOX_TYPE_HVC2 : GF_ISOM_BOX_TYPE_HVC1;
} else {
if (!entry->lhvc_config) entry->lhvc_config = (GF_HEVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_LHVC);
if (entry->lhvc_config->config) gf_odf_hevc_cfg_del(entry->lhvc_config->config);
entry->lhvc_config->config = HEVC_DuplicateConfig(cfg);
if (operand_type==GF_ISOM_HVCC_SET_LHVC_WITH_BASE_BACKWARD) {
if (entry->type==GF_ISOM_BOX_TYPE_HEV2) entry->type = GF_ISOM_BOX_TYPE_HEV1;
else entry->type = GF_ISOM_BOX_TYPE_HVC1;
} else {
if (entry->type==GF_ISOM_BOX_TYPE_HEV1) entry->type = GF_ISOM_BOX_TYPE_HEV2;
else entry->type = GF_ISOM_BOX_TYPE_HVC2;
}
}
}
else if (operand_type==GF_ISOM_HVCC_SET_LHVC) {
if (entry->hevc_config) {
gf_isom_box_del((GF_Box*)entry->hevc_config);
entry->hevc_config=NULL;
}
if (!cfg) return GF_BAD_PARAM;
if (!entry->lhvc_config) entry->lhvc_config = (GF_HEVCConfigurationBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_LHVC);
if (entry->lhvc_config->config) gf_odf_hevc_cfg_del(entry->lhvc_config->config);
entry->lhvc_config->config = HEVC_DuplicateConfig(cfg);
if ((entry->type==GF_ISOM_BOX_TYPE_HEV1) || (entry->type==GF_ISOM_BOX_TYPE_HEV2)) entry->type = GF_ISOM_BOX_TYPE_LHE1;
else entry->type = GF_ISOM_BOX_TYPE_LHV1;
}
}
HEVC_RewriteESDescriptor(entry);
return GF_OK;
}
GF_EXPORT
GF_Err gf_isom_hevc_config_update(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, GF_HEVCConfig *cfg)
{
return gf_isom_hevc_config_update_ex(the_file, trackNumber, DescriptionIndex, cfg, GF_ISOM_HVCC_UPDATE);
}
GF_EXPORT
GF_Err gf_isom_hevc_set_inband_config(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
return gf_isom_hevc_config_update_ex(the_file, trackNumber, DescriptionIndex, NULL, GF_ISOM_HVCC_SET_INBAND);
}
GF_EXPORT
GF_Err gf_isom_hevc_set_tile_config(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, GF_HEVCConfig *cfg, Bool is_base_track)
{
return gf_isom_hevc_config_update_ex(the_file, trackNumber, DescriptionIndex, cfg, is_base_track ? GF_ISOM_HVCC_SET_TILE_BASE_TRACK : GF_ISOM_HVCC_SET_TILE);
}
GF_Err gf_isom_lhvc_config_update(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex, GF_HEVCConfig *cfg, GF_ISOMLHEVCTrackType track_type)
{
if (cfg) cfg->is_lhvc = GF_TRUE;
switch (track_type) {
case GF_ISOM_LEHVC_ONLY:
return gf_isom_hevc_config_update_ex(the_file, trackNumber, DescriptionIndex, cfg, GF_ISOM_HVCC_SET_LHVC);
case GF_ISOM_LEHVC_WITH_BASE:
return gf_isom_hevc_config_update_ex(the_file, trackNumber, DescriptionIndex, cfg, GF_ISOM_HVCC_SET_LHVC_WITH_BASE);
case GF_ISOM_LEHVC_WITH_BASE_BACKWARD:
return gf_isom_hevc_config_update_ex(the_file, trackNumber, DescriptionIndex, cfg, GF_ISOM_HVCC_SET_LHVC_WITH_BASE_BACKWARD);
default:
return GF_BAD_PARAM;
}
}
#endif
GF_EXPORT
GF_Box *gf_isom_clone_config_box(GF_Box *box)
{
GF_Box *clone;
switch (box->type)
{
case GF_ISOM_BOX_TYPE_AVCC:
case GF_ISOM_BOX_TYPE_SVCC:
case GF_ISOM_BOX_TYPE_MVCC:
clone = gf_isom_box_new(box->type);
((GF_AVCConfigurationBox *)clone)->config = AVC_DuplicateConfig(((GF_AVCConfigurationBox *)box)->config);
break;
case GF_ISOM_BOX_TYPE_HVCC:
clone = gf_isom_box_new(box->type);
((GF_HEVCConfigurationBox *)clone)->config = HEVC_DuplicateConfig(((GF_HEVCConfigurationBox *)box)->config);
break;
default:
clone = NULL;
GF_LOG(GF_LOG_WARNING, GF_LOG_CONTAINER, ("Cloning of config not supported for type %s\n", gf_4cc_to_str(box->type)));
break;
}
return clone;
}
GF_EXPORT
GF_AVCConfig *gf_isom_avc_config_get(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
GF_TrackBox *trak;
GF_MPEGVisualSampleEntryBox *entry;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return NULL;
if (gf_isom_get_avc_svc_type(the_file, trackNumber, DescriptionIndex)==GF_ISOM_AVCTYPE_NONE)
return NULL;
entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return NULL;
if (!entry->avc_config) return NULL;
return AVC_DuplicateConfig(entry->avc_config->config);
}
GF_EXPORT
GF_HEVCConfig *gf_isom_hevc_config_get(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
GF_TrackBox *trak;
GF_MPEGVisualSampleEntryBox *entry;
if (gf_isom_get_reference_count(the_file, trackNumber, GF_ISOM_REF_TBAS)) {
u32 ref_track;
GF_Err e = gf_isom_get_reference(the_file, trackNumber, GF_ISOM_REF_TBAS, 1, &ref_track);
if (e == GF_OK) {
trackNumber = ref_track;
}
}
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return NULL;
if (gf_isom_get_hevc_lhvc_type(the_file, trackNumber, DescriptionIndex)==GF_ISOM_HEVCTYPE_NONE)
return NULL;
entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return NULL;
if (!entry->hevc_config) return NULL;
return HEVC_DuplicateConfig(entry->hevc_config->config);
}
GF_EXPORT
GF_AVCConfig *gf_isom_svc_config_get(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
GF_TrackBox *trak;
GF_MPEGVisualSampleEntryBox *entry;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return NULL;
if (gf_isom_get_avc_svc_type(the_file, trackNumber, DescriptionIndex)==GF_ISOM_AVCTYPE_NONE)
return NULL;
entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return NULL;
if (!entry->svc_config) return NULL;
return AVC_DuplicateConfig(entry->svc_config->config);
}
GF_EXPORT
GF_AVCConfig *gf_isom_mvc_config_get(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
GF_TrackBox *trak;
GF_MPEGVisualSampleEntryBox *entry;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return NULL;
if (gf_isom_get_avc_svc_type(the_file, trackNumber, DescriptionIndex)==GF_ISOM_AVCTYPE_NONE)
return NULL;
entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return NULL;
if (!entry->mvc_config) return NULL;
return AVC_DuplicateConfig(entry->mvc_config->config);
}
GF_EXPORT
u32 gf_isom_get_avc_svc_type(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
u32 type;
GF_TrackBox *trak;
GF_MPEGVisualSampleEntryBox *entry;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return GF_ISOM_AVCTYPE_NONE;
if (trak->Media->handler->handlerType != GF_ISOM_MEDIA_VISUAL) return GF_ISOM_AVCTYPE_NONE;
entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return GF_ISOM_AVCTYPE_NONE;
type = entry->type;
if (type == GF_ISOM_BOX_TYPE_ENCV) {
GF_ProtectionSchemeInfoBox *sinf = (GF_ProtectionSchemeInfoBox *) gf_list_get(entry->protections, 0);
if (sinf && sinf->original_format) type = sinf->original_format->data_format;
}
else if (type == GF_ISOM_BOX_TYPE_RESV) {
if (entry->rinf && entry->rinf->original_format) type = entry->rinf->original_format->data_format;
}
switch (type) {
case GF_ISOM_BOX_TYPE_AVC1:
case GF_ISOM_BOX_TYPE_AVC2:
case GF_ISOM_BOX_TYPE_AVC3:
case GF_ISOM_BOX_TYPE_AVC4:
case GF_ISOM_BOX_TYPE_SVC1:
case GF_ISOM_BOX_TYPE_MVC1:
break;
default:
return GF_ISOM_AVCTYPE_NONE;
}
if (entry->avc_config && !entry->svc_config && !entry->mvc_config) return GF_ISOM_AVCTYPE_AVC_ONLY;
if (entry->avc_config && entry->svc_config) return GF_ISOM_AVCTYPE_AVC_SVC;
if (entry->avc_config && entry->mvc_config) return GF_ISOM_AVCTYPE_AVC_MVC;
if (!entry->avc_config && entry->svc_config) return GF_ISOM_AVCTYPE_SVC_ONLY;
if (!entry->avc_config && entry->mvc_config) return GF_ISOM_AVCTYPE_MVC_ONLY;
return GF_ISOM_AVCTYPE_NONE;
}
GF_EXPORT
u32 gf_isom_get_hevc_lhvc_type(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
u32 type;
GF_TrackBox *trak;
GF_MPEGVisualSampleEntryBox *entry;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return GF_ISOM_HEVCTYPE_NONE;
if (trak->Media->handler->handlerType != GF_ISOM_MEDIA_VISUAL) return GF_ISOM_HEVCTYPE_NONE;
entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return GF_ISOM_AVCTYPE_NONE;
type = entry->type;
if (type == GF_ISOM_BOX_TYPE_ENCV) {
GF_ProtectionSchemeInfoBox *sinf = (GF_ProtectionSchemeInfoBox *) gf_list_get(entry->protections, 0);
if (sinf && sinf->original_format) type = sinf->original_format->data_format;
}
else if (type == GF_ISOM_BOX_TYPE_RESV) {
if (entry->rinf && entry->rinf->original_format) type = entry->rinf->original_format->data_format;
}
switch (type) {
case GF_ISOM_BOX_TYPE_HVC1:
case GF_ISOM_BOX_TYPE_HEV1:
case GF_ISOM_BOX_TYPE_HVC2:
case GF_ISOM_BOX_TYPE_HEV2:
case GF_ISOM_BOX_TYPE_LHV1:
case GF_ISOM_BOX_TYPE_LHE1:
case GF_ISOM_BOX_TYPE_HVT1:
break;
default:
return GF_ISOM_HEVCTYPE_NONE;
}
if (entry->hevc_config && !entry->lhvc_config) return GF_ISOM_HEVCTYPE_HEVC_ONLY;
if (entry->hevc_config && entry->lhvc_config) return GF_ISOM_HEVCTYPE_HEVC_LHVC;
if (!entry->hevc_config && entry->lhvc_config) return GF_ISOM_HEVCTYPE_LHVC_ONLY;
return GF_ISOM_HEVCTYPE_NONE;
}
GF_EXPORT
GF_HEVCConfig *gf_isom_lhvc_config_get(GF_ISOFile *the_file, u32 trackNumber, u32 DescriptionIndex)
{
GF_HEVCConfig *lhvc;
GF_OperatingPointsInformation *oinf=NULL;
GF_TrackBox *trak;
GF_MPEGVisualSampleEntryBox *entry;
trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak || !trak->Media || !DescriptionIndex) return NULL;
if (gf_isom_get_hevc_lhvc_type(the_file, trackNumber, DescriptionIndex)==GF_ISOM_HEVCTYPE_NONE)
return NULL;
entry = (GF_MPEGVisualSampleEntryBox*)gf_list_get(trak->Media->information->sampleTable->SampleDescription->other_boxes, DescriptionIndex-1);
if (!entry) return NULL;
if (!entry->lhvc_config) return NULL;
lhvc = HEVC_DuplicateConfig(entry->lhvc_config->config);
if (!lhvc) return NULL;
gf_isom_get_oinf_info(the_file, trackNumber, &oinf);
if (oinf) {
LHEVC_ProfileTierLevel *ptl = (LHEVC_ProfileTierLevel *)gf_list_last(oinf->profile_tier_levels);
if (ptl) {
lhvc->profile_space = ptl->general_profile_space;
lhvc->tier_flag = ptl->general_tier_flag;
lhvc->profile_idc = ptl->general_profile_idc;
lhvc->general_profile_compatibility_flags = ptl->general_profile_compatibility_flags;
lhvc->constraint_indicator_flags = ptl->general_constraint_indicator_flags;
}
}
return lhvc;
}
void btrt_del(GF_Box *s)
{
GF_BitRateBox *ptr = (GF_BitRateBox *)s;
if (ptr) gf_free(ptr);
}
GF_Err btrt_Read(GF_Box *s, GF_BitStream *bs)
{
GF_BitRateBox *ptr = (GF_BitRateBox *)s;
ptr->bufferSizeDB = gf_bs_read_u32(bs);
ptr->maxBitrate = gf_bs_read_u32(bs);
ptr->avgBitrate = gf_bs_read_u32(bs);
return GF_OK;
}
GF_Box *btrt_New()
{
GF_BitRateBox *tmp = (GF_BitRateBox *) gf_malloc(sizeof(GF_BitRateBox));
if (tmp == NULL) return NULL;
memset(tmp, 0, sizeof(GF_BitRateBox));
tmp->type = GF_ISOM_BOX_TYPE_BTRT;
return (GF_Box *)tmp;
}
#ifndef GPAC_DISABLE_ISOM_WRITE
GF_Err btrt_Write(GF_Box *s, GF_BitStream *bs)
{
GF_Err e;
GF_BitRateBox *ptr = (GF_BitRateBox *) s;
if (!s) return GF_BAD_PARAM;
e = gf_isom_box_write_header(s, bs);
if (e) return e;
gf_bs_write_u32(bs, ptr->bufferSizeDB);
gf_bs_write_u32(bs, ptr->maxBitrate);
gf_bs_write_u32(bs, ptr->avgBitrate);
return GF_OK;
}
GF_Err btrt_Size(GF_Box *s)
{
GF_BitRateBox *ptr = (GF_BitRateBox *)s;
ptr->size += 12;
return GF_OK;
}
#endif
void m4ds_del(GF_Box *s)
{
GF_MPEG4ExtensionDescriptorsBox *ptr = (GF_MPEG4ExtensionDescriptorsBox *)s;
gf_odf_desc_list_del(ptr->descriptors);
gf_list_del(ptr->descriptors);
gf_free(ptr);
}
GF_Err m4ds_Read(GF_Box *s, GF_BitStream *bs)
{
GF_Err e;
char *enc_od;
GF_MPEG4ExtensionDescriptorsBox *ptr = (GF_MPEG4ExtensionDescriptorsBox *)s;
u32 od_size = (u32) ptr->size;
if (!od_size) return GF_OK;
enc_od = (char *)gf_malloc(sizeof(char) * od_size);
gf_bs_read_data(bs, enc_od, od_size);
e = gf_odf_desc_list_read((char *)enc_od, od_size, ptr->descriptors);
gf_free(enc_od);
return e;
}
GF_Box *m4ds_New()
{
GF_MPEG4ExtensionDescriptorsBox *tmp = (GF_MPEG4ExtensionDescriptorsBox *) gf_malloc(sizeof(GF_MPEG4ExtensionDescriptorsBox));
if (tmp == NULL) return NULL;
memset(tmp, 0, sizeof(GF_MPEG4ExtensionDescriptorsBox));
tmp->type = GF_ISOM_BOX_TYPE_M4DS;
tmp->descriptors = gf_list_new();
return (GF_Box *)tmp;
}
#ifndef GPAC_DISABLE_ISOM_WRITE
GF_Err m4ds_Write(GF_Box *s, GF_BitStream *bs)
{
GF_Err e;
char *enc_ods;
u32 enc_od_size;
GF_MPEG4ExtensionDescriptorsBox *ptr = (GF_MPEG4ExtensionDescriptorsBox *) s;
if (!s) return GF_BAD_PARAM;
e = gf_isom_box_write_header(s, bs);
if (e) return e;
enc_ods = NULL;
enc_od_size = 0;
e = gf_odf_desc_list_write(ptr->descriptors, &enc_ods, &enc_od_size);
if (e) return e;
if (enc_od_size) {
gf_bs_write_data(bs, enc_ods, enc_od_size);
gf_free(enc_ods);
}
return GF_OK;
}
GF_Err m4ds_Size(GF_Box *s)
{
GF_Err e;
u32 descSize = 0;
GF_MPEG4ExtensionDescriptorsBox *ptr = (GF_MPEG4ExtensionDescriptorsBox *)s;
e = gf_odf_desc_list_size(ptr->descriptors, &descSize);
ptr->size += descSize;
return e;
}
#endif
void avcc_del(GF_Box *s)
{
GF_AVCConfigurationBox *ptr = (GF_AVCConfigurationBox *)s;
if (ptr->config) gf_odf_avc_cfg_del(ptr->config);
gf_free(ptr);
}
GF_Err avcc_Read(GF_Box *s, GF_BitStream *bs)
{
u32 i, count;
GF_AVCConfigurationBox *ptr = (GF_AVCConfigurationBox *)s;
if (ptr->config) gf_odf_avc_cfg_del(ptr->config);
ptr->config = gf_odf_avc_cfg_new();
ptr->config->configurationVersion = gf_bs_read_u8(bs);
ptr->config->AVCProfileIndication = gf_bs_read_u8(bs);
ptr->config->profile_compatibility = gf_bs_read_u8(bs);
ptr->config->AVCLevelIndication = gf_bs_read_u8(bs);
if (ptr->type==GF_ISOM_BOX_TYPE_AVCC) {
gf_bs_read_int(bs, 6);
} else {
ptr->config->complete_representation = gf_bs_read_int(bs, 1);
gf_bs_read_int(bs, 5);
}
ptr->config->nal_unit_size = 1 + gf_bs_read_int(bs, 2);
gf_bs_read_int(bs, 3);
count = gf_bs_read_int(bs, 5);
ptr->size -= 7;
for (i=0; i<count; i++) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot *) gf_malloc(sizeof(GF_AVCConfigSlot));
sl->size = gf_bs_read_u16(bs);
sl->data = (char *)gf_malloc(sizeof(char) * sl->size);
gf_bs_read_data(bs, sl->data, sl->size);
gf_list_add(ptr->config->sequenceParameterSets, sl);
ptr->size -= 2+sl->size;
}
count = gf_bs_read_u8(bs);
for (i=0; i<count; i++) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot *)gf_malloc(sizeof(GF_AVCConfigSlot));
sl->size = gf_bs_read_u16(bs);
if (gf_bs_available(bs) < sl->size) {
gf_free(sl);
GF_LOG(GF_LOG_ERROR, GF_LOG_CODING, ("AVCC: Not enough bits to parse. Aborting.\n"));
return GF_ISOM_INVALID_FILE;
}
sl->data = (char *)gf_malloc(sizeof(char) * sl->size);
gf_bs_read_data(bs, sl->data, sl->size);
gf_list_add(ptr->config->pictureParameterSets, sl);
ptr->size -= 2+sl->size;
}
if (ptr->type==GF_ISOM_BOX_TYPE_AVCC) {
if (gf_avc_is_rext_profile(ptr->config->AVCProfileIndication)) {
if (!ptr->size) {
#ifndef GPAC_DISABLE_AV_PARSERS
AVCState avc;
s32 idx, vui_flag_pos;
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot*)gf_list_get(ptr->config->sequenceParameterSets, 0);
idx = gf_media_avc_read_sps(sl->data+1, sl->size-1, &avc, 0, (u32 *) &vui_flag_pos);
if (idx>=0) {
ptr->config->chroma_format = avc.sps[idx].chroma_format;
ptr->config->luma_bit_depth = 8 + avc.sps[idx].luma_bit_depth_m8;
ptr->config->chroma_bit_depth = 8 + avc.sps[idx].chroma_bit_depth_m8;
}
#else
ptr->config->chroma_format = 1;
ptr->config->luma_bit_depth = 8;
ptr->config->chroma_bit_depth = 8;
#endif
return GF_OK;
}
gf_bs_read_int(bs, 6);
ptr->config->chroma_format = gf_bs_read_int(bs, 2);
gf_bs_read_int(bs, 5);
ptr->config->luma_bit_depth = 8 + gf_bs_read_int(bs, 3);
gf_bs_read_int(bs, 5);
ptr->config->chroma_bit_depth = 8 + gf_bs_read_int(bs, 3);
count = gf_bs_read_int(bs, 8);
ptr->size -= 4;
if (count*2 > ptr->size) {
GF_LOG(GF_LOG_WARNING, GF_LOG_CODING, ("AVCC: invalid numOfSequenceParameterSetExt value. Skipping.\n"));
return GF_OK;
}
if (count) {
ptr->config->sequenceParameterSetExtensions = gf_list_new();
for (i=0; i<count; i++) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot *)gf_malloc(sizeof(GF_AVCConfigSlot));
sl->size = gf_bs_read_u16(bs);
if (gf_bs_available(bs) < sl->size) {
gf_free(sl);
GF_LOG(GF_LOG_ERROR, GF_LOG_CODING, ("AVCC: Not enough bits to parse. Aborting.\n"));
return GF_ISOM_INVALID_FILE;
}
sl->data = (char *)gf_malloc(sizeof(char) * sl->size);
gf_bs_read_data(bs, sl->data, sl->size);
gf_list_add(ptr->config->sequenceParameterSetExtensions, sl);
ptr->size -= sl->size + 2;
}
}
}
}
return GF_OK;
}
GF_Box *avcc_New()
{
GF_AVCConfigurationBox *tmp = (GF_AVCConfigurationBox *) gf_malloc(sizeof(GF_AVCConfigurationBox));
if (tmp == NULL) return NULL;
memset(tmp, 0, sizeof(GF_AVCConfigurationBox));
tmp->type = GF_ISOM_BOX_TYPE_AVCC;
return (GF_Box *)tmp;
}
#ifndef GPAC_DISABLE_ISOM_WRITE
GF_Err avcc_Write(GF_Box *s, GF_BitStream *bs)
{
u32 i, count;
GF_Err e;
GF_AVCConfigurationBox *ptr = (GF_AVCConfigurationBox *) s;
if (!s) return GF_BAD_PARAM;
if (!ptr->config) return GF_OK;
e = gf_isom_box_write_header(s, bs);
if (e) return e;
gf_bs_write_u8(bs, ptr->config->configurationVersion);
gf_bs_write_u8(bs, ptr->config->AVCProfileIndication);
gf_bs_write_u8(bs, ptr->config->profile_compatibility);
gf_bs_write_u8(bs, ptr->config->AVCLevelIndication);
if (ptr->type==GF_ISOM_BOX_TYPE_AVCC) {
gf_bs_write_int(bs, 0x3F, 6);
} else {
gf_bs_write_int(bs, ptr->config->complete_representation, 1);
gf_bs_write_int(bs, 0x1F, 5);
}
gf_bs_write_int(bs, ptr->config->nal_unit_size - 1, 2);
gf_bs_write_int(bs, 0x7, 3);
count = gf_list_count(ptr->config->sequenceParameterSets);
gf_bs_write_int(bs, count, 5);
for (i=0; i<count; i++) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot *) gf_list_get(ptr->config->sequenceParameterSets, i);
gf_bs_write_u16(bs, sl->size);
gf_bs_write_data(bs, sl->data, sl->size);
}
count = gf_list_count(ptr->config->pictureParameterSets);
gf_bs_write_u8(bs, count);
for (i=0; i<count; i++) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot *) gf_list_get(ptr->config->pictureParameterSets, i);
gf_bs_write_u16(bs, sl->size);
gf_bs_write_data(bs, sl->data, sl->size);
}
if (ptr->type==GF_ISOM_BOX_TYPE_AVCC) {
if (gf_avc_is_rext_profile(ptr->config->AVCProfileIndication)) {
gf_bs_write_int(bs, 0xFF, 6);
gf_bs_write_int(bs, ptr->config->chroma_format, 2);
gf_bs_write_int(bs, 0xFF, 5);
gf_bs_write_int(bs, ptr->config->luma_bit_depth - 8, 3);
gf_bs_write_int(bs, 0xFF, 5);
gf_bs_write_int(bs, ptr->config->chroma_bit_depth - 8, 3);
count = ptr->config->sequenceParameterSetExtensions ? gf_list_count(ptr->config->sequenceParameterSetExtensions) : 0;
gf_bs_write_u8(bs, count);
for (i=0; i<count; i++) {
GF_AVCConfigSlot *sl = (GF_AVCConfigSlot *) gf_list_get(ptr->config->sequenceParameterSetExtensions, i);
gf_bs_write_u16(bs, sl->size);
gf_bs_write_data(bs, sl->data, sl->size);
}
}
}
return GF_OK;
}
GF_Err avcc_Size(GF_Box *s)
{
u32 i, count;
GF_AVCConfigurationBox *ptr = (GF_AVCConfigurationBox *)s;
if (!ptr->config) {
ptr->size = 0;
return GF_OK;
}
ptr->size += 7;
count = gf_list_count(ptr->config->sequenceParameterSets);
for (i=0; i<count; i++)
ptr->size += 2 + ((GF_AVCConfigSlot *)gf_list_get(ptr->config->sequenceParameterSets, i))->size;
count = gf_list_count(ptr->config->pictureParameterSets);
for (i=0; i<count; i++)
ptr->size += 2 + ((GF_AVCConfigSlot *)gf_list_get(ptr->config->pictureParameterSets, i))->size;
if (ptr->type==GF_ISOM_BOX_TYPE_AVCC) {
if (gf_avc_is_rext_profile(ptr->config->AVCProfileIndication)) {
ptr->size += 4;
count = ptr->config->sequenceParameterSetExtensions ?gf_list_count(ptr->config->sequenceParameterSetExtensions) : 0;
for (i=0; i<count; i++)
ptr->size += 2 + ((GF_AVCConfigSlot *)gf_list_get(ptr->config->sequenceParameterSetExtensions, i))->size;
}
}
return GF_OK;
}
#endif
void hvcc_del(GF_Box *s)
{
GF_HEVCConfigurationBox *ptr = (GF_HEVCConfigurationBox*)s;
if (ptr->config) gf_odf_hevc_cfg_del(ptr->config);
gf_free(ptr);
}
GF_Err hvcc_Read(GF_Box *s, GF_BitStream *bs)
{
u64 pos;
GF_HEVCConfigurationBox *ptr = (GF_HEVCConfigurationBox *)s;
if (ptr->config) gf_odf_hevc_cfg_del(ptr->config);
pos = gf_bs_get_position(bs);
ptr->config = gf_odf_hevc_cfg_read_bs(bs, (s->type == GF_ISOM_BOX_TYPE_HVCC) ? GF_FALSE : GF_TRUE);
pos = gf_bs_get_position(bs) - pos ;
if (pos < ptr->size)
ptr->size -= (u32) pos;
return ptr->config ? GF_OK : GF_ISOM_INVALID_FILE;
}
GF_Box *hvcc_New()
{
GF_HEVCConfigurationBox *tmp = (GF_HEVCConfigurationBox *) gf_malloc(sizeof(GF_HEVCConfigurationBox));
if (tmp == NULL) return NULL;
memset(tmp, 0, sizeof(GF_HEVCConfigurationBox));
tmp->type = GF_ISOM_BOX_TYPE_HVCC;
return (GF_Box *)tmp;
}
#ifndef GPAC_DISABLE_ISOM_WRITE
GF_Err hvcc_Write(GF_Box *s, GF_BitStream *bs)
{
GF_Err e;
GF_HEVCConfigurationBox *ptr = (GF_HEVCConfigurationBox *) s;
if (!s) return GF_BAD_PARAM;
if (!ptr->config) return GF_OK;
e = gf_isom_box_write_header(s, bs);
if (e) return e;
return gf_odf_hevc_cfg_write_bs(ptr->config, bs);
}
GF_Err hvcc_Size(GF_Box *s)
{
u32 i, count, j, subcount;
GF_HEVCConfigurationBox *ptr = (GF_HEVCConfigurationBox *)s;
if (!ptr->config) {
ptr->size = 0;
return GF_OK;
}
if (!ptr->config->is_lhvc)
ptr->size += 23;
else
ptr->size += 6;
count = gf_list_count(ptr->config->param_array);
for (i=0; i<count; i++) {
GF_HEVCParamArray *ar = (GF_HEVCParamArray*)gf_list_get(ptr->config->param_array, i);
ptr->size += 3;
subcount = gf_list_count(ar->nalus);
for (j=0; j<subcount; j++) {
ptr->size += 2 + ((GF_AVCConfigSlot *)gf_list_get(ar->nalus, j))->size;
}
}
return GF_OK;
}
#endif
GF_OperatingPointsInformation *gf_isom_oinf_new_entry()
{
GF_OperatingPointsInformation* ptr;
GF_SAFEALLOC(ptr, GF_OperatingPointsInformation);
if (ptr) {
ptr->profile_tier_levels = gf_list_new();
ptr->operating_points = gf_list_new();
ptr->dependency_layers = gf_list_new();
}
return ptr;
}
void gf_isom_oinf_del_entry(void *entry)
{
GF_OperatingPointsInformation* ptr = (GF_OperatingPointsInformation *)entry;
if (!ptr) return;
if (ptr->profile_tier_levels) {
while (gf_list_count(ptr->profile_tier_levels)) {
LHEVC_ProfileTierLevel *ptl = (LHEVC_ProfileTierLevel *)gf_list_get(ptr->profile_tier_levels, 0);
gf_free(ptl);
gf_list_rem(ptr->profile_tier_levels, 0);
}
gf_list_del(ptr->profile_tier_levels);
}
if (ptr->operating_points) {
while (gf_list_count(ptr->operating_points)) {
LHEVC_OperatingPoint *op = (LHEVC_OperatingPoint *)gf_list_get(ptr->operating_points, 0);
gf_free(op);
gf_list_rem(ptr->operating_points, 0);
}
gf_list_del(ptr->operating_points);
}
if (ptr->dependency_layers) {
while (gf_list_count(ptr->dependency_layers)) {
LHEVC_DependentLayer *dep = (LHEVC_DependentLayer *)gf_list_get(ptr->dependency_layers, 0);
gf_free(dep);
gf_list_rem(ptr->dependency_layers, 0);
}
gf_list_del(ptr->dependency_layers);
}
gf_free(ptr);
return;
}
GF_Err gf_isom_oinf_read_entry(void *entry, GF_BitStream *bs)
{
GF_OperatingPointsInformation* ptr = (GF_OperatingPointsInformation *)entry;
u32 i, j, count;
if (!ptr) return GF_BAD_PARAM;
ptr->scalability_mask = gf_bs_read_u16(bs);
gf_bs_read_int(bs, 2);
count = gf_bs_read_int(bs, 6);
for (i = 0; i < count; i++) {
LHEVC_ProfileTierLevel *ptl;
GF_SAFEALLOC(ptl, LHEVC_ProfileTierLevel);
if (!ptl) return GF_OUT_OF_MEM;
ptl->general_profile_space = gf_bs_read_int(bs, 2);
ptl->general_tier_flag= gf_bs_read_int(bs, 1);
ptl->general_profile_idc = gf_bs_read_int(bs, 5);
ptl->general_profile_compatibility_flags = gf_bs_read_u32(bs);
ptl->general_constraint_indicator_flags = gf_bs_read_long_int(bs, 48);
ptl->general_level_idc = gf_bs_read_u8(bs);
gf_list_add(ptr->profile_tier_levels, ptl);
}
count = gf_bs_read_u16(bs);
for (i = 0; i < count; i++) {
LHEVC_OperatingPoint *op;
GF_SAFEALLOC(op, LHEVC_OperatingPoint);
if (!op) return GF_OUT_OF_MEM;
op->output_layer_set_idx = gf_bs_read_u16(bs);
op->max_temporal_id = gf_bs_read_u8(bs);
op->layer_count = gf_bs_read_u8(bs);
for (j = 0; j < op->layer_count; j++) {
op->layers_info[j].ptl_idx = gf_bs_read_u8(bs);
op->layers_info[j].layer_id = gf_bs_read_int(bs, 6);
op->layers_info[j].is_outputlayer = gf_bs_read_int(bs, 1) ? GF_TRUE : GF_FALSE;
op->layers_info[j].is_alternate_outputlayer = gf_bs_read_int(bs, 1) ? GF_TRUE : GF_FALSE;
}
op->minPicWidth = gf_bs_read_u16(bs);
op->minPicHeight = gf_bs_read_u16(bs);
op->maxPicWidth = gf_bs_read_u16(bs);
op->maxPicHeight = gf_bs_read_u16(bs);
op->maxChromaFormat = gf_bs_read_int(bs, 2);
op->maxBitDepth = gf_bs_read_int(bs, 3) + 8;
gf_bs_read_int(bs, 1);
op->frame_rate_info_flag = gf_bs_read_int(bs, 1) ? GF_TRUE : GF_FALSE;
op->bit_rate_info_flag = gf_bs_read_int(bs, 1) ? GF_TRUE : GF_FALSE;
if (op->frame_rate_info_flag) {
op->avgFrameRate = gf_bs_read_u16(bs);
gf_bs_read_int(bs, 6);
op->constantFrameRate = gf_bs_read_int(bs, 2);
}
if (op->bit_rate_info_flag) {
op->maxBitRate = gf_bs_read_u32(bs);
op->avgBitRate = gf_bs_read_u32(bs);
}
gf_list_add(ptr->operating_points, op);
}
count = gf_bs_read_u8(bs);
for (i = 0; i < count; i++) {
LHEVC_DependentLayer *dep;
GF_SAFEALLOC(dep, LHEVC_DependentLayer);
if (!dep) return GF_OUT_OF_MEM;
dep->dependent_layerID = gf_bs_read_u8(bs);
dep->num_layers_dependent_on = gf_bs_read_u8(bs);
for (j = 0; j < dep->num_layers_dependent_on; j++)
dep->dependent_on_layerID[j] = gf_bs_read_u8(bs);
for (j = 0; j < 16; j++) {
if (ptr->scalability_mask & (1 << j))
dep->dimension_identifier[j] = gf_bs_read_u8(bs);
}
gf_list_add(ptr->dependency_layers, dep);
}
return GF_OK;
}
GF_Err gf_isom_oinf_write_entry(void *entry, GF_BitStream *bs)
{
GF_OperatingPointsInformation* ptr = (GF_OperatingPointsInformation *)entry;
u32 i, j, count;
if (!ptr) return GF_OK;
gf_bs_write_u16(bs, ptr->scalability_mask);
gf_bs_write_int(bs, 0xFF, 2);
count=gf_list_count(ptr->profile_tier_levels);
gf_bs_write_int(bs, count, 6);
for (i = 0; i < count; i++) {
LHEVC_ProfileTierLevel *ptl = (LHEVC_ProfileTierLevel *)gf_list_get(ptr->profile_tier_levels, i);
gf_bs_write_int(bs, ptl->general_profile_space, 2);
gf_bs_write_int(bs, ptl->general_tier_flag, 1);
gf_bs_write_int(bs, ptl->general_profile_idc, 5);
gf_bs_write_u32(bs, ptl->general_profile_compatibility_flags);
gf_bs_write_long_int(bs, ptl->general_constraint_indicator_flags, 48);
gf_bs_write_u8(bs, ptl->general_level_idc);
}
count=gf_list_count(ptr->operating_points);
gf_bs_write_u16(bs, count);
for (i = 0; i < count; i++) {
LHEVC_OperatingPoint *op = (LHEVC_OperatingPoint *)gf_list_get(ptr->operating_points, i);;
gf_bs_write_u16(bs, op->output_layer_set_idx);
gf_bs_write_u8(bs, op->max_temporal_id);
gf_bs_write_u8(bs, op->layer_count);
for (j = 0; j < op->layer_count; j++) {
gf_bs_write_u8(bs, op->layers_info[j].ptl_idx);
gf_bs_write_int(bs, op->layers_info[j].layer_id, 6);
op->layers_info[j].is_outputlayer ? gf_bs_write_int(bs, 0x1, 1) : gf_bs_write_int(bs, 0x0, 1);
op->layers_info[j].is_alternate_outputlayer ? gf_bs_write_int(bs, 0x1, 1) : gf_bs_write_int(bs, 0x0, 1);
}
gf_bs_write_u16(bs, op->minPicWidth);
gf_bs_write_u16(bs, op->minPicHeight);
gf_bs_write_u16(bs, op->maxPicWidth);
gf_bs_write_u16(bs, op->maxPicHeight);
gf_bs_write_int(bs, op->maxChromaFormat, 2);
gf_bs_write_int(bs, op->maxBitDepth - 8, 3);
gf_bs_write_int(bs, 0x1, 1);
op->frame_rate_info_flag ? gf_bs_write_int(bs, 0x1, 1) : gf_bs_write_int(bs, 0x0, 1);
op->bit_rate_info_flag ? gf_bs_write_int(bs, 0x1, 1) : gf_bs_write_int(bs, 0x0, 1);
if (op->frame_rate_info_flag) {
gf_bs_write_u16(bs, op->avgFrameRate);
gf_bs_write_int(bs, 0xFF, 6);
gf_bs_write_int(bs, op->constantFrameRate, 2);
}
if (op->bit_rate_info_flag) {
gf_bs_write_u32(bs, op->maxBitRate);
gf_bs_write_u32(bs, op->avgBitRate);
}
}
count=gf_list_count(ptr->dependency_layers);
gf_bs_write_u8(bs, count);
for (i = 0; i < count; i++) {
LHEVC_DependentLayer *dep = (LHEVC_DependentLayer *)gf_list_get(ptr->dependency_layers, i);
gf_bs_write_u8(bs, dep->dependent_layerID);
gf_bs_write_u8(bs, dep->num_layers_dependent_on);
for (j = 0; j < dep->num_layers_dependent_on; j++)
gf_bs_write_u8(bs, dep->dependent_on_layerID[j]);
for (j = 0; j < 16; j++) {
if (ptr->scalability_mask & (1 << j))
gf_bs_write_u8(bs, dep->dimension_identifier[j]);
}
}
return GF_OK;
}
u32 gf_isom_oinf_size_entry(void *entry)
{
GF_OperatingPointsInformation* ptr = (GF_OperatingPointsInformation *)entry;
u32 size = 0, i ,j, count;
if (!ptr) return 0;
size += 3;
count=gf_list_count(ptr->profile_tier_levels);
size += count * 12;
size += 2;
count=gf_list_count(ptr->operating_points);
for (i = 0; i < count; i++) {
LHEVC_OperatingPoint *op = (LHEVC_OperatingPoint *)gf_list_get(ptr->operating_points, i);;
size += 2 + 1 + 1;
size += op->layer_count * 2;
size += 9;
if (op->frame_rate_info_flag) {
size += 3;
}
if (op->bit_rate_info_flag) {
size += 8;
}
}
size += 1;
count=gf_list_count(ptr->dependency_layers);
for (i = 0; i < count; i++) {
LHEVC_DependentLayer *dep = (LHEVC_DependentLayer *)gf_list_get(ptr->dependency_layers, i);
size += 1 + 1;
size += dep->num_layers_dependent_on * 1;
for (j = 0; j < 16; j++) {
if (ptr->scalability_mask & (1 << j))
size += 1;
}
}
return size;
}
GF_LHVCLayerInformation *gf_isom_linf_new_entry()
{
GF_LHVCLayerInformation* ptr;
GF_SAFEALLOC(ptr, GF_LHVCLayerInformation);
if (ptr) ptr->num_layers_in_track = gf_list_new();
return ptr;
}
void gf_isom_linf_del_entry(void *entry)
{
GF_LHVCLayerInformation* ptr = (GF_LHVCLayerInformation *)entry;
if (!ptr) return;
while (gf_list_count(ptr->num_layers_in_track)) {
LHVCLayerInfoItem *li = (LHVCLayerInfoItem *)gf_list_get(ptr->num_layers_in_track, 0);
gf_free(li);
gf_list_rem(ptr->num_layers_in_track, 0);
}
gf_list_del(ptr->num_layers_in_track);
gf_free(ptr);
return;
}
GF_Err gf_isom_linf_read_entry(void *entry, GF_BitStream *bs)
{
GF_LHVCLayerInformation* ptr = (GF_LHVCLayerInformation *)entry;
u32 i, count;
if (!ptr) return GF_BAD_PARAM;
gf_bs_read_int(bs, 2);
count = gf_bs_read_int(bs, 6);
for (i = 0; i < count; i++) {
LHVCLayerInfoItem *li;
GF_SAFEALLOC(li, LHVCLayerInfoItem);
if (!li) return GF_OUT_OF_MEM;
gf_bs_read_int(bs, 4);
li->layer_id = gf_bs_read_int(bs, 6);
li->min_TemporalId = gf_bs_read_int(bs, 3);
li->max_TemporalId = gf_bs_read_int(bs, 3);
gf_bs_read_int(bs, 1);
li->sub_layer_presence_flags = gf_bs_read_int(bs, 7);
gf_list_add(ptr->num_layers_in_track, li);
}
return GF_OK;
}
GF_Err gf_isom_linf_write_entry(void *entry, GF_BitStream *bs)
{
GF_LHVCLayerInformation* ptr = (GF_LHVCLayerInformation *)entry;
u32 i, count;
if (!ptr) return GF_OK;
gf_bs_write_int(bs, 0, 2);
count=gf_list_count(ptr->num_layers_in_track);
gf_bs_write_int(bs, count, 6);
for (i = 0; i < count; i++) {
LHVCLayerInfoItem *li = (LHVCLayerInfoItem *)gf_list_get(ptr->num_layers_in_track, i);
gf_bs_write_int(bs, 0, 4);
gf_bs_write_int(bs, li->layer_id, 6);
gf_bs_write_int(bs, li->min_TemporalId, 3);
gf_bs_write_int(bs, li->max_TemporalId, 3);
gf_bs_write_int(bs, 0, 1);
gf_bs_write_int(bs, li->sub_layer_presence_flags, 7);
}
return GF_OK;
}
u32 gf_isom_linf_size_entry(void *entry)
{
GF_LHVCLayerInformation* ptr = (GF_LHVCLayerInformation *)entry;
u32 size = 0, count;
if (!ptr) return 0;
size += 1;
count=gf_list_count(ptr->num_layers_in_track);
size += count * 3;
return size;
}
#endif