// Include own headers
#include "MainWindow.hpp"
#include "../gui/ui_MainWindow.h"

// Include modules
#include "TrackPointRenderer.hpp"
#include "PickHandler.hpp"

// Include dependencies
#include <QFileDialog>
#include <QMessageBox>

MainWindow* globalPointer;

MainWindow* MainWindow::getInstance() {
  return globalPointer;
}

MainWindow::MainWindow(QWidget* parent): QMainWindow(parent), ui(new Ui::MainWindow) {
  globalPointer = this;

  ui->setupUi(this);

  osgWidget = new OSGWidget(this);
  ui->sceneWidget->layout()->addWidget(osgWidget);

  noMeshWidget = new NoMeshWidget();
  editWidget = new EditWidget();

  // TODO: Add option for opening a project via double click
  projectStore = new ProjectStore();
  renderView(NoMesh);

  // Setup menu
  newAct = new QAction(tr("&New project"), this);
  newAct->setShortcuts(QKeySequence::New);
  newAct->setStatusTip(tr("Create a new file"));
  connect(newAct, &QAction::triggered, this, &MainWindow::newFile);

  loadAct = new QAction(tr("&Load project"), this);
  loadAct->setShortcuts(QKeySequence::Open);
  loadAct->setStatusTip(tr("Load an existing project"));
  connect(loadAct, &QAction::triggered, this, &MainWindow::load);

  saveAct = new QAction(tr("&Save project"), this);
  saveAct->setShortcuts(QKeySequence::Save);
  saveAct->setStatusTip(tr("Update the project file"));
  connect(saveAct, &QAction::triggered, this, &MainWindow::save);

  saveAsAct = new QAction(tr("&Save project as"), this);
  saveAsAct->setStatusTip(tr("Save the project in a new file"));
  connect(saveAsAct, &QAction::triggered, this, &MainWindow::saveAs);

  closeAct = new QAction(tr("&Close project"), this);
  closeAct->setShortcuts(QKeySequence::Close);
  closeAct->setStatusTip(tr("Closing the current project"));
  connect(closeAct, &QAction::triggered, this, &MainWindow::close);

  fileMenu = menuBar()->addMenu(tr("&File"));
  fileMenu->addAction(newAct);
  fileMenu->addSeparator();
  fileMenu->addAction(loadAct);
  fileMenu->addSeparator();
  fileMenu->addAction(saveAct);
  fileMenu->addAction(saveAsAct);
  fileMenu->addSeparator();
  fileMenu->addAction(closeAct);
}

MainWindow::~MainWindow() {
  delete ui;
  delete osgWidget;
}

void MainWindow::renderView(GuiView view) {
  switch(view) {
    case NoMesh: {
      editWidget->setParent(nullptr);
      ui->controlWidget->layout()->addWidget(noMeshWidget);
      break;
    }
    case Edit: {
      noMeshWidget->setParent(nullptr);
      ui->controlWidget->layout()->addWidget(editWidget);
      break;
    }
  }
}

OSGWidget* MainWindow::getOsgWidget() {
  return osgWidget;
}

ProjectStore* MainWindow::getStore() {
  return projectStore;
}

EditWidget* MainWindow::getEditWiget() {
  return editWidget;
}

void MainWindow::showErrorMessage(std::string message, std::string title) {
  QMessageBox msg(this);
  msg.setText(QString::fromUtf8(message.c_str()));
  msg.setWindowTitle(QString::fromUtf8(title.c_str()));
  msg.exec();
}

void MainWindow::closeEvent(QCloseEvent *event) {
  if (projectStore->isModified()) {
    if (!saveChangesPopup()) {
      event->ignore();
      return;
    }
  }
  if (projectStore->isRendering()) {
    if (!isRenderingPopup()) {
      event->ignore();
      return;
    }
  }
  projectStore->closeProject();
  event->accept();
}

void MainWindow::newFile() {
  if (projectStore->isProjectOpen()) {
    if (projectStore->isModified()) {
      if (!saveChangesPopup()) return;
    }
    if (projectStore->isRendering()) {
      if (!isRenderingPopup()) return;
    }
    projectStore->closeProject();
  }
  renderView(NoMesh);
  cleanup();
  noMeshWidget->loadMeshFile();
}

void MainWindow::load() {
  QString fileName = QFileDialog::getOpenFileName(this, tr("Open a TrackpointApp Project"), "", tr("TrackpointApp Projects (*.trackproj)"));
  std::string projectFile = fileName.toUtf8().constData();
  projectStore->loadProject(projectFile);
}

bool MainWindow::save() {
  if (!projectStore->saveProject()) {
    return saveAs();
  }
  return true;
}

bool MainWindow::saveAs() {
  QString fileName = QFileDialog::getSaveFileName(this, tr("Save your TrackpointApp Project"), "", tr("TrackpointApp Projects (*.trackproj)"));
  std::string fileString = fileName.toUtf8().constData();
  if (!projectStore->saveProject(fileString)) {
    showErrorMessage("Something went wrong while saving the project. Please try again.", "Error saving project.");
    return false;
  }
  return true;
}

void MainWindow::close() {
  if (projectStore->isModified()) {
    if (!saveChangesPopup()) return;
  }
  if (projectStore->isRendering()) {
    if (!isRenderingPopup()) return;
  }
  projectStore->closeProject();
  renderView(NoMesh);
  cleanup();
}

void MainWindow::cleanup() {
  editWidget->resetAllSettings();
  osgWidget->clear();
  osgWidget->getPointRenderer()->clear();
}

bool MainWindow::saveChangesPopup() {
  QMessageBox msgBox;
  msgBox.setText("The current project has been modified.");
  msgBox.setInformativeText("Do you want to save your changes?");
  msgBox.setStandardButtons(QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel);
  msgBox.setDefaultButton(QMessageBox::Save);
  int ret = msgBox.exec();
  switch(ret) {
    case QMessageBox::Save:
      return save();
    case QMessageBox::Discard:
      return true;
    case QMessageBox::Cancel:
      return false;
    default:
      return false;
  }
}

bool MainWindow::isRenderingPopup() {
  QMessageBox msgBox;
  msgBox.setText("A rendering is in progress.");
  msgBox.setInformativeText("Do you really want to cancel the rendering?");
  msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
  msgBox.setDefaultButton(QMessageBox::No);
  int ret = msgBox.exec();
  switch(ret) {
    case QMessageBox::Yes:
      return true;
    case QMessageBox::No:
      return false;
    default:
      return false;
  }
}