KisWarpTransformWorker Class Reference

#include <kis_warptransform_worker.h>

Inheritance diagram for KisWarpTransformWorker:

Classes
struct	FunctionTransformOp

Public Types
typedef enum KisWarpTransformWorker::WarpCalculation_	WarpCalculation
enum	WarpCalculation_ { GRID = 0 , DRAW }
typedef enum KisWarpTransformWorker::WarpType_	WarpType
enum	WarpType_ { AFFINE_TRANSFORM = 0 , SIMILITUDE_TRANSFORM , RIGID_TRANSFORM , N_MODES }

Public Member Functions
QRect	approxChangeRect (const QRect &rc)
QRect	approxNeedRect (const QRect &rc, const QRect &fullBounds)
	KisWarpTransformWorker (WarpType warpType, QVector< QPointF > origPoint, QVector< QPointF > transfPoint, qreal alpha, KoUpdater *progress)
void	run (KisPaintDeviceSP srcDev, KisPaintDeviceSP dstDev)
	~KisWarpTransformWorker () override

Static Public Member Functions
static QPointF	affineTransformMath (QPointF v, QVector< QPointF > p, QVector< QPointF > q, qreal alpha)
static QPointF	rigidTransformMath (QPointF v, QVector< QPointF > p, QVector< QPointF > q, qreal alpha)
static QPointF	similitudeTransformMath (QPointF v, QVector< QPointF > p, QVector< QPointF > q, qreal alpha)
static QImage	transformQImage (WarpType warpType, const QVector< QPointF > &origPoint, const QVector< QPointF > &transfPoint, qreal alpha, const QImage &srcImage, const QPointF &srcQImageOffset, QPointF *newOffset)

Private Types
typedef QPointF(*	WarpMathFunction) (QPointF, QVector< QPointF >, QVector< QPointF >, qreal)

Private Attributes
qreal	m_alpha {1.0}
QVector< QPointF >	m_origPoint
KoUpdater *	m_progress {0}
QVector< QPointF >	m_transfPoint
WarpCalculation	m_warpCalc {GRID}
WarpMathFunction	m_warpMathFunction

Detailed Description

Class to apply a transformation (affine, similitude, MLS) to a paintDevice or a QImage according an original set of points p, a new set of points q, and the constant alpha. The algorithms are based a paper entitled "Image Deformation Using Moving Least Squares", by Scott Schaefer (Texas A&M University), Travis McPhail (Rice University) and Joe Warren (Rice University)

Definition at line 32 of file kis_warptransform_worker.h.

Member Typedef Documentation

WarpCalculation

typedef enum KisWarpTransformWorker::WarpCalculation_ KisWarpTransformWorker::WarpCalculation

WarpMathFunction

typedef QPointF(* KisWarpTransformWorker::WarpMathFunction) (QPointF, QVector< QPointF >, QVector< QPointF >, qreal)

private

Definition at line 63 of file kis_warptransform_worker.h.

WarpType

typedef enum KisWarpTransformWorker::WarpType_ KisWarpTransformWorker::WarpType

Member Enumeration Documentation

WarpCalculation_

enum KisWarpTransformWorker::WarpCalculation_

Enumerator
GRID
DRAW

Definition at line 38 of file kis_warptransform_worker.h.

{GRID = 0, DRAW} WarpCalculation;

WarpType_

enum KisWarpTransformWorker::WarpType_

Enumerator
AFFINE_TRANSFORM
SIMILITUDE_TRANSFORM
RIGID_TRANSFORM
N_MODES

Definition at line 37 of file kis_warptransform_worker.h.

{AFFINE_TRANSFORM = 0, SIMILITUDE_TRANSFORM, RIGID_TRANSFORM, N_MODES} WarpType;

Constructor & Destructor Documentation

KisWarpTransformWorker()

KisWarpTransformWorker::KisWarpTransformWorker	(	WarpType	warpType,
		QVector< QPointF >	origPoint,
		QVector< QPointF >	transfPoint,
		qreal	alpha,
		KoUpdater *	progress )

Definition at line 172 of file kis_warptransform_worker.cc.

        : m_progress(progress)
{
    m_origPoint = origPoint;
    m_transfPoint = transfPoint;
    m_alpha = alpha;
 
    switch(warpType) {
    case AFFINE_TRANSFORM:
        m_warpMathFunction = &affineTransformMath;
        break;
    case SIMILITUDE_TRANSFORM:
        m_warpMathFunction = &similitudeTransformMath;
        break;
    case RIGID_TRANSFORM:
        m_warpMathFunction = &rigidTransformMath;
        break;
    default:
        m_warpMathFunction = 0;
        break;
    }
}

