You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
342 lines
12 KiB
342 lines
12 KiB
/*
|
|
* 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);
|
|
TQ_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 mask = brush->mask(info, xFraction, yFraction);
|
|
dab = computeDab(mask, 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");
|
|
}
|
|
TQ_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->maskWidth(info), brush->maskHeight(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);
|
|
}
|