Unity Compute Shader的一个简单完整的示例

代码示例简介

       这个例子是 新建一个数组，这个数组里的值在GPU中进行一次计算，并把结果保存回CPU，最终输出出来。代码如下：

c#脚本：

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class ComputeScript : MonoBehaviour
{
    //ComputeShader对象定义，需要给它赋值一个computeShader的shader资源
    public ComputeShader computeShader;
    
    //ComputeBuffer, c#端与gpu数据通信的容器，我们组织好需要计算的数据（本例中是inOutBuffer），装在这个buffer里面，然后把这个buffer塞到
    //computeShader里面，为gpu计算做数据准备
    private ComputeBuffer buffer;

    //我们总共有多少个数据
    public int count = 128;

    //表示我们要计算的方法在computeShader里的索引，这个索引在绑定buffer时会用到
    private int kernal;

    //我们自己定义的数据，用这个对象把我们要的数据装起来，然后塞给ComputeBuffer
    //注意，数组里面的数据必须是blittable 类型的数据，可以认为必须是c#基础类型，或者由基础类型组成的struct类型
    MixData[] inOutBuffer;
    struct MixData
    {
        public int myVal;
        public float