#include <QGridLayout>
#include <QPixmap>
#include <QLabel>

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>

#include "EditSimShotDialog.h"
#include "Convert.h"

#include <QDebug>

using namespace cv;

EditSimShotDialog::EditSimShotDialog(const QString &fName, int currIdx, const QList<qint64> &shotPositions, int nVignettes, int frameWidth, QWidget *parent)
  : QDialog(parent),
    m_nVignettes(nVignettes),
    m_refIdx(currIdx),
    m_currIdx(currIdx),
    m_shotPositions(shotPositions),
    m_nCamera(0)
{
  int fWidth;
  int fHeight;

  // vector containing the camera ids
  m_shotCamera.resize(shotPositions.size());

  // retrieving movie features
  m_cap.release();
  m_cap.open(fName.toStdString());
  fWidth = m_cap.get(CV_CAP_PROP_FRAME_WIDTH);
  fHeight = m_cap.get(CV_CAP_PROP_FRAME_HEIGHT);
  m_vignetteWidth = frameWidth / m_nVignettes;
  m_vignetteHeight = m_vignetteWidth * fHeight / fWidth;

  // doesn't work anymore after updating Ubuntu to 14.04
  // m_frameDur = 1 / m_cap.get(CV_CAP_PROP_FPS) * 1000;
  m_frameDur = 1 / 25.0 * 1000;

  // initializing vignette list
  int i(m_refIdx);

  while (i >= m_refIdx - m_nVignettes + 1) {
    if (i < 0)
      m_vignettePositions.push_front(-1);
    else
      m_vignettePositions.push_front(m_shotPositions[i] - m_frameDur);
    i--;
  }

  m_currentVignette = new QLabel;
  m_currentVignette->setPixmap(refShot());

  m_vignetteWidget = new VignetteWidget(m_nVignettes, frameWidth);
  m_vignetteWidget->setVideoCapture(fName);
  m_vignetteWidget->updateVignette(m_vignettePositions);

  QGridLayout *layout = new QGridLayout;
  layout->addWidget(m_currentVignette, 0, 0, Qt::AlignRight);
  layout->addWidget(m_vignetteWidget, 1, 0, Qt::AlignRight);

  setLayout(layout);
  
  setWindowTitle("Annotate similar shots - Shot " + QString::number(m_refIdx + 1));

  // setting deletion operations vector
  m_delOp.resize(m_shotPositions.size());
  m_delOp.fill(false);
}

///////////////////////////////////
// slot called when pressing key //
///////////////////////////////////

void EditSimShotDialog::keyPressEvent(QKeyEvent *event)
{
  int toAddIdx; // index of shot to add to visible ones
  int idxToDel; // index of single linked shot to untie

  /********************************/
  /* going through previous shots */
  /********************************/

  if (event->key() == Qt::Key_Left && m_currIdx > m_nVignettes / 2) {

    m_currIdx--;
    toAddIdx = m_currIdx - m_nVignettes + 1;

    if (toAddIdx < 0)
      m_vignettePositions.push_front(-1);
    else
      m_vignettePositions.push_front(m_shotPositions[toAddIdx] - m_frameDur);
    m_vignettePositions.pop_back();
  }
  
  else if (event->key() == Qt::Key_Right && m_currIdx < m_refIdx) {

    m_vignettePositions.push_back(m_shotPositions[++m_currIdx] - m_frameDur);
    m_vignettePositions.pop_front();
  }

  /*******************************************************/
  /* linking current shot to previous one or ignoring it */
  /*******************************************************/

  else if (event->key() == Qt::Key_Return || event->key() == Qt::Key_Backspace) {

    // no similar shot found: increment camera label
    if (event->key() == Qt::Key_Backspace) {
      m_shotCamera[m_refIdx] = m_nCamera++;

      // setting corresponding deletion operation to true
      m_delOp[m_refIdx] = true;
    }

    // similar shot found: label current shot with selected one
    else if (event->key() == Qt::Key_Return) {
      
      // linking the shots
      m_shotCamera[m_refIdx] = m_shotCamera[m_currIdx - 1];
      emit labelSimShot(m_shotPositions[m_currIdx - 1], m_shotCamera[m_currIdx - 1], Segment::Manual);
      emit labelSimShot(m_shotPositions[m_refIdx], m_shotCamera[m_refIdx], Segment::Manual);

      // setting corresponding deletion operation to false
      m_delOp[m_refIdx] = false;
    }

    // move to next shot position
    m_currIdx = ++m_refIdx;

    // last shot annotated: exit dialog
    if (m_refIdx == m_shotPositions.size())
      accept();

    else {

      // re-initializing vignette list
      int i(m_refIdx);
      m_vignettePositions.clear();

      while (i >= m_refIdx - m_nVignettes + 1) {
	if (i < 0)
	  m_vignettePositions.push_front(-1);
	else
	  m_vignettePositions.push_front(m_shotPositions[i] - m_frameDur);
	i--;
      }

      // set pixmap of current shot and window title
      m_currentVignette->setPixmap(refShot());
      setWindowTitle("Annotate similar shots - Shot " + QString::number(m_refIdx + 1));
      
      update();
    }
  }

  /*****************************/
  /* cancelling last operation */
  /*****************************/

  else if (event->key() == Qt::Key_Escape && m_refIdx > 0) {

    // updating indices
    m_currIdx = --m_refIdx;

    // last operation consisted in ignoring new shot
    if (m_delOp[m_refIdx])
      m_nCamera--;

    // last operation consisted in linking the shot to a previous one
    else {
      // retrieving camera label to cancel
      const int camToDel(m_shotCamera[m_refIdx]);

      // deleting last shot annotation
      emit labelSimShot(m_shotPositions[m_refIdx], -1, Segment::Manual);

      // if linked with only one shot, delete its associated label
      if (m_shotCamera.count(camToDel) == 2) {
	idxToDel = m_shotCamera.lastIndexOf(camToDel);
	emit labelSimShot(m_shotPositions[idxToDel], -1, Segment::Manual);
      }
    }

    // updating vignette position list
    toAddIdx = m_currIdx - m_nVignettes + 1;

    if (toAddIdx < 0)
      m_vignettePositions.push_front(-1);
    else
      m_vignettePositions.push_front(m_shotPositions[toAddIdx] - m_frameDur);

    m_vignettePositions.pop_back();

    // set pixmap of current shot and window title
    m_currentVignette->setPixmap(refShot());
    setWindowTitle("Annotate similar shots - Shot " + QString::number(m_refIdx + 1));

    update();
  }

  // updating corresponding images
  m_vignetteWidget->updateVignette(m_vignettePositions);
  m_vignetteWidget->update();
}

///////////////////////
// auxiliary methods //
///////////////////////

QPixmap EditSimShotDialog::refShot()
{
  Mat frame;

  // retrieving selected frame
  m_cap.set(CV_CAP_PROP_POS_MSEC, m_shotPositions[m_refIdx]);
  m_cap >> frame;
  cv::resize(frame, frame, Size(m_vignetteWidth, m_vignetteHeight));

  return QPixmap::fromImage(Convert::fromBGRMatToQImage(frame));
}