Cycles: constant folding for RGB/Vector Curves and Color Ramp.
These are complex nodes, and it's conceivable they may end up constant in some circumstances within node groups, so folding support is useful. Reviewed By: brecht Differential Revision: https://developer.blender.org/D2084
This commit is contained in:
Alexander Gavrilov
committed by
Brecht Van Lommel
Brecht Van Lommel
parent
f4bcc97729
commit
ea2ebf7a00
89
intern/cycles/kernel/shaders/node_ramp_util.h
Normal file
89
intern/cycles/kernel/shaders/node_ramp_util.h
Normal file
@@ -0,0 +1,89 @@
|
||||
/*
|
||||
* Copyright 2011-2013 Blender Foundation
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
/* 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;
|
||||
}
|
||||
@@ -14,43 +14,7 @@
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "stdosl.h"
|
||||
#include "oslutil.h"
|
||||
|
||||
float ramp_lookup(color ramp[], float at, int component)
|
||||
{
|
||||
int table_size = arraylength(ramp);
|
||||
|
||||
if (at < 0.0 || at > 1.0) {
|
||||
float t0, dy;
|
||||
if (at < 0.0) {
|
||||
t0 = ramp[0][component];
|
||||
dy = t0 - ramp[1][component];
|
||||
at = -at;
|
||||
}
|
||||
else {
|
||||
t0 = ramp[table_size - 1][component];
|
||||
dy = t0 - ramp[table_size - 2][component];
|
||||
at = at - 1.0;
|
||||
}
|
||||
return t0 + dy * at * (table_size - 1);
|
||||
}
|
||||
|
||||
float 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][component];
|
||||
|
||||
if (t > 0.0)
|
||||
result = (1.0 - t) * result + t * ramp[i + 1][component];
|
||||
|
||||
return result;
|
||||
}
|
||||
#include "node_ramp_util.h"
|
||||
|
||||
shader node_rgb_curves(
|
||||
color ramp[] = {0.0},
|
||||
@@ -63,9 +27,13 @@ shader node_rgb_curves(
|
||||
{
|
||||
color c = (ColorIn - color(min_x, min_x, min_x)) / (max_x - min_x);
|
||||
|
||||
ColorOut[0] = ramp_lookup(ramp, c[0], 0);
|
||||
ColorOut[1] = ramp_lookup(ramp, c[1], 1);
|
||||
ColorOut[2] = ramp_lookup(ramp, c[2], 2);
|
||||
color r = rgb_ramp_lookup(ramp, c[0], 1, 1);
|
||||
color g = rgb_ramp_lookup(ramp, c[0], 1, 1);
|
||||
color b = rgb_ramp_lookup(ramp, c[0], 1, 1);
|
||||
|
||||
ColorOut[0] = r[0];
|
||||
ColorOut[1] = g[1];
|
||||
ColorOut[2] = b[2];
|
||||
|
||||
ColorOut = mix(ColorIn, ColorOut, Fac);
|
||||
}
|
||||
|
||||
@@ -14,8 +14,7 @@
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "stdosl.h"
|
||||
#include "oslutil.h"
|
||||
#include "node_ramp_util.h"
|
||||
|
||||
shader node_rgb_ramp(
|
||||
color ramp_color[] = {0.0},
|
||||
@@ -26,21 +25,7 @@ shader node_rgb_ramp(
|
||||
output color Color = 0.0,
|
||||
output float Alpha = 1.0)
|
||||
{
|
||||
int table_size = arraylength(ramp_color);
|
||||
float f = clamp(Fac, 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 = ramp_color[i];
|
||||
Alpha = ramp_alpha[i];
|
||||
|
||||
if (interpolate && t > 0.0) {
|
||||
Color = (1.0 - t) * Color + t * ramp_color[i + 1];
|
||||
Alpha = (1.0 - t) * Alpha + t * ramp_alpha[i + 1];
|
||||
}
|
||||
Color = rgb_ramp_lookup(ramp_color, Fac, interpolate, 0);
|
||||
Alpha = rgb_ramp_lookup(ramp_alpha, Fac, interpolate, 0);
|
||||
}
|
||||
|
||||
|
||||
@@ -14,43 +14,7 @@
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "stdosl.h"
|
||||
#include "oslutil.h"
|
||||
|
||||
float ramp_lookup(color ramp[], float at, int component)
|
||||
{
|
||||
int table_size = arraylength(ramp);
|
||||
|
||||
if (at < 0.0 || at > 1.0) {
|
||||
float t0, dy;
|
||||
if (at < 0.0) {
|
||||
t0 = ramp[0][component];
|
||||
dy = t0 - ramp[1][component];
|
||||
at = -at;
|
||||
}
|
||||
else {
|
||||
t0 = ramp[table_size - 1][component];
|
||||
dy = t0 - ramp[table_size - 2][component];
|
||||
at = at - 1.0;
|
||||
}
|
||||
return t0 + dy * at * (table_size - 1);
|
||||
}
|
||||
|
||||
float 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][component];
|
||||
|
||||
if (t > 0.0)
|
||||
result = (1.0 - t) * result + t * ramp[i + 1][component];
|
||||
|
||||
return result;
|
||||
}
|
||||
#include "node_ramp_util.h"
|
||||
|
||||
shader node_vector_curves(
|
||||
color ramp[] = {0.0},
|
||||
@@ -63,9 +27,13 @@ shader node_vector_curves(
|
||||
{
|
||||
vector c = (VectorIn - vector(min_x, min_x, min_x)) / (max_x - min_x);
|
||||
|
||||
VectorOut[0] = ramp_lookup(ramp, c[0], 0);
|
||||
VectorOut[1] = ramp_lookup(ramp, c[1], 1);
|
||||
VectorOut[2] = ramp_lookup(ramp, c[2], 2);
|
||||
color r = rgb_ramp_lookup(ramp, c[0], 1, 1);
|
||||
color g = rgb_ramp_lookup(ramp, c[0], 1, 1);
|
||||
color b = rgb_ramp_lookup(ramp, c[0], 1, 1);
|
||||
|
||||
VectorOut[0] = r[0];
|
||||
VectorOut[1] = g[1];
|
||||
VectorOut[2] = b[2];
|
||||
|
||||
VectorOut = mix(VectorIn, VectorOut, Fac);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user