void KisDuplicateOp::paintAt(const KisPaintInformation& info)
{
if (!painter()) return;
if (!m_duplicateStartIsSet) {
m_duplicateStartIsSet = true;
m_duplicateStart = info.pos();
}
bool heal = settings->healing();
if (!source()) return;
KisBrushSP brush = m_brush;
if (!brush) return;
if (! brush->canPaintFor(info))
return;
double scale = KisPaintOp::scaleForPressure(info.pressure());
QPointF hotSpot = brush->hotSpot(scale, scale);
QPointF pt = info.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.
qint32 x;
double xFraction;
qint32 y;
double yFraction;
splitCoordinate(pt.x(), &x, &xFraction);
splitCoordinate(pt.y(), &y, &yFraction);
xFraction = yFraction = 0.0;
QPointF srcPointF = pt - settings->offset();
QPoint srcPoint = QPoint(x - static_cast<qint32>(settings->offset().x()),
y - static_cast<qint32>(settings->offset().y()));
qint32 sw = brush->maskWidth(scale, 0.0);
qint32 sh = brush->maskHeight(scale, 0.0);
if (srcPoint.x() < 0)
srcPoint.setX(0);
if (srcPoint.y() < 0)
srcPoint.setY(0);
if (!(m_srcdev && !(*m_srcdev->colorSpace() == *source()->colorSpace()))) {
m_srcdev = new KisPaintDevice(source()->colorSpace());
}
Q_CHECK_PTR(m_srcdev);
// Perspective correction ?
KisPainter copyPainter(m_srcdev);
KisImageSP image = settings->m_image;
if (settings->perspectiveCorrection() && image && image->perspectiveGrid()->countSubGrids() == 1) {
Matrix3qreal startM = Matrix3qreal::Identity();
Matrix3qreal endM = Matrix3qreal::Identity();
// First look for the grid corresponding to the start point
KisSubPerspectiveGrid* subGridStart = *image->perspectiveGrid()->begin();
QRect r = QRect(0, 0, image->width(), image->height());
#if 1
if (subGridStart) {
startM = KisPerspectiveMath::computeMatrixTransfoFromPerspective(r, *subGridStart->topLeft(), *subGridStart->topRight(), *subGridStart->bottomLeft(), *subGridStart->bottomRight());
}
#endif
#if 1
// Second look for the grid corresponding to the end point
KisSubPerspectiveGrid* subGridEnd = *image->perspectiveGrid()->begin();
if (subGridEnd) {
endM = KisPerspectiveMath::computeMatrixTransfoToPerspective(*subGridEnd->topLeft(), *subGridEnd->topRight(), *subGridEnd->bottomLeft(), *subGridEnd->bottomRight(), r);
}
#endif
// Compute the translation in the perspective transformation space:
QPointF positionStartPaintingT = KisPerspectiveMath::matProd(endM, QPointF(m_duplicateStart));
QPointF duplicateStartPositionT = KisPerspectiveMath::matProd(endM, QPointF(m_duplicateStart) - QPointF(settings->offset()));
QPointF translat = duplicateStartPositionT - positionStartPaintingT;
KisRectIteratorPixel dstIt = m_srcdev->createRectIterator(0, 0, sw, sh);
KisRandomSubAccessorPixel srcAcc = source()->createRandomSubAccessor();
//Action
while (!dstIt.isDone()) {
if (dstIt.isSelected()) {
QPointF p = KisPerspectiveMath::matProd(startM, KisPerspectiveMath::matProd(endM, QPointF(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.setCompositeOp(COMPOSITE_COPY);
copyPainter.bitBlt(0, 0, source(), srcPoint.x(), srcPoint.y(), sw, sh);
copyPainter.end();
}
// heal ?
if (heal) {
quint16 dataDevice[4];
quint16 dataSrcDev[4];
double* matrix = new double[ 3 * sw * sh ];
// First divide
const KoColorSpace* deviceCs = source()->colorSpace();
KisHLineConstIteratorPixel deviceIt = source()->createHLineConstIterator(x, y, sw);
KisHLineIteratorPixel srcDevIt = m_srcdev->createHLineIterator(0, 0, sw);
double* matrixIt = &matrix[0];
for (int j = 0; j < sh; j++) {
for (int i = 0; !srcDevIt.isDone(); i++) {
deviceCs->toLabA16(deviceIt.rawData(), (quint8*)dataDevice, 1);
deviceCs->toLabA16(srcDevIt.rawData(), (quint8*)dataSrcDev, 1);
// Division
for (int k = 0; k < 3; k++) {
matrixIt[k] = dataDevice[k] / (double)qMax((int)dataSrcDev [k], 1);
}
++deviceIt;
++srcDevIt;
matrixIt += 3;
}
deviceIt.nextRow();
srcDevIt.nextRow();
}
// Minimize energy
{
int iter = 0;
double err;
double* solution = new double [ 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);
delete [] solution;
}
// Finaly multiply
deviceIt = source()->createHLineIterator(x, y, sw);
srcDevIt = m_srcdev->createHLineIterator(0, 0, sw);
matrixIt = &matrix[0];
for (int j = 0; j < sh; j++) {
for (int i = 0; !srcDevIt.isDone(); i++) {
deviceCs->toLabA16(deviceIt.rawData(), (quint8*)dataDevice, 1);
deviceCs->toLabA16(srcDevIt.rawData(), (quint8*)dataSrcDev, 1);
// Multiplication
for (int k = 0; k < 3; k++) {
dataSrcDev[k] = (int)CLAMP(matrixIt[k] * qMax((int) dataSrcDev[k], 1), 0, 65535);
}
deviceCs->fromLabA16((quint8*)dataSrcDev, srcDevIt.rawData(), 1);
++deviceIt;
++srcDevIt;
matrixIt += 3;
}
deviceIt.nextRow();
srcDevIt.nextRow();
}
delete [] matrix;
}
KisFixedPaintDeviceSP fixedDab = new KisFixedPaintDevice(m_srcdev->colorSpace());
fixedDab->setRect(QRect(0, 0, sw, sh));
fixedDab->initialize();
m_srcdev->readBytes(fixedDab->data(), fixedDab->bounds());
brush->mask(fixedDab, scale, scale, 0.0, info, xFraction, yFraction);
m_srcdev->writeBytes(fixedDab->data(), fixedDab->bounds());
QRect dabRect = QRect(0, 0, brush->maskWidth(scale, 0.0), brush->maskHeight(scale, 0.0));
QRect dstRect = QRect(x, y, dabRect.width(), dabRect.height());
if (painter()->bounds().isValid()) {
dstRect &= painter()->bounds();
}
if (dstRect.isNull() || dstRect.isEmpty() || !dstRect.isValid()) return;
qint32 sx = dstRect.x() - x;
qint32 sy = dstRect.y() - y;
sw = dstRect.width();
sh = dstRect.height();
painter()->bitBlt(dstRect.x(), dstRect.y(), m_srcdev, sx, sy, sw, sh);
}
KisSpacingInformation KisDuplicateOp::paintAt(const KisPaintInformation& info)
{
if (!painter()->device()) return 1.0;
KisBrushSP brush = m_brush;
if (!brush)
return 1.0;
if (!brush->canPaintFor(info))
return 1.0;
if (!m_duplicateStartIsSet) {
m_duplicateStartIsSet = true;
m_duplicateStart = info.pos();
}
KisPaintDeviceSP realSourceDevice = settings->node()->paintDevice();
qreal scale = m_sizeOption.apply(info);
if (checkSizeTooSmall(scale)) return KisSpacingInformation();
QPointF hotSpot = brush->hotSpot(scale, scale, 0, info);
QPointF pt = info.pos() - hotSpot;
setCurrentScale(scale);
// 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.
qint32 x, y;
qreal xFraction, yFraction; // will not be used
splitCoordinate(pt.x(), &x, &xFraction);
splitCoordinate(pt.y(), &y, &yFraction);
QPoint srcPoint;
if(m_moveSourcePoint)
{
srcPoint = QPoint(x - static_cast<qint32>(settings->offset().x()),
y - static_cast<qint32>(settings->offset().y()));
} else {
srcPoint = QPoint(static_cast<qint32>(settings->position().x() - hotSpot.x()),
static_cast<qint32>(settings->position().y() - hotSpot.y()));
}
qint32 sw = brush->maskWidth(scale, 0.0, xFraction, yFraction, info);
qint32 sh = brush->maskHeight(scale, 0.0, xFraction, yFraction, info);
if (srcPoint.x() < 0)
srcPoint.setX(0);
if (srcPoint.y() < 0)
srcPoint.setY(0);
// Perspective correction ?
KisImageWSP image = settings->m_image;
if (m_perspectiveCorrection && image && image->perspectiveGrid()->countSubGrids() == 1) {
Matrix3qreal startM = Matrix3qreal::Identity();
Matrix3qreal endM = Matrix3qreal::Identity();
// First look for the grid corresponding to the start point
KisSubPerspectiveGrid* subGridStart = *image->perspectiveGrid()->begin();
QRect r = QRect(0, 0, image->width(), image->height());
#if 1
if (subGridStart) {
startM = KisPerspectiveMath::computeMatrixTransfoFromPerspective(r, *subGridStart->topLeft(), *subGridStart->topRight(), *subGridStart->bottomLeft(), *subGridStart->bottomRight());
}
#endif
#if 1
// Second look for the grid corresponding to the end point
KisSubPerspectiveGrid* subGridEnd = *image->perspectiveGrid()->begin();
if (subGridEnd) {
endM = KisPerspectiveMath::computeMatrixTransfoToPerspective(*subGridEnd->topLeft(), *subGridEnd->topRight(), *subGridEnd->bottomLeft(), *subGridEnd->bottomRight(), r);
}
#endif
// Compute the translation in the perspective transformation space:
QPointF positionStartPaintingT = KisPerspectiveMath::matProd(endM, QPointF(m_duplicateStart));
QPointF duplicateStartPositionT = KisPerspectiveMath::matProd(endM, QPointF(m_duplicateStart) - QPointF(settings->offset()));
QPointF translat = duplicateStartPositionT - positionStartPaintingT;
KisRectIteratorSP dstIt = m_srcdev->createRectIteratorNG(0, 0, sw, sh);
KisRandomSubAccessorSP srcAcc = realSourceDevice->createRandomSubAccessor();
//Action
do {
QPointF p = KisPerspectiveMath::matProd(startM, KisPerspectiveMath::matProd(endM, QPointF(dstIt->x() + x, dstIt->y() + y)) + translat);
srcAcc->moveTo(p);
srcAcc->sampledOldRawData(dstIt->rawData());
} while (dstIt->nextPixel());
} else {
KisPainter copyPainter(m_srcdev);
copyPainter.setCompositeOp(COMPOSITE_COPY);
copyPainter.bitBltOldData(0, 0, realSourceDevice, srcPoint.x(), srcPoint.y(), sw, sh);
copyPainter.end();
}
// heal ?
if (m_healing) {
quint16 srcData[4];
quint16 tmpData[4];
qreal* matrix = new qreal[ 3 * sw * sh ];
// First divide
const KoColorSpace* srcCs = realSourceDevice->colorSpace();
const KoColorSpace* tmpCs = m_srcdev->colorSpace();
KisHLineConstIteratorSP srcIt = realSourceDevice->createHLineConstIteratorNG(x, y, sw);
KisHLineIteratorSP tmpIt = m_srcdev->createHLineIteratorNG(0, 0, sw);
qreal* matrixIt = &matrix[0];
for (int j = 0; j < sh; j++) {
for (int i = 0; i < sw; i++) {
srcCs->toLabA16(srcIt->oldRawData(), (quint8*)srcData, 1);
tmpCs->toLabA16(tmpIt->rawData(), (quint8*)tmpData, 1);
// Division
for (int k = 0; k < 3; k++) {
matrixIt[k] = srcData[k] / (qreal)qMax((int)tmpData [k], 1);
}
srcIt->nextPixel();
tmpIt->nextPixel();
matrixIt += 3;
}
srcIt->nextRow();
tmpIt->nextRow();
}
// Minimize energy
{
int iter = 0;
qreal err;
qreal* solution = new qreal [ 3 * sw * sh ];
do {
err = minimizeEnergy(&matrix[0], &solution[0], sw, sh);
// swap pointers
qreal *tmp = matrix;
matrix = solution;
solution = tmp;
iter++;
} while (err > 0.00001 && iter < 100);
delete [] solution;
}
// Finaly multiply
KisHLineIteratorSP srcIt2 = realSourceDevice->createHLineIteratorNG(x, y, sw);
KisHLineIteratorSP tmpIt2 = m_srcdev->createHLineIteratorNG(0, 0, sw);
matrixIt = &matrix[0];
for (int j = 0; j < sh; j++) {
for (int i = 0; i < sw; i++) {
srcCs->toLabA16(srcIt2->rawData(), (quint8*)srcData, 1);
tmpCs->toLabA16(tmpIt2->rawData(), (quint8*)tmpData, 1);
// Multiplication
for (int k = 0; k < 3; k++) {
tmpData[k] = (int)CLAMP(matrixIt[k] * qMax((int) tmpData[k], 1), 0, 65535);
}
tmpCs->fromLabA16((quint8*)tmpData, tmpIt2->rawData(), 1);
srcIt2->nextPixel();
tmpIt2->nextPixel();
matrixIt += 3;
}
srcIt2->nextRow();
tmpIt2->nextRow();
}
delete [] matrix;
}
static const KoColorSpace *cs = KoColorSpaceRegistry::instance()->alpha8();
static KoColor color(Qt::black, cs);
KisFixedPaintDeviceSP dab =
m_dabCache->fetchDab(cs, color, scale, scale,
0.0, info);
QRect dstRect = QRect(x, y, dab->bounds().width(), dab->bounds().height());
if (dstRect.isEmpty()) return 1.0;
painter()->bitBltWithFixedSelection(dstRect.x(), dstRect.y(),
m_srcdev, dab,
dstRect.width(),
dstRect.height());
painter()->renderMirrorMaskSafe(dstRect, m_srcdev, 0, 0, dab,
!m_dabCache->needSeparateOriginal());
return effectiveSpacing(dstRect.width(), dstRect.height());
}
