diff options
Diffstat (limited to 'krita/core/kis_selection.cc')
| -rw-r--r-- | krita/core/kis_selection.cc | 582 |
1 files changed, 582 insertions, 0 deletions
diff --git a/krita/core/kis_selection.cc b/krita/core/kis_selection.cc new file mode 100644 index 00000000..9964a355 --- /dev/null +++ b/krita/core/kis_selection.cc @@ -0,0 +1,582 @@ +/* + * Copyright (c) 2004 Boudewijn Rempt <boud@valdyas.org> + * + * 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., 675 mass ave, cambridge, ma 02139, usa. + */ + +#include <qimage.h> + +#include <kdebug.h> +#include <klocale.h> +#include <qcolor.h> + +#include "kis_layer.h" +#include "kis_debug_areas.h" +#include "kis_types.h" +#include "kis_colorspace_factory_registry.h" +#include "kis_fill_painter.h" +#include "kis_iterators_pixel.h" +#include "kis_integer_maths.h" +#include "kis_image.h" +#include "kis_datamanager.h" +#include "kis_fill_painter.h" +#include "kis_selection.h" + +KisSelection::KisSelection(KisPaintDeviceSP dev) + : super(dev->parentLayer() + , KisMetaRegistry::instance()->csRegistry()->getAlpha8() + , (QString("selection for ") + dev->name()).latin1()) + , m_parentPaintDevice(dev) + , m_doCacheExactRect(false) + , m_dirty(false) +{ + Q_ASSERT(dev); +} + +KisSelection::KisSelection() + : super(KisMetaRegistry::instance()->csRegistry()->getAlpha8(), "anonymous selection") + , m_parentPaintDevice(0), m_dirty(false) +{ +} + +KisSelection::KisSelection(const KisSelection& rhs) + : super(rhs), m_parentPaintDevice(rhs.m_parentPaintDevice), m_doCacheExactRect(rhs.m_doCacheExactRect), + m_cachedExactRect(rhs.m_cachedExactRect), m_dirty(rhs.m_dirty) +{ +} + +KisSelection::~KisSelection() +{ +} + +Q_UINT8 KisSelection::selected(Q_INT32 x, Q_INT32 y) +{ + KisHLineIteratorPixel iter = createHLineIterator(x, y, 1, false); + + Q_UINT8 *pix = iter.rawData(); + + return *pix; +} + +void KisSelection::setSelected(Q_INT32 x, Q_INT32 y, Q_UINT8 s) +{ + KisHLineIteratorPixel iter = createHLineIterator(x, y, 1, true); + + Q_UINT8 *pix = iter.rawData(); + + *pix = s; +} + +QImage KisSelection::maskImage() +{ + // If part of a KisAdjustmentLayer, there may be no parent device. + QImage img; + QRect bounds; + if (m_parentPaintDevice) { + + bounds = m_parentPaintDevice->exactBounds(); + bounds = bounds.intersect( m_parentPaintDevice->image()->bounds() ); + img = QImage(bounds.width(), bounds.height(), 32); + } + else { + bounds = QRect( 0, 0, image()->width(), image()->height()); + img = QImage(bounds.width(), bounds.height(), 32); + } + + KisHLineIteratorPixel it = createHLineIterator(bounds.x(), bounds.y(), bounds.width(), false); + for (int y2 = bounds.y(); y2 < bounds.height() - bounds.y(); ++y2) { + int x2 = 0; + while (!it.isDone()) { + Q_UINT8 s = MAX_SELECTED - *(it.rawData()); + Q_INT32 c = qRgb(s, s, s); + img.setPixel(x2, y2, c); + ++x2; + ++it; + } + it.nextRow(); + } + return img; +} +void KisSelection::select(QRect r) +{ + KisFillPainter painter(this); + KisColorSpace * cs = KisMetaRegistry::instance()->csRegistry()->getRGB8(); + painter.fillRect(r, KisColor(Qt::white, cs), MAX_SELECTED); + Q_INT32 x, y, w, h; + extent(x, y, w, h); +} + +void KisSelection::clear(QRect r) +{ + KisFillPainter painter(this); + KisColorSpace * cs = KisMetaRegistry::instance()->csRegistry()->getRGB8(); + painter.fillRect(r, KisColor(Qt::white, cs), MIN_SELECTED); +} + +void KisSelection::clear() +{ + Q_UINT8 defPixel = MIN_SELECTED; + m_datamanager->setDefaultPixel(&defPixel); + m_datamanager->clear(); +} + +void KisSelection::invert() +{ + Q_INT32 x,y,w,h; + + extent(x, y, w, h); + KisRectIterator it = createRectIterator(x, y, w, h, true); + while ( ! it.isDone() ) + { + // CBR this is wrong only first byte is inverted + // BSAR: But we have always only one byte in this color model :-). + *(it.rawData()) = MAX_SELECTED - *(it.rawData()); + ++it; + } + Q_UINT8 defPixel = MAX_SELECTED - *(m_datamanager->defaultPixel()); + m_datamanager->setDefaultPixel(&defPixel); +} + +bool KisSelection::isTotallyUnselected(QRect r) +{ + if(*(m_datamanager->defaultPixel()) != MIN_SELECTED) + return false; + QRect sr = selectedExactRect(); + return ! r.intersects(sr); +} + +bool KisSelection::isProbablyTotallyUnselected(QRect r) +{ + if(*(m_datamanager->defaultPixel()) != MIN_SELECTED) + return false; + QRect sr = selectedRect(); + return ! r.intersects(sr); +} + + +QRect KisSelection::selectedRect() const +{ + if(*(m_datamanager->defaultPixel()) == MIN_SELECTED || !m_parentPaintDevice) + return extent(); + else + return extent().unite(m_parentPaintDevice->extent()); +} + +QRect KisSelection::selectedExactRect() const +{ + if(m_doCacheExactRect) + return m_cachedExactRect; + else if(*(m_datamanager->defaultPixel()) == MIN_SELECTED || !m_parentPaintDevice) + return exactBounds(); + else + return exactBounds().unite(m_parentPaintDevice->exactBounds()); +} + +void KisSelection::stopCachingExactRect() +{ + kdDebug() << "stop caching the exact rect" << endl; + m_doCacheExactRect = false; +} + + +void KisSelection::startCachingExactRect() +{ + kdDebug() << "start caching the exact rect" << endl; + if(*(m_datamanager->defaultPixel()) == MIN_SELECTED || !m_parentPaintDevice) + m_cachedExactRect = exactBounds(); + else + m_cachedExactRect = exactBounds().unite(m_parentPaintDevice->exactBounds()); + m_doCacheExactRect = true; +} + +void KisSelection::paintUniformSelectionRegion(QImage img, const QRect& imageRect, const QRegion& uniformRegion) +{ + Q_ASSERT(img.size() == imageRect.size()); + Q_ASSERT(imageRect.contains(uniformRegion.boundingRect())); + + if (img.isNull() || img.size() != imageRect.size() || !imageRect.contains(uniformRegion.boundingRect())) { + return; + } + + if (*m_datamanager->defaultPixel() == MIN_SELECTED) { + + QRegion region = uniformRegion & QRegion(imageRect); + + if (!region.isEmpty()) { + QMemArray<QRect> rects = region.rects(); + + for (unsigned int i = 0; i < rects.count(); i++) { + QRect r = rects[i]; + + for (Q_INT32 y = 0; y < r.height(); ++y) { + + QRgb *imagePixel = reinterpret_cast<QRgb *>(img.scanLine(r.y() - imageRect.y() + y)); + imagePixel += r.x() - imageRect.x(); + + Q_INT32 numPixels = r.width(); + + while (numPixels > 0) { + + QRgb srcPixel = *imagePixel; + Q_UINT8 srcGrey = (qRed(srcPixel) + qGreen(srcPixel) + qBlue(srcPixel)) / 9; + Q_UINT8 srcAlpha = qAlpha(srcPixel); + + srcGrey = UINT8_MULT(srcGrey, srcAlpha); + Q_UINT8 dstAlpha = QMAX(srcAlpha, 192); + + QRgb dstPixel = qRgba(128 + srcGrey, 128 + srcGrey, 165 + srcGrey, dstAlpha); + *imagePixel = dstPixel; + + ++imagePixel; + --numPixels; + } + } + } + } + } +} + +void KisSelection::paintSelection(QImage img, Q_INT32 imageRectX, Q_INT32 imageRectY, Q_INT32 imageRectWidth, Q_INT32 imageRectHeight) +{ + Q_ASSERT(img.size() == QSize(imageRectWidth, imageRectHeight)); + + if (img.isNull() || img.size() != QSize(imageRectWidth, imageRectHeight)) { + return; + } + + QRect imageRect(imageRectX, imageRectY, imageRectWidth, imageRectHeight); + QRect selectionExtent = extent(); + + selectionExtent.setLeft(selectionExtent.left() - 1); + selectionExtent.setTop(selectionExtent.top() - 1); + selectionExtent.setWidth(selectionExtent.width() + 2); + selectionExtent.setHeight(selectionExtent.height() + 2); + + QRegion uniformRegion = QRegion(imageRect); + uniformRegion -= QRegion(selectionExtent); + + if (!uniformRegion.isEmpty()) { + paintUniformSelectionRegion(img, imageRect, uniformRegion); + } + + QRect nonuniformRect = imageRect & selectionExtent; + + if (!nonuniformRect.isEmpty()) { + + const Q_INT32 imageRectOffsetX = nonuniformRect.x() - imageRectX; + const Q_INT32 imageRectOffsetY = nonuniformRect.y() - imageRectY; + + imageRectX = nonuniformRect.x(); + imageRectY = nonuniformRect.y(); + imageRectWidth = nonuniformRect.width(); + imageRectHeight = nonuniformRect.height(); + + const Q_INT32 NUM_SELECTION_ROWS = 3; + + Q_UINT8 *selectionRow[NUM_SELECTION_ROWS]; + + Q_INT32 aboveRowIndex = 0; + Q_INT32 centreRowIndex = 1; + Q_INT32 belowRowIndex = 2; + + selectionRow[aboveRowIndex] = new Q_UINT8[imageRectWidth + 2]; + selectionRow[centreRowIndex] = new Q_UINT8[imageRectWidth + 2]; + selectionRow[belowRowIndex] = new Q_UINT8[imageRectWidth + 2]; + + readBytes(selectionRow[centreRowIndex], imageRectX - 1, imageRectY - 1, imageRectWidth + 2, 1); + readBytes(selectionRow[belowRowIndex], imageRectX - 1, imageRectY, imageRectWidth + 2, 1); + + for (Q_INT32 y = 0; y < imageRectHeight; ++y) { + + Q_INT32 oldAboveRowIndex = aboveRowIndex; + aboveRowIndex = centreRowIndex; + centreRowIndex = belowRowIndex; + belowRowIndex = oldAboveRowIndex; + + readBytes(selectionRow[belowRowIndex], imageRectX - 1, imageRectY + y + 1, imageRectWidth + 2, 1); + + const Q_UINT8 *aboveRow = selectionRow[aboveRowIndex] + 1; + const Q_UINT8 *centreRow = selectionRow[centreRowIndex] + 1; + const Q_UINT8 *belowRow = selectionRow[belowRowIndex] + 1; + + QRgb *imagePixel = reinterpret_cast<QRgb *>(img.scanLine(imageRectOffsetY + y)); + imagePixel += imageRectOffsetX; + + for (Q_INT32 x = 0; x < imageRectWidth; ++x) { + + Q_UINT8 centre = *centreRow; + + if (centre != MAX_SELECTED) { + + // this is where we come if the pixels should be blue or bluish + + QRgb srcPixel = *imagePixel; + Q_UINT8 srcGrey = (qRed(srcPixel) + qGreen(srcPixel) + qBlue(srcPixel)) / 9; + Q_UINT8 srcAlpha = qAlpha(srcPixel); + + // Colour influence is proportional to alphaPixel. + srcGrey = UINT8_MULT(srcGrey, srcAlpha); + + QRgb dstPixel; + + if (centre == MIN_SELECTED) { + //this is where we come if the pixels should be blue (or red outline) + + Q_UINT8 left = *(centreRow - 1); + Q_UINT8 right = *(centreRow + 1); + Q_UINT8 above = *aboveRow; + Q_UINT8 below = *belowRow; + + // Stop unselected transparent areas from appearing the same + // as selected transparent areas. + Q_UINT8 dstAlpha = QMAX(srcAlpha, 192); + + // now for a simple outline based on 4-connectivity + if (left != MIN_SELECTED || right != MIN_SELECTED || above != MIN_SELECTED || below != MIN_SELECTED) { + dstPixel = qRgba(255, 0, 0, dstAlpha); + } else { + dstPixel = qRgba(128 + srcGrey, 128 + srcGrey, 165 + srcGrey, dstAlpha); + } + } else { + dstPixel = qRgba(UINT8_BLEND(qRed(srcPixel), srcGrey + 128, centre), + UINT8_BLEND(qGreen(srcPixel), srcGrey + 128, centre), + UINT8_BLEND(qBlue(srcPixel), srcGrey + 165, centre), + srcAlpha); + } + + *imagePixel = dstPixel; + } + + aboveRow++; + centreRow++; + belowRow++; + imagePixel++; + } + } + + delete [] selectionRow[aboveRowIndex]; + delete [] selectionRow[centreRowIndex]; + delete [] selectionRow[belowRowIndex]; + } +} + +void KisSelection::paintSelection(QImage img, const QRect& scaledImageRect, const QSize& scaledImageSize, const QSize& imageSize) +{ + if (img.isNull() || scaledImageRect.isEmpty() || scaledImageSize.isEmpty() || imageSize.isEmpty()) { + return; + } + + Q_ASSERT(img.size() == scaledImageRect.size()); + + if (img.size() != scaledImageRect.size()) { + return; + } + + Q_INT32 imageWidth = imageSize.width(); + Q_INT32 imageHeight = imageSize.height(); + + QRect selectionExtent = extent(); + + selectionExtent.setLeft(selectionExtent.left() - 1); + selectionExtent.setTop(selectionExtent.top() - 1); + selectionExtent.setWidth(selectionExtent.width() + 2); + selectionExtent.setHeight(selectionExtent.height() + 2); + + double xScale = static_cast<double>(scaledImageSize.width()) / imageWidth; + double yScale = static_cast<double>(scaledImageSize.height()) / imageHeight; + + QRect scaledSelectionExtent; + + scaledSelectionExtent.setLeft(static_cast<int>(selectionExtent.left() * xScale)); + scaledSelectionExtent.setRight(static_cast<int>(ceil((selectionExtent.right() + 1) * xScale)) - 1); + scaledSelectionExtent.setTop(static_cast<int>(selectionExtent.top() * yScale)); + scaledSelectionExtent.setBottom(static_cast<int>(ceil((selectionExtent.bottom() + 1) * yScale)) - 1); + + QRegion uniformRegion = QRegion(scaledImageRect); + uniformRegion -= QRegion(scaledSelectionExtent); + + if (!uniformRegion.isEmpty()) { + paintUniformSelectionRegion(img, scaledImageRect, uniformRegion); + } + + QRect nonuniformRect = scaledImageRect & scaledSelectionExtent; + + if (!nonuniformRect.isEmpty()) { + + const Q_INT32 scaledImageRectXOffset = nonuniformRect.x() - scaledImageRect.x(); + const Q_INT32 scaledImageRectYOffset = nonuniformRect.y() - scaledImageRect.y(); + + const Q_INT32 scaledImageRectX = nonuniformRect.x(); + const Q_INT32 scaledImageRectY = nonuniformRect.y(); + const Q_INT32 scaledImageRectWidth = nonuniformRect.width(); + const Q_INT32 scaledImageRectHeight = nonuniformRect.height(); + + const Q_INT32 imageRowLeft = static_cast<Q_INT32>(scaledImageRectX / xScale); + const Q_INT32 imageRowRight = static_cast<Q_INT32>((ceil((scaledImageRectX + scaledImageRectWidth - 1 + 1) / xScale)) - 1); + + const Q_INT32 imageRowWidth = imageRowRight - imageRowLeft + 1; + const Q_INT32 imageRowStride = imageRowWidth + 2; + + const Q_INT32 NUM_SELECTION_ROWS = 3; + + Q_INT32 aboveRowIndex = 0; + Q_INT32 centreRowIndex = 1; + Q_INT32 belowRowIndex = 2; + + Q_INT32 aboveRowSrcY = -3; + Q_INT32 centreRowSrcY = -3; + Q_INT32 belowRowSrcY = -3; + + Q_UINT8 *selectionRows = new Q_UINT8[imageRowStride * NUM_SELECTION_ROWS]; + Q_UINT8 *selectionRow[NUM_SELECTION_ROWS]; + + selectionRow[0] = selectionRows + 1; + selectionRow[1] = selectionRow[0] + imageRowStride; + selectionRow[2] = selectionRow[0] + (2 * imageRowStride); + + for (Q_INT32 y = 0; y < scaledImageRectHeight; ++y) { + + Q_INT32 scaledY = scaledImageRectY + y; + Q_INT32 srcY = (scaledY * imageHeight) / scaledImageSize.height(); + + Q_UINT8 *aboveRow; + Q_UINT8 *centreRow; + Q_UINT8 *belowRow; + + if (srcY - 1 == aboveRowSrcY) { + aboveRow = selectionRow[aboveRowIndex]; + centreRow = selectionRow[centreRowIndex]; + belowRow = selectionRow[belowRowIndex]; + } else if (srcY - 1 == centreRowSrcY) { + + Q_INT32 oldAboveRowIndex = aboveRowIndex; + + aboveRowIndex = centreRowIndex; + centreRowIndex = belowRowIndex; + belowRowIndex = oldAboveRowIndex; + + aboveRow = selectionRow[aboveRowIndex]; + centreRow = selectionRow[centreRowIndex]; + belowRow = selectionRow[belowRowIndex]; + + readBytes(belowRow - 1, imageRowLeft - 1, srcY + 1, imageRowStride, 1); + + } else if (srcY - 1 == belowRowSrcY) { + + Q_INT32 oldAboveRowIndex = aboveRowIndex; + Q_INT32 oldCentreRowIndex = centreRowIndex; + + aboveRowIndex = belowRowIndex; + centreRowIndex = oldAboveRowIndex; + belowRowIndex = oldCentreRowIndex; + + aboveRow = selectionRow[aboveRowIndex]; + centreRow = selectionRow[centreRowIndex]; + belowRow = selectionRow[belowRowIndex]; + + if (belowRowIndex == centreRowIndex + 1) { + readBytes(centreRow - 1, imageRowLeft - 1, srcY, imageRowStride, 2); + } else { + readBytes(centreRow - 1, imageRowLeft - 1, srcY, imageRowStride, 1); + readBytes(belowRow - 1, imageRowLeft - 1, srcY + 1, imageRowStride, 1); + } + + } else { + + aboveRowIndex = 0; + centreRowIndex = 1; + belowRowIndex = 2; + + aboveRow = selectionRow[aboveRowIndex]; + centreRow = selectionRow[centreRowIndex]; + belowRow = selectionRow[belowRowIndex]; + + readBytes(selectionRows, imageRowLeft - 1, srcY - 1, imageRowStride, NUM_SELECTION_ROWS); + } + + aboveRowSrcY = srcY - 1; + centreRowSrcY = aboveRowSrcY + 1; + belowRowSrcY = centreRowSrcY + 1; + + QRgb *imagePixel = reinterpret_cast<QRgb *>(img.scanLine(scaledImageRectYOffset + y)); + imagePixel += scaledImageRectXOffset; + + for (Q_INT32 x = 0; x < scaledImageRectWidth; ++x) { + + Q_INT32 scaledX = scaledImageRectX + x; + Q_INT32 srcX = (scaledX * imageWidth) / scaledImageSize.width(); + + Q_UINT8 centre = *(centreRow + srcX - imageRowLeft); + + if (centre != MAX_SELECTED) { + + // this is where we come if the pixels should be blue or bluish + + QRgb srcPixel = *imagePixel; + Q_UINT8 srcGrey = (qRed(srcPixel) + qGreen(srcPixel) + qBlue(srcPixel)) / 9; + Q_UINT8 srcAlpha = qAlpha(srcPixel); + + // Colour influence is proportional to alphaPixel. + srcGrey = UINT8_MULT(srcGrey, srcAlpha); + + QRgb dstPixel; + + if (centre == MIN_SELECTED) { + //this is where we come if the pixels should be blue (or red outline) + + Q_UINT8 left = *(centreRow + (srcX - imageRowLeft) - 1); + Q_UINT8 right = *(centreRow + (srcX - imageRowLeft) + 1); + Q_UINT8 above = *(aboveRow + (srcX - imageRowLeft)); + Q_UINT8 below = *(belowRow + (srcX - imageRowLeft)); + + // Stop unselected transparent areas from appearing the same + // as selected transparent areas. + Q_UINT8 dstAlpha = QMAX(srcAlpha, 192); + + // now for a simple outline based on 4-connectivity + if (left != MIN_SELECTED || right != MIN_SELECTED || above != MIN_SELECTED || below != MIN_SELECTED) { + dstPixel = qRgba(255, 0, 0, dstAlpha); + } else { + dstPixel = qRgba(128 + srcGrey, 128 + srcGrey, 165 + srcGrey, dstAlpha); + } + } else { + dstPixel = qRgba(UINT8_BLEND(qRed(srcPixel), srcGrey + 128, centre), + UINT8_BLEND(qGreen(srcPixel), srcGrey + 128, centre), + UINT8_BLEND(qBlue(srcPixel), srcGrey + 165, centre), + srcAlpha); + } + + *imagePixel = dstPixel; + } + + imagePixel++; + } + } + + delete [] selectionRows; + } +} + +void KisSelection::setDirty(const QRect& rc) +{ + if (m_dirty) + super::setDirty(rc); +} + +void KisSelection::setDirty() +{ + if (m_dirty) + super::setDirty(); +} |