References AFFINE_TRANSFORM, affineTransformMath(), m_alpha, m_origPoint, m_transfPoint, m_warpMathFunction, RIGID_TRANSFORM, rigidTransformMath(), SIMILITUDE_TRANSFORM, and similitudeTransformMath().

~KisWarpTransformWorker()

KisWarpTransformWorker::~KisWarpTransformWorker ( )

override

Definition at line 195 of file kis_warptransform_worker.cc.

{
}

Member Function Documentation

affineTransformMath()

QPointF KisWarpTransformWorker::affineTransformMath ( QPointF v, QVector< QPointF > p, QVector< QPointF > q, qreal alpha )

static

Definition at line 25 of file kis_warptransform_worker.cc.

{
    int nbPoints = p.size();
    QVarLengthArray<qreal> w(nbPoints);
    qreal sumWi = 0;
    QPointF pStar(0, 0), qStar(0, 0);
    QVarLengthArray<QPointF> pHat(nbPoints), qHat(nbPoints);
 
    for (int i = 0; i < nbPoints; ++i) {
        if (v == p[i])
            return q[i];
 
        QVector2D tmp(p[i] - v);
        w[i] = 1. / pow(tmp.lengthSquared(), alpha);
        pStar += w[i] * p[i];
        qStar += w[i] * q[i];
        sumWi += w[i];
    }
    pStar /= sumWi;
    qStar /= sumWi;
 
    qreal A_tmp[4] = {0, 0, 0, 0};
    for (int i = 0; i < nbPoints; ++i) {
        pHat[i] = p[i] - pStar;
        qHat[i] = q[i] - qStar;
 
        A_tmp[0] += w[i] * pow(pHat[i].x(), 2);
        A_tmp[3] += w[i] * pow(pHat[i].y(), 2);
        A_tmp[1] += w[i] * pHat[i].x() * pHat[i].y();
    }
    A_tmp[2] = A_tmp[1];
    qreal det_A_tmp = A_tmp[0] * A_tmp[3] - A_tmp[1] * A_tmp[2];
 
    qreal A_tmp_inv[4];
 
    if (det_A_tmp == 0)
        return v;
 
    A_tmp_inv[0] = A_tmp[3] / det_A_tmp;
    A_tmp_inv[1] = - A_tmp[1] / det_A_tmp;
    A_tmp_inv[2] = A_tmp_inv[1];
    A_tmp_inv[3] = A_tmp[0] / det_A_tmp;
 
    QPointF t = v - pStar;
    QPointF A_precalc(t.x() * A_tmp_inv[0] + t.y() * A_tmp_inv[1], t.x() * A_tmp_inv[2] + t.y() * A_tmp_inv[3]);
    qreal A_j;
 
    QPointF res = qStar;
    for (int j = 0; j < nbPoints; ++j) {
        A_j = A_precalc.x() * pHat[j].x() + A_precalc.y() * pHat[j].y();
 
        res += w[j] * A_j * qHat[j];
    }
 
    return res;
}

References p, and v.

approxChangeRect()

QRect KisWarpTransformWorker::approxChangeRect

(

const QRect &

rc

)

Definition at line 254 of file kis_warptransform_worker.cc.

{
    const qreal margin = 0.05;
 
    FunctionTransformOp functionOp(m_warpMathFunction, m_origPoint, m_transfPoint, m_alpha);
    QRect resultRect = KisAlgebra2D::approximateRectWithPointTransform(rc, functionOp);
 
    return KisAlgebra2D::blowRect(resultRect, margin);
}

References KisAlgebra2D::approximateRectWithPointTransform(), KisAlgebra2D::blowRect(), m_alpha, m_origPoint, m_transfPoint, and m_warpMathFunction.

approxNeedRect()

QRect KisWarpTransformWorker::approxNeedRect	(	const QRect &	rc,
		const QRect &	fullBounds )

Definition at line 264 of file kis_warptransform_worker.cc.

{
    Q_UNUSED(rc);
    return fullBounds;
}

rigidTransformMath()

QPointF KisWarpTransformWorker::rigidTransformMath ( QPointF v, QVector< QPointF > p, QVector< QPointF > q, qreal alpha )

static

Definition at line 129 of file kis_warptransform_worker.cc.

