ComputeShader绘制全屏纯色纹理-CSDN博客

本文链接：https://blog.csdn.net/weixin_43796392/article/details/147638199

参考

Getting Started With Compute Shaders In Unity

环境

Win10
Unity20194.40

全屏纯色纹理示例

使用ComputerShader逐个像素设置颜色

ComputeShader脚本
设置纹理颜色

#pragma kernel CSMain

RWTexture2D<float4> Result;//纹理
half4 solidColor;//颜色

[numthreads(8,8,1)]
void CSMain (uint3 id : SV_DispatchThreadID)
{
      Result[id.xy] = solidColor;//设置颜色 id.xy对应屏幕上的像素点
}

UseSolidComputeShader脚本
调用ComputeShader,绘制纯色纹理

using UnityEngine;
public class UseSolidComputeShader : MonoBehaviour
{
    [SerializeField] ComputeShader computeShader;
    RenderTexture renderTexture;
    int functionId;
    int groupsX;
    int groupsY;

    public RenderTexture Texture => renderTexture;

    public void Init(int width, int height)
    {
        renderTexture = new RenderTexture(width, height, 24, RenderTextureFormat.ARGB32, RenderTextureReadWrite.Default);
        renderTexture.enableRandomWrite = true;
        renderTexture.Create();

        functionId = computeShader.FindKernel("CSMain");
        groupsX = Mathf.CeilToInt(renderTexture.width / 8f);//向上取整，避免小于纹理的宽度
        groupsY = Mathf.CeilToInt(renderTexture.height / 8f);
        computeShader.SetTexture(functionId, "Result", renderTexture);
    }

    public void Draw(Color color)
    {
        computeShader.SetVector("solidColor", color);
        computeShader.Dispatch(functionId, groupsX, groupsY, 1);
    }
}

TestSolidColor脚本
按下D键，绘制纯色纹理，显示在Rawimage上

using UnityEngine;
using UnityEngine.Profiling;
using UnityEngine.UI;
public class TestSolidColor : MonoBehaviour
{
    [SerializeField] UseSolidComputeShader useSolidComputeShader;

    [SerializeField] RawImage rawImage;

    private void Awake()
    {
        useSolidComputeShader.Init(Screen.width, Screen.height);
        rawImage.texture = useSolidComputeShader.Texture;
    }

    private void Update()
    {
        if (Input.GetKeyDown(KeyCode.D))
        {
            Profiler.BeginSample("Random Solid Color");
            useSolidComputeShader.Draw(Random.ColorHSV());
            Profiler.EndSample();
        }
    }
}