Cycles: svn merge -r41225:41232 ^/trunk/blender
[blender.git] / intern / cycles / kernel / osl / osl_services.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_SERVICES_H__
20 #define __OSL_SERVICES_H__
21
22 /* OSL Render Services
23  *
24  * Implementation of OSL render services, to retriever matrices, attributes,
25  * textures and point clouds. In principle this should only be accessing
26  * kernel data, but currently we also reach back into the Scene to retrieve
27  * attributes.
28  */
29
30 #include <OSL/oslexec.h>
31 #include <OSL/oslclosure.h>
32
33 #ifdef WITH_PARTIO
34 #include <Partio.h>
35 #endif
36
37 CCL_NAMESPACE_BEGIN
38
39 class Object;
40 class Scene;
41 class Shader;
42 class ShaderData;
43 class float3;
44 class KernelGlobals;
45
46 class OSLRenderServices : public OSL::RendererServices
47 {
48 public:
49         OSLRenderServices();
50         ~OSLRenderServices();
51         
52         void thread_init(KernelGlobals *kernel_globals);
53
54         bool get_matrix(OSL::Matrix44 &result, OSL::TransformationPtr xform, float time);
55         bool get_inverse_matrix(OSL::Matrix44 &result, OSL::TransformationPtr xform, float time);
56         bool get_matrix(OSL::Matrix44 &result, ustring from, float time);
57         bool get_inverse_matrix(OSL::Matrix44 &result, ustring to, float time);
58
59         bool get_array_attribute(void *renderstate, bool derivatives, 
60                 ustring object, TypeDesc type, ustring name,
61                 int index, void *val);
62         bool get_attribute(void *renderstate, bool derivatives, ustring object,
63                 TypeDesc type, ustring name, void *val);
64
65         bool get_userdata(bool derivatives, ustring name, TypeDesc type, 
66                 void *renderstate, void *val);
67         bool has_userdata(ustring name, TypeDesc type, void *renderstate);
68
69         void *get_pointcloud_attr_query(ustring *attr_names,
70                 TypeDesc *attr_types, int nattrs);
71         int pointcloud(ustring filename, const OSL::Vec3 &center, float radius,
72                 int max_points, void *attr_query, void **attr_outdata);
73
74 private:
75         KernelGlobals *kernel_globals;
76
77 #ifdef WITH_PARTIO
78         /* OSL gets pointers to this but its definition is private.
79            right now it only caches the types already converted to
80            Partio constants. this is what get_pointcloud_attr_query
81            returns */
82         struct AttrQuery
83         {
84                 /* names of the attributes to query */
85                 std::vector<ustring> attr_names;
86                 /* types as (enum Partio::ParticleAttributeType) of the
87                    attributes in the query */
88                 std::vector<int> attr_partio_types;
89                 /* for sanity checks, capacity of the output arrays */
90                 int capacity;
91         };
92
93         Partio::ParticlesData *get_pointcloud(ustring filename);
94
95         /* keep a list so adding elements doesn't invalidate pointers */
96         std::list<AttrQuery> m_attr_queries;
97 #endif
98
99         static ustring u_distance;
100         static ustring u_index;
101         static ustring u_camera;
102         static ustring u_screen;
103         static ustring u_raster;
104         static ustring u_ndc;
105         static ustring u_empty;
106 };
107
108 CCL_NAMESPACE_END
109
110 #endif /* __OSL_SERVICES_H__  */
111