291 lines
6.7 KiB
C++
291 lines
6.7 KiB
C++
/*
|
|
* Copyright 2003-2018 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 "AudiofileDecoderPlugin.hxx"
|
|
#include "../DecoderAPI.hxx"
|
|
#include "input/InputStream.hxx"
|
|
#include "CheckAudioFormat.hxx"
|
|
#include "tag/Handler.hxx"
|
|
#include "util/ScopeExit.hxx"
|
|
#include "util/Domain.hxx"
|
|
#include "Log.hxx"
|
|
|
|
#include <audiofile.h>
|
|
#include <af_vfs.h>
|
|
|
|
#include <exception>
|
|
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
static constexpr Domain audiofile_domain("audiofile");
|
|
|
|
static void
|
|
audiofile_error_func(long, const char *msg) noexcept
|
|
{
|
|
LogWarning(audiofile_domain, msg);
|
|
}
|
|
|
|
static bool
|
|
audiofile_init(const ConfigBlock &)
|
|
{
|
|
afSetErrorHandler(audiofile_error_func);
|
|
return true;
|
|
}
|
|
|
|
struct AudioFileInputStream {
|
|
DecoderClient *const client;
|
|
InputStream &is;
|
|
|
|
size_t Read(void *buffer, size_t size) noexcept {
|
|
/* libaudiofile does not like partial reads at all,
|
|
and will abort playback; therefore always force full
|
|
reads */
|
|
return decoder_read_full(client, is, buffer, size)
|
|
? size
|
|
: 0;
|
|
}
|
|
};
|
|
|
|
gcc_pure
|
|
static SongTime
|
|
audiofile_get_duration(AFfilehandle fh) noexcept
|
|
{
|
|
return SongTime::FromScale<uint64_t>(afGetFrameCount(fh, AF_DEFAULT_TRACK),
|
|
afGetRate(fh, AF_DEFAULT_TRACK));
|
|
}
|
|
|
|
static ssize_t
|
|
audiofile_file_read(AFvirtualfile *vfile, void *data, size_t length) noexcept
|
|
{
|
|
AudioFileInputStream &afis = *(AudioFileInputStream *)vfile->closure;
|
|
|
|
return afis.Read(data, length);
|
|
}
|
|
|
|
static AFfileoffset
|
|
audiofile_file_length(AFvirtualfile *vfile) noexcept
|
|
{
|
|
AudioFileInputStream &afis = *(AudioFileInputStream *)vfile->closure;
|
|
InputStream &is = afis.is;
|
|
|
|
return is.GetSize();
|
|
}
|
|
|
|
static AFfileoffset
|
|
audiofile_file_tell(AFvirtualfile *vfile) noexcept
|
|
{
|
|
AudioFileInputStream &afis = *(AudioFileInputStream *)vfile->closure;
|
|
InputStream &is = afis.is;
|
|
|
|
return is.GetOffset();
|
|
}
|
|
|
|
static void
|
|
audiofile_file_destroy(AFvirtualfile *vfile) noexcept
|
|
{
|
|
assert(vfile->closure != nullptr);
|
|
|
|
vfile->closure = nullptr;
|
|
}
|
|
|
|
static AFfileoffset
|
|
audiofile_file_seek(AFvirtualfile *vfile, AFfileoffset _offset,
|
|
int is_relative) noexcept
|
|
{
|
|
AudioFileInputStream &afis = *(AudioFileInputStream *)vfile->closure;
|
|
InputStream &is = afis.is;
|
|
|
|
offset_type offset = _offset;
|
|
if (is_relative)
|
|
offset += is.GetOffset();
|
|
|
|
try {
|
|
is.LockSeek(offset);
|
|
return is.GetOffset();
|
|
} catch (...) {
|
|
LogError(std::current_exception(), "Seek failed");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static AFvirtualfile *
|
|
setup_virtual_fops(AudioFileInputStream &afis) noexcept
|
|
{
|
|
AFvirtualfile *vf = (AFvirtualfile *)malloc(sizeof(*vf));
|
|
vf->closure = &afis;
|
|
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;
|
|
}
|
|
|
|
gcc_const
|
|
static SampleFormat
|
|
audiofile_bits_to_sample_format(int bits) noexcept
|
|
{
|
|
switch (bits) {
|
|
case 8:
|
|
return SampleFormat::S8;
|
|
|
|
case 16:
|
|
return SampleFormat::S16;
|
|
|
|
case 24:
|
|
return SampleFormat::S24_P32;
|
|
|
|
case 32:
|
|
return SampleFormat::S32;
|
|
}
|
|
|
|
return SampleFormat::UNDEFINED;
|
|
}
|
|
|
|
static SampleFormat
|
|
audiofile_setup_sample_format(AFfilehandle af_fp) noexcept
|
|
{
|
|
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 AudioFormat
|
|
CheckAudioFormat(AFfilehandle fh)
|
|
{
|
|
return CheckAudioFormat(afGetRate(fh, AF_DEFAULT_TRACK),
|
|
audiofile_setup_sample_format(fh),
|
|
afGetVirtualChannels(fh, AF_DEFAULT_TRACK));
|
|
}
|
|
|
|
static void
|
|
audiofile_stream_decode(DecoderClient &client, InputStream &is)
|
|
{
|
|
if (!is.IsSeekable() || !is.KnownSize()) {
|
|
LogWarning(audiofile_domain, "not seekable");
|
|
return;
|
|
}
|
|
|
|
AudioFileInputStream afis{&client, is};
|
|
AFvirtualfile *const vf = setup_virtual_fops(afis);
|
|
|
|
const AFfilehandle fh = afOpenVirtualFile(vf, "r", nullptr);
|
|
if (fh == AF_NULL_FILEHANDLE)
|
|
return;
|
|
|
|
AtScopeExit(fh) { afCloseFile(fh); };
|
|
|
|
const auto audio_format = CheckAudioFormat(fh);
|
|
const auto total_time = audiofile_get_duration(fh);
|
|
|
|
const uint16_t kbit_rate = (uint16_t)
|
|
(is.GetSize() * uint64_t(8) / total_time.ToMS());
|
|
|
|
const unsigned frame_size = (unsigned)
|
|
afGetVirtualFrameSize(fh, AF_DEFAULT_TRACK, true);
|
|
|
|
client.Ready(audio_format, true, total_time);
|
|
|
|
DecoderCommand cmd;
|
|
do {
|
|
uint8_t chunk[8192];
|
|
const int nframes =
|
|
afReadFrames(fh, AF_DEFAULT_TRACK, chunk,
|
|
sizeof(chunk) / frame_size);
|
|
if (nframes <= 0)
|
|
break;
|
|
|
|
cmd = client.SubmitData(nullptr,
|
|
chunk, nframes * frame_size,
|
|
kbit_rate);
|
|
|
|
if (cmd == DecoderCommand::SEEK) {
|
|
AFframecount frame = client.GetSeekFrame();
|
|
afSeekFrame(fh, AF_DEFAULT_TRACK, frame);
|
|
|
|
client.CommandFinished();
|
|
cmd = DecoderCommand::NONE;
|
|
}
|
|
} while (cmd == DecoderCommand::NONE);
|
|
}
|
|
|
|
static bool
|
|
audiofile_scan_stream(InputStream &is, TagHandler &handler) noexcept
|
|
{
|
|
if (!is.IsSeekable() || !is.KnownSize())
|
|
return false;
|
|
|
|
AudioFileInputStream afis{nullptr, is};
|
|
AFvirtualfile *vf = setup_virtual_fops(afis);
|
|
AFfilehandle fh = afOpenVirtualFile(vf, "r", nullptr);
|
|
if (fh == AF_NULL_FILEHANDLE)
|
|
return false;
|
|
|
|
AtScopeExit(fh) { afCloseFile(fh); };
|
|
|
|
handler.OnDuration(audiofile_get_duration(fh));
|
|
|
|
try {
|
|
handler.OnAudioFormat(CheckAudioFormat(fh));
|
|
} catch (...) {
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
static const char *const audiofile_suffixes[] = {
|
|
"wav", "au", "aiff", "aif", nullptr
|
|
};
|
|
|
|
static const char *const audiofile_mime_types[] = {
|
|
"audio/wav",
|
|
"audio/aiff",
|
|
"audio/x-wav",
|
|
"audio/x-aiff",
|
|
nullptr
|
|
};
|
|
|
|
const struct DecoderPlugin audiofile_decoder_plugin = {
|
|
"audiofile",
|
|
audiofile_init,
|
|
nullptr,
|
|
audiofile_stream_decode,
|
|
nullptr,
|
|
nullptr,
|
|
audiofile_scan_stream,
|
|
nullptr,
|
|
audiofile_suffixes,
|
|
audiofile_mime_types,
|
|
};
|