summaryrefslogtreecommitdiffstats
path: root/chalk/plugins/paintops/defaultpaintops/kis_duplicateop.cc
diff options
context:
space:
mode:
authortpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da>2011-06-26 00:41:16 +0000
committertpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da>2011-06-26 00:41:16 +0000
commit698569f8428ca088f764d704034a1330517b98c0 (patch)
treebf45be6946ebbbee9cce5a5bcf838f4c952d87e6 /chalk/plugins/paintops/defaultpaintops/kis_duplicateop.cc
parent2785103a6bd4de55bd26d79e34d0fdd4b329a73a (diff)
downloadkoffice-698569f8428ca088f764d704034a1330517b98c0.tar.gz
koffice-698569f8428ca088f764d704034a1330517b98c0.zip
Finish rebranding of Krita as Chalk
git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/koffice@1238363 283d02a7-25f6-0310-bc7c-ecb5cbfe19da
Diffstat (limited to 'chalk/plugins/paintops/defaultpaintops/kis_duplicateop.cc')
-rw-r--r--chalk/plugins/paintops/defaultpaintops/kis_duplicateop.cc341
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);
+}