#include "speeddialog.h"

#include <QHBoxLayout>
#include <QGridLayout>
#include <QLabel>
#include <QCheckBox>
#include <QPushButton>
#include <QDialogButtonBox>
#include <QDebug>

#include "ui/labelslider.h"
#include "project/clip.h"
#include "project/sequence.h"
#include "io/media.h"
#include "playback/playback.h"

SpeedDialog::SpeedDialog(QWidget *parent) : QDialog(parent) {
	QVBoxLayout* main_layout = new QVBoxLayout();
	setLayout(main_layout);

	QGridLayout* grid = new QGridLayout();
	grid->setSpacing(6);

	grid->addWidget(new QLabel("Speed:"), 0, 0);
	percent = new LabelSlider();
	percent->decimal_places = 2;
	percent->set_display_type(LABELSLIDER_PERCENT);
	percent->set_default_value(1);
	grid->addWidget(percent, 0, 1);

	grid->addWidget(new QLabel("Frame Rate:"), 1, 0);
	frame_rate = new LabelSlider();
	frame_rate->decimal_places = 3;
	grid->addWidget(frame_rate, 1, 1);

	grid->addWidget(new QLabel("Duration:"), 2, 0);
	duration = new LabelSlider();
	duration->set_display_type(LABELSLIDER_FRAMENUMBER);
	grid->addWidget(duration, 2, 1);

	main_layout->addLayout(grid);

	reverse = new QCheckBox("Reverse");
	main_layout->addWidget(reverse);
	main_layout->addWidget(new QCheckBox("Maintain Audio Pitch"));
	main_layout->addWidget(new QCheckBox("Ripple"));

	QDialogButtonBox* buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel);
	buttonBox->setCenterButtons(true);
	main_layout->addWidget(buttonBox);
	connect(buttonBox, SIGNAL(rejected()), this, SLOT(reject()));
	connect(buttonBox, SIGNAL(accepted()), this, SLOT(accept()));

	connect(percent, SIGNAL(valueChanged()), this, SLOT(percent_update()));
	connect(frame_rate, SIGNAL(valueChanged()), this, SLOT(frame_rate_update()));
	connect(duration, SIGNAL(valueChanged()), this, SLOT(duration_update()));
}

void SpeedDialog::run() {
	bool enable_frame_rate = false;

	default_frame_rate = NAN;
	current_frame_rate = NAN;
	current_percent = NAN;
	default_length = -1;
	current_length = -1;

	for (int i=0;i<clips.size();i++) {
		Clip* c = clips.at(i);

		double clip_percent;

		// get default frame rate/percentage
		if (c->track < 0) {
			double media_frame_rate = c->getMediaFrameRate();

			clip_percent = c->frame_rate / media_frame_rate;

			// get "default" frame rate"
			if (enable_frame_rate) {
				// check if frame rate is equal to default
				if (!qIsNaN(default_frame_rate) && media_frame_rate != default_frame_rate) {
					default_frame_rate = NAN;
				}
			} else {
				default_frame_rate = media_frame_rate;
			}

			enable_frame_rate = true;
		} else {
			clip_percent = c->frame_rate;

			if (i > 0) {
				if (!qIsNaN(default_percent) && c->frame_rate != default_percent) {
					default_percent = NAN;
				}
			} else {
				default_percent = c->frame_rate;
			}
		}

		// get default length
		if (i == 0) {
			current_length = c->getLength();
			default_length = qRound(c->getLength() * clip_percent);
		} else if (current_length != -1 && c->getLength() != current_length) {
			current_length = -1;
		}
	}

	frame_rate->setEnabled(enable_frame_rate);
	frame_rate->set_default_value(default_frame_rate);
	frame_rate->set_value(current_frame_rate, false);
	percent->set_value(current_percent, false);
	duration->set_default_value(default_length);

	exec();
}

void SpeedDialog::percent_update() {
	frame_rate->set_value(default_frame_rate * percent->value(), false);
	duration->set_value(clips.at(0)->getLength() / percent->value(), false);
}

void SpeedDialog::duration_update() {
	double pc = clips.at(0)->getLength() / duration->value();
	frame_rate->set_value(default_frame_rate * pc, false);
	percent->set_value(pc, false);
}

void SpeedDialog::frame_rate_update() {
	double fr = (frame_rate->value());
	double pc = (fr / default_frame_rate);
	percent->set_value(pc, false);
	duration->set_value(clips.at(0)->getLength() / pc, false);
}

void SpeedDialog::accept() {
	for (int i=0;i<clips.size();i++) {
		Clip* c = clips.at(i);
		if (c->track < 0) {
			c->frame_rate = frame_rate->value();
		} else {
			c->frame_rate = percent->value();
		}
		close_clip(c);
	}
	QDialog::accept();
}