KoColorSpace.cpp 9.91 KB
Newer Older
1 2 3
/*
 *  Copyright (c) 2005 Boudewijn Rempt <boud@valdyas.org>
 *
4
 * This library is free software; you can redistribute it and/or
5
 * modify it under the terms of the GNU Lesser General Public
6 7
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
8
 *
9 10 11
 * This library 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
12
 * Lesser General Public License for more details.
13
 *
14
 * You should have received a copy of the GNU Lesser General Public License
15 16 17 18
 * along with this library; see the file COPYING.LIB.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
*/
19 20
#include <QThreadStorage>
#include <QByteArray>
21

22
#include "KoColorSpace.h"
23 24 25

#include <kdebug.h>

26
#include "KoCompositeOp.h"
Cyrille Berger's avatar
Cyrille Berger committed
27
#include "KoColorTransformation.h"
28

29 30 31 32 33 34 35 36
struct KoColorSpace::Private {
    QString id;
    QString name;
    QHash<QString, KoCompositeOp *> compositeOps;
    KoColorSpaceRegistry * parent;
    Q3ValueVector<KoChannelInfo *> channels;
    KoMixColorsOp* mixColorsOp;
    KoConvolutionOp* convolutionOp;
37 38 39
    QThreadStorage< QVector<quint8>* > conversionCache;

    DWORD cmType;  // The colorspace type as defined by littlecms
40
    icColorSpaceSignature colorSpaceSignature; // The colorspace signature as defined in icm/icc files
41 42
};

Stefan Nikolaus's avatar
Stefan Nikolaus committed
43 44 45 46 47
KoColorSpace::KoColorSpace()
    : d (new Private())
{
}

48
KoColorSpace::KoColorSpace(const QString &id, const QString &name, KoColorSpaceRegistry * parent, KoMixColorsOp* mixColorsOp, KoConvolutionOp* convolutionOp, DWORD cmType, icColorSpaceSignature colorSpaceSignature )
49
    : d (new Private())
50
{
51 52 53 54 55 56 57
    d->id = id;
    d->name = name;
    d->parent = parent;
    d->mixColorsOp = mixColorsOp;
    d->convolutionOp = convolutionOp;
    d->cmType = cmType;
    d->colorSpaceSignature = colorSpaceSignature;
58 59
}

60
KoColorSpace::~KoColorSpace()
61
{
62 63 64 65 66 67 68 69 70
    delete d->mixColorsOp;
    delete d->convolutionOp;
    delete d;
}

QString KoColorSpace::id() const {return d->id;}

QString KoColorSpace::name() const {return d->name;}

71 72 73
quint32 KoColorSpace::colorSpaceType() const { return d->cmType; }
icColorSpaceSignature KoColorSpace::colorSpaceSignature() const { return d->colorSpaceSignature; }

74 75 76 77 78 79 80 81
Q3ValueVector<KoChannelInfo *> KoColorSpace::channels() const
{
    return d->channels;
}

void KoColorSpace::addChannel(KoChannelInfo * ci)
{
    d->channels.push_back(ci);
82 83
}

84
quint8 *KoColorSpace::allocPixelBuffer(quint32 numPixels) const
85 86 87 88
{
    return new quint8[pixelSize()*numPixels];
}

89
QList<KoCompositeOp*> KoColorSpace::userVisiblecompositeOps() const
90
{
91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
    return d->compositeOps.values();
}

void KoColorSpace::mixColors(const quint8 **colors, const quint8 *weights, quint32 nColors, quint8 *dst) const
{
    Q_ASSERT(d->mixColorsOp);
    d->mixColorsOp->mixColors(colors, weights, nColors, dst);
}

KoMixColorsOp* KoColorSpace::mixColorsOp() const {
    return d->mixColorsOp;
}

