#include "toonzqt/keyframenavigator.h"

#include "toonzqt/styleselection.h"

#include "toonzqt/gutil.h"

#include "toonz/txsheet.h"

#include "toonz/txshcolumn.h"

#include "toonz/tstageobjectkeyframe.h"

#include "toonz/stageobjectutil.h"

#include "tpixelutils.h"

#include "tfx.h"

#include "tparamcontainer.h"

#include "tspectrumparam.h"

#include "ttonecurveparam.h"

#include <qpainter></qpainter>

#include <qpixmap></qpixmap>

#include <qicon></qicon>

#ifdef _WIN32

#pragma warning(disable : 4251)

#endif

#include <memory></memory>

#include <qaction></qaction>

#include <qmouseevent></qmouseevent>

using namespace std;

//=============================================================================

// KeyframeNavigator

//-----------------------------------------------------------------------------

KeyframeNavigator::KeyframeNavigator(QWidget *parent, TFrameHandle *frameHandle)

    : QToolBar(parent), m_frameHandle(frameHandle) {

  setLayoutDirection(Qt::LeftToRight);

  setIconSize(QSize(15, 23));

  setObjectName("keyFrameNavigator");

  QIcon previewActIcon = createQIconPNG("prevkey");

  previewActIcon.addFile(QString(":Resources/prevkey_disabled.png"), QSize(),

                         QIcon::Disabled);

  m_actPreviewKey = new QAction(previewActIcon, tr("Previous Key"), this);

  connect(m_actPreviewKey, SIGNAL(triggered()), SLOT(togglePrevKeyAct()));

  addAction(m_actPreviewKey);

  m_actKeyNo = new QAction(createQIconPNG("key_no"), tr("Set Key"), this);

  connect(m_actKeyNo, SIGNAL(triggered()), SLOT(toggleKeyAct()));

  addAction(m_actKeyNo);

  m_actKeyPartial =

      new QAction(createQIconPNG("key_partial"), tr("Set Key"), this);

  connect(m_actKeyPartial, SIGNAL(triggered()), SLOT(toggleKeyAct()));

  addAction(m_actKeyPartial);

  m_actKeyTotal = new QAction(createQIconPNG("key_total"), tr("Set Key"), this);

  connect(m_actKeyTotal, SIGNAL(triggered()), SLOT(toggleKeyAct()));

  addAction(m_actKeyTotal);

  QIcon nextActIcon = createQIconPNG("nextkey");

  nextActIcon.addFile(QString(":Resources/nextkey_disabled.png"), QSize(),

                      QIcon::Disabled);

  m_actNextKey = new QAction(nextActIcon, tr("Next Key"), this);

  connect(m_actNextKey, SIGNAL(triggered()), SLOT(toggleNextKeyAct()));

  addAction(m_actNextKey);

}

//-----------------------------------------------------------------------------

void KeyframeNavigator::update() {

  // Prev button

  if (hasPrev())

    m_actPreviewKey->setDisabled(false);

  else

    m_actPreviewKey->setDisabled(true);

  bool isFullKey = isFullKeyframe();

  bool isKey     = isKeyframe();

  if (isKey && !isFullKey) {

    m_actKeyNo->setVisible(false);

    m_actKeyTotal->setVisible(false);

    m_actKeyPartial->setVisible(true);

    m_actKeyPartial->setDisabled(false);

  }

  if (isFullKey) {

    m_actKeyNo->setVisible(false);

    m_actKeyPartial->setVisible(false);

    m_actKeyTotal->setVisible(true);

    m_actKeyTotal->setDisabled(false);

  }

  if (!isKey && !isFullKey) {

    m_actKeyPartial->setVisible(false);

    m_actKeyTotal->setVisible(false);

    m_actKeyNo->setVisible(true);

    m_actKeyNo->setDisabled(false);

  }

  // Next button

  if (hasNext())

    m_actNextKey->setDisabled(false);

  else

    m_actNextKey->setDisabled(true);

}

//-----------------------------------------------------------------------------

void KeyframeNavigator::showEvent(QShowEvent *e) {

  update();

  if (!m_frameHandle) return;

  connect(m_frameHandle, SIGNAL(frameSwitched()), this, SLOT(update()));

}

//-----------------------------------------------------------------------------

