tag/Chromaprint: OO wrapper for a ChromaprintContext

1 files changed, 273 insertions, 0 deletions

diff --git a/test/RunChromaprint.cxx b/test/RunChromaprint.cxx
new file mode 100644
index 000000000..311ad4eb7
--- /dev/null
+++ b/test/RunChromaprint.cxx
@@ -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;
+}