tahoma2d/toonz/sources/toonzlib/scriptbinding_image.cpp

#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<Image *>(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