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

// Models two intervals (so 4 colors), that are connected at a specific threshold point.

// Bias and scale for 0 to threshold
layout(ctype=GrColor4f, tracked) in uniform half4 scale01;
layout(ctype=GrColor4f, tracked) in uniform half4 bias01;

// Bias and scale for threshold to 1
layout(ctype=GrColor4f, tracked) in uniform half4 scale23;
layout(ctype=GrColor4f, tracked) in uniform half4 bias23;

layout(tracked) in uniform float threshold;

void main() {
    half t = sk_InColor.x;

    half4 scale, bias;
    if (t < threshold) {
        scale = scale01;
        bias = bias01;
    } else {
        scale = scale23;
        bias = bias23;
    }

    sk_OutColor = t * scale + bias;
}

//////////////////////////////////////////////////////////////////////////////

@make {
    static std::unique_ptr<GrFragmentProcessor> Make(const GrColor4f& c0, const GrColor4f& c1,
                                                     const GrColor4f& c2, const GrColor4f& c3,
                                                     float threshold);
}

@cppEnd {
    std::unique_ptr<GrFragmentProcessor> GrDualIntervalGradientColorizer::Make(
            const GrColor4f& c0, const GrColor4f& c1, const GrColor4f& c2,  const GrColor4f& c3, float threshold) {
        // Derive scale and biases from the 4 colors and threshold
        auto vc0 = Sk4f::Load(c0.fRGBA);
        auto vc1 = Sk4f::Load(c1.fRGBA);
        auto scale01 = (vc1 - vc0) / threshold;
        // bias01 = c0

        auto vc2 = Sk4f::Load(c2.fRGBA);
        auto vc3 = Sk4f::Load(c3.fRGBA);
        auto scale23 = (vc3 - vc2) / (1 - threshold);
        auto bias23 = vc2 - threshold * scale23;

        return std::unique_ptr<GrFragmentProcessor>(new GrDualIntervalGradientColorizer(
                GrColor4f(scale01[0], scale01[1], scale01[2], scale01[3]), c0,
                GrColor4f(scale23[0], scale23[1], scale23[2], scale23[3]),
                GrColor4f(bias23[0], bias23[1], bias23[2], bias23[3]), threshold));
    }
}