/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            lv2.cc
 *
 *  Thu Feb 12 14:55:41 CET 2015
 *  Copyright 2015 Bent Bisballe Nyeng
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *  This file is part of DrumGizmo.
 *
 *  DrumGizmo is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation; either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  DrumGizmo 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 Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with DrumGizmo; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */
#include "dgunit.h"

#include <config.h>

#include <thread>
#include <chrono>
#include <memory.h>
#include <stdio.h>
#include <arpa/inet.h>

#include "drumkit_creator.h"
#include "lv2_test_host.h"

#define DG_URI "http://drumgizmo.org/lv2"

enum class Ports {
	FreeWheel = 0,
	Latency,
	MidiPort,
	AudioPortOffset,
};

/**
 * Tests that should be performed:
 * -------------------------------
 * - Run without port connects (shouldn't crash)
 * - Run without output ports connects (shouldn't crash)
 * - Run with buffer size 0
 * - Run with VERY LARGE buffer size (>1MB?)
 * - Run with buffer size a prime number (and thereby not power of 2)
 * - Run with HUGE number of midi events in one buffer (10000)
 */
class test_lv2 : public DGUnit
{
public:
	test_lv2()
	{
		DGUNIT_TEST(test_lv2::open_and_verify);
		DGUNIT_TEST(test_lv2::run_no_ports_connected);
		DGUNIT_TEST(test_lv2::run_no_output_ports_connected);
		DGUNIT_TEST(test_lv2::test1);
	}

	DrumkitCreator drumkit_creator;

	void open_and_verify()
	{
		int res;

		LV2TestHost h(LV2_PATH);

		res = h.open(DG_URI);
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.verify();
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.close();
		DGUNIT_ASSERT_EQUAL(0, res);
	}

	void run_no_ports_connected()
	{
		int res;

		LV2TestHost h(LV2_PATH);

		res = h.open(DG_URI);
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.verify();
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.createInstance(44100);
		DGUNIT_ASSERT_EQUAL(0, res);

		const char config_fmt[] =
			"<config>\n"
			"  <value name=\"drumkitfile\">%s</value>\n"
			"  <value name=\"midimapfile\">%s</value>\n"
			"  <value name=\"enable_velocity_modifier\">%s</value>\n"
			"  <value name=\"velocity_modifier_falloff\">%f</value>\n"
			"  <value name=\"velocity_modifier_weight\">%f</value>\n"
			"  <value name=\"enable_velocity_randomiser\">%s</value>\n"
			"  <value name=\"velocity_randomiser_weight\">%f</value>\n"
			"  <value name=\"enable_resampling\">%s</value>\n"
			"  <value name=\"enable_resampling\">%s</value>\n"
			"  <value name=\"disk_cache_upper_limit\">%d</value>\n"
			"  <value name=\"disk_cache_chunk_size\">%d</value>\n"
			"  <value name=\"disk_cache_enable\">%s</value>\n"
			"  <value name=\"enable_bleed_control\">%s</value>\n"
			"  <value name=\"master_bleed\">%f</value>\n"
			"  <value name=\"enable_latency_modifier\">%s</value>\n"
			"  <value name=\"latency_max\">%d</value>\n"
			"  <value name=\"latency_laid_back\">%d</value>\n"
			"  <value name=\"latency_stddev\">%f</value>\n"
			"  <value name=\"latency_regain\">%f</value>\n"
			"</config>";

		// Create drumkit
		auto kit1_file = drumkit_creator.createStdKit("kit1");

		auto midimapfile = drumkit_creator.createStdMidimap("midimap");
		bool enable_velocity_modifier = true;
		float velocity_modifier_falloff = 0.5;
		float velocity_modifier_weight = 0.25;
		bool enable_velocity_randomiser = false;
		float velocity_randomiser_weight = 0.1;
		bool enable_resampling = false;
		int disk_cache_upper_limit = 1024 * 1024;
		int disk_cache_chunk_size = 1024 * 1024 * 1024;
		bool disk_cache_enable = true;
		bool enable_bleed_control = false;
		float master_bleed = 1.0f;
		bool enable_latency_modifier = false;
		int latency_max = 0u;
		int latency_laid_back = 0u;
		float latency_stddev = 100.0f;
		float latency_regain = 0.9f;

		char config[sizeof(config_fmt) * 2];
		sprintf(config, config_fmt,
		        kit1_file.c_str(),
		        midimapfile.c_str(),
		        enable_velocity_modifier?"true":"false",
		        velocity_modifier_falloff,
		        velocity_modifier_weight,
		        enable_velocity_randomiser?"true":"false",
		        velocity_randomiser_weight,
		        enable_resampling?"true":"false",
		        enable_resampling?"true":"false",
		        disk_cache_upper_limit,
		        disk_cache_chunk_size,
		        disk_cache_enable?"true":"false",
		        enable_bleed_control?"true":"false",
		        master_bleed,
		        enable_latency_modifier?"true":"false",
		        latency_max,
		        latency_laid_back,
		        latency_stddev,
		        latency_regain);

		res = h.loadConfig(config, strlen(config));
		DGUNIT_ASSERT_EQUAL(0, res);

		// run for 1 samples to trigger kit loading
		res = h.run(1);
		DGUNIT_ASSERT_EQUAL(0, res);
		std::this_thread::sleep_for(std::chrono::milliseconds(1)); // wait for kit to get loaded (async),

		res = h.run(100);
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.destroyInstance();
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.close();
		DGUNIT_ASSERT_EQUAL(0, res);
	}