void KoColorSpace::convolveColors(quint8** colors, qint32* kernelValues, KoChannelInfo::enumChannelFlags channelFlags, quint8 *dst, qint32 factor, qint32 offset, qint32 nPixels) const
{
    Q_ASSERT(d->convolutionOp);
    d->convolutionOp->convolveColors(colors, kernelValues, channelFlags, dst, factor, offset, nPixels);
}

KoConvolutionOp* KoColorSpace::convolutionOp() const {
    return d->convolutionOp;
112 113
}

Cyrille Berger's avatar
Cyrille Berger committed
114
const KoCompositeOp * KoColorSpace::compositeOp(const QString & id) const
115
{
116 117
    if ( d->compositeOps.contains( id ) )
        return d->compositeOps.value( id );
118
    else
119
        return d->compositeOps.value( COMPOSITE_OVER );
120 121 122 123 124
}

void KoColorSpace::addCompositeOp(const KoCompositeOp * op)
{
    if ( op->colorSpace()->id() == id()) {
125
        d->compositeOps.insert( op->id(), const_cast<KoCompositeOp*>( op ) );
126 127 128 129
    }
}


130
bool KoColorSpace::convertPixelsTo(const quint8 * src,
131
                                   quint8 * dst,
Cyrille Berger's avatar
Cyrille Berger committed
132
                                   const KoColorSpace * dstColorSpace,
133
                                   quint32 numPixels,
Cyrille Berger's avatar
Cyrille Berger committed
134
                                   qint32 renderingIntent) const
135
{
136
    Q_UNUSED(renderingIntent);
137
    // 4 channels: labA, 2 bytes per lab channel
138
    quint8 *pixels = new quint8[sizeof(quint16)*4*numPixels];
139 140 141 142 143 144 145
    toLabA16(src, pixels,numPixels);
    dstColorSpace->fromLabA16(pixels, dst,numPixels);

    delete [] pixels;

    return true;
}
146 147 148 149 150 151 152 153 154 155 156 157

void KoColorSpace::bitBlt(quint8 *dst,
                          qint32 dststride,
                          KoColorSpace * srcSpace,
                          const quint8 *src,
                          qint32 srcRowStride,
                          const quint8 *srcAlphaMask,
                          qint32 maskRowStride,
                          quint8 opacity,
                          qint32 rows,
                          qint32 cols,
                          const QString & op,
Cyrille Berger's avatar
Cyrille Berger committed
158
                          const QBitArray & channelFlags) const
159
{
160 161
    if ( d->compositeOps.contains( op ) ) {
        bitBlt(dst, dststride, srcSpace, src, srcRowStride, srcAlphaMask, maskRowStride, opacity, rows, cols, d->compositeOps.value( op ), channelFlags);
162 163
    }
    else {
164
        bitBlt(dst, dststride, srcSpace, src, srcRowStride, srcAlphaMask, maskRowStride, opacity, rows, cols, d->compositeOps.value( COMPOSITE_OVER ), channelFlags);
165 166 167 168 169 170 171 172 173 174 175 176 177 178
    }

}

void KoColorSpace::bitBlt(quint8 *dst,
                          qint32 dststride,
                          KoColorSpace * srcSpace,
                          const quint8 *src,
                          qint32 srcRowStride,
                          const quint8 *srcAlphaMask,
                          qint32 maskRowStride,
                          quint8 opacity,
                          qint32 rows,
                          qint32 cols,
Cyrille Berger's avatar
Cyrille Berger committed
179
                          const QString& op) const
180
{
181 182
    if ( d->compositeOps.contains( op ) ) {
        bitBlt(dst, dststride, srcSpace, src, srcRowStride, srcAlphaMask, maskRowStride, opacity, rows, cols, d->compositeOps.value( op ));
183 184
    }
    else {
185
        bitBlt(dst, dststride, srcSpace, src, srcRowStride, srcAlphaMask, maskRowStride, opacity, rows, cols, d->compositeOps.value( COMPOSITE_OVER ) );
186 187
    }
}
188 189

