2 * This file is part of DisOrder.
3 * Copyright (C) 2004, 2005, 2007 Richard Kettlewell
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 /** @file plugins/tracklength.c
19 * @brief Plugin to compute track lengths
21 * Currently implements MP3, OGG, FLAC and WAV.
29 #include <sys/types.h>
36 #include <vorbis/vorbisfile.h>
38 /* libFLAC has had an API change and stupidly taken away the old API */
39 #if HAVE_FLAC_FILE_DECODER_H
40 # include <FLAC/file_decoder.h>
42 # include <FLAC/stream_decoder.h>
43 #define FLAC__FileDecoder FLAC__StreamDecoder
44 #define FLAC__FileDecoderState FLAC__StreamDecoderState
53 static void *mmap_file(const char *path, size_t *lengthp) {
58 if((fd = open(path, O_RDONLY)) < 0) {
59 disorder_error(errno, "error opening %s", path);
62 if(fstat(fd, &sb) < 0) {
63 disorder_error(errno, "error calling stat on %s", path);
66 if(sb.st_size == 0) /* can't map 0-length files */
68 if((base = mmap(0, sb.st_size, PROT_READ,
69 MAP_SHARED, fd, 0)) == (void *)-1) {
70 disorder_error(errno, "error calling mmap on %s", path);
73 *lengthp = sb.st_size;
81 static long tl_mp3(const char *path) {
86 if(!(base = mmap_file(path, &length))) return -1;
87 b.duration = mad_timer_zero;
88 scan_mp3(base, length, &b);
90 return b.duration.seconds + !!b.duration.fraction;
93 static long tl_ogg(const char *path) {
99 if(!(fp = fopen(path, "rb"))) goto error;
100 if(ov_open(fp, &vf, 0, 0)) goto error;
102 length = ov_time_total(&vf, -1);
110 static long tl_wav(const char *path) {
112 int err, sample_frame_size;
115 if((err = wav_init(f, path))) {
116 disorder_error(err, "error opening %s", path);
119 sample_frame_size = (f->bits + 7) / 8 * f->channels;
120 if(sample_frame_size) {
121 const long long n_samples = f->datasize / sample_frame_size;
122 duration = (n_samples + f->rate - 1) / f->rate;
129 /* libFLAC's "simplified" interface is rather heavyweight... */
136 static void flac_metadata(const FLAC__FileDecoder attribute((unused)) *decoder,
137 const FLAC__StreamMetadata *metadata,
139 struct flac_state *const state = client_data;
140 const FLAC__StreamMetadata_StreamInfo *const stream_info
141 = &metadata->data.stream_info;
143 if(metadata->type == FLAC__METADATA_TYPE_STREAMINFO)
144 /* FLAC uses 0 to mean unknown and conveniently so do we */
145 state->duration = (stream_info->total_samples
146 + stream_info->sample_rate - 1)
147 / stream_info->sample_rate;
150 static void flac_error(const FLAC__FileDecoder attribute((unused)) *decoder,
151 FLAC__StreamDecoderErrorStatus status,
153 const struct flac_state *const state = client_data;
155 disorder_error(0, "error decoding %s: %s", state->path,
156 FLAC__StreamDecoderErrorStatusString[status]);
159 static FLAC__StreamDecoderWriteStatus flac_write
160 (const FLAC__FileDecoder attribute((unused)) *decoder,
161 const FLAC__Frame attribute((unused)) *frame,
162 const FLAC__int32 attribute((unused)) *const buffer_[],
163 void attribute((unused)) *client_data) {
164 const struct flac_state *const state = client_data;
166 if(state->duration >= 0)
167 return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
169 return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
172 static long tl_flac(const char *path) {
173 struct flac_state state[1];
175 state->duration = -1; /* error */
177 #if HAVE_FLAC_FILE_DECODER_H
179 FLAC__FileDecoder *fd = 0;
180 FLAC__FileDecoderState fs;
182 if(!(fd = FLAC__file_decoder_new())) {
183 disorder_error(0, "FLAC__file_decoder_new failed");
186 if(!(FLAC__file_decoder_set_filename(fd, path))) {
187 disorder_error(0, "FLAC__file_set_filename failed");
190 FLAC__file_decoder_set_metadata_callback(fd, flac_metadata);
191 FLAC__file_decoder_set_error_callback(fd, flac_error);
192 FLAC__file_decoder_set_write_callback(fd, flac_write);
193 FLAC__file_decoder_set_client_data(fd, state);
194 if((fs = FLAC__file_decoder_init(fd))) {
195 disorder_error(0, "FLAC__file_decoder_init: %s",
196 FLAC__FileDecoderStateString[fs]);
199 FLAC__file_decoder_process_until_end_of_metadata(fd);
202 FLAC__file_decoder_delete(fd);
206 FLAC__StreamDecoder *sd = 0;
207 FLAC__StreamDecoderInitStatus is;
209 if(!(sd = FLAC__stream_decoder_new())) {
210 disorder_error(0, "FLAC__stream_decoder_new failed");
213 if((is = FLAC__stream_decoder_init_file(sd, path, flac_write, flac_metadata,
214 flac_error, state))) {
215 disorder_error(0, "FLAC__stream_decoder_init_file %s: %s",
216 path, FLAC__StreamDecoderInitStatusString[is]);
219 FLAC__stream_decoder_process_until_end_of_metadata(sd);
222 FLAC__stream_decoder_delete(sd);
225 return state->duration;
228 static const struct {
230 long (*fn)(const char *path);
232 { ".FLAC", tl_flac },
236 { ".flac", tl_flac },
241 #define N_FILE_FORMATS (int)(sizeof file_formats / sizeof *file_formats)
243 long disorder_tracklength(const char attribute((unused)) *track,
245 const char *ext = strrchr(path, '.');
246 int l, r, m = 0, c = 0; /* quieten compiler */
250 r = N_FILE_FORMATS - 1;
251 while(l <= r && (c = strcmp(ext, file_formats[m = (l + r) / 2].ext)))
257 return file_formats[m].fn(path);