void KeyframeNavigator::hideEvent(QHideEvent *e) {

  if (!m_frameHandle) return;

  disconnect(m_frameHandle);

}

//=============================================================================

// ViewerKeyframeNavigator

//-----------------------------------------------------------------------------

TStageObject *ViewerKeyframeNavigator::getStageObject() const {

  if (!m_xsheetHandle || !m_objectHandle) return 0;

  TStageObjectId objectId = m_objectHandle->getObjectId();

  TXsheet *xsh            = m_xsheetHandle->getXsheet();

  // Se e' una colonna sound non posso settare chiavi

  if (objectId.isColumn()) {

    TXshColumn *column = xsh->getColumn(objectId.getIndex());

    if (column && column->getSoundColumn()) return 0;

  }

  return xsh->getStageObject(objectId);

}

//-----------------------------------------------------------------------------

bool ViewerKeyframeNavigator::hasNext() const {

  TStageObject *pegbar = getStageObject();

  if (!pegbar) return false;

  int r0, r1;

  pegbar->getKeyframeRange(r0, r1);

  return r0 <= r1 && getCurrentFrame() < r1;

}

//-------------------------------------------------------------------

bool ViewerKeyframeNavigator::hasPrev() const {

  TStageObject *pegbar = getStageObject();

  if (!pegbar) return false;

  int r0, r1;

  pegbar->getKeyframeRange(r0, r1);

  return r0 <= r1 && getCurrentFrame() > r0;

}

//-------------------------------------------------------------------

bool ViewerKeyframeNavigator::hasKeyframes() const {

  TStageObject *pegbar = getStageObject();

  if (!pegbar) return false;

  int r0, r1;

  pegbar->getKeyframeRange(r0, r1);

  return r0 <= r1;

}

//-------------------------------------------------------------------

bool ViewerKeyframeNavigator::isKeyframe() const {

  TStageObject *pegbar = getStageObject();

  if (!pegbar) return false;

  return pegbar->isKeyframe(getCurrentFrame());

}

//-------------------------------------------------------------------

bool ViewerKeyframeNavigator::isFullKeyframe() const {

  TStageObject *pegbar = getStageObject();

  if (!pegbar) return false;

  return pegbar->isFullKeyframe(getCurrentFrame());

}

//-------------------------------------------------------------------

void ViewerKeyframeNavigator::toggle() {

  TStageObject *pegbar = getStageObject();

  if (!pegbar) return;

  int frame = getCurrentFrame();

  if (pegbar->isFullKeyframe(frame)) {

    TStageObject::Keyframe key = pegbar->getKeyframe(frame);

    pegbar->removeKeyframeWithoutUndo(frame);

    UndoRemoveKeyFrame *undo =

        new UndoRemoveKeyFrame(pegbar->getId(), frame, key, m_xsheetHandle);

    undo->setObjectHandle(m_objectHandle);

    TUndoManager::manager()->add(undo);

  } else {

    UndoSetKeyFrame *undo =

        new UndoSetKeyFrame(pegbar->getId(), frame, m_xsheetHandle);

    pegbar->setKeyframeWithoutUndo(frame);

    undo->setObjectHandle(m_objectHandle);

    TUndoManager::manager()->add(undo);

  }

  m_objectHandle->notifyObjectIdChanged(false);

}

//-------------------------------------------------------------------

void ViewerKeyframeNavigator::goNext() {

  TStageObject *pegbar = getStageObject();

  if (!pegbar) return;

  int frame = getCurrentFrame();

  TStageObject::KeyframeMap keyframes;

  pegbar->getKeyframes(keyframes);

  TStageObject::KeyframeMap::iterator it;

  for (it = keyframes.begin(); it != keyframes.end(); ++it)

    if (it->first > frame) {

      setCurrentFrame(it->first);

      return;

    }

}

//-------------------------------------------------------------------

void ViewerKeyframeNavigator::goPrev() {

  TStageObject *pegbar = getStageObject();

  if (!pegbar) return;

  int frame = getCurrentFrame();

  TStageObject::KeyframeMap keyframes;

  pegbar->getKeyframes(keyframes);

  TStageObject::KeyframeMap::reverse_iterator it;

  for (it = keyframes.rbegin(); it != keyframes.rend(); ++it)

    if (it->first < frame) {

      setCurrentFrame(it->first);

      return;

    }

}

