303 lines
8.4 KiB
C++
303 lines
8.4 KiB
C++
|
|
|
|
#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
|