/***

  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 <QVariant>
#include <QVBoxLayout>
#include <QComboBox>
#include <QGroupBox>
#include <QGridLayout>
#include <QSpinBox>
#include <QLabel>
#include <QLineEdit>
#include <QDialogButtonBox>

#include "panels/panels.h"
#include "panels/project.h"
#include "timeline/sequence.h"
#include "undo/undostack.h"
#include "undo/undo.h"
#include "timeline/clip.h"
#include "panels/timeline.h"
#include "project/media.h"
#include "rendering/audio.h"
#include "global/config.h"

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

NewSequenceDialog::NewSequenceDialog(QWidget *parent, Media *existing, Sequence* iexisting_sequence) :
  QDialog(parent),
  existing_item(existing),
  existing_sequence(iexisting_sequence)
{
  Q_ASSERT(!(existing != nullptr && iexisting_sequence != nullptr));

  setup_ui();

  if (existing != nullptr) {
    existing_sequence = existing->to_sequence().get();
  }

  if (existing_sequence != nullptr) {
    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"));
  }
}

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

void NewSequenceDialog::SetNameEditable(bool enabled)
{
  sequence_name_edit->setVisible(enabled);
  sequence_name_label->setVisible(enabled);
}

void NewSequenceDialog::accept() {
  if (existing_sequence == nullptr) {

    // The dialog wasn't given an existing Sequence object, so we'll make a new one

    SequencePtr s = std::make_shared<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();
    olive::project_model.CreateSequence(ca, s, true, nullptr);
    olive::undo_stack.push(ca);

  } else if (existing_item != nullptr) {

    // The dialog was given an existing Sequence object, so we'll apply the changes to it

    ComboAction* ca = new ComboAction();

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

    EditSequenceCommand* esc = new EditSequenceCommand(existing_item, existing_item->to_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);

    QVector<Clip*> existing_sequence_clips = existing_sequence->GetAllClips();
    for (int i=0;i<existing_sequence_clips.size();i++) {
      existing_sequence_clips.at(i)->refactor_frame_rate(ca, multiplier, true);
    }

    olive::undo_stack.push(ca);

  } else if (existing_sequence != nullptr) {

    // This dialog was given an existing Sequence without a Media wrapper - therefore just directly apply the settings

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

  }

  QDialog::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(olive::config.default_sequence_width);
  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(olive::config.default_sequence_height);
  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);
  for (int i=0;i<frame_rate_combobox->count();i++) {
    if (qFuzzyCompare(frame_rate_combobox->itemData(i).toDouble(), olive::config.default_sequence_framerate)) {
      frame_rate_combobox->setCurrentIndex(i);
    }
  }
  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)"));
  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);
  for (int i=0;i<audio_frequency_combobox->count();i++) {
    if (audio_frequency_combobox->itemData(i) == olive::config.default_sequence_audio_frequency) {
      audio_frequency_combobox->setCurrentIndex(i);
    }
  }

  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);

  sequence_name_label = new QLabel(tr("Name:"));
  nameLayout->addWidget(sequence_name_label);

  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(accept()));
  connect(buttonBox, SIGNAL(rejected()), this, SLOT(reject()));
}