//-----------------------------------------------------------------------------

void ViewerKeyframeNavigator::showEvent(QShowEvent *e) {

  if (!m_objectHandle) return;

  connect(m_objectHandle, SIGNAL(objectSwitched()), this, SLOT(update()));

  connect(m_objectHandle, SIGNAL(objectChanged(bool)), this, SLOT(update()));

  KeyframeNavigator::showEvent(e);

}

//-----------------------------------------------------------------------------

void ViewerKeyframeNavigator::hideEvent(QHideEvent *e) {

  if (!m_objectHandle) return;

  disconnect(m_objectHandle);

  KeyframeNavigator::hideEvent(e);

}

//=============================================================================

namespace {

//-----------------------------------------------------------------------------

class UndoPaletteSetKeyFrame final : public TUndo {

  int m_frame;

  int m_styleId;

  TPaletteHandle *m_paletteHandle;

public:

  UndoPaletteSetKeyFrame(int styleId, int frame, TPaletteHandle *paletteHandle)

      : m_frame(frame), m_styleId(styleId), m_paletteHandle(paletteHandle) {}

  void undo() const override { setKeyFrame(); }

  void redo() const override { setKeyFrame(); }

  int getSize() const override { return sizeof(*this); }

protected:

  void setKeyFrame() const {

    TPalette *palette = m_paletteHandle->getPalette();

    if (palette->isKeyframe(m_styleId, m_frame))

      palette->clearKeyframe(m_styleId, m_frame);

    else

      palette->setKeyframe(m_styleId, m_frame);

    m_paletteHandle->notifyPaletteChanged();

  }

};

//-----------------------------------------------------------------------------

}  // namespace

//=============================================================================

//=============================================================================

// PaletteKeyframeNavigator

//-----------------------------------------------------------------------------

bool PaletteKeyframeNavigator::hasNext() const {

  TPalette *palette = getPalette();

  if (!palette) return false;

  int styleId = getStyleIndex();

  int frame   = getCurrentFrame();

  int n       = palette->getKeyframeCount(styleId);

  for (int i = n - 1; i >= 0; i--) {

    int f = palette->getKeyframe(styleId, i);

    if (f > frame)

      return true;

    else if (f <= frame)

      return false;

  }

  return false;

}

//-----------------------------------------------------------------------------

bool PaletteKeyframeNavigator::hasPrev() const {

  TPalette *palette = getPalette();

  if (!palette) return false;

  int styleId = getStyleIndex();

  int frame   = getCurrentFrame();

  int n       = palette->getKeyframeCount(styleId);

  for (int i = 0; i < n; i++) {

    int f = palette->getKeyframe(styleId, i);

    if (f < frame)

      return true;

    else if (f >= frame)

      return false;

  }

  return false;

}

//-----------------------------------------------------------------------------

bool PaletteKeyframeNavigator::hasKeyframes() const {

  TPalette *palette = getPalette();

  if (!palette) return false;

  return palette->getKeyframeCount(getStyleIndex()) > 0;

}

//-----------------------------------------------------------------------------

bool PaletteKeyframeNavigator::isKeyframe() const {

  TPalette *palette = getPalette();

  if (!palette) return false;

  int frame = getCurrentFrame();

  return palette->isKeyframe(getStyleIndex(), frame);

}

//-----------------------------------------------------------------------------

void PaletteKeyframeNavigator::toggle() {

  TPalette *palette = getPalette();

  if (!palette) return;

  int styleId = getStyleIndex(), frame = getCurrentFrame();

  std::unique_ptr<undopalettesetkeyframe> undo(</undopalettesetkeyframe>

      new UndoPaletteSetKeyFrame(styleId, frame, m_paletteHandle));

  undo->redo();

  TUndoManager::manager()->add(undo.release());

}

//-----------------------------------------------------------------------------

void PaletteKeyframeNavigator::goNext() {

  TPalette *palette = getPalette();

  if (!palette) return;

  int styleId = getStyleIndex();

  int frame   = getCurrentFrame();

  int n       = palette->getKeyframeCount(styleId);

  for (int i = 0; i < n; i++) {

    int f = palette->getKeyframe(styleId, i);

    if (f > frame) {

      setCurrentFrame(f);

      break;

    }

  }

}

//-----------------------------------------------------------------------------

