978c8e5335ade55dd98dbf0863cbaafc9f190a43
[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
22 CCL_NAMESPACE_BEGIN
23
24 class ShaderGraph;
25 class ShaderInput;
26 class ShaderNode;
27 class ShaderOutput;
28
29 class ConstantFolder {
30 public:
31         ShaderGraph *const graph;
32         ShaderNode *const node;
33         ShaderOutput *const output;
34
35         ConstantFolder(ShaderGraph *graph, ShaderNode *node, ShaderOutput *output);
36
37         bool all_inputs_constant() const;
38
39         /* Constant folding helpers, always return true for convenience. */
40         void make_constant(float value) const;
41         void make_constant(float3 value) const;
42         void make_constant_clamp(float value, bool clamp) const;
43         void make_constant_clamp(float3 value, bool clamp) const;
44
45         /* Bypass node, relinking to another output socket. */
46         void bypass(ShaderOutput *output) const;
47
48         /* For closure nodes, discard node entirely or bypass to one of its inputs. */
49         void discard() const;
50         void bypass_or_discard(ShaderInput *input) const;
51
52         /* Bypass or make constant, unless we can't due to clamp being true. */
53         bool try_bypass_or_make_constant(ShaderInput *input, float3 input_value, bool clamp) const;
54 };
55
56 CCL_NAMESPACE_END
57
58 #endif /* __CONSTANT_FOLD_H__ */
59