/**

 * Relational pipes

 * Copyright © 2020 František Kučera (Frantovo.cz, GlobalCode.info)

 *

 * 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, version 3 of the License.

 *

 * 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, see <http://www.gnu.org/licenses/>.

 */

#pragma once

#include <memory>

#include <atomic>

#include <string>

#include <cstring>

#include <vector>

#include <iostream>

#include <sstream>

#include <locale>

#include <codecvt>

#include <sys/mman.h>

#include <signal.h>

#include <unistd.h>

#include <jack/jack.h>

#include <jack/midiport.h>

#include <jack/ringbuffer.h>

#include <relpipe/common/type/typedefs.h>

#include <relpipe/reader/TypeId.h>

#include <relpipe/reader/handlers/RelationalReaderStringHandler.h>

#include <relpipe/reader/handlers/AttributeMetadata.h>

#include "Configuration.h"

#include "JackException.h"

namespace relpipe {

namespace out {

namespace jack {

int dequeueMessages(jack_nframes_t frames, void* arg);

class JackHandler : public relpipe::reader::handlers::RelationalReaderStringHandler {

private:

	Configuration& configuration;

	std::wstring_convert<std::codecvt_utf8<wchar_t>> convertor; // TODO: local system encoding

	std::atomic<bool> continueProcessing{true};

	/**

	 * Is passed through the ring buffer

	 * from the relpipe-reading thread to the jack-writing thread (callback).

	 */

	struct MidiMessage {

		uint8_t buffer[4096] = {0};

		uint32_t size;

		uint32_t time;

	};

	/**

	 * JACK callbacks (called from the real-time thread)

	 */

	class RealTimeContext {

	public:

		jack_client_t* jackClient = nullptr;

		jack_port_t* jackPort = nullptr;

		jack_ringbuffer_t* ringBuffer = nullptr;

		const int RING_BUFFER_SIZE = 100;

		int processCallback(jack_nframes_t frames) {

			const size_t queuedMessages = jack_ringbuffer_read_space(ringBuffer) / sizeof (MidiMessage);

			void* jackPortBuffer = jack_port_get_buffer(jackPort, frames); // jack_port_get_buffer() must be called outside the loop, otherwise it will multiply the MIDI events

			jack_midi_clear_buffer(jackPortBuffer); // TODO: clean buffer?

			for (size_t i = 0; i < queuedMessages && i < frames; i++) {

				// TODO: correct timing?

				MidiMessage m;

				jack_ringbuffer_read(ringBuffer, (char*) &m, sizeof (MidiMessage));

				jack_midi_data_t* midiData = jack_midi_event_reserve(jackPortBuffer, m.time, m.size);

				memcpy(midiData, m.buffer, m.size);

			}

			return 0;

		}

		}

	} realTimeContext;

	/**

	 * Temporary storage of values read from relational input.

	 */

	class RelationContext {

	public:

		bool skip = false;

		size_t indexOfEvent = -1;

		size_t indexOfChannel = -1;

		size_t indexOfNoteOn = -1;

		size_t indexOfNotePitch = -1;

		size_t indexOfNoteVelocity = -1;

		size_t indexOfControllerId = -1;

		size_t indexOfControllerValue = -1;

		size_t indexOfRaw = -1;

		size_t attributeCount = 0;

		class RecordContext {

		public:

			enum class Event {

				NOTE,

				CONTROL,

				SYSEX,

				UNKNOWN

			};

			static Event parseEventType(const relpipe::common::type::StringX name) {

				if (name == L"note") return Event::NOTE;

				else if (name == L"control") return Event::CONTROL;

				else if (name == L"sysex") return Event::SYSEX;

				else return Event::UNKNOWN;

			}

			Event event = Event::UNKNOWN;

			relpipe::common::type::Integer channel;

			relpipe::common::type::Boolean noteOn;

			relpipe::common::type::Integer notePitch;

			relpipe::common::type::Integer noteVelocity;

			relpipe::common::type::Integer controllerId;

			relpipe::common::type::Integer controllerValue;

			relpipe::common::type::StringX raw;

			size_t attributeIndex = 0;

		} recordContext;

	} relationContext;

	static void jackErrorCallback(const char * message) {

		std::wstring_convert < std::codecvt_utf8<wchar_t>> convertor; // TODO: local system encoding

		std::wcerr << L"JACK: " << convertor.from_bytes(message) << std::endl;

	}

public:

	JackHandler(Configuration& configuration) : configuration(configuration) {

		// Initialize JACK connection:

		std::string clientName = convertor.to_bytes(configuration.jackClientName);

		realTimeContext.jackClient = jack_client_open(clientName.c_str(), JackNullOption, nullptr);

		if (realTimeContext.jackClient == nullptr) throw JackException(L"Could not create JACK client.");

		realTimeContext.ringBuffer = jack_ringbuffer_create(realTimeContext.RING_BUFFER_SIZE * sizeof (MidiMessage));

		jack_set_process_callback(realTimeContext.jackClient, RealTimeContext::processCallback, &realTimeContext);

		// TODO: report also other events (connections etc.)

		jack_set_error_function(jackErrorCallback);

		jack_set_info_function(jackErrorCallback);

		realTimeContext.jackPort = jack_port_register(realTimeContext.jackClient, "output", JACK_DEFAULT_MIDI_TYPE, JackPortIsOutput, 0);

		if (realTimeContext.jackPort == nullptr) throw JackException(L"Could not register port.");

		if (mlockall(MCL_CURRENT | MCL_FUTURE)) fwprintf(stderr, L"Warning: Can not lock memory.\n");

		int jackError = jack_activate(realTimeContext.jackClient);

		if (jackError) throw JackException(L"Could not activate client.");

		// Wait for a port connection, because it does not make much sense to send MIDI events nowhere:

		while (jack_port_connected(realTimeContext.jackPort) < configuration.requiredJackConnections) usleep(10000);

	}

