/*
  This file is part of Ingen.
  Copyright 2007-2016 David Robillard <http://drobilla.net/>

  Ingen is free software: you can redistribute it and/or modify it under the
  terms of the GNU Affero General Public License as published by the Free
  Software Foundation, either version 3 of the License, or 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 Affero General Public License for details.

  You should have received a copy of the GNU Affero General Public License
  along with Ingen.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <cassert>
#include <cmath>
#include <string>

#include "ingen/URIs.hpp"
#include "ingen/World.hpp"
#include "ingen/client/BlockModel.hpp"
#include "ingen/client/ParameterModel.hpp"

namespace Ingen {
namespace Client {

BlockModel::BlockModel(URIs&             uris,
                       SPtr<PluginModel> plugin,
                       const Raul::Path& path)
	: ObjectModel(uris, path)
	, _plugin_uri(plugin->uri())
	, _plugin(plugin)
	, _num_values(0)
	, _min_values(0)
	, _max_values(0)
{
}

BlockModel::BlockModel(URIs&             uris,
                       const Raul::URI&  plugin_uri,
                       const Raul::Path& path)
	: ObjectModel(uris, path)
	, _plugin_uri(plugin_uri)
	, _num_values(0)
	, _min_values(0)
	, _max_values(0)
{
}

BlockModel::BlockModel(const BlockModel& copy)
	: ObjectModel(copy)
	, _plugin_uri(copy._plugin_uri)
	, _num_values(copy._num_values)
	, _min_values((float*)malloc(sizeof(float) * _num_values))
	, _max_values((float*)malloc(sizeof(float) * _num_values))
{
	memcpy(_min_values, copy._min_values, sizeof(float) * _num_values);
	memcpy(_max_values, copy._max_values, sizeof(float) * _num_values);
}

BlockModel::~BlockModel()
{
	clear();
}

void
BlockModel::remove_port(SPtr<PortModel> port)
{
	for (Ports::iterator i = _ports.begin(); i != _ports.end(); ++i) {
		if ((*i) == port) {
			_ports.erase(i);
			break;
		}
	}
	_signal_removed_port.emit(port);
}

void
BlockModel::remove_port(const Raul::Path& port_path)
{
	for (Ports::iterator i = _ports.begin(); i != _ports.end(); ++i) {
		if ((*i)->path() == port_path) {
			_ports.erase(i);
			break;
		}
	}
}

void
BlockModel::clear()
{
	_ports.clear();
	assert(_ports.empty());
	delete[] _min_values;
	delete[] _max_values;
	_min_values = 0;
	_max_values = 0;
}

void
BlockModel::add_child(SPtr<ObjectModel> c)
{
	assert(c->parent().get() == this);

	//ObjectModel::add_child(c);

	SPtr<PortModel> pm = dynamic_ptr_cast<PortModel>(c);
	if (pm) {
		add_port(pm);
		return;
	}

	SPtr<ParameterModel> am = dynamic_ptr_cast<ParameterModel>(c);
	if (am) {
		add_parameter(am);
		return;
	}
}

bool
BlockModel::remove_child(SPtr<ObjectModel> c)
{
	assert(c->path().is_child_of(path()));
	assert(c->parent().get() == this);

	//bool ret = ObjectModel::remove_child(c);

	SPtr<PortModel> pm = dynamic_ptr_cast<PortModel>(c);
	assert(pm);
	remove_port(pm);

	//return ret;
	return true;
}

void
BlockModel::add_port(SPtr<PortModel> pm)
{
	assert(pm);
	assert(pm->path().is_child_of(path()));
	assert(pm->parent().get() == this);

	// Store should have handled this by merging the two
	assert(find(_ports.begin(), _ports.end(), pm) == _ports.end());

	_ports.push_back(pm);
	_signal_new_port.emit(pm);
}

void
BlockModel::add_parameter(SPtr<ParameterModel> pm)
{
	assert(pm);
	assert(pm->path().is_child_of(path()));
	assert(pm->parent().get() == this);

	// Store should have handled this by merging the two
	assert(find(_parameters.begin(), _parameters.end(), pm) == _parameters.end());

	_parameters.push_back(pm);
	_signal_new_parameter.emit(pm);
}

SPtr<const PortModel>
BlockModel::get_port(const Raul::Symbol& symbol) const
{
	for (auto p : _ports)
		if (p->symbol() == symbol)
			return p;
	return SPtr<PortModel>();
}

SPtr<const PortModel>
BlockModel::get_port(uint32_t index) const
{
	return _ports[index];
}

Ingen::Node*
BlockModel::port(uint32_t index) const
{
	assert(index < num_ports());
	return const_cast<Ingen::Node*>(
		dynamic_cast<const Ingen::Node*>(_ports[index].get()));
}

void
BlockModel::default_port_value_range(SPtr<const PortModel> port,
                                     float&                min,
                                     float&                max,
                                     uint32_t              srate) const
{
	// Default control values
	min = 0.0;
	max = 1.0;

	// Get range from client-side LV2 data
	if (_plugin && _plugin->lilv_plugin()) {
		if (!_min_values) {
			_num_values = lilv_plugin_get_num_ports(_plugin->lilv_plugin());
			_min_values = new float[_num_values];
			_max_values = new float[_num_values];
			lilv_plugin_get_port_ranges_float(_plugin->lilv_plugin(),
			                                  _min_values, _max_values, 0);
		}

		if (!std::isnan(_min_values[port->index()]))
			min = _min_values[port->index()];
		if (!std::isnan(_max_values[port->index()]))
			max = _max_values[port->index()];
	}

	if (port->port_property(_uris.lv2_sampleRate)) {
		min *= srate;
		max *= srate;
	}
}

void
BlockModel::port_value_range(SPtr<const PortModel> port,
                             float&                min,
                             float&                max,
                             uint32_t              srate) const
{
	assert(port->parent().get() == this);

	default_port_value_range(port, min, max);

	// Possibly overriden
	const Atom& min_atom = port->get_property(_uris.lv2_minimum);
	const Atom& max_atom = port->get_property(_uris.lv2_maximum);
	if (min_atom.type() == _uris.forge.Float)
		min = min_atom.get<float>();
	if (max_atom.type() == _uris.forge.Float)
		max = max_atom.get<float>();

	if (max <= min)
		max = min + 1.0;

	if (port->port_property(_uris.lv2_sampleRate)) {
		min *= srate;
		max *= srate;
	}
}

std::string
BlockModel::label() const
{
	const Atom& name_property = get_property(_uris.lv2_name);
	if (name_property.type() == _uris.forge.String) {
		return name_property.ptr<char>();
	} else if (plugin_model()) {
		return plugin_model()->human_name();
	} else {
		return symbol().c_str();
	}
}

std::string
BlockModel::port_label(SPtr<const PortModel> port) const
{
	const Atom& name = port->get_property(Raul::URI(LV2_CORE__name));
	if (name.is_valid() && name.type() == _uris.forge.String) {
		return name.ptr<char>();
	}

	if (_plugin && _plugin->lilv_plugin()) {
		LilvWorld*        w     = _plugin->lilv_world();
		const LilvPlugin* plug  = _plugin->lilv_plugin();
		LilvNode*         sym   = lilv_new_string(w, port->symbol().c_str());
		const LilvPort*   lport = lilv_plugin_get_port_by_symbol(plug, sym);
		if (lport) {
			LilvNode* lname = lilv_port_get_name(plug, lport);
			if (lname && lilv_node_is_string(lname)) {
				std::string ret(lilv_node_as_string(lname));
				lilv_node_free(lname);
				return ret;
			}
			lilv_node_free(lname);
		}
	}

	return port->symbol().c_str();
}

void
BlockModel::set(SPtr<ObjectModel> model)
{
	SPtr<BlockModel> block = dynamic_ptr_cast<BlockModel>(model);
	if (block) {
		_plugin_uri = block->_plugin_uri;
		_plugin     = block->_plugin;
	}

	ObjectModel::set(model);
}

} // namespace Client
} // namespace Ingen