Cycles: svn merge -r37904:37957 https://svn.blender.org/svnroot/bf-blender/trunk...
[blender.git] / intern / cycles / render / svm.h
1 /*
2  * Copyright 2011, Blender Foundation.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  */
18
19 #ifndef __SVM_H__
20 #define __SVM_H__
21
22 #include "attribute.h"
23 #include "graph.h"
24 #include "shader.h"
25
26 #include "util_set.h"
27
28 CCL_NAMESPACE_BEGIN
29
30 class Device;
31 class DeviceScene;
32 class ImageManager;
33 struct KernelSunSky;
34 class Scene;
35 class ShaderGraph;
36 class ShaderInput;
37 class ShaderNode;
38 class ShaderOutput;
39
40 /* Shader Manager */
41
42 class SVMShaderManager : public ShaderManager {
43 public:
44         SVMShaderManager();
45         ~SVMShaderManager();
46
47         void device_update(Device *device, DeviceScene *dscene, Scene *scene, Progress& progress);
48         void device_free(Device *device, DeviceScene *dscene);
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         void stack_link(ShaderInput *input, ShaderOutput *output);
61         void add_node(NodeType type, int a = 0, int b = 0, int c = 0);
62         void add_node(int a = 0, int b = 0, int c = 0, int d = 0);
63         void add_node(NodeType type, const float3& f);
64         void add_node(const float4& f);
65         uint attribute(ustring name);
66         uint attribute(Attribute::Standard std);
67         uint encode_uchar4(uint x, uint y = 0, uint z = 0, uint w = 0);
68
69         ShaderType output_type() { return current_type; }
70
71         ImageManager *image_manager;
72         ShaderManager *shader_manager;
73         KernelSunSky *sunsky;
74         bool background;
75
76 protected:
77         struct Stack {
78                 int users[SVM_STACK_SIZE];
79         };
80
81         void stack_clear_temporary(ShaderNode *node);
82         int stack_size(ShaderSocketType type);
83         int stack_find_offset(ShaderSocketType type);
84         void stack_clear_users(ShaderNode *node, set<ShaderNode*>& done);
85
86         bool node_skip_input(ShaderNode *node, ShaderInput *input);
87
88         void find_dependencies(set<ShaderNode*>& dependencies, const set<ShaderNode*>& done, ShaderInput *input);
89         void generate_svm_nodes(const set<ShaderNode*>& nodes, set<ShaderNode*>& done);
90         void generate_closure(ShaderNode *node, set<ShaderNode*> done, Stack stack);
91
92         void compile_type(Shader *shader, ShaderGraph *graph, ShaderType type);
93
94         vector<int4> svm_nodes;
95         ShaderType current_type;
96         Shader *current_shader;
97         Stack active_stack;
98         int max_stack_use;
99 };
100
101 CCL_NAMESPACE_END
102
103 #endif /* __SVM_H__ */
104