115 lines
5.3 KiB
HLSL
115 lines
5.3 KiB
HLSL
int2 tint_clamp(int2 e, int2 low, int2 high) {
|
|
return min(max(e, low), high);
|
|
}
|
|
|
|
struct GammaTransferParams {
|
|
float G;
|
|
float A;
|
|
float B;
|
|
float C;
|
|
float D;
|
|
float E;
|
|
float F;
|
|
uint padding;
|
|
};
|
|
struct ExternalTextureParams {
|
|
uint numPlanes;
|
|
uint doYuvToRgbConversionOnly;
|
|
float3x4 yuvToRgbConversionMatrix;
|
|
GammaTransferParams gammaDecodeParams;
|
|
GammaTransferParams gammaEncodeParams;
|
|
float3x3 gamutConversionMatrix;
|
|
float3x2 coordTransformationMatrix;
|
|
};
|
|
|
|
Texture2D<float4> ext_tex_plane_1 : register(t2);
|
|
cbuffer cbuffer_ext_tex_params : register(b3) {
|
|
uint4 ext_tex_params[13];
|
|
};
|
|
Texture2D<float4> t : register(t0);
|
|
RWTexture2D<float4> outImage : register(u1);
|
|
|
|
float3 gammaCorrection(float3 v, GammaTransferParams params) {
|
|
const bool3 cond = (abs(v) < float3((params.D).xxx));
|
|
const float3 t = (float3(sign(v)) * ((params.C * abs(v)) + params.F));
|
|
const float3 f = (float3(sign(v)) * (pow(((params.A * abs(v)) + params.B), float3((params.G).xxx)) + params.E));
|
|
return (cond ? t : f);
|
|
}
|
|
|
|
float4 textureLoadExternal(Texture2D<float4> plane0, Texture2D<float4> plane1, int2 coord, ExternalTextureParams params) {
|
|
const int2 coord1 = (coord >> (1u).xx);
|
|
float3 color = float3(0.0f, 0.0f, 0.0f);
|
|
if ((params.numPlanes == 1u)) {
|
|
color = plane0.Load(int3(coord, 0)).rgb;
|
|
} else {
|
|
color = mul(params.yuvToRgbConversionMatrix, float4(plane0.Load(int3(coord, 0)).r, plane1.Load(int3(coord1, 0)).rg, 1.0f));
|
|
}
|
|
if ((params.doYuvToRgbConversionOnly == 0u)) {
|
|
color = gammaCorrection(color, params.gammaDecodeParams);
|
|
color = mul(color, params.gamutConversionMatrix);
|
|
color = gammaCorrection(color, params.gammaEncodeParams);
|
|
}
|
|
return float4(color, 1.0f);
|
|
}
|
|
|
|
float3x4 ext_tex_params_load_2(uint offset) {
|
|
const uint scalar_offset = ((offset + 0u)) / 4;
|
|
const uint scalar_offset_1 = ((offset + 16u)) / 4;
|
|
const uint scalar_offset_2 = ((offset + 32u)) / 4;
|
|
return float3x4(asfloat(ext_tex_params[scalar_offset / 4]), asfloat(ext_tex_params[scalar_offset_1 / 4]), asfloat(ext_tex_params[scalar_offset_2 / 4]));
|
|
}
|
|
|
|
GammaTransferParams ext_tex_params_load_4(uint offset) {
|
|
const uint scalar_offset_3 = ((offset + 0u)) / 4;
|
|
const uint scalar_offset_4 = ((offset + 4u)) / 4;
|
|
const uint scalar_offset_5 = ((offset + 8u)) / 4;
|
|
const uint scalar_offset_6 = ((offset + 12u)) / 4;
|
|
const uint scalar_offset_7 = ((offset + 16u)) / 4;
|
|
const uint scalar_offset_8 = ((offset + 20u)) / 4;
|
|
const uint scalar_offset_9 = ((offset + 24u)) / 4;
|
|
const uint scalar_offset_10 = ((offset + 28u)) / 4;
|
|
const GammaTransferParams tint_symbol = {asfloat(ext_tex_params[scalar_offset_3 / 4][scalar_offset_3 % 4]), asfloat(ext_tex_params[scalar_offset_4 / 4][scalar_offset_4 % 4]), asfloat(ext_tex_params[scalar_offset_5 / 4][scalar_offset_5 % 4]), asfloat(ext_tex_params[scalar_offset_6 / 4][scalar_offset_6 % 4]), asfloat(ext_tex_params[scalar_offset_7 / 4][scalar_offset_7 % 4]), asfloat(ext_tex_params[scalar_offset_8 / 4][scalar_offset_8 % 4]), asfloat(ext_tex_params[scalar_offset_9 / 4][scalar_offset_9 % 4]), ext_tex_params[scalar_offset_10 / 4][scalar_offset_10 % 4]};
|
|
return tint_symbol;
|
|
}
|
|
|
|
float3x3 ext_tex_params_load_6(uint offset) {
|
|
const uint scalar_offset_11 = ((offset + 0u)) / 4;
|
|
const uint scalar_offset_12 = ((offset + 16u)) / 4;
|
|
const uint scalar_offset_13 = ((offset + 32u)) / 4;
|
|
return float3x3(asfloat(ext_tex_params[scalar_offset_11 / 4].xyz), asfloat(ext_tex_params[scalar_offset_12 / 4].xyz), asfloat(ext_tex_params[scalar_offset_13 / 4].xyz));
|
|
}
|
|
|
|
float3x2 ext_tex_params_load_8(uint offset) {
|
|
const uint scalar_offset_14 = ((offset + 0u)) / 4;
|
|
uint4 ubo_load = ext_tex_params[scalar_offset_14 / 4];
|
|
const uint scalar_offset_15 = ((offset + 8u)) / 4;
|
|
uint4 ubo_load_1 = ext_tex_params[scalar_offset_15 / 4];
|
|
const uint scalar_offset_16 = ((offset + 16u)) / 4;
|
|
uint4 ubo_load_2 = ext_tex_params[scalar_offset_16 / 4];
|
|
return float3x2(asfloat(((scalar_offset_14 & 2) ? ubo_load.zw : ubo_load.xy)), asfloat(((scalar_offset_15 & 2) ? ubo_load_1.zw : ubo_load_1.xy)), asfloat(((scalar_offset_16 & 2) ? ubo_load_2.zw : ubo_load_2.xy)));
|
|
}
|
|
|
|
ExternalTextureParams ext_tex_params_load(uint offset) {
|
|
const uint scalar_offset_17 = ((offset + 0u)) / 4;
|
|
const uint scalar_offset_18 = ((offset + 4u)) / 4;
|
|
const ExternalTextureParams tint_symbol_1 = {ext_tex_params[scalar_offset_17 / 4][scalar_offset_17 % 4], ext_tex_params[scalar_offset_18 / 4][scalar_offset_18 % 4], ext_tex_params_load_2((offset + 16u)), ext_tex_params_load_4((offset + 64u)), ext_tex_params_load_4((offset + 96u)), ext_tex_params_load_6((offset + 128u)), ext_tex_params_load_8((offset + 176u))};
|
|
return tint_symbol_1;
|
|
}
|
|
|
|
[numthreads(1, 1, 1)]
|
|
void main() {
|
|
uint2 tint_tmp;
|
|
t.GetDimensions(tint_tmp.x, tint_tmp.y);
|
|
float4 red = textureLoadExternal(t, ext_tex_plane_1, tint_clamp((10).xx, (0).xx, int2((tint_tmp - (1u).xx))), ext_tex_params_load(0u));
|
|
uint2 tint_tmp_1;
|
|
outImage.GetDimensions(tint_tmp_1.x, tint_tmp_1.y);
|
|
outImage[tint_clamp((0).xx, (0).xx, int2((tint_tmp_1 - (1u).xx)))] = red;
|
|
uint2 tint_tmp_2;
|
|
t.GetDimensions(tint_tmp_2.x, tint_tmp_2.y);
|
|
float4 green = textureLoadExternal(t, ext_tex_plane_1, tint_clamp(int2(70, 118), (0).xx, int2((tint_tmp_2 - (1u).xx))), ext_tex_params_load(0u));
|
|
uint2 tint_tmp_3;
|
|
outImage.GetDimensions(tint_tmp_3.x, tint_tmp_3.y);
|
|
outImage[tint_clamp(int2(1, 0), (0).xx, int2((tint_tmp_3 - (1u).xx)))] = green;
|
|
return;
|
|
}
|