#include "toonzqt/functionviewer.h"

// TnzQt includes
#include "toonzqt/functionselection.h"
#include "toonzqt/functionpanel.h"
#include "toonzqt/functiontreeviewer.h"
#include "toonzqt/functionsheet.h"
#include "toonzqt/functionsegmentviewer.h"
#include "toonzqt/functiontoolbar.h"
#include "toonzqt/swatchviewer.h"
#include "toonzqt/dvscrollwidget.h"

// TnzLib includes
#include "toonz/tstageobjecttree.h"
#include "toonz/txsheet.h"
#include "toonz/txsheethandle.h"
#include "toonz/tobjecthandle.h"
#include "toonz/tfxhandle.h"
#include "toonz/tcolumnhandle.h"
#include "toonz/fxdag.h"
#include "toonz/txshzeraryfxcolumn.h"
#include "toonz/tcolumnfx.h"
#include "toonz/toonzscene.h"
#include "toonz/tproject.h"
#include "toonz/tscenehandle.h"
#include "toonz/sceneproperties.h"

// TnzBase includes
#include "tparamcontainer.h"
#include "tunit.h"

// TnzCore includes
#include "tstopwatch.h"

// Qt includes
#include <QPainter>
#include <QMouseEvent>
#include <QWheelEvent>
#include <QBoxLayout>
#include <QPushButton>
#include <QStackedWidget>
#include <QLabel>
#include <QToolBar>
#include <QAction>

using namespace DVGui;

//=============================================================================
//
// FunctionPanel
//
//-----------------------------------------------------------------------------

