#include "grapheditor.h"

#include <QVBoxLayout>
#include <QLabel>
#include <QPushButton>
#include <QVariant>

#include "ui/keyframenavigator.h"
#include "ui/timelineheader.h"
#include "ui/timelinetools.h"
#include "ui/labelslider.h"
#include "ui/graphview.h"
#include "project/effect.h"
#include "project/effectfield.h"
#include "project/effectrow.h"
#include "project/clip.h"
#include "panels.h"
#include "debug.h"

GraphEditor::GraphEditor(QWidget* parent) : QDockWidget(parent), row(nullptr) {
	setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

    setWindowTitle(tr("Graph Editor"));
	resize(720, 480);

	QWidget* main_widget = new QWidget();
	setWidget(main_widget);
	QVBoxLayout* layout = new QVBoxLayout();
	main_widget->setLayout(layout);

	QWidget* tool_widget = new QWidget();
	tool_widget->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Maximum);
	QHBoxLayout* tools = new QHBoxLayout();
	tool_widget->setLayout(tools);

	QWidget* left_tool_widget = new QWidget();
	QHBoxLayout* left_tool_layout = new QHBoxLayout();
	left_tool_layout->setSpacing(0);
	left_tool_layout->setMargin(0);
	left_tool_widget->setLayout(left_tool_layout);
	tools->addWidget(left_tool_widget);
	QWidget* center_tool_widget = new QWidget();
	QHBoxLayout* center_tool_layout = new QHBoxLayout();
	center_tool_layout->setSpacing(0);
	center_tool_layout->setMargin(0);
	center_tool_widget->setLayout(center_tool_layout);
	tools->addWidget(center_tool_widget);
	QWidget* right_tool_widget = new QWidget();
	QHBoxLayout* right_tool_layout = new QHBoxLayout();
	right_tool_layout->setSpacing(0);
	right_tool_layout->setMargin(0);
	right_tool_widget->setLayout(right_tool_layout);
	tools->addWidget(right_tool_widget);

	keyframe_nav = new KeyframeNavigator();
	keyframe_nav->enable_keyframes(true);
	keyframe_nav->enable_keyframe_toggle(false);
	left_tool_layout->addWidget(keyframe_nav);
	left_tool_layout->addStretch();

    linear_button = new QPushButton(tr("Linear"));
	linear_button->setProperty("type", KEYFRAME_TYPE_LINEAR);
	linear_button->setCheckable(true);
    bezier_button = new QPushButton(tr("Bezier"));
	bezier_button->setProperty("type", KEYFRAME_TYPE_BEZIER);
	bezier_button->setCheckable(true);
    hold_button = new QPushButton(tr("Hold"));
	hold_button->setProperty("type", KEYFRAME_TYPE_HOLD);
	hold_button->setCheckable(true);

	center_tool_layout->addStretch();
	center_tool_layout->addWidget(linear_button);
	center_tool_layout->addWidget(bezier_button);
	center_tool_layout->addWidget(hold_button);

	layout->addWidget(tool_widget);

	QWidget* central_widget = new QWidget();
	QVBoxLayout* central_layout = new QVBoxLayout();
	central_widget->setLayout(central_layout);
	central_layout->setSpacing(0);
	central_layout->setMargin(0);
	header = new TimelineHeader();
	header->viewer = panel_sequence_viewer;
	central_layout->addWidget(header);
	view = new GraphView();
	central_layout->addWidget(view);

	layout->addWidget(central_widget);

	QWidget* value_widget = new QWidget();
	value_widget->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Maximum);
	QHBoxLayout* values = new QHBoxLayout();
	value_widget->setLayout(values);
	values->addStretch();

	QWidget* central_value_widget = new QWidget();
	value_layout = new QHBoxLayout();
	value_layout->setMargin(0);
	value_layout->addWidget(new QLabel("")); // a spacer so the layout doesn't jump
	central_value_widget->setLayout(value_layout);
	values->addWidget(central_value_widget);

	values->addStretch();
	layout->addWidget(value_widget);

	current_row_desc = new QLabel();
	current_row_desc->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Maximum);
	current_row_desc->setAlignment(Qt::AlignCenter);
	layout->addWidget(current_row_desc);

	connect(view, SIGNAL(zoom_changed(double)), header, SLOT(update_zoom(double)));
	connect(view, SIGNAL(x_scroll_changed(int)), header, SLOT(set_scroll(int)));
	connect(view, SIGNAL(selection_changed(bool, int)), this, SLOT(set_key_button_enabled(bool, int)));

	connect(linear_button, SIGNAL(clicked(bool)), this, SLOT(set_keyframe_type()));
	connect(bezier_button, SIGNAL(clicked(bool)), this, SLOT(set_keyframe_type()));
	connect(hold_button, SIGNAL(clicked(bool)), this, SLOT(set_keyframe_type()));
}

