]> git.tdb.fi Git - libs/gl.git/blob - source/glsl/finalize.cpp
Combine the precision handling visitors and rewrite the logic
[libs/gl.git] / source / glsl / finalize.cpp
1 #include <msp/core/algorithm.h>
2 #include <msp/core/raii.h>
3 #include <msp/strings/lexicalcast.h>
4 #include "finalize.h"
5 #include "glsl_error.h"
6
7 using namespace std;
8
9 namespace Msp {
10 namespace GL {
11 namespace SL {
12
13 PrecisionConverter::PrecisionConverter():
14         stage(0)
15 { }
16
17 void PrecisionConverter::apply(Stage &s)
18 {
19         stage = &s;
20         s.content.visit(*this);
21         NodeRemover().apply(s, nodes_to_remove);
22 }
23
24 void PrecisionConverter::visit(Block &block)
25 {
26         for(NodeList<Statement>::iterator i=block.body.begin(); i!=block.body.end(); ++i)
27         {
28                 if(&block==&stage->content)
29                         insert_point = i;
30                 (*i)->visit(*this);
31         }
32 }
33
34 void PrecisionConverter::visit(Precision &prec)
35 {
36         if(stage->required_features.gl_api==OPENGL_ES2)
37                 have_default.insert(prec.type);
38         else
39                 nodes_to_remove.insert(&prec);
40 }
41
42 void PrecisionConverter::visit(VariableDeclaration &var)
43 {
44         if(stage->required_features.gl_api!=OPENGL_ES2)
45         {
46                 var.precision.clear();
47                 return;
48         }
49
50         const char *default_prec = (stage->type==Stage::FRAGMENT ? "mediump" : "highp");
51         const TypeDeclaration *type = var.type_declaration;
52         while(type)
53         {
54                 if(dynamic_cast<const ImageTypeDeclaration *>(type))
55                 {
56                         default_prec = "lowp";
57                         break;
58                 }
59                 else if(const BasicTypeDeclaration *basic = dynamic_cast<const BasicTypeDeclaration *>(type))
60                 {
61                         if(basic->kind==BasicTypeDeclaration::INT || basic->kind==BasicTypeDeclaration::FLOAT)
62                                 break;
63                         type = basic->base_type;
64                 }
65                 else
66                         return;
67         }
68         if(!type)
69                 return;
70
71         if(!have_default.count(type->name))
72         {
73                 Precision *prec = new Precision;
74                 prec->precision = default_prec;
75                 prec->type = type->name;
76                 stage->content.body.insert(insert_point, prec);
77
78                 have_default.insert(type->name);
79         }
80 }
81
82
83 LegacyConverter::LegacyConverter():
84         frag_out(0)
85 { }
86
87 void LegacyConverter::apply(Stage &s, const Features &feat)
88 {
89         stage = &s;
90         features = feat;
91         if(supports_stage(s.type))
92                 s.content.visit(*this);
93         else
94                 unsupported(format("Stage %s is not supported", Stage::get_stage_name(s.type)));
95 }
96
97 void LegacyConverter::unsupported(const string &reason)
98 {
99         Diagnostic diagnostic;
100         diagnostic.severity = Diagnostic::ERR;
101         diagnostic.source = GENERATED_SOURCE;
102         diagnostic.line = 0;
103         diagnostic.message = reason;
104         stage->diagnostics.push_back(diagnostic);
105 }
106
107 void LegacyConverter::visit(Block &block)
108 {
109         for(NodeList<Statement>::iterator i=block.body.begin(); i!=block.body.end(); ++i)
110         {
111                 if(&block==&stage->content)
112                         uniform_insert_point = i;
113                 (*i)->visit(*this);
114         }
115 }
116
117 bool LegacyConverter::check_version(const Version &feature_version) const
118 {
119         if(features.glsl_version<feature_version)
120                 return false;
121         else if(stage->required_features.glsl_version<feature_version)
122                 stage->required_features.glsl_version = feature_version;
123
124         return true;
125 }
126
127 bool LegacyConverter::check_extension(bool Features::*extension) const
128 {
129         if(!(features.*extension))
130                 return false;
131
132         stage->required_features.*extension = true;
133
134         return true;
135 }
136
137 bool LegacyConverter::supports_stage(Stage::Type st) const
138 {
139         if(st==Stage::GEOMETRY)
140         {
141                 if(features.gl_api==OPENGL_ES2)
142                         return check_version(Version(3, 20));
143                 else
144                         return check_version(Version(1, 50));
145         }
146         else
147                 return true;
148 }
149
150 bool LegacyConverter::supports_unified_interface_syntax() const
151 {
152         if(features.gl_api==OPENGL_ES2)
153                 return check_version(Version(3, 0));
154         else
155                 return check_version(Version(1, 30));
156 }
157
158 void LegacyConverter::visit(VariableReference &var)
159 {
160         if(var.declaration==frag_out && !supports_unified_interface_syntax())
161         {
162                 var.name = "gl_FragColor";
163                 var.declaration = 0;
164                 r_type = "vec4";
165         }
166         else if(var.declaration)
167                 r_type = var.declaration->type;
168         else
169                 r_type.clear();
170 }
171
172 void LegacyConverter::visit(Assignment &assign)
173 {
174         TraversingVisitor::visit(assign);
175         if(assign.target.declaration==frag_out && !supports_unified_interface_syntax())
176                 assign.target.declaration = 0;
177 }
178
179 bool LegacyConverter::supports_unified_sampling_functions() const
180 {
181         if(features.gl_api==OPENGL_ES2)
182                 return check_version(Version(3, 0));
183         else
184                 return check_version(Version(1, 30));
185 }
186
187 void LegacyConverter::visit(FunctionCall &call)
188 {
189         if(call.name=="texture")
190         {
191                 string sampler_type;
192                 r_type.clear();
193                 NodeArray<Expression>::iterator i = call.arguments.begin();
194                 if(i!=call.arguments.end())
195                 {
196                         (*i)->visit(*this);
197                         sampler_type = r_type;
198
199                         for(; i!=call.arguments.end(); ++i)
200                                 (*i)->visit(*this);
201                 }
202
203                 if(!supports_unified_sampling_functions())
204                 {
205                         if(sampler_type=="sampler1D")
206                                 call.name = "texture1D";
207                         else if(sampler_type=="sampler2D")
208                                 call.name = "texture2D";
209                         else if(sampler_type=="sampler3D")
210                                 call.name = "texture3D";
211                         else if(sampler_type=="samplerCube")
212                                 call.name = "textureCube";
213                         else if(sampler_type=="sampler1DShadow")
214                                 call.name = "shadow1D";
215                         else if(sampler_type=="sampler2DShadow")
216                                 call.name = "shadow2D";
217                         else if(sampler_type=="sampler1DArray")
218                         {
219                                 check_extension(&Features::ext_texture_array);
220                                 call.name = "texture1DArray";
221                         }
222                         else if(sampler_type=="sampler2DArray")
223                         {
224                                 check_extension(&Features::ext_texture_array);
225                                 call.name = "texture2DArray";
226                         }
227                         else if(sampler_type=="sampler1DArrayShadow")
228                         {
229                                 check_extension(&Features::ext_texture_array);
230                                 call.name = "shadow1DArray";
231                         }
232                         else if(sampler_type=="sampler2DArrayShadow")
233                         {
234                                 check_extension(&Features::ext_texture_array);
235                                 call.name = "shadow2DArray";
236                         }
237                 }
238         }
239         else
240                 TraversingVisitor::visit(call);
241 }
242
243 bool LegacyConverter::supports_interface_layouts() const
244 {
245         if(features.gl_api==OPENGL_ES2)
246                 return check_version(Version(3, 0));
247         else if(check_version(Version(3, 30)))
248                 return true;
249         else
250                 return check_extension(&Features::arb_explicit_attrib_location);
251 }
252
253 bool LegacyConverter::supports_centroid_sampling() const
254 {
255         if(features.gl_api==OPENGL_ES2)
256                 return check_version(Version(3, 0));
257         else if(check_version(Version(1, 20)))
258                 return true;
259         else
260                 return check_extension(&Features::ext_gpu_shader4);
261 }
262
263 bool LegacyConverter::supports_sample_sampling() const
264 {
265         if(features.gl_api==OPENGL_ES2)
266                 return check_version(Version(3, 20));
267         else if(check_version(Version(4, 0)))
268                 return true;
269         else
270                 return check_extension(&Features::arb_gpu_shader5);
271 }
272
273 void LegacyConverter::visit(VariableDeclaration &var)
274 {
275         if(var.layout && !supports_interface_layouts())
276         {
277                 vector<Layout::Qualifier>::iterator i;
278                 for(i=var.layout->qualifiers.begin(); (i!=var.layout->qualifiers.end() && i->name!="location"); ++i) ;
279                 if(i!=var.layout->qualifiers.end())
280                 {
281                         if(stage->type==Stage::VERTEX && var.interface=="in")
282                         {
283                                 stage->locations[var.name] = i->value;
284                                 var.layout->qualifiers.erase(i);
285                         }
286                         else if(stage->type==Stage::FRAGMENT && var.interface=="out")
287                         {
288                                 if(check_extension(&Features::ext_gpu_shader4))
289                                         stage->locations[var.name] = i->value;
290                                 else if(i->value!=0)
291                                         unsupported("EXT_gpu_shader4 required for multiple fragment shader outputs");
292                                 var.layout->qualifiers.erase(i);
293                         }
294
295                         if(var.layout->qualifiers.empty())
296                                 var.layout = 0;
297                 }
298         }
299
300         if(var.sampling=="centroid")
301         {
302                 if(!supports_centroid_sampling())
303                         var.sampling = string();
304         }
305         else if(var.sampling=="sample")
306         {
307                 if(!supports_sample_sampling())
308                         var.sampling = string();
309         }
310
311         if((var.interface=="in" || var.interface=="out") && !supports_unified_interface_syntax())
312         {
313                 if(stage->type==Stage::FRAGMENT && var.interface=="out")
314                 {
315                         frag_out = &var;
316                         nodes_to_remove.insert(&var);
317                 }
318         }
319
320         TraversingVisitor::visit(var);
321 }
322
323 bool LegacyConverter::supports_interface_blocks(const string &iface) const
324 {
325         if(features.gl_api==OPENGL_ES2)
326         {
327                 if(iface=="uniform")
328                         return check_version(Version(3, 0));
329                 else
330                         return check_version(Version(3, 20));
331         }
332         else if(check_version(Version(1, 50)))
333                 return true;
334         else if(iface=="uniform")
335                 return check_extension(&Features::arb_uniform_buffer_object);
336         else
337                 return false;
338 }
339
340 void LegacyConverter::visit(InterfaceBlock &iface)
341 {
342         if(!supports_interface_blocks(iface.interface) && iface.type_declaration)
343         {
344                 if(!iface.instance_name.empty())
345                         unsupported("ARB_uniform_buffer_object required for interface block instances");
346                 else if(iface.struct_declaration)
347                 {
348                         stage->content.body.splice(uniform_insert_point, iface.struct_declaration->members.body);
349                         nodes_to_remove.insert(&iface);
350                         nodes_to_remove.insert(iface.struct_declaration);
351                 }
352                 else
353                         /* If the interface block is an array, it should have an instance
354                         name too, so this should never be reached */
355                         throw logic_error("Unexpected interface block configuration");
356         }
357 }
358
359 } // namespace SL
360 } // namespace GL
361 } // namespace Msp