#include "toonzqt/functiontoolbar.h"

// TnzQt includes

#include "toonzqt/doublefield.h"

#include "toonzqt/functionkeyframenavigator.h"

#include "toonzqt/functionselection.h"

#include "toonzqt/framenavigator.h"

#include "toonzqt/gutil.h"

#include "toonzqt/menubarcommand.h"

// TnzLib includes

#include "toonz/tframehandle.h"

#include "toonz/doubleparamcmd.h"

// Qt includes

#include <qlabel></qlabel>

#include <qapplication></qapplication>

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

/*

   * two states: curve_defined / curve_not_defined

   * three widgets: value_fld, frame_nav, keyframe_nav

                    (if curve_not_defined only keyframe_nav is visible)

   * user actions

     - value_fld changed / keyframe_nav changed (key icon) => modify current

   curve

     - frame_nav changed / keyframe_nav changed (arrows) => modify the 'extern'

   frame

   * 'extern' events:

     - 'extern' frame changed => update fields

     - curve selected => update fields (and possibly change status)

*/

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

FunctionToolbar::FunctionToolbar(QWidget *parent)

    : DVGui::ToolBar(parent), m_frameHandle(0), m_curve(0), m_selection(0) {

  setFixedHeight(28);

  setIconSize(QSize(20, 20));

  m_valueToolbar    = new DVGui::ToolBar();

  m_keyframeToolbar = new DVGui::ToolBar();

  // value field

  m_valueToolbar->addWidget(new QLabel(tr("Value")));

  m_valueFld = new DVGui::MeasuredDoubleLineEdit();

  // frame navigator

  m_frameNavigator = new FrameNavigator(this);

  // keyframe navigator

  m_keyframeNavigator          = new FunctionKeyframeNavigator(this);

  QWidget *space               = new QWidget(this);

  DVGui::ToolBar *spaceToolBar = new DVGui::ToolBar();

  spaceToolBar->addWidget(space);

  m_valueFld->setStyleSheet("height:14px;margin-right:5px;margin-top:2px;");

  space->setMinimumHeight(22);

  space->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

  QIcon toggleIcon      = createQIcon("swap");

  QAction *toggleAction = new QAction(tr("&Open Function Curve Editor"), this);

  assert(toggleAction);

  toggleAction->setIcon(toggleIcon);

  //-----layout

  m_valueFldAction          = addWidget(m_valueToolbar);

  m_keyframeNavigatorAction = addWidget(m_keyframeToolbar);

  m_valueToolbar->addWidget(m_valueFld);

  m_valueToolbar->addSeparator();

  m_keyframeToolbar->addWidget(m_keyframeNavigator);

  m_keyframeToolbar->addSeparator();

  addWidget(m_frameNavigator);

  addSeparator();

  addWidget(spaceToolBar);

  addSeparator();

  addAction(toggleAction);

  bool ret = connect(m_valueFld, SIGNAL(valueChanged()), this,

                     SLOT(onValueFieldChanged()));

  ret = ret && connect(m_frameNavigator, SIGNAL(frameSwitched()), this,

                       SLOT(onNavFrameSwitched()));

  ret = ret && connect(toggleAction, SIGNAL(triggered()), this,

                       SIGNAL(numericalColumnToggled()));

  m_valueFldAction->setVisible(false);

  m_keyframeNavigatorAction->setVisible(false);

  assert(ret);

}

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

FunctionToolbar::~FunctionToolbar() {

  if (m_curve) {

    m_curve->removeObserver(this);

    m_curve->release();

  }

}

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

bool FunctionToolbar::anyWidgetHasFocus() {

  return hasFocus() || m_valueToolbar->hasFocus() ||

         m_keyframeToolbar->hasFocus() || m_valueFld->hasFocus() ||

         m_frameNavigator->anyWidgetHasFocus() ||

         m_keyframeNavigator->hasFocus();

}

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

void FunctionToolbar::setCurve(TDoubleParam *curve) {

  if (curve == m_curve) return;

  bool curveDefined = curve != 0;

  m_valueFldAction->setVisible(curveDefined);

  m_keyframeNavigatorAction->setVisible(curveDefined);

  m_keyframeNavigator->setCurve(curve);

  if (curve) {

    curve->addObserver(this);

    curve->addRef();

  }

  if (m_curve) {

    m_curve->removeObserver(this);

    m_curve->release();

  }

  m_curve = curve;

  if (m_curve) {

    m_valueFld->setMeasure(m_curve->getMeasureName());

    setFrame(m_frameNavigator->getFrame());

  } else {

    m_valueFld->setMeasure("");

    m_valueFld->setValue(0);

  }

}

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

void FunctionToolbar::setFrame(double frame) {

  m_frameNavigator->setFrame(tround(frame), false);

  if (m_curve)

    m_valueFld->setValue(m_curve->getValue(frame));

  else

    m_valueFld->setValue(0);

}

void FunctionToolbar::onNextKeyframe(QWidget *panel) {

  m_keyframeNavigator->onNextKeyframe(panel);

}

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

void FunctionToolbar::onPrevKeyframe(QWidget *panel) {

  m_keyframeNavigator->onPrevKeyframe(panel);

}

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

void FunctionToolbar::onValueFieldChanged() {

  if (!m_curve) return;

  int frame     = m_frameNavigator->getFrame();

  double value  = m_valueFld->getValue();

  double fValue = m_curve->getValue(frame);

  if (value == fValue) return;

  KeyframeSetter::setValue(m_curve, frame, value);

}

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

void FunctionToolbar::onFrameSwitched() { setFrame(m_frameHandle->getFrame()); }

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

void FunctionToolbar::onNavFrameSwitched() {

  if (m_frameHandle) m_frameHandle->setFrame(m_frameNavigator->getFrame());

}

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

void FunctionToolbar::onSelectionChanged() {

  if (m_selection) {

    if (m_selection->getSelectedKeyframeCount() == 1) {

      QPair<tdoubleparam *,="" int=""> k = m_selection->getSelectedKeyframe(0);</tdoubleparam>

      if (k.first == m_curve) {

        int frame = m_curve->keyframeIndexToFrame(k.second);

        setFrame(frame);

        return;

      }

    } else if (m_selection->getSelectedKeyframeCount() > 1) {

      setFrame(m_frameHandle->getFrame());

      m_valueFld->setText("");

      return;

    }

  }

  if (m_frameHandle && m_curve)

    setFrame(m_frameHandle->getFrame());

  else

    m_valueFld->setText("");

}

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

void FunctionToolbar::setFrameHandle(TFrameHandle *frameHandle) {

  if (m_frameHandle) m_frameHandle->disconnect(this);

  m_frameHandle = frameHandle;

  if (m_frameHandle)

    connect(m_frameHandle, SIGNAL(frameSwitched()), this,

            SLOT(onFrameSwitched()));

  m_keyframeNavigator->setFrameHandle(frameHandle);

}

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

void FunctionToolbar::setSelection(FunctionSelection *selection) {

  if (m_selection != selection) {

    if (m_selection)

      disconnect(m_selection, SIGNAL(selectionChanged()), this,

                 SLOT(onSelectionChanged()));

    m_selection = selection;

    if (m_selection)

      connect(m_selection, SIGNAL(selectionChanged()), this,

              SLOT(onSelectionChanged()));

  }

}

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

void FunctionToolbar::onChange(const TParamChange &) {

  m_keyframeNavigator->update();

}