comparison matroskadec.c @ 3702:b4f34511ccd3 libavformat

matroskadec: add basic tags support (metadata)

3701:5cf9918994ed

    uint64_t time;
    EbmlList pos;
} MatroskaIndex;

typedef struct {
    uint64_t id;
    uint64_t pos;
} MatroskaSeekhead;

typedef struct {

    char    *title;
    EbmlList tracks;
    EbmlList attachments;
    EbmlList chapters;
    EbmlList index;
    EbmlList seekhead;

    /* byte position of the segment inside the stream */
    offset_t segment_start;

    { MATROSKA_ID_POINTENTRY,         EBML_NEST, sizeof(MatroskaIndex), offsetof(MatroskaDemuxContext,index), {.n=matroska_index_entry} },
    { EBML_ID_VOID,                   EBML_NONE },
    { 0 }
};

static EbmlSyntax matroska_tags[] = {
    { EBML_ID_VOID,                   EBML_NONE },
    { 0 }
};

static EbmlSyntax matroska_seekhead_entry[] = {

};

static EbmlSyntax matroska_clusters[] = {
    { MATROSKA_ID_CLUSTER,        EBML_NEST, 0, 0, {.n=matroska_cluster} },
    { 0 }
};

/*
 * Return: Whether we reached the end of a level in the hierarchy or not.
 */

 failed:
    av_free(pkt_data);
    return -1;
}

static void matroska_execute_seekhead(MatroskaDemuxContext *matroska)
{
    EbmlList *seekhead_list = &matroska->seekhead;
    MatroskaSeekhead *seekhead = seekhead_list->elem;
    uint32_t level_up = matroska->level_up;

        matroska->ctx->duration = matroska->duration * matroska->time_scale
                                  * 1000 / AV_TIME_BASE;
    if (matroska->title)
        strncpy(matroska->ctx->title, matroska->title,
                sizeof(matroska->ctx->title)-1);

    tracks = matroska->tracks.elem;
    for (i=0; i < matroska->tracks.nb_elem; i++) {
        MatroskaTrack *track = &tracks[i];
        enum CodecID codec_id = CODEC_ID_NONE;

3702:b4f34511ccd3

    uint64_t time;
    EbmlList pos;
} MatroskaIndex;

typedef struct {
    char *name;
    char *string;
    EbmlList sub;
} MatroskaTag;

typedef struct {
    uint64_t id;
    uint64_t pos;
} MatroskaSeekhead;

typedef struct {

    char    *title;
    EbmlList tracks;
    EbmlList attachments;
    EbmlList chapters;
    EbmlList index;
    EbmlList tags;
    EbmlList seekhead;

    /* byte position of the segment inside the stream */
    offset_t segment_start;

    { MATROSKA_ID_POINTENTRY,         EBML_NEST, sizeof(MatroskaIndex), offsetof(MatroskaDemuxContext,index), {.n=matroska_index_entry} },
    { EBML_ID_VOID,                   EBML_NONE },
    { 0 }
};

static EbmlSyntax matroska_simpletag[] = {
    { MATROSKA_ID_TAGNAME,            EBML_UTF8, 0, offsetof(MatroskaTag,name) },
    { MATROSKA_ID_TAGSTRING,          EBML_UTF8, 0, offsetof(MatroskaTag,string) },
    { MATROSKA_ID_SIMPLETAG,          EBML_NEST, sizeof(MatroskaTag), offsetof(MatroskaTag,sub), {.n=matroska_simpletag} },
    { MATROSKA_ID_TAGLANG,            EBML_NONE },
    { MATROSKA_ID_TAGDEFAULT,         EBML_NONE },
    { EBML_ID_VOID,                   EBML_NONE },
    { 0 }
};

static EbmlSyntax matroska_tag[] = {
    { MATROSKA_ID_SIMPLETAG,          EBML_NEST, sizeof(MatroskaTag), 0, {.n=matroska_simpletag} },
    { MATROSKA_ID_TAGTARGETS,         EBML_NONE },
    { EBML_ID_VOID,                   EBML_NONE },
    { 0 }
};

static EbmlSyntax matroska_tags[] = {
    { MATROSKA_ID_TAG,                EBML_NEST, 0, offsetof(MatroskaDemuxContext,tags), {.n=matroska_tag} },
    { EBML_ID_VOID,                   EBML_NONE },
    { 0 }
};

static EbmlSyntax matroska_seekhead_entry[] = {

};

static EbmlSyntax matroska_clusters[] = {
    { MATROSKA_ID_CLUSTER,        EBML_NEST, 0, 0, {.n=matroska_cluster} },
    { 0 }
};

#define SIZE_OFF(x) sizeof(((AVFormatContext*)0)->x),offsetof(AVFormatContext,x)
const struct {
    const char name[16];
    int   size;
    int   offset;
} metadata[] = {
    { "TITLE",           SIZE_OFF(title)      },
    { "ARTIST",          SIZE_OFF(author)     },
    { "WRITTEN_BY",      SIZE_OFF(author)     },
    { "LEAD_PERFORMER",  SIZE_OFF(author)     },
    { "COPYRIGHT",       SIZE_OFF(copyright)  },
    { "COMMENT",         SIZE_OFF(comment)    },
    { "ALBUM",           SIZE_OFF(album)      },
    { "DATE_WRITTEN",    SIZE_OFF(year)       },
    { "DATE_RELEASED",   SIZE_OFF(year)       },
    { "PART_NUMBER",     SIZE_OFF(track)      },
    { "GENRE",           SIZE_OFF(genre)      },
};

/*
 * Return: Whether we reached the end of a level in the hierarchy or not.
 */

 failed:
    av_free(pkt_data);
    return -1;
}

static void matroska_convert_tags(AVFormatContext *s, EbmlList *list)
{
    MatroskaTag *tags = list->elem;
    int i, j;

    for (i=0; i < list->nb_elem; i++) {
        for (j=0; j < ARRAY_SIZE(metadata); j++){
            if (!strcmp(tags[i].name, metadata[j].name)) {
                int *ptr = (int *)((char *)s + metadata[j].offset);
                if (*ptr)  continue;
                if (metadata[j].size > sizeof(int))
                    av_strlcpy((char *)ptr, tags[i].string, metadata[j].size);
                else
                    *ptr = atoi(tags[i].string);
            }
        }
        if (tags[i].sub.nb_elem)
            matroska_convert_tags(s, &tags[i].sub);
    }
}

static void matroska_execute_seekhead(MatroskaDemuxContext *matroska)
{
    EbmlList *seekhead_list = &matroska->seekhead;
    MatroskaSeekhead *seekhead = seekhead_list->elem;
    uint32_t level_up = matroska->level_up;

        matroska->ctx->duration = matroska->duration * matroska->time_scale
                                  * 1000 / AV_TIME_BASE;
    if (matroska->title)
        strncpy(matroska->ctx->title, matroska->title,
                sizeof(matroska->ctx->title)-1);
    matroska_convert_tags(s, &matroska->tags);

    tracks = matroska->tracks.elem;
    for (i=0; i < matroska->tracks.nb_elem; i++) {
        MatroskaTrack *track = &tracks[i];
        enum CodecID codec_id = CODEC_ID_NONE;