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.
345 lines
12 KiB
345 lines
12 KiB
/*
|
|
* Copyright (c) 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_ycbcr_u8_colorspace.h"
|
|
|
|
#include <tqimage.h>
|
|
|
|
#include <kis_integer_maths.h>
|
|
|
|
const TQ_INT32 MAX_CHANNEL_YCbCr = 3;
|
|
const TQ_INT32 MAX_CHANNEL_YCbCrA = 4;
|
|
|
|
KisYCbCrU8ColorSpace::KisYCbCrU8ColorSpace(KisColorSpaceFactoryRegistry* parent, KisProfile* /*p*/)
|
|
: KisU8BaseColorSpace(KisID("YCbCrAU8", i18n("YCbCr (8-bit integer/channel)")), TYPE_YCbCr_8, icSigYCbCrData, parent, 0)
|
|
{
|
|
m_channels.push_back(new KisChannelInfo(i18n("Y"), "Y", PIXEL_Y * sizeof(TQ_UINT8), KisChannelInfo::COLOR, KisChannelInfo::UINT8, sizeof(TQ_UINT8)));
|
|
m_channels.push_back(new KisChannelInfo(i18n("Cb"), "Cb", PIXEL_Cb * sizeof(TQ_UINT8), KisChannelInfo::COLOR, KisChannelInfo::UINT8, sizeof(TQ_UINT8)));
|
|
m_channels.push_back(new KisChannelInfo(i18n("Cr"), "Cr", PIXEL_Cr * sizeof(TQ_UINT8), KisChannelInfo::COLOR, KisChannelInfo::UINT8, sizeof(TQ_UINT8)));
|
|
m_channels.push_back(new KisChannelInfo(i18n("Alpha"), "A", PIXEL_ALPHA * sizeof(TQ_UINT8), KisChannelInfo::ALPHA, KisChannelInfo::UINT8, sizeof(TQ_UINT8)));
|
|
|
|
m_alphaPos = PIXEL_ALPHA * sizeof(TQ_UINT8);
|
|
KisAbstractColorSpace::init();
|
|
}
|
|
|
|
|
|
KisYCbCrU8ColorSpace::~KisYCbCrU8ColorSpace()
|
|
{
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::setPixel(TQ_UINT8 *dst, TQ_UINT8 Y, TQ_UINT8 Cb, TQ_UINT8 Cr, TQ_UINT8 alpha) const
|
|
{
|
|
Pixel *dstPixel = reinterpret_cast<Pixel *>(dst);
|
|
|
|
dstPixel->Y = Y;
|
|
dstPixel->Cb = Cb;
|
|
dstPixel->Cr = Cr;
|
|
dstPixel->alpha = alpha;
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::getPixel(const TQ_UINT8 *src, TQ_UINT8 *Y, TQ_UINT8 *Cb, TQ_UINT8 *Cr, TQ_UINT8 *alpha) const
|
|
{
|
|
const Pixel *srcPixel = reinterpret_cast<const Pixel *>(src);
|
|
|
|
*Y = srcPixel->Y;
|
|
*Cb = srcPixel->Cb;
|
|
*Cr = srcPixel->Cr;
|
|
*alpha = srcPixel->alpha;
|
|
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::fromTQColor(const TQColor& c, TQ_UINT8 *dstU8, KisProfile * profile )
|
|
{
|
|
if(getProfile())
|
|
{
|
|
KisU8BaseColorSpace::fromTQColor(c, dstU8, profile);
|
|
} else {
|
|
Pixel *dst = reinterpret_cast<Pixel *>(dstU8);
|
|
dst->Y = computeY( c.red(), c.green(), c.blue());
|
|
dst->Cb = computeCb( c.red(), c.green(), c.blue());
|
|
dst->Cr = computeCr( c.red(), c.green(), c.blue());
|
|
}
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::fromTQColor(const TQColor& c, TQ_UINT8 opacity, TQ_UINT8 *dstU8, KisProfile * profile )
|
|
{
|
|
if(getProfile())
|
|
{
|
|
KisU8BaseColorSpace::fromTQColor(c, opacity, dstU8, profile);
|
|
} else {
|
|
Pixel *dst = reinterpret_cast<Pixel *>(dstU8);
|
|
dst->Y = computeY( c.red(), c.green(), c.blue());
|
|
dst->Cb = computeCb( c.red(), c.green(), c.blue());
|
|
dst->Cr = computeCr( c.red(), c.green(), c.blue());
|
|
dst->alpha = opacity;
|
|
}
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::toTQColor(const TQ_UINT8 *srcU8, TQColor *c, KisProfile * profile)
|
|
{
|
|
if(getProfile())
|
|
{
|
|
KisU8BaseColorSpace::toTQColor(srcU8, c, profile);
|
|
|
|
} else {
|
|
const Pixel *src = reinterpret_cast<const Pixel *>(srcU8);
|
|
c->setRgb(computeRed(src->Y,src->Cb,src->Cr), computeGreen(src->Y,src->Cb,src->Cr), computeBlue(src->Y,src->Cb,src->Cr));
|
|
}
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::toTQColor(const TQ_UINT8 *srcU8, TQColor *c, TQ_UINT8 *opacity, KisProfile * profile)
|
|
{
|
|
if(getProfile())
|
|
{
|
|
KisU8BaseColorSpace::toTQColor(srcU8, c, opacity, profile);
|
|
} else {
|
|
const Pixel *src = reinterpret_cast<const Pixel *>(srcU8);
|
|
c->setRgb(computeRed(src->Y,src->Cb,src->Cr), computeGreen(src->Y,src->Cb,src->Cr), computeBlue(src->Y,src->Cb,src->Cr));
|
|
*opacity = src->alpha;
|
|
}
|
|
}
|
|
|
|
TQ_UINT8 KisYCbCrU8ColorSpace::difference(const TQ_UINT8 *src1U8, const TQ_UINT8 *src2U8)
|
|
{
|
|
if(getProfile())
|
|
return KisU8BaseColorSpace::difference(src1U8, src2U8);
|
|
const Pixel *src1 = reinterpret_cast<const Pixel *>(src1U8);
|
|
const Pixel *src2 = reinterpret_cast<const Pixel *>(src2U8);
|
|
|
|
return TQMAX(TQABS(src2->Y - src1->Y), TQMAX(TQABS(src2->Cb - src1->Cb), TQABS(src2->Cr - src1->Cr)));
|
|
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::mixColors(const TQ_UINT8 **colors, const TQ_UINT8 *weights, TQ_UINT32 nColors, TQ_UINT8 *dst) const
|
|
{
|
|
TQ_UINT8 totalY = 0, totalCb = 0, totalCr = 0, newAlpha = 0;
|
|
|
|
while (nColors--)
|
|
{
|
|
const Pixel *pixel = reinterpret_cast<const Pixel *>(*colors);
|
|
|
|
TQ_UINT8 alpha = pixel->alpha;
|
|
float alphaTimesWeight = alpha * *weights;
|
|
|
|
totalY += (TQ_UINT8)(pixel->Y * alphaTimesWeight);
|
|
totalCb += (TQ_UINT8)(pixel->Cb * alphaTimesWeight);
|
|
totalCr += (TQ_UINT8)(pixel->Cr * alphaTimesWeight);
|
|
newAlpha += (TQ_UINT8)(alphaTimesWeight);
|
|
|
|
weights++;
|
|
colors++;
|
|
}
|
|
|
|
Pixel *dstPixel = reinterpret_cast<Pixel *>(dst);
|
|
|
|
dstPixel->alpha = newAlpha;
|
|
|
|
if (newAlpha > 0) {
|
|
totalY = totalY / newAlpha;
|
|
totalCb = totalCb / newAlpha;
|
|
totalCr = totalCr / newAlpha;
|
|
}
|
|
|
|
dstPixel->Y = totalY;
|
|
dstPixel->Cb = totalCb;
|
|
dstPixel->Cr = totalCr;
|
|
}
|
|
|
|
TQValueVector<KisChannelInfo *> KisYCbCrU8ColorSpace::channels() const {
|
|
return m_channels;
|
|
}
|
|
|
|
TQ_UINT32 KisYCbCrU8ColorSpace::nChannels() const {
|
|
return MAX_CHANNEL_YCbCrA;
|
|
}
|
|
|
|
TQ_UINT32 KisYCbCrU8ColorSpace::nColorChannels() const {
|
|
return MAX_CHANNEL_YCbCr;
|
|
}
|
|
|
|
TQ_UINT32 KisYCbCrU8ColorSpace::pixelSize() const {
|
|
return MAX_CHANNEL_YCbCrA*sizeof(TQ_UINT8);
|
|
}
|
|
|
|
|
|
TQImage KisYCbCrU8ColorSpace::convertToTQImage(const TQ_UINT8 *data, TQ_INT32 width, TQ_INT32 height, KisProfile * dstProfile, TQ_INT32 renderingIntent, float exposure )
|
|
{
|
|
if(getProfile())
|
|
return KisU8BaseColorSpace::convertToTQImage( data, width, height, dstProfile, renderingIntent, exposure);
|
|
|
|
TQImage img = TQImage(width, height, 32, 0, TQImage::LittleEndian);
|
|
img.setAlphaBuffer(true);
|
|
|
|
TQ_INT32 i = 0;
|
|
uchar *j = img.bits();
|
|
|
|
while ( i < width * height * MAX_CHANNEL_YCbCrA) {
|
|
TQ_UINT8 Y = *( data + i + PIXEL_Y );
|
|
TQ_UINT8 Cb = *( data + i + PIXEL_Cb );
|
|
TQ_UINT8 Cr = *( data + i + PIXEL_Cr );
|
|
*( j + 3) = *( data + i + PIXEL_ALPHA );
|
|
*( j + 2 ) = computeRed(Y,Cb,Cr);
|
|
*( j + 1 ) = computeGreen(Y,Cb,Cr);
|
|
*( j + 0 ) = computeBlue(Y,Cb,Cr);
|
|
i += MAX_CHANNEL_YCbCrA;
|
|
j += 4;
|
|
}
|
|
return img;
|
|
}
|
|
|
|
|
|
void KisYCbCrU8ColorSpace::bitBlt(TQ_UINT8 *dst, TQ_INT32 dstRowStride, const TQ_UINT8 *src, TQ_INT32 srcRowStride, const TQ_UINT8 *srcAlphaMask, TQ_INT32 maskRowStride, TQ_UINT8 opacity, TQ_INT32 rows, TQ_INT32 cols, const KisCompositeOp& op)
|
|
{
|
|
switch (op.op()) {
|
|
case COMPOSITE_UNDEF:
|
|
// Undefined == no composition
|
|
break;
|
|
case COMPOSITE_OVER:
|
|
compositeOver(dst, dstRowStride, src, srcRowStride, srcAlphaMask, maskRowStride, rows, cols, opacity);
|
|
break;
|
|
case COMPOSITE_COPY:
|
|
compositeCopy(dst, dstRowStride, src, srcRowStride, srcAlphaMask, maskRowStride, rows, cols, opacity);
|
|
break;
|
|
case COMPOSITE_ERASE:
|
|
compositeErase(dst, dstRowStride, src, srcRowStride, srcAlphaMask, maskRowStride, rows, cols, opacity);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::compositeOver(TQ_UINT8 *dstRowStart, TQ_INT32 dstRowStride, const TQ_UINT8 *srcRowStart, TQ_INT32 srcRowStride, const TQ_UINT8 *maskRowStart, TQ_INT32 maskRowStride, TQ_INT32 rows, TQ_INT32 numColumns, TQ_UINT8 opacity)
|
|
{
|
|
while (rows > 0) {
|
|
|
|
const TQ_UINT8 *src = srcRowStart;
|
|
TQ_UINT8 *dst = dstRowStart;
|
|
const TQ_UINT8 *mask = maskRowStart;
|
|
TQ_INT32 columns = numColumns;
|
|
|
|
while (columns > 0) {
|
|
|
|
TQ_UINT8 srcAlpha = src[PIXEL_ALPHA];
|
|
|
|
// apply the alphamask
|
|
if (mask != 0) {
|
|
if (*mask != OPACITY_OPAQUE) {
|
|
srcAlpha *= *mask;
|
|
}
|
|
mask++;
|
|
}
|
|
|
|
if (srcAlpha > OPACITY_TRANSPARENT) {
|
|
|
|
if (opacity < OPACITY_OPAQUE) {
|
|
srcAlpha *= opacity;
|
|
}
|
|
|
|
if (srcAlpha == OPACITY_OPAQUE) {
|
|
memcpy(dst, src, MAX_CHANNEL_YCbCrA * sizeof(TQ_UINT8));
|
|
} else {
|
|
TQ_UINT8 dstAlpha = dst[PIXEL_ALPHA];
|
|
|
|
TQ_UINT8 srcBlend;
|
|
|
|
if (dstAlpha == OPACITY_OPAQUE ) {
|
|
srcBlend = srcAlpha;
|
|
} else {
|
|
TQ_UINT8 newAlpha = dstAlpha + (OPACITY_OPAQUE - dstAlpha) * srcAlpha;
|
|
dst[PIXEL_ALPHA] = newAlpha;
|
|
|
|
if (newAlpha > 0) {
|
|
srcBlend = srcAlpha / newAlpha;
|
|
} else {
|
|
srcBlend = srcAlpha;
|
|
}
|
|
}
|
|
|
|
if (srcBlend == OPACITY_OPAQUE) {
|
|
memcpy(dst, src, MAX_CHANNEL_YCbCr * sizeof(TQ_UINT8));
|
|
} else {
|
|
dst[PIXEL_Y] = UINT8_BLEND(src[PIXEL_Y], dst[PIXEL_Y], srcBlend);
|
|
dst[PIXEL_Cb] = UINT8_BLEND(src[PIXEL_Cb], dst[PIXEL_Cb], srcBlend);
|
|
dst[PIXEL_Cr] = UINT8_BLEND(src[PIXEL_Cr], dst[PIXEL_Cr], srcBlend);
|
|
}
|
|
}
|
|
}
|
|
|
|
columns--;
|
|
src += MAX_CHANNEL_YCbCrA;
|
|
dst += MAX_CHANNEL_YCbCrA;
|
|
}
|
|
|
|
rows--;
|
|
srcRowStart += srcRowStride;
|
|
dstRowStart += dstRowStride;
|
|
if(maskRowStart) {
|
|
maskRowStart += maskRowStride;
|
|
}
|
|
}
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::compositeErase(TQ_UINT8 *dst, TQ_INT32 dstRowSize, const TQ_UINT8 *src, TQ_INT32 srcRowSize, const TQ_UINT8 *srcAlphaMask, TQ_INT32 maskRowStride, TQ_INT32 rows, TQ_INT32 cols, TQ_UINT8 /*opacity*/)
|
|
{
|
|
while (rows-- > 0)
|
|
{
|
|
const Pixel *s = reinterpret_cast<const Pixel *>(src);
|
|
Pixel *d = reinterpret_cast<Pixel *>(dst);
|
|
const TQ_UINT8 *mask = srcAlphaMask;
|
|
|
|
for (TQ_INT32 i = cols; i > 0; i--, s++, d++)
|
|
{
|
|
TQ_UINT8 srcAlpha = s -> alpha;
|
|
|
|
// apply the alphamask
|
|
if (mask != 0) {
|
|
if (*mask != OPACITY_OPAQUE) {
|
|
srcAlpha = *mask;
|
|
}
|
|
mask++;
|
|
}
|
|
d -> alpha = srcAlpha * d -> alpha;
|
|
}
|
|
|
|
dst += dstRowSize;
|
|
src += srcRowSize;
|
|
if(srcAlphaMask) {
|
|
srcAlphaMask += maskRowStride;
|
|
}
|
|
}
|
|
}
|
|
|
|
void KisYCbCrU8ColorSpace::compositeCopy(TQ_UINT8 *dstRowStart, TQ_INT32 dstRowStride, const TQ_UINT8 *srcRowStart, TQ_INT32 srcRowStride, const TQ_UINT8 */*mask*/, TQ_INT32 /*maskRowStride*/, TQ_INT32 rows, TQ_INT32 numColumns, TQ_UINT8 /*opacity*/)
|
|
{
|
|
while (rows > 0) {
|
|
memcpy(dstRowStart, srcRowStart, numColumns * sizeof(Pixel));
|
|
--rows;
|
|
srcRowStart += srcRowStride;
|
|
dstRowStart += dstRowStride;
|
|
}
|
|
}
|
|
|
|
KisCompositeOpList KisYCbCrU8ColorSpace::userVisiblecompositeOps() const
|
|
{
|
|
KisCompositeOpList list;
|
|
|
|
list.append(KisCompositeOp(COMPOSITE_OVER));
|
|
return list;
|
|
}
|