void KoColorSpace::bitBlt(quint8 *dst,
190 191 192 193 194 195 196 197 198 199 200
                          qint32 dststride,
                          KoColorSpace * srcSpace,
                          const quint8 *src,
                          qint32 srcRowStride,
                          const quint8 *srcAlphaMask,
                          qint32 maskRowStride,
                          quint8 opacity,
                          qint32 rows,
                          qint32 cols,
                          const KoCompositeOp * op,
                          const QBitArray & channelFlags) const
201 202 203 204 205 206 207
{
    if (rows <= 0 || cols <= 0)
        return;

    if (this != srcSpace) {
        quint32 len = pixelSize() * rows * cols;

208 209
        QVector<quint8> * conversionCache = threadLocalConversionCache(len);
        quint8* conversionData = conversionCache->data();
210 211 212

        for (qint32 row = 0; row < rows; row++) {
            srcSpace->convertPixelsTo(src + row * srcRowStride,
213
                                      conversionData + row * cols * pixelSize(), this,
214 215 216 217 218 219 220
                                      cols);
        }

        // The old srcRowStride is no longer valid because we converted to the current cs
        srcRowStride = cols * pixelSize();

        op->composite( dst, dststride,
221
                       conversionData, srcRowStride,
222 223 224 225 226 227 228 229 230 231 232 233 234 235
                       srcAlphaMask, maskRowStride,
                       rows,  cols,
                       opacity, channelFlags );

    }
    else {
        op->composite( dst, dststride,
                       src, srcRowStride,
                       srcAlphaMask, maskRowStride,
                       rows,  cols,
                       opacity, channelFlags );
    }
}

236
// XXX: I don't want this code duplication, but also don't want an
237 238 239
//      extra function call in this critical section of code. What to
//      do?
void KoColorSpace::bitBlt(quint8 *dst,
240 241 242 243 244 245 246 247 248 249
                          qint32 dststride,
                          KoColorSpace * srcSpace,
                          const quint8 *src,
                          qint32 srcRowStride,
                          const quint8 *srcAlphaMask,
                          qint32 maskRowStride,
                          quint8 opacity,
                          qint32 rows,
                          qint32 cols,
                          const KoCompositeOp * op) const
250 251 252 253 254 255 256
{
    if (rows <= 0 || cols <= 0)
        return;

    if (this != srcSpace) {
        quint32 len = pixelSize() * rows * cols;

257 258
        QVector<quint8> * conversionCache = threadLocalConversionCache(len);
        quint8* conversionData = conversionCache->data();
259 260 261

        for (qint32 row = 0; row < rows; row++) {
            srcSpace->convertPixelsTo(src + row * srcRowStride,
262
                                      conversionData + row * cols * pixelSize(), this,
263 264 265 266 267 268 269
                                      cols);
        }

        // The old srcRowStride is no longer valid because we converted to the current cs
        srcRowStride = cols * pixelSize();

        op->composite( dst, dststride,
270
                       conversionData, srcRowStride,
271 272 273 274 275 276 277 278 279 280 281 282 283
                       srcAlphaMask, maskRowStride,
                       rows,  cols,
                       opacity);

    }
    else {
        op->composite( dst, dststride,
                       src,srcRowStride,
                       srcAlphaMask, maskRowStride,
                       rows,  cols,
                       opacity);
    }
}
284 285 286 287 288 289 290 291 292 293 294 295 296 297 298

QVector<quint8> * KoColorSpace::threadLocalConversionCache(quint32 size) const
{
    QVector<quint8> * ba = 0;
    if ( !d->conversionCache.hasLocalData() ) {
        ba = new QVector<quint8>( '0', size );
        d->conversionCache.setLocalData( ba );
    }
    else {
        ba = d->conversionCache.localData();
        if ( ( quint8 )ba->size() < size )
            ba->resize( size );
    }
    return ba;
}