#include "toonz/scriptbinding_image_builder.h"

#include "toonz/scriptbinding_image.h"

#include "ttoonzimage.h"

#include "trop.h"

#include <cmath></cmath>

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

  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