tag/Chromaprint: OO wrapper for a ChromaprintContext
This commit is contained in:
Max Kellermann
273
test/RunChromaprint.cxx
Normal file
273
test/RunChromaprint.cxx
Normal file
@@ -0,0 +1,273 @@
|
||||
/*
|
||||
* 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 "config.h"
|
||||
#include "config/File.hxx"
|
||||
#include "config/Migrate.hxx"
|
||||
#include "config/Data.hxx"
|
||||
#include "tag/Chromaprint.hxx"
|
||||
#include "pcm/PcmConvert.hxx"
|
||||
#include "event/Thread.hxx"
|
||||
#include "decoder/DecoderList.hxx"
|
||||
#include "decoder/DecoderPlugin.hxx"
|
||||
#include "decoder/Client.hxx"
|
||||
#include "input/Init.hxx"
|
||||
#include "input/InputStream.hxx"
|
||||
#include "fs/Path.hxx"
|
||||
#include "AudioFormat.hxx"
|
||||
#include "util/OptionDef.hxx"
|
||||
#include "util/OptionParser.hxx"
|
||||
#include "util/PrintException.hxx"
|
||||
#include "Log.hxx"
|
||||
#include "LogBackend.hxx"
|
||||
|
||||
#include <stdexcept>
|
||||
|
||||
#include <assert.h>
|
||||
#include <unistd.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
|
||||
struct CommandLine {
|
||||
const char *decoder = nullptr;
|
||||
const char *uri = nullptr;
|
||||
|
||||
Path config_path = nullptr;
|
||||
|
||||
bool verbose = false;
|
||||
};
|
||||
|
||||
enum Option {
|
||||
OPTION_CONFIG,
|
||||
OPTION_VERBOSE,
|
||||
};
|
||||
|
||||
static constexpr OptionDef option_defs[] = {
|
||||
{"config", 0, true, "Load a MPD configuration file"},
|
||||
{"verbose", 'v', false, "Verbose logging"},
|
||||
};
|
||||
|
||||
static CommandLine
|
||||
ParseCommandLine(int argc, char **argv)
|
||||
{
|
||||
CommandLine c;
|
||||
|
||||
OptionParser option_parser(option_defs, argc, argv);
|
||||
while (auto o = option_parser.Next()) {
|
||||
switch (Option(o.index)) {
|
||||
case OPTION_CONFIG:
|
||||
c.config_path = Path::FromFS(o.value);
|
||||
break;
|
||||
|
||||
case OPTION_VERBOSE:
|
||||
c.verbose = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
auto args = option_parser.GetRemaining();
|
||||
if (args.size != 2)
|
||||
throw std::runtime_error("Usage: RunChromaprint [--verbose] [--config=FILE] DECODER URI");
|
||||
|
||||
c.decoder = args[0];
|
||||
c.uri = args[1];
|
||||
return c;
|
||||
}
|
||||
|
||||
class GlobalInit {
|
||||
ConfigData config;
|
||||
EventThread io_thread;
|
||||
|
||||
public:
|
||||
GlobalInit(Path config_path, bool verbose) {
|
||||
SetLogThreshold(verbose ? LogLevel::DEBUG : LogLevel::INFO);
|
||||
|
||||
if (!config_path.IsNull()) {
|
||||
ReadConfigFile(config, config_path);
|
||||
Migrate(config);
|
||||
}
|
||||
|
||||
io_thread.Start();
|
||||
|
||||
input_stream_global_init(config,
|
||||
io_thread.GetEventLoop());
|
||||
decoder_plugin_init_all(config);
|
||||
|
||||
pcm_convert_global_init(config);
|
||||
}
|
||||
|
||||
~GlobalInit() {
|
||||
decoder_plugin_deinit_all();
|
||||
input_stream_global_finish();
|
||||
}
|
||||
};
|
||||
|
||||
class ChromaprintDecoderClient final : public DecoderClient {
|
||||
bool ready = false;
|
||||
|
||||
bool need_convert = false;
|
||||
|
||||
PcmConvert convert;
|
||||
|
||||
Chromaprint::Context chromaprint;
|
||||
|
||||
uint64_t remaining_bytes;
|
||||
|
||||
public:
|
||||
Mutex mutex;
|
||||
|
||||
~ChromaprintDecoderClient() noexcept {
|
||||
if (need_convert)
|
||||
convert.Close();
|
||||
}
|
||||
|
||||
void PrintResult();
|
||||
|
||||
/* virtual methods from DecoderClient */
|
||||
void Ready(AudioFormat audio_format,
|
||||
bool seekable, SignedSongTime duration) override;
|
||||
|
||||
DecoderCommand GetCommand() noexcept override {
|
||||
return remaining_bytes > 0
|
||||
? DecoderCommand::NONE
|
||||
: DecoderCommand::STOP;
|
||||
}
|
||||
|
||||
void CommandFinished() override {}
|
||||
|
||||
SongTime GetSeekTime() noexcept override {
|
||||
return SongTime::zero();
|
||||
}
|
||||
|
||||
uint64_t GetSeekFrame() noexcept override {
|
||||
return 0;
|
||||
}
|
||||
|
||||
void SeekError() override {}
|
||||
|
||||
InputStreamPtr OpenUri(const char *) override {
|
||||
throw std::runtime_error("Not implemented");
|
||||
}
|
||||
|
||||
size_t Read(InputStream &is, void *buffer, size_t length) override {
|
||||
return is.LockRead(buffer, length);
|
||||
}
|
||||
|
||||
void SubmitTimestamp(FloatDuration) override {}
|
||||
DecoderCommand SubmitData(InputStream *is,
|
||||
const void *data, size_t length,
|
||||
uint16_t kbit_rate) override;
|
||||
|
||||
DecoderCommand SubmitTag(InputStream *, Tag &&) override {
|
||||
return GetCommand();
|
||||
}
|
||||
|
||||
void SubmitReplayGain(const ReplayGainInfo *) {}
|
||||
void SubmitMixRamp(MixRampInfo &&) {}
|
||||
};
|
||||
|
||||
void
|
||||
ChromaprintDecoderClient::PrintResult()
|
||||
{
|
||||
if (!ready)
|
||||
throw std::runtime_error("Decoding failed");
|
||||
|
||||
if (need_convert) {
|
||||
auto flushed = convert.Flush();
|
||||
auto data = ConstBuffer<int16_t>::FromVoid(flushed);
|
||||
chromaprint.Feed(data.data, data.size);
|
||||
}
|
||||
|
||||
chromaprint.Finish();
|
||||
|
||||
printf("%s\n", chromaprint.GetFingerprint().c_str());
|
||||
}
|
||||
|
||||
void
|
||||
ChromaprintDecoderClient::Ready(AudioFormat audio_format, bool, SignedSongTime)
|
||||
{
|
||||
/* feed the first two minutes into libchromaprint */
|
||||
remaining_bytes = audio_format.TimeToSize(std::chrono::minutes(2));
|
||||
|
||||
if (audio_format.format != SampleFormat::S16) {
|
||||
const AudioFormat src_audio_format = audio_format;
|
||||
audio_format.format = SampleFormat::S16;
|
||||
|
||||
convert.Open(src_audio_format, audio_format);
|
||||
need_convert = true;
|
||||
}
|
||||
|
||||
chromaprint.Start(audio_format.sample_rate, audio_format.channels);
|
||||
|
||||
ready = true;
|
||||
}
|
||||
|
||||
DecoderCommand
|
||||
ChromaprintDecoderClient::SubmitData(InputStream *,
|
||||
const void *_data, size_t length,
|
||||
uint16_t)
|
||||
{
|
||||
if (length > remaining_bytes)
|
||||
remaining_bytes = 0;
|
||||
else
|
||||
remaining_bytes -= length;
|
||||
|
||||
ConstBuffer<void> src{_data, length};
|
||||
ConstBuffer<int16_t> data;
|
||||
|
||||
if (need_convert) {
|
||||
auto result = convert.Convert(src);
|
||||
data = ConstBuffer<int16_t>::FromVoid(result);
|
||||
} else
|
||||
data = ConstBuffer<int16_t>::FromVoid(src);
|
||||
|
||||
chromaprint.Feed(data.data, data.size);
|
||||
|
||||
return GetCommand();
|
||||
}
|
||||
|
||||
int main(int argc, char **argv)
|
||||
try {
|
||||
const auto c = ParseCommandLine(argc, argv);
|
||||
|
||||
const GlobalInit init(c.config_path, c.verbose);
|
||||
|
||||
const DecoderPlugin *plugin = decoder_plugin_from_name(c.decoder);
|
||||
if (plugin == nullptr) {
|
||||
fprintf(stderr, "No such decoder: %s\n", c.decoder);
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
ChromaprintDecoderClient client;
|
||||
if (plugin->file_decode != nullptr) {
|
||||
plugin->FileDecode(client, Path::FromFS(c.uri));
|
||||
} else if (plugin->stream_decode != nullptr) {
|
||||
auto is = InputStream::OpenReady(c.uri, client.mutex);
|
||||
plugin->StreamDecode(client, *is);
|
||||
} else {
|
||||
fprintf(stderr, "Decoder plugin is not usable\n");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
client.PrintResult();
|
||||
return EXIT_SUCCESS;
|
||||
} catch (...) {
|
||||
PrintException(std::current_exception());
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
Reference in New Issue
Block a user