Imported Debian version 2.5.2~trusty
[deb_ffmpeg.git] / ffmpeg / libavformat / replaygain.c
... / ...
CommitLineData
1/*
2 * This file is part of FFmpeg.
3 *
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19/**
20 * @file
21 * replaygain tags parsing
22 */
23
24#include <stdint.h>
25#include <stdlib.h>
26#include <string.h>
27
28#include "libavutil/avstring.h"
29#include "libavutil/dict.h"
30#include "libavutil/intreadwrite.h"
31#include "libavutil/mathematics.h"
32#include "libavutil/mem.h"
33#include "libavutil/replaygain.h"
34
35#include "avformat.h"
36#include "replaygain.h"
37
38static int32_t parse_value(const char *value, int32_t min)
39{
40 char *fraction;
41 int scale = 10000;
42 int32_t mb = 0;
43 int sign = 1;
44 int db;
45
46 if (!value)
47 return min;
48
49 value += strspn(value, " \t");
50
51 if (*value == '-')
52 sign = -1;
53
54 db = strtol(value, &fraction, 0);
55 if (*fraction++ == '.') {
56 while (av_isdigit(*fraction) && scale) {
57 mb += scale * (*fraction - '0');
58 scale /= 10;
59 fraction++;
60 }
61 }
62
63 if (abs(db) > (INT32_MAX - mb) / 100000)
64 return min;
65
66 return db * 100000 + sign * mb;
67}
68
69int ff_replaygain_export_raw(AVStream *st, int32_t tg, uint32_t tp,
70 int32_t ag, uint32_t ap)
71{
72 AVPacketSideData *sd, *tmp;
73 AVReplayGain *replaygain;
74 int i;
75
76 if (tg == INT32_MIN && ag == INT32_MIN)
77 return 0;
78
79 for (i = 0; i < st->nb_side_data; i++) {
80 AVPacketSideData *src_sd = &st->side_data[i];
81
82 if (src_sd->type == AV_PKT_DATA_REPLAYGAIN)
83 return 0;
84 }
85
86 replaygain = av_mallocz(sizeof(*replaygain));
87 if (!replaygain)
88 return AVERROR(ENOMEM);
89
90 tmp = av_realloc_array(st->side_data, st->nb_side_data + 1, sizeof(*tmp));
91 if (!tmp) {
92 av_freep(&replaygain);
93 return AVERROR(ENOMEM);
94 }
95 st->side_data = tmp;
96 st->nb_side_data++;
97
98 sd = &st->side_data[st->nb_side_data - 1];
99 sd->type = AV_PKT_DATA_REPLAYGAIN;
100 sd->data = (uint8_t*)replaygain;
101 sd->size = sizeof(*replaygain);
102
103 replaygain->track_gain = tg;
104 replaygain->track_peak = tp;
105 replaygain->album_gain = ag;
106 replaygain->album_peak = ap;
107
108 return 0;
109}
110
111int ff_replaygain_export(AVStream *st, AVDictionary *metadata)
112{
113 const AVDictionaryEntry *tg, *tp, *ag, *ap;
114
115 tg = av_dict_get(metadata, "REPLAYGAIN_TRACK_GAIN", NULL, 0);
116 tp = av_dict_get(metadata, "REPLAYGAIN_TRACK_PEAK", NULL, 0);
117 ag = av_dict_get(metadata, "REPLAYGAIN_ALBUM_GAIN", NULL, 0);
118 ap = av_dict_get(metadata, "REPLAYGAIN_ALBUM_PEAK", NULL, 0);
119
120 return ff_replaygain_export_raw(st,
121 parse_value(tg ? tg->value : NULL, INT32_MIN),
122 parse_value(tp ? tp->value : NULL, 0),
123 parse_value(ag ? ag->value : NULL, INT32_MIN),
124 parse_value(ap ? ap->value : NULL, 0));
125}