a0f1c8817dcb2941574bf7f2de8da550b77531b3
[ffmpeg.git] / 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     bytestream_put_le32(p, strlen(vendor_string));
59     bytestream_put_buffer(p, vendor_string, strlen(vendor_string));
60     if (m) {
61         AVMetadataTag *tag = NULL;
62         bytestream_put_le32(p, count);
63         while ((tag = av_metadata_get(m, "", tag, AV_METADATA_IGNORE_SUFFIX))) {
64             unsigned int len1 = strlen(tag->key);
65             unsigned int len2 = strlen(tag->value);
66             bytestream_put_le32(p, len1+1+len2);
67             bytestream_put_buffer(p, tag->key, len1);
68             bytestream_put_byte(p, '=');
69             bytestream_put_buffer(p, tag->value, len2);
70         }
71     } else
72         bytestream_put_le32(p, 0);
73     return 0;
74 }