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