#if QT_VERSION >= 0x050500
FunctionViewer::FunctionViewer(QWidget *parent, Qt::WindowFlags flags)
#else
FunctionViewer::FunctionViewer(QWidget *parent, Qt::WFlags flags)
#endif
	: QSplitter(parent), m_xshHandle(0), m_frameHandle(0), m_objectHandle(0), m_fxHandle(0), m_columnHandle(0), m_curve(0), m_selection(new FunctionSelection()), m_sceneHandle(0)
{
	setObjectName("FunctionEditor");

	// Prepare local timeline
	m_localFrame.setFrame(0);
	setFocusPolicy(Qt::NoFocus);

	m_treeView = new FunctionTreeView(this);
	m_functionGraph = new FunctionPanel(this);
	m_numericalColumns = new FunctionSheet();
	m_toolbar = new FunctionToolbar;
	m_segmentViewer = new FunctionSegmentViewer(this, m_numericalColumns, m_functionGraph);
	QWidget *leftPanel = new QWidget();
	QWidget *rightPanel = new QWidget();

	//----
	m_treeView->resize(150, m_treeView->size().height());
	m_treeView->setMinimumWidth(0);

	FunctionTreeModel *ftModel =
		dynamic_cast<FunctionTreeModel *>(m_treeView->model());
	m_functionGraph->setModel(ftModel);
	m_numericalColumns->setModel(ftModel);

	m_functionGraph->setSelection(getSelection());
	m_numericalColumns->setSelection(getSelection());

	m_numericalColumns->setViewer(this);

	m_toolbar->setSelection(m_selection);
	m_toolbar->setFocusPolicy(Qt::NoFocus);

	m_toolbar->setFrameHandle(&m_localFrame);		   // The function editor adopts an internal timeline
	m_functionGraph->setFrameHandle(&m_localFrame);	// synchronized among its various sub-widgets.
	m_numericalColumns->setFrameHandle(&m_localFrame); // In case an external m_frameHandle is specified,
													   // it synchronizes with that, too.

	//---- layout

	QVBoxLayout *leftLayout = new QVBoxLayout();
	leftLayout->setMargin(0);
	leftLayout->setSpacing(0);
	{
		leftLayout->addWidget(m_toolbar);
		leftLayout->addSpacing(36);
		leftLayout->addWidget(m_numericalColumns);
	}
	leftPanel->setLayout(leftLayout);

	addWidget(leftPanel);

	QVBoxLayout *rightLayout = new QVBoxLayout();
	rightLayout->setMargin(0);
	rightLayout->setSpacing(5);
	{
		rightLayout->addWidget(m_segmentViewer, 0);
		rightLayout->addWidget(m_treeView, 1);
	}
	rightPanel->setLayout(rightLayout);

	addWidget(rightPanel);

	//--- set the splitter's default size
	setSizes(QList<int>() << 500 << 200);
	setStretchFactor(0, 5);
	setStretchFactor(1, 2);

	//---- signal-slot connections
	bool ret = true;
	ret = ret && connect(m_toolbar, SIGNAL(numericalColumnToggled()), this, SLOT(toggleMode()));
	ret = ret && connect(ftModel, SIGNAL(activeChannelsChanged()), m_functionGraph, SLOT(update()));
	ret = ret && connect(ftModel, SIGNAL(activeChannelsChanged()), m_numericalColumns, SLOT(updateAll()));
	ret = ret && connect(ftModel, SIGNAL(curveChanged(bool)), m_treeView, SLOT(update()));
	ret = ret && connect(ftModel, SIGNAL(curveChanged(bool)), m_functionGraph, SLOT(update()));
	ret = ret && connect(ftModel, SIGNAL(curveChanged(bool)), m_numericalColumns, SLOT(updateAll()));
	ret = ret && connect(ftModel, SIGNAL(curveSelected(TDoubleParam *)), this, SLOT(onCurveSelected(TDoubleParam *)));
	ret = ret && connect(ftModel, SIGNAL(curveChanged(bool)), m_segmentViewer, SLOT(onCurveChanged()));
	ret = ret && connect(ftModel, SIGNAL(curveChanged(bool)), this, SLOT(onCurveChanged(bool)));
	ret = ret && connect(&m_localFrame, SIGNAL(frameSwitched()), this, SLOT(onFrameSwitched()));
	ret = ret && connect(getSelection(), SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
	ret = ret && connect(m_functionGraph, SIGNAL(keyframeSelected(double)), m_toolbar, SLOT(setFrame(double)));

	ret = ret && connect(m_treeView, SIGNAL(switchCurrentObject(TStageObject *)),
						 this, SLOT(doSwitchCurrentObject(TStageObject *)));
	ret = ret && connect(m_treeView, SIGNAL(switchCurrentFx(TFx *)),
						 this, SLOT(doSwitchCurrentFx(TFx *)));

	ret = ret && connect(ftModel, SIGNAL(currentChannelChanged(FunctionTreeModel::Channel *)), m_numericalColumns, SLOT(onCurrentChannelChanged(FunctionTreeModel::Channel *)));

	assert(ret);

	m_functionGraph->hide();
}

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

FunctionViewer::~FunctionViewer()
{
	delete m_selection;
	m_toolbar->setFrameHandle(0);
}

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

void FunctionViewer::showEvent(QShowEvent *)
{
	refreshModel();

	// Connect handles
	bool ret = true;

	if (m_xshHandle) {
		ret = connect(m_xshHandle, SIGNAL(xsheetChanged()), this, SLOT(refreshModel())) && ret;
		ret = connect(m_xshHandle, SIGNAL(xsheetSwitched()), this, SLOT(rebuildModel())) && ret;
	}

	if (m_frameHandle)
		ret = connect(m_frameHandle, SIGNAL(frameSwitched()), this, SLOT(propagateExternalSetFrame())) && ret;

	if (m_objectHandle) {
		ret = connect(m_objectHandle, SIGNAL(objectSwitched()), this, SLOT(onStageObjectSwitched())) && ret;
		ret = connect(m_objectHandle, SIGNAL(objectChanged(bool)), this, SLOT(onStageObjectChanged(bool))) && ret;
	}

	if (m_fxHandle)
		ret = connect(m_fxHandle, SIGNAL(fxSwitched()), this, SLOT(onFxSwitched())) && ret;

	//display animated channels when the scene is switched
	if (m_sceneHandle)
		ret = connect(m_sceneHandle, SIGNAL(sceneSwitched()), m_treeView, SLOT(displayAnimatedChannels())) && ret;

	assert(ret);

	m_treeView->setExpanded(m_treeView->model()->index(0, 0), true);
	m_treeView->setExpanded(m_treeView->model()->index(1, 0), true);

	FunctionTreeModel *ftm = static_cast<FunctionTreeModel *>(m_treeView->model());

	if (m_objectHandle) {
		assert(m_xshHandle);
		TXsheet *xsh = m_xshHandle->getXsheet();

		const TStageObjectId &objId = m_objectHandle->getObjectId();
		ftm->setCurrentStageObject((objId == TStageObjectId::NoneId) ? (TStageObject *)0 : xsh->getStageObject(objId));
	}

	if (m_fxHandle)
		ftm->setCurrentFx(m_fxHandle->getFx());
}

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

void FunctionViewer::hideEvent(QHideEvent *)
{
	if (m_xshHandle)
		m_xshHandle->disconnect(this);
	if (m_frameHandle)
		m_frameHandle->disconnect(this);
	if (m_objectHandle)
		m_objectHandle->disconnect(this);
	if (m_fxHandle)
		m_fxHandle->disconnect(this);
	if (m_sceneHandle)
		m_sceneHandle->disconnect(this);

	if (m_functionGraph->isVisible())
		m_functionGraph->hide();
}

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

void FunctionViewer::openContextMenu(TreeModel::Item *item, const QPoint &globalPos)
{
	m_treeView->openContextMenu(item, globalPos);
}

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

void FunctionViewer::rebuildModel()
{
	getSelection()->selectNone();
	onCurveSelected(0);
	m_functionGraph->getModel()->resetAll();
	refreshModel();
	m_treeView->displayAnimatedChannels();
}

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

void FunctionViewer::refreshModel()
{
	TXsheet *xsh = m_xshHandle ? m_xshHandle->getXsheet() : 0;

	m_functionGraph->getModel()->refreshData(xsh);

	if (xsh) {
		int rowCount = xsh->getFrameCount();
		m_numericalColumns->setRowCount(rowCount);
		m_numericalColumns->updateAll();

		ToonzScene *scene = xsh->getScene();
		if (!scene) // This seems wrong. It should rather be
			return; // asserted - though I'm not touching it now...

		TFilePath scenePath = scene->getScenePath().getParentDir();
		if (scene->isUntitled())
			scenePath = TProjectManager::instance()->getCurrentProject()->getScenesPath();

		m_treeView->setCurrentScenePath(scenePath);

		int distance, offset;
		scene->getProperties()->getMarkers(distance, offset);
		m_numericalColumns->setMarkRow(distance, offset);
	}

	m_treeView->updateAll();
	m_toolbar->setCurve(0);
}

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

void FunctionViewer::setXsheetHandle(TXsheetHandle *xshHandle)
{
	if (xshHandle == m_xshHandle)
		return;

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

	m_xshHandle = xshHandle;
	m_segmentViewer->setXsheetHandle(xshHandle);

	if (m_xshHandle && isVisible()) {
		TXsheet *xsh = m_xshHandle->getXsheet();
		m_functionGraph->getModel()->refreshData(xsh);

		bool ret = connect(m_xshHandle, SIGNAL(xsheetChanged), this, SLOT(refreshModel()));
		assert(ret);
	}
}

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

void FunctionViewer::setFrameHandle(TFrameHandle *frameHandle)
{
	if (m_frameHandle == frameHandle)
		return;

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

	m_frameHandle = frameHandle;

	if (m_frameHandle && isVisible()) {
		bool ret = connect(m_frameHandle, SIGNAL(frameSwitched()), this, SLOT(propagateExternalSetFrame()));
		assert(ret);
	}
}

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

void FunctionViewer::setObjectHandle(TObjectHandle *objectHandle)
{
	if (objectHandle == m_objectHandle)
		return;

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

	m_objectHandle = objectHandle;

	if (m_objectHandle && isVisible()) {
		m_treeView->updateAll();

		bool ret = true;
		ret = connect(m_objectHandle, SIGNAL(objectSwitched()), this, SLOT(onStageObjectSwitched())) && ret;
		ret = connect(m_objectHandle, SIGNAL(objectChanged(bool)), this, SLOT(onStageObjectChanged(bool))) && ret;

		assert(ret);
	}

	FunctionTreeModel *ftModel = static_cast<FunctionTreeModel *>(m_treeView->model());
	if (ftModel)
		ftModel->setObjectHandle(objectHandle);
}
//-----------------------------------------------------------------------------

void FunctionViewer::setFxHandle(TFxHandle *fxHandle)
{
	if (fxHandle == m_fxHandle)
		return;

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

	m_fxHandle = fxHandle;

	if (isVisible()) {
		m_treeView->updateAll();

		bool ret = connect(m_fxHandle, SIGNAL(fxSwitched()), this, SLOT(onFxSwitched()));
		assert(ret);
	}

	FunctionTreeModel *ftModel = static_cast<FunctionTreeModel *>(m_treeView->model());
	if (ftModel)
		ftModel->setFxHandle(fxHandle);
}

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

void FunctionViewer::setColumnHandle(TColumnHandle *columnHandle)
{
	if (columnHandle == m_columnHandle)
		return;

	m_columnHandle = columnHandle;

	if (isVisible())
		m_treeView->updateAll();
}

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

void FunctionViewer::onFrameSwitched()
{
	int frame = m_localFrame.getFrame();
	m_segmentViewer->setSegmentByFrame(m_curve, frame);

	if (m_frameHandle)
		m_frameHandle->setFrame(frame);
}

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

void FunctionViewer::toggleMode()
{
	if (isHidden())
		return;

	if (m_functionGraph->isVisible())
		m_functionGraph->hide();
	else
		m_functionGraph->show();
}

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

void FunctionViewer::onCurveChanged(bool isDragging)
{
	if (m_objectHandle)
		m_objectHandle->notifyObjectIdChanged(isDragging);

	//emit signal if the current channel belongs to Fx in order to update the preview fx
	if (m_fxHandle) {
		FunctionTreeModel *ftModel =
			dynamic_cast<FunctionTreeModel *>(m_treeView->model());
		if (ftModel) {
			FunctionTreeModel::Channel *currChan = ftModel->getCurrentChannel();
			if (currChan) {
				//カレントチャンネルがFxChannelGroupに含まれていたらEmit
				FxChannelGroup *fxChanGroup =
					dynamic_cast<FxChannelGroup *>(currChan->getChannelGroup());
				if (fxChanGroup)
					m_fxHandle->notifyFxChanged();
			}
		}
	}
}

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

void FunctionViewer::onCurveSelected(TDoubleParam *curve)
{
	m_curve = curve;
	m_toolbar->setCurve(curve);

	QPair<TDoubleParam *, int> selectedSegment = getSelection()->getSelectedSegment();
	if (selectedSegment.first)
		m_segmentViewer->setSegment(selectedSegment.first, selectedSegment.second);
	else
		m_segmentViewer->setSegment(m_curve, -1);
}

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

void FunctionViewer::onValueFieldChanged()
{
}

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

void FunctionViewer::onXsheetChanged()
{
	TXsheet *xsh = m_xshHandle->getXsheet();
	int rowCount = xsh->getFrameCount();
	m_numericalColumns->setRowCount(rowCount);
}

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

void FunctionViewer::onStageObjectSwitched()
{
	TXsheet *xsh = m_xshHandle->getXsheet();

	const TStageObjectId &objId = m_objectHandle->getObjectId();
	TStageObject *obj = (objId == TStageObjectId::NoneId) ? (TStageObject *)0 : xsh->getStageObject(objId);

	static_cast<FunctionTreeModel *>(m_treeView->model())->setCurrentStageObject(obj);

	m_treeView->updateAll();
	m_functionGraph->update();
}

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

void FunctionViewer::onStageObjectChanged(bool isDragging)
{
	TXsheet *xsh = m_xshHandle->getXsheet();

	const TStageObjectId &objId = m_objectHandle->getObjectId();
	TStageObject *obj = (objId == TStageObjectId::NoneId) ? (TStageObject *)0 : xsh->getStageObject(objId);

	static_cast<FunctionTreeModel *>(m_treeView->model())->setCurrentStageObject(obj);

	if (!isDragging)
		m_treeView->updateAll();

	m_functionGraph->update();
}

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

void FunctionViewer::onFxSwitched()
{
	TFx *fx = m_fxHandle->getFx();
	TZeraryColumnFx *zfx = dynamic_cast<TZeraryColumnFx *>(fx);
	if (zfx)
		fx = zfx->getZeraryFx();
	static_cast<FunctionTreeModel *>(m_treeView->model())->setCurrentFx(fx);
	m_treeView->updateAll();
	m_functionGraph->update();
}

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

void FunctionViewer::onSelectionChanged()
{
	QPair<TDoubleParam *, int> selectedSegment = getSelection()->getSelectedSegment();
	if (selectedSegment.first) {
		if (selectedSegment.first != m_curve) {
			m_curve = selectedSegment.first;
			m_toolbar->setCurve(selectedSegment.first);
		}
		m_segmentViewer->setSegment(selectedSegment.first, selectedSegment.second);
	} else {
		m_segmentViewer->setSegment(m_curve, -1);
	}
	// update curves
	if (m_functionGraph->isVisible() && !m_functionGraph->hasFocus())
		m_functionGraph->update();
}

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

void FunctionViewer::doSwitchCurrentObject(TStageObject *obj)
{
	TStageObjectId id = obj->getId();
	if (id.isColumn())
		m_columnHandle->setColumnIndex(id.getIndex());
	else
		m_objectHandle->setObjectId(id);
}

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

void FunctionViewer::doSwitchCurrentFx(TFx *fx)
{
	if (!fx) {
		m_fxHandle->setFx(0);
		return;
	}
	if (fx->isZerary()) {
		TXsheet *xsh = m_xshHandle->getXsheet();
		int i, columnCount = xsh->getColumnCount();
		for (i = 0; i < columnCount; i++) {
			TXshColumn *column = xsh->getColumn(i);
			TXshZeraryFxColumn *zfx = dynamic_cast<TXshZeraryFxColumn *>(xsh->getColumn(i));
			if (!zfx)
				continue;
			if (zfx->getZeraryColumnFx()->getZeraryFx() == fx) {
				fx = zfx->getZeraryColumnFx();
				m_columnHandle->setColumnIndex(i);
				break;
			}
		}
	}
	// Forbid update of the swatch upon column switch. This would generate
	// a useless render...
	SwatchViewer::suspendRendering(true, false);

	int columnIndex = fx->getReferenceColumnIndex();
	if (columnIndex >= 0)
		m_columnHandle->setColumnIndex(columnIndex);

	SwatchViewer::suspendRendering(false);
	m_fxHandle->setFx(fx);
	emit editObject();
}

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

void FunctionViewer::propagateExternalSetFrame()
{
	assert(m_frameHandle);
	m_localFrame.setFrame(m_frameHandle->getFrame());
}

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

void FunctionViewer::addParameter(
	TParam *parameter, const TFilePath &folder)
{
	static_cast<FunctionTreeModel *>(m_treeView->model())->addParameter(parameter, folder);
}

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

void FunctionViewer::setFocusColumnsOrGraph()
{
	m_numericalColumns->setFocus();
}

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

void FunctionViewer::clearFocusColumnsAndGraph()
{
	m_functionGraph->clearFocus();
	m_numericalColumns->clearFocus();
}

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

bool FunctionViewer::columnsOrGraphHasFocus()
{
	return m_functionGraph->hasFocus() || m_numericalColumns->anyWidgetHasFocus() || m_toolbar->anyWidgetHasFocus() || m_segmentViewer->anyWidgetHasFocus();
}

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

void FunctionViewer::setSceneHandle(TSceneHandle *sceneHandle)
{
	m_sceneHandle = sceneHandle;
}

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

bool FunctionViewer::isExpressionPageActive()
{
	return m_segmentViewer->isExpressionPageActive();
}