	void run_no_output_ports_connected()
	{
		int res;

		LV2TestHost h(LV2_PATH);

		res = h.open(DG_URI);
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.verify();
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.createInstance(44100);
		DGUNIT_ASSERT_EQUAL(0, res);

		const char config_fmt[] =
			"<config>\n"
			"  <value name=\"drumkitfile\">%s</value>\n"
			"  <value name=\"midimapfile\">%s</value>\n"
			"  <value name=\"enable_velocity_modifier\">%s</value>\n"
			"  <value name=\"velocity_modifier_falloff\">%f</value>\n"
			"  <value name=\"velocity_modifier_weight\">%f</value>\n"
			"  <value name=\"enable_velocity_randomiser\">%s</value>\n"
			"  <value name=\"velocity_randomiser_weight\">%f</value>\n"
			"  <value name=\"enable_resampling\">%s</value>\n"
			"  <value name=\"enable_resampling\">%s</value>\n"
			"  <value name=\"disk_cache_upper_limit\">%d</value>\n"
			"  <value name=\"disk_cache_chunk_size\">%d</value>\n"
			"  <value name=\"disk_cache_enable\">%s</value>\n"
			"  <value name=\"enable_bleed_control\">%s</value>\n"
			"  <value name=\"master_bleed\">%f</value>\n"
			"  <value name=\"enable_latency_modifier\">%s</value>\n"
			"  <value name=\"latency_max\">%d</value>\n"
			"  <value name=\"latency_laid_back\">%d</value>\n"
			"  <value name=\"latency_stddev\">%f</value>\n"
			"  <value name=\"latency_regain\">%f</value>\n"
			"</config>";

		// Create drumkit
		auto kit1_file = drumkit_creator.createStdKit("kit1");

		auto midimapfile = drumkit_creator.createStdMidimap("midimap");
		bool enable_velocity_modifier = true;
		float velocity_modifier_falloff = 0.5;
		float velocity_modifier_weight = 0.25;
		bool enable_velocity_randomiser = false;
		float velocity_randomiser_weight = 0.1;
		bool enable_resampling = false;
		int disk_cache_upper_limit = 1024 * 1024;
		int disk_cache_chunk_size = 1024 * 1024 * 1024;
		bool disk_cache_enable = true;
		bool enable_bleed_control = false;
		float master_bleed = 1.0f;
		bool enable_latency_modifier = false;
		int latency_max = 0u;
		int latency_laid_back = 0u;
		float latency_stddev = 100.0f;
		float latency_regain = 0.9f;

		char config[sizeof(config_fmt) * 2];
		sprintf(config, config_fmt,
		        kit1_file.c_str(),
		        midimapfile.c_str(),
		        enable_velocity_modifier?"true":"false",
		        velocity_modifier_falloff,
		        velocity_modifier_weight,
		        enable_velocity_randomiser?"true":"false",
		        velocity_randomiser_weight,
		        enable_resampling?"true":"false",
		        enable_resampling?"true":"false",
		        disk_cache_upper_limit,
		        disk_cache_chunk_size,
		        disk_cache_enable?"true":"false",
		        enable_bleed_control?"true":"false",
		        master_bleed,
		        enable_latency_modifier?"true":"false",
		        latency_max,
		        latency_laid_back,
		        latency_stddev,
		        latency_regain);

		res = h.loadConfig(config, strlen(config));
		DGUNIT_ASSERT_EQUAL(0, res);

		// Port buffers:
		char sequence_buffer[4096];
		bool freeWheel = false;

		// Free wheel port
		res = h.connectPort((int)Ports::FreeWheel, (void*)&freeWheel);

		LV2TestHost::Sequence seq(sequence_buffer, sizeof(sequence_buffer));
		res = h.connectPort((int)Ports::MidiPort, seq.data());
		DGUNIT_ASSERT_EQUAL(0, res);

		// run for 1 samples to trigger kit loading
		res = h.run(1);
		DGUNIT_ASSERT_EQUAL(0, res);
		std::this_thread::sleep_for(std::chrono::milliseconds(1)); // wait for kit to get loaded (async),

		seq.addMidiNote(5, 1, 127);
		res = h.run(100);
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.destroyInstance();
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.close();
		DGUNIT_ASSERT_EQUAL(0, res);
	}

