Cycles: constant folding for RGB/Vector Curves and Color Ramp.
[blender.git] / intern / cycles / render / constant_fold.h
1 /*
2  * Copyright 2011-2013 Blender Foundation
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #ifndef __CONSTANT_FOLD_H__
18 #define __CONSTANT_FOLD_H__
19
20 #include "util_types.h"
21 #include "util_vector.h"
22
23 CCL_NAMESPACE_BEGIN
24
25 class ShaderGraph;
26 class ShaderInput;
27 class ShaderNode;
28 class ShaderOutput;
29
30 class ConstantFolder {
31 public:
32         ShaderGraph *const graph;
33         ShaderNode *const node;
34         ShaderOutput *const output;
35
36         ConstantFolder(ShaderGraph *graph, ShaderNode *node, ShaderOutput *output);
37
38         bool all_inputs_constant() const;
39
40         /* Constant folding helpers, always return true for convenience. */
41         void make_constant(float value) const;
42         void make_constant(float3 value) const;
43         void make_constant_clamp(float value, bool clamp) const;
44         void make_constant_clamp(float3 value, bool clamp) const;
45
46         /* Bypass node, relinking to another output socket. */
47         void bypass(ShaderOutput *output) const;
48
49         /* For closure nodes, discard node entirely or bypass to one of its inputs. */
50         void discard() const;
51         void bypass_or_discard(ShaderInput *input) const;
52
53         /* Bypass or make constant, unless we can't due to clamp being true. */
54         bool try_bypass_or_make_constant(ShaderInput *input, float3 input_value, bool clamp) const;
55 };
56
57 CCL_NAMESPACE_END
58
59 #endif /* __CONSTANT_FOLD_H__ */
60