void PaletteKeyframeNavigator::goPrev() {

  TPalette *palette = getPalette();

  if (!palette) return;

  int styleId = getStyleIndex();

  int frame   = getCurrentFrame();

  int n       = palette->getKeyframeCount(styleId);

  for (int i = n - 1; i >= 0; i--) {

    int f = palette->getKeyframe(styleId, i);

    if (f < frame) {

      setCurrentFrame(f);

      break;

    }

  }

}

//-----------------------------------------------------------------------------

void PaletteKeyframeNavigator::showEvent(QShowEvent *e) {

  if (!m_paletteHandle) return;

  connect(m_paletteHandle, SIGNAL(paletteSwitched()), this, SLOT(update()));

  connect(m_paletteHandle, SIGNAL(paletteChanged()), this, SLOT(update()));

  connect(m_paletteHandle, SIGNAL(colorStyleSwitched()), this, SLOT(update()));

  KeyframeNavigator::showEvent(e);

}

//-----------------------------------------------------------------------------

void PaletteKeyframeNavigator::hideEvent(QHideEvent *e) {

  if (!m_paletteHandle) return;

  disconnect(m_paletteHandle);

  KeyframeNavigator::hideEvent(e);

}

//=============================================================================

namespace {

//-----------------------------------------------------------------------------

//! Se non c'e' un keyframe successivo ritorna frame

int getNextKeyframe(TFxP fx, int frame) {

  if (!fx) return frame;

  int targetFrame = frame;

  for (int i = 0; i < fx->getParams()->getParamCount(); i++) {

    TParamP param = fx->getParams()->getParam(i);

    int j         = param->getNextKeyframe(frame);

    if (j < 0) continue;

    int f = (int)param->keyframeIndexToFrame(j);

    if (targetFrame == frame || f < targetFrame) targetFrame = f;

  }

  return targetFrame;

}

//-----------------------------------------------------------------------------

//! Se non c'e' un keyframe precedente ritorna frame

int getPrevKeyframe(TFxP fx, int frame) {

  if (!fx) return frame;

  int targetFrame = frame;

  for (int i = 0; i < fx->getParams()->getParamCount(); i++) {

    TParamP param = fx->getParams()->getParam(i);

    int j         = param->getPrevKeyframe(frame);

    if (j < 0) continue;

    int f = (int)param->keyframeIndexToFrame(j);

    if (targetFrame == frame || f > targetFrame) targetFrame = f;

  }

  return targetFrame;

}

//-----------------------------------------------------------------------------

void setKeyframe(TFxP fx, int frame, bool on) {

  if (fx)

    for (int i = 0; i < fx->getParams()->getParamCount(); i++) {

      TParamP param = fx->getParams()->getParam(i);

      if (TDoubleParamP dp = param) {

        if (on)

          dp->setValue(frame, dp->getValue(frame));

        else

          dp->deleteKeyframe(frame);

      }

    }

}

//-----------------------------------------------------------------------------

}  // anonymous namespace

//=============================================================================

//=============================================================================

// FxKeyframeNavigator

//-----------------------------------------------------------------------------

bool FxKeyframeNavigator::hasNext() const {

  TFx *fx = getFx();

  if (!fx)

    return false;

  else

    return getNextKeyframe(fx, getCurrentFrame()) > getCurrentFrame();

}

//-----------------------------------------------------------------------------

bool FxKeyframeNavigator::hasPrev() const {

  TFx *fx = getFx();

  if (!fx)

    return false;

  else

    return getPrevKeyframe(fx, getCurrentFrame()) < getCurrentFrame();

}

//-----------------------------------------------------------------------------

bool FxKeyframeNavigator::hasKeyframes() const {

  TFx *fx = getFx();

  if (!fx) return false;

  for (int i = 0; i < fx->getParams()->getParamCount(); i++) {

    TParamP param = fx->getParams()->getParam(i);

    if (param->hasKeyframes()) return true;

  }

  return false;

}

//-----------------------------------------------------------------------------

bool FxKeyframeNavigator::isKeyframe() const {

  TFx *fx = getFx();

  if (!fx) return false;

  for (int i = 0; i < fx->getParams()->getParamCount(); i++) {

    TParamP param = fx->getParams()->getParam(i);

    if (param->isKeyframe(getCurrentFrame())) return true;

  }

  return false;

}

