KoColorTransformation* KisBurnShadowsAdjustmentFactory::createTransformation(const KoColorSpace* colorSpace, QHash<QString, QVariant> parameters) const
{
KoColorTransformation * adj;
if (colorSpace->colorModelId() != RGBAColorModelID) {
dbgKrita << "Unsupported color space " << colorSpace->id() << " in KisBurnShadowsAdjustment::createTransformation";
return 0;
}
if (colorSpace->colorDepthId() == Integer8BitsColorDepthID) {
adj = new KisBurnShadowsAdjustment< quint8, KoBgrTraits < quint8 > >();
}
#ifdef HAVE_OPENEXR
else if (colorSpace->colorDepthId() == Float16BitsColorDepthID) {
adj = new KisBurnShadowsAdjustment< half, KoRgbTraits < half > >();
}
#endif
else if (colorSpace->colorDepthId() == Integer16BitsColorDepthID) {
adj = new KisBurnShadowsAdjustment< quint16, KoBgrTraits < quint8 > >();
} else if (colorSpace->colorDepthId() == Float32BitsColorDepthID) {
adj = new KisBurnShadowsAdjustment< float, KoRgbTraits < float > >();
} else {
dbgKrita << "Unsupported color space " << colorSpace->id() << " in KisBurnShadowsAdjustment::createTransformation";
return 0;
}
adj->setParameters(parameters);
return adj;
}
void ChalkBrush::paint(KisPaintDeviceSP dev, qreal x, qreal y, const KoColor &color)
{
m_inkColor = color;
m_counter++;
qreal decr = (m_counter * m_counter * m_counter) / 1000000.0f;
Bristle *bristle;
qint32 pixelSize = dev->colorSpace()->pixelSize();
KisRandomAccessor accessor = dev->createRandomAccessor((int)x, (int)y);
qreal dirt, result;
//count decrementing of saturation and alpha
result = log( ( qreal )m_counter);
result = -(result * 10) / 100.0;
QHash<QString, QVariant> params;
params["h"] = 0.0;
params["s"] = result;
params["v"] = 0.0;
KoColorTransformation* transfo = dev->colorSpace()->createColorTransformation("hsv_adjustment", params);
transfo->transform(m_inkColor.data(), m_inkColor.data(), 1);
int opacity = static_cast<int>((1.0f + result) * 255.0);
m_inkColor.setOpacity(opacity);
for (int i = 0;i < m_bristles.size();i++) {
bristle = &m_bristles[i];
// let's call that noise from ground to chalk :)
dirt = drand48();
if (bristle->distanceCenter() > m_radius || dirt < 0.5) {
continue;
}
int dx, dy;
dx = (int)(x + bristle->x());
dy = (int)(y + bristle->y());
accessor.moveTo(dx, dy);
memcpy(accessor.rawData(), m_inkColor.data(), pixelSize);
bristle->setInkAmount(1.0f - decr);
}
}
void KisGmicSimpleConvertor::convertToGmicImageFast(KisPaintDeviceSP dev, CImg< float >& gmicImage, QRect rc)
{
KoColorTransformation * pixelToGmicPixelFormat = createTransformation(dev->colorSpace());
if (pixelToGmicPixelFormat == 0)
{
dbgPlugins << "Fall-back to slow color conversion method";
convertToGmicImage(dev, gmicImage, rc);
return;
}
if (rc.isEmpty())
{
dbgPlugins << "Image rectangle is empty! Using supplied gmic layer dimension";
rc = QRect(0,0,gmicImage._width, gmicImage._height);
}
qint32 x = rc.x();
qint32 y = rc.y();
qint32 width = rc.width();
qint32 height = rc.height();
width = width < 0 ? 0 : width;
height = height < 0 ? 0 : height;
const qint32 numChannels = 4;
int greenOffset = gmicImage._width * gmicImage._height;
int blueOffset = greenOffset * 2;
int alphaOffset = greenOffset * 3;
QVector<float *> planes;
planes.append(gmicImage._data);
planes.append(gmicImage._data + greenOffset);
planes.append(gmicImage._data + blueOffset);
planes.append(gmicImage._data + alphaOffset);
KisRandomConstAccessorSP it = dev->createRandomConstAccessorNG(dev->x(), dev->y());
int tileWidth = it->numContiguousColumns(dev->x());
int tileHeight = it->numContiguousRows(dev->y());
Q_ASSERT(tileWidth == 64);
Q_ASSERT(tileHeight == 64);
const KoColorSpace *rgbaFloat32bitcolorSpace = KoColorSpaceRegistry::instance()->colorSpace(RGBAColorModelID.id(),
Float32BitsColorDepthID.id(),
KoColorSpaceRegistry::instance()->rgb8()->profile());
Q_CHECK_PTR(rgbaFloat32bitcolorSpace);
const qint32 dstPixelSize = rgbaFloat32bitcolorSpace->pixelSize();
const qint32 srcPixelSize = dev->pixelSize();
quint8 * dstTile = new quint8[rgbaFloat32bitcolorSpace->pixelSize() * tileWidth * tileHeight];
qint32 dataY = 0;
qint32 imageX = x;
qint32 imageY = y;
it->moveTo(imageX, imageY);
qint32 rowsRemaining = height;
while (rowsRemaining > 0) {
qint32 dataX = 0;
imageX = x;
qint32 columnsRemaining = width;
qint32 numContiguousImageRows = it->numContiguousRows(imageY);
qint32 rowsToWork = qMin(numContiguousImageRows, rowsRemaining);
qint32 convertedTileY = tileHeight - rowsToWork;
Q_ASSERT(convertedTileY >= 0);
while (columnsRemaining > 0) {
qint32 numContiguousImageColumns = it->numContiguousColumns(imageX);
qint32 columnsToWork = qMin(numContiguousImageColumns, columnsRemaining);
qint32 convertedTileX = tileWidth - columnsToWork;
Q_ASSERT(convertedTileX >= 0);
const qint32 dataIdx = dataX + dataY * width;
const qint32 dstTileIndex = convertedTileX + convertedTileY * tileWidth;
const qint32 tileRowStride = (tileWidth - columnsToWork) * dstPixelSize;
const qint32 srcTileRowStride = (tileWidth - columnsToWork) * srcPixelSize;
it->moveTo(imageX, imageY);
quint8 *tileItStart = dstTile + dstTileIndex * dstPixelSize;
// transform tile row by row
quint8 *dstTileIt = tileItStart;
quint8 *srcTileIt = const_cast<quint8*>(it->rawDataConst());
qint32 row = rowsToWork;
while (row > 0)
{
pixelToGmicPixelFormat->transform(srcTileIt, dstTileIt , columnsToWork);
srcTileIt += columnsToWork * srcPixelSize;
srcTileIt += srcTileRowStride;
dstTileIt += columnsToWork * dstPixelSize;
dstTileIt += tileRowStride;
row--;
}
// here we want to copy floats to dstTile, so tileItStart has to point to float buffer
qint32 channelSize = sizeof(float);
for(qint32 i=0; i<numChannels;i++)
{
float * planeIt = planes[i] + dataIdx;
qint32 dataStride = (width - columnsToWork);
quint8* tileIt = tileItStart;
for (qint32 row = 0; row < rowsToWork; row++) {
for (int col = 0; col < columnsToWork; col++) {
memcpy(planeIt, tileIt, channelSize);
tileIt += dstPixelSize;
planeIt += 1;
}
tileIt += tileRowStride;
planeIt += dataStride;
}
// skip channel in tile: red, green, blue, alpha
tileItStart += channelSize;
}
imageX += columnsToWork;
dataX += columnsToWork;
columnsRemaining -= columnsToWork;
}
imageY += rowsToWork;
dataY += rowsToWork;
rowsRemaining -= rowsToWork;
}
delete [] dstTile;
delete pixelToGmicPixelFormat;
}
void KisGmicSimpleConvertor::convertFromGmicFast(gmic_image<float>& gmicImage, KisPaintDeviceSP dst, float gmicUnitValue)
{
const KoColorSpace * dstColorSpace = dst->colorSpace();
KoColorTransformation * gmicToDstPixelFormat = createTransformationFromGmic(dstColorSpace,gmicImage._spectrum,gmicUnitValue);
if (gmicToDstPixelFormat == 0)
{
dbgPlugins << "Fall-back to slow color conversion";
convertFromGmicImage(gmicImage, dst, gmicUnitValue);
return;
}
qint32 x = 0;
qint32 y = 0;
qint32 width = gmicImage._width;
qint32 height = gmicImage._height;
width = width < 0 ? 0 : width;
height = height < 0 ? 0 : height;
const KoColorSpace *rgbaFloat32bitcolorSpace = KoColorSpaceRegistry::instance()->colorSpace(RGBAColorModelID.id(),
Float32BitsColorDepthID.id(),
KoColorSpaceRegistry::instance()->rgb8()->profile());
// this function always convert to rgba or rgb with various color depth
quint32 dstNumChannels = rgbaFloat32bitcolorSpace->channelCount();
// number of channels that we will copy
quint32 numChannels = gmicImage._spectrum;
// gmic image has 4, 3, 2, 1 channel
QVector<float *> planes(dstNumChannels);
int channelOffset = gmicImage._width * gmicImage._height;
for (unsigned int channelIndex = 0; channelIndex < gmicImage._spectrum; channelIndex++)
{
planes[channelIndex] = gmicImage._data + channelOffset * channelIndex;
}
for (unsigned int channelIndex = gmicImage._spectrum; channelIndex < dstNumChannels; channelIndex++)
{
planes[channelIndex] = 0; //turn off
}
qint32 dataY = 0;
qint32 imageY = y;
qint32 rowsRemaining = height;
const qint32 floatPixelSize = rgbaFloat32bitcolorSpace->pixelSize();
KisRandomAccessorSP it = dst->createRandomAccessorNG(dst->x(), dst->y()); // 0,0
int tileWidth = it->numContiguousColumns(dst->x());
int tileHeight = it->numContiguousRows(dst->y());
Q_ASSERT(tileWidth == 64);
Q_ASSERT(tileHeight == 64);
quint8 * convertedTile = new quint8[rgbaFloat32bitcolorSpace->pixelSize() * tileWidth * tileHeight];
// grayscale and rgb case does not have alpha, so let's fill 4th channel of rgba tile with opacity opaque
if (gmicImage._spectrum == 1 || gmicImage._spectrum == 3)
{
quint32 nPixels = tileWidth * tileHeight;
quint32 pixelIndex = 0;
KoRgbF32Traits::Pixel* srcPixel = reinterpret_cast<KoRgbF32Traits::Pixel*>(convertedTile);
while (pixelIndex < nPixels)
{
srcPixel->alpha = gmicUnitValue;
++srcPixel;
++pixelIndex;
}
}
while (rowsRemaining > 0) {
qint32 dataX = 0;
qint32 imageX = x;
qint32 columnsRemaining = width;
qint32 numContiguousImageRows = it->numContiguousRows(imageY);
qint32 rowsToWork = qMin(numContiguousImageRows, rowsRemaining);
while (columnsRemaining > 0) {
qint32 numContiguousImageColumns = it->numContiguousColumns(imageX);
qint32 columnsToWork = qMin(numContiguousImageColumns, columnsRemaining);
const qint32 dataIdx = dataX + dataY * width;
const qint32 tileRowStride = (tileWidth - columnsToWork) * floatPixelSize;
quint8 *tileItStart = convertedTile;
// copy gmic channels to float tile
qint32 channelSize = sizeof(float);
for(quint32 i=0; i<numChannels; i++)
{
float * planeIt = planes[i] + dataIdx;
qint32 dataStride = (width - columnsToWork);
quint8* tileIt = tileItStart;
for (qint32 row = 0; row < rowsToWork; row++) {
for (int col = 0; col < columnsToWork; col++) {
memcpy(tileIt, planeIt, channelSize);
tileIt += floatPixelSize;
planeIt += 1;
}
tileIt += tileRowStride;
planeIt += dataStride;
}
tileItStart += channelSize;
}
it->moveTo(imageX, imageY);
quint8 *dstTileItStart = it->rawData();
tileItStart = convertedTile; // back to the start of the converted tile
// copy float tile to dst colorspace based on input colorspace (rgb or grayscale)
for (qint32 row = 0; row < rowsToWork; row++)
{
gmicToDstPixelFormat->transform(tileItStart, dstTileItStart, columnsToWork);
dstTileItStart += dstColorSpace->pixelSize() * tileWidth;
tileItStart += floatPixelSize * tileWidth;
}
imageX += columnsToWork;
dataX += columnsToWork;
columnsRemaining -= columnsToWork;
}
imageY += rowsToWork;
dataY += rowsToWork;
rowsRemaining -= rowsToWork;
}
delete [] convertedTile;
delete gmicToDstPixelFormat;
}
void KisUniformColorSource::darken(qint32 v)
{
KoColorTransformation* transfo = m_color->colorSpace()->createDarkenAdjustment(v, false, 0.0);
transfo->transform(m_color->data(), m_color->data(), 1);
delete transfo;
}