KoOptimizedCompositeOpAlphaDarken128.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2016 Thorsten Zachmann <zachmann@kde.org>
 * SPDX-FileCopyrightText: 2022 L. E. Segovia <amy@amyspark.me>
 *
 * SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
#ifndef KOOPTIMIZEDCOMPOSITEOPALPHADARKEN128_H
#define KOOPTIMIZEDCOMPOSITEOPALPHADARKEN128_H
 
#include "KoCompositeOpBase.h"
#include "KoCompositeOpRegistry.h"
#include "KoStreamedMath.h"
#include "KoAlphaDarkenParamsWrapper.h"
 
template<typename channels_type, typename ParamsWrapperT>
struct AlphaDarkenCompositor128 {
    using ParamsWrapper = ParamsWrapperT;
 
    struct Pixel {
        channels_type red;
        channels_type green;
        channels_type blue;
        channels_type alpha;
    };
 
    template<bool haveMask, bool src_aligned, typename _impl>
    static ALWAYS_INLINE void compositeVector(const quint8 *src, quint8 *dst, const quint8 *mask, float opacity, const ParamsWrapper &oparams)
    {
        using float_v = typename KoStreamedMath<_impl>::float_v;
        using float_m = typename float_v::batch_bool_type;
 
        float_v src_c1;
        float_v src_c2;
        float_v src_c3;
        float_v src_alpha;
 
        PixelWrapper<channels_type, _impl> dataWrapper;
        dataWrapper.read(src, src_c1, src_c2, src_c3, src_alpha);
 
        const float_v msk_norm_alpha = [&](){;
            if (haveMask) {
                const float_v uint8Rec1(1.0f / 255.0f);
                const float_v mask_vec = KoStreamedMath<_impl>::fetch_mask_8(mask);
                return mask_vec * uint8Rec1 * src_alpha;
            }
            else {
                return src_alpha;
            }
        }();
 
        // we don't use directly passed value
        Q_UNUSED(opacity);
 
        // instead we use value calculated by ParamsWrapper
        opacity = oparams.opacity;
        const float_v opacity_vec(opacity);
 
        src_alpha = msk_norm_alpha * opacity_vec;
 
        const float_v zeroValue(static_cast<float>(KoColorSpaceMathsTraits<channels_type>::zeroValue));
 
        float_v dst_c1;
        float_v dst_c2;
        float_v dst_c3;
        float_v dst_alpha;
 
        dataWrapper.read(dst, dst_c1, dst_c2, dst_c3, dst_alpha);
 
        const float_m empty_dst_pixels_mask = dst_alpha == zeroValue;
 
        if (!xsimd::all(empty_dst_pixels_mask)) {
            if (xsimd::none(empty_dst_pixels_mask)) {
                dst_c1 = (src_c1 - dst_c1) * src_alpha + dst_c1;
                dst_c2 = (src_c2 - dst_c2) * src_alpha + dst_c2;
                dst_c3 = (src_c3 - dst_c3) * src_alpha + dst_c3;
            }
            else {
                dst_c1 = xsimd::select(empty_dst_pixels_mask, src_c1, (src_c1 - dst_c1) * src_alpha + dst_c1);
                dst_c2 = xsimd::select(empty_dst_pixels_mask, src_c2, (src_c2 - dst_c2) * src_alpha + dst_c2);
                dst_c3 = xsimd::select(empty_dst_pixels_mask, src_c3, (src_c3 - dst_c3) * src_alpha + dst_c3);
            }
        }
        else {
            dst_c1 = src_c1;
            dst_c2 = src_c2;
            dst_c3 = src_c3;
        }
 
        const float_v fullFlowAlpha = [&]() {
            if (oparams.averageOpacity > opacity) {
                const float_v average_opacity_vec(oparams.averageOpacity);
                const float_m fullFlowAlpha_mask = average_opacity_vec > dst_alpha;
                return xsimd::select(fullFlowAlpha_mask,
                                (average_opacity_vec - src_alpha)
                                        * (dst_alpha / average_opacity_vec)
                                    + src_alpha,
                                dst_alpha);
            } else {
                const float_m fullFlowAlpha_mask = opacity_vec > dst_alpha;
                return xsimd::select(
                    fullFlowAlpha_mask,
                    (opacity_vec - dst_alpha) * msk_norm_alpha + dst_alpha,
                    dst_alpha);
            }
        }();
 
        dst_alpha = [&]() {
            if (oparams.flow == 1.0) {
                return fullFlowAlpha;
            }
            else {
                const float_v zeroFlowAlpha = ParamsWrapper::calculateZeroFlowAlpha(src_alpha, dst_alpha);
                const float_v flow_norm_vec(oparams.flow);
                return (fullFlowAlpha - zeroFlowAlpha) * flow_norm_vec + zeroFlowAlpha;
            }
        }();
 
        dataWrapper.write(dst, dst_c1, dst_c2, dst_c3, dst_alpha);
    }
 