	void test1()
	{
		int res;

		LV2TestHost h(LV2_PATH);

		res = h.open(DG_URI);
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.verify();
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.createInstance(44100);
		DGUNIT_ASSERT_EQUAL(0, res);

		const char config_fmt[] =
			"<config>\n"
			"  <value name=\"drumkitfile\">%s</value>\n"
			"  <value name=\"midimapfile\">%s</value>\n"
			"  <value name=\"enable_velocity_modifier\">%s</value>\n"
			"  <value name=\"velocity_modifier_falloff\">%f</value>\n"
			"  <value name=\"velocity_modifier_weight\">%f</value>\n"
			"  <value name=\"enable_velocity_randomiser\">%s</value>\n"
			"  <value name=\"velocity_randomiser_weight\">%f</value>\n"
			"  <value name=\"enable_resampling\">%s</value>\n"
			"  <value name=\"enable_resampling\">%s</value>\n"
			"  <value name=\"disk_cache_upper_limit\">%d</value>\n"
			"  <value name=\"disk_cache_chunk_size\">%d</value>\n"
			"  <value name=\"disk_cache_enable\">%s</value>\n"
			"  <value name=\"enable_bleed_control\">%s</value>\n"
			"  <value name=\"master_bleed\">%f</value>\n"
			"  <value name=\"enable_latency_modifier\">%s</value>\n"
			"  <value name=\"latency_max\">%d</value>\n"
			"  <value name=\"latency_laid_back\">%d</value>\n"
			"  <value name=\"latency_stddev\">%f</value>\n"
			"  <value name=\"latency_regain\">%f</value>\n"
			"</config>";

		// Create drumkit
		auto kit1_file = drumkit_creator.createStdKit("kit1");

		auto midimapfile = drumkit_creator.createStdMidimap("midimap");
		bool enable_velocity_modifier = true;
		float velocity_modifier_falloff = 0.5;
		float velocity_modifier_weight = 0.25;
		bool enable_velocity_randomiser = false;
		float velocity_randomiser_weight = 0.1;
		bool enable_resampling = false;
		int disk_cache_upper_limit = 1024 * 1024;
		int disk_cache_chunk_size = 1024 * 1024 * 1024;
		bool disk_cache_enable = true;
		bool enable_bleed_control = false;
		float master_bleed = 1.0f;
		bool enable_latency_modifier = false;
		int latency_max = 0u;
		int latency_laid_back = 0u;
		float latency_stddev = 100.0f;
		float latency_regain = 0.9f;

		char config[sizeof(config_fmt) * 2];
		sprintf(config, config_fmt,
		        kit1_file.c_str(),
		        midimapfile.c_str(),
		        enable_velocity_modifier?"true":"false",
		        velocity_modifier_falloff,
		        velocity_modifier_weight,
		        enable_velocity_randomiser?"true":"false",
		        velocity_randomiser_weight,
		        enable_resampling?"true":"false",
		        enable_resampling?"true":"false",
		        disk_cache_upper_limit,
		        disk_cache_chunk_size,
		        disk_cache_enable?"true":"false",
		        enable_bleed_control?"true":"false",
		        master_bleed,
		        enable_latency_modifier?"true":"false",
		        latency_max,
		        latency_laid_back,
		        latency_stddev,
		        latency_regain);

		res = h.loadConfig(config, strlen(config));
		DGUNIT_ASSERT_EQUAL(0, res);

		// Port buffers:
		char sequence_buffer[4096];
		float pcm_buffer[NUM_CHANNELS][10];
		bool freeWheel = true;

		// Free wheel port
		res = h.connectPort((int)Ports::FreeWheel, (void*)&freeWheel);

		LV2TestHost::Sequence seq(sequence_buffer, sizeof(sequence_buffer));
		res = h.connectPort((int)Ports::MidiPort, seq.data());
		DGUNIT_ASSERT_EQUAL(0, res);

		for(int i = 0; i < NUM_CHANNELS; ++i)
		{
			for(int j = 0; j < 10; ++j)
			{
				pcm_buffer[i][j] = 0.42;
			}
			res += h.connectPort((int)Ports::AudioPortOffset + i, pcm_buffer[i]);
		}
		DGUNIT_ASSERT_EQUAL(0, res);

		// run for 1 samples to trigger kit loading
		res = h.run(1);
		DGUNIT_ASSERT_EQUAL(0, res);
		std::this_thread::sleep_for(std::chrono::seconds(1));  // wait for kit to get loaded (async),

		seq.addMidiNote(5, 1, 127);
		for(int i = 0; i < 10; i++)
		{
			res = h.run(10);
			std::this_thread::sleep_for(std::chrono::milliseconds(1));
			DGUNIT_ASSERT_EQUAL(0, res);

			//printf("Iteration:\n");
			//for(int k = 0; k < 16; k++) {
			//	printf("#%d ", k);
			//	for(int j = 0; j < 10; j++) printf("[%f]", pcm_buffer[k][j]);
			//	printf("\n");
			//}
			//printf("\n");

			seq.clear();
		}


		seq.addMidiNote(5, 1, 127);
		res = h.run(10);
		std::this_thread::sleep_for(std::chrono::milliseconds(1));
		DGUNIT_ASSERT_EQUAL(0, res);

		/*
		printf("Iteration:\n");
		for(int k = 0; k < 4; k++) {
			printf("#%d ", k);
			for(int j = 0; j < 10; j++) printf("[%f]", pcm_buffer[k][j]);
			printf("\n");
		}
		printf("\n");
		*/

		union {
			float f;
			unsigned int u;
		} comp_val;

		comp_val.u = 1040744448; // floating point value 0.133301....

		for(int k = 0; k < 4; k++)
		{
			for(int j = 0; j < 10; j++)
			{
				DGUNIT_ASSERT_EQUAL(((j==5)?comp_val.f:0), pcm_buffer[k][j]);
			}
		}
		seq.clear();

		res = h.destroyInstance();
		DGUNIT_ASSERT_EQUAL(0, res);

		res = h.close();
		DGUNIT_ASSERT_EQUAL(0, res);
	}
};

// Registers the fixture into the 'registry'
static test_lv2 test;