#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);

	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 = createQIconOnOffPNG("swap", false);

	QAction *toggleAction = new QAction(tr("&Function Editor Toggle"), 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::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();

}