#include "MainWindow.h"
#include "ui_mainwindow.h"

#include <QFile>
#include <QKeyEvent>
#include <QFileDialog>
#include <QMessageBox>

#include <vtkDataSetReader.h>

#include "../Program.h"
#include "../advection/UVGrid.h"
#include "../advection/kernel/RK4AdvectionKernel.h"
#include "../advection/kernel/SnapBoundaryConditionKernel.h"
#include "../commands/TimerCallbackCommand.h"
#include "../layers/BackgroundImage.h"
#include "../layers/EColLayer.h"
#include "../layers/EGlyphLayer.h"
#include "../layers/LColLayer.h"
#include "../layers/LGlyphLayer.h"
#include "../layers/enums.h"
#include "../layers/LegendLayer.h"

using namespace std;

#define DT 3600

MainWindow::MainWindow(QWidget* parent)
  : QMainWindow(parent)
  , ui(new Ui::MainWindow)
{
  ui->setupUi(this);
  this->setupTechniques();
}

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


void MainWindow::setupTechniques() {
  cout << "Reading data..." << endl;
  string dataPath = "../../../../data";
  shared_ptr<UVGrid> uvGrid = make_shared<UVGrid>(dataPath);

  // initialize techniques
  Program *program = this->ui->program;
  auto technique1 = new Technique(program->getCamera());
  auto technique2 = new Technique(program->getCamera());
  
  // add bg layer
  auto bg = new BackgroundImage(dataPath + "/map_qgis_1035.png");
  technique1->addLayer(bg);
  technique2->addLayer(bg);

  // add legend layer
  technique1->addLayer(new LegendLayer(dataPath + "/legends", "COLGLYPH"));
  technique2->addLayer(new LegendLayer(dataPath + "/legends", "GLYPHCOL"));

  // add Euler layers
  technique1->addLayer(new EColLayer(uvGrid));
  technique2->addLayer(new EGlyphLayer(uvGrid));

  // setup LGlyphLayer
  auto kernelRK4 = make_unique<RK4AdvectionKernel>(uvGrid);
  auto kernelRK4BoundaryChecked = make_unique<SnapBoundaryConditionKernel>(std::move(kernelRK4), uvGrid);
  auto lGlyph = new LGlyphLayer(uvGrid, std::move(kernelRK4BoundaryChecked));

  auto kernelRK4_2 = make_unique<RK4AdvectionKernel>(uvGrid);
  auto kernelRK4BoundaryChecked_2 = make_unique<SnapBoundaryConditionKernel>(std::move(kernelRK4_2), uvGrid);
  auto lCol = new LColLayer(uvGrid, std::move(kernelRK4BoundaryChecked_2));
  lGlyph->setDt(DT);
  lCol->setDt(DT);

  technique1->addLayer(lGlyph);
  technique2->addLayer(lCol); 

  program->addTechnique(technique1);
  program->addTechnique(technique2);

  program->setActiveTechnique(COLGLYPH);

  // Setup timer
  auto intr = program->getInteractor();
  this->timer = vtkSmartPointer<TimerCallbackCommand>::New(program);
  this->timer->SetClientData(program);
  this->timer->setDt(DT);
  intr->AddObserver(vtkCommand::TimerEvent, this->timer);
  intr->CreateRepeatingTimer(17); // 60 fps == 1000 / 60 == 16.7 ms per frame
}


void MainWindow::keyPressEvent(QKeyEvent *ev) {
  if (ev->key() == Qt::Key_Space)  {
    this->timer->togglePaused();
  }
}

/* --------------------------------------------------------------------
 * +                    QTWidget callbacks                            +
 * --------------------------------------------------------------------*/


void MainWindow::on_FirstButton_clicked(bool checked) {
  if (checked) {
    ui->program->setActiveTechnique(COLGLYPH);
    ui->program->requestRender();
  }
}


void MainWindow::on_SecondButton_clicked(bool checked) {
  if (checked) {
    ui->program->setActiveTechnique(GLYPHCOL);
    ui->program->requestRender();
  }
}


void MainWindow::on_ComplementaryButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setColourMode(COMPLEMENTARY);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_ContrastingButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setColourMode(CONTRASTING);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_MonochromaticButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setColourMode(MONOCHROMATIC);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_SaturateButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setSaturationMode(SATURATED);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_DesaturateButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setSaturationMode(DESATURATED);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_CircleButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setGlyphStyle(CIRCLE);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_TriangleButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setGlyphStyle(TRIANGLE);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_SquareButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setGlyphStyle(SQUARE);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_CrossButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setGlyphStyle(CROSS);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_FullySampledButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setSamplingMode(FULLYSAMPLED);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_RegularlySubsampledButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setSamplingMode(REGULARLYSUBSAMPLED);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_IregularlySubsampledButton_clicked(bool checked) {
  if (checked) {
    for (Technique *t : ui->program->getTechniques()) {
      t->setSamplingMode(IRREGULARLYSUBSAMPLED);
    }
    ui->program->requestRender();
  }
}


void MainWindow::on_pauseButton_clicked(bool checked) {
  this->timer->togglePaused();
}