/***

  Olive - Non-Linear Video Editor
  Copyright (C) 2022 Olive Team
  Modifications Copyright (C) 2025 mikesolar

  This program 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 3 of the License, or
  (at your option) any later version.

  This program 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 more details.

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

***/

#include "nodeparamviewkeyframecontrol.h"

#include <QHBoxLayout>
#include <QMessageBox>

#include "core.h"
#include "node/nodeundo.h"
#include "ui/icons/icons.h"

namespace olive
{

NodeParamViewKeyframeControl::NodeParamViewKeyframeControl(bool right_align,
														   QWidget *parent)
	: QWidget(parent)
{
	QHBoxLayout *layout = new QHBoxLayout(this);
	layout->setContentsMargins(0, 0, 0, 0);
	layout->setSpacing(0);

	if (right_align) {
		// Automatically right aligns all buttons
		layout->addStretch();
	}

	prev_key_btn_ = CreateNewToolButton(icon::TriLeft);
	prev_key_btn_->setIconSize(prev_key_btn_->iconSize() / 2);
	layout->addWidget(prev_key_btn_);

	toggle_key_btn_ = CreateNewToolButton(icon::Diamond);
	toggle_key_btn_->setCheckable(true);
	toggle_key_btn_->setIconSize(toggle_key_btn_->iconSize() / 2);
	layout->addWidget(toggle_key_btn_);

	next_key_btn_ = CreateNewToolButton(icon::TriRight);
	next_key_btn_->setIconSize(next_key_btn_->iconSize() / 2);
	layout->addWidget(next_key_btn_);

	enable_key_btn_ = CreateNewToolButton(icon::Clock);
	enable_key_btn_->setCheckable(true);
	enable_key_btn_->setIconSize(enable_key_btn_->iconSize() / 4 * 3);
	layout->addWidget(enable_key_btn_);

	connect(prev_key_btn_, &QPushButton::clicked, this,
			&NodeParamViewKeyframeControl::GoToPreviousKey);
	connect(next_key_btn_, &QPushButton::clicked, this,
			&NodeParamViewKeyframeControl::GoToNextKey);
	connect(toggle_key_btn_, &QPushButton::clicked, this,
			&NodeParamViewKeyframeControl::ToggleKeyframe);
	connect(enable_key_btn_, &QPushButton::toggled, this,
			&NodeParamViewKeyframeControl::ShowButtonsFromKeyframeEnable);
	connect(enable_key_btn_, &QPushButton::clicked, this,
			&NodeParamViewKeyframeControl::KeyframeEnableBtnClicked);

	// Set defaults
	SetInput(NodeInput());
	ShowButtonsFromKeyframeEnable(false);
}

void NodeParamViewKeyframeControl::SetInput(const NodeInput &input)
{
	if (input_.IsValid()) {
		disconnect(input_.node(), &Node::KeyframeEnableChanged, this,
				   &NodeParamViewKeyframeControl::KeyframeEnableChanged);
		disconnect(input_.node(), &Node::KeyframeAdded, this,
				   &NodeParamViewKeyframeControl::UpdateState);
		disconnect(input_.node(), &Node::KeyframeRemoved, this,
				   &NodeParamViewKeyframeControl::UpdateState);
		disconnect(input_.node(), &Node::KeyframeTimeChanged, this,
				   &NodeParamViewKeyframeControl::UpdateState);
	}

	input_ = input;
	SetButtonsEnabled(input_.IsValid());

	// Pick up keyframing value
	enable_key_btn_->setChecked(input_.IsValid() && input_.IsKeyframing());

	// Update buttons
	UpdateState();

	if (input_.IsValid()) {
		connect(input_.node(), &Node::KeyframeEnableChanged, this,
				&NodeParamViewKeyframeControl::KeyframeEnableChanged);
		connect(input_.node(), &Node::KeyframeAdded, this,
				&NodeParamViewKeyframeControl::UpdateState);
		connect(input_.node(), &Node::KeyframeRemoved, this,
				&NodeParamViewKeyframeControl::UpdateState);
		connect(input_.node(), &Node::KeyframeTimeChanged, this,
				&NodeParamViewKeyframeControl::UpdateState);
	}
}

void NodeParamViewKeyframeControl::TimeTargetDisconnectEvent(ViewerOutput *v)
{
	disconnect(v, &ViewerOutput::PlayheadChanged, this,
			   &NodeParamViewKeyframeControl::UpdateState);
}

void NodeParamViewKeyframeControl::TimeTargetConnectEvent(ViewerOutput *v)
{
	connect(v, &ViewerOutput::PlayheadChanged, this,
			&NodeParamViewKeyframeControl::UpdateState);
	UpdateState();
}

QPushButton *
NodeParamViewKeyframeControl::CreateNewToolButton(const QIcon &icon) const
{
	QPushButton *btn = new QPushButton();
	btn->setSizePolicy(QSizePolicy::Maximum, QSizePolicy::Maximum);
	btn->setIcon(icon);

	return btn;
}

void NodeParamViewKeyframeControl::SetButtonsEnabled(bool e)
{
	prev_key_btn_->setEnabled(e);
	toggle_key_btn_->setEnabled(e);
	next_key_btn_->setEnabled(e);
	enable_key_btn_->setEnabled(e);
}

rational NodeParamViewKeyframeControl::GetCurrentTimeAsNodeTime() const
{
	return GetAdjustedTime(GetTimeTarget(), input_.node(),
						   GetTimeTarget()->GetPlayhead(),
						   Node::kTransformTowardsInput);
}

rational
NodeParamViewKeyframeControl::ConvertToViewerTime(const rational &r) const
{
	return GetAdjustedTime(input_.node(), GetTimeTarget(), r,
						   Node::kTransformTowardsOutput);
}

void NodeParamViewKeyframeControl::ShowButtonsFromKeyframeEnable(bool e)
{
	prev_key_btn_->setVisible(e);
	toggle_key_btn_->setVisible(e);
	next_key_btn_->setVisible(e);
}

void NodeParamViewKeyframeControl::ToggleKeyframe(bool e)
{
	rational node_time = GetCurrentTimeAsNodeTime();

	QVector<NodeKeyframe *> keys =
		input_.node()->GetKeyframesAtTime(input_, node_time);

	MultiUndoCommand *command = new MultiUndoCommand();

	int nb_tracks = input_.node()->GetNumberOfKeyframeTracks(input_);

	if (e && keys.isEmpty()) {
		// Add a keyframe here (one for each track)
		for (int i = 0; i < nb_tracks; i++) {
			NodeKeyframe *key = new NodeKeyframe(
				node_time,
				input_.node()->GetSplitValueAtTimeOnTrack(input_, node_time, i),
				input_.node()->GetBestKeyframeTypeForTimeOnTrack(input_,
																 node_time, i),
				i, input_.element(), input_.input());

			command->add_child(
				new NodeParamInsertKeyframeCommand(input_.node(), key));
		}
	} else if (!e && !keys.isEmpty()) {
		// Remove all keyframes at this time
		foreach (NodeKeyframe *key, keys) {
			command->add_child(new NodeParamRemoveKeyframeCommand(key));

			if (input_.node()->GetKeyframeTracks(input_).size() == 1) {
				// If this was the last keyframe on this track, set the standard value to the value at this time too
				command->add_child(new NodeParamSetStandardValueCommand(
					NodeKeyframeTrackReference(input_, key->track()),
					input_.node()->GetSplitValueAtTimeOnTrack(input_, node_time,
															  key->track())));
			}
		}
	}

	Core::instance()->undo_stack()->push(command, tr("Toggled Keyframe"));
}

void NodeParamViewKeyframeControl::UpdateState()
{
	if (!input_.IsValid() || !input_.IsKeyframing() || !GetTimeTarget()) {
		return;
	}

	NodeKeyframe *earliest_key = input_.node()->GetEarliestKeyframe(input_);
	NodeKeyframe *latest_key = input_.node()->GetLatestKeyframe(input_);

	rational node_time = GetCurrentTimeAsNodeTime();

	prev_key_btn_->setEnabled(earliest_key && node_time > earliest_key->time());
	next_key_btn_->setEnabled(latest_key && node_time < latest_key->time());
	toggle_key_btn_->setChecked(
		input_.node()->HasKeyframeAtTime(input_, node_time));
}

void NodeParamViewKeyframeControl::GoToPreviousKey()
{
	rational node_time = GetCurrentTimeAsNodeTime();

	NodeKeyframe *previous_key =
		input_.node()->GetClosestKeyframeBeforeTime(input_, node_time);

	if (previous_key && GetTimeTarget()) {
		rational key_time = ConvertToViewerTime(previous_key->time());
		GetTimeTarget()->SetPlayhead(key_time);
	}
}

void NodeParamViewKeyframeControl::GoToNextKey()
{
	rational node_time = GetCurrentTimeAsNodeTime();

	NodeKeyframe *next_key =
		input_.node()->GetClosestKeyframeAfterTime(input_, node_time);

	if (next_key && GetTimeTarget()) {
		rational key_time = ConvertToViewerTime(next_key->time());
		GetTimeTarget()->SetPlayhead(key_time);
	}
}

void NodeParamViewKeyframeControl::KeyframeEnableBtnClicked(bool e)
{
	if (e == input_.IsKeyframing()) {
		// No-op
		return;
	}

	MultiUndoCommand *command = new MultiUndoCommand();

	QString command_name;

	if (e) {
		// Enable keyframing
		command->add_child(new NodeParamSetKeyframingCommand(input_, true));

		// Create one keyframe across all tracks here
		const QVector<QVariant> &key_vals =
			input_.node()->GetSplitStandardValue(input_);

		for (int i = 0; i < key_vals.size(); i++) {
			NodeKeyframe *key =
				new NodeKeyframe(GetCurrentTimeAsNodeTime(), key_vals.at(i),
								 NodeKeyframe::kDefaultType, i,
								 input_.element(), input_.input());

			command->add_child(
				new NodeParamInsertKeyframeCommand(input_.node(), key));
		}

		command_name =
			tr("Enabled Keyframing On %1 - %2")
				.arg(input_.node()->GetLabelAndName(), input_.GetInputName());
	} else {
		// Confirm the user wants to clear all keyframes
		if (QMessageBox::warning(
				this, tr("Warning"),
				tr("Are you sure you want to disable keyframing on this value? This will clear all existing keyframes."),
				QMessageBox::Yes | QMessageBox::No) == QMessageBox::Yes) {
			// Store value at this time, we'll set this as the persistent value later
			const QVector<QVariant> &stored_vals =
				input_.node()->GetSplitValueAtTime(input_,
												   GetCurrentTimeAsNodeTime());

			// Delete all keyframes
			foreach (const NodeKeyframeTrack &track,
					 input_.node()->GetKeyframeTracks(input_)) {
				for (int i = track.size() - 1; i >= 0; i--) {
					command->add_child(
						new NodeParamRemoveKeyframeCommand(track.at(i)));
				}
			}

			// Update standard value
			for (int i = 0; i < stored_vals.size(); i++) {
				command->add_child(new NodeParamSetStandardValueCommand(
					NodeKeyframeTrackReference(input_, i), stored_vals.at(i)));
			}

			// Disable keyframing
			command->add_child(
				new NodeParamSetKeyframingCommand(input_, false));

			command_name = tr("Disabled Keyframing On %1 - %2")
							   .arg(input_.node()->GetLabelAndName(),
									input_.GetInputName());
		} else {
			// Disable action has effectively been ignored
			enable_key_btn_->setChecked(true);
		}
	}

	Core::instance()->undo_stack()->push(command, command_name);
}

void NodeParamViewKeyframeControl::KeyframeEnableChanged(const NodeInput &input,
														 bool e)
{
	if (input_ == input) {
		enable_key_btn_->setChecked(e);
	}
}

}