#include "toonz/scriptbinding_level.h"

#include "toonz/scriptbinding_files.h"

#include <qscriptengine></qscriptengine>

#include "timage_io.h"

#include "tlevel_io.h"

#include "tlevel.h"

#include "toonz/tcenterlinevectorizer.h"

#include "toonz/tcamera.h"

#include "trop.h"

#include "trasterimage.h"

#include "ttoonzimage.h"

#include "tvectorimage.h"

#include "tpalette.h"

#include "tofflinegl.h"

#include "tvectorrenderdata.h"

#include "tgeometry.h"

#include "toonz/stage.h"

#include "toonz/tcamera.h"

#include "toonz/txshleveltypes.h"

#include "toonz/levelproperties.h"

#include "toonz/toonzscene.h"

#include "toonz/txshsimplelevel.h"

#include "toonz/levelset.h"

#include "tfiletype.h"

#include "tsystem.h"

#include <qregexp></qregexp>

#include <qcolor></qcolor>

#include "tfiletype.h"

namespace TScriptBinding

{

Level::Level()

	: m_sl(0), m_type(NO_XSHLEVEL), m_scene(new ToonzScene()), m_sceneOwner(true)

{

}

Level::Level(TXshSimpleLevel *sl)

	: m_sl(sl), m_type(sl->getType()), m_scene(sl->getScene()), m_sceneOwner(false)

{

	sl->addRef();

}

Level::~Level()

{

	if (m_sceneOwner)

		delete m_scene;

	if (m_sl)

		m_sl->release();

}

QScriptValue Level::ctor(QScriptContext *context, QScriptEngine *engine)

{

	Level *level = new Level();

	QScriptValue obj = engine->newQObject(level, QScriptEngine::AutoOwnership);

	if (context->argumentCount() == 1) {

		return obj.property("load").call(obj, context->argumentsObject());

	}

	return obj;

}

QScriptValue Level::toString()

{

	QString info = "(";

	QString comma = "";

	if (getName() != "") {

		info.append(comma).append(getName());

		comma = ", ";

	}

	info.append(comma).append(tr("%1 frames").arg(getFrameCount()));

	info.append(")");

	if (m_type == PLI_XSHLEVEL)

		return QString("Vector level %1").arg(info);

	else if (m_type == TZP_XSHLEVEL)

		return QString("Toonz level %1").arg(info);

	else if (m_type == NO_XSHLEVEL)

		return QString("Empty level");

	else if (m_type == OVL_XSHLEVEL)

		return QString("Raster level %1").arg(info);

	else

		return QString("Level %1").arg(info);

}

QString Level::getType() const

{

	if (m_type == NO_XSHLEVEL)

		return "Empty";

	else if (m_type == PLI_XSHLEVEL)

		return "Vector";

	else if (m_type == TZP_XSHLEVEL)

		return "ToonzRaster";

	else if (m_type == OVL_XSHLEVEL)

		return "Raster";

	else

		return "Unknown";

}

int Level::getFrameCount() const

{

	return m_sl ? m_sl->getFrameCount() : 0;

}

QString Level::getName() const

{

	return m_sl ? QString::fromStdWString(m_sl->getName()) : "";

}

void Level::setName(const QString &name)

{

	if (m_sl)

		m_sl->setName(name.toStdWString());

}

QScriptValue Level::getPath() const

{

	if (m_sl) {

		FilePath *result = new FilePath(m_sl->getPath());

		return result->create<filepath>(engine());</filepath>

	} else

		return QScriptValue();

	return m_sl ? QString::fromStdWString(m_sl->getName()) : "";

}

void Level::setPath(const QScriptValue &pathArg)

{

	TFilePath fp;

	FilePath *filePath = qscriptvalue_cast<filepath *="">(pathArg);</filepath>

	if (filePath)

		fp = filePath->getToonzFilePath();

	else if (pathArg.isString())

		fp = TFilePath(pathArg.toString().toStdString());

	else

		context()->throwError(tr("Bad argument (%1). It should be FilePath or string").arg(pathArg.toString()));

	if (m_sl) {

		m_sl->setPath(fp);

		try {

			m_sl->load();

		} catch (...) {

			context()->throwError(tr("Exception loading level (%1)").arg(QString::fromStdWString(fp.getWideString())));

		}

	}

}

QScriptValue Level::load(const QScriptValue &fpArg)

{

	if (m_sl) {

		m_scene->getLevelSet()->removeLevel(m_sl, true);

		m_sl->release();

		m_sl = 0;

	}

	// get the path

	TFilePath fp;

	QScriptValue err = checkFilePath(context(), fpArg, fp);

	if (err.isError())

		return err;

	QString fpStr = fpArg.toString();

	try {

		if (!TSystem::doesExistFileOrLevel(fp)) {

			return context()->throwError(tr("File %1 doesn't exist").arg(fpStr));

		}

		TFileType::Type fileType = TFileType::getInfo(fp);

		if (TFileType::isVector(fileType))

			m_type = PLI_XSHLEVEL;

		else if (0 != (fileType & TFileType::CMAPPED_IMAGE))

			m_type = TZP_XSHLEVEL;

		else if (0 != (fileType & TFileType::RASTER_IMAGE))

			m_type = OVL_XSHLEVEL;

		else {

			return context()->throwError(tr("File %1 is unsupported").arg(fpStr));

		}

		TXshLevel *xl = m_scene->loadLevel(fp);

		if (xl) {

			m_sl = xl->getSimpleLevel();

			m_sl->addRef();

		}

		return context()->thisObject();

	} catch (...) {

		return context()->throwError(tr("Exception reading %1").arg(fpStr));

	}

}

QScriptValue Level::save(const QScriptValue &fpArg)

{

	if (getFrameCount() == 0) {

		return context()->throwError(tr("Can't save an empty level"));

	}

	// get the path

	TFilePath fp;

	QScriptValue err = checkFilePath(context(), fpArg, fp);

	if (err.isError())

		return err;

	QString fpStr = fpArg.toString();

	// handle conversion (if it is needed and possible)

	TFileType::Type fileType = TFileType::getInfo(fp);

	bool isCompatible = false;

	if (TFileType::isFullColor(fileType)) {

		if (m_sl->getType() == OVL_XSHLEVEL)

			isCompatible = true;

	} else if (TFileType::isVector(fileType)) {

		if (m_sl->getType() == PLI_XSHLEVEL)

			isCompatible = true;

	} else if (fileType & TFileType::CMAPPED_IMAGE) {

		if (m_sl->getType() == TZP_XSHLEVEL)

			isCompatible = true;

	} else {

		return context()->throwError(tr("Unrecognized file type :").arg(fpStr));

	}

	if (!isCompatible) {

		return context()->throwError(tr("Can't save a %1 level to this file type : %2").arg(getType()).arg(fpStr));

	}

	try {

		m_sl->save(fp);

	} catch (TSystemException se) {

		return context()->throwError(tr("Exception writing %1").arg(QString::fromStdWString(se.getMessage())));

	}

	return context()->thisObject();

}

TFrameId Level::getFid(const QScriptValue &arg, QString &err)

{

	if (arg.isNumber() || arg.isString()) {

		QString s = arg.toString();

		QRegExp re("(\\d+)(\\w?)");

		if (re.exactMatch(s)) {

			int d = re.cap(1).toInt();

			QString c = re.cap(2);

			TFrameId fid;

			if (c.length() == 1)

#if QT_VERSION >= 0x050500

				fid = TFrameId(d, c[0].unicode());

#else

				fid = TFrameId(d, c[0].toAscii());

#endif

			else

				fid = TFrameId(d);

			err = "";

			return fid;

		}

	}

	err = QObject::tr("Argument '%1' does not look like a FrameId").arg(arg.toString());

	return TFrameId();

}

TImageP Level::getImg(const TFrameId &fid)

{

	if (m_sl)

		return m_sl->getFrame(fid, false);

	else

		return TImageP();

}

QScriptValue Level::getFrame(const QScriptValue &fidArg)

{

	if (getFrameCount() == 0)

		return context()->throwError("An empty level has no frames");

	QString err;

	TFrameId fid = getFid(fidArg, err);

	if (err != "")

		return context()->throwError(err);

	TImageP content = m_sl->getFrame(fid, false);

	if (content) {

		Image *img = new Image(content.getPointer());

		return create(img);

	} else {

		return QScriptValue();

	}

}

QScriptValue Level::getFrameByIndex(const QScriptValue &indexArg)

{

	if (getFrameCount() == 0)

		return context()->throwError("An empty level has no frames");

	if (!indexArg.isNumber()) {

		return context()->throwError(tr("frame index (%1) must be a number").arg(indexArg.toString()));

	}

	int index = indexArg.toInteger();

	if (index < 0 || index >= getFrameCount()) {

		return context()->throwError(tr("frame index (%1) is out of range (0-%2)").arg(index).arg(getFrameCount() - 1));

	}

	TFrameId fid = m_sl->index2fid(index);

	TImageP content = m_sl->getFrame(fid, false);

	if (content) {

		Image *img = new Image(content.getPointer());

		return create(img);

	} else {

		return QScriptValue();

	}

}

// TODO: chiamare setFrame(const TFrameId &fid, const TImageP &img)

QScriptValue Level::setFrame(const QScriptValue &fidArg, const QScriptValue &imageArg)

{

	QString err;

	TFrameId fid = getFid(fidArg, err);

	if (err != "")

		return context()->throwError(err);

	Image *img = qscriptvalue_cast(imageArg);

	if (!img) {

		return context()->throwError(tr("second argument (%1) is not an image").arg(imageArg.toString()));

	}

	QString imgType = img->getType();

	int levelType = NO_XSHLEVEL;

	if (imgType == "ToonzRaster")

		levelType = TZP_XSHLEVEL;

	else if (imgType == "Raster")

		levelType = OVL_XSHLEVEL;

	else if (imgType == "Vector")

		levelType = PLI_XSHLEVEL;

	else {

		return context()->throwError(tr("can not insert a %1 image into a level").arg(imgType));

	}

	if (m_type == NO_XSHLEVEL) {

		m_type = levelType;

		TXshLevel *xl = m_scene->createNewLevel(levelType);

		m_sl = xl->getSimpleLevel();

		m_sl->addRef();

		m_sl->setPalette(img->getImg()->getPalette());

		if (levelType != PLI_XSHLEVEL) {

			LevelProperties *lprop = m_sl->getProperties();

			lprop->setDpiPolicy(LevelProperties::DP_ImageDpi);

			int xres = img->getWidth();

			int yres = img->getHeight();

			double dpi = img->getDpi();

			lprop->setDpi(dpi);

			lprop->setImageDpi(TPointD(dpi, dpi));

			lprop->setImageRes(TDimension(xres, yres));

			// lprop->setHasAlpha(true);

		}

	} else if (m_type != levelType) {

		return context()->throwError(tr("can not insert a %1 image to a %2 level").arg(imgType).arg(getType()));

	}

	if (m_sl->getFrameCount() == 0)

		m_sl->setPalette(img->getImg()->getPalette());

	m_sl->setFrame(fid, img->getImg());

	m_sl->setDirtyFlag(true);

	return context()->thisObject();

}

QScriptValue Level::getFrameIds()

{

	QList<tframeid> fids;</tframeid>

	getFrameIds(fids);

	QScriptValue result = engine()->newArray();

	quint32 index = 0;

	foreach (TFrameId fid, fids) {

		QString fidStr = QString::fromStdString(fid.expand());

		result.setProperty(index++, fidStr);

	}

	return result;

}

void Level::getFrameIds(QList<tframeid> &result)</tframeid>

{

	if (getFrameCount() > 0) {

		std::vector<tframeid> fids;</tframeid>

		m_sl->getFids(fids);

		for (std::vector<tframeid>::iterator it = fids.begin(); it != fids.end(); ++it) {</tframeid>

			result.append(*it);

		}

	}

}

int Level::setFrame(const TFrameId &fid, const TImageP &img)

{

	TImage::Type imgType = img->getType();

	int levelType = NO_XSHLEVEL;

	if (imgType == TImage::TOONZ_RASTER)

		levelType = TZP_XSHLEVEL;

	else if (imgType == TImage::RASTER)

		levelType = OVL_XSHLEVEL;

	else if (imgType == TImage::VECTOR)

		levelType = PLI_XSHLEVEL;

	else {

		return -1;

	}

	if (m_type == NO_XSHLEVEL) {

		m_type = levelType;

		TXshLevel *xl = m_scene->createNewLevel(levelType);

		m_sl = xl->getSimpleLevel();

		m_sl->addRef();

		m_sl->setPalette(img->getPalette());

		if (levelType != PLI_XSHLEVEL) {

			LevelProperties *lprop = m_sl->getProperties();

			lprop->setDpiPolicy(LevelProperties::DP_ImageDpi);

			int xres = 0, yres = 0;

			double dpix = 0, dpiy = 0;

			if (TRasterImageP ri = img) {

				if (ri->getRaster()) {

					TDimension size = ri->getRaster()->getSize();

					xres = size.lx;

					yres = size.ly;

					ri->getDpi(dpix, dpiy);

				}

			} else if (TToonzImageP ti = img) {

				if (ti->getRaster()) {

					TDimension size = ri->getRaster()->getSize();

					xres = size.lx;

					yres = size.ly;

					ri->getDpi(dpix, dpiy);

				}

			}

			lprop->setDpi(dpix);

			lprop->setImageDpi(TPointD(dpix, dpiy));

			lprop->setImageRes(TDimension(xres, yres));

			// lprop->setHasAlpha(true);

		}

	} else if (m_type != levelType) {

		return -2;

	}

	if (m_sl->getFrameCount() == 0)

		m_sl->setPalette(img->getPalette());

	m_sl->setFrame(fid, img);

	m_sl->setDirtyFlag(true);

	return 1;

}

} // namespace TScriptBinding