unity|Unity Compute Shader 例子 unity|shader

Unity Compute Shader Demo

本文提供一个简单的可运行的Unity Compute Shader实例

运行结果 【unity|Unity Compute Shader 例子】

文章图片

Shader代码

// Each #kernel tells which function to compile; you can have many kernels #pragma kernel CSMain// Create a RenderTexture with enableRandomWrite flag and set it // with cs.SetTexture RWTexture2D Result; [numthreads(8,8,1)] void CSMain (uint3 id : SV_DispatchThreadID) { Result[id.xy] = float4(id.x & id.y, (id.x & 15)/15.0, (id.y & 15)/15.0, 1); // Result[id.xy] = float4(1,0,0,1); }

Script代码

using System.Collections; using System.Collections.Generic; using UnityEngine; using UnityEngine.UI; public class Sample01 : MonoBehaviour { public ComputeShader _ComputeShader; public RawImage rawImg; private Vector2Int size = new Vector2Int(8,8); private RenderTexture _Result; private int propId; private int kernelIndex; void Start() { var desc = new RenderTextureDescriptor(size.x, size.y, RenderTextureFormat.ARGBHalf); desc.enableRandomWrite = true; _Result = new RenderTexture(desc); _Result.Create(); propId = Shader.PropertyToID("Result"); kernelIndex = _ComputeShader.FindKernel("CSMain"); _ComputeShader.SetTexture(kernelIndex, propId, _Result); _ComputeShader.Dispatch(kernelIndex, size.x, size.y, 1); rawImg.texture = _Result; }void OnDestroy() {if (null != _Result) { _Result.Release(); }}}

需要手动做的操作