/*
  This file is part of Ingen.
  Copyright 2007-2012 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 "App.hpp"
#include "GraphTreeWindow.hpp"
#include "SubgraphModule.hpp"
#include "WindowFactory.hpp"
#include "ingen/Interface.hpp"
#include "ingen/Log.hpp"
#include "ingen/client/ClientStore.hpp"
#include "ingen/client/GraphModel.hpp"
#include "raul/Path.hpp"

using namespace std;

namespace Ingen {

using namespace Client;

namespace GUI {

GraphTreeWindow::GraphTreeWindow(BaseObjectType*                   cobject,
                                 const Glib::RefPtr<Gtk::Builder>& xml)
	: Window(cobject)
	, _app(NULL)
	, _enable_signal(true)
{
	xml->get_widget_derived("graphs_treeview", _graphs_treeview);

	_graph_treestore = Gtk::TreeStore::create(_graph_tree_columns);
	_graphs_treeview->set_window(this);
	_graphs_treeview->set_model(_graph_treestore);
	Gtk::TreeViewColumn* name_col = Gtk::manage(
		new Gtk::TreeViewColumn("Graph", _graph_tree_columns.name_col));
	Gtk::TreeViewColumn* enabled_col = Gtk::manage(
		new Gtk::TreeViewColumn("Run", _graph_tree_columns.enabled_col));
	name_col->set_resizable(true);
	name_col->set_expand(true);

	_graphs_treeview->append_column(*name_col);
	_graphs_treeview->append_column(*enabled_col);
	Gtk::CellRendererToggle* enabled_renderer = dynamic_cast<Gtk::CellRendererToggle*>(
		_graphs_treeview->get_column_cell_renderer(1));
	enabled_renderer->property_activatable() = true;

	_graph_tree_selection = _graphs_treeview->get_selection();

	_graphs_treeview->signal_row_activated().connect(
		sigc::mem_fun(this, &GraphTreeWindow::event_graph_activated));
	enabled_renderer->signal_toggled().connect(
		sigc::mem_fun(this, &GraphTreeWindow::event_graph_enabled_toggled));

	_graphs_treeview->columns_autosize();
}

void
GraphTreeWindow::init(App& app, ClientStore& store)
{
	_app = &app;
	store.signal_new_object().connect(
		sigc::mem_fun(this, &GraphTreeWindow::new_object));
}

void
GraphTreeWindow::new_object(SPtr<ObjectModel> object)
{
	SPtr<GraphModel> graph = dynamic_ptr_cast<GraphModel>(object);
	if (graph)
		add_graph(graph);
}

void
GraphTreeWindow::add_graph(SPtr<GraphModel> pm)
{
	if (!pm->parent()) {
		Gtk::TreeModel::iterator iter = _graph_treestore->append();
		Gtk::TreeModel::Row row = *iter;
		if (pm->path().is_root()) {
			row[_graph_tree_columns.name_col] = _app->interface()->uri();
		} else {
			row[_graph_tree_columns.name_col] = pm->symbol().c_str();
		}
		row[_graph_tree_columns.enabled_col] = pm->enabled();
		row[_graph_tree_columns.graph_model_col] = pm;
		_graphs_treeview->expand_row(_graph_treestore->get_path(iter), true);
	} else {
		Gtk::TreeModel::Children children = _graph_treestore->children();
		Gtk::TreeModel::iterator c = find_graph(children, pm->parent());

		if (c != children.end()) {
			Gtk::TreeModel::iterator iter = _graph_treestore->append(c->children());
			Gtk::TreeModel::Row row = *iter;
			row[_graph_tree_columns.name_col] = pm->symbol().c_str();
			row[_graph_tree_columns.enabled_col] = pm->enabled();
			row[_graph_tree_columns.graph_model_col] = pm;
			_graphs_treeview->expand_row(_graph_treestore->get_path(iter), true);
		}
	}

	pm->signal_property().connect(
		sigc::bind(sigc::mem_fun(this, &GraphTreeWindow::graph_property_changed),
		           pm));

	pm->signal_moved().connect(
		sigc::bind(sigc::mem_fun(this, &GraphTreeWindow::graph_moved),
		           pm));

	pm->signal_destroyed().connect(
		sigc::bind(sigc::mem_fun(this, &GraphTreeWindow::remove_graph),
		           pm));
}

void
GraphTreeWindow::remove_graph(SPtr<GraphModel> pm)
{
	Gtk::TreeModel::iterator i = find_graph(_graph_treestore->children(), pm);
	if (i != _graph_treestore->children().end())
		_graph_treestore->erase(i);
}

Gtk::TreeModel::iterator
GraphTreeWindow::find_graph(Gtk::TreeModel::Children  root,
                            SPtr<Client::ObjectModel> graph)
{
	for (Gtk::TreeModel::iterator c = root.begin(); c != root.end(); ++c) {
		SPtr<GraphModel> pm = (*c)[_graph_tree_columns.graph_model_col];
		if (graph == pm) {
			return c;
		} else if ((*c)->children().size() > 0) {
			Gtk::TreeModel::iterator ret = find_graph(c->children(), graph);
			if (ret != c->children().end())
				return ret;
		}
	}
	return root.end();
}

/** Show the context menu for the selected graph in the graphs treeview.
 */
