Add inline data items to the collection
[libs/gl.git] / source / materials / renderpass.cpp
1 #include <msp/core/algorithm.h>
2 #include <msp/datafile/collection.h>
3 #include <msp/io/print.h>
4 #include <msp/strings/format.h>
5 #include "error.h"
6 #include "renderpass.h"
7 #include "program.h"
8 #include "programdata.h"
9 #include "renderer.h"
10 #include "texture.h"
11 #include "texture2d.h"
12 #include "texturing.h"
13 #include "uniform.h"
14
15 using namespace std;
16
17 namespace Msp {
18 namespace GL {
19
20 RenderPass::RenderPass():
21         shprog(0),
22         shprog_from_material(false),
23         shdata(0),
24         material(0),
25         back_faces(false),
26         receive_shadows(false),
27         image_based_lighting(false)
28 { }
29
30 void RenderPass::set_material_textures()
31 {
32         const Tag *material_texture_tags = material->get_texture_tags();
33         for(const Tag *tag=material_texture_tags; tag->id; ++tag)
34                 set_texture(*tag, material->get_texture(*tag), material->get_sampler());
35 }
36
37 void RenderPass::maybe_create_material_shader()
38 {
39         if(shprog && !shprog_from_material)
40                 return;
41
42         map<string, int> extra_spec;
43         if(receive_shadows)
44                 extra_spec["use_shadow_map"] = true;
45         if(image_based_lighting)
46                 extra_spec["use_image_based_lighting"] = true;
47
48         shprog = material->create_compatible_shader(extra_spec);
49
50         if(shdata)
51                 shdata = new ProgramData(*shdata, shprog);
52
53         shprog_from_material = true;
54 }
55
56 void RenderPass::set_shader_program(const Program *prog, const ProgramData *data)
57 {
58         shprog = prog;
59         shprog_from_material = false;
60         shdata = (data ? new ProgramData(*data) : 0);
61 }
62
63 Tag RenderPass::get_slotted_uniform_tag(Tag slot) const
64 {
65         map<Tag, Tag>::const_iterator i = uniform_slots.find(slot);
66         if(i==uniform_slots.end())
67                 return Tag();
68         return i->second;
69 }
70
71 void RenderPass::set_material(const Material *mat)
72 {
73         material = mat;
74         maybe_create_material_shader();
75         set_material_textures();
76 }
77
78 void RenderPass::set_texture(Tag tag, const Texture *tex, const Sampler *samp)
79 {
80         vector<TextureSlot>::iterator i = find_member(textures, tag, &TextureSlot::tag);
81         if(i==textures.end())
82         {
83                 textures.push_back(TextureSlot(tag));
84                 i = textures.end()-1;
85         }
86         i->texture = tex;
87         if(samp)
88                 i->sampler = samp;
89 }
90
91 Tag RenderPass::get_texture_tag(const string &slot) const
92 {
93         vector<TextureSlot>::const_iterator i = find_member(textures, slot, &TextureSlot::slot_name);
94         return (i!=textures.end() ? i->tag : Tag());
95 }
96
97 void RenderPass::set_texture(unsigned index, const Texture *tex, const Sampler *samp)
98 {
99         if(!shprog)
100                 throw invalid_operation("RenderPass::set_texture");
101
102         const vector<Program::UniformInfo> &uniforms = shprog->get_uniforms();
103         for(vector<Program::UniformInfo>::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
104                 if(is_image(i->type) && i->binding==static_cast<int>(index))
105                         return set_texture(i->tag, tex, samp);
106
107         if(shdata)
108         {
109                 const vector<Tag> &tags = shdata->get_uniform_tags();
110                 for(vector<Tag>::const_iterator i=tags.begin(); i!=tags.end(); ++i)
111                 {
112                         vector<Program::UniformInfo>::const_iterator j = find_member(uniforms, *i, &Program::UniformInfo::tag);
113                         if(j==uniforms.end() || !is_image(j->type))
114                                 continue;
115                         if(const Uniform1i *uni1i = dynamic_cast<const Uniform1i *>(shdata->find_uniform(*i)))
116                                 if(uni1i->get()==static_cast<int>(index))
117                                         return set_texture(*i, tex, samp);
118                 }
119         }
120 }
121
122 int RenderPass::get_texture_index(const string &n) const
123 {
124         vector<TextureSlot>::const_iterator i = find_member(textures, n, &TextureSlot::slot_name);
125         return (shprog && i!=textures.end() ? shprog->get_uniform_binding(i->tag) : -1);
126 }
127
128 void RenderPass::set_back_faces(bool bf)
129 {
130         back_faces = bf;
131 }
132
133 void RenderPass::set_receive_shadows(bool rs)
134 {
135         receive_shadows = rs;
136 }
137
138 void RenderPass::apply(Renderer &renderer) const
139 {
140         for(vector<TextureSlot>::const_iterator i=textures.begin(); i!=textures.end(); ++i)
141                 renderer.set_texture(i->tag, i->texture, i->sampler);
142         renderer.set_material(material);
143         renderer.set_shader_program(shprog, shdata.get());
144         renderer.set_reverse_winding(back_faces);
145 }
146
147 void RenderPass::set_debug_name(const string &name)
148 {
149 #ifdef DEBUG
150         if(shdata.refcount()==1)
151                 shdata->set_debug_name(name+" [UBO]");
152 #else
153         (void)name;
154 #endif
155 }
156
157
158 DataFile::Loader::ActionMap RenderPass::Loader::shared_actions;
159
160 RenderPass::Loader::Loader(RenderPass &p):
161         DataFile::CollectionObjectLoader<RenderPass>(p, 0)
162 {
163         set_actions(shared_actions);
164 }
165
166 RenderPass::Loader::Loader(RenderPass &p, Collection &c):
167         DataFile::CollectionObjectLoader<RenderPass>(p, &c)
168 {
169         set_actions(shared_actions);
170 }
171
172 void RenderPass::Loader::init_actions()
173 {
174         add("shader",   &Loader::shader);
175         add("image_based_lighting", &RenderPass::image_based_lighting);
176         add("material", &Loader::material_inline);
177         add("material", &Loader::material);
178         add("material_slot", &RenderPass::material_slot);
179         add("back_faces",&RenderPass::back_faces);
180         add("receive_shadows", &RenderPass::receive_shadows);
181         add("texture", &Loader::texture);
182         add("uniforms", &Loader::uniforms);
183         add("uniform_slot", &Loader::uniform_slot);
184         add("uniform_slot", &Loader::uniform_slot2);
185
186         // Deprecated
187         add("texunit",  &Loader::texunit);
188         add("texunit",  &Loader::texture);
189         add("texunit",  &Loader::texunit_named);
190 }
191
192 void RenderPass::Loader::set_inline_base_name(const string &n)
193 {
194         inline_base_name = n;
195 }
196
197 void RenderPass::Loader::finish()
198 {
199         if(obj.material)
200                 obj.maybe_create_material_shader();
201 }
202
203 // Temporary compatibility feature
204 string RenderPass::Loader::get_shader_name(const string &n)
205 {
206         if(n.size()>=5 && !n.compare(n.size()-5, 5, ".glsl"))
207         {
208                 IO::print(IO::cerr, "Warning: Loading module '%s' as shader is deprecated\n", n);
209                 return n+".shader";
210         }
211         return n;
212 }
213
214 void RenderPass::Loader::material_inline()
215 {
216         Material::GenericLoader ldr(coll);
217         load_sub_with(ldr);
218         RefPtr<Material> mat = ldr.get_material();
219         get_collection().add(inline_base_name+".mat", mat.get());
220         obj.material = mat.release();
221         obj.set_material_textures();
222 }
223
224 void RenderPass::Loader::material(const string &name)
225 {
226         obj.material = &get_collection().get<Material>(name);
227         obj.set_material_textures();
228 }
229
230 void RenderPass::Loader::shader(const string &n)
231 {
232         obj.shprog = &get_collection().get<Program>(get_shader_name(n));
233         obj.shprog_from_material = false;
234         if(obj.shdata)
235                 obj.shdata = new ProgramData(*obj.shdata, obj.shprog);
236 }
237
238 void RenderPass::Loader::texture(const string &n)
239 {
240         vector<TextureSlot>::iterator i = find_member(obj.textures, Tag(n), &TextureSlot::tag);
241         if(i==obj.textures.end())
242         {
243                 obj.textures.push_back(TextureSlot(n));
244                 i = obj.textures.end()-1;
245         }
246         TextureSlot::Loader ldr(*i, n, coll);
247         load_sub_with(ldr);
248 }
249
250 void RenderPass::Loader::texunit(unsigned)
251 {
252         IO::print(IO::cerr, "Warning: specifying textures by unit number is deprecated and may not produce expected results");
253         string name;
254         if(obj.shprog)
255         {
256                 const vector<Program::UniformInfo> &uniforms = obj.shprog->get_uniforms();
257                 for(vector<Program::UniformInfo>::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
258                         if(is_image(i->type) && i->binding>=0)
259                         {
260                                 if(!name.empty())
261                                 {
262                                         name.clear();
263                                         break;
264                                 }
265                                 name = i->name;
266                         }
267         }
268
269         if(name.empty())
270                 throw runtime_error("Could not determine name for texture");
271
272         texture(name);
273 }
274
275 void RenderPass::Loader::texunit_named(unsigned, const string &n)
276 {
277         texture(n);
278 }
279
280 void RenderPass::Loader::uniforms()
281 {
282         if(!obj.shprog || obj.shprog_from_material)
283                 throw runtime_error("Shader is required for uniforms");
284         if(!obj.shdata)
285                 obj.shdata = new ProgramData(obj.shprog);
286         else if(obj.shdata.refcount()>1)
287                 obj.shdata = new ProgramData(*obj.shdata);
288         load_sub(*obj.shdata);
289 }
290
291 void RenderPass::Loader::uniform_slot(const string &name)
292 {
293         uniform_slot2(name, name);
294 }
295
296 void RenderPass::Loader::uniform_slot2(const string &name, const string &slot)
297 {
298         obj.uniform_slots[slot] = name;
299 }
300
301
302 RenderPass::TextureSlot::Loader::Loader(TextureSlot &ts, const string &an, Collection *c):
303         CollectionObjectLoader<TextureSlot>(ts, c),
304         auto_slot_name(an)
305 {
306         add("sampler", &TextureSlot::sampler);
307         add("slot", &Loader::slot_auto);
308         add("slot", &TextureSlot::slot_name);
309         add("texture", &TextureSlot::texture);
310 }
311
312 void RenderPass::TextureSlot::Loader::slot_auto()
313 {
314         obj.slot_name = auto_slot_name;
315 }
316
317 } // namespace GL
318 } // namespace Msp