mpd/src/decoder/AudiofileDecoderPlugin.cxx

268 lines
6.1 KiB
C++
Raw Normal View History

/*
2014-01-13 22:30:36 +01:00
* Copyright (C) 2003-2014 The Music Player Daemon Project
* http://www.musicpd.org
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include "config.h"
2013-07-28 12:37:55 +02:00
#include "AudiofileDecoderPlugin.hxx"
2013-07-28 13:18:48 +02:00
#include "DecoderAPI.hxx"
#include "InputStream.hxx"
2013-07-29 07:50:08 +02:00
#include "CheckAudioFormat.hxx"
2013-09-05 18:22:02 +02:00
#include "tag/TagHandler.hxx"
#include "util/Error.hxx"
#include "util/Domain.hxx"
#include "Log.hxx"
#include <audiofile.h>
#include <af_vfs.h>
2008-11-21 20:13:36 +01:00
#include <assert.h>
/* pick 1020 since its devisible for 8,16,24, and 32-bit audio */
#define CHUNK_SIZE 1020
static constexpr Domain audiofile_domain("audiofile");
static int audiofile_get_duration(const char *file)
{
int total_time;
2013-07-28 12:37:55 +02:00
AFfilehandle af_fp = afOpenFile(file, "r", nullptr);
if (af_fp == AF_NULL_FILEHANDLE) {
return -1;
}
total_time = (int)
((double)afGetFrameCount(af_fp, AF_DEFAULT_TRACK)
/ afGetRate(af_fp, AF_DEFAULT_TRACK));
afCloseFile(af_fp);
return total_time;
}
static ssize_t
audiofile_file_read(AFvirtualfile *vfile, void *data, size_t length)
{
InputStream &is = *(InputStream *)vfile->closure;
Error error;
size_t nbytes = is.LockRead(data, length, error);
if (nbytes == 0 && error.IsDefined()) {
LogError(error);
return -1;
}
return nbytes;
}
static AFfileoffset
audiofile_file_length(AFvirtualfile *vfile)
{
InputStream &is = *(InputStream *)vfile->closure;
return is.GetSize();
}
static AFfileoffset
audiofile_file_tell(AFvirtualfile *vfile)
{
InputStream &is = *(InputStream *)vfile->closure;
return is.GetOffset();
}
static void
audiofile_file_destroy(AFvirtualfile *vfile)
{
2013-07-28 12:37:55 +02:00
assert(vfile->closure != nullptr);
2013-07-28 12:37:55 +02:00
vfile->closure = nullptr;
}
static AFfileoffset
audiofile_file_seek(AFvirtualfile *vfile, AFfileoffset offset, int is_relative)
{
InputStream &is = *(InputStream *)vfile->closure;
int whence = (is_relative ? SEEK_CUR : SEEK_SET);
Error error;
if (is.LockSeek(offset, whence, error)) {
return is.GetOffset();
} else {
return -1;
}
}
static AFvirtualfile *
setup_virtual_fops(InputStream &stream)
{
2013-07-28 12:37:55 +02:00
AFvirtualfile *vf = new AFvirtualfile();
vf->closure = &stream;
2013-07-28 12:37:55 +02:00
vf->write = nullptr;
vf->read = audiofile_file_read;
vf->length = audiofile_file_length;
vf->destroy = audiofile_file_destroy;
vf->seek = audiofile_file_seek;
vf->tell = audiofile_file_tell;
return vf;
}
2013-08-03 21:00:50 +02:00
static SampleFormat
audiofile_bits_to_sample_format(int bits)
{
switch (bits) {
case 8:
2013-08-03 21:00:50 +02:00
return SampleFormat::S8;
case 16:
2013-08-03 21:00:50 +02:00
return SampleFormat::S16;
case 24:
2013-08-03 21:00:50 +02:00
return SampleFormat::S24_P32;
case 32:
2013-08-03 21:00:50 +02:00
return SampleFormat::S32;
}
2013-08-03 21:00:50 +02:00
return SampleFormat::UNDEFINED;
}
2013-08-03 21:00:50 +02:00
static SampleFormat
audiofile_setup_sample_format(AFfilehandle af_fp)
{
int fs, bits;
afGetSampleFormat(af_fp, AF_DEFAULT_TRACK, &fs, &bits);
if (!audio_valid_sample_format(audiofile_bits_to_sample_format(bits))) {
FormatDebug(audiofile_domain,
"input file has %d bit samples, converting to 16",
bits);
bits = 16;
}
afSetVirtualSampleFormat(af_fp, AF_DEFAULT_TRACK,
AF_SAMPFMT_TWOSCOMP, bits);
afGetVirtualSampleFormat(af_fp, AF_DEFAULT_TRACK, &fs, &bits);
return audiofile_bits_to_sample_format(bits);
}
static void
audiofile_stream_decode(Decoder &decoder, InputStream &is)
{
AFvirtualfile *vf;
int fs, frame_count;
AFfilehandle af_fp;
2013-08-03 21:00:50 +02:00
AudioFormat audio_format;
float total_time;
uint16_t bit_rate;
int ret;
char chunk[CHUNK_SIZE];
if (!is.IsSeekable()) {
LogWarning(audiofile_domain, "not seekable");
return;
}
vf = setup_virtual_fops(is);
2013-07-28 12:37:55 +02:00
af_fp = afOpenVirtualFile(vf, "r", nullptr);
if (af_fp == AF_NULL_FILEHANDLE) {
LogWarning(audiofile_domain, "failed to input stream");
return;
}
Error error;
2013-08-03 21:00:50 +02:00
if (!audio_format_init_checked(audio_format,
afGetRate(af_fp, AF_DEFAULT_TRACK),
audiofile_setup_sample_format(af_fp),
afGetVirtualChannels(af_fp, AF_DEFAULT_TRACK),
error)) {
LogError(error);
afCloseFile(af_fp);
return;
}
frame_count = afGetFrameCount(af_fp, AF_DEFAULT_TRACK);
total_time = ((float)frame_count / (float)audio_format.sample_rate);
bit_rate = (uint16_t)(is.GetSize() * 8.0 / total_time / 1000.0 + 0.5);
fs = (int)afGetVirtualFrameSize(af_fp, AF_DEFAULT_TRACK, 1);
2013-08-03 21:00:50 +02:00
decoder_initialized(decoder, audio_format, true, total_time);
DecoderCommand cmd;
do {
ret = afReadFrames(af_fp, AF_DEFAULT_TRACK, chunk,
CHUNK_SIZE / fs);
if (ret <= 0)
break;
2013-07-28 12:37:55 +02:00
cmd = decoder_data(decoder, nullptr,
chunk, ret * fs,
bit_rate);
if (cmd == DecoderCommand::SEEK) {
AFframecount frame = decoder_seek_where(decoder) *
audio_format.sample_rate;
afSeekFrame(af_fp, AF_DEFAULT_TRACK, frame);
decoder_command_finished(decoder);
cmd = DecoderCommand::NONE;
}
} while (cmd == DecoderCommand::NONE);
afCloseFile(af_fp);
}
static bool
audiofile_scan_file(const char *file,
const struct tag_handler *handler, void *handler_ctx)
{
int total_time = audiofile_get_duration(file);
if (total_time < 0) {
FormatWarning(audiofile_domain,
"Failed to get total song time from: %s",
file);
return false;
}
tag_handler_invoke_duration(handler, handler_ctx, total_time);
return true;
}
static const char *const audiofile_suffixes[] = {
2013-07-28 12:37:55 +02:00
"wav", "au", "aiff", "aif", nullptr
};
static const char *const audiofile_mime_types[] = {
"audio/x-wav",
"audio/x-aiff",
2013-07-28 12:37:55 +02:00
nullptr
};
const struct DecoderPlugin audiofile_decoder_plugin = {
2013-07-28 12:37:55 +02:00
"audiofile",
nullptr,
nullptr,
audiofile_stream_decode,
nullptr,
audiofile_scan_file,
nullptr,
nullptr,
audiofile_suffixes,
audiofile_mime_types,
};