/*
 * Copyright (c) 2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include "ge_render.h"

#include "ge_aibar_shader_filter.h"
#include "ge_aurora_noise_shader.h"
#include "ge_bezier_warp_shader_filter.h"
#include "ge_border_light_shader.h"
#include "ge_color_gradient_shader_filter.h"
#include "ge_content_light_shader_filter.h"
#include "ge_contour_diagonal_flow_light_shader.h"
#include "ge_direction_light_shader_filter.h"
#include "ge_displacement_distort_shader_filter.h"
#include "ge_edge_light_shader_filter.h"
#include "ge_external_dynamic_loader.h"
#include "ge_grey_shader_filter.h"
#include "ge_kawase_blur_shader_filter.h"
#include "ge_linear_gradient_blur_shader_filter.h"
#include "ge_hps_effect_filter.h"
#include "ge_log.h"
#include "ge_magnifier_shader_filter.h"
#include "ge_mask_transition_shader_filter.h"
#include "ge_mesa_blur_shader_filter.h"
#include "ge_particle_circular_halo_shader.h"
#include "ge_sound_wave_filter.h"
#include "ge_visual_effect_impl.h"
#include "ge_variable_radius_blur_shader_filter.h"
#include "ge_water_ripple_filter.h"
#include "ge_wavy_ripple_light_shader.h"

namespace OHOS {
namespace GraphicsEffectEngine {
using namespace Rosen::Drawing;
using ShaderCreator = std::function<std::shared_ptr<GEShader>(std::shared_ptr<GEVisualEffectImpl>)>;

static std::unordered_map<GEVisualEffectImpl::FilterType, ShaderCreator> g_shaderCreatorLUT = {
    {GEVisualEffectImpl::FilterType::CONTOUR_DIAGONAL_FLOW_LIGHT, [] (std::shared_ptr<GEVisualEffectImpl> ve)
        {
            std::shared_ptr<GEShader> out = nullptr;
            if (ve == nullptr || ve->GetContenDiagonalParams() == nullptr) {
                return out;
            }
            const auto& params = ve->GetContenDiagonalParams();
            out = std::make_shared<GEContourDiagonalFlowLightShader>(*params);
            return out;
        }
    },
    {GEVisualEffectImpl::FilterType::WAVY_RIPPLE_LIGHT, [] (std::shared_ptr<GEVisualEffectImpl> ve)
        {
            std::shared_ptr<GEShader> out = nullptr;
            if (ve == nullptr || ve->GetWavyRippleLightParams() == nullptr) {
                return out;
            }
            const auto& params = ve->GetWavyRippleLightParams();
            out = std::make_shared<GEWavyRippleLightShader>(*params);
            return out;
        }
    },
    {GEVisualEffectImpl::FilterType::AURORA_NOISE, [] (std::shared_ptr<GEVisualEffectImpl> ve)
        {
            std::shared_ptr<GEShader> out = nullptr;
            if (ve == nullptr || ve->GetAuroraNoiseParams() == nullptr) {
                return out;
            }
            const auto& params = ve->GetAuroraNoiseParams();
            out = GEAuroraNoiseShader::CreateAuroraNoiseShader(*params);
            return out;
        }
    },
    {GEVisualEffectImpl::FilterType::PARTICLE_CIRCULAR_HALO, [] (std::shared_ptr<GEVisualEffectImpl> ve)
        {
            std::shared_ptr<GEShader> out = nullptr;
            if (ve == nullptr || ve->GetParticleCircularHaloParams() == nullptr) {
                return out;
            }
            const auto& params = ve->GetParticleCircularHaloParams();
            out = std::make_shared<GEParticleCircularHaloShader>(*params);
            return out;
        }
    },
    {GEVisualEffectImpl::FilterType::COLOR_GRADIENT_EFFECT, [] (std::shared_ptr<GEVisualEffectImpl> ve)
        {
            std::shared_ptr<GEShader> out = nullptr;
            if (ve == nullptr) {
                return out;
            }
            const auto& params = ve->GetColorGradientEffectParams();
            if (params == nullptr) {
                return out;
            }
            auto type = static_cast<uint32_t>(Drawing::GEVisualEffectImpl::FilterType::COLOR_GRADIENT_EFFECT);
            auto impl = GEExternalDynamicLoader::GetInstance().CreateGEXObjectByType(
                type, sizeof(GEXColorGradientEffectParams), static_cast<void*>(params.get()));
            if (!impl) {
                GE_LOGE("GEXColorGradientEffect::CreateDynamicImpl create object failed.");
                return out;
            }
            std::shared_ptr<GEShader> dmShader(static_cast<GEShader*>(impl));
            return dmShader;
        }
    },
    {GEVisualEffectImpl::FilterType::LIGHT_CAVE, [] (std::shared_ptr<GEVisualEffectImpl> ve)
        {
            std::shared_ptr<GEShader> out = nullptr;
            if (ve == nullptr) {
                return out;
            }
            const auto& params = ve->GetLightCaveParams();
            if (params == nullptr) {
                return out;
            }
            auto type = static_cast<uint32_t>(Drawing::GEVisualEffectImpl::FilterType::LIGHT_CAVE);
            auto impl = GEExternalDynamicLoader::GetInstance().CreateGEXObjectByType(
                type, sizeof(GEXLightCaveShaderParams), static_cast<void*>(params.get()));
            if (!impl) {
                GE_LOGE("GEXLightCaveShader::CreateDynamicImpl create object failed.");
                return out;
            }
            std::shared_ptr<GEShader> dmShader(static_cast<GEShader*>(impl));
            return dmShader;
        }
    },
    {GEVisualEffectImpl::FilterType::BORDER_LIGHT, [] (std::shared_ptr<GEVisualEffectImpl> ve)
        {
            std::shared_ptr<GEShader> out = nullptr;
            if (ve == nullptr || ve->GetBorderLightParams() == nullptr) {
                return out;
            }
            const auto& params = ve->GetBorderLightParams();
            out = std::make_shared<GEBorderLightShader>(*params);
            return out;
        }
    }
};

GERender::GERender() {}

GERender::~GERender() {}

void GERender::DrawImageEffect(Drawing::Canvas& canvas, Drawing::GEVisualEffectContainer& veContainer,
    const std::shared_ptr<Drawing::Image>& image, const Drawing::Rect& src, const Drawing::Rect& dst,
    const Drawing::SamplingOptions& sampling)
{
    if (!image) {
        LOGE("GERender::DrawImageRect image is null");
        return;
    }

    auto resImage = ApplyImageEffect(canvas, veContainer, image, src, dst, sampling);
    Drawing::Brush brush;
    canvas.AttachBrush(brush);
    canvas.DrawImageRect(*resImage, src, dst, Drawing::SamplingOptions());
    canvas.DetachBrush();
}

std::shared_ptr<Drawing::Image> GERender::ApplyImageEffect(Drawing::Canvas& canvas,
    Drawing::GEVisualEffectContainer& veContainer, const std::shared_ptr<Drawing::Image>& image,
    const Drawing::Rect& src, const Drawing::Rect& dst, const Drawing::SamplingOptions& sampling)
{
    if (!image) {
        LOGE("GERender::ApplyImageEffect image is null");
        return nullptr;
    }
    std::vector<std::shared_ptr<GEShaderFilter>> geShaderFilters;
    auto resImage = image;
    for (auto vef : veContainer.GetFilters()) {
        if (vef == nullptr) {
            LOGD("GERender::ApplyImageEffect vef is null");
            continue;
        }
        auto ve = vef->GetImpl();
        std::shared_ptr<GEShaderFilter> geShaderFilter = GenerateShaderFilter(vef);
        if (geShaderFilter == nullptr) {
            LOGD("GERender::ApplyImageEffect filter is null");
            continue;
        }
        geShaderFilter->SetSupportHeadroom(vef->GetSupportHeadroom());
        geShaderFilter->SetCache(ve->GetCache());
        geShaderFilter->Preprocess(canvas, src, dst);
        resImage = geShaderFilter->ProcessImage(canvas, resImage, src, dst);
        ve->SetCache(geShaderFilter->GetCache());
    }

    return resImage;
}

bool GERender::ExcuteRangeEmpty(Drawing::Canvas& canvas, Drawing::GEVisualEffectContainer& veContainer,
    const HpsGEImageEffectContext& context, std::shared_ptr<Drawing::Image>& outImage, Drawing::Brush& brush)
{
    auto visualEffects = veContainer.GetFilters();
    if (visualEffects.empty()) {
        return false;
    }
    HpsGEImageEffectContext fullGEContext = { context.image, context.src, context.src, context.sampling, true,
        context.alpha, context.colorFilter, context.maskColor, context.saturationForHPS, context.brightnessForHPS };
    bool status = ApplyGEEffects(canvas, visualEffects, fullGEContext, outImage);
    if (status) {
        DrawToCanvas(canvas, context, outImage, brush);
    }
    return status;
}

bool GERender::ApplyHpsGEImageEffect(Drawing::Canvas& canvas, Drawing::GEVisualEffectContainer& veContainer,
    const HpsGEImageEffectContext& context, std::shared_ptr<Drawing::Image>& outImage, Drawing::Brush& brush)
{
    if (!context.image) {
        LOGE("GERender::ApplyHpsGEImageEffect image is null");
        return false;
    }
    auto visualEffects = veContainer.GetFilters();
    if (visualEffects.empty()) {
        return false;
    }

    auto hpsEffectFilter = std::make_shared<HpsEffectFilter>(canvas);
    std::vector<IndexRange> hpsSupportedIndexRanges = hpsEffectFilter->HpsSupportedEffectsIndexRanges(visualEffects);
    if (hpsSupportedIndexRanges.empty()) {
        return ExcuteRangeEmpty(canvas, veContainer, context, outImage, brush);
    }

    auto resImage = context.image;
    bool appliedBlur = false; // compatibility with HPS
    bool lastAppliedGE = true;
    auto indexRangeInfos = CategorizeRanges(hpsSupportedIndexRanges, visualEffects.size());
    for (auto& indexRangeInfo : indexRangeInfos) {
        std::vector<std::shared_ptr<Drawing::GEVisualEffect>> subVisualEffects(
            visualEffects.begin() + indexRangeInfo.range[0], visualEffects.begin() + indexRangeInfo.range[1] + 1);
        auto currentImage = resImage;
        if (indexRangeInfo.mode == EffectMode::GE) {
            HpsGEImageEffectContext partialGEContext = { currentImage, context.src, context.src,
                context.sampling, false, context.alpha, context.colorFilter, context.maskColor,
                context.saturationForHPS, context.brightnessForHPS };
            // When hps support exists, don't set compatibility switch
            ApplyGEEffects(canvas, subVisualEffects, partialGEContext, resImage);
            lastAppliedGE = true;
            continue;
        }
        for (auto vef : subVisualEffects) {
            if (vef == nullptr) {
                continue;
            }
            auto ve = vef->GetImpl();
            hpsEffectFilter->GenerateVisualEffectFromGE(ve, context.src, context.dst, context.saturationForHPS,
                context.brightnessForHPS, context.image);
        }
        HpsEffectFilter::HpsEffectContext hpsEffectContext = {context.alpha, context.colorFilter, context.maskColor};
        appliedBlur = hpsEffectFilter->ApplyHpsEffect(canvas, currentImage, resImage, hpsEffectContext);
        lastAppliedGE = false;
    }
    outImage = resImage;
    if (lastAppliedGE) { // ApplyHpsEffect have done canvas sync
        DrawToCanvas(canvas, context, outImage, brush);
    }
    return appliedBlur;
}

void GERender::DrawToCanvas(Drawing::Canvas& canvas, const HpsGEImageEffectContext& context,
                            std::shared_ptr<Drawing::Image>& outImage, Drawing::Brush& brush)
{
    if (outImage != nullptr) {
        canvas.AttachBrush(brush);
        canvas.DrawImageRect(*outImage, context.src, context.dst, context.sampling);
        canvas.DetachBrush();
    }
}

std::vector<GERender::IndexRangeInfo> GERender::CategorizeRanges(
    const std::vector<IndexRange>& hpsIndexRanges, const int32_t veContainerSize)
{
    std::vector<IndexRangeInfo> categorizedRanges;
    // The first GE range before the first hps range
    if (!hpsIndexRanges.empty()) {
        int start = 0;
        int end = hpsIndexRanges[0][0] - 1;
        if (start <= end) {
            categorizedRanges.emplace_back(EffectMode::GE, IndexRange{start, end});
        }
    }

    for (size_t i = 0; i < hpsIndexRanges.size(); ++i) {
        categorizedRanges.emplace_back(EffectMode::HPS, hpsIndexRanges[i]);

        // If it is not the last HPS range，deal with middle GE ranges
        if (i >= hpsIndexRanges.size() - 1) {
            continue;
        }
        int start = hpsIndexRanges[i][1] + 1;
        int end = hpsIndexRanges[i + 1][0] - 1;
        if (start <= end) {
            categorizedRanges.emplace_back(EffectMode::GE, IndexRange{start, end});
        }
    }

    // The last GE range after the last hps range
    if (!hpsIndexRanges.empty()) {
        int start = hpsIndexRanges.back()[1] + 1;
        int end = veContainerSize - 1;
        if (start <= end) {
            categorizedRanges.emplace_back(EffectMode::GE, IndexRange{start, end});
        }
    } else {
        // No hps ranges, the full veContainer are GE ranges
        categorizedRanges.emplace_back(EffectMode::GE, IndexRange{0, veContainerSize - 1});
    }

    return categorizedRanges;
}

bool GERender::ComposeGEEffects(std::vector<std::shared_ptr<Drawing::GEVisualEffect>>& visualEffects,
                                std::vector<std::shared_ptr<GEShaderFilter>>& geShaderFiltersOut)
{
    bool anyFilterComposed = false;
    std::shared_ptr<GEFilterComposer> filterComposer = nullptr;
    geShaderFiltersOut.clear();
    for (auto vef : visualEffects) {
        if (vef == nullptr) {
            LOGE("GERender::ComposeGEEffects vef is null");
            continue;
        }
        auto ve = vef->GetImpl();
        auto currentFilter = GenerateShaderFilter(vef);
        if (currentFilter == nullptr) {
            continue;
        }
        auto& filterParams = currentFilter->Params();
        if (currentFilter->Type().empty() || !filterParams || !filterParams.has_value()) {
            // Current filter not supported, save current composed filter
            if (filterComposer) {
                geShaderFiltersOut.push_back(filterComposer->GetComposedFilter());
                filterComposer.reset();
            }
            geShaderFiltersOut.push_back(currentFilter);
        } else if (filterComposer == nullptr) {
            // Create a new composer for potential composion
            filterComposer = std::make_shared<GEFilterComposer>(currentFilter);
        } else if (filterComposer->Compose(currentFilter)) {
            // Compose success
            anyFilterComposed = true;
            continue;
        } else {
            // Compose fail, save current composed filter and create a new composer for potential composion
            geShaderFiltersOut.push_back(filterComposer->GetComposedFilter());
            filterComposer = std::make_shared<GEFilterComposer>(currentFilter);
        }
    }

    if (filterComposer) {
        // Save any filter left in composer
        geShaderFiltersOut.push_back(filterComposer->GetComposedFilter());
        filterComposer.reset();
    }

    return anyFilterComposed;
}

bool GERender::ApplyGEEffects(Drawing::Canvas& canvas,
    std::vector<std::shared_ptr<Drawing::GEVisualEffect>>& visualEffects,
    const HpsGEImageEffectContext& context, std::shared_ptr<Drawing::Image>& outImage)
{
    auto image = context.image;
    auto src = context.src;
    auto dst = context.dst;
    auto sampling = context.sampling;

    std::vector<std::shared_ptr<GEShaderFilter>> geShaderFilters;
    bool anyFilterComposed = ComposeGEEffects(visualEffects, geShaderFilters);

    // Compatibility issues with Kawase skip in RSDrawingFilter::DrawImageRectInternal()
    bool skipKawase = !anyFilterComposed && context.compatibleWithHpsSkipBlur;
    if (skipKawase) {
        return false;
    }

    outImage = image;
    for (auto filter : geShaderFilters) {
        outImage = filter->ProcessImage(canvas, outImage, src, dst);
    }
    return true;
}

bool GERender::HpsSupportEffect(Drawing::GEVisualEffectContainer& veContainer,
                                std::shared_ptr<HpsEffectFilter>& hpsEffectFilter)
{
    if (hpsEffectFilter == nullptr) {
        return false;
    }
    return hpsEffectFilter->HpsSupportEffectGE(veContainer);
}

// true represent Draw Kawase or Mesa succ, false represent Draw Kawase or Mesa false or no Kawase and Mesa
bool GERender::ApplyHpsImageEffect(Drawing::Canvas& canvas, Drawing::GEVisualEffectContainer& veContainer,
    const HpsGEImageEffectContext& context, std::shared_ptr<Drawing::Image>& outImage)
{
    auto hpsEffectFilter = std::make_shared<HpsEffectFilter>(canvas);
    if (!context.image) {
        LOGE("GERender::ApplyImageEffect image is null");
        return false;
    }

    if (veContainer.GetFilters().empty()) {
        return false;
    }

    if (hpsEffectFilter->HpsSupportEffectGE(veContainer)) {
        for (auto vef : veContainer.GetFilters()) {
            auto ve = vef->GetImpl();
            hpsEffectFilter->GenerateVisualEffectFromGE(ve, context.src, context.dst, context.saturationForHPS,
                context.brightnessForHPS, context.image);
        }

        HpsEffectFilter::HpsEffectContext hpsEffectContext = {context.alpha, context.colorFilter, context.maskColor};
        return hpsEffectFilter->ApplyHpsEffect(canvas, context.image, outImage, hpsEffectContext);
    }

    return false;
}

std::shared_ptr<GEShaderFilter> GERender::GenerateExtShaderFilter(
    const std::shared_ptr<Drawing::GEVisualEffectImpl>& ve)
{
    auto type = ve->GetFilterType();
    switch (type) {
        case Drawing::GEVisualEffectImpl::FilterType::MESA_BLUR: {
            const auto& mesaParams = ve->GetMESAParams();
            auto object = GEExternalDynamicLoader::GetInstance().CreateGEXObjectByType(
                static_cast<uint32_t>(type), sizeof(Drawing::GEMESABlurShaderFilterParams),
                static_cast<void*>(mesaParams.get()));
            if (!object) {
                return std::make_shared<GEMESABlurShaderFilter>(*mesaParams);
            }
            std::shared_ptr<GEMESABlurShaderFilter> dmShader(static_cast<GEMESABlurShaderFilter*>(object));
            return dmShader;
        }
        case Drawing::GEVisualEffectImpl::FilterType::LINEAR_GRADIENT_BLUR: {
            const auto& linearGradientBlurParams = ve->GetLinearGradientBlurParams();
            if (linearGradientBlurParams->isRadiusGradient) {
                auto object = GEExternalDynamicLoader::GetInstance().CreateGEXObjectByType(
                    static_cast<uint32_t>(type), sizeof(Drawing::GELinearGradientBlurShaderFilterParams),
                    static_cast<void*>(linearGradientBlurParams.get()));
                if (object) {
                    std::shared_ptr<GELinearGradientBlurShaderFilter>
                        dmShader(static_cast<GELinearGradientBlurShaderFilter*>(object));
                    return dmShader;
                }
            }
            return std::make_shared<GELinearGradientBlurShaderFilter>(*linearGradientBlurParams);
        }
        case Drawing::GEVisualEffectImpl::FilterType::EDGE_LIGHT: {
            const auto& edgeLightParams = ve->GetEdgeLightParams();
            auto object = GEExternalDynamicLoader::GetInstance().CreateGEXObjectByType(
                static_cast<uint32_t>(type), sizeof(Drawing::GEEdgeLightShaderFilterParams),
                static_cast<void*>(edgeLightParams.get()));
            if (!object) {
                return std::make_shared<GEEdgeLightShaderFilter>(*edgeLightParams);
            }
            std::shared_ptr<GEEdgeLightShaderFilter> dmShader(static_cast<GEEdgeLightShaderFilter*>(object));
            return dmShader;
        }
        case Drawing::GEVisualEffectImpl::FilterType::DISPERSION: {
            const auto& dispersionParams = ve->GetDispersionParams();
            auto object = GEExternalDynamicLoader::GetInstance().CreateGEXObjectByType(
                static_cast<uint32_t>(type), sizeof(Drawing::GEDispersionShaderFilterParams),
                static_cast<void*>(dispersionParams.get()));
            if (!object) {
                return nullptr;
            }
            std::shared_ptr<GEShaderFilter> dmShader(static_cast<GEShaderFilter*>(object));
            return dmShader;
        }
        case Drawing::GEVisualEffectImpl::FilterType::VARIABLE_RADIUS_BLUR: {
            const auto& variableRadiusBlurParams = ve->GetVariableRadiusBlurParams();
            auto object = GEExternalDynamicLoader::GetInstance().CreateGEXObjectByType(
                static_cast<uint32_t>(type),
                sizeof(Drawing::GEVariableRadiusBlurShaderFilterParams),
                static_cast<void*>(variableRadiusBlurParams.get()));
            if (!object) {
                return std::make_shared<GEVariableRadiusBlurShaderFilter>(*variableRadiusBlurParams);
            }
            std::shared_ptr<GEVariableRadiusBlurShaderFilter> dmShader(
                static_cast<GEVariableRadiusBlurShaderFilter*>(object));
            return dmShader;
        }
        default:
            break;
    }
    return nullptr;
}

std::shared_ptr<GEShaderFilter> GERender::GenerateShaderFilter(
    const std::shared_ptr<Drawing::GEVisualEffect>& vef)
{
    auto ve = vef->GetImpl();
    std::shared_ptr<GEShaderFilter> shaderFilter;
    LOGD("GERender::GenerateShaderFilter %{public}d", (int)ve->GetFilterType());
    switch (ve->GetFilterType()) {
        case Drawing::GEVisualEffectImpl::FilterType::KAWASE_BLUR: {
            const auto& kawaseParams = ve->GetKawaseParams();
            shaderFilter = std::make_shared<GEKawaseBlurShaderFilter>(*kawaseParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::MESA_BLUR: {
            shaderFilter = GenerateExtShaderFilter(ve);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::AIBAR: {
            const auto& aiBarParams = ve->GetAIBarParams();
            shaderFilter = std::make_shared<GEAIBarShaderFilter>(*aiBarParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::COLOR_GRADIENT: {
            const auto& colorGradientParams = ve->GetColorGradientParams();
            shaderFilter = std::make_shared<GEColorGradientShaderFilter>(*colorGradientParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::GREY: {
            const auto& greyParams = ve->GetGreyParams();
            shaderFilter = std::make_shared<GEGreyShaderFilter>(*greyParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::LINEAR_GRADIENT_BLUR: {
            shaderFilter = GenerateExtShaderFilter(ve);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::MAGNIFIER: {
            const auto& magnifierParams = ve->GetMagnifierParams();
            shaderFilter = std::make_shared<GEMagnifierShaderFilter>(*magnifierParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::WATER_RIPPLE: {
            const auto& waterRippleParams = ve->GetWaterRippleParams();
            shaderFilter = std::make_shared<GEWaterRippleFilter>(*waterRippleParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::DISPLACEMENT_DISTORT_FILTER: {
            const auto& displacementDistortParams = ve->GetDisplacementDistortParams();
            shaderFilter = std::make_shared<GEDisplacementDistortFilter>(*displacementDistortParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::SOUND_WAVE: {
            const auto& soundWaveParams = ve->GetSoundWaveParams();
            shaderFilter = std::make_shared<GESoundWaveFilter>(*soundWaveParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::EDGE_LIGHT: {
            shaderFilter = GenerateExtShaderFilter(ve);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::BEZIER_WARP: {
            const auto& bezierWarpParams = ve->GetBezierWarpParams();
            shaderFilter = std::make_shared<GEBezierWarpShaderFilter>(*bezierWarpParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::DISPERSION: {
            shaderFilter = GenerateExtShaderFilter(ve);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::CONTENT_LIGHT: {
            const auto& contentLightParams = ve->GetContentLightParams();
            shaderFilter = std::make_shared<GEContentLightFilter>(*contentLightParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::DIRECTION_LIGHT: {
            const auto& directionLightParams = ve->GetDirectionLightParams();
            shaderFilter = std::make_shared<GEDirectionLightShaderFilter>(*directionLightParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::MASK_TRANSITION: {
            const auto& maskTransitionParams = ve->GetMaskTransitionParams();
            shaderFilter = std::make_shared<GEMaskTransitionShaderFilter>(*maskTransitionParams);
            break;
        }
        case Drawing::GEVisualEffectImpl::FilterType::VARIABLE_RADIUS_BLUR: {
            shaderFilter = GenerateExtShaderFilter(ve);
            break;
        }
        default:
            break;
    }
    if (shaderFilter) {
        shaderFilter->SetShaderFilterCanvasinfo(vef->GetCanvasInfo());
        shaderFilter->SetSupportHeadroom(vef->GetSupportHeadroom());
    }
    return shaderFilter;
}

std::vector<std::shared_ptr<GEShaderFilter>> GERender::GenerateShaderFilters(
    Drawing::GEVisualEffectContainer& veContainer)
{
    LOGD("GERender::shaderFilters %{public}d", (int)veContainer.GetFilters().size());
    std::vector<std::shared_ptr<GEShaderFilter>> shaderFilters;
    for (auto vef : veContainer.GetFilters()) {
        std::shared_ptr<GEShaderFilter> shaderFilter = GenerateShaderFilter(vef);
        shaderFilters.push_back(shaderFilter);
    }
    return shaderFilters;
}

void GERender::DrawShaderEffect(Drawing::Canvas& canvas, Drawing::GEVisualEffectContainer& veContainer,
    const Drawing::Rect& bounds)
{
    LOGD("GERender::DrawShaderEffect %{public}zu", veContainer.GetFilters().size());
    for (auto vef : veContainer.GetFilters()) {
        if (vef == nullptr) {
            LOGD("GERender::DrawShaderEffect vef is null");
            continue;
        }
        auto ve = vef->GetImpl();
        std::shared_ptr<GEShader> geShaderEffect = GenerateShaderEffect(ve);
        if (geShaderEffect == nullptr) {
            LOGD("GERender::DrawShaderEffect shader is null");
            continue;
        }
        geShaderEffect->SetCache(ve->GetCache());
        geShaderEffect->DrawShader(canvas, bounds);
        ve->SetCache(geShaderEffect->GetCache());
    }
}

std::shared_ptr<GEShader> GERender::GenerateShaderEffect(const std::shared_ptr<Drawing::GEVisualEffectImpl>& ve)
{
    auto it = g_shaderCreatorLUT.find(ve->GetFilterType());
    return it != g_shaderCreatorLUT.end() ? it->second(ve) : nullptr;
}
} // namespace GraphicsEffectEngine
} // namespace OHOS