72 lines
2.0 KiB
C#
72 lines
2.0 KiB
C#
using UnityEngine;
|
|
|
|
public class ComputeController : MonoBehaviour
|
|
{
|
|
[Header("Conections")]
|
|
[SerializeField] private ComputeShader _computeShader;
|
|
[SerializeField] private Texture2D _inputTexture;
|
|
[SerializeField] private RenderTexture _resultTexture;
|
|
|
|
[Header("State")]
|
|
[SerializeField] private int _width;
|
|
[SerializeField] private int _height;
|
|
[SerializeField] private float _sin;
|
|
|
|
private int _kernelHandle;
|
|
private const int NumThreadX = 8;
|
|
private const int NumThreadY = 8;
|
|
|
|
#region ShaderVariables
|
|
|
|
private static readonly int Result = Shader.PropertyToID("result");
|
|
private static readonly int SourceTexture = Shader.PropertyToID("source_texture");
|
|
private static readonly int SinScaled = Shader.PropertyToID("sin_scaled");
|
|
|
|
#endregion
|
|
|
|
private void Start()
|
|
{
|
|
InitializeTextures();
|
|
|
|
InitializeShader();
|
|
|
|
RunComputeShader();
|
|
}
|
|
|
|
private void InitializeTextures()
|
|
{
|
|
_width = Mathf.RoundToInt(_inputTexture.width);
|
|
_height = Mathf.RoundToInt(_inputTexture.height);
|
|
_resultTexture = new(_width, _height, 24);
|
|
_resultTexture.name = "ResultTexture";
|
|
_resultTexture.enableRandomWrite = true;
|
|
_resultTexture.Create();
|
|
}
|
|
|
|
private void InitializeShader()
|
|
{
|
|
_kernelHandle = _computeShader.FindKernel("cs_main");
|
|
_computeShader.SetTexture(_kernelHandle, SourceTexture, _inputTexture);
|
|
_computeShader.SetTexture(_kernelHandle, Result, _resultTexture);
|
|
}
|
|
|
|
private void RunComputeShader()
|
|
{
|
|
_computeShader.Dispatch(_kernelHandle, _width / NumThreadX, _height / NumThreadY, 1);
|
|
}
|
|
|
|
private void OnGUI()
|
|
{
|
|
ShaderUpdate();
|
|
|
|
GUI.DrawTexture(new Rect(0, 0, _width, _height), _inputTexture);
|
|
GUI.DrawTexture(new Rect(0, _height, _width, _height), _resultTexture);
|
|
}
|
|
|
|
private void ShaderUpdate()
|
|
{
|
|
_sin = (Mathf.Sin(Time.unscaledTime) + 1) / 2;
|
|
_computeShader.SetFloat(SinScaled, _sin);
|
|
RunComputeShader();
|
|
}
|
|
} |