Fix T39793: cycles SVM shading bug with tangled up nodes after recent optimization.
[blender.git] / intern / cycles / render / svm.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 __SVM_H__
18 #define __SVM_H__
19
20 #include "attribute.h"
21 #include "graph.h"
22 #include "shader.h"
23
24 #include "util_set.h"
25 #include "util_string.h"
26
27 CCL_NAMESPACE_BEGIN
28
29 class Device;
30 class DeviceScene;
31 class ImageManager;
32 class Scene;
33 class ShaderGraph;
34 class ShaderInput;
35 class ShaderNode;
36 class ShaderOutput;
37
38 /* Shader Manager */
39
40 class SVMShaderManager : public ShaderManager {
41 public:
42         SVMShaderManager();
43         ~SVMShaderManager();
44
45         void reset(Scene *scene);
46
47         void device_update(Device *device, DeviceScene *dscene, Scene *scene, Progress& progress);
48         void device_free(Device *device, DeviceScene *dscene, Scene *scene);
49 };
50
51 /* Graph Compiler */
52
53 class SVMCompiler {
54 public:
55         SVMCompiler(ShaderManager *shader_manager, ImageManager *image_manager);
56         void compile(Shader *shader, vector<int4>& svm_nodes, int index);
57
58         void stack_assign(ShaderOutput *output);
59         void stack_assign(ShaderInput *input);
60         int stack_find_offset(ShaderSocketType type);
61         void stack_clear_offset(ShaderSocketType type, int offset);
62         void stack_link(ShaderInput *input, ShaderOutput *output);
63
64         void add_node(NodeType type, int a = 0, int b = 0, int c = 0);
65         void add_node(int a = 0, int b = 0, int c = 0, int d = 0);
66         void add_node(NodeType type, const float3& f);
67         void add_node(const float4& f);
68         void add_array(float4 *f, int num);
69         uint attribute(ustring name);
70         uint attribute(AttributeStandard std);
71         uint encode_uchar4(uint x, uint y = 0, uint z = 0, uint w = 0);
72         uint closure_mix_weight_offset() { return mix_weight_offset; }
73
74         ShaderType output_type() { return current_type; }
75
76         ImageManager *image_manager;
77         ShaderManager *shader_manager;
78         bool background;
79
80 protected:
81         /* stack */
82         struct Stack {
83                 Stack() { memset(users, 0, sizeof(users)); }
84                 Stack(const Stack& other) { memcpy(users, other.users, sizeof(users)); }
85                 Stack& operator=(const Stack& other) { memcpy(users, other.users, sizeof(users)); return *this; }
86
87                 bool empty()
88                 {
89                         for(int i = 0; i < SVM_STACK_SIZE; i++)
90                                 if(users[i])
91                                         return false;
92
93                         return true;
94                 }
95
96                 void print()
97                 {
98                         printf("stack <");
99
100                         for(int i = 0; i < SVM_STACK_SIZE; i++)
101                                 printf((users[i])? "*": " ");
102
103                         printf(">\n");
104                 }
105
106                 int users[SVM_STACK_SIZE];
107         };
108
109         struct StackBackup {
110                 Stack stack;
111                 vector<int> offsets;
112                 set<ShaderNode*> done;
113         };
114
115         void stack_backup(StackBackup& backup, set<ShaderNode*>& done);
116         void stack_restore(StackBackup& backup, set<ShaderNode*>& done);
117
118         void stack_clear_temporary(ShaderNode *node);
119         int stack_size(ShaderSocketType type);
120         void stack_clear_users(ShaderNode *node, set<ShaderNode*>& done);
121
122         bool node_skip_input(ShaderNode *node, ShaderInput *input);
123
124         /* single closure */
125         void find_dependencies(set<ShaderNode*>& dependencies,
126                 const set<ShaderNode*>& done, ShaderInput *input);
127         void generate_node(ShaderNode *node, set<ShaderNode*>& done);
128         void generate_closure_node(ShaderNode *node, set<ShaderNode*>& done);
129         void generated_shared_closure_nodes(ShaderNode *node, set<ShaderNode*>& done,
130                 set<ShaderNode*>& closure_done, const set<ShaderNode*>& shared);
131         void generate_svm_nodes(const set<ShaderNode*>& nodes, set<ShaderNode*>& done);
132
133         /* multi closure */
134         void generate_multi_closure(ShaderNode *node, set<ShaderNode*>& done, set<ShaderNode*>& closure_done);
135
136         /* compile */
137         void compile_type(Shader *shader, ShaderGraph *graph, ShaderType type);
138
139         vector<int4> svm_nodes;
140         ShaderType current_type;
141         Shader *current_shader;
142         ShaderGraph *current_graph;
143         Stack active_stack;
144         int max_stack_use;
145         uint mix_weight_offset;
146         bool compile_failed;
147 };
148
149 CCL_NAMESPACE_END
150
151 #endif /* __SVM_H__ */
152