Skiko项目中RuntimeShaderBuilder对数组uniform的支持解析

背景介绍

Skiko是JetBrains推出的一个跨平台Skia绑定库，它允许开发者在多种平台上使用Skia图形库的功能。在图形编程中，着色器(Shader)是实现各种视觉效果的核心组件之一。RuntimeShaderBuilder是Skiko中用于构建运行时着色器的重要工具类。

问题发现

在Skiko的早期版本中，开发者发现RuntimeShaderBuilder缺少对数组类型uniform变量的支持。具体表现为，当着色器代码中声明了一个vec3数组类型的uniform变量时，无法通过现有的API来设置这个数组的值。

const int COLOR_SIZE = 4;
uniform vec3 uColor[COLOR_SIZE];

技术分析

现有API限制

原生的RuntimeShaderBuilder提供了多种uniform设置方法，如设置单个浮点数、整数、向量等，但缺少直接设置浮点数数组的方法。这在处理需要传递多个颜色值或其他向量数组的场景时造成了不便。

Android平台的对比

在Android平台上，类似的RuntimeShader实现已经支持通过FloatArray来设置数组类型的uniform变量。开发者可以这样使用：

compositeRuntimeEffect.setFloatUniform("uColor", floatArrayOf(
    1f, 0f, 0f,
    0f, 1f, 1f,
    0f, 0f, 0f,
    0f, 1f, 0f
))

底层实现机制

在Skia底层，SkRuntimeEffectBuilder::BuilderUniform类提供了set()方法来处理数组类型的uniform变量。Skiko需要在其JNI绑定层添加对应的方法调用，才能实现这一功能。

解决方案实现

社区贡献者sphrak实现了这一功能，主要改动包括：

1. 在RuntimeShaderBuilder.kt中添加新的uniform(String, FloatArray)方法
2. 在RuntimeShaderBuilder.cc中添加对应的JNI绑定
3. 确保底层调用SkRuntimeEffectBuilder::BuilderUniform.set()方法

使用示例

实现后，开发者可以这样使用数组uniform：

val shaderBuilder = RuntimeShaderBuilder(RuntimeEffect.makeForShader(shaderCode))
val colors = floatArrayOf(
    1.0f, 0.0f, 0.0f, // 红色
    0.0f, 1.0f, 0.0f, // 绿色
    0.0f, 0.0f, 1.0f  // 蓝色
)
shaderBuilder.uniform("colors", colors)

类型转换处理

值得注意的是，Skia底层会自动处理浮点数数组到不同向量类型的转换。无论是vec2[]还是vec3[]，都可以使用同一个FloatArray来设置，系统会根据着色器中的声明自动进行正确的解释。

跨平台兼容性

这一功能已经在多个平台上得到验证：

• JVM平台
• macOS Arm64平台
• WebAssembly(wasmJs)平台

实际应用场景

这个功能特别适用于需要传递多个颜色值或其他向量参数的着色器效果，例如：

1. 多色渐变效果
2. 粒子系统中的多个属性值
3. 复杂材质的多参数配置
4. 动画中的关键帧数据传递

总结

Skiko通过添加对数组类型uniform的支持，进一步提升了其在图形编程中的灵活性和实用性。这一改进使得Skiko在功能上更加接近原生Android平台的实现，为跨平台图形开发提供了更一致的体验。开发者现在可以更方便地在着色器中使用数组参数，实现更复杂的视觉效果。