Generate compatible OSL shader parameter names in cases where the node has and input...
[blender.git] / intern / cycles / render / osl.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 __OSL_H__
20 #define __OSL_H__
21
22 #include "util_set.h"
23
24 #include "shader.h"
25
26 #ifdef WITH_OSL
27 #include <OSL/oslexec.h>
28 #endif
29
30 CCL_NAMESPACE_BEGIN
31
32 class Device;
33 class DeviceScene;
34 class ImageManager;
35 class OSLRenderServices;
36 class OSLGlobals;
37 class Scene;
38 class ShaderGraph;
39 class ShaderNode;
40 class ShaderInput;
41 class ShaderOutput;
42
43 #ifdef WITH_OSL
44
45 /* Shader Manage */
46
47 class OSLShaderManager : public ShaderManager {
48 public:
49         OSLShaderManager();
50         ~OSLShaderManager();
51
52         void device_update(Device *device, DeviceScene *dscene, Scene *scene, Progress& progress);
53         void device_free(Device *device, DeviceScene *dscene);
54
55 private:
56         OSL::ShadingSystem *ss;
57         OSL::TextureSystem *ts;
58         OSLRenderServices *services;
59         OSL::ErrorHandler errhandler;
60 };
61
62 #endif
63
64 /* Graph Compiler */
65
66 class OSLCompiler {
67 public:
68         OSLCompiler(void *shadingsys);
69         void compile(OSLGlobals *og, Shader *shader);
70
71         void add(ShaderNode *node, const char *name);
72
73         void parameter(const char *name, float f);
74         void parameter_color(const char *name, float3 f);
75         void parameter_vector(const char *name, float3 f);
76         void parameter_normal(const char *name, float3 f);
77         void parameter_point(const char *name, float3 f);
78         void parameter(const char *name, int f);
79         void parameter(const char *name, const char *s);
80         void parameter(const char *name, ustring str);
81         void parameter(const char *name, const Transform& tfm);
82
83         void parameter_array(const char *name, const float f[], int arraylen);
84         void parameter_color_array(const char *name, const float f[][3], int arraylen);
85         void parameter_vector_array(const char *name, const float f[][3], int arraylen);
86         void parameter_normal_array(const char *name, const float f[][3], int arraylen);
87         void parameter_point_array(const char *name, const float f[][3], int arraylen);
88         void parameter_array(const char *name, const int f[], int arraylen);
89         void parameter_array(const char *name, const char * const s[], int arraylen);
90         void parameter_array(const char *name, const Transform tfm[], int arraylen);
91
92         ShaderType output_type() { return current_type; }
93
94         bool background;
95
96 private:
97         string id(ShaderNode *node);
98         void compile_type(Shader *shader, ShaderGraph *graph, ShaderType type);
99         bool node_skip_input(ShaderNode *node, ShaderInput *input);
100         string compatible_name(ShaderNode *node, ShaderInput *input);
101         string compatible_name(ShaderNode *node, ShaderOutput *output);
102
103         void find_dependencies(set<ShaderNode*>& dependencies, ShaderInput *input);
104         void generate_nodes(const set<ShaderNode*>& nodes);
105
106         void *shadingsys;
107         ShaderType current_type;
108         Shader *current_shader;
109 };
110
111 CCL_NAMESPACE_END
112
113 #endif /* __OSL_H__  */
114