KisSpacingInformation KisFilterOp::paintAt(const KisPaintInformation& info)
{
if (!painter()) {
return KisSpacingInformation(1.0);
}
if (!m_filter) {
return KisSpacingInformation(1.0);
}
if (!source()) {
return KisSpacingInformation(1.0);
}
KisBrushSP brush = m_brush;
if (!brush) return KisSpacingInformation(1.0);
if (! brush->canPaintFor(info))
return KisSpacingInformation(1.0);
qreal scale = m_sizeOption.apply(info);
if (checkSizeTooSmall(scale)) return KisSpacingInformation();
setCurrentScale(scale);
qreal rotation = m_rotationOption.apply(info);
static const KoColorSpace *cs = KoColorSpaceRegistry::instance()->alpha8();
static KoColor color(Qt::black, cs);
QRect dstRect;
KisFixedPaintDeviceSP dab =
m_dabCache->fetchDab(cs, color, info.pos(),
scale, scale, rotation,
info, 1.0,
&dstRect);
if (dstRect.isEmpty()) return KisSpacingInformation(1.0);
QRect dabRect = dab->bounds();
// sanity check
Q_ASSERT(dstRect.size() == dabRect.size());
// Filter the paint device
QRect neededRect = m_filter->neededRect(dstRect, m_filterConfiguration);
KisPainter p(m_tmpDevice);
if (!m_smudgeMode) {
p.setCompositeOp(COMPOSITE_COPY);
}
p.bitBltOldData(neededRect.topLeft() - dstRect.topLeft(), source(), neededRect);
KisTransaction transaction(m_tmpDevice);
m_filter->process(m_tmpDevice, dabRect, m_filterConfiguration, 0);
transaction.end();
painter()->
bitBltWithFixedSelection(dstRect.x(), dstRect.y(),
m_tmpDevice, dab,
0, 0,
dabRect.x(), dabRect.y(),
dabRect.width(), dabRect.height());
painter()->renderMirrorMaskSafe(dstRect, m_tmpDevice, 0, 0, dab,
!m_dabCache->needSeparateOriginal());
return effectiveSpacing(scale, rotation);
}
KisSpacingInformation KisColorSmudgeOp::paintAt(const KisPaintInformation& info)
{
KisBrushSP brush = m_brush;
// Simple error catching
if (!painter()->device() || !brush || !brush->canPaintFor(info))
return 1.0;
// get the scaling factor calculated by the size option
qreal scale = m_sizeOption.apply(info);
qreal rotation = m_rotationOption.apply(info);
if (checkSizeTooSmall(scale)) return KisSpacingInformation();
setCurrentScale(scale);
setCurrentRotation(rotation);
QPointF scatteredPos =
m_scatterOption.apply(info,
brush->maskWidth(scale, rotation, 0, 0, info),
brush->maskHeight(scale, rotation, 0, 0, info));
QPointF hotSpot = brush->hotSpot(scale, scale, rotation, info);
/**
* Update the brush mask.
*
* Upon leaving the function:
* o m_maskDab stores the new mask
* o m_maskBounds stroes the extents of the mask paint device
* o m_dstDabRect stores the destination rect where the mask is going
* to be written to
*/
updateMask(info, scale, rotation, scatteredPos);
QPointF newCenterPos = QRectF(m_dstDabRect).center();
QRect srcDabRect = m_dstDabRect.translated((m_lastPaintPos - newCenterPos).toPoint());
/**
* Save the center of the current dab to know where to read the
* data during the next pass. We do not save scatteredPos here,
* because it may slightly differ from the real center of the
* brush (due to some rounding effects), which will result in
* really weird quality.
*/
m_lastPaintPos = QRectF(m_dstDabRect).center();
KisSpacingInformation spacingInfo =
effectiveSpacing(m_dstDabRect.width(), m_dstDabRect.height(),
m_spacingOption, info);
if (m_firstRun) {
m_firstRun = false;
return spacingInfo;
}
// save the old opacity value and composite mode
quint8 oldOpacity = painter()->opacity();
QString oldModeId = painter()->compositeOp()->id();
qreal fpOpacity = (qreal(oldOpacity) / 255.0) * m_opacityOption.getOpacityf(info);
KoColor color = painter()->paintColor();
if (m_image && m_overlayModeOption.isChecked()) {
m_image->blockUpdates();
m_backgroundPainter->bitBlt(QPoint(), m_image->projection(), srcDabRect);
m_image->unblockUpdates();
}
else {
// IMPORTANT: clear the temporary painting device to color black with zero opacity
// it will only clear the extents of the brush
m_tempDev->clear(QRect(QPoint(), m_dstDabRect.size()));
}
if (m_smudgeRateOption.getMode() == KisSmudgeOption::SMEARING_MODE) {
// cut out the area from the canvas under the brush
// and blit it to the temporary painting device
if(m_smudgeRadiusOption.isChecked())
{
m_smudgeRadiusOption.apply(*m_smudgePainter,info,m_brush->width(),m_dstDabRect.center().x(),m_dstDabRect.center().y(),painter()->device());
color = m_smudgePainter->paintColor();
}
m_smudgePainter->bitBlt(QPoint(), painter()->device(), srcDabRect);
}
else {
QPoint pt = (srcDabRect.topLeft() + hotSpot).toPoint();
// get the pixel on the canvas that lies beneath the hot spot
// of the dab and fill the temporary paint device with that color
KoColor color = painter()->paintColor();
if(m_smudgeRadiusOption.isChecked())
{
m_smudgeRadiusOption.apply(*m_smudgePainter,info,m_brush->width(),pt.x(),pt.y(),painter()->device());
KoColor color2 = m_smudgePainter->paintColor();
m_smudgePainter->fill(0, 0, m_dstDabRect.width(), m_dstDabRect.height(),color2);
}
else
{
KisCrossDeviceColorPickerInt colorPicker(painter()->device(), color);
colorPicker.pickColor(pt.x(), pt.y(), color.data());
m_smudgePainter->fill(0, 0, m_dstDabRect.width(), m_dstDabRect.height(), color);
}
}
// if the user selected the color smudge option
// we will mix some color into the temporary painting device (m_tempDev)
if (m_colorRateOption.isChecked()) {
// this will apply the opacy (selected by the user) to copyPainter
// (but fit the rate inbetween the range 0.0 to (1.0-SmudgeRate))
qreal maxColorRate = qMax<qreal>(1.0 - m_smudgeRateOption.getRate(), 0.2);
m_colorRateOption.apply(*m_colorRatePainter, info, 0.0, maxColorRate, fpOpacity);
// paint a rectangle with the current color (foreground color)
// or a gradient color (if enabled)
// into the temporary painting device and use the user selected
// composite mode
KoColor color = painter()->paintColor();
m_gradientOption.apply(color, m_gradient, info);
m_colorRatePainter->fill(0, 0, m_dstDabRect.width(), m_dstDabRect.height(), color);
}
// if color is disabled (only smudge) and "overlay mode is enabled
// then first blit the region under the brush from the image projection
// to the painting device to prevent a rapid build up of alpha value
// if the color to be smudged is semi transparent
if (m_image && m_overlayModeOption.isChecked() && !m_colorRateOption.isChecked()) {
painter()->setCompositeOp(COMPOSITE_COPY);
painter()->setOpacity(OPACITY_OPAQUE_U8);
m_image->blockUpdates();
painter()->bitBlt(m_dstDabRect.topLeft(), m_image->projection(), m_dstDabRect);
m_image->unblockUpdates();
}
// set opacity calculated by the rate option
m_smudgeRateOption.apply(*painter(), info, 0.0, 1.0, fpOpacity);
// then blit the temporary painting device on the canvas at the current brush position
// the alpha mask (maskDab) will be used here to only blit the pixels that are in the area (shape) of the brush
painter()->setCompositeOp(COMPOSITE_COPY);
painter()->bitBltWithFixedSelection(m_dstDabRect.x(), m_dstDabRect.y(), m_tempDev, m_maskDab, m_dstDabRect.width(), m_dstDabRect.height());
painter()->renderMirrorMaskSafe(m_dstDabRect, m_tempDev, 0, 0, m_maskDab,
!m_dabCache->needSeparateOriginal());
// restore orginal opacy and composite mode values
painter()->setOpacity(oldOpacity);
painter()->setCompositeOp(oldModeId);
return spacingInfo;
}
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());
}