	void startRelation(const relpipe::common::type::StringX name, std::vector<relpipe::reader::handlers::AttributeMetadata> attributes) override {

		// TODO: validate metadata and prepare attribute mappings (names and types are important, order does not matter)

		relationContext = RelationContext();

		relationContext.attributeCount = attributes.size();

		for (int i = 0; i < relationContext.attributeCount; i++) {

			const relpipe::common::type::StringX attributeName = attributes[i].getAttributeName();

			if (attributeName == L"event") relationContext.indexOfEvent = i;

			else if (attributeName == L"channel") relationContext.indexOfChannel = i;

			else if (attributeName == L"controller_id") relationContext.indexOfControllerId = i;

			else if (attributeName == L"controller_value") relationContext.indexOfControllerValue = i;

			else if (attributeName == L"note_on") relationContext.indexOfNoteOn = i;

			else if (attributeName == L"note_pitch") relationContext.indexOfNotePitch = i;

			else if (attributeName == L"note_velocity") relationContext.indexOfNoteVelocity = i;

			else if (attributeName == L"raw") relationContext.indexOfRaw = i;

		}

		// TODO: check also data types and skipt relation if important attributes are missing

		// TODO: configurable relation name

		if (relationContext.indexOfRaw == -1) {

			relationContext.skip = true;

			fwprintf(stderr, L"Relation „%ls“ will be ignored because mandatory attribute „raw“ is missing.\n", name.c_str());

		}

	}

	void attribute(const relpipe::common::type::StringX& value) override {

		if (relationContext.skip) return;

		// TODO: switch to RelationalReaderStringHandler

		// TODO: if (continueProcessing) {} ?

		RelationContext& rel = relationContext;

		RelationContext::RecordContext& rec = rel.recordContext;

		if (rel.indexOfEvent == rec.attributeIndex) rec.event = rec.parseEventType(value);

		else if (rel.indexOfChannel == rec.attributeIndex) rec.channel = std::stoi(value);

		else if (rel.indexOfControllerId == rec.attributeIndex) rec.controllerId = std::stoi(value);

		else if (rel.indexOfControllerValue == rec.attributeIndex) rec.controllerValue = std::stoi(value);

		else if (rel.indexOfNoteOn == rec.attributeIndex) rec.noteOn = value == L"true";

		else if (rel.indexOfNotePitch == rec.attributeIndex) rec.notePitch = std::stoi(value);

		else if (rel.indexOfNoteVelocity == rec.attributeIndex) rec.noteVelocity = std::stoi(value);

		else if (rel.indexOfRaw == rec.attributeIndex) rec.raw = value;

		rec.attributeIndex++;

		if (rec.attributeIndex == rel.attributeCount) {

			while (jack_ringbuffer_write_space(realTimeContext.ringBuffer) < sizeof (MidiMessage)) usleep(1000); // should not happen, the real-time thread should be faster

			MidiMessage m;

			// TODO: correct timing?

			m.time = 0;

			if (rec.event == RelationContext::RecordContext::Event::NOTE) {

				m.size = 3;

				m.buffer[0] = (rec.noteOn ? 0x90 : 0x80) | rec.channel;

				m.buffer[1] = rec.notePitch;

				m.buffer[2] = rec.noteVelocity;

			} else if (rec.event == RelationContext::RecordContext::Event::CONTROL) {

				m.size = 3;

				m.buffer[0] = 0xB0 | rec.channel;

				m.buffer[1] = rec.controllerId;

				m.buffer[2] = rec.controllerValue;

			} else { // SysEx and other raw messages

				m.size = 0;

				size_t nibble = 0;

				for (int i = 0; i < rec.raw.size() && m.size < sizeof (m.buffer); i++) {

					wchar_t ch = rec.raw[i];

					if (ch == L' ') continue;

					else if (ch >= L'0' && ch <= L'9') m.buffer[m.size] += (ch - '0') /**/ << (nibble % 2 ? 0 : 4);

					else if (ch >= L'a' && ch <= L'f') m.buffer[m.size] += (ch - 'a' + 10) << (nibble % 2 ? 0 : 4);

					else if (ch >= L'A' && ch <= L'F') m.buffer[m.size] += (ch - 'A' + 10) << (nibble % 2 ? 0 : 4);

					else throw JackException(L"Invalid character in the hexadeximal sequence: „" + std::wstring(1, ch) + L"“.");

					nibble++;

					if (nibble % 2 == 0) m.size++;

				}

			}

			// TODO: if (m.size == 0) fwprintf(stderr, L"Missing data\n");

			jack_ringbuffer_write(realTimeContext.ringBuffer, (const char *) &m, sizeof (m));

			relationContext.recordContext = RelationContext::RecordContext();

		}

	}

	void endOfPipe() {

		// TODO: send optional (configurable) MIDI events

		// Wait until the ring buffer is empty

		while (continueProcessing && jack_ringbuffer_read_space(realTimeContext.ringBuffer)) usleep(1000);

		usleep(1000000);

		// TODO: better waiting (ringBuffer might be empty, but events have not been sent to JACK yet)

		// TODO: optionally mute all; probably enabled by default

	}

	void finish(int sig) {

		continueProcessing = false;

	}

	virtual ~JackHandler() {

		// Close JACK connection:

		jack_deactivate(realTimeContext.jackClient);

		jack_client_close(realTimeContext.jackClient);

		jack_ringbuffer_free(realTimeContext.ringBuffer);

	}

};

}

}

}