KoColorSpace.cpp 13.9 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 21

#include "KoColorSpace.h"

22 23
#include <QThreadStorage>
#include <QByteArray>
24
#include <QBitArray>
25

26
#include "KoChannelInfo.h"
Boudewijn Rempt's avatar
Boudewijn Rempt committed
27
#include "DebugPigment.h"
28
#include "KoCompositeOp.h"
Cyrille Berger's avatar
Cyrille Berger committed
29
#include "KoColorTransformation.h"
30 31
#include "KoColorTransformationFactory.h"
#include "KoColorTransformationFactoryRegistry.h"
32
#include "KoColorConversionCache.h"
33 34
#include "KoColorConversionSystem.h"
#include "KoColorSpaceRegistry.h"
35
#include "KoColorProfile.h"
36
#include "KoCopyColorConversionTransformation.h"
37
#include "KoFallBackColorTransformation.h"
38
#include "KoUniqueNumberForIdServer.h"
39

40
#include "KoColorSpace_p.h"
41

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

47 48
KoColorSpace::KoColorSpace(const QString &id, const QString &name, KoMixColorsOp* mixColorsOp, KoConvolutionOp* convolutionOp)
        : d(new Private())
49
{
50
    d->id = id;
51
    d->idNumber = KoUniqueNumberForIdServer::instance()->numberForId(d->id);
52 53 54
    d->name = name;
    d->mixColorsOp = mixColorsOp;
    d->convolutionOp = convolutionOp;
55 56 57 58
    d->transfoToRGBA16 = 0;
    d->transfoFromRGBA16 = 0;
    d->transfoToLABA16 = 0;
    d->transfoFromLABA16 = 0;
59
    d->deletability = NotOwnedByRegistry;
60 61
}

62
KoColorSpace::~KoColorSpace()
63
{
64 65
    Q_ASSERT(d->deletability != OwnedByRegistryDoNotDelete);

66
    qDeleteAll(d->compositeOps);
67
    foreach(KoChannelInfo * channel, d->channels) {
Cyrille Berger's avatar
Cyrille Berger committed
68 69
        delete channel;
    }
70 71 72 73 74
    if (d->deletability == NotOwnedByRegistry) {
        KoColorConversionCache* cache = KoColorSpaceRegistry::instance()->colorConversionCache();
        if (cache) {
            cache->colorSpaceIsDestroyed(this);
        }
75
    }
76 77
    delete d->mixColorsOp;
    delete d->convolutionOp;
Cyrille Berger's avatar
Cyrille Berger committed
78 79 80 81
    delete d->transfoToRGBA16;
    delete d->transfoFromRGBA16;
    delete d->transfoToLABA16;
    delete d->transfoFromLABA16;
82 83 84
    delete d;
}

85 86
bool KoColorSpace::operator==(const KoColorSpace& rhs) const
{
87 88
    const KoColorProfile* p1 = rhs.profile();
    const KoColorProfile* p2 = profile();
89
    return d->idNumber == rhs.d->idNumber && ((p1 == p2) || (*p1 == *p2));
90 91
}

92 93 94 95
QString KoColorSpace::id() const
{
    return d->id;
}
96

97 98 99 100
QString KoColorSpace::name() const
{
    return d->name;
}
101

Adrian Page's avatar
q3--  
Adrian Page committed
102
QList<KoChannelInfo *> KoColorSpace::channels() const
103 104 105 106
{
    return d->channels;
}

107 108
QBitArray KoColorSpace::channelFlags(bool color, bool alpha, bool substance, bool substrate) const
{
109 110 111 112 113 114 115 116 117
    QBitArray ba(d->channels.size());
    if (!color && !alpha && !substance && !substrate) return ba;

    for (int i = 0; i < d->channels.size(); ++i) {
        KoChannelInfo * channel = d->channels.at(i);
        if ((color && channel->channelType() == KoChannelInfo::COLOR) ||
                (alpha && channel->channelType() == KoChannelInfo::ALPHA) ||
                (substrate && channel->channelType() == KoChannelInfo::SUBSTRATE) ||
                (substance && channel->channelType() == KoChannelInfo::SUBSTANCE))
118
            ba.setBit(channel->index(), true);
119 120 121 122
    }
    return ba;
}

123 124 125
void KoColorSpace::addChannel(KoChannelInfo * ci)
{
    d->channels.push_back(ci);
126 127
}

128
quint8 *KoColorSpace::allocPixelBuffer(quint32 numPixels, bool clear, quint8 defaultvalue) const
129
{
130 131 132 133 134 135 136 137
    if (numPixels == 0) {
        return 0;
    }
    quint8* buf = new quint8[pixelSize()*numPixels];
    if (clear) {
        memset(buf, defaultvalue, pixelSize() * numPixels);
    }
    return buf;
138 139
}

140 141 142 143 144
bool KoColorSpace::hasCompositeOp(const QString& id) const
{
    return d->compositeOps.contains(id);
}

145
QList<KoCompositeOp*> KoColorSpace::compositeOps() const
146
{
147 148 149 150
    return d->compositeOps.values();
}


151 152
KoMixColorsOp* KoColorSpace::mixColorsOp() const
{
153 154 155 156
    return d->mixColorsOp;
}


157 158
KoConvolutionOp* KoColorSpace::convolutionOp() const
{
159
    return d->convolutionOp;
160 161
}

Cyrille Berger's avatar
Cyrille Berger committed
162
const KoCompositeOp * KoColorSpace::compositeOp(const QString & id) const
163
{
164 165
    if (d->compositeOps.contains(id))
        return d->compositeOps.value(id);
166
    else {
Boudewijn Rempt's avatar
Boudewijn Rempt committed
167
        warnPigment << "Asking for non-existent composite operation " << id << ", returning " << COMPOSITE_OVER;
168
        return d->compositeOps.value(COMPOSITE_OVER);
169
    }
170 171 172 173
}

void KoColorSpace::addCompositeOp(const KoCompositeOp * op)
{
174 175
    if (op->colorSpace()->id() == id()) {
        d->compositeOps.insert(op->id(), const_cast<KoCompositeOp*>(op));
176 177 178
    }
}

179
const KoColorConversionTransformation* KoColorSpace::toLabA16Converter() const
180
{
181 182
    if (!d->transfoToLABA16) {
        d->transfoToLABA16 = KoColorSpaceRegistry::instance()->colorConversionSystem()->createColorConverter(this, KoColorSpaceRegistry::instance()->lab16("")) ;
183
    }
184
    return d->transfoToLABA16;
185
}
186

187
const KoColorConversionTransformation* KoColorSpace::fromLabA16Converter() const
188
{
189 190
    if (!d->transfoFromLABA16) {
        d->transfoFromLABA16 = KoColorSpaceRegistry::instance()->colorConversionSystem()->createColorConverter(KoColorSpaceRegistry::instance()->lab16("") , this) ;
191
    }
192
    return d->transfoFromLABA16;
193
}
194
const KoColorConversionTransformation* KoColorSpace::toRgbA16Converter() const
195
{
196 197
    if (!d->transfoToRGBA16) {
        d->transfoToRGBA16 = KoColorSpaceRegistry::instance()->colorConversionSystem()->createColorConverter(this, KoColorSpaceRegistry::instance()->rgb16("")) ;
198
    }
199
    return d->transfoToRGBA16;
200
}
201
const KoColorConversionTransformation* KoColorSpace::fromRgbA16Converter() const
202
{
203 204
    if (!d->transfoFromRGBA16) {
        d->transfoFromRGBA16 = KoColorSpaceRegistry::instance()->colorConversionSystem()->createColorConverter(KoColorSpaceRegistry::instance()->rgb16("") , this) ;
205
    }
206 207 208 209 210
    return d->transfoFromRGBA16;
}

void KoColorSpace::toLabA16(const quint8 * src, quint8 * dst, quint32 nPixels) const
{
211
    toLabA16Converter()->transform(src, dst, nPixels);
212 213 214 215
}

void KoColorSpace::fromLabA16(const quint8 * src, quint8 * dst, quint32 nPixels) const
{
216
    fromLabA16Converter()->transform(src, dst, nPixels);
217 218 219 220
}

void KoColorSpace::toRgbA16(const quint8 * src, quint8 * dst, quint32 nPixels) const
{
221
    toRgbA16Converter()->transform(src, dst, nPixels);
222 223 224 225
}

void KoColorSpace::fromRgbA16(const quint8 * src, quint8 * dst, quint32 nPixels) const
{
226
    fromRgbA16Converter()->transform(src, dst, nPixels);
227 228
}

229 230
KoColorConversionTransformation* KoColorSpace::createColorConverter(const KoColorSpace * dstColorSpace, KoColorConversionTransformation::Intent renderingIntent) const
{
231
    if (*this == *dstColorSpace) {
232
        return new KoCopyColorConversionTransformation(this);
233
    } else {
234
        return KoColorSpaceRegistry::instance()->colorConversionSystem()->createColorConverter(this, dstColorSpace, renderingIntent);
235
    }
236 237
}

238
bool KoColorSpace::convertPixelsTo(const quint8 * src,
Boudewijn Rempt's avatar
Boudewijn Rempt committed
239 240 241 242
                                   quint8 * dst,
                                   const KoColorSpace * dstColorSpace,
                                   quint32 numPixels,
                                   KoColorConversionTransformation::Intent renderingIntent) const
243
{
244 245
    if (*this == *dstColorSpace) {
        memcpy(dst, src, numPixels * sizeof(quint8) * pixelSize());
246 247 248 249
    } else {
        KoCachedColorConversionTransformation cct = KoColorSpaceRegistry::instance()->colorConversionCache()->cachedConverter(this, dstColorSpace, renderingIntent);
        cct.transformation()->transform(src, dst, numPixels);
    }
250 251
    return true;
}
252

253

Silvio Heinrich's avatar
Silvio Heinrich committed
254
void KoColorSpace::bitBlt(quint8* dst,
255
                          qint32 dststride,
Silvio Heinrich's avatar
Silvio Heinrich committed
256 257
                          const KoColorSpace* srcSpace,
                          const quint8* src,
258
                          qint32 srcRowStride,
Silvio Heinrich's avatar
Silvio Heinrich committed
259
                          const quint8* srcAlphaMask,
260 261 262 263
                          qint32 maskRowStride,
                          quint8 opacity,
                          qint32 rows,
                          qint32 cols,
Silvio Heinrich's avatar
Silvio Heinrich committed
264 265
                          const QString& op,
                          const QBitArray& channelFlags) const
266
{
267
    if (d->compositeOps.contains(op)) {
268
        bitBlt(dst, dststride, srcSpace, src, srcRowStride, srcAlphaMask, maskRowStride, opacity, rows, cols, d->compositeOps.value(op), channelFlags);
269
    } else {
270
        bitBlt(dst, dststride, srcSpace, src, srcRowStride, srcAlphaMask, maskRowStride, opacity, rows, cols, d->compositeOps.value(COMPOSITE_OVER), channelFlags);
271
    }
272

273 274
}

Silvio Heinrich's avatar
Silvio Heinrich committed
275
void KoColorSpace::bitBlt(quint8* dst,
276
                          qint32 dstRowStride,
Silvio Heinrich's avatar
Silvio Heinrich committed
277 278
                          const KoColorSpace* srcSpace,
                          const quint8* src,
279
                          qint32 srcRowStride,
280
                          const quint8 *srcAlphaMask,
281 282 283 284
                          qint32 maskRowStride,
                          quint8 opacity,
                          qint32 rows,
                          qint32 cols,
Silvio Heinrich's avatar
Silvio Heinrich committed
285 286
                          const KoCompositeOp* op,
                          const QBitArray& channelFlags) const
287
{
288
    Q_ASSERT_X(*op->colorSpace() == *this, "KoColorSpace::bitBlt", QString("Composite op is for color space %1 (%2) while this is %3 (%4)").arg(op->colorSpace()->id()).arg(op->colorSpace()->profile()->name()).arg(id()).arg(profile()->name()).toLatin1());
289

290 291 292
    if (rows <= 0 || cols <= 0)
        return;

293
    if (!(*this == *srcSpace)) {
294

Boudewijn Rempt's avatar
Boudewijn Rempt committed
295
        quint32 conversionBufferStride = cols * pixelSize();
296 297
        QVector<quint8> * conversionCache =
            threadLocalConversionCache(rows * conversionBufferStride);
298

Boudewijn Rempt's avatar
Boudewijn Rempt committed
299
        quint8* conversionData = conversionCache->data();
300 301 302

        for (qint32 row = 0; row < rows; row++) {
            srcSpace->convertPixelsTo(src + row * srcRowStride,
303
                                      conversionData + row * conversionBufferStride,
Boudewijn Rempt's avatar
Boudewijn Rempt committed
304
                                      this, cols);
305 306
        }

307 308 309 310
        op->composite(dst, dstRowStride,
                      conversionData, conversionBufferStride,
                      srcAlphaMask, maskRowStride,
                      rows,  cols,
311
                      opacity, channelFlags);
312 313 314 315 316
    } else {
        op->composite(dst, dstRowStride,
                      src, srcRowStride,
                      srcAlphaMask, maskRowStride,
                      rows,  cols,
317
                      opacity, channelFlags);
318
    }
Boudewijn Rempt's avatar
Boudewijn Rempt committed
319

320 321
}

322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347
void KoColorSpace::bitBlt(const KoColorSpace* srcSpace, const KoCompositeOp::ParameterInfo& params, const KoCompositeOp* op) const
{
    Q_ASSERT_X(*op->colorSpace() == *this, "KoColorSpace::bitBlt", QString("Composite op is for color space %1 (%2) while this is %3 (%4)").arg(op->colorSpace()->id()).arg(op->colorSpace()->profile()->name()).arg(id()).arg(profile()->name()).toLatin1());
    
    if(params.rows <= 0 || params.cols <= 0)
        return;
    
    if(!(*this == *srcSpace)) {
        quint32           conversionBufferStride = params.cols * pixelSize();
        QVector<quint8> * conversionCache        = threadLocalConversionCache(params.rows * conversionBufferStride);
        quint8*           conversionData         = conversionCache->data();
        
        for(qint32 row=0; row<params.rows; row++) {
            srcSpace->convertPixelsTo(params.srcRowStart + row*params.srcRowStride    ,
                                      conversionData     + row*conversionBufferStride, this, params.cols);
        }
        
        KoCompositeOp::ParameterInfo paramInfo(params);
        paramInfo.srcRowStart  = conversionData;
        paramInfo.srcRowStride = conversionBufferStride;
        op->composite(paramInfo);
    }
    else op->composite(params);
}


348 349 350
QVector<quint8> * KoColorSpace::threadLocalConversionCache(quint32 size) const
{
    QVector<quint8> * ba = 0;
351 352 353 354
    if (!d->conversionCache.hasLocalData()) {
        ba = new QVector<quint8>(size, '0');
        d->conversionCache.setLocalData(ba);
    } else {
355
        ba = d->conversionCache.localData();
356 357
        if ((quint8)ba->size() < size)
            ba->resize(size);
358 359 360
    }
    return ba;
}
361

362
KoColorTransformation* KoColorSpace::createColorTransformation(const QString & id, const QHash<QString, QVariant> & parameters) const
363
{
364 365 366
    KoColorTransformationFactory* factory = KoColorTransformationFactoryRegistry::instance()->get(id);
    if (!factory) return 0;
    QPair<KoID, KoID> model(colorModelId(), colorDepthId());
367
    QList< QPair<KoID, KoID> > models = factory->supportedModels();
368 369
    if (models.isEmpty() || models.contains(model)) {
        return factory->createTransformation(this, parameters);
370 371
    } else {
        // Find the best solution
372
        // TODO use the color conversion cache
373 374 375 376 377 378
        KoColorConversionTransformation* csToFallBack = 0;
        KoColorConversionTransformation* fallBackToCs = 0;
        KoColorSpaceRegistry::instance()->colorConversionSystem()->createColorConverters(this, models, csToFallBack, fallBackToCs);
        Q_ASSERT(csToFallBack);
        Q_ASSERT(fallBackToCs);
        KoColorTransformation* transfo = factory->createTransformation(fallBackToCs->srcColorSpace(), parameters);
379
        return new KoFallBackColorTransformation(csToFallBack, fallBackToCs, transfo);
380 381
    }
}
382

383
QImage KoColorSpace::convertToQImage(const quint8 *data, qint32 width, qint32 height,
Boudewijn Rempt's avatar
Boudewijn Rempt committed
384 385
                                     const KoColorProfile *dstProfile,
                                     KoColorConversionTransformation::Intent renderingIntent) const
386 387 388 389 390 391 392 393 394 395 396

{
    QImage img = QImage(width, height, QImage::Format_ARGB32);

    const KoColorSpace * dstCS = KoColorSpaceRegistry::instance()->rgb8(dstProfile);

    if (data)
        this->convertPixelsTo(const_cast<quint8 *>(data), img.bits(), dstCS, width * height, renderingIntent);

    return img;
}