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