wehub-resource-sync
38 lines
892 B
Plaintext
38 lines
892 B
Plaintext
layout(set = 0, binding = 0) writeonly buffer outputBuffer{
|
|
FLOAT4 data[];
|
|
}uOutput;
|
|
|
|
layout(set = 0, binding = 1) readonly buffer inputBuffer{
|
|
FLOAT4 data[];
|
|
}uInput;
|
|
|
|
layout(set = 0, binding = 2) readonly buffer scaleBuffer{
|
|
FLOAT4 data[];
|
|
}uScale;
|
|
|
|
layout(set = 0, binding = 3) readonly buffer biasBuffer{
|
|
FLOAT4 data[];
|
|
}uBias;
|
|
|
|
layout(set = 0, binding = 4) uniform constBuffer{
|
|
ivec4 imgSize;
|
|
}uConst;
|
|
|
|
layout(local_size_x = 256, local_size_y = 1, local_size_z = 1) in;
|
|
|
|
void main()
|
|
{
|
|
int posIndex = int(gl_GlobalInvocationID.x);
|
|
|
|
if(posIndex < uConst.imgSize.w)
|
|
{
|
|
int channelIndex = posIndex / uConst.imgSize.x;
|
|
FLOAT4 scale = uScale.data[channelIndex];
|
|
FLOAT4 bias = uBias.data[channelIndex];
|
|
|
|
FLOAT4 color = uInput.data[posIndex];
|
|
FLOAT4 res = color * scale + bias;
|
|
uOutput.data[posIndex] = res;
|
|
}
|
|
}
|