/***

  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 "timelinemarker.h"

#include <QApplication>

#include "common/qtutils.h"
#include "common/xmlutils.h"
#include "config/config.h"
#include "node/project.h"
#include "ui/colorcoding.h"

namespace olive
{

TimelineMarker::TimelineMarker(QObject *parent)
	: color_(OAK_CONFIG("MarkerColor").toInt())
{
	setParent(parent);
}

TimelineMarker::TimelineMarker(int color, const TimeRange &time,
							   const QString &name, QObject *parent)
	: time_(time)
	, name_(name)
	, color_(color)
{
	setParent(parent);
}

void TimelineMarker::set_time(const TimeRange &time)
{
	time_ = time;
	emit time_changed(time_);
}

void TimelineMarker::set_time(const Rational &time)
{
	set_time(TimeRange(time, time + time_.length()));
}

bool TimelineMarker::has_sibling_at_time(const Rational &t) const
{
	TimelineMarker *m =
		static_cast<TimelineMarkerList *>(parent())->get_marker_at_time(t);
	return m && m != this;
}

void TimelineMarker::set_name(const QString &name)
{
	name_ = name;
	emit name_changed(name_);
}

void TimelineMarker::set_color(int c)
{
	color_ = c;
	emit color_changed(color_);
}

int TimelineMarker::get_marker_height(const QFontMetrics &fm)
{
	return fm.height();
}

QRect TimelineMarker::draw(QPainter *p, const QPoint &pt, int max_right,
						   double scale, bool selected)
{
	QFontMetrics fm = p->fontMetrics();

	int marker_height = get_marker_height(fm);
	int marker_width = QtUtils::q_font_metrics_width(fm, QStringLiteral("H"));

	int half_width = marker_width / 2;

	QColor c = QtUtils::to_q_color(ColorCoding::get_color(color()));
	if (selected) {
		p->setPen(Qt::white);
		p->setBrush(c.lighter());
	} else {
		p->setPen(Qt::black);
		p->setBrush(c);
	}

	int top = pt.y() - marker_height;

	QTextOption op(Qt::AlignLeft | Qt::AlignVCenter);
	op.setWrapMode(QTextOption::NoWrap);

	if (time_.out() != time_.in()) {
		QRect marker_rect(pt.x(), top, time_.length().to_double() * scale,
						  marker_height);

		p->drawRect(marker_rect);

		if (!name_.isEmpty()) {
			p->setPen(
				ColorCoding::get_ui_selector_color(ColorCoding::get_color(color_)));
			p->drawText(marker_rect.adjusted(marker_width / 4, 0, 0, 0), name_,
						op);
		}

		return marker_rect;
	} else {
		int half_marker_height = marker_height / 3;
		int left = pt.x() - half_width;
		int right = pt.x() + half_width;
		int center_y = pt.y() - half_marker_height;

		QPoint points[] = {
			pt,
			QPoint(left, center_y),
			QPoint(left, top),
			QPoint(right, top),
			QPoint(right, center_y),
			pt,
		};

		p->setRenderHint(QPainter::Antialiasing);
		p->drawPolygon(points, 6);

		if (!name_.isEmpty() && max_right != -1) {
			QRect text_rect(right, top, max_right - right, marker_height);

			int padding = QtUtils::q_font_metrics_width(p->fontMetrics(),
													 QStringLiteral(" "));
			text_rect.adjust(padding, 0, -padding - half_width, 0);

			p->setPen(qApp->palette().text().color());
			p->drawText(text_rect, name_, op);
		}

		return QRect(left, top, marker_width, marker_height);
	}
}

bool TimelineMarker::load(QXmlStreamReader *reader)
{
	Rational in, out;

	XMLAttributeLoop(reader, attr)
	{
		if (attr.name() == QStringLiteral("name")) {
			this->set_name(attr.value().toString());
		} else if (attr.name() == QStringLiteral("in")) {
			in = Rational::from_string(attr.value().toString().toStdString());
		} else if (attr.name() == QStringLiteral("out")) {
			out = Rational::from_string(attr.value().toString().toStdString());
		} else if (attr.name() == QStringLiteral("color")) {
			this->set_color(attr.value().toInt());
		}
	}

	this->set_time(TimeRange(in, out));

	// This element has no inner text, so just skip it
	reader->skipCurrentElement();

	return true;
}

void TimelineMarker::save(QXmlStreamWriter *writer) const
{
	writer->writeAttribute(QStringLiteral("name"), this->name());
	writer->writeAttribute(
		QStringLiteral("in"),
		QString::fromStdString(this->time().in().to_string()));
	writer->writeAttribute(
		QStringLiteral("out"),
		QString::fromStdString(this->time().out().to_string()));
	writer->writeAttribute(QStringLiteral("color"),
						   QString::number(this->color()));
}

bool TimelineMarkerList::load(QXmlStreamReader *reader)
{
	while (xml_read_next_start_element(reader)) {
		if (reader->name() == QStringLiteral("marker")) {
			TimelineMarker *marker = new TimelineMarker(this);
			if (!marker->load(reader)) {
				return false;
			}
		} else {
			reader->skipCurrentElement();
		}
	}

	return true;
}

void TimelineMarkerList::save(QXmlStreamWriter *writer) const
{
	for (auto it = this->cbegin(); it != this->cend(); it++) {
		TimelineMarker *marker = *it;

		writer->writeStartElement(QStringLiteral("marker"));

		marker->save(writer);

		writer->writeEndElement(); // marker
	}
}

void TimelineMarkerList::childEvent(QChildEvent *e)
{
	QObject::childEvent(e);

	if (TimelineMarker *marker = dynamic_cast<TimelineMarker *>(e->child())) {
		if (e->type() == QChildEvent::ChildAdded) {
			connect(marker, &TimelineMarker::time_changed, this,
					&TimelineMarkerList::handle_marker_time_change);
			connect(marker, &TimelineMarker::time_changed, this,
					&TimelineMarkerList::handle_marker_modification);
			connect(marker, &TimelineMarker::name_changed, this,
					&TimelineMarkerList::handle_marker_modification);
			connect(marker, &TimelineMarker::color_changed, this,
					&TimelineMarkerList::handle_marker_modification);

			insert_into_list(marker);

			emit marker_added(marker);

		} else if (e->type() == QChildEvent::ChildRemoved) {
			remove_from_list(marker);

			disconnect(marker, &TimelineMarker::time_changed, this,
					   &TimelineMarkerList::handle_marker_time_change);
			disconnect(marker, &TimelineMarker::time_changed, this,
					   &TimelineMarkerList::handle_marker_modification);
			disconnect(marker, &TimelineMarker::name_changed, this,
					   &TimelineMarkerList::handle_marker_modification);
			disconnect(marker, &TimelineMarker::color_changed, this,
					   &TimelineMarkerList::handle_marker_modification);

			emit marker_removed(marker);
		}
	}
}

void TimelineMarkerList::insert_into_list(TimelineMarker *marker)
{
	// Insertion sort by time to allow some loop optimizations
	bool found = false;
	for (auto it = markers_.begin(); it != markers_.end(); it++) {
		TimelineMarker *m = *it;

		Q_ASSERT(m->time().in() != marker->time().in());

		if (m->time().in() > marker->time().in()) {
			markers_.insert(it, marker);
			found = true;
			break;
		}
	}

	if (!found) {
		markers_.push_back(marker);
	}
}

bool TimelineMarkerList::remove_from_list(TimelineMarker *marker)
{
	auto it = std::find(markers_.begin(), markers_.end(), marker);

	if (it != markers_.end()) {
		markers_.erase(it);
		return true;
	}

	return false;
}

void TimelineMarkerList::handle_marker_modification()
{
	emit marker_modified(static_cast<TimelineMarker *>(sender()));
}

void TimelineMarkerList::handle_marker_time_change()
{
	TimelineMarker *m = static_cast<TimelineMarker *>(sender());

	auto it = std::find(markers_.begin(), markers_.end(), m);

	if (it != markers_.end()) {
		markers_.erase(it);
		insert_into_list(m);
	}
}

MarkerAddCommand::MarkerAddCommand(TimelineMarkerList *marker_list,
								   const TimeRange &range, const QString &name,
								   int color)
	: marker_list_(marker_list)
	, added_marker_(nullptr)
{
	added_marker_ = new TimelineMarker(color, range, name, &memory_manager_);
	added_marker_->setParent(&memory_manager_);
}

MarkerAddCommand::MarkerAddCommand(TimelineMarkerList *marker_list,
								   TimelineMarker *marker)
	: marker_list_(marker_list)
	, added_marker_(marker)
{
	added_marker_->setParent(&memory_manager_);
}

Project *MarkerAddCommand::get_relevant_project() const
{
	return Project::get_project_from_object(marker_list_);
}

void MarkerAddCommand::redo()
{
	added_marker_->setParent(marker_list_);
}

void MarkerAddCommand::undo()
{
	added_marker_->setParent(&memory_manager_);
}

MarkerRemoveCommand::MarkerRemoveCommand(TimelineMarker *marker)
	: marker_(marker)
{
}

Project *MarkerRemoveCommand::get_relevant_project() const
{
	return Project::get_project_from_object(marker_);
}

void MarkerRemoveCommand::redo()
{
	marker_list_ = marker_->parent();
	marker_->setParent(&memory_manager_);
}

void MarkerRemoveCommand::undo()
{
	marker_->setParent(marker_list_);
}

MarkerChangeColorCommand::MarkerChangeColorCommand(TimelineMarker *marker,
												   int new_color)
	: marker_(marker)
	, new_color_(new_color)
{
}

Project *MarkerChangeColorCommand::get_relevant_project() const
{
	return Project::get_project_from_object(marker_);
}

void MarkerChangeColorCommand::redo()
{
	old_color_ = marker_->color();
	marker_->set_color(new_color_);
}

void MarkerChangeColorCommand::undo()
{
	marker_->set_color(old_color_);
}

MarkerChangeNameCommand::MarkerChangeNameCommand(TimelineMarker *marker,
												 QString new_name)
	: marker_(marker)
	, new_name_(new_name)
{
}

Project *MarkerChangeNameCommand::get_relevant_project() const
{
	return Project::get_project_from_object(marker_);
}

void MarkerChangeNameCommand::redo()
{
	old_name_ = marker_->name();
	marker_->set_name(new_name_);
}

void MarkerChangeNameCommand::undo()
{
	marker_->set_name(old_name_);
}

MarkerChangeTimeCommand::MarkerChangeTimeCommand(TimelineMarker *marker,
												 const TimeRange &time,
												 const TimeRange &old_time)
	: marker_(marker)
	, old_time_(old_time)
	, new_time_(time)
{
}

Project *MarkerChangeTimeCommand::get_relevant_project() const
{
	return Project::get_project_from_object(marker_);
}

void MarkerChangeTimeCommand::redo()
{
	marker_->set_time(new_time_);
}

void MarkerChangeTimeCommand::undo()
{
	marker_->set_time(old_time_);
}

}