void
GraphTreeWindow::show_graph_menu(GdkEventButton* ev)
{
	Gtk::TreeModel::iterator active = _graph_tree_selection->get_selected();
	if (active) {
		Gtk::TreeModel::Row row = *active;
		SPtr<GraphModel> pm = row[_graph_tree_columns.graph_model_col];
		if (pm) {
			_app->log().warn("TODO: graph menu from tree window");
		}
	}
}

void
GraphTreeWindow::event_graph_activated(const Gtk::TreeModel::Path& path,
                                       Gtk::TreeView::Column*      col)
{
	Gtk::TreeModel::iterator active = _graph_treestore->get_iter(path);
	Gtk::TreeModel::Row row = *active;
	SPtr<GraphModel> pm = row[_graph_tree_columns.graph_model_col];

	_app->window_factory()->present_graph(pm);
}

void
GraphTreeWindow::event_graph_enabled_toggled(const Glib::ustring& path_str)
{
	Gtk::TreeModel::Path path(path_str);
	Gtk::TreeModel::iterator active = _graph_treestore->get_iter(path);
	Gtk::TreeModel::Row row = *active;

	SPtr<GraphModel> pm = row[_graph_tree_columns.graph_model_col];
	assert(pm);

	if (_enable_signal)
		_app->set_property(pm->uri(),
		                   _app->uris().ingen_enabled,
		                   _app->forge().make((bool)!pm->enabled()));
}

void
GraphTreeWindow::graph_property_changed(const Raul::URI& key,
                                        const Atom&      value,
                                        SPtr<GraphModel> graph)
{
	const URIs& uris = _app->uris();
	_enable_signal = false;
	if (key == uris.ingen_enabled && value.type() == uris.forge.Bool) {
		Gtk::TreeModel::iterator i = find_graph(_graph_treestore->children(), graph);
		if (i != _graph_treestore->children().end()) {
			Gtk::TreeModel::Row row = *i;
			row[_graph_tree_columns.enabled_col] = value.get<int32_t>();
		} else {
			_app->log().error(fmt("Unable to find graph %1%\n")
			                  % graph->path());
		}
	}
	_enable_signal = true;
}

void
GraphTreeWindow::graph_moved(SPtr<GraphModel> graph)
{
	_enable_signal = false;

	Gtk::TreeModel::iterator i
		= find_graph(_graph_treestore->children(), graph);

	if (i != _graph_treestore->children().end()) {
		Gtk::TreeModel::Row row = *i;
		row[_graph_tree_columns.name_col] = graph->symbol().c_str();
	} else {
		_app->log().error(fmt("Unable to find graph %1%\n")
		                  % graph->path());
	}

	_enable_signal = true;
}

} // namespace GUI
} // namespace Ingen