/***

    Olive - Non-Linear Video Editor
    Copyright (C) 2019  Olive Team

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

#include "panels/panels.h"
#include "panels/project.h"
#include "project/sequence.h"
#include "project/undo.h"
#include "project/clip.h"
#include "panels/timeline.h"
#include "playback/playback.h"
#include "project/media.h"
#include "playback/audio.h"

#include <QVariant>
#include <QVBoxLayout>
#include <QComboBox>
#include <QGroupBox>
#include <QGridLayout>
#include <QSpinBox>
#include <QLabel>
#include <QLineEdit>
#include <QDialogButtonBox>

extern "C" {
	#include <libavcodec/avcodec.h>
}

NewSequenceDialog::NewSequenceDialog(QWidget *parent, Media *existing) :
	QDialog(parent),
	existing_item(existing)
{
	setup_ui();

	if (existing != nullptr) {
		existing_sequence = existing->to_sequence();
		setWindowTitle(tr("Editing \"%1\"").arg(existing_sequence->name));

		width_numeric->setValue(existing_sequence->width);
		height_numeric->setValue(existing_sequence->height);
		int comp_rate = qRound(existing_sequence->frame_rate*100);
		for (int i=0;i<frame_rate_combobox->count();i++) {
			if (qRound(frame_rate_combobox->itemData(i).toDouble()*100) == comp_rate) {
				frame_rate_combobox->setCurrentIndex(i);
				break;
			}
		}
		sequence_name_edit->setText(existing_sequence->name);
		for (int i=0;i<audio_frequency_combobox->count();i++) {
			if (audio_frequency_combobox->itemData(i) == existing_sequence->audio_frequency) {
				audio_frequency_combobox->setCurrentIndex(i);
				break;
			}
		}
	} else {
		existing_sequence = nullptr;
		setWindowTitle(tr("New Sequence"));
	}
}

NewSequenceDialog::~NewSequenceDialog()
{}

void NewSequenceDialog::set_sequence_name(const QString& s) {
	sequence_name_edit->setText(s);
}

void NewSequenceDialog::create() {
	if (existing_sequence == nullptr) {
		Sequence* s = new Sequence();

		s->name = sequence_name_edit->text();
		s->width = width_numeric->value();
		s->height = height_numeric->value();
		s->frame_rate = frame_rate_combobox->currentData().toDouble();
		s->audio_frequency = audio_frequency_combobox->currentData().toInt();
		s->audio_layout = AV_CH_LAYOUT_STEREO;

		ComboAction* ca = new ComboAction();
		panel_project->create_sequence_internal(ca, s, true, nullptr);
		Olive::UndoStack.push(ca);
	} else {
		ComboAction* ca = new ComboAction();

		double multiplier = frame_rate_combobox->currentData().toDouble() / existing_sequence->frame_rate;

		EditSequenceCommand* esc = new EditSequenceCommand(existing_item, existing_sequence);
		esc->name = sequence_name_edit->text();
		esc->width = width_numeric->value();
		esc->height = height_numeric->value();
		esc->frame_rate = frame_rate_combobox->currentData().toDouble();
		esc->audio_frequency = audio_frequency_combobox->currentData().toInt();
		esc->audio_layout = AV_CH_LAYOUT_STEREO;
		ca->append(esc);

		for (int i=0;i<existing_sequence->clips.size();i++) {
			Clip* c = existing_sequence->clips.at(i);
			if (c != nullptr) {
				c->refactor_frame_rate(ca, multiplier, true);
			}
		}

		Olive::UndoStack.push(ca);
	}

	accept();
}

void NewSequenceDialog::preset_changed(int index) {
	switch (index) {
	case 0: // FILM 4K
		width_numeric->setValue(4096);
		height_numeric->setValue(2160);
		break;
	case 1: // TV 4K
		width_numeric->setValue(3840);
		height_numeric->setValue(2160);
		break;
	case 2: // 1080p
		width_numeric->setValue(1920);
		height_numeric->setValue(1080);
		break;
	case 3: // 720p
		width_numeric->setValue(1280);
		height_numeric->setValue(720);
		break;
	case 4: // 480p
		width_numeric->setValue(640);
		height_numeric->setValue(480);
		break;
	case 5: // 360p
		width_numeric->setValue(640);
		height_numeric->setValue(360);
		break;
	case 6: // 240p
		width_numeric->setValue(320);
		height_numeric->setValue(240);
		break;
	case 7: // 144p
		width_numeric->setValue(192);
		height_numeric->setValue(144);
		break;
	case 8: // NTSC (480i)
		width_numeric->setValue(720);
		height_numeric->setValue(480);
		break;
	case 9: // PAL (576i)
		width_numeric->setValue(720);
		height_numeric->setValue(576);
		break;
	}
}

void NewSequenceDialog::setup_ui() {
	QVBoxLayout* verticalLayout = new QVBoxLayout(this);

	QWidget* widget = new QWidget(this);

	QHBoxLayout* preset_layout = new QHBoxLayout(widget);
	preset_layout->setContentsMargins(0, 0, 0, 0);

	preset_layout->addWidget(new QLabel(tr("Preset:"), this));

	preset_combobox = new QComboBox(widget);

	preset_combobox->addItem(tr("Film 4K"));
	preset_combobox->addItem(tr("TV 4K (Ultra HD/2160p)"));
	preset_combobox->addItem(tr("1080p"));
	preset_combobox->addItem(tr("720p"));
	preset_combobox->addItem(tr("480p"));
	preset_combobox->addItem(tr("360p"));
	preset_combobox->addItem(tr("240p"));
	preset_combobox->addItem(tr("144p"));
	preset_combobox->addItem(tr("NTSC (480i)"));
	preset_combobox->addItem(tr("PAL (576i)"));
	preset_combobox->addItem(tr("Custom"));
	preset_combobox->setCurrentIndex(2);

	preset_layout->addWidget(preset_combobox);

	verticalLayout->addWidget(widget);

	QGroupBox* videoGroupBox = new QGroupBox(this);
	videoGroupBox->setTitle(tr("Video"));

	QGridLayout* videoLayout = new QGridLayout(videoGroupBox);

	videoLayout->addWidget(new QLabel(tr("Width:"), this), 0, 0, 1, 1);
	width_numeric = new QSpinBox(videoGroupBox);
	width_numeric->setMaximum(9999);
	width_numeric->setValue(1920);
	videoLayout->addWidget(width_numeric, 0, 2, 1, 2);

	videoLayout->addWidget(new QLabel(tr("Height:"), this), 1, 0, 1, 2);
	height_numeric = new QSpinBox(videoGroupBox);
	height_numeric->setMaximum(9999);
	height_numeric->setValue(1080);
	videoLayout->addWidget(height_numeric, 1, 2, 1, 2);

	videoLayout->addWidget(new QLabel(tr("Frame Rate:"), this), 2, 0, 1, 1);
	frame_rate_combobox = new QComboBox(videoGroupBox);
	frame_rate_combobox->addItem("10 FPS", 10.0);
	frame_rate_combobox->addItem("12.5 FPS", 12.5);
	frame_rate_combobox->addItem("15 FPS", 15.0);
	frame_rate_combobox->addItem("23.976 FPS", 23.976);
	frame_rate_combobox->addItem("24 FPS", 24.0);
	frame_rate_combobox->addItem("25 FPS", 25.0);
	frame_rate_combobox->addItem("29.97 FPS", 29.97);
	frame_rate_combobox->addItem("30 FPS", 30.0);
	frame_rate_combobox->addItem("50 FPS", 50.0);
	frame_rate_combobox->addItem("59.94 FPS", 59.94);
	frame_rate_combobox->addItem("60 FPS", 60.0);
	frame_rate_combobox->setCurrentIndex(6);
	videoLayout->addWidget(frame_rate_combobox, 2, 2, 1, 2);

	videoLayout->addWidget(new QLabel(tr("Pixel Aspect Ratio:"), this), 4, 0, 1, 1);
	par_combobox = new QComboBox(videoGroupBox);
	par_combobox->addItem(tr("Square Pixels (1.0)"));
	videoLayout->addWidget(par_combobox, 4, 2, 1, 2);

	videoLayout->addWidget(new QLabel(tr("Interlacing:"), this), 6, 0, 1, 1);
	interlacing_combobox = new QComboBox(videoGroupBox);
	interlacing_combobox->addItem(tr("None (Progressive)"));
//	interlacing_combobox->addItem("Upper Field First");
//	interlacing_combobox->addItem("Lower Field First");
	videoLayout->addWidget(interlacing_combobox, 6, 2, 1, 2);

	verticalLayout->addWidget(videoGroupBox);

	QGroupBox* audioGroupBox = new QGroupBox(this);
	audioGroupBox->setTitle(tr("Audio"));

	QGridLayout* audioLayout = new QGridLayout(audioGroupBox);

	audioLayout->addWidget(new QLabel(tr("Sample Rate: "), this), 0, 0, 1, 1);

	audio_frequency_combobox = new QComboBox(audioGroupBox);
	combobox_audio_sample_rates(audio_frequency_combobox);
	audio_frequency_combobox->setCurrentIndex(4);

	audioLayout->addWidget(audio_frequency_combobox, 0, 1, 1, 1);

	verticalLayout->addWidget(audioGroupBox);

	QWidget* nameWidget = new QWidget(this);
	QHBoxLayout* nameLayout = new QHBoxLayout(nameWidget);
	nameLayout->setContentsMargins(0, 0, 0, 0);

	nameLayout->addWidget(new QLabel(tr("Name:"), this));

	sequence_name_edit = new QLineEdit(nameWidget);

	nameLayout->addWidget(sequence_name_edit);

	verticalLayout->addWidget(nameWidget);

	QDialogButtonBox* buttonBox = new QDialogButtonBox(this);
	buttonBox->setOrientation(Qt::Horizontal);
	buttonBox->setStandardButtons(QDialogButtonBox::Cancel|QDialogButtonBox::Ok);
	buttonBox->setCenterButtons(true);

	verticalLayout->addWidget(buttonBox);

	connect(preset_combobox, SIGNAL(currentIndexChanged(int)), this, SLOT(preset_changed(int)));
	connect(buttonBox, SIGNAL(accepted()), this, SLOT(create()));
	connect(buttonBox, SIGNAL(rejected()), this, SLOT(reject()));
}