    template <bool haveMask, typename _impl>
    static ALWAYS_INLINE void compositeOnePixelScalar(const quint8 *s, quint8 *d, const quint8 *mask, float opacity, const ParamsWrapper &oparams)
    {
        using namespace Arithmetic;
        const qint32 alpha_pos = 3;
 
        const auto *src = reinterpret_cast<const channels_type*>(s);
        auto *dst = reinterpret_cast<channels_type*>(d);
 
        float dstAlphaNorm = dst[alpha_pos];
        PixelWrapper<channels_type, _impl>::normalizeAlpha(dstAlphaNorm);
 
        const float uint8Rec1 = 1.0f / 255.0f;
        float mskAlphaNorm = haveMask ? float(*mask) * uint8Rec1 * src[alpha_pos] : src[alpha_pos];
        PixelWrapper<channels_type, _impl>::normalizeAlpha(mskAlphaNorm);
 
        Q_UNUSED(opacity);
        opacity = oparams.opacity;
 
        const float srcAlphaNorm = mskAlphaNorm * opacity;
 
        if (dstAlphaNorm != 0) {
            dst[0] = PixelWrapper<channels_type, _impl>::lerpMixedUintFloat(dst[0], src[0], srcAlphaNorm);
            dst[1] = PixelWrapper<channels_type, _impl>::lerpMixedUintFloat(dst[1], src[1], srcAlphaNorm);
            dst[2] = PixelWrapper<channels_type, _impl>::lerpMixedUintFloat(dst[2], src[2], srcAlphaNorm);
        } else {
            const auto *s = reinterpret_cast<const Pixel*>(src);
            auto *d = reinterpret_cast<Pixel*>(dst);
            *d = *s;
        }
 
        const float flow = oparams.flow;
        const float averageOpacity = oparams.averageOpacity;
 
        const float fullFlowAlpha = [&]() {
            if (averageOpacity > opacity) {
                return averageOpacity > dstAlphaNorm ? lerp(srcAlphaNorm, averageOpacity, dstAlphaNorm / averageOpacity) : dstAlphaNorm;
            } else {
                return opacity > dstAlphaNorm ? lerp(dstAlphaNorm, opacity, mskAlphaNorm) : dstAlphaNorm;
            }
        }();
 
        dstAlphaNorm = [&]() {
            if (flow == 1.0f) {
                return fullFlowAlpha;
            } else {
                const float zeroFlowAlpha = ParamsWrapper::calculateZeroFlowAlpha(srcAlphaNorm, dstAlphaNorm);
                return lerp(zeroFlowAlpha, fullFlowAlpha, flow);
            }
        }();
 
        PixelWrapper<channels_type, _impl>::denormalizeAlpha(dstAlphaNorm);
        dst[alpha_pos] = PixelWrapper<channels_type, _impl>::roundFloatToUint(dstAlphaNorm);
    }
};
 
