#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;</tthickpoint>

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

}