comparison mxfenc.c @ 4397:6d1626886974 libavformat

only use 2 slices for index, one video(vbr) and one audio(cbr)

4396:a34ee37a164e

} MXFLocalTagPair;

typedef struct {
    uint8_t flags;
    uint64_t offset;
    unsigned slice_offset[17]; // one video, 16 audio
} MXFIndexEntry;

typedef struct {
    AudioInterleaveContext aic;
    UID track_essence_element_key;

    AVRational time_base;
    int header_written;
    MXFIndexEntry *index_entries;
    unsigned edit_units_count;
    uint64_t timestamp;   ///< timestamp, as year(16),month(8),day(8),hour(8),minutes(8),msec/4(8)
} MXFContext;

static const uint8_t uuid_base[]            = { 0xAD,0xAB,0x44,0x24,0x2f,0x25,0x4d,0xc7,0x92,0xff,0x29,0xbd };
static const uint8_t umid_base[]            = { 0x06,0x0A,0x2B,0x34,0x01,0x01,0x01,0x05,0x01,0x01,0x0D,0x00,0x13,0x00,0x00,0x00 };

    mxf_write_package(s, SourcePackage);
    mxf_write_essence_container_data(s);
    return 0;
}

static unsigned klv_fill_size(AVFormatContext *s)
{
    unsigned pad = KAG_SIZE - (url_ftell(s->pb) & (KAG_SIZE-1));
    if (pad < 17) // smallest fill item possible
        return pad + KAG_SIZE;
    else
        return pad & (KAG_SIZE-1);
}

static int mxf_write_index_table_segment(AVFormatContext *s)
{
    MXFContext *mxf = s->priv_data;
    ByteIOContext *pb = s->pb;
    int i, j, k, ret;
    int temporal_reordering = 0;
    int last_key_index = 0, key_index = 0;

    av_log(s, AV_LOG_DEBUG, "edit units count %d\n", mxf->edit_units_count);

    put_buffer(pb, index_table_segment_key, 16);
    ret = klv_encode_ber_length(pb, 109 + (s->nb_streams+1)*6 +
                                mxf->edit_units_count*(11+(s->nb_streams-1)*4));

    // instance id
    mxf_write_local_tag(pb, 16, 0x3C0A);
    mxf_write_uuid(pb, IndexTableSegment, 0);

    // body sid
    mxf_write_local_tag(pb, 4, 0x3F07);
    put_be32(pb, 1);

    // slice count - 1
    mxf_write_local_tag(pb, 1, 0x3F08);
    put_byte(pb, s->nb_streams-1);

    // delta entry array
    mxf_write_local_tag(pb, 8 + (s->nb_streams+1)*6, 0x3F09);
    put_be32(pb, s->nb_streams+1); // num of entries
    put_be32(pb, 6);             // size of one entry

        AVStream *st = s->streams[i];
        MXFStreamContext *sc = st->priv_data;
        put_byte(pb, sc->temporal_reordering);
        if (sc->temporal_reordering)
            temporal_reordering = 1;
        put_byte(pb, i);
        if (i == 0)
            put_be32(pb, KAG_SIZE); // system item size including klv fill
        else
            put_be32(pb, 0); // element delta
    }

    mxf_write_local_tag(pb, 8 + mxf->edit_units_count*(11+(s->nb_streams-1)*4), 0x3F0A);
    put_be32(pb, mxf->edit_units_count);  // num of entries
    put_be32(pb, 11+(s->nb_streams-1)*4); // size of one entry
    for (i = 0; i < mxf->edit_units_count; i++) {
        if (temporal_reordering) {
            int temporal_offset = 0;
            for (j = i+1; j < mxf->edit_units_count; j++) {
                temporal_offset++;

                last_key_index = key_index;
        }
        put_byte(pb, mxf->index_entries[i].flags);
        // stream offset
        put_be64(pb, mxf->index_entries[i].offset - mxf->index_entries[0].offset);
        for (k = 0; k < s->nb_streams; k++) {
            if (mxf->index_entries[i].slice_offset[k])
                put_be32(pb, mxf->index_entries[i].slice_offset[k]);
        }
    }

    return ret;
}

        unsigned header_byte_count;

        mxf_write_primer_pack(s);
        mxf_write_header_metadata_sets(s);
        pos = url_ftell(s->pb);
        header_byte_count = pos - start + klv_fill_size(s);

        // update header_byte_count
        url_fseek(pb, header_byte_count_offset, SEEK_SET);
        put_be64(pb, header_byte_count);
        url_fseek(pb, pos, SEEK_SET);

            if (st->codec->sample_rate != 48000) {
                av_log(s, AV_LOG_ERROR, "only 48khz is implemented\n");
                return -1;
            }
            av_set_pts_info(st, 64, 1, st->codec->sample_rate);
        }
        sc->duration = -1;

        sc->index = mxf_get_essence_container_ul_index(st->codec->codec_id);
        if (sc->index == -1) {

    return 0;
}

