/*
  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 "ingen/Store.hpp"
#include "ingen/URIs.hpp"
#include "raul/Maid.hpp"
#include "raul/Path.hpp"

#include "Broadcaster.hpp"
#include "Driver.hpp"
#include "Engine.hpp"
#include "GraphImpl.hpp"
#include "PreProcessContext.hpp"
#include "events/CreateGraph.hpp"
#include "events/CreatePort.hpp"

namespace Ingen {
namespace Server {
namespace Events {

CreateGraph::CreateGraph(Engine&                     engine,
                         SPtr<Interface>             client,
                         int32_t                     id,
                         SampleCount                 timestamp,
                         const Raul::Path&           path,
                         const Resource::Properties& properties)
	: Event(engine, client, id, timestamp)
	, _path(path)
	, _properties(properties)
	, _graph(NULL)
	, _parent(NULL)
	, _compiled_graph(NULL)
{}

void
CreateGraph::build_child_events()
{
	const Ingen::URIs& uris = _engine.world()->uris();

	// Properties common to both ports
	Resource::Properties control_properties;
	control_properties.put(uris.atom_bufferType, uris.atom_Sequence);
	control_properties.put(uris.atom_supports, uris.patch_Message);
	control_properties.put(uris.lv2_designation, uris.lv2_control);
	control_properties.put(uris.lv2_portProperty, uris.lv2_connectionOptional);
	control_properties.put(uris.rdf_type, uris.atom_AtomPort);
	control_properties.put(uris.rsz_minimumSize, uris.forge.make(4096));

	// Add control port (message receive)
	Resource::Properties in_properties(control_properties);
	in_properties.put(uris.lv2_index, uris.forge.make(0));
	in_properties.put(uris.lv2_name, uris.forge.alloc("Control"));
	in_properties.put(uris.rdf_type, uris.lv2_InputPort);
	in_properties.put(uris.ingen_canvasX, uris.forge.make(32.0f),
	                  Resource::Graph::EXTERNAL);
	in_properties.put(uris.ingen_canvasY, uris.forge.make(32.0f),
	                  Resource::Graph::EXTERNAL);

	_child_events.push_back(
		SPtr<Events::CreatePort>(
			new Events::CreatePort(
				_engine, _request_client, -1, _time,
				_path.child(Raul::Symbol("control")),
				in_properties)));

	// Add notify port (message respond)
	Resource::Properties out_properties(control_properties);
	out_properties.put(uris.lv2_index, uris.forge.make(1));
	out_properties.put(uris.lv2_name, uris.forge.alloc("Notify"));
	out_properties.put(uris.rdf_type, uris.lv2_OutputPort);
	out_properties.put(uris.ingen_canvasX, uris.forge.make(128.0f),
	                   Resource::Graph::EXTERNAL);
	out_properties.put(uris.ingen_canvasY, uris.forge.make(32.0f),
	                   Resource::Graph::EXTERNAL);

	_child_events.push_back(
		SPtr<Events::CreatePort>(
			new Events::CreatePort(_engine, _request_client, -1, _time,
			                       _path.child(Raul::Symbol("notify")),
			                       out_properties)));
}

bool
CreateGraph::pre_process(PreProcessContext& ctx)
{
	if (_engine.store()->get(_path)) {
		return Event::pre_process_done(Status::EXISTS, _path);
	}

	if (!_path.is_root()) {
		const Raul::Path up(_path.parent());
		if (!(_parent = dynamic_cast<GraphImpl*>(_engine.store()->get(up)))) {
			return Event::pre_process_done(Status::PARENT_NOT_FOUND, up);
		}
	}

	const Ingen::URIs& uris = _engine.world()->uris();

	typedef Resource::Properties::const_iterator iterator;

	uint32_t ext_poly = 1;
	uint32_t int_poly = 1;
	iterator p        = _properties.find(uris.ingen_polyphony);
	if (p != _properties.end() && p->second.type() == uris.forge.Int) {
		int_poly = p->second.get<int32_t>();
	}

	if (int_poly < 1 || int_poly > 128) {
		return Event::pre_process_done(Status::INVALID_POLY, _path);
	}

	if (!_parent || int_poly == _parent->internal_poly()) {
		ext_poly = int_poly;
	}

	const Raul::Symbol symbol(_path.is_root() ? "graph" : _path.symbol());

	// Get graph prototype
	iterator t = _properties.find(uris.lv2_prototype);
	if (t == _properties.end()) {
		t = _properties.find(uris.lv2_prototype);
	}

	if (t != _properties.end() &&
	    uris.forge.is_uri(t->second) &&
	    Raul::URI::is_valid(uris.forge.str(t->second, false)) &&
	    Node::uri_is_path(Raul::URI(uris.forge.str(t->second, false)))) {
		// Create a duplicate of an existing graph
		const Raul::URI prototype(uris.forge.str(t->second, false));
		GraphImpl*      ancestor = dynamic_cast<GraphImpl*>(
			_engine.store()->get(Node::uri_to_path(prototype)));
		if (!ancestor) {
			return Event::pre_process_done(Status::PROTOTYPE_NOT_FOUND, prototype);
		} else if (!(_graph = dynamic_cast<GraphImpl*>(
			      ancestor->duplicate(_engine, symbol, _parent)))) {
			return Event::pre_process_done(Status::CREATION_FAILED, _path);
		}
	} else {
		// Create a new graph
		_graph = new GraphImpl(_engine, symbol, ext_poly, _parent,
		                       _engine.driver()->sample_rate(), int_poly);
		_graph->add_property(uris.rdf_type, uris.ingen_Graph.urid);
		_graph->add_property(uris.rdf_type,
		                     Resource::Property(uris.ingen_Block,
		                                        Resource::Graph::EXTERNAL));
	}

	_graph->set_properties(_properties);

	if (_parent) {
		// Add graph to parent
		_parent->add_block(*_graph);
		if (_parent->enabled()) {
			_graph->enable();
		}
		if (ctx.must_compile(_parent)) {
			_compiled_graph = CompiledGraph::compile(_parent);
		}
	}

	_graph->activate(*_engine.buffer_factory());

	// Insert into store and build update to send to clients
	_engine.store()->add(_graph);
	_update.put_graph(_graph);
	for (BlockImpl& block : _graph->blocks()) {
		_engine.store()->add(&block);
	}

	// Build and pre-process child events to create standard ports
	build_child_events();
	for (SPtr<Event> ev : _child_events) {
		ev->pre_process(ctx);
	}

	return Event::pre_process_done(Status::SUCCESS);
}

void
CreateGraph::execute(RunContext& context)
{
	if (_graph) {
		if (_parent && _compiled_graph) {
			_compiled_graph = _parent->swap_compiled_graph(_compiled_graph);
		}

		for (SPtr<Event> ev : _child_events) {
			ev->execute(context);
		}
	}
}

void
CreateGraph::post_process()
{
	Broadcaster::Transfer t(*_engine.broadcaster());
	if (respond() == Status::SUCCESS) {
		_update.send(_engine.broadcaster());
	}

	if (_graph) {
		for (SPtr<Event> ev : _child_events) {
			ev->post_process();
		}
	}
	_child_events.clear();

	delete _compiled_graph;
}

void
CreateGraph::undo(Interface& target)
{
	target.del(_graph->uri());
}

} // namespace Events
} // namespace Server
} // namespace Ingen