/* This file is part of Ingen.
 * Copyright (C) 2007-2009 Dave Robillard <http://drobilla.net>
 *
 * Ingen 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.
 *
 * Ingen 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 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 St, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <math.h>
#include "raul/midi_events.h"
#include "internals/Controller.hpp"
#include "PostProcessor.hpp"
#include "events/MidiLearn.hpp"
#include "events/SendPortValue.hpp"
#include "InputPort.hpp"
#include "OutputPort.hpp"
#include "InternalPlugin.hpp"
#include "AudioBuffer.hpp"
#include "ProcessContext.hpp"
#include "EventBuffer.hpp"
#include "util.hpp"

using namespace std;

namespace Ingen {
namespace Internals {

using namespace Shared;

static InternalPlugin controller_plugin(NS_INTERNALS "Controller", "controller");

ControllerNode::ControllerNode(BufferFactory& bufs,
                               const string&  path,
                               bool           polyphonic,
                               PatchImpl*     parent,
                               SampleRate     srate,
                               size_t         buffer_size)
	: NodeBase(&controller_plugin, path, false, parent, srate, buffer_size)
	, _learning(false)
{
	_ports = new Raul::Array<PortImpl*>(6);

	_midi_in_port = new InputPort(bufs, this, "input", 0, 1, DataType::EVENTS, Raul::Atom(), _buffer_size);
	_ports->at(0) = _midi_in_port;

	_param_port = new InputPort(bufs, this, "controller", 1, 1, DataType::CONTROL, 0.0f, sizeof(Sample));
	_param_port->set_property("lv2:minimum", 0.0f);
	_param_port->set_property("lv2:maximum", 127.0f);
	_param_port->set_property("lv2:integer", true);
	_ports->at(1) = _param_port;

	_log_port = new InputPort(bufs, this, "logarithmic", 2, 1, DataType::CONTROL, 0.0f, sizeof(Sample));
	_log_port->set_property("lv2:toggled", true);
	_ports->at(2) = _log_port;

	_min_port = new InputPort(bufs, this, "minimum", 3, 1, DataType::CONTROL, 0.0f, sizeof(Sample));
	_ports->at(3) = _min_port;

	_max_port = new InputPort(bufs, this, "maximum", 4, 1, DataType::CONTROL, 1.0f, sizeof(Sample));
	_ports->at(4) = _max_port;

	_audio_port = new OutputPort(bufs, this, "ar_output", 5, 1, DataType::AUDIO, 0.0f, _buffer_size);
	_ports->at(5) = _audio_port;
}


void
ControllerNode::process(ProcessContext& context)
{
	NodeBase::pre_process(context);

	uint32_t frames = 0;
	uint32_t subframes = 0;
	uint16_t type = 0;
	uint16_t size = 0;
	uint8_t* buf = NULL;

	EventBuffer* const midi_in = (EventBuffer*)_midi_in_port->buffer(0).get();

	midi_in->rewind();

	while (midi_in->get_event(&frames, &subframes, &type, &size, &buf)) {
		// FIXME: type
		if (size >= 3 && (buf[0] & 0xF0) == MIDI_CMD_CONTROL)
			control(context, buf[1], buf[2], frames + context.start());

		midi_in->increment();
	}

	NodeBase::post_process(context);
}


void
ControllerNode::control(ProcessContext& context, uint8_t control_num, uint8_t val, FrameTime time)
{
	assert(time - context.start() < _buffer_size);

	Sample scaled_value;

	const Sample nval = (val / 127.0f); // normalized [0, 1]

	if (_learning) {
		_param_port->set_value(control_num);
		((AudioBuffer*)_param_port->buffer(0).get())->set_value(
				(float)control_num, context.start(), context.end());
		_param_port->broadcast_value(context, true);
		_learning = false;
	}

	const Sample min_port_val = ((AudioBuffer*)_min_port->buffer(0).get())->value_at(0);
	const Sample max_port_val = ((AudioBuffer*)_max_port->buffer(0).get())->value_at(0);
	const Sample log_port_val = ((AudioBuffer*)_log_port->buffer(0).get())->value_at(0);

	if (log_port_val > 0.0f) {
		// haaaaack, stupid negatives and logarithms
		Sample log_offset = 0;
		if (min_port_val < 0)
			log_offset = fabs(min_port_val);
		const Sample min = log(min_port_val + 1 + log_offset);
		const Sample max = log(max_port_val + 1 + log_offset);
		scaled_value = expf(nval * (max - min) + min) - 1 - log_offset;
	} else {
		scaled_value = ((nval) * (max_port_val - min_port_val)) + min_port_val;
	}

	if (control_num == ((AudioBuffer*)_param_port->buffer(0).get())->value_at(0))
		((AudioBuffer*)_audio_port->buffer(0).get())->set_value(scaled_value, context.start(), time);
}


} // namespace Ingen
} // namespace Internals