Files
test2/intern/cycles/kernel/osl/shaders/node_ramp_util.h
2023-09-24 14:52:38 +10:00

88 lines
1.7 KiB
C

/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
*
* SPDX-License-Identifier: Apache-2.0 */
/* NOTE: svm_ramp.h, svm_ramp_util.h and node_ramp_util.h must stay consistent */
color rgb_ramp_lookup(color ramp[], float at, int interpolate, int extrapolate)
{
float f = at;
int table_size = arraylength(ramp);
if ((f < 0.0 || f > 1.0) && extrapolate) {
color t0, dy;
if (f < 0.0) {
t0 = ramp[0];
dy = t0 - ramp[1];
f = -f;
}
else {
t0 = ramp[table_size - 1];
dy = t0 - ramp[table_size - 2];
f = f - 1.0;
}
return t0 + dy * f * (table_size - 1);
}
f = clamp(at, 0.0, 1.0) * (table_size - 1);
/* clamp int as well in case of NaN */
int i = (int)f;
if (i < 0) {
i = 0;
}
if (i >= table_size) {
i = table_size - 1;
}
float t = f - (float)i;
color result = ramp[i];
if (interpolate && t > 0.0) {
result = (1.0 - t) * result + t * ramp[i + 1];
}
return result;
}
float rgb_ramp_lookup(float ramp[], float at, int interpolate, int extrapolate)
{
float f = at;
int table_size = arraylength(ramp);
if ((f < 0.0 || f > 1.0) && extrapolate) {
float t0, dy;
if (f < 0.0) {
t0 = ramp[0];
dy = t0 - ramp[1];
f = -f;
}
else {
t0 = ramp[table_size - 1];
dy = t0 - ramp[table_size - 2];
f = f - 1.0;
}
return t0 + dy * f * (table_size - 1);
}
f = clamp(at, 0.0, 1.0) * (table_size - 1);
/* clamp int as well in case of NaN */
int i = (int)f;
if (i < 0) {
i = 0;
}
if (i >= table_size) {
i = table_size - 1;
}
float t = f - (float)i;
float result = ramp[i];
if (interpolate && t > 0.0) {
result = (1.0 - t) * result + t * ramp[i + 1];
}
return result;
}