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

   1 /*
   2  * VorbisComment writer
   3  * Copyright (c) 2009 James Darnley
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 #include "avformat.h"
  23 #include "metadata.h"
  24 #include "vorbiscomment.h"
  25 #include "libavcodec/bytestream.h"
  26
  27 /**
  28  * VorbisComment metadata conversion mapping.
  29  * from Ogg Vorbis I format specification: comment field and header specification
  30  * http://xiph.org/vorbis/doc/v-comment.html
  31  */
  32 const AVMetadataConv ff_vorbiscomment_metadata_conv[] = {
  33     { "ALBUMARTIST", "album_artist"},
  34     { "TRACKNUMBER", "track"  },
  35     { "DISCNUMBER",  "disc"   },
  36     { 0 }
  37 };
  38
  39 int ff_vorbiscomment_length(AVMetadata *m, const char *vendor_string,
  40                             unsigned *count)
  41 {
  42     int len = 8;
  43     len += strlen(vendor_string);
  44     *count = 0;
  45     if (m) {
  46         AVMetadataTag *tag = NULL;
  47         while ((tag = av_metadata_get(m, "", tag, AV_METADATA_IGNORE_SUFFIX))) {
  48             len += 4 +strlen(tag->key) + 1 + strlen(tag->value);
  49             (*count)++;
  50         }
  51     }
  52     return len;
  53 }
  54
  55 int ff_vorbiscomment_write(uint8_t **p, AVMetadata **m,
  56                            const char *vendor_string, const unsigned count)
  57 {
  58     ff_metadata_conv(m, ff_vorbiscomment_metadata_conv, NULL);
  59     bytestream_put_le32(p, strlen(vendor_string));
  60     bytestream_put_buffer(p, vendor_string, strlen(vendor_string));
  61     if (*m) {
  62         AVMetadataTag *tag = NULL;
  63         bytestream_put_le32(p, count);
  64         while ((tag = av_metadata_get(*m, "", tag, AV_METADATA_IGNORE_SUFFIX))) {
  65             unsigned int len1 = strlen(tag->key);
  66             unsigned int len2 = strlen(tag->value);
  67             bytestream_put_le32(p, len1+1+len2);
  68             bytestream_put_buffer(p, tag->key, len1);
  69             bytestream_put_byte(p, '=');
  70             bytestream_put_buffer(p, tag->value, len2);
  71         }
  72     } else
  73         bytestream_put_le32(p, 0);
  74     return 0;
  75 }