static void mxf_write_klv_fill(AVFormatContext *s)
{
    unsigned pad = klv_fill_size(s);
    if (pad) {
        put_buffer(s->pb, klv_fill_key, 16);
        pad -= 16;
        pad -= klv_ber_length(pad);
        klv_encode_ber_length(s->pb, pad);

    mxf_write_klv_fill(s);

    if (st->index == 0) {
        mxf->index_entries[mxf->edit_units_count].offset = url_ftell(pb);
        mxf->index_entries[mxf->edit_units_count].slice_offset[st->index] = 0;

        mxf_write_system_item(s);
        mxf_write_klv_fill(s);

        mxf->edit_units_count++;
    } else {
        mxf->index_entries[mxf->edit_units_count-1].slice_offset[st->index] =
            url_ftell(pb) - mxf->index_entries[mxf->edit_units_count-1].offset;
    }

    put_buffer(pb, sc->track_essence_element_key, 16); // write key
    klv_encode_ber_length(pb, pkt->size); // write length

{
    MXFContext *mxf = s->priv_data;
    ByteIOContext *pb = s->pb;
    unsigned index_byte_count =
        109 + (s->nb_streams+1)*6 +
        mxf->edit_units_count*(11+(s->nb_streams-1)*4);

    // add encoded ber length
    index_byte_count += 16 + klv_ber_length(index_byte_count);

    mxf_write_klv_fill(s);

4397:6d1626886974

} MXFLocalTagPair;

typedef struct {
    uint8_t flags;
    uint64_t offset;
    unsigned slice_offset; // offset of audio slice
} MXFIndexEntry;

typedef struct {
    AudioInterleaveContext aic;
    UID track_essence_element_key;

    AVRational time_base;
    int header_written;
    MXFIndexEntry *index_entries;
    unsigned edit_units_count;
    uint64_t timestamp;   ///< timestamp, as year(16),month(8),day(8),hour(8),minutes(8),msec/4(8)
    uint8_t slice_count;  /// index slice count minus 1 (1 if no audio, 0 otherwise)
} MXFContext;

static const uint8_t uuid_base[]            = { 0xAD,0xAB,0x44,0x24,0x2f,0x25,0x4d,0xc7,0x92,0xff,0x29,0xbd };
static const uint8_t umid_base[]            = { 0x06,0x0A,0x2B,0x34,0x01,0x01,0x01,0x05,0x01,0x01,0x0D,0x00,0x13,0x00,0x00,0x00 };

    mxf_write_package(s, SourcePackage);
    mxf_write_essence_container_data(s);
    return 0;
}

static unsigned klv_fill_size(uint64_t size)
{
    unsigned pad = KAG_SIZE - (size & (KAG_SIZE-1));
    if (pad < 17) // smallest fill item possible
        return pad + KAG_SIZE;
    else
        return pad & (KAG_SIZE-1);
}

static int mxf_write_index_table_segment(AVFormatContext *s)
{
    MXFContext *mxf = s->priv_data;
    ByteIOContext *pb = s->pb;
    int i, j, ret;
    int temporal_reordering = 0;
    int last_key_index = 0, key_index = 0;

    av_log(s, AV_LOG_DEBUG, "edit units count %d\n", mxf->edit_units_count);

    put_buffer(pb, index_table_segment_key, 16);
    ret = klv_encode_ber_length(pb, 109 + (s->nb_streams+1)*6 +
                                mxf->edit_units_count*(11+mxf->slice_count*4));

    // instance id
    mxf_write_local_tag(pb, 16, 0x3C0A);
    mxf_write_uuid(pb, IndexTableSegment, 0);

    // body sid
    mxf_write_local_tag(pb, 4, 0x3F07);
    put_be32(pb, 1);

    // real slice count - 1
    mxf_write_local_tag(pb, 1, 0x3F08);
    put_byte(pb, mxf->slice_count);

    // delta entry array
    mxf_write_local_tag(pb, 8 + (s->nb_streams+1)*6, 0x3F09);
    put_be32(pb, s->nb_streams+1); // num of entries
    put_be32(pb, 6);             // size of one entry

        AVStream *st = s->streams[i];
        MXFStreamContext *sc = st->priv_data;
        put_byte(pb, sc->temporal_reordering);
        if (sc->temporal_reordering)
            temporal_reordering = 1;
        // slice number
        if (i == 0) { // video track
            put_byte(pb, 0); // slice number
            put_be32(pb, KAG_SIZE); // system item size including klv fill
        } else { // audio track
            unsigned audio_frame_size = sc->aic.samples[0]*sc->aic.sample_size;
            audio_frame_size += klv_fill_size(audio_frame_size);
            put_byte(pb, 1);
            put_be32(pb, (i-1)*audio_frame_size); // element delta
        }
    }

    mxf_write_local_tag(pb, 8 + mxf->edit_units_count*(11+mxf->slice_count*4), 0x3F0A);
    put_be32(pb, mxf->edit_units_count);  // num of entries
    put_be32(pb, 11+mxf->slice_count*4);  // size of one entry
    for (i = 0; i < mxf->edit_units_count; i++) {
        if (temporal_reordering) {
            int temporal_offset = 0;
            for (j = i+1; j < mxf->edit_units_count; j++) {
                temporal_offset++;

                last_key_index = key_index;
        }
        put_byte(pb, mxf->index_entries[i].flags);
        // stream offset
        put_be64(pb, mxf->index_entries[i].offset - mxf->index_entries[0].offset);
        if (s->nb_streams > 1)
            put_be32(pb, mxf->index_entries[i].slice_offset);
    }

    return ret;
}

        unsigned header_byte_count;

        mxf_write_primer_pack(s);
        mxf_write_header_metadata_sets(s);
        pos = url_ftell(s->pb);
        header_byte_count = pos - start + klv_fill_size(pos);

        // update header_byte_count
        url_fseek(pb, header_byte_count_offset, SEEK_SET);
        put_be64(pb, header_byte_count);
        url_fseek(pb, pos, SEEK_SET);

            if (st->codec->sample_rate != 48000) {
                av_log(s, AV_LOG_ERROR, "only 48khz is implemented\n");
                return -1;
            }
            av_set_pts_info(st, 64, 1, st->codec->sample_rate);
            mxf->slice_count = 1;
        }
        sc->duration = -1;

        sc->index = mxf_get_essence_container_ul_index(st->codec->codec_id);
        if (sc->index == -1) {

    return 0;
}