void GraphEditor::update_panel() {
	if (isVisible()) {
		if (row != nullptr) {
			int slider_index = 0;
			for (int i=0;i<row->fieldCount();i++) {
				EffectField* field = row->field(i);
				if (field->type == EFFECT_FIELD_DOUBLE) {
					slider_proxies.at(slider_index)->set_value(row->field(i)->get_current_data().toDouble(), false);
					slider_index++;
				}
			}
		}

		header->update();
		view->update();
	}
}

void GraphEditor::set_row(EffectRow *r) {
	for (int i=0;i<slider_proxies.size();i++) {
		delete slider_proxies.at(i);
		delete slider_proxy_buttons.at(i);
	}
	slider_proxies.clear();
	slider_proxy_buttons.clear();
	slider_proxy_sources.clear();

	if (row != nullptr) {
		// clear old row connections
		disconnect(keyframe_nav, SIGNAL(goto_previous_key()), row, SLOT(goto_previous_key()));
		disconnect(keyframe_nav, SIGNAL(toggle_key()), row, SLOT(toggle_key()));
		disconnect(keyframe_nav, SIGNAL(goto_next_key()), row, SLOT(goto_next_key()));
	}

	bool found_vals = false;

	if (r != nullptr && r->isKeyframing()) {
		for (int i=0;i<r->fieldCount();i++) {
			EffectField* field = r->field(i);
			if (field->type == EFFECT_FIELD_DOUBLE) {
				QPushButton* slider_button = new QPushButton();
				slider_button->setCheckable(true);
				slider_button->setChecked(true);
				slider_button->setIcon(QIcon(":/icons/record.png"));
				slider_button->setProperty("field", i);
				slider_button->setIconSize(QSize(8, 8));
				slider_button->setMaximumSize(QSize(12, 12));
				connect(slider_button, SIGNAL(toggled(bool)), this, SLOT(set_field_visibility(bool)));
				slider_proxy_buttons.append(slider_button);
				value_layout->addWidget(slider_button);

				LabelSlider* slider = new LabelSlider();
				slider->set_color(get_curve_color(i, r->fieldCount()).name());
				connect(slider, SIGNAL(valueChanged()), this, SLOT(passthrough_slider_value()));
				slider_proxies.append(slider);
				value_layout->addWidget(slider);

				slider_proxy_sources.append(static_cast<LabelSlider*>(field->ui_element));

				found_vals = true;
			}
		}
	}

	if (found_vals) {
		row = r;
		current_row_desc->setText(row->parent_effect->parent_clip->name + " :: " + row->parent_effect->meta->name + " :: " + row->get_name());
		header->set_visible_in(r->parent_effect->parent_clip->timeline_in);

		connect(keyframe_nav, SIGNAL(goto_previous_key()), row, SLOT(goto_previous_key()));
		connect(keyframe_nav, SIGNAL(toggle_key()), row, SLOT(toggle_key()));
		connect(keyframe_nav, SIGNAL(goto_next_key()), row, SLOT(goto_next_key()));
	} else {
		row = nullptr;
		current_row_desc->setText(0);
	}
	view->set_row(row);
	update_panel();
}

bool GraphEditor::view_is_focused() {
    return view->hasFocus() || header->hasFocus();
}

bool GraphEditor::view_is_under_mouse() {
    return view->underMouse() || header->underMouse();
}

void GraphEditor::delete_selected_keys() {
	view->delete_selected_keys();
}

void GraphEditor::select_all() {
	view->select_all();
}

void GraphEditor::set_key_button_enabled(bool e, int type) {
	linear_button->setEnabled(e);
	linear_button->setChecked(type == KEYFRAME_TYPE_LINEAR);
	bezier_button->setEnabled(e);
	bezier_button->setChecked(type == KEYFRAME_TYPE_BEZIER);
	hold_button->setEnabled(e);
	hold_button->setChecked(type == KEYFRAME_TYPE_HOLD);
}

void GraphEditor::passthrough_slider_value() {
	for (int i=0;i<slider_proxies.size();i++) {
		if (slider_proxies.at(i) == sender()) {
			slider_proxy_sources.at(i)->set_value(slider_proxies.at(i)->value(), true);
		}
	}
}

void GraphEditor::set_keyframe_type() {
	linear_button->setChecked(linear_button == sender());
	bezier_button->setChecked(bezier_button == sender());
	hold_button->setChecked(hold_button == sender());
	view->set_selected_keyframe_type(sender()->property("type").toInt());
}

void GraphEditor::set_field_visibility(bool b) {
	view->set_field_visibility(sender()->property("field").toInt(), b);
}