46 lines
1.5 KiB
C#
46 lines
1.5 KiB
C#
using System;
|
|
using UnityEngine;
|
|
using UnityEngine.Events;
|
|
|
|
namespace cylvester
|
|
{
|
|
[Serializable]
|
|
class UnityMovementTextureEvent : UnityEvent<Texture> { }
|
|
|
|
public class MovementBaker : MonoBehaviour
|
|
{
|
|
[SerializeField] private UnityMovementTextureEvent DifferenceTextureReceived;
|
|
[SerializeField] private ComputeShader computeShader;
|
|
[SerializeField] private float factor = 1;
|
|
|
|
private RenderTexture resultTexture_;
|
|
private RenderTexture previousFrameTexture_;
|
|
private int kernelHandle_;
|
|
|
|
private void Start()
|
|
{
|
|
resultTexture_ = new RenderTexture(512, 424, 0, RenderTextureFormat.R16) {enableRandomWrite = true};
|
|
resultTexture_.Create();
|
|
|
|
previousFrameTexture_ = new RenderTexture(512, 424, 0, RenderTextureFormat.R16) {enableRandomWrite = true};
|
|
previousFrameTexture_.Create();
|
|
|
|
kernelHandle_ = computeShader.FindKernel("CSMain");
|
|
}
|
|
|
|
public void OnInfraredFrameReceived(Texture2D texture)
|
|
{
|
|
computeShader.SetTexture(kernelHandle_, "Input", texture);
|
|
computeShader.SetTexture(kernelHandle_, "Previous", previousFrameTexture_);
|
|
computeShader.SetTexture(kernelHandle_, "Result", resultTexture_);
|
|
computeShader.SetFloat("Factor", factor);
|
|
|
|
computeShader.Dispatch(kernelHandle_, 1, 1, 1);
|
|
|
|
DifferenceTextureReceived.Invoke(resultTexture_);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|