static void mxf_write_klv_fill(AVFormatContext *s)
{
    unsigned pad = klv_fill_size(url_ftell(s->pb));
    if (pad) {
        put_buffer(s->pb, klv_fill_key, 16);
        pad -= 16;
        pad -= klv_ber_length(pad);
        klv_encode_ber_length(s->pb, pad);

    mxf_write_klv_fill(s);

    if (st->index == 0) {
        mxf->index_entries[mxf->edit_units_count].offset = url_ftell(pb);

        mxf_write_system_item(s);
        mxf_write_klv_fill(s);

        mxf->edit_units_count++;
    } else if (st->index == 1) {
        mxf->index_entries[mxf->edit_units_count-1].slice_offset =
            url_ftell(pb) - mxf->index_entries[mxf->edit_units_count-1].offset;
    }

    put_buffer(pb, sc->track_essence_element_key, 16); // write key
    klv_encode_ber_length(pb, pkt->size); // write length

{
    MXFContext *mxf = s->priv_data;
    ByteIOContext *pb = s->pb;
    unsigned index_byte_count =
        109 + (s->nb_streams+1)*6 +
        mxf->edit_units_count*(11+mxf->slice_count*4);

    // add encoded ber length
    index_byte_count += 16 + klv_ber_length(index_byte_count);

    mxf_write_klv_fill(s);