#include "toonz/scriptbinding_image_builder.h"
#include "toonz/scriptbinding_image.h"
#include "ttoonzimage.h"
#include "trop.h"
#define _USE_MATH_DEFINES
#include <math.h>

namespace TScriptBinding
{

Transform::Transform()
{
}

Transform::Transform(const TAffine &aff)
	: m_affine(aff)
{
}

Transform::~Transform()
{
}

QScriptValue Transform::ctor(QScriptContext *context, QScriptEngine *engine)
{
	return create(engine, new Transform());
}

QScriptValue Transform::toString()
{
	if (m_affine.isIdentity())
		return tr("Identity");
	else if (m_affine.isTranslation())
		return tr("Translation(%1,%2)").arg(m_affine.a13).arg(m_affine.a23);
	else {
		QString translationPart = "";
		if (m_affine.a13 != 0.0 || m_affine.a23 != 0.0)
			translationPart = tr("Translation(%1,%2)").arg(m_affine.a13).arg(m_affine.a23);

		if (fabs(m_affine.det() - 1.0) < 1.0e-8) {
			double phi = atan2(m_affine.a12, m_affine.a11) * 180.0 / M_PI;
			phi = -(0.001 * floor(1000 * phi + 0.5));
			QString rotationPart = tr("Rotation(%1)").arg(phi);
			if (translationPart == "")
				return rotationPart;
			else
				return translationPart + "*" + rotationPart;
		} else if (m_affine.a12 == 0.0 && m_affine.a21 == 0.0) {
			QString scalePart;
			if (m_affine.a11 == m_affine.a22)
				scalePart = tr("Scale(%1%)").arg(m_affine.a11 * 100);
			else
				scalePart = tr("Scale(%1%, %2%)").arg(m_affine.a11 * 100).arg(m_affine.a22 * 100);
			if (translationPart == "")
				return scalePart;
			else
				return translationPart + "*" + scalePart;
		} else {
			return tr("Transform(%1, %2, %3;  %4, %5, %6)")
				.arg(m_affine.a11)
				.arg(m_affine.a12)
				.arg(m_affine.a13)
				.arg(m_affine.a21)
				.arg(m_affine.a22)
				.arg(m_affine.a23);
		}
	}
}

QScriptValue Transform::translate(double x, double y)
{
	return create(engine(), new Transform(TTranslation(x, y) * m_affine));
}

QScriptValue Transform::rotate(double degrees)
{
	return create(engine(), new Transform(TRotation(degrees) * m_affine));
}

QScriptValue Transform::scale(double s)
{
	return create(engine(), new Transform(TScale(s) * m_affine));
}

QScriptValue Transform::scale(double sx, double sy)
{
	return create(engine(), new Transform(TScale(sx, sy) * m_affine));
}

//=============================================================================

ImageBuilder::ImageBuilder()
	: m_width(0), m_height(0)
{
}

ImageBuilder::~ImageBuilder()
{
}

QScriptValue ImageBuilder::ctor(QScriptContext *context, QScriptEngine *engine)
{
	ImageBuilder *imageBuilder = 0;
	if (context->argumentCount() == 2 || context->argumentCount() == 3) {
		if (!context->argument(0).isNumber() || !context->argument(1).isNumber())
			return context->throwError("Bad arguments: expected width,height[,type]");
		int width = (int)(context->argument(0).toNumber());
		int height = (int)(context->argument(1).toNumber());
		if (width <= 0 || height <= 0)
			return context->throwError("Bad size");
		QString type;
		if (context->argumentCount() == 3) {
			if (context->argument(2).isString())
				type = context->argument(2).toString();
			if (type != "Raster" && type != "ToonzRaster")
				return context->throwError(tr("Bad argument (%1): should be 'Raster' or ToonzRaster'").arg(context->argument(2).toString()));
		}
		imageBuilder = new ImageBuilder();
		imageBuilder->m_width = width;
		imageBuilder->m_height = height;
		if (type == "Raster")
			imageBuilder->m_img = new TRasterImage(TRaster32P(width, height));
		else if (type == "ToonzRaster") {
			imageBuilder->m_img = new TToonzImage(TRasterCM32P(width, height), TRect(0, 0, width, height));
		}
	} else {
		if (context->argumentCount() != 0)
			return context->throwError("Bad argument count. expected: width,height[,type]");
		imageBuilder = new ImageBuilder();
	}
	QScriptValue obj = engine->newQObject(imageBuilder, QScriptEngine::AutoOwnership, QScriptEngine::ExcludeSuperClassContents | QScriptEngine::ExcludeSuperClassMethods);
	return obj;
}

QScriptValue ImageBuilder::toString()
{
	QString type = "Empty";
	if (m_img) {
		if (m_img->getType() == TImage::RASTER)
			type = "Raster";
		else if (m_img->getType() == TImage::TOONZ_RASTER)
			type = "ToonzRaster";
		else if (m_img->getType() == TImage::VECTOR)
			type = "Vector";
		else
			type = "Bad";
	}
	return tr("ImageBuilder(%1 image)").arg(type);
}

QScriptValue ImageBuilder::getImage()
{
	return create(engine(), new Image(m_img));
}

QScriptValue ImageBuilder::fill(const QString &colorName)
{
	QColor color;
	QScriptValue err = checkColor(context(), colorName, color);
	if (err.isError())
		return err;
	TPixel32 pix(color.red(), color.green(), color.blue(), color.alpha());
	if (m_img) {
		if (m_img->getType() != TImage::RASTER)
			context()->throwError("Can't fill a non-'Raster' image");
		TRaster32P ras = m_img->raster();
		if (ras)
			ras->fill(pix);
	} else if (m_width > 0 && m_height > 0) {
		TRaster32P ras(m_width, m_height);
		ras->fill(pix);
		m_img = TRasterImageP(ras);
	}
	return context()->thisObject();
}

void ImageBuilder::clear()
{
	m_img = TImageP();
}

QString ImageBuilder::add(const TImageP &img, const TAffine &aff)
{
	if (fabs(aff.det()) < 0.001)
		return "";
	if (m_img && (m_img->getType() != img->getType()))
		return "Image type mismatch";

	if (!m_img && img->getType() != TImage::VECTOR && m_width > 0 && m_height > 0) {
		TRasterP ras = img->raster()->create(m_width, m_height);
		if (img->getType() == TImage::RASTER)
			m_img = TRasterImageP(ras);
		else if (img->getType() == TImage::TOONZ_RASTER) {
			m_img = TToonzImageP(ras, ras->getBounds());
			m_img->setPalette(img->getPalette());
		} else
			return "Bad image type";
	}

	if (!m_img && aff.isIdentity()) {
		m_img = img->cloneImage();
	} else if (img->getType() == TImage::VECTOR) {
		// vector image
		if (!m_img) {
			// transform
			TVectorImageP vi = img->cloneImage();
			vi->transform(aff);
			m_img = vi;
		} else {
			// merge
			TVectorImageP up = img;
			TVectorImageP dn = m_img;
			dn->mergeImage(up, aff);
		}
	} else {
		if (img->getType() != TImage::TOONZ_RASTER && img->getType() != TImage::RASTER) {
			// this should not ever happen
			return "Bad image type";
		}
		TRasterP up = img->raster();
		if (!m_img) {
			// create an empty bg
			TRasterP ras = up->create();
			ras->clear();
			if (img->getType() == TImage::TOONZ_RASTER) {
				TRasterCM32P rasCm = ras;
				m_img = TToonzImageP(rasCm, TRect(0, 0, ras->getLx() - 1, ras->getLy() - 1));
				m_img->setPalette(img->getPalette());
			} else {
				m_img = TRasterImageP(ras);
			}
		}
		TRasterP dn = m_img->raster();
		if (aff.isTranslation() && aff.a13 == floor(aff.a13) && aff.a23 == floor(aff.a23)) {
			// just a integer coord translation
			TPoint delta = -up->getCenter() + dn->getCenter() + TPoint((int)aff.a13, (int)aff.a23);
			TRop::over(dn, up, delta);
		} else {
			TAffine aff1 = TTranslation(dn->getCenterD()) * aff * TTranslation(-up->getCenterD());
			TRop::over(dn, up, aff1, TRop::Mitchell);
		}
	}
	return "";
}

/*
  TImageP srcImg = img->getImg();
  if(srcImg->getType()==TImage::RASTER || srcImg->getType()==TImage::TOONZ_RASTER)
  {
    TRasterP in = srcImg->raster();
    TRasterP out = in->create();
    TPointD center = in->getCenterD();
    TAffine aff1 = TTranslation(center) * aff * TTranslation(-center);
    TRop::resample(out,in,aff1,TRop::Mitchell);
    m_img = TRasterImageP(out);
  }
  else if(srcImg->getType()==TImage::VECTOR)
  {
    TVectorImageP vi = srcImg->cloneImage();
    vi->transform(aff);
    m_img = vi;
  }
  else
  {
    return context()->throwError(tr("Bad image type"));
  }
  return context()->thisObject();
  */

QScriptValue ImageBuilder::add(QScriptValue imgArg)
{
	Image *simg = 0;
	QScriptValue err = checkImage(context(), imgArg, simg);
	if (err.isError())
		return err;
	QString errStr = add(simg->getImg(), TAffine());
	if (errStr != "")
		return context()->throwError(tr("%1 : %2").arg(errStr).arg(imgArg.toString()));
	else
		return context()->thisObject();
}

QScriptValue ImageBuilder::add(QScriptValue imgArg, QScriptValue transformationArg)
{
	Image *simg = 0;
	QScriptValue err = checkImage(context(), imgArg, simg);
	if (err.isError())
		return err;
	Transform *transformation = qscriptvalue_cast<Transform *>(transformationArg);
	if (!transformation) {
		return context()->throwError(tr("Bad argument (%1): should be a Transformation").arg(transformationArg.toString()));
	}
	TAffine aff = transformation->getAffine();
	QString errStr = add(simg->getImg(), aff);
	if (errStr != "")
		return context()->throwError(tr("%1 : %2").arg(errStr).arg(imgArg.toString()));
	else
		return context()->thisObject();
}

} // namespace TScriptBinding