//-----------------------------------------------------------------------------

bool FxKeyframeNavigator::isFullKeyframe() const {

  TFx *fx = getFx();

  if (!fx) return false;

  int keyFrameCount        = 0;

  int animatableParamCount = 0;

  for (int i = 0; i < fx->getParams()->getParamCount(); i++) {

    TParamP param = fx->getParams()->getParam(i);

    if (param->isAnimatable()) {

      animatableParamCount++;

      if (param->isKeyframe(getCurrentFrame())) keyFrameCount++;

    }

  }

  return animatableParamCount > 0 && keyFrameCount == animatableParamCount;

}

//-----------------------------------------------------------------------------

void FxKeyframeNavigator::toggle() {

  TFx *fx = getFx();

  if (!fx) return;

  int i, paramCount = fx->getParams()->getParamCount();

  // determino in quale caso ci troviamo:

  // il frame corrente non e' keyframe di nessun parametro (isKeyframe=false),

  // di qualche parametro o di tutti i parametri (isFullKeyframe=true)

  bool isFullKeyframe = true;

  bool isKeyframe     = false;

  int frame           = getCurrentFrame();

  for (i = 0; i < paramCount; i++) {

    TParamP param = fx->getParams()->getParam(i);

    if (!param->isAnimatable()) continue;

    if (param->isKeyframe(frame))

      isKeyframe = true;

    else

      isFullKeyframe = false;

  }

  if (!isKeyframe) isFullKeyframe = false;

  // modifico lo stato: nokeyframe->full, full->no, partial->full

  bool on = !isKeyframe || isKeyframe && !isFullKeyframe;

  for (i = 0; i < fx->getParams()->getParamCount();

       i++) {  // TODO. spostare questo codice in TParam

    TParamP param = fx->getParams()->getParam(i);

    if (TDoubleParamP dp = param) {

      if (on)

        dp->setValue(frame, dp->getValue(frame));

      else

        dp->deleteKeyframe(frame);

    } else if (TRangeParamP rp = param) {

      if (on)

        rp->setValue(frame, rp->getValue(frame));

      else

        rp->deleteKeyframe(frame);

    } else if (TPointParamP pp = param) {

      if (on)

        pp->setValue(frame, pp->getValue(frame));

      else

        pp->deleteKeyframe(frame);

    } else if (TPixelParamP pip = param) {

      if (on)

        pip->setValue(frame, pip->getValue(frame));

      else

        pip->deleteKeyframe(frame);

    } else if (TSpectrumParamP sp = param) {

      if (on)

        sp->setValue(frame, sp->getValue(frame), false);

      else

        sp->deleteKeyframe(frame);

    } else if (TToneCurveParamP tcp = param) {

      if (on)

        tcp->setValue(frame, tcp->getValue(frame), false);

      else

        tcp->deleteKeyframe(frame);

    }

  }

  m_fxHandle->notifyFxChanged();

}

//-----------------------------------------------------------------------------

void FxKeyframeNavigator::goNext() {

  TFx *fx = getFx();

  if (!fx) return;

  int frame = getNextKeyframe(fx, getCurrentFrame());

  if (frame > getCurrentFrame()) {

    setCurrentFrame(frame);

    //		m_fxHandle->notifyFxChanged();

  }

}

//-----------------------------------------------------------------------------

void FxKeyframeNavigator::goPrev() {

  TFx *fx = getFx();

  if (!fx) return;

  int frame = getPrevKeyframe(fx, getCurrentFrame());

  if (frame < getCurrentFrame()) {

    setCurrentFrame(frame);

    //		m_fxHandle->notifyFxChanged();

  }

}

//-----------------------------------------------------------------------------

void FxKeyframeNavigator::showEvent(QShowEvent *e) {

  if (!m_fxHandle) return;

  connect(m_fxHandle, SIGNAL(fxSwitched()), this, SLOT(update()));

  connect(m_fxHandle, SIGNAL(fxChanged()), this, SLOT(update()));

  KeyframeNavigator::showEvent(e);

}

//-----------------------------------------------------------------------------

void FxKeyframeNavigator::hideEvent(QHideEvent *e) {

  if (!m_fxHandle) return;

  disconnect(m_fxHandle);

  KeyframeNavigator::hideEvent(e);

}