#include "toonz/scriptbinding_image.h"

#include "toonz/scriptbinding_level.h"

#include "toonz/scriptbinding_files.h"

#include "tsystem.h"

#include "ttoonzimage.h"

#include "tfiletype.h"

#include "timage_io.h"

#include "tlevel_io.h"

namespace TScriptBinding

{

Image::Image()

{

}

Image::Image(const TImageP img)

	: m_img(img)

{

}

Image::Image(TImage *img)

	: m_img(img)

{

}

Image::~Image()

{

}

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

{

	Image *img = new Image();

	QScriptValue obj = create(engine, img);

	QScriptValue err = checkArgumentCount(context, "the Image constructor", 0, 1);

	if (err.isError())

		return err;

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

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

	}

	return obj;

}

QScriptValue Image::toString()

{

	if (m_img) {

		TImage::Type type = m_img->getType();

		if (type == TImage::RASTER)

			return QString("Raster image ( %1 x %2 )").arg(getWidth()).arg(getHeight());

		else if (type == TImage::TOONZ_RASTER)

			return QString("Toonz raster image ( %1 x %2 )").arg(getWidth()).arg(getHeight());

		else if (type == TImage::VECTOR)

			return QString("Vector image");

		else

			return QString("Image");

	} else {

		return "Empty image";

	}

}

int Image::getWidth()

{

	return !!m_img && !!m_img->raster() ? m_img->raster()->getSize().lx : 0;

}

int Image::getHeight()

{

	return !!m_img && !!m_img->raster() ? m_img->raster()->getSize().ly : 0;

}

double Image::getDpi()

{

	if (TRasterImageP ri = m_img) {

		double dpix = 0, dpiy = 0;

		ri->getDpi(dpix, dpiy);

		return dpix;

	} else if (TToonzImageP ti = m_img) {

		double dpix = 0, dpiy = 0;

		ti->getDpi(dpix, dpiy);

		return dpix;

	} else

		return 0;

}

QString Image::getType() const

{

	if (m_img) {

		TImage::Type type = m_img->getType();

		if (type == TImage::RASTER)

			return "Raster";

		else if (type == TImage::TOONZ_RASTER)

			return "ToonzRaster";

		else if (type == TImage::VECTOR)

			return "Vector";

		else

			return "Unknown";

	} else

		return "Empty";

}

QScriptValue Image::load(const QScriptValue &fpArg)

{

	// clear the old image (if any)

	m_img = TImageP();

	// get the path

	TFilePath fp;

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

	if (err.isError())

		return err;

	QString fpStr = fpArg.toString();

	try {

		// check if the file/level does exist

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

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

		}

		// the file could be a level

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

		if (TFileType::isLevel(fileType)) {

			// file is a level: read first frame

			TLevelReaderP lr(fp);

			TLevelP level = lr->loadInfo();

			int n = level->getFrameCount();

			if (n > 0) {

				// there are some frames

				TFrameId fid = fp.getFrame();

				if (fid == TFrameId::NO_FRAME || fid == TFrameId::EMPTY_FRAME)

					fid = level->begin()->first;

				m_img = lr->getFrameReader(fid)->load();

				if (!m_img) {

					return context()->throwError(QString("Could not read %1").arg(fpStr));

				}

				m_img->setPalette(level->getPalette());

				if (n > 1 && (fp.getFrame() == TFrameId::EMPTY_FRAME || fp.getFrame() == TFrameId::NO_FRAME)) {

					// warning: a multi-frame level read into an Image

					warning(tr("Loaded first frame of %1").arg(n));

				}

			} else {

				// level contains no frame (not sure it can even happen)

				return context()->throwError(QString("%1 contains no frames").arg(fpStr));

			}

		} else {

			// plain image: try to read it

			if (!TImageReader::load(fp, m_img)) {

				return context()->throwError(QString("File %1 not found or not readable").arg(fpStr));

			}

		}

		// return a reference to the Image object

		return context()->thisObject();

	} catch (...) {

		return context()->throwError(tr("Unexpected error while reading image").arg(fpStr));

	}

}

QScriptValue Image::save(const QScriptValue &fpArg)

{

	// clear the old image (if any)

	if (!m_img) {

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

	}

	// 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_img->getType() == TImage::RASTER)

			isCompatible = true;

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

		if (m_img->getType() == TImage::VECTOR)

			isCompatible = true;

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

		if (m_img->getType() == TImage::TOONZ_RASTER)

			isCompatible = true;

	} else {

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

	}

	if (!isCompatible) {

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

	}

	try {

		if (TFileType::isLevel(fileType)) {

			TLevelP level = new TLevel();

			level->setPalette(m_img->getPalette());

			level->setFrame(TFrameId(1), m_img);

			TLevelWriterP lw(fp);

			if (m_img->getPalette())

				lw->setPalette(m_img->getPalette());

			lw->save(level);

		} else {

			TImageWriterP iw(fp);

			iw->save(fp, m_img);

		}

		// return a reference to the Image object

		return context()->thisObject();

	} catch (...) {

		return context()->throwError(tr("Unexpected error while writing image").arg(fpStr));

	}

}

QScriptValue checkImage(QScriptContext *context, const QScriptValue &value, Image *&img)

{

	img = qscriptvalue_cast(value);

	if (!img || !img->getImg())

		return context->throwError(QObject::tr("Bad argument (%1): should be an Image (not empty)").arg(value.toString()));

	else

		return QScriptValue();

}

} // namespace TScriptBinding