svn merge ^/trunk/blender -r55776:55813
[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 #include "util_string.h"
24 #include "util_thread.h"
25
26 #include "shader.h"
27
28 #ifdef WITH_OSL
29 #include <OSL/oslcomp.h>
30 #include <OSL/oslexec.h>
31 #include <OSL/oslquery.h>
32 #endif
33
34 CCL_NAMESPACE_BEGIN
35
36 class Device;
37 class DeviceScene;
38 class ImageManager;
39 class OSLRenderServices;
40 struct OSLGlobals;
41 class Scene;
42 class ShaderGraph;
43 class ShaderNode;
44 class ShaderInput;
45 class ShaderOutput;
46
47 #ifdef WITH_OSL
48
49 /* OSL Shader Info
50  * to auto detect closures in the shader for MIS and transparent shadows */
51
52 struct OSLShaderInfo {
53         OSLShaderInfo()
54         : has_surface_emission(false), has_surface_transparent(false),
55           has_surface_bssrdf(false)
56         {}
57
58         bool has_surface_emission;
59         bool has_surface_transparent;
60         bool has_surface_bssrdf;
61 };
62
63 /* Shader Manage */
64
65 class OSLShaderManager : public ShaderManager {
66 public:
67         OSLShaderManager();
68         ~OSLShaderManager();
69
70         void reset(Scene *scene);
71
72         bool use_osl() { return true; }
73
74         void device_update(Device *device, DeviceScene *dscene, Scene *scene, Progress& progress);
75         void device_free(Device *device, DeviceScene *dscene, Scene *scene);
76
77         /* osl compile and query */
78         static bool osl_compile(const string& inputfile, const string& outputfile);
79         static bool osl_query(OSL::OSLQuery& query, const string& filepath);
80
81         /* shader file loading, all functions return pointer to hash string if found */
82         const char *shader_test_loaded(const string& hash);
83         const char *shader_load_bytecode(const string& hash, const string& bytecode);
84         const char *shader_load_filepath(string filepath);
85         OSLShaderInfo *shader_loaded_info(const string& hash);
86
87 protected:
88         void texture_system_init();
89         void shading_system_init();
90
91         OSL::ShadingSystem *ss;
92         OSL::TextureSystem *ts;
93         OSLRenderServices *services;
94         OSL::ErrorHandler errhandler;
95         map<string, OSLShaderInfo> loaded_shaders;
96
97         static OSL::TextureSystem *ts_shared;
98         static thread_mutex ts_shared_mutex;
99         static int ts_shared_users;
100 };
101
102 #endif
103
104 /* Graph Compiler */
105
106 class OSLCompiler {
107 public:
108         OSLCompiler(void *manager, void *shadingsys, ImageManager *image_manager);
109         void compile(OSLGlobals *og, Shader *shader);
110
111         void add(ShaderNode *node, const char *name, bool isfilepath = false);
112
113         void parameter(const char *name, float f);
114         void parameter_color(const char *name, float3 f);
115         void parameter_vector(const char *name, float3 f);
116         void parameter_normal(const char *name, float3 f);
117         void parameter_point(const char *name, float3 f);
118         void parameter(const char *name, int f);
119         void parameter(const char *name, const char *s);
120         void parameter(const char *name, ustring str);
121         void parameter(const char *name, const Transform& tfm);
122
123         void parameter_array(const char *name, const float f[], int arraylen);
124         void parameter_color_array(const char *name, const float f[][3], int arraylen);
125         void parameter_vector_array(const char *name, const float f[][3], int arraylen);
126         void parameter_normal_array(const char *name, const float f[][3], int arraylen);
127         void parameter_point_array(const char *name, const float f[][3], int arraylen);
128         void parameter_array(const char *name, const int f[], int arraylen);
129         void parameter_array(const char *name, const char * const s[], int arraylen);
130         void parameter_array(const char *name, const Transform tfm[], int arraylen);
131
132         ShaderType output_type() { return current_type; }
133
134         bool background;
135         ImageManager *image_manager;
136
137 private:
138         string id(ShaderNode *node);
139         void compile_type(Shader *shader, ShaderGraph *graph, ShaderType type);
140         bool node_skip_input(ShaderNode *node, ShaderInput *input);
141         string compatible_name(ShaderNode *node, ShaderInput *input);
142         string compatible_name(ShaderNode *node, ShaderOutput *output);
143
144         void find_dependencies(set<ShaderNode*>& dependencies, ShaderInput *input);
145         void generate_nodes(const set<ShaderNode*>& nodes);
146
147         void *shadingsys;
148         void *manager;
149         ShaderType current_type;
150         Shader *current_shader;
151 };
152
153 CCL_NAMESPACE_END
154
155 #endif /* __OSL_H__  */
156