#include "toonz/tobjecthandle.h"

#include "toonz/tstageobjectid.h"
#include "toonz/tstageobject.h"
#include "toonz/tstageobjectspline.h"
#include "toonz/txsheet.h"
#include "tvectorimage.h"
#include "tstroke.h"

namespace
{

void clearImage(TVectorImage *img)
{
	while (img->getStrokeCount())
		img->deleteStroke(0);
}

// obj -> image
void getSpline(TVectorImage *dstImg, TStageObjectId objId, TStageObjectSpline *currentSpline)
{
	QMutexLocker lock(dstImg->getMutex());
	clearImage(dstImg);
	if (currentSpline)
		dstImg->addStroke(new TStroke(*currentSpline->getStroke()));
}

// image -> obj
void setSpline(TVectorImage *srcImg, TStageObjectId objId, TStageObjectSpline *currentSpline)
{
	if (!currentSpline)
		return;
	if (srcImg->getStrokeCount() == 0) {
		double d = 30;
		std::vector<TThickPoint> points;
		points.push_back(TPointD(-d, 0));
		points.push_back(TPointD(0, 0));
		points.push_back(TPointD(d, 0));
		srcImg->addStroke(new TStroke(points), false);
	}
	TStroke *stroke = srcImg->getStroke(0);
	currentSpline->setStroke(new TStroke(*stroke));
}

} // namespace

//=============================================================================
// TXshLevelHandle
//-----------------------------------------------------------------------------

TObjectHandle::TObjectHandle()
	: m_objectId(TStageObjectId::NoneId), m_isSpline(false), m_splineImage(0), m_currentSpline(0)
{
	m_splineImage = new TVectorImage();
	m_splineImage->addRef();
}

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

TObjectHandle::~TObjectHandle()
{
	m_splineImage->release();
}

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

TStageObjectId TObjectHandle::getObjectId()
{
	return m_objectId;
}

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

void TObjectHandle::setObjectId(TStageObjectId objectId)
{
	if (m_objectId != objectId) {
		m_objectId = objectId;
		m_isSpline = false;
		clearImage(m_splineImage);
		emit objectSwitched();
	}
}

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

void TObjectHandle::setIsSpline(bool isSpline)
{
	if (m_isSpline != isSpline) {
		m_isSpline = isSpline;
		emit objectSwitched();
	}
}

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

void TObjectHandle::commitSplineChanges()
{
	setSpline(m_splineImage, m_objectId, m_currentSpline);
	emit splineChanged();
}

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

void TObjectHandle::setSplineObject(TStageObjectSpline *splineObject)
{
	m_currentSpline = splineObject;
	if (m_isSpline && m_currentSpline)
		getSpline(m_splineImage, m_objectId, m_currentSpline);
}