template<typename _impl, typename ParamsWrapper>
class KoOptimizedCompositeOpAlphaDarken128Impl : public KoCompositeOp
{
public:
    KoOptimizedCompositeOpAlphaDarken128Impl(const KoColorSpace* cs)
        : KoCompositeOp(cs, COMPOSITE_ALPHA_DARKEN, KoCompositeOp::categoryMix()) {}
 
    using KoCompositeOp::composite;
 
    void composite(const KoCompositeOp::ParameterInfo& params) const override
    {
        if(params.maskRowStart) {
            KoStreamedMath<_impl>::template genericComposite128<true, true, AlphaDarkenCompositor128<float, ParamsWrapper> >(params);
        } else {
            KoStreamedMath<_impl>::template genericComposite128<false, true, AlphaDarkenCompositor128<float, ParamsWrapper> >(params);
        }
    }
};
 
template<typename _impl>
class KoOptimizedCompositeOpAlphaDarkenHard128
    : public KoOptimizedCompositeOpAlphaDarken128Impl<_impl, KoAlphaDarkenParamsWrapperHard>
{
public:
    KoOptimizedCompositeOpAlphaDarkenHard128(const KoColorSpace* cs)
        : KoOptimizedCompositeOpAlphaDarken128Impl<_impl, KoAlphaDarkenParamsWrapperHard>(cs) {}
};
 
template<typename _impl>
class KoOptimizedCompositeOpAlphaDarkenCreamy128
    : public KoOptimizedCompositeOpAlphaDarken128Impl<_impl, KoAlphaDarkenParamsWrapperCreamy>
{
public:
    KoOptimizedCompositeOpAlphaDarkenCreamy128(const KoColorSpace* cs)
        : KoOptimizedCompositeOpAlphaDarken128Impl<_impl, KoAlphaDarkenParamsWrapperCreamy>(cs) {}
};
 
template<typename _impl, typename ParamsWrapper>
class KoOptimizedCompositeOpAlphaDarkenU64Impl : public KoCompositeOp
{
public:
    KoOptimizedCompositeOpAlphaDarkenU64Impl(const KoColorSpace* cs)
        : KoCompositeOp(cs, COMPOSITE_ALPHA_DARKEN, KoCompositeOp::categoryMix()) {}
 
    using KoCompositeOp::composite;
 
    void composite(const KoCompositeOp::ParameterInfo& params) const override
    {
        if(params.maskRowStart) {
            KoStreamedMath<_impl>::template genericComposite64<true, true, AlphaDarkenCompositor128<quint16, ParamsWrapper> >(params);
        } else {
            KoStreamedMath<_impl>::template genericComposite64<false, true, AlphaDarkenCompositor128<quint16, ParamsWrapper> >(params);
        }
    }
};
 
template<typename _impl>
class KoOptimizedCompositeOpAlphaDarkenHardU64
    : public KoOptimizedCompositeOpAlphaDarkenU64Impl<_impl, KoAlphaDarkenParamsWrapperHard>
{
public:
    KoOptimizedCompositeOpAlphaDarkenHardU64(const KoColorSpace* cs)
        : KoOptimizedCompositeOpAlphaDarkenU64Impl<_impl, KoAlphaDarkenParamsWrapperHard>(cs) {}
};
 
template<typename _impl>
class KoOptimizedCompositeOpAlphaDarkenCreamyU64
    : public KoOptimizedCompositeOpAlphaDarkenU64Impl<_impl, KoAlphaDarkenParamsWrapperCreamy>
{
public:
    KoOptimizedCompositeOpAlphaDarkenCreamyU64(const KoColorSpace* cs)
        : KoOptimizedCompositeOpAlphaDarkenU64Impl<_impl, KoAlphaDarkenParamsWrapperCreamy>(cs) {}
};
 
 
#endif // KOOPTIMIZEDCOMPOSITEOPALPHADARKEN128_H