You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.3 KiB
49 lines
1.3 KiB
#include "UnityCG.cginc"
|
|
|
|
RWStructuredBuffer<uint> _Vectorscope;
|
|
Texture2D<float4> _Source;
|
|
|
|
CBUFFER_START (Params)
|
|
uint _IsLinear;
|
|
float4 _Res;
|
|
CBUFFER_END
|
|
|
|
#define GROUP_SIZE 32
|
|
|
|
float3 RgbToYUV(float3 c)
|
|
{
|
|
float Y = 0.299 * c.r + 0.587 * c.g + 0.114 * c.b;
|
|
float U = -0.169 * c.r - 0.331 * c.g + 0.500 * c.b;
|
|
float V = 0.500 * c.r - 0.419 * c.g - 0.081 * c.b;
|
|
return float3(Y, U, V);
|
|
}
|
|
|
|
#pragma kernel KVectorscope
|
|
[numthreads(GROUP_SIZE,GROUP_SIZE,1)]
|
|
void KVectorscope(uint2 dispatchThreadId : SV_DispatchThreadID)
|
|
{
|
|
if (dispatchThreadId.x < (uint)_Res.x && dispatchThreadId.y < (uint)_Res.y)
|
|
{
|
|
float3 color = saturate(_Source[dispatchThreadId].xyz);
|
|
if (_IsLinear > 0)
|
|
color = LinearToGammaSpace(color);
|
|
|
|
float3 yuv = RgbToYUV(color);
|
|
|
|
if (length(yuv.yz) > 0.49)
|
|
yuv.yz = normalize(yuv.yz) * 0.49;
|
|
|
|
yuv.yz += (0.5).xx;
|
|
uint u = (uint)floor(yuv.y * _Res.x);
|
|
uint v = (uint)floor(yuv.z * _Res.y);
|
|
InterlockedAdd(_Vectorscope[v * _Res.x + u], 1);
|
|
}
|
|
}
|
|
|
|
#pragma kernel KVectorscopeClear
|
|
[numthreads(GROUP_SIZE,GROUP_SIZE,1)]
|
|
void KVectorscopeClear(uint2 dispatchThreadId : SV_DispatchThreadID)
|
|
{
|
|
if (dispatchThreadId.x < (uint)_Res.x && dispatchThreadId.y < (uint)_Res.y)
|
|
_Vectorscope[dispatchThreadId.y * _Res.x + dispatchThreadId.x] = 0u;
|
|
}
|