{
    int nbPoints = p.size();
    QVarLengthArray<qreal> w(nbPoints);
    qreal sumWi = 0;
    QPointF pStar(0, 0), qStar(0, 0);
    QVarLengthArray<QPointF> pHat(nbPoints), qHat(nbPoints);
 
    for (int i = 0; i < nbPoints; ++i) {
        if (v == p[i])
            return q[i];
 
        QVector2D tmp(p[i] - v);
        w[i] = 1. / pow(tmp.lengthSquared(), alpha);
        pStar += w[i] * p[i];
        qStar += w[i] * q[i];
        sumWi += w[i];
    }
    pStar /= sumWi;
    qStar /= sumWi;
 
    QVector2D res_tmp(0, 0);
    qreal qx, qy, px, py;
    for (int i = 0; i < nbPoints; ++i) {
        pHat[i] = p[i] - pStar;
        qHat[i] = q[i] - qStar;
 
        qx = w[i] * qHat[i].x();
        qy = w[i] * qHat[i].y();
        px = pHat[i].x();
        py = pHat[i].y();
 
        res_tmp += QVector2D(qx * px + qy * py, qx * py - qy * px);
    }
 
    QPointF f_arrow(res_tmp.normalized().toPointF());
    QVector2D v_m_pStar(v - pStar);
    QPointF res(f_arrow.x() * v_m_pStar.x() + f_arrow.y() * v_m_pStar.y(), f_arrow.x() * v_m_pStar.y() - f_arrow.y() * v_m_pStar.x());
    res += qStar;
 
    return res;
}

References p, and v.

run()

void KisWarpTransformWorker::run	(	KisPaintDeviceSP	srcDev,
		KisPaintDeviceSP	dstDev )

Definition at line 222 of file kis_warptransform_worker.cc.

{
    KIS_SAFE_ASSERT_RECOVER_RETURN(*srcDev->colorSpace() == *dstDev->colorSpace());
 
    if (!m_warpMathFunction ||
        m_origPoint.isEmpty() ||
        m_origPoint.size() != m_transfPoint.size()) {
 
        return;
    }
 
    if (m_origPoint.size() == 1) {
        dstDev->makeCloneFromRough(srcDev, srcDev->extent());
        QPointF translate(QPointF(srcDev->x(), srcDev->y()) + m_transfPoint[0] - m_origPoint[0]);
        dstDev->moveTo(translate.toPoint());
        return;
    }
 
    const QRect srcBounds = srcDev->region().boundingRect();
 
    dstDev->clear();
 
    const int pixelPrecision = 8;
 
    FunctionTransformOp functionOp(m_warpMathFunction, m_origPoint, m_transfPoint, m_alpha);
    GridIterationTools::PaintDevicePolygonOp polygonOp(srcDev, dstDev);
    GridIterationTools::processGrid(polygonOp, functionOp,
                                    srcBounds, pixelPrecision);
}

References KisRegion::boundingRect(), KisPaintDevice::clear(), KisPaintDevice::colorSpace(), KisPaintDevice::extent(), KIS_SAFE_ASSERT_RECOVER_RETURN, m_alpha, m_origPoint, m_transfPoint, m_warpMathFunction, KisPaintDevice::makeCloneFromRough(), KisPaintDevice::moveTo(), GridIterationTools::processGrid(), KisPaintDevice::region(), KisPaintDevice::x(), and KisPaintDevice::y().

similitudeTransformMath()

QPointF KisWarpTransformWorker::similitudeTransformMath ( QPointF v, QVector< QPointF > p, QVector< QPointF > q, qreal alpha )

static

Definition at line 82 of file kis_warptransform_worker.cc.

{
    int nbPoints = p.size();
    QVarLengthArray<qreal> w(nbPoints);
    qreal sumWi = 0;
    QPointF pStar(0, 0), qStar(0, 0);
    QVarLengthArray<QPointF> pHat(nbPoints), qHat(nbPoints);
 
    for (int i = 0; i < nbPoints; ++i) {
        if (v == p[i])
            return q[i];
 
        QVector2D tmp(p[i] - v);
        w[i] = 1. / pow(tmp.lengthSquared(), alpha);
        pStar += w[i] * p[i];
        qStar += w[i] * q[i];
        sumWi += w[i];
    }
    pStar /= sumWi;
    qStar /= sumWi;
 
    qreal mu_s = 0;
    QPointF res_tmp(0, 0);
    qreal qx, qy, px, py;
    for (int i = 0; i < nbPoints; ++i) {
        pHat[i] = p[i] - pStar;
        qHat[i] = q[i] - qStar;
 
        QVector2D tmp(pHat[i]);
        mu_s += w[i] * tmp.lengthSquared();
 
        qx = w[i] * qHat[i].x();
        qy = w[i] * qHat[i].y();
        px = pHat[i].x();
        py = pHat[i].y();
 
        res_tmp += QPointF(qx * px + qy * py, qx * py - qy * px);
    }
 
    res_tmp /= mu_s;
    QPointF v_m_pStar(v - pStar);
    QPointF res(res_tmp.x() * v_m_pStar.x() + res_tmp.y() * v_m_pStar.y(), res_tmp.x() * v_m_pStar.y() - res_tmp.y() * v_m_pStar.x());
    res += qStar;
 
    return res;
}

References p, and v.

transformQImage()

QImage KisWarpTransformWorker::transformQImage ( WarpType warpType, const QVector< QPointF > & origPoint, const QVector< QPointF > & transfPoint, qreal alpha, const QImage & srcImage, const QPointF & srcQImageOffset, QPointF * newOffset )

static

Definition at line 270 of file kis_warptransform_worker.cc.

{
    KIS_ASSERT_RECOVER(srcImage.format() == QImage::Format_ARGB32) {
        return QImage();
    }
 
    WarpMathFunction warpMathFunction = &rigidTransformMath;
 
    switch (warpType) {
    case AFFINE_TRANSFORM:
        warpMathFunction = &affineTransformMath;
        break;
    case SIMILITUDE_TRANSFORM:
        warpMathFunction = &similitudeTransformMath;
        break;
    case RIGID_TRANSFORM:
        warpMathFunction = &rigidTransformMath;
        break;
    default:
        KIS_ASSERT_RECOVER(0 && "Unknown warp mode") { return QImage(); }
    }
 
    if (!warpMathFunction ||
        origPoint.isEmpty() ||
        origPoint.size() != transfPoint.size()) {
 
        return srcImage;
    }
 
    if (origPoint.size() == 1) {
        *newOffset = srcQImageOffset + (transfPoint[0] - origPoint[0]).toPoint();
        return srcImage;
    }
 
    FunctionTransformOp functionOp(warpMathFunction, origPoint, transfPoint, alpha);
 
    const QRectF srcBounds = QRectF(srcQImageOffset, srcImage.size());
    QRectF dstBounds;
 
    {
        QPolygonF testPoints;
        testPoints << srcBounds.topLeft();
        testPoints << srcBounds.topRight();
        testPoints << srcBounds.bottomRight();
        testPoints << srcBounds.bottomLeft();
        testPoints << srcBounds.topLeft();
 
        QPolygonF::iterator it = testPoints.begin() + 1;
 
        while (it != testPoints.end()) {
            it = testPoints.insert(it, 0.5 * (*it + *(it - 1)));
            it += 2;
        }
 
        it = testPoints.begin();
 
        while (it != testPoints.end()) {
            *it = functionOp(*it);
            ++it;
        }
 
        dstBounds = testPoints.boundingRect();
    }
 
    QPointF dstQImageOffset = dstBounds.topLeft();
    *newOffset = dstQImageOffset;
 
    QRect dstBoundsI = dstBounds.toAlignedRect();
    QImage dstImage(dstBoundsI.size(), srcImage.format());
    dstImage.fill(0);
 
    const int pixelPrecision = 32;
    GridIterationTools::QImagePolygonOp polygonOp(srcImage, dstImage, srcQImageOffset, dstQImageOffset);
    GridIterationTools::processGrid(polygonOp, functionOp, srcBounds.toAlignedRect(), pixelPrecision);
 
    return dstImage;
}

References AFFINE_TRANSFORM, affineTransformMath(), KIS_ASSERT_RECOVER, GridIterationTools::processGrid(), RIGID_TRANSFORM, rigidTransformMath(), SIMILITUDE_TRANSFORM, and similitudeTransformMath().

Member Data Documentation

m_alpha

qreal KisWarpTransformWorker::m_alpha {1.0}

private

Definition at line 70 of file kis_warptransform_worker.h.

{1.0};

m_origPoint

QVector<QPointF> KisWarpTransformWorker::m_origPoint

private

Definition at line 68 of file kis_warptransform_worker.h.

m_progress

KoUpdater* KisWarpTransformWorker::m_progress {0}

private

Definition at line 71 of file kis_warptransform_worker.h.

{0};

m_transfPoint

QVector<QPointF> KisWarpTransformWorker::m_transfPoint

private

Definition at line 69 of file kis_warptransform_worker.h.

m_warpCalc

WarpCalculation KisWarpTransformWorker::m_warpCalc {GRID}

private

Definition at line 67 of file kis_warptransform_worker.h.

{GRID};

m_warpMathFunction

WarpMathFunction KisWarpTransformWorker::m_warpMathFunction

private

Definition at line 66 of file kis_warptransform_worker.h.

---

The documentation for this class was generated from the following files:

• libs/image/kis_warptransform_worker.h
• libs/image/kis_warptransform_worker.cc