git.sesse.net Git - ffmpeg/blob - libavformat/metadata.c

   1 /*
   2  * copyright (c) 2009 Michael Niedermayer
   3  *
   4  * This file is part of FFmpeg.
   5  *
   6  * FFmpeg is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * FFmpeg is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with FFmpeg; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 #include <strings.h>
  22 #include "avformat.h"
  23 #include "metadata.h"
  24
  25 AVMetadataTag *
  26 av_metadata_get(AVMetadata *m, const char *key, const AVMetadataTag *prev, int flags)
  27 {
  28     unsigned int i, j;
  29
  30     if(!m)
  31         return NULL;
  32
  33     if(prev) i= prev - m->elems + 1;
  34     else     i= 0;
  35
  36     for(; i<m->count; i++){
  37         const char *s= m->elems[i].key;
  38         if(flags & AV_METADATA_MATCH_CASE) for(j=0;         s[j]  ==         key[j]  && key[j]; j++);
  39         else                               for(j=0; toupper(s[j]) == toupper(key[j]) && key[j]; j++);
  40         if(key[j])
  41             continue;
  42         if(s[j] && !(flags & AV_METADATA_IGNORE_SUFFIX))
  43             continue;
  44         return &m->elems[i];
  45     }
  46     return NULL;
  47 }
  48
  49 int av_metadata_set2(AVMetadata **pm, const char *key, const char *value, int flags)
  50 {
  51     AVMetadata *m= *pm;
  52     AVMetadataTag *tag= av_metadata_get(m, key, NULL, flags);
  53
  54     if(!m)
  55         m=*pm= av_mallocz(sizeof(*m));
  56
  57     if(tag){
  58         if (flags & AV_METADATA_DONT_OVERWRITE)
  59             return 0;
  60         av_free(tag->value);
  61         av_free(tag->key);
  62         *tag= m->elems[--m->count];
  63     }else{
  64         AVMetadataTag *tmp= av_realloc(m->elems, (m->count+1) * sizeof(*m->elems));
  65         if(tmp){
  66             m->elems= tmp;
  67         }else
  68             return AVERROR(ENOMEM);
  69     }
  70     if(value){
  71         if(flags & AV_METADATA_DONT_STRDUP_KEY){
  72             m->elems[m->count].key  = key;
  73         }else
  74         m->elems[m->count].key  = av_strdup(key  );
  75         if(flags & AV_METADATA_DONT_STRDUP_VAL){
  76             m->elems[m->count].value= value;
  77         }else
  78         m->elems[m->count].value= av_strdup(value);
  79         m->count++;
  80     }
  81     if(!m->count) {
  82         av_free(m->elems);
  83         av_freep(pm);
  84     }
  85
  86     return 0;
  87 }
  88
  89 #if FF_API_OLD_METADATA
  90 int av_metadata_set(AVMetadata **pm, const char *key, const char *value)
  91 {
  92     return av_metadata_set2(pm, key, value, 0);
  93 }
  94 #endif
  95
  96 #if FF_API_OLD_METADATA2
  97 void av_metadata_conv(AVFormatContext *ctx, const AVMetadataConv *d_conv,
  98                                             const AVMetadataConv *s_conv)
  99 {
 100     return;
 101 }
 102 #endif
 103
 104 void av_metadata_free(AVMetadata **pm)
 105 {
 106     AVMetadata *m= *pm;
 107
 108     if(m){
 109         while(m->count--){
 110             av_free(m->elems[m->count].key);
 111             av_free(m->elems[m->count].value);
 112         }
 113         av_free(m->elems);
 114     }
 115     av_freep(pm);
 116 }
 117
 118 void ff_metadata_conv(AVMetadata **pm, const AVMetadataConv *d_conv,
 119                                        const AVMetadataConv *s_conv)
 120 {
 121     /* TODO: use binary search to look up the two conversion tables
 122        if the tables are getting big enough that it would matter speed wise */
 123     const AVMetadataConv *sc, *dc;
 124     AVMetadataTag *mtag = NULL;
 125     AVMetadata *dst = NULL;
 126     const char *key;
 127
 128     if (d_conv == s_conv)
 129         return;
 130
 131     while((mtag=av_metadata_get(*pm, "", mtag, AV_METADATA_IGNORE_SUFFIX))) {
 132         key = mtag->key;
 133         if (s_conv)
 134             for (sc=s_conv; sc->native; sc++)
 135                 if (!strcasecmp(key, sc->native)) {
 136                     key = sc->generic;
 137                     break;
 138                 }
 139         if (d_conv)
 140             for (dc=d_conv; dc->native; dc++)
 141                 if (!strcasecmp(key, dc->generic)) {
 142                     key = dc->native;
 143                     break;
 144                 }
 145         av_metadata_set2(&dst, key, mtag->value, 0);
 146     }
 147     av_metadata_free(pm);
 148     *pm = dst;
 149 }
 150
 151 void ff_metadata_conv_ctx(AVFormatContext *ctx, const AVMetadataConv *d_conv,
 152                                                 const AVMetadataConv *s_conv)
 153 {
 154     int i;
 155     ff_metadata_conv(&ctx->metadata, d_conv, s_conv);
 156     for (i=0; i<ctx->nb_streams ; i++)
 157         ff_metadata_conv(&ctx->streams [i]->metadata, d_conv, s_conv);
 158     for (i=0; i<ctx->nb_chapters; i++)
 159         ff_metadata_conv(&ctx->chapters[i]->metadata, d_conv, s_conv);
 160     for (i=0; i<ctx->nb_programs; i++)
 161         ff_metadata_conv(&ctx->programs[i]->metadata, d_conv, s_conv);
 162 }
 163
 164 void av_metadata_copy(AVMetadata **dst, AVMetadata *src, int flags)
 165 {
 166     AVMetadataTag *t = NULL;
 167
 168     while ((t = av_metadata_get(src, "", t, AV_METADATA_IGNORE_SUFFIX)))
 169         av_metadata_set2(dst, t->key, t->value, flags);
 170 }