1 files changed, 341 insertions, 0 deletions
diff --git a/chalk/plugins/paintops/defaultpaintops/kis_duplicateop.cc b/chalk/plugins/paintops/defaultpaintops/kis_duplicateop.cc
new file mode 100644
index 00000000..139ed51d
--- /dev/null
+++ b/chalk/plugins/paintops/defaultpaintops/kis_duplicateop.cc
@@ -0,0 +1,341 @@
+/*
+ *  Copyright (c) 2002 Patrick Julien <freak@codepimps.org>
+ *  Copyright (c) 2004 Boudewijn Rempt <boud@valdyas.org>
+ *  Copyright (c) 2004 Clarence Dang <dang@kde.org>
+ *  Copyright (c) 2004 Adrian Page <adrian@pagenet.plus.com>
+ *  Copyright (c) 2004-2006 Cyrille Berger <cberger@cberger.net>
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ */
+
+#include "kis_duplicateop.h"
+
+#include <tqrect.h>
+
+#include <kdebug.h>
+
+#include "kis_brush.h"
+#include "kis_global.h"
+#include "kis_paint_device.h"
+#include "kis_layer.h"
+#include "kis_painter.h"
+#include "kis_types.h"
+#include "kis_meta_registry.h"
+#include "kis_colorspace_factory_registry.h"
+#include "kis_paintop.h"
+#include "kis_iterators_pixel.h"
+#include "kis_selection.h"
+#include "kis_perspective_grid.h"
+#include "kis_random_sub_accessor.h"
+
+KisPaintOp * KisDuplicateOpFactory::createOp(const KisPaintOpSettings */*settings*/, KisPainter * painter)
+{
+    KisPaintOp * op = new KisDuplicateOp(painter);
+    Q_CHECK_PTR(op);
+    return op;
+}
+
+
+KisDuplicateOp::KisDuplicateOp(KisPainter * painter)
+    : super(painter)
+    , m_target(0)
+    , m_srcdev(0)
+{
+}
+
+KisDuplicateOp::~KisDuplicateOp()
+{
+}
+
+double KisDuplicateOp::minimizeEnergy(const double* m, double* sol, int w, int h)
+{
+    int rowstride = 3*w;
+    double err = 0;
+    memcpy(sol, m, 3* sizeof(double) * w);
+    m+=rowstride;
+    sol+=rowstride;
+    for ( int i = 1; i < h - 1; i++)
+    {
+        memcpy(sol, m, 3* sizeof(double));
+        m+=3; sol+=3;
+        for ( int j = 3; j < rowstride-3; j++)
+        {
+            double tmp = *sol;
+            *sol = ( ( *(m - 3 ) + *(m + 3) + *(m - rowstride ) + *(m + rowstride )) + 2 * *m ) /6;
+            double diff = *sol - tmp;
+            err += diff*diff;
+            m ++; sol ++;
+        }
+        memcpy(sol, m, 3* sizeof(double));
+        m+=3; sol+=3;
+}
+    memcpy(sol, m, 3* sizeof(double) * w);
+    return err;
+}
+
+
+void KisDuplicateOp::paintAt(const KisPoint &pos, const KisPaintInformation& info)
+{
+    if (!m_painter) return;
+
+    bool heal = m_painter->duplicateHealing();
+//     int healradius = m_painter->duplicateHealingRadius();
+
+    KisPaintDeviceSP device = m_painter->device();
+    if (m_source) device = m_source;
+    if (!device) return;
+
+    KisBrush * brush = m_painter->brush();
+    if (!brush) return;
+    if (! brush->canPaintFor(info) )
+        return;
+
+    KisPoint hotSpot = brush->hotSpot(info);
+    KisPoint pt = pos - hotSpot;
+
+    // Split the coordinates into integer plus fractional parts. The integer
+    // is where the dab will be positioned and the fractional part determines
+    // the sub-pixel positioning.
+    TQ_INT32 x;
+    double xFraction;
+    TQ_INT32 y;
+    double yFraction;
+
+    splitCoordinate(pt.x(), &x, &xFraction);
+    splitCoordinate(pt.y(), &y, &yFraction);
+    xFraction = yFraction = 0.0;
+
+    KisPaintDeviceSP dab = 0;
+
+    if (brush->brushType() == IMAGE ||
+        brush->brushType() == PIPE_IMAGE) {
+        dab = brush->image(device->colorSpace(), info, xFraction, yFraction);
+        dab->convertTo(KisMetaRegistry::instance()->csRegistry()->getAlpha8());
+    }
+    else {
+        KisAlphaMaskSP tqmask = brush->tqmask(info, xFraction, yFraction);
+        dab = computeDab(tqmask, KisMetaRegistry::instance()->csRegistry()->getAlpha8());
+    }
+
+    m_painter->setPressure(info.pressure);
+
+    KisPoint srcPointF = pt - m_painter->duplicateOffset();
+    TQPoint srcPoint = TQPoint(x - static_cast<TQ_INT32>(m_painter->duplicateOffset().x()),
+                             y - static_cast<TQ_INT32>(m_painter->duplicateOffset().y()));
+
+
+    TQ_INT32 sw = dab->extent().width();
+    TQ_INT32 sh = dab->extent().height();
+
+    if (srcPoint.x() < 0 )
+        srcPoint.setX(0);
+
+    if( srcPoint.y() < 0)
+        srcPoint.setY(0);
+    if( !(m_srcdev && m_srcdev->colorSpace() != device->colorSpace()) )
+    {
+        m_srcdev = new KisPaintDevice(device->colorSpace(), "duplicate source dev");
+        m_target = new KisPaintDevice(device->colorSpace(), "duplicate target dev");
+    }
+    Q_CHECK_PTR(m_srcdev);
+
+    // Perspective correction ?
+    KisPainter copyPainter(m_srcdev);
+    if(m_painter->duplicatePerspectiveCorrection())
+    {
+        double startM[3][3];
+        double endM[3][3];
+        for(int i = 0; i < 3; i++)
+        {
+            for(int j = 0; j < 3; j++)
+            {
+                startM[i][j] = 0.;
+                endM[i][j] = 0.;
+            }
+            startM[i][i] = 1.;
+            endM[i][i] = 1.;
+        }
+        // First look for the grid corresponding to the start point
+        KisSubPerspectiveGrid* subGridStart = *device->image()->perspectiveGrid()->begin();//device->image()->perspectiveGrid()->gridAt(KisPoint(srcPoint.x() +hotSpot.x(),srcPoint.y() +hotSpot.y()));
+        TQRect r = TQRect(0,0, device->image()->width(), device->image()->height());
+
+#if 1
+        if(subGridStart)
+        {
+//             kdDebug() << "fgrid" << endl;
+//             kdDebug() << *subGridStart->topLeft() << " " << *subGridStart->topRight() << " " << *subGridStart->bottomLeft() << " " <<  *subGridStart->bottomRight() << endl;
+            double* b = KisPerspectiveMath::computeMatrixTransfoFromPerspective( r, *subGridStart->topLeft(), *subGridStart->topRight(), *subGridStart->bottomLeft(), *subGridStart->bottomRight());
+            for(int i = 0; i < 3; i++)
+            {
+                for(int j = 0; j < 3; j++)
+                {
+//                     kdDebug() << "sol[" << 3*i+j << "]=" << b[3*i+j] << endl;
+                    startM[i][j] = b[3*i+j];
+                }
+            }
+
+        }
+#endif
+#if 1
+        // Second look for the grid corresponding to the end point
+        KisSubPerspectiveGrid* subGridEnd = *device->image()->perspectiveGrid()->begin();// device->image()->perspectiveGrid()->gridAt(pos);
+        if(subGridEnd)
+        {
+//             kdDebug() << "second grid" << endl;
+            double* b = KisPerspectiveMath::computeMatrixTransfoToPerspective(*subGridEnd->topLeft(), *subGridEnd->topRight(), *subGridEnd->bottomLeft(), *subGridEnd->bottomRight(), r);
+            for(int i = 0; i < 3; i++)
+            {
+                for(int j = 0; j < 3; j++)
+                {
+//                     kdDebug() << "sol[" << 3*i+j << "]=" << b[3*i+j] << endl;
+                    endM[i][j] = b[3*i+j];
+                }
+            }
+        }
+#endif
+//         kdDebug()<< " oouuuuh" << srcPointF << KisPerspectiveMath::matProd(startM,  KisPerspectiveMath::matProd(endM, srcPointF ) ) << KisPerspectiveMath::matProd(endM,  KisPerspectiveMath::matProd(startM, srcPointF ) );
+
+        // Compute the translation in the perspective transformation space:
+        KisPoint positionStartPaintingT = KisPerspectiveMath::matProd(endM, m_painter->duplicateStart() );
+        KisPoint duplicateStartPoisitionT = KisPerspectiveMath::matProd(endM, m_painter->duplicateStart() - m_painter->duplicateOffset() );
+        KisPoint translat = duplicateStartPoisitionT - positionStartPaintingT;
+        KisRectIteratorPixel dstIt = m_srcdev->createRectIterator(0, 0, sw, sh, true);
+        KisRandomSubAccessorPixel srcAcc = device->createRandomSubAccessor();
+        //Action
+        while(!dstIt.isDone())
+        {
+            if(dstIt.isSelected())
+            {
+                KisPoint p =  KisPerspectiveMath::matProd(startM, KisPerspectiveMath::matProd(endM, KisPoint(dstIt.x() + x, dstIt.y() + y) ) + translat );
+                srcAcc.moveTo( p );
+                srcAcc.sampledOldRawData( dstIt.rawData() );
+            }
+            ++dstIt;
+        }
+
+
+    } else {
+        // Or, copy the source data on the temporary device:
+        copyPainter.bitBlt(0, 0, COMPOSITE_COPY, device, srcPoint.x(), srcPoint.y(), sw, sh);
+        copyPainter.end();
+    }
+
+    // heal ?
+
+    if(heal)
+    {
+        TQ_UINT16 dataDevice[4];
+        TQ_UINT16 dataSrcDev[4];
+        TQMemArray<double> matrix ( 3 * sw * sh );
+        // First divide
+        KisColorSpace* deviceCs = device->colorSpace();
+        KisHLineIteratorPixel deviceIt = device->createHLineIterator(x, y, sw, false );
+        KisHLineIteratorPixel srcDevIt = m_srcdev->createHLineIterator(0, 0, sw, true );
+        double* matrixIt = &matrix[0];
+        for(int j = 0; j < sh; j++)
+        {
+            for(int i= 0; !srcDevIt.isDone(); i++)
+            {
+                deviceCs->toLabA16(deviceIt.rawData(), (TQ_UINT8*)dataDevice, 1);
+                deviceCs->toLabA16(srcDevIt.rawData(), (TQ_UINT8*)dataSrcDev, 1);
+                // Division
+                for( int k = 0; k < 3; k++)
+                {
+                    matrixIt[k] = dataDevice[k] / (double)TQMAX(dataSrcDev [k], 1);
+                }
+                ++deviceIt;
+                ++srcDevIt;
+                matrixIt +=3;
+            }
+            deviceIt.nextRow();
+            srcDevIt.nextRow();
+        }
+        // Minimize energy
+        {
+            int iter = 0;
+            double err;
+            TQMemArray<double> solution ( 3 * sw * sh );
+            do {
+                err = minimizeEnergy(&matrix[0], &solution[0],sw,sh);
+                memcpy (&matrix[0], &solution[0], sw * sh * 3 * sizeof(double));
+                iter++;
+            } while( err < 0.00001 && iter < 100);
+        }
+
+        // Finaly multiply
+        deviceIt = device->createHLineIterator(x, y, sw, false );
+        srcDevIt = m_srcdev->createHLineIterator(0, 0, sw, true );
+        matrixIt = &matrix[0];
+        for(int j = 0; j < sh; j++)
+        {
+            for(int i= 0; !srcDevIt.isDone(); i++)
+            {
+                deviceCs->toLabA16(deviceIt.rawData(), (TQ_UINT8*)dataDevice, 1);
+                deviceCs->toLabA16(srcDevIt.rawData(), (TQ_UINT8*)dataSrcDev, 1);
+                // Multiplication
+                for( int k = 0; k < 3; k++)
+                {
+                    dataSrcDev[k] = (int)CLAMP( matrixIt[k] * TQMAX( dataSrcDev[k], 1), 0, 65535 );
+                }
+                deviceCs->fromLabA16((TQ_UINT8*)dataSrcDev, srcDevIt.rawData(), 1);
+                ++deviceIt;
+                ++srcDevIt;
+                matrixIt +=3;
+            }
+            deviceIt.nextRow();
+            srcDevIt.nextRow();
+        }
+    }
+
+
+    // Add the dab as selection to the srcdev
+//     KisPainter copySelection(srcdev->selection().data());
+//     copySelection.bitBlt(0, 0, COMPOSITE_OVER, dab, 0, 0, sw, sh);
+//     copySelection.end();
+
+    // copy the srcdev onto a new device, after applying the dab selection
+    copyPainter.begin(m_target);
+
+    copyPainter.bltMask(0, 0, COMPOSITE_OVER, m_srcdev, dab,
+                             OPACITY_OPAQUE, 0, 0, sw, sh);
+    copyPainter.end();
+
+    TQRect dabRect = TQRect(0, 0, brush->tqmaskWidth(info), brush->tqmaskHeight(info));
+    TQRect dstRect = TQRect(x, y, dabRect.width(), dabRect.height());
+
+    KisImage * image = device->image();
+
+    if (image != 0) {
+        dstRect &= image->bounds();
+    }
+
+    if (dstRect.isNull() || dstRect.isEmpty() || !dstRect.isValid()) return;
+
+    TQ_INT32 sx = dstRect.x() - x;
+    TQ_INT32 sy = dstRect.y() - y;
+    sw = dstRect.width();
+    sh = dstRect.height();
+
+    if (m_source->hasSelection()) {
+        m_painter->bltSelection(dstRect.x(), dstRect.y(), m_painter->compositeOp(), m_target,
+                                m_source->selection(), m_painter->opacity(), sx, sy, sw, sh);
+    }
+    else {
+        m_painter->bitBlt(dstRect.x(), dstRect.y(), m_painter->compositeOp(), m_target, m_painter->opacity(), sx, sy, sw, sh);
+    }
+
+
+    m_painter->addDirtyRect(dstRect);
+}