/*
 * Copyright 2018 Google Inc.
 *
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#include "SkColorSpaceXformSteps.h"

// TODO: explain

SkColorSpaceXformSteps::SkColorSpaceXformSteps(SkColorSpace* src, SkAlphaType srcAT,
                                               SkColorSpace* dst) {
    // Set all bools to false, all floats to 0.0f.
    memset(this, 0, sizeof(*this));

    // We have some options about what to do with null src or dst here.
    SkASSERT(src && dst);

    this->unpremul        = srcAT == kPremul_SkAlphaType;
    this->linearize       = !src->gammaIsLinear();
    this->gamut_transform = src->toXYZD50Hash() != dst->toXYZD50Hash();
    this->encode          = !dst->gammaIsLinear();
    this->premul          = srcAT != kOpaque_SkAlphaType;

    if (this->gamut_transform && src->toXYZD50() && dst->fromXYZD50()) {
        auto xform = SkMatrix44(*src->toXYZD50(),  *dst->fromXYZD50());
        if (xform.get(3,0) == 0 && xform.get(3,1) == 0 && xform.get(3,2) == 0 &&
            xform.get(3,3) == 1 &&
            xform.get(0,3) == 0 && xform.get(1,3) == 0 && xform.get(2,3) == 0) {

            for (int r = 0; r < 3; r++)
            for (int c = 0; c < 3; c++) {
                this->src_to_dst_matrix[3*r+c] = xform.get(r,c);
            }
        }
    }

    // Fill out all the transfer functions we'll use:
    SkColorSpaceTransferFn srcTF, dstTF;
    SkAssertResult(src->isNumericalTransferFn(&srcTF));
    SkAssertResult(dst->isNumericalTransferFn(&dstTF));
    this->srcTFInv = srcTF.invert();
    this->dstTF    = dstTF;

    // If we linearize then immediately reencode with the same transfer function, skip both.
    if ( this->linearize       &&
        !this->gamut_transform &&
         this->encode          &&
        0 == memcmp(&srcTF, &dstTF, sizeof(SkColorSpaceTransferFn)))
    {
        this->linearize  = false;
        this->encode     = false;
    }

    // Skip unpremul...premul if there are no non-linear operations between.
    if ( this->unpremul   &&
        !this->linearize  &&
        !this->encode     &&
         this->premul)
    {
        this->unpremul = false;
        this->premul   = false;
    }
}