nir: pack nir_variable::data::stream
[mesa.git] / src / compiler / glsl / glsl_to_nir.cpp
1 /*
2 * Copyright © 2014 Intel Corporation
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
13 * Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
21 * IN THE SOFTWARE.
22 *
23 * Authors:
24 * Connor Abbott (cwabbott0@gmail.com)
25 *
26 */
27
28 #include "float64_glsl.h"
29 #include "glsl_to_nir.h"
30 #include "ir_visitor.h"
31 #include "ir_hierarchical_visitor.h"
32 #include "ir.h"
33 #include "ir_optimization.h"
34 #include "program.h"
35 #include "compiler/nir/nir_control_flow.h"
36 #include "compiler/nir/nir_builder.h"
37 #include "compiler/nir/nir_builtin_builder.h"
38 #include "compiler/nir/nir_deref.h"
39 #include "main/errors.h"
40 #include "main/imports.h"
41 #include "main/mtypes.h"
42 #include "main/shaderobj.h"
43 #include "util/u_math.h"
44
45 /*
46 * pass to lower GLSL IR to NIR
47 *
48 * This will lower variable dereferences to loads/stores of corresponding
49 * variables in NIR - the variables will be converted to registers in a later
50 * pass.
51 */
52
53 namespace {
54
55 class nir_visitor : public ir_visitor
56 {
57 public:
58 nir_visitor(gl_context *ctx, nir_shader *shader);
59 ~nir_visitor();
60
61 virtual void visit(ir_variable *);
62 virtual void visit(ir_function *);
63 virtual void visit(ir_function_signature *);
64 virtual void visit(ir_loop *);
65 virtual void visit(ir_if *);
66 virtual void visit(ir_discard *);
67 virtual void visit(ir_demote *);
68 virtual void visit(ir_loop_jump *);
69 virtual void visit(ir_return *);
70 virtual void visit(ir_call *);
71 virtual void visit(ir_assignment *);
72 virtual void visit(ir_emit_vertex *);
73 virtual void visit(ir_end_primitive *);
74 virtual void visit(ir_expression *);
75 virtual void visit(ir_swizzle *);
76 virtual void visit(ir_texture *);
77 virtual void visit(ir_constant *);
78 virtual void visit(ir_dereference_variable *);
79 virtual void visit(ir_dereference_record *);
80 virtual void visit(ir_dereference_array *);
81 virtual void visit(ir_barrier *);
82
83 void create_function(ir_function_signature *ir);
84
85 private:
86 void add_instr(nir_instr *instr, unsigned num_components, unsigned bit_size);
87 nir_ssa_def *evaluate_rvalue(ir_rvalue *ir);
88
89 nir_alu_instr *emit(nir_op op, unsigned dest_size, nir_ssa_def **srcs);
90 nir_alu_instr *emit(nir_op op, unsigned dest_size, nir_ssa_def *src1);
91 nir_alu_instr *emit(nir_op op, unsigned dest_size, nir_ssa_def *src1,
92 nir_ssa_def *src2);
93 nir_alu_instr *emit(nir_op op, unsigned dest_size, nir_ssa_def *src1,
94 nir_ssa_def *src2, nir_ssa_def *src3);
95
96 bool supports_std430;
97
98 nir_shader *shader;
99 nir_function_impl *impl;
100 nir_builder b;
101 nir_ssa_def *result; /* result of the expression tree last visited */
102
103 nir_deref_instr *evaluate_deref(ir_instruction *ir);
104
105 nir_constant *constant_copy(ir_constant *ir, void *mem_ctx);
106
107 /* most recent deref instruction created */
108 nir_deref_instr *deref;
109
110 /* whether the IR we're operating on is per-function or global */
111 bool is_global;
112
113 ir_function_signature *sig;
114
115 /* map of ir_variable -> nir_variable */
116 struct hash_table *var_table;
117
118 /* map of ir_function_signature -> nir_function_overload */
119 struct hash_table *overload_table;
120 };
121
122 /*
123 * This visitor runs before the main visitor, calling create_function() for
124 * each function so that the main visitor can resolve forward references in
125 * calls.
126 */
127
128 class nir_function_visitor : public ir_hierarchical_visitor
129 {
130 public:
131 nir_function_visitor(nir_visitor *v) : visitor(v)
132 {
133 }
134 virtual ir_visitor_status visit_enter(ir_function *);
135
136 private:
137 nir_visitor *visitor;
138 };
139
140 /* glsl_to_nir can only handle converting certain function paramaters
141 * to NIR. This visitor checks for parameters it can't currently handle.
142 */
143 class ir_function_param_visitor : public ir_hierarchical_visitor
144 {
145 public:
146 ir_function_param_visitor()
147 : unsupported(false)
148 {
149 }
150
151 virtual ir_visitor_status visit_enter(ir_function_signature *ir)
152 {
153
154 if (ir->is_intrinsic())
155 return visit_continue;
156
157 foreach_in_list(ir_variable, param, &ir->parameters) {
158 if (!param->type->is_vector() || !param->type->is_scalar()) {
159 unsupported = true;
160 return visit_stop;
161 }
162
163 if (param->data.mode == ir_var_function_inout) {
164 unsupported = true;
165 return visit_stop;
166 }
167 }
168
169 return visit_continue;
170 }
171
172 bool unsupported;
173 };
174
175 } /* end of anonymous namespace */
176
177
178 static bool
179 has_unsupported_function_param(exec_list *ir)
180 {
181 ir_function_param_visitor visitor;
182 visit_list_elements(&visitor, ir);
183 return visitor.unsupported;
184 }
185
186 nir_shader *
187 glsl_to_nir(struct gl_context *ctx,
188 const struct gl_shader_program *shader_prog,
189 gl_shader_stage stage,
190 const nir_shader_compiler_options *options)
191 {
192 struct gl_linked_shader *sh = shader_prog->_LinkedShaders[stage];
193
194 const struct gl_shader_compiler_options *gl_options =
195 &ctx->Const.ShaderCompilerOptions[stage];
196
197 /* glsl_to_nir can only handle converting certain function paramaters
198 * to NIR. If we find something we can't handle then we get the GLSL IR
199 * opts to remove it before we continue on.
200 *
201 * TODO: add missing glsl ir to nir support and remove this loop.
202 */
203 while (has_unsupported_function_param(sh->ir)) {
204 do_common_optimization(sh->ir, true, true, gl_options,
205 ctx->Const.NativeIntegers);
206 }
207
208 nir_shader *shader = nir_shader_create(NULL, stage, options,
209 &sh->Program->info);
210
211 nir_visitor v1(ctx, shader);
212 nir_function_visitor v2(&v1);
213 v2.run(sh->ir);
214 visit_exec_list(sh->ir, &v1);
215
216 nir_validate_shader(shader, "after glsl to nir, before function inline");
217
218 /* We have to lower away local constant initializers right before we
219 * inline functions. That way they get properly initialized at the top
220 * of the function and not at the top of its caller.
221 */
222 nir_lower_constant_initializers(shader, (nir_variable_mode)~0);
223 nir_lower_returns(shader);
224 nir_inline_functions(shader);
225 nir_opt_deref(shader);
226
227 nir_validate_shader(shader, "after function inlining and return lowering");
228
229 /* Now that we have inlined everything remove all of the functions except
230 * main().
231 */
232 foreach_list_typed_safe(nir_function, function, node, &(shader)->functions){
233 if (strcmp("main", function->name) != 0) {
234 exec_node_remove(&function->node);
235 }
236 }
237
238 /* Remap the locations to slots so those requiring two slots will occupy
239 * two locations. For instance, if we have in the IR code a dvec3 attr0 in
240 * location 0 and vec4 attr1 in location 1, in NIR attr0 will use
241 * locations/slots 0 and 1, and attr1 will use location/slot 2 */
242 if (shader->info.stage == MESA_SHADER_VERTEX)
243 nir_remap_dual_slot_attributes(shader, &sh->Program->DualSlotInputs);
244
245 shader->info.name = ralloc_asprintf(shader, "GLSL%d", shader_prog->Name);
246 if (shader_prog->Label)
247 shader->info.label = ralloc_strdup(shader, shader_prog->Label);
248
249 /* Check for transform feedback varyings specified via the API */
250 shader->info.has_transform_feedback_varyings =
251 shader_prog->TransformFeedback.NumVarying > 0;
252
253 /* Check for transform feedback varyings specified in the Shader */
254 if (shader_prog->last_vert_prog)
255 shader->info.has_transform_feedback_varyings |=
256 shader_prog->last_vert_prog->sh.LinkedTransformFeedback->NumVarying > 0;
257
258 if (shader->info.stage == MESA_SHADER_FRAGMENT) {
259 shader->info.fs.pixel_center_integer = sh->Program->info.fs.pixel_center_integer;
260 shader->info.fs.origin_upper_left = sh->Program->info.fs.origin_upper_left;
261 }
262
263 return shader;
264 }
265
266 nir_visitor::nir_visitor(gl_context *ctx, nir_shader *shader)
267 {
268 this->supports_std430 = ctx->Const.UseSTD430AsDefaultPacking;
269 this->shader = shader;
270 this->is_global = true;
271 this->var_table = _mesa_pointer_hash_table_create(NULL);
272 this->overload_table = _mesa_pointer_hash_table_create(NULL);
273 this->result = NULL;
274 this->impl = NULL;
275 this->deref = NULL;
276 this->sig = NULL;
277 memset(&this->b, 0, sizeof(this->b));
278 }
279
280 nir_visitor::~nir_visitor()
281 {
282 _mesa_hash_table_destroy(this->var_table, NULL);
283 _mesa_hash_table_destroy(this->overload_table, NULL);
284 }
285
286 nir_deref_instr *
287 nir_visitor::evaluate_deref(ir_instruction *ir)
288 {
289 ir->accept(this);
290 return this->deref;
291 }
292
293 nir_constant *
294 nir_visitor::constant_copy(ir_constant *ir, void *mem_ctx)
295 {
296 if (ir == NULL)
297 return NULL;
298
299 nir_constant *ret = rzalloc(mem_ctx, nir_constant);
300
301 const unsigned rows = ir->type->vector_elements;
302 const unsigned cols = ir->type->matrix_columns;
303 unsigned i;
304
305 ret->num_elements = 0;
306 switch (ir->type->base_type) {
307 case GLSL_TYPE_UINT:
308 /* Only float base types can be matrices. */
309 assert(cols == 1);
310
311 for (unsigned r = 0; r < rows; r++)
312 ret->values[r].u32 = ir->value.u[r];
313
314 break;
315
316 case GLSL_TYPE_INT:
317 /* Only float base types can be matrices. */
318 assert(cols == 1);
319
320 for (unsigned r = 0; r < rows; r++)
321 ret->values[r].i32 = ir->value.i[r];
322
323 break;
324
325 case GLSL_TYPE_FLOAT:
326 case GLSL_TYPE_DOUBLE:
327 if (cols > 1) {
328 ret->elements = ralloc_array(mem_ctx, nir_constant *, cols);
329 ret->num_elements = cols;
330 for (unsigned c = 0; c < cols; c++) {
331 nir_constant *col_const = rzalloc(mem_ctx, nir_constant);
332 col_const->num_elements = 0;
333 switch (ir->type->base_type) {
334 case GLSL_TYPE_FLOAT:
335 for (unsigned r = 0; r < rows; r++)
336 col_const->values[r].f32 = ir->value.f[c * rows + r];
337 break;
338
339 case GLSL_TYPE_DOUBLE:
340 for (unsigned r = 0; r < rows; r++)
341 col_const->values[r].f64 = ir->value.d[c * rows + r];
342 break;
343
344 default:
345 unreachable("Cannot get here from the first level switch");
346 }
347 ret->elements[c] = col_const;
348 }
349 } else {
350 switch (ir->type->base_type) {
351 case GLSL_TYPE_FLOAT:
352 for (unsigned r = 0; r < rows; r++)
353 ret->values[r].f32 = ir->value.f[r];
354 break;
355
356 case GLSL_TYPE_DOUBLE:
357 for (unsigned r = 0; r < rows; r++)
358 ret->values[r].f64 = ir->value.d[r];
359 break;
360
361 default:
362 unreachable("Cannot get here from the first level switch");
363 }
364 }
365 break;
366
367 case GLSL_TYPE_UINT64:
368 /* Only float base types can be matrices. */
369 assert(cols == 1);
370
371 for (unsigned r = 0; r < rows; r++)
372 ret->values[r].u64 = ir->value.u64[r];
373 break;
374
375 case GLSL_TYPE_INT64:
376 /* Only float base types can be matrices. */
377 assert(cols == 1);
378
379 for (unsigned r = 0; r < rows; r++)
380 ret->values[r].i64 = ir->value.i64[r];
381 break;
382
383 case GLSL_TYPE_BOOL:
384 /* Only float base types can be matrices. */
385 assert(cols == 1);
386
387 for (unsigned r = 0; r < rows; r++)
388 ret->values[r].b = ir->value.b[r];
389
390 break;
391
392 case GLSL_TYPE_STRUCT:
393 case GLSL_TYPE_ARRAY:
394 ret->elements = ralloc_array(mem_ctx, nir_constant *,
395 ir->type->length);
396 ret->num_elements = ir->type->length;
397
398 for (i = 0; i < ir->type->length; i++)
399 ret->elements[i] = constant_copy(ir->const_elements[i], mem_ctx);
400 break;
401
402 default:
403 unreachable("not reached");
404 }
405
406 return ret;
407 }
408
409 static const glsl_type *
410 wrap_type_in_array(const glsl_type *elem_type, const glsl_type *array_type)
411 {
412 if (!array_type->is_array())
413 return elem_type;
414
415 elem_type = wrap_type_in_array(elem_type, array_type->fields.array);
416
417 return glsl_type::get_array_instance(elem_type, array_type->length);
418 }
419
420 void
421 nir_visitor::visit(ir_variable *ir)
422 {
423 /* TODO: In future we should switch to using the NIR lowering pass but for
424 * now just ignore these variables as GLSL IR should have lowered them.
425 * Anything remaining are just dead vars that weren't cleaned up.
426 */
427 if (ir->data.mode == ir_var_shader_shared)
428 return;
429
430 /* FINISHME: inout parameters */
431 assert(ir->data.mode != ir_var_function_inout);
432
433 if (ir->data.mode == ir_var_function_out)
434 return;
435
436 nir_variable *var = rzalloc(shader, nir_variable);
437 var->type = ir->type;
438 var->name = ralloc_strdup(var, ir->name);
439
440 var->data.always_active_io = ir->data.always_active_io;
441 var->data.read_only = ir->data.read_only;
442 var->data.centroid = ir->data.centroid;
443 var->data.sample = ir->data.sample;
444 var->data.patch = ir->data.patch;
445 var->data.invariant = ir->data.invariant;
446 var->data.location = ir->data.location;
447 var->data.stream = ir->data.stream;
448 if (ir->data.stream & (1u << 31))
449 var->data.stream |= NIR_STREAM_PACKED;
450 var->data.compact = false;
451
452 switch(ir->data.mode) {
453 case ir_var_auto:
454 case ir_var_temporary:
455 if (is_global)
456 var->data.mode = nir_var_shader_temp;
457 else
458 var->data.mode = nir_var_function_temp;
459 break;
460
461 case ir_var_function_in:
462 case ir_var_const_in:
463 var->data.mode = nir_var_function_temp;
464 break;
465
466 case ir_var_shader_in:
467 if (shader->info.stage == MESA_SHADER_GEOMETRY &&
468 ir->data.location == VARYING_SLOT_PRIMITIVE_ID) {
469 /* For whatever reason, GLSL IR makes gl_PrimitiveIDIn an input */
470 var->data.location = SYSTEM_VALUE_PRIMITIVE_ID;
471 var->data.mode = nir_var_system_value;
472 } else {
473 var->data.mode = nir_var_shader_in;
474
475 if (shader->info.stage == MESA_SHADER_TESS_EVAL &&
476 (ir->data.location == VARYING_SLOT_TESS_LEVEL_INNER ||
477 ir->data.location == VARYING_SLOT_TESS_LEVEL_OUTER)) {
478 var->data.compact = ir->type->without_array()->is_scalar();
479 }
480
481 if (shader->info.stage > MESA_SHADER_VERTEX &&
482 ir->data.location >= VARYING_SLOT_CLIP_DIST0 &&
483 ir->data.location <= VARYING_SLOT_CULL_DIST1) {
484 var->data.compact = ir->type->without_array()->is_scalar();
485 }
486 }
487 break;
488
489 case ir_var_shader_out:
490 var->data.mode = nir_var_shader_out;
491 if (shader->info.stage == MESA_SHADER_TESS_CTRL &&
492 (ir->data.location == VARYING_SLOT_TESS_LEVEL_INNER ||
493 ir->data.location == VARYING_SLOT_TESS_LEVEL_OUTER)) {
494 var->data.compact = ir->type->without_array()->is_scalar();
495 }
496
497 if (shader->info.stage <= MESA_SHADER_GEOMETRY &&
498 ir->data.location >= VARYING_SLOT_CLIP_DIST0 &&
499 ir->data.location <= VARYING_SLOT_CULL_DIST1) {
500 var->data.compact = ir->type->without_array()->is_scalar();
501 }
502 break;
503
504 case ir_var_uniform:
505 if (ir->get_interface_type())
506 var->data.mode = nir_var_mem_ubo;
507 else
508 var->data.mode = nir_var_uniform;
509 break;
510
511 case ir_var_shader_storage:
512 var->data.mode = nir_var_mem_ssbo;
513 break;
514
515 case ir_var_system_value:
516 var->data.mode = nir_var_system_value;
517 break;
518
519 default:
520 unreachable("not reached");
521 }
522
523 unsigned image_access = 0;
524 if (ir->data.memory_read_only)
525 image_access |= ACCESS_NON_WRITEABLE;
526 if (ir->data.memory_write_only)
527 image_access |= ACCESS_NON_READABLE;
528 if (ir->data.memory_coherent)
529 image_access |= ACCESS_COHERENT;
530 if (ir->data.memory_volatile)
531 image_access |= ACCESS_VOLATILE;
532 if (ir->data.memory_restrict)
533 image_access |= ACCESS_RESTRICT;
534
535 /* For UBO and SSBO variables, we need explicit types */
536 if (var->data.mode & (nir_var_mem_ubo | nir_var_mem_ssbo)) {
537 const glsl_type *explicit_ifc_type =
538 ir->get_interface_type()->get_explicit_interface_type(supports_std430);
539
540 if (ir->type->without_array()->is_interface()) {
541 /* If the type contains the interface, wrap the explicit type in the
542 * right number of arrays.
543 */
544 var->type = wrap_type_in_array(explicit_ifc_type, ir->type);
545 } else {
546 /* Otherwise, this variable is one entry in the interface */
547 UNUSED bool found = false;
548 for (unsigned i = 0; i < explicit_ifc_type->length; i++) {
549 const glsl_struct_field *field =
550 &explicit_ifc_type->fields.structure[i];
551 if (strcmp(ir->name, field->name) != 0)
552 continue;
553
554 var->type = field->type;
555 if (field->memory_read_only)
556 image_access |= ACCESS_NON_WRITEABLE;
557 if (field->memory_write_only)
558 image_access |= ACCESS_NON_READABLE;
559 if (field->memory_coherent)
560 image_access |= ACCESS_COHERENT;
561 if (field->memory_volatile)
562 image_access |= ACCESS_VOLATILE;
563 if (field->memory_restrict)
564 image_access |= ACCESS_RESTRICT;
565
566 found = true;
567 break;
568 }
569 assert(found);
570 }
571 }
572
573 var->data.interpolation = ir->data.interpolation;
574 var->data.location_frac = ir->data.location_frac;
575
576 switch (ir->data.depth_layout) {
577 case ir_depth_layout_none:
578 var->data.depth_layout = nir_depth_layout_none;
579 break;
580 case ir_depth_layout_any:
581 var->data.depth_layout = nir_depth_layout_any;
582 break;
583 case ir_depth_layout_greater:
584 var->data.depth_layout = nir_depth_layout_greater;
585 break;
586 case ir_depth_layout_less:
587 var->data.depth_layout = nir_depth_layout_less;
588 break;
589 case ir_depth_layout_unchanged:
590 var->data.depth_layout = nir_depth_layout_unchanged;
591 break;
592 default:
593 unreachable("not reached");
594 }
595
596 var->data.index = ir->data.index;
597 var->data.descriptor_set = 0;
598 var->data.binding = ir->data.binding;
599 var->data.explicit_binding = ir->data.explicit_binding;
600 var->data.bindless = ir->data.bindless;
601 var->data.offset = ir->data.offset;
602
603 var->data.image.access = (gl_access_qualifier)image_access;
604 var->data.image.format = ir->data.image_format;
605
606 var->data.fb_fetch_output = ir->data.fb_fetch_output;
607 var->data.explicit_xfb_buffer = ir->data.explicit_xfb_buffer;
608 var->data.explicit_xfb_stride = ir->data.explicit_xfb_stride;
609 var->data.xfb_buffer = ir->data.xfb_buffer;
610 var->data.xfb_stride = ir->data.xfb_stride;
611
612 var->num_state_slots = ir->get_num_state_slots();
613 if (var->num_state_slots > 0) {
614 var->state_slots = rzalloc_array(var, nir_state_slot,
615 var->num_state_slots);
616
617 ir_state_slot *state_slots = ir->get_state_slots();
618 for (unsigned i = 0; i < var->num_state_slots; i++) {
619 for (unsigned j = 0; j < 5; j++)
620 var->state_slots[i].tokens[j] = state_slots[i].tokens[j];
621 var->state_slots[i].swizzle = state_slots[i].swizzle;
622 }
623 } else {
624 var->state_slots = NULL;
625 }
626
627 var->constant_initializer = constant_copy(ir->constant_initializer, var);
628
629 var->interface_type = ir->get_interface_type();
630
631 if (var->data.mode == nir_var_function_temp)
632 nir_function_impl_add_variable(impl, var);
633 else
634 nir_shader_add_variable(shader, var);
635
636 _mesa_hash_table_insert(var_table, ir, var);
637 }
638
639 ir_visitor_status
640 nir_function_visitor::visit_enter(ir_function *ir)
641 {
642 foreach_in_list(ir_function_signature, sig, &ir->signatures) {
643 visitor->create_function(sig);
644 }
645 return visit_continue_with_parent;
646 }
647
648 void
649 nir_visitor::create_function(ir_function_signature *ir)
650 {
651 if (ir->is_intrinsic())
652 return;
653
654 nir_function *func = nir_function_create(shader, ir->function_name());
655 if (strcmp(ir->function_name(), "main") == 0)
656 func->is_entrypoint = true;
657
658 func->num_params = ir->parameters.length() +
659 (ir->return_type != glsl_type::void_type);
660 func->params = ralloc_array(shader, nir_parameter, func->num_params);
661
662 unsigned np = 0;
663
664 if (ir->return_type != glsl_type::void_type) {
665 /* The return value is a variable deref (basically an out parameter) */
666 func->params[np].num_components = 1;
667 func->params[np].bit_size = 32;
668 np++;
669 }
670
671 foreach_in_list(ir_variable, param, &ir->parameters) {
672 /* FINISHME: pass arrays, structs, etc by reference? */
673 assert(param->type->is_vector() || param->type->is_scalar());
674
675 if (param->data.mode == ir_var_function_in) {
676 func->params[np].num_components = param->type->vector_elements;
677 func->params[np].bit_size = glsl_get_bit_size(param->type);
678 } else {
679 func->params[np].num_components = 1;
680 func->params[np].bit_size = 32;
681 }
682 np++;
683 }
684 assert(np == func->num_params);
685
686 _mesa_hash_table_insert(this->overload_table, ir, func);
687 }
688
689 void
690 nir_visitor::visit(ir_function *ir)
691 {
692 foreach_in_list(ir_function_signature, sig, &ir->signatures)
693 sig->accept(this);
694 }
695
696 void
697 nir_visitor::visit(ir_function_signature *ir)
698 {
699 if (ir->is_intrinsic())
700 return;
701
702 this->sig = ir;
703
704 struct hash_entry *entry =
705 _mesa_hash_table_search(this->overload_table, ir);
706
707 assert(entry);
708 nir_function *func = (nir_function *) entry->data;
709
710 if (ir->is_defined) {
711 nir_function_impl *impl = nir_function_impl_create(func);
712 this->impl = impl;
713
714 this->is_global = false;
715
716 nir_builder_init(&b, impl);
717 b.cursor = nir_after_cf_list(&impl->body);
718
719 unsigned i = (ir->return_type != glsl_type::void_type) ? 1 : 0;
720
721 foreach_in_list(ir_variable, param, &ir->parameters) {
722 nir_variable *var =
723 nir_local_variable_create(impl, param->type, param->name);
724
725 if (param->data.mode == ir_var_function_in) {
726 nir_store_var(&b, var, nir_load_param(&b, i), ~0);
727 }
728
729 _mesa_hash_table_insert(var_table, param, var);
730 i++;
731 }
732
733 visit_exec_list(&ir->body, this);
734
735 this->is_global = true;
736 } else {
737 func->impl = NULL;
738 }
739 }
740
741 void
742 nir_visitor::visit(ir_loop *ir)
743 {
744 nir_push_loop(&b);
745 visit_exec_list(&ir->body_instructions, this);
746 nir_pop_loop(&b, NULL);
747 }
748
749 void
750 nir_visitor::visit(ir_if *ir)
751 {
752 nir_push_if(&b, evaluate_rvalue(ir->condition));
753 visit_exec_list(&ir->then_instructions, this);
754 nir_push_else(&b, NULL);
755 visit_exec_list(&ir->else_instructions, this);
756 nir_pop_if(&b, NULL);
757 }
758
759 void
760 nir_visitor::visit(ir_discard *ir)
761 {
762 /*
763 * discards aren't treated as control flow, because before we lower them
764 * they can appear anywhere in the shader and the stuff after them may still
765 * be executed (yay, crazy GLSL rules!). However, after lowering, all the
766 * discards will be immediately followed by a return.
767 */
768
769 nir_intrinsic_instr *discard;
770 if (ir->condition) {
771 discard = nir_intrinsic_instr_create(this->shader,
772 nir_intrinsic_discard_if);
773 discard->src[0] =
774 nir_src_for_ssa(evaluate_rvalue(ir->condition));
775 } else {
776 discard = nir_intrinsic_instr_create(this->shader, nir_intrinsic_discard);
777 }
778
779 nir_builder_instr_insert(&b, &discard->instr);
780 }
781
782 void
783 nir_visitor::visit(ir_demote *ir)
784 {
785 nir_intrinsic_instr *demote =
786 nir_intrinsic_instr_create(this->shader, nir_intrinsic_demote);
787
788 nir_builder_instr_insert(&b, &demote->instr);
789 }
790
791 void
792 nir_visitor::visit(ir_emit_vertex *ir)
793 {
794 nir_intrinsic_instr *instr =
795 nir_intrinsic_instr_create(this->shader, nir_intrinsic_emit_vertex);
796 nir_intrinsic_set_stream_id(instr, ir->stream_id());
797 nir_builder_instr_insert(&b, &instr->instr);
798 }
799
800 void
801 nir_visitor::visit(ir_end_primitive *ir)
802 {
803 nir_intrinsic_instr *instr =
804 nir_intrinsic_instr_create(this->shader, nir_intrinsic_end_primitive);
805 nir_intrinsic_set_stream_id(instr, ir->stream_id());
806 nir_builder_instr_insert(&b, &instr->instr);
807 }
808
809 void
810 nir_visitor::visit(ir_loop_jump *ir)
811 {
812 nir_jump_type type;
813 switch (ir->mode) {
814 case ir_loop_jump::jump_break:
815 type = nir_jump_break;
816 break;
817 case ir_loop_jump::jump_continue:
818 type = nir_jump_continue;
819 break;
820 default:
821 unreachable("not reached");
822 }
823
824 nir_jump_instr *instr = nir_jump_instr_create(this->shader, type);
825 nir_builder_instr_insert(&b, &instr->instr);
826 }
827
828 void
829 nir_visitor::visit(ir_return *ir)
830 {
831 if (ir->value != NULL) {
832 nir_deref_instr *ret_deref =
833 nir_build_deref_cast(&b, nir_load_param(&b, 0),
834 nir_var_function_temp, ir->value->type, 0);
835
836 nir_ssa_def *val = evaluate_rvalue(ir->value);
837 nir_store_deref(&b, ret_deref, val, ~0);
838 }
839
840 nir_jump_instr *instr = nir_jump_instr_create(this->shader, nir_jump_return);
841 nir_builder_instr_insert(&b, &instr->instr);
842 }
843
844 static void
845 intrinsic_set_std430_align(nir_intrinsic_instr *intrin, const glsl_type *type)
846 {
847 unsigned bit_size = type->is_boolean() ? 32 : glsl_get_bit_size(type);
848 unsigned pow2_components = util_next_power_of_two(type->vector_elements);
849 nir_intrinsic_set_align(intrin, (bit_size / 8) * pow2_components, 0);
850 }
851
852 /* Accumulate any qualifiers along the deref chain to get the actual
853 * load/store qualifier.
854 */
855
856 static enum gl_access_qualifier
857 deref_get_qualifier(nir_deref_instr *deref)
858 {
859 nir_deref_path path;
860 nir_deref_path_init(&path, deref, NULL);
861
862 unsigned qualifiers = path.path[0]->var->data.image.access;
863
864 const glsl_type *parent_type = path.path[0]->type;
865 for (nir_deref_instr **cur_ptr = &path.path[1]; *cur_ptr; cur_ptr++) {
866 nir_deref_instr *cur = *cur_ptr;
867
868 if (parent_type->is_interface()) {
869 const struct glsl_struct_field *field =
870 &parent_type->fields.structure[cur->strct.index];
871 if (field->memory_read_only)
872 qualifiers |= ACCESS_NON_WRITEABLE;
873 if (field->memory_write_only)
874 qualifiers |= ACCESS_NON_READABLE;
875 if (field->memory_coherent)
876 qualifiers |= ACCESS_COHERENT;
877 if (field->memory_volatile)
878 qualifiers |= ACCESS_VOLATILE;
879 if (field->memory_restrict)
880 qualifiers |= ACCESS_RESTRICT;
881 }
882
883 parent_type = cur->type;
884 }
885
886 nir_deref_path_finish(&path);
887
888 return (gl_access_qualifier) qualifiers;
889 }
890
891 void
892 nir_visitor::visit(ir_call *ir)
893 {
894 if (ir->callee->is_intrinsic()) {
895 nir_intrinsic_op op;
896
897 switch (ir->callee->intrinsic_id) {
898 case ir_intrinsic_generic_atomic_add:
899 op = ir->return_deref->type->is_integer_32_64()
900 ? nir_intrinsic_deref_atomic_add : nir_intrinsic_deref_atomic_fadd;
901 break;
902 case ir_intrinsic_generic_atomic_and:
903 op = nir_intrinsic_deref_atomic_and;
904 break;
905 case ir_intrinsic_generic_atomic_or:
906 op = nir_intrinsic_deref_atomic_or;
907 break;
908 case ir_intrinsic_generic_atomic_xor:
909 op = nir_intrinsic_deref_atomic_xor;
910 break;
911 case ir_intrinsic_generic_atomic_min:
912 assert(ir->return_deref);
913 if (ir->return_deref->type == glsl_type::int_type)
914 op = nir_intrinsic_deref_atomic_imin;
915 else if (ir->return_deref->type == glsl_type::uint_type)
916 op = nir_intrinsic_deref_atomic_umin;
917 else if (ir->return_deref->type == glsl_type::float_type)
918 op = nir_intrinsic_deref_atomic_fmin;
919 else
920 unreachable("Invalid type");
921 break;
922 case ir_intrinsic_generic_atomic_max:
923 assert(ir->return_deref);
924 if (ir->return_deref->type == glsl_type::int_type)
925 op = nir_intrinsic_deref_atomic_imax;
926 else if (ir->return_deref->type == glsl_type::uint_type)
927 op = nir_intrinsic_deref_atomic_umax;
928 else if (ir->return_deref->type == glsl_type::float_type)
929 op = nir_intrinsic_deref_atomic_fmax;
930 else
931 unreachable("Invalid type");
932 break;
933 case ir_intrinsic_generic_atomic_exchange:
934 op = nir_intrinsic_deref_atomic_exchange;
935 break;
936 case ir_intrinsic_generic_atomic_comp_swap:
937 op = ir->return_deref->type->is_integer_32_64()
938 ? nir_intrinsic_deref_atomic_comp_swap
939 : nir_intrinsic_deref_atomic_fcomp_swap;
940 break;
941 case ir_intrinsic_atomic_counter_read:
942 op = nir_intrinsic_atomic_counter_read_deref;
943 break;
944 case ir_intrinsic_atomic_counter_increment:
945 op = nir_intrinsic_atomic_counter_inc_deref;
946 break;
947 case ir_intrinsic_atomic_counter_predecrement:
948 op = nir_intrinsic_atomic_counter_pre_dec_deref;
949 break;
950 case ir_intrinsic_atomic_counter_add:
951 op = nir_intrinsic_atomic_counter_add_deref;
952 break;
953 case ir_intrinsic_atomic_counter_and:
954 op = nir_intrinsic_atomic_counter_and_deref;
955 break;
956 case ir_intrinsic_atomic_counter_or:
957 op = nir_intrinsic_atomic_counter_or_deref;
958 break;
959 case ir_intrinsic_atomic_counter_xor:
960 op = nir_intrinsic_atomic_counter_xor_deref;
961 break;
962 case ir_intrinsic_atomic_counter_min:
963 op = nir_intrinsic_atomic_counter_min_deref;
964 break;
965 case ir_intrinsic_atomic_counter_max:
966 op = nir_intrinsic_atomic_counter_max_deref;
967 break;
968 case ir_intrinsic_atomic_counter_exchange:
969 op = nir_intrinsic_atomic_counter_exchange_deref;
970 break;
971 case ir_intrinsic_atomic_counter_comp_swap:
972 op = nir_intrinsic_atomic_counter_comp_swap_deref;
973 break;
974 case ir_intrinsic_image_load:
975 op = nir_intrinsic_image_deref_load;
976 break;
977 case ir_intrinsic_image_store:
978 op = nir_intrinsic_image_deref_store;
979 break;
980 case ir_intrinsic_image_atomic_add:
981 op = ir->return_deref->type->is_integer_32_64()
982 ? nir_intrinsic_image_deref_atomic_add
983 : nir_intrinsic_image_deref_atomic_fadd;
984 break;
985 case ir_intrinsic_image_atomic_min:
986 if (ir->return_deref->type == glsl_type::int_type)
987 op = nir_intrinsic_image_deref_atomic_imin;
988 else if (ir->return_deref->type == glsl_type::uint_type)
989 op = nir_intrinsic_image_deref_atomic_umin;
990 else
991 unreachable("Invalid type");
992 break;
993 case ir_intrinsic_image_atomic_max:
994 if (ir->return_deref->type == glsl_type::int_type)
995 op = nir_intrinsic_image_deref_atomic_imax;
996 else if (ir->return_deref->type == glsl_type::uint_type)
997 op = nir_intrinsic_image_deref_atomic_umax;
998 else
999 unreachable("Invalid type");
1000 break;
1001 case ir_intrinsic_image_atomic_and:
1002 op = nir_intrinsic_image_deref_atomic_and;
1003 break;
1004 case ir_intrinsic_image_atomic_or:
1005 op = nir_intrinsic_image_deref_atomic_or;
1006 break;
1007 case ir_intrinsic_image_atomic_xor:
1008 op = nir_intrinsic_image_deref_atomic_xor;
1009 break;
1010 case ir_intrinsic_image_atomic_exchange:
1011 op = nir_intrinsic_image_deref_atomic_exchange;
1012 break;
1013 case ir_intrinsic_image_atomic_comp_swap:
1014 op = nir_intrinsic_image_deref_atomic_comp_swap;
1015 break;
1016 case ir_intrinsic_image_atomic_inc_wrap:
1017 op = nir_intrinsic_image_deref_atomic_inc_wrap;
1018 break;
1019 case ir_intrinsic_image_atomic_dec_wrap:
1020 op = nir_intrinsic_image_deref_atomic_dec_wrap;
1021 break;
1022 case ir_intrinsic_memory_barrier:
1023 op = nir_intrinsic_memory_barrier;
1024 break;
1025 case ir_intrinsic_image_size:
1026 op = nir_intrinsic_image_deref_size;
1027 break;
1028 case ir_intrinsic_image_samples:
1029 op = nir_intrinsic_image_deref_samples;
1030 break;
1031 case ir_intrinsic_ssbo_store:
1032 op = nir_intrinsic_store_ssbo;
1033 break;
1034 case ir_intrinsic_ssbo_load:
1035 op = nir_intrinsic_load_ssbo;
1036 break;
1037 case ir_intrinsic_ssbo_atomic_add:
1038 op = ir->return_deref->type->is_integer_32_64()
1039 ? nir_intrinsic_ssbo_atomic_add : nir_intrinsic_ssbo_atomic_fadd;
1040 break;
1041 case ir_intrinsic_ssbo_atomic_and:
1042 op = nir_intrinsic_ssbo_atomic_and;
1043 break;
1044 case ir_intrinsic_ssbo_atomic_or:
1045 op = nir_intrinsic_ssbo_atomic_or;
1046 break;
1047 case ir_intrinsic_ssbo_atomic_xor:
1048 op = nir_intrinsic_ssbo_atomic_xor;
1049 break;
1050 case ir_intrinsic_ssbo_atomic_min:
1051 assert(ir->return_deref);
1052 if (ir->return_deref->type == glsl_type::int_type)
1053 op = nir_intrinsic_ssbo_atomic_imin;
1054 else if (ir->return_deref->type == glsl_type::uint_type)
1055 op = nir_intrinsic_ssbo_atomic_umin;
1056 else if (ir->return_deref->type == glsl_type::float_type)
1057 op = nir_intrinsic_ssbo_atomic_fmin;
1058 else
1059 unreachable("Invalid type");
1060 break;
1061 case ir_intrinsic_ssbo_atomic_max:
1062 assert(ir->return_deref);
1063 if (ir->return_deref->type == glsl_type::int_type)
1064 op = nir_intrinsic_ssbo_atomic_imax;
1065 else if (ir->return_deref->type == glsl_type::uint_type)
1066 op = nir_intrinsic_ssbo_atomic_umax;
1067 else if (ir->return_deref->type == glsl_type::float_type)
1068 op = nir_intrinsic_ssbo_atomic_fmax;
1069 else
1070 unreachable("Invalid type");
1071 break;
1072 case ir_intrinsic_ssbo_atomic_exchange:
1073 op = nir_intrinsic_ssbo_atomic_exchange;
1074 break;
1075 case ir_intrinsic_ssbo_atomic_comp_swap:
1076 op = ir->return_deref->type->is_integer_32_64()
1077 ? nir_intrinsic_ssbo_atomic_comp_swap
1078 : nir_intrinsic_ssbo_atomic_fcomp_swap;
1079 break;
1080 case ir_intrinsic_shader_clock:
1081 op = nir_intrinsic_shader_clock;
1082 break;
1083 case ir_intrinsic_begin_invocation_interlock:
1084 op = nir_intrinsic_begin_invocation_interlock;
1085 break;
1086 case ir_intrinsic_end_invocation_interlock:
1087 op = nir_intrinsic_end_invocation_interlock;
1088 break;
1089 case ir_intrinsic_group_memory_barrier:
1090 op = nir_intrinsic_group_memory_barrier;
1091 break;
1092 case ir_intrinsic_memory_barrier_atomic_counter:
1093 op = nir_intrinsic_memory_barrier_atomic_counter;
1094 break;
1095 case ir_intrinsic_memory_barrier_buffer:
1096 op = nir_intrinsic_memory_barrier_buffer;
1097 break;
1098 case ir_intrinsic_memory_barrier_image:
1099 op = nir_intrinsic_memory_barrier_image;
1100 break;
1101 case ir_intrinsic_memory_barrier_shared:
1102 op = nir_intrinsic_memory_barrier_shared;
1103 break;
1104 case ir_intrinsic_shared_load:
1105 op = nir_intrinsic_load_shared;
1106 break;
1107 case ir_intrinsic_shared_store:
1108 op = nir_intrinsic_store_shared;
1109 break;
1110 case ir_intrinsic_shared_atomic_add:
1111 op = ir->return_deref->type->is_integer_32_64()
1112 ? nir_intrinsic_shared_atomic_add
1113 : nir_intrinsic_shared_atomic_fadd;
1114 break;
1115 case ir_intrinsic_shared_atomic_and:
1116 op = nir_intrinsic_shared_atomic_and;
1117 break;
1118 case ir_intrinsic_shared_atomic_or:
1119 op = nir_intrinsic_shared_atomic_or;
1120 break;
1121 case ir_intrinsic_shared_atomic_xor:
1122 op = nir_intrinsic_shared_atomic_xor;
1123 break;
1124 case ir_intrinsic_shared_atomic_min:
1125 assert(ir->return_deref);
1126 if (ir->return_deref->type == glsl_type::int_type)
1127 op = nir_intrinsic_shared_atomic_imin;
1128 else if (ir->return_deref->type == glsl_type::uint_type)
1129 op = nir_intrinsic_shared_atomic_umin;
1130 else if (ir->return_deref->type == glsl_type::float_type)
1131 op = nir_intrinsic_shared_atomic_fmin;
1132 else
1133 unreachable("Invalid type");
1134 break;
1135 case ir_intrinsic_shared_atomic_max:
1136 assert(ir->return_deref);
1137 if (ir->return_deref->type == glsl_type::int_type)
1138 op = nir_intrinsic_shared_atomic_imax;
1139 else if (ir->return_deref->type == glsl_type::uint_type)
1140 op = nir_intrinsic_shared_atomic_umax;
1141 else if (ir->return_deref->type == glsl_type::float_type)
1142 op = nir_intrinsic_shared_atomic_fmax;
1143 else
1144 unreachable("Invalid type");
1145 break;
1146 case ir_intrinsic_shared_atomic_exchange:
1147 op = nir_intrinsic_shared_atomic_exchange;
1148 break;
1149 case ir_intrinsic_shared_atomic_comp_swap:
1150 op = ir->return_deref->type->is_integer_32_64()
1151 ? nir_intrinsic_shared_atomic_comp_swap
1152 : nir_intrinsic_shared_atomic_fcomp_swap;
1153 break;
1154 case ir_intrinsic_vote_any:
1155 op = nir_intrinsic_vote_any;
1156 break;
1157 case ir_intrinsic_vote_all:
1158 op = nir_intrinsic_vote_all;
1159 break;
1160 case ir_intrinsic_vote_eq:
1161 op = nir_intrinsic_vote_ieq;
1162 break;
1163 case ir_intrinsic_ballot:
1164 op = nir_intrinsic_ballot;
1165 break;
1166 case ir_intrinsic_read_invocation:
1167 op = nir_intrinsic_read_invocation;
1168 break;
1169 case ir_intrinsic_read_first_invocation:
1170 op = nir_intrinsic_read_first_invocation;
1171 break;
1172 case ir_intrinsic_helper_invocation:
1173 op = nir_intrinsic_is_helper_invocation;
1174 break;
1175 default:
1176 unreachable("not reached");
1177 }
1178
1179 nir_intrinsic_instr *instr = nir_intrinsic_instr_create(shader, op);
1180 nir_ssa_def *ret = &instr->dest.ssa;
1181
1182 switch (op) {
1183 case nir_intrinsic_deref_atomic_add:
1184 case nir_intrinsic_deref_atomic_imin:
1185 case nir_intrinsic_deref_atomic_umin:
1186 case nir_intrinsic_deref_atomic_imax:
1187 case nir_intrinsic_deref_atomic_umax:
1188 case nir_intrinsic_deref_atomic_and:
1189 case nir_intrinsic_deref_atomic_or:
1190 case nir_intrinsic_deref_atomic_xor:
1191 case nir_intrinsic_deref_atomic_exchange:
1192 case nir_intrinsic_deref_atomic_comp_swap:
1193 case nir_intrinsic_deref_atomic_fadd:
1194 case nir_intrinsic_deref_atomic_fmin:
1195 case nir_intrinsic_deref_atomic_fmax:
1196 case nir_intrinsic_deref_atomic_fcomp_swap: {
1197 int param_count = ir->actual_parameters.length();
1198 assert(param_count == 2 || param_count == 3);
1199
1200 /* Deref */
1201 exec_node *param = ir->actual_parameters.get_head();
1202 ir_rvalue *rvalue = (ir_rvalue *) param;
1203 ir_dereference *deref = rvalue->as_dereference();
1204 ir_swizzle *swizzle = NULL;
1205 if (!deref) {
1206 /* We may have a swizzle to pick off a single vec4 component */
1207 swizzle = rvalue->as_swizzle();
1208 assert(swizzle && swizzle->type->vector_elements == 1);
1209 deref = swizzle->val->as_dereference();
1210 assert(deref);
1211 }
1212 nir_deref_instr *nir_deref = evaluate_deref(deref);
1213 if (swizzle) {
1214 nir_deref = nir_build_deref_array_imm(&b, nir_deref,
1215 swizzle->mask.x);
1216 }
1217 instr->src[0] = nir_src_for_ssa(&nir_deref->dest.ssa);
1218
1219 nir_intrinsic_set_access(instr, deref_get_qualifier(nir_deref));
1220
1221 /* data1 parameter (this is always present) */
1222 param = param->get_next();
1223 ir_instruction *inst = (ir_instruction *) param;
1224 instr->src[1] = nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1225
1226 /* data2 parameter (only with atomic_comp_swap) */
1227 if (param_count == 3) {
1228 assert(op == nir_intrinsic_deref_atomic_comp_swap ||
1229 op == nir_intrinsic_deref_atomic_fcomp_swap);
1230 param = param->get_next();
1231 inst = (ir_instruction *) param;
1232 instr->src[2] = nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1233 }
1234
1235 /* Atomic result */
1236 assert(ir->return_deref);
1237 nir_ssa_dest_init(&instr->instr, &instr->dest,
1238 ir->return_deref->type->vector_elements, 32, NULL);
1239 nir_builder_instr_insert(&b, &instr->instr);
1240 break;
1241 }
1242 case nir_intrinsic_atomic_counter_read_deref:
1243 case nir_intrinsic_atomic_counter_inc_deref:
1244 case nir_intrinsic_atomic_counter_pre_dec_deref:
1245 case nir_intrinsic_atomic_counter_add_deref:
1246 case nir_intrinsic_atomic_counter_min_deref:
1247 case nir_intrinsic_atomic_counter_max_deref:
1248 case nir_intrinsic_atomic_counter_and_deref:
1249 case nir_intrinsic_atomic_counter_or_deref:
1250 case nir_intrinsic_atomic_counter_xor_deref:
1251 case nir_intrinsic_atomic_counter_exchange_deref:
1252 case nir_intrinsic_atomic_counter_comp_swap_deref: {
1253 /* Set the counter variable dereference. */
1254 exec_node *param = ir->actual_parameters.get_head();
1255 ir_dereference *counter = (ir_dereference *)param;
1256
1257 instr->src[0] = nir_src_for_ssa(&evaluate_deref(counter)->dest.ssa);
1258 param = param->get_next();
1259
1260 /* Set the intrinsic destination. */
1261 if (ir->return_deref) {
1262 nir_ssa_dest_init(&instr->instr, &instr->dest, 1, 32, NULL);
1263 }
1264
1265 /* Set the intrinsic parameters. */
1266 if (!param->is_tail_sentinel()) {
1267 instr->src[1] =
1268 nir_src_for_ssa(evaluate_rvalue((ir_dereference *)param));
1269 param = param->get_next();
1270 }
1271
1272 if (!param->is_tail_sentinel()) {
1273 instr->src[2] =
1274 nir_src_for_ssa(evaluate_rvalue((ir_dereference *)param));
1275 param = param->get_next();
1276 }
1277
1278 nir_builder_instr_insert(&b, &instr->instr);
1279 break;
1280 }
1281 case nir_intrinsic_image_deref_load:
1282 case nir_intrinsic_image_deref_store:
1283 case nir_intrinsic_image_deref_atomic_add:
1284 case nir_intrinsic_image_deref_atomic_imin:
1285 case nir_intrinsic_image_deref_atomic_umin:
1286 case nir_intrinsic_image_deref_atomic_imax:
1287 case nir_intrinsic_image_deref_atomic_umax:
1288 case nir_intrinsic_image_deref_atomic_and:
1289 case nir_intrinsic_image_deref_atomic_or:
1290 case nir_intrinsic_image_deref_atomic_xor:
1291 case nir_intrinsic_image_deref_atomic_exchange:
1292 case nir_intrinsic_image_deref_atomic_comp_swap:
1293 case nir_intrinsic_image_deref_atomic_fadd:
1294 case nir_intrinsic_image_deref_samples:
1295 case nir_intrinsic_image_deref_size:
1296 case nir_intrinsic_image_deref_atomic_inc_wrap:
1297 case nir_intrinsic_image_deref_atomic_dec_wrap: {
1298 nir_ssa_undef_instr *instr_undef =
1299 nir_ssa_undef_instr_create(shader, 1, 32);
1300 nir_builder_instr_insert(&b, &instr_undef->instr);
1301
1302 /* Set the image variable dereference. */
1303 exec_node *param = ir->actual_parameters.get_head();
1304 ir_dereference *image = (ir_dereference *)param;
1305 nir_deref_instr *deref = evaluate_deref(image);
1306 const glsl_type *type = deref->type;
1307
1308 nir_intrinsic_set_access(instr, deref_get_qualifier(deref));
1309
1310 instr->src[0] = nir_src_for_ssa(&deref->dest.ssa);
1311 param = param->get_next();
1312
1313 /* Set the intrinsic destination. */
1314 if (ir->return_deref) {
1315 unsigned num_components = ir->return_deref->type->vector_elements;
1316 nir_ssa_dest_init(&instr->instr, &instr->dest,
1317 num_components, 32, NULL);
1318 }
1319
1320 if (op == nir_intrinsic_image_deref_size) {
1321 instr->num_components = instr->dest.ssa.num_components;
1322 } else if (op == nir_intrinsic_image_deref_load ||
1323 op == nir_intrinsic_image_deref_store) {
1324 instr->num_components = 4;
1325 }
1326
1327 if (op == nir_intrinsic_image_deref_size ||
1328 op == nir_intrinsic_image_deref_samples) {
1329 nir_builder_instr_insert(&b, &instr->instr);
1330 break;
1331 }
1332
1333 /* Set the address argument, extending the coordinate vector to four
1334 * components.
1335 */
1336 nir_ssa_def *src_addr =
1337 evaluate_rvalue((ir_dereference *)param);
1338 nir_ssa_def *srcs[4];
1339
1340 for (int i = 0; i < 4; i++) {
1341 if (i < type->coordinate_components())
1342 srcs[i] = nir_channel(&b, src_addr, i);
1343 else
1344 srcs[i] = &instr_undef->def;
1345 }
1346
1347 instr->src[1] = nir_src_for_ssa(nir_vec(&b, srcs, 4));
1348 param = param->get_next();
1349
1350 /* Set the sample argument, which is undefined for single-sample
1351 * images.
1352 */
1353 if (type->sampler_dimensionality == GLSL_SAMPLER_DIM_MS) {
1354 instr->src[2] =
1355 nir_src_for_ssa(evaluate_rvalue((ir_dereference *)param));
1356 param = param->get_next();
1357 } else {
1358 instr->src[2] = nir_src_for_ssa(&instr_undef->def);
1359 }
1360
1361 /* Set the intrinsic parameters. */
1362 if (!param->is_tail_sentinel()) {
1363 instr->src[3] =
1364 nir_src_for_ssa(evaluate_rvalue((ir_dereference *)param));
1365 param = param->get_next();
1366 }
1367
1368 if (!param->is_tail_sentinel()) {
1369 instr->src[4] =
1370 nir_src_for_ssa(evaluate_rvalue((ir_dereference *)param));
1371 param = param->get_next();
1372 }
1373 nir_builder_instr_insert(&b, &instr->instr);
1374 break;
1375 }
1376 case nir_intrinsic_memory_barrier:
1377 case nir_intrinsic_group_memory_barrier:
1378 case nir_intrinsic_memory_barrier_atomic_counter:
1379 case nir_intrinsic_memory_barrier_buffer:
1380 case nir_intrinsic_memory_barrier_image:
1381 case nir_intrinsic_memory_barrier_shared:
1382 nir_builder_instr_insert(&b, &instr->instr);
1383 break;
1384 case nir_intrinsic_shader_clock:
1385 nir_ssa_dest_init(&instr->instr, &instr->dest, 2, 32, NULL);
1386 instr->num_components = 2;
1387 nir_builder_instr_insert(&b, &instr->instr);
1388 break;
1389 case nir_intrinsic_begin_invocation_interlock:
1390 nir_builder_instr_insert(&b, &instr->instr);
1391 break;
1392 case nir_intrinsic_end_invocation_interlock:
1393 nir_builder_instr_insert(&b, &instr->instr);
1394 break;
1395 case nir_intrinsic_store_ssbo: {
1396 exec_node *param = ir->actual_parameters.get_head();
1397 ir_rvalue *block = ((ir_instruction *)param)->as_rvalue();
1398
1399 param = param->get_next();
1400 ir_rvalue *offset = ((ir_instruction *)param)->as_rvalue();
1401
1402 param = param->get_next();
1403 ir_rvalue *val = ((ir_instruction *)param)->as_rvalue();
1404
1405 param = param->get_next();
1406 ir_constant *write_mask = ((ir_instruction *)param)->as_constant();
1407 assert(write_mask);
1408
1409 nir_ssa_def *nir_val = evaluate_rvalue(val);
1410 if (val->type->is_boolean())
1411 nir_val = nir_b2i32(&b, nir_val);
1412
1413 instr->src[0] = nir_src_for_ssa(nir_val);
1414 instr->src[1] = nir_src_for_ssa(evaluate_rvalue(block));
1415 instr->src[2] = nir_src_for_ssa(evaluate_rvalue(offset));
1416 intrinsic_set_std430_align(instr, val->type);
1417 nir_intrinsic_set_write_mask(instr, write_mask->value.u[0]);
1418 instr->num_components = val->type->vector_elements;
1419
1420 nir_builder_instr_insert(&b, &instr->instr);
1421 break;
1422 }
1423 case nir_intrinsic_load_ssbo: {
1424 exec_node *param = ir->actual_parameters.get_head();
1425 ir_rvalue *block = ((ir_instruction *)param)->as_rvalue();
1426
1427 param = param->get_next();
1428 ir_rvalue *offset = ((ir_instruction *)param)->as_rvalue();
1429
1430 instr->src[0] = nir_src_for_ssa(evaluate_rvalue(block));
1431 instr->src[1] = nir_src_for_ssa(evaluate_rvalue(offset));
1432
1433 const glsl_type *type = ir->return_deref->var->type;
1434 instr->num_components = type->vector_elements;
1435 intrinsic_set_std430_align(instr, type);
1436
1437 /* Setup destination register */
1438 unsigned bit_size = type->is_boolean() ? 32 : glsl_get_bit_size(type);
1439 nir_ssa_dest_init(&instr->instr, &instr->dest,
1440 type->vector_elements, bit_size, NULL);
1441
1442 /* Insert the created nir instruction now since in the case of boolean
1443 * result we will need to emit another instruction after it
1444 */
1445 nir_builder_instr_insert(&b, &instr->instr);
1446
1447 /*
1448 * In SSBO/UBO's, a true boolean value is any non-zero value, but we
1449 * consider a true boolean to be ~0. Fix this up with a != 0
1450 * comparison.
1451 */
1452 if (type->is_boolean())
1453 ret = nir_i2b(&b, &instr->dest.ssa);
1454 break;
1455 }
1456 case nir_intrinsic_ssbo_atomic_add:
1457 case nir_intrinsic_ssbo_atomic_imin:
1458 case nir_intrinsic_ssbo_atomic_umin:
1459 case nir_intrinsic_ssbo_atomic_imax:
1460 case nir_intrinsic_ssbo_atomic_umax:
1461 case nir_intrinsic_ssbo_atomic_and:
1462 case nir_intrinsic_ssbo_atomic_or:
1463 case nir_intrinsic_ssbo_atomic_xor:
1464 case nir_intrinsic_ssbo_atomic_exchange:
1465 case nir_intrinsic_ssbo_atomic_comp_swap:
1466 case nir_intrinsic_ssbo_atomic_fadd:
1467 case nir_intrinsic_ssbo_atomic_fmin:
1468 case nir_intrinsic_ssbo_atomic_fmax:
1469 case nir_intrinsic_ssbo_atomic_fcomp_swap: {
1470 int param_count = ir->actual_parameters.length();
1471 assert(param_count == 3 || param_count == 4);
1472
1473 /* Block index */
1474 exec_node *param = ir->actual_parameters.get_head();
1475 ir_instruction *inst = (ir_instruction *) param;
1476 instr->src[0] = nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1477
1478 /* Offset */
1479 param = param->get_next();
1480 inst = (ir_instruction *) param;
1481 instr->src[1] = nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1482
1483 /* data1 parameter (this is always present) */
1484 param = param->get_next();
1485 inst = (ir_instruction *) param;
1486 instr->src[2] = nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1487
1488 /* data2 parameter (only with atomic_comp_swap) */
1489 if (param_count == 4) {
1490 assert(op == nir_intrinsic_ssbo_atomic_comp_swap ||
1491 op == nir_intrinsic_ssbo_atomic_fcomp_swap);
1492 param = param->get_next();
1493 inst = (ir_instruction *) param;
1494 instr->src[3] = nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1495 }
1496
1497 /* Atomic result */
1498 assert(ir->return_deref);
1499 nir_ssa_dest_init(&instr->instr, &instr->dest,
1500 ir->return_deref->type->vector_elements, 32, NULL);
1501 nir_builder_instr_insert(&b, &instr->instr);
1502 break;
1503 }
1504 case nir_intrinsic_load_shared: {
1505 exec_node *param = ir->actual_parameters.get_head();
1506 ir_rvalue *offset = ((ir_instruction *)param)->as_rvalue();
1507
1508 nir_intrinsic_set_base(instr, 0);
1509 instr->src[0] = nir_src_for_ssa(evaluate_rvalue(offset));
1510
1511 const glsl_type *type = ir->return_deref->var->type;
1512 instr->num_components = type->vector_elements;
1513 intrinsic_set_std430_align(instr, type);
1514
1515 /* Setup destination register */
1516 unsigned bit_size = type->is_boolean() ? 32 : glsl_get_bit_size(type);
1517 nir_ssa_dest_init(&instr->instr, &instr->dest,
1518 type->vector_elements, bit_size, NULL);
1519
1520 nir_builder_instr_insert(&b, &instr->instr);
1521
1522 /* The value in shared memory is a 32-bit value */
1523 if (type->is_boolean())
1524 ret = nir_i2b(&b, &instr->dest.ssa);
1525 break;
1526 }
1527 case nir_intrinsic_store_shared: {
1528 exec_node *param = ir->actual_parameters.get_head();
1529 ir_rvalue *offset = ((ir_instruction *)param)->as_rvalue();
1530
1531 param = param->get_next();
1532 ir_rvalue *val = ((ir_instruction *)param)->as_rvalue();
1533
1534 param = param->get_next();
1535 ir_constant *write_mask = ((ir_instruction *)param)->as_constant();
1536 assert(write_mask);
1537
1538 nir_intrinsic_set_base(instr, 0);
1539 instr->src[1] = nir_src_for_ssa(evaluate_rvalue(offset));
1540
1541 nir_intrinsic_set_write_mask(instr, write_mask->value.u[0]);
1542
1543 nir_ssa_def *nir_val = evaluate_rvalue(val);
1544 /* The value in shared memory is a 32-bit value */
1545 if (val->type->is_boolean())
1546 nir_val = nir_b2i32(&b, nir_val);
1547
1548 instr->src[0] = nir_src_for_ssa(nir_val);
1549 instr->num_components = val->type->vector_elements;
1550 intrinsic_set_std430_align(instr, val->type);
1551
1552 nir_builder_instr_insert(&b, &instr->instr);
1553 break;
1554 }
1555 case nir_intrinsic_shared_atomic_add:
1556 case nir_intrinsic_shared_atomic_imin:
1557 case nir_intrinsic_shared_atomic_umin:
1558 case nir_intrinsic_shared_atomic_imax:
1559 case nir_intrinsic_shared_atomic_umax:
1560 case nir_intrinsic_shared_atomic_and:
1561 case nir_intrinsic_shared_atomic_or:
1562 case nir_intrinsic_shared_atomic_xor:
1563 case nir_intrinsic_shared_atomic_exchange:
1564 case nir_intrinsic_shared_atomic_comp_swap:
1565 case nir_intrinsic_shared_atomic_fadd:
1566 case nir_intrinsic_shared_atomic_fmin:
1567 case nir_intrinsic_shared_atomic_fmax:
1568 case nir_intrinsic_shared_atomic_fcomp_swap: {
1569 int param_count = ir->actual_parameters.length();
1570 assert(param_count == 2 || param_count == 3);
1571
1572 /* Offset */
1573 exec_node *param = ir->actual_parameters.get_head();
1574 ir_instruction *inst = (ir_instruction *) param;
1575 instr->src[0] = nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1576
1577 /* data1 parameter (this is always present) */
1578 param = param->get_next();
1579 inst = (ir_instruction *) param;
1580 instr->src[1] = nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1581
1582 /* data2 parameter (only with atomic_comp_swap) */
1583 if (param_count == 3) {
1584 assert(op == nir_intrinsic_shared_atomic_comp_swap ||
1585 op == nir_intrinsic_shared_atomic_fcomp_swap);
1586 param = param->get_next();
1587 inst = (ir_instruction *) param;
1588 instr->src[2] =
1589 nir_src_for_ssa(evaluate_rvalue(inst->as_rvalue()));
1590 }
1591
1592 /* Atomic result */
1593 assert(ir->return_deref);
1594 unsigned bit_size = glsl_get_bit_size(ir->return_deref->type);
1595 nir_ssa_dest_init(&instr->instr, &instr->dest,
1596 ir->return_deref->type->vector_elements,
1597 bit_size, NULL);
1598 nir_builder_instr_insert(&b, &instr->instr);
1599 break;
1600 }
1601 case nir_intrinsic_vote_any:
1602 case nir_intrinsic_vote_all:
1603 case nir_intrinsic_vote_ieq: {
1604 nir_ssa_dest_init(&instr->instr, &instr->dest, 1, 1, NULL);
1605 instr->num_components = 1;
1606
1607 ir_rvalue *value = (ir_rvalue *) ir->actual_parameters.get_head();
1608 instr->src[0] = nir_src_for_ssa(evaluate_rvalue(value));
1609
1610 nir_builder_instr_insert(&b, &instr->instr);
1611 break;
1612 }
1613
1614 case nir_intrinsic_ballot: {
1615 nir_ssa_dest_init(&instr->instr, &instr->dest,
1616 ir->return_deref->type->vector_elements, 64, NULL);
1617 instr->num_components = ir->return_deref->type->vector_elements;
1618
1619 ir_rvalue *value = (ir_rvalue *) ir->actual_parameters.get_head();
1620 instr->src[0] = nir_src_for_ssa(evaluate_rvalue(value));
1621
1622 nir_builder_instr_insert(&b, &instr->instr);
1623 break;
1624 }
1625 case nir_intrinsic_read_invocation: {
1626 nir_ssa_dest_init(&instr->instr, &instr->dest,
1627 ir->return_deref->type->vector_elements, 32, NULL);
1628 instr->num_components = ir->return_deref->type->vector_elements;
1629
1630 ir_rvalue *value = (ir_rvalue *) ir->actual_parameters.get_head();
1631 instr->src[0] = nir_src_for_ssa(evaluate_rvalue(value));
1632
1633 ir_rvalue *invocation = (ir_rvalue *) ir->actual_parameters.get_head()->next;
1634 instr->src[1] = nir_src_for_ssa(evaluate_rvalue(invocation));
1635
1636 nir_builder_instr_insert(&b, &instr->instr);
1637 break;
1638 }
1639 case nir_intrinsic_read_first_invocation: {
1640 nir_ssa_dest_init(&instr->instr, &instr->dest,
1641 ir->return_deref->type->vector_elements, 32, NULL);
1642 instr->num_components = ir->return_deref->type->vector_elements;
1643
1644 ir_rvalue *value = (ir_rvalue *) ir->actual_parameters.get_head();
1645 instr->src[0] = nir_src_for_ssa(evaluate_rvalue(value));
1646
1647 nir_builder_instr_insert(&b, &instr->instr);
1648 break;
1649 }
1650 case nir_intrinsic_is_helper_invocation: {
1651 nir_ssa_dest_init(&instr->instr, &instr->dest, 1, 1, NULL);
1652 instr->num_components = 1;
1653 nir_builder_instr_insert(&b, &instr->instr);
1654 break;
1655 }
1656 default:
1657 unreachable("not reached");
1658 }
1659
1660 if (ir->return_deref)
1661 nir_store_deref(&b, evaluate_deref(ir->return_deref), ret, ~0);
1662
1663 return;
1664 }
1665
1666 struct hash_entry *entry =
1667 _mesa_hash_table_search(this->overload_table, ir->callee);
1668 assert(entry);
1669 nir_function *callee = (nir_function *) entry->data;
1670
1671 nir_call_instr *call = nir_call_instr_create(this->shader, callee);
1672
1673 unsigned i = 0;
1674 nir_deref_instr *ret_deref = NULL;
1675 if (ir->return_deref) {
1676 nir_variable *ret_tmp =
1677 nir_local_variable_create(this->impl, ir->return_deref->type,
1678 "return_tmp");
1679 ret_deref = nir_build_deref_var(&b, ret_tmp);
1680 call->params[i++] = nir_src_for_ssa(&ret_deref->dest.ssa);
1681 }
1682
1683 foreach_two_lists(formal_node, &ir->callee->parameters,
1684 actual_node, &ir->actual_parameters) {
1685 ir_rvalue *param_rvalue = (ir_rvalue *) actual_node;
1686 ir_variable *sig_param = (ir_variable *) formal_node;
1687
1688 if (sig_param->data.mode == ir_var_function_out) {
1689 nir_deref_instr *out_deref = evaluate_deref(param_rvalue);
1690 call->params[i] = nir_src_for_ssa(&out_deref->dest.ssa);
1691 } else if (sig_param->data.mode == ir_var_function_in) {
1692 nir_ssa_def *val = evaluate_rvalue(param_rvalue);
1693 nir_src src = nir_src_for_ssa(val);
1694
1695 nir_src_copy(&call->params[i], &src, call);
1696 } else if (sig_param->data.mode == ir_var_function_inout) {
1697 unreachable("unimplemented: inout parameters");
1698 }
1699
1700 i++;
1701 }
1702
1703 nir_builder_instr_insert(&b, &call->instr);
1704
1705 if (ir->return_deref)
1706 nir_store_deref(&b, evaluate_deref(ir->return_deref), nir_load_deref(&b, ret_deref), ~0);
1707 }
1708
1709 void
1710 nir_visitor::visit(ir_assignment *ir)
1711 {
1712 unsigned num_components = ir->lhs->type->vector_elements;
1713
1714 b.exact = ir->lhs->variable_referenced()->data.invariant ||
1715 ir->lhs->variable_referenced()->data.precise;
1716
1717 if ((ir->rhs->as_dereference() || ir->rhs->as_constant()) &&
1718 (ir->write_mask == (1 << num_components) - 1 || ir->write_mask == 0)) {
1719 nir_deref_instr *lhs = evaluate_deref(ir->lhs);
1720 nir_deref_instr *rhs = evaluate_deref(ir->rhs);
1721 enum gl_access_qualifier lhs_qualifiers = deref_get_qualifier(lhs);
1722 enum gl_access_qualifier rhs_qualifiers = deref_get_qualifier(rhs);
1723 if (ir->condition) {
1724 nir_push_if(&b, evaluate_rvalue(ir->condition));
1725 nir_copy_deref_with_access(&b, lhs, rhs, lhs_qualifiers,
1726 rhs_qualifiers);
1727 nir_pop_if(&b, NULL);
1728 } else {
1729 nir_copy_deref_with_access(&b, lhs, rhs, lhs_qualifiers,
1730 rhs_qualifiers);
1731 }
1732 return;
1733 }
1734
1735 assert(ir->rhs->type->is_scalar() || ir->rhs->type->is_vector());
1736
1737 ir->lhs->accept(this);
1738 nir_deref_instr *lhs_deref = this->deref;
1739 nir_ssa_def *src = evaluate_rvalue(ir->rhs);
1740
1741 if (ir->write_mask != (1 << num_components) - 1 && ir->write_mask != 0) {
1742 /* GLSL IR will give us the input to the write-masked assignment in a
1743 * single packed vector. So, for example, if the writemask is xzw, then
1744 * we have to swizzle x -> x, y -> z, and z -> w and get the y component
1745 * from the load.
1746 */
1747 unsigned swiz[4];
1748 unsigned component = 0;
1749 for (unsigned i = 0; i < 4; i++) {
1750 swiz[i] = ir->write_mask & (1 << i) ? component++ : 0;
1751 }
1752 src = nir_swizzle(&b, src, swiz, num_components);
1753 }
1754
1755 enum gl_access_qualifier qualifiers = deref_get_qualifier(lhs_deref);
1756 if (ir->condition) {
1757 nir_push_if(&b, evaluate_rvalue(ir->condition));
1758 nir_store_deref_with_access(&b, lhs_deref, src, ir->write_mask,
1759 qualifiers);
1760 nir_pop_if(&b, NULL);
1761 } else {
1762 nir_store_deref_with_access(&b, lhs_deref, src, ir->write_mask,
1763 qualifiers);
1764 }
1765 }
1766
1767 /*
1768 * Given an instruction, returns a pointer to its destination or NULL if there
1769 * is no destination.
1770 *
1771 * Note that this only handles instructions we generate at this level.
1772 */
1773 static nir_dest *
1774 get_instr_dest(nir_instr *instr)
1775 {
1776 nir_alu_instr *alu_instr;
1777 nir_intrinsic_instr *intrinsic_instr;
1778 nir_tex_instr *tex_instr;
1779
1780 switch (instr->type) {
1781 case nir_instr_type_alu:
1782 alu_instr = nir_instr_as_alu(instr);
1783 return &alu_instr->dest.dest;
1784
1785 case nir_instr_type_intrinsic:
1786 intrinsic_instr = nir_instr_as_intrinsic(instr);
1787 if (nir_intrinsic_infos[intrinsic_instr->intrinsic].has_dest)
1788 return &intrinsic_instr->dest;
1789 else
1790 return NULL;
1791
1792 case nir_instr_type_tex:
1793 tex_instr = nir_instr_as_tex(instr);
1794 return &tex_instr->dest;
1795
1796 default:
1797 unreachable("not reached");
1798 }
1799
1800 return NULL;
1801 }
1802
1803 void
1804 nir_visitor::add_instr(nir_instr *instr, unsigned num_components,
1805 unsigned bit_size)
1806 {
1807 nir_dest *dest = get_instr_dest(instr);
1808
1809 if (dest)
1810 nir_ssa_dest_init(instr, dest, num_components, bit_size, NULL);
1811
1812 nir_builder_instr_insert(&b, instr);
1813
1814 if (dest) {
1815 assert(dest->is_ssa);
1816 this->result = &dest->ssa;
1817 }
1818 }
1819
1820 nir_ssa_def *
1821 nir_visitor::evaluate_rvalue(ir_rvalue* ir)
1822 {
1823 ir->accept(this);
1824 if (ir->as_dereference() || ir->as_constant()) {
1825 /*
1826 * A dereference is being used on the right hand side, which means we
1827 * must emit a variable load.
1828 */
1829
1830 enum gl_access_qualifier access = deref_get_qualifier(this->deref);
1831 this->result = nir_load_deref_with_access(&b, this->deref, access);
1832 }
1833
1834 return this->result;
1835 }
1836
1837 static bool
1838 type_is_float(glsl_base_type type)
1839 {
1840 return type == GLSL_TYPE_FLOAT || type == GLSL_TYPE_DOUBLE ||
1841 type == GLSL_TYPE_FLOAT16;
1842 }
1843
1844 static bool
1845 type_is_signed(glsl_base_type type)
1846 {
1847 return type == GLSL_TYPE_INT || type == GLSL_TYPE_INT64 ||
1848 type == GLSL_TYPE_INT16;
1849 }
1850
1851 void
1852 nir_visitor::visit(ir_expression *ir)
1853 {
1854 /* Some special cases */
1855 switch (ir->operation) {
1856 case ir_binop_ubo_load: {
1857 nir_intrinsic_instr *load =
1858 nir_intrinsic_instr_create(this->shader, nir_intrinsic_load_ubo);
1859 unsigned bit_size = ir->type->is_boolean() ? 32 :
1860 glsl_get_bit_size(ir->type);
1861 load->num_components = ir->type->vector_elements;
1862 load->src[0] = nir_src_for_ssa(evaluate_rvalue(ir->operands[0]));
1863 load->src[1] = nir_src_for_ssa(evaluate_rvalue(ir->operands[1]));
1864 intrinsic_set_std430_align(load, ir->type);
1865 add_instr(&load->instr, ir->type->vector_elements, bit_size);
1866
1867 /*
1868 * In UBO's, a true boolean value is any non-zero value, but we consider
1869 * a true boolean to be ~0. Fix this up with a != 0 comparison.
1870 */
1871
1872 if (ir->type->is_boolean())
1873 this->result = nir_i2b(&b, &load->dest.ssa);
1874
1875 return;
1876 }
1877
1878 case ir_unop_interpolate_at_centroid:
1879 case ir_binop_interpolate_at_offset:
1880 case ir_binop_interpolate_at_sample: {
1881 ir_dereference *deref = ir->operands[0]->as_dereference();
1882 ir_swizzle *swizzle = NULL;
1883 if (!deref) {
1884 /* the api does not allow a swizzle here, but the varying packing code
1885 * may have pushed one into here.
1886 */
1887 swizzle = ir->operands[0]->as_swizzle();
1888 assert(swizzle);
1889 deref = swizzle->val->as_dereference();
1890 assert(deref);
1891 }
1892
1893 deref->accept(this);
1894
1895 nir_intrinsic_op op;
1896 if (this->deref->mode == nir_var_shader_in) {
1897 switch (ir->operation) {
1898 case ir_unop_interpolate_at_centroid:
1899 op = nir_intrinsic_interp_deref_at_centroid;
1900 break;
1901 case ir_binop_interpolate_at_offset:
1902 op = nir_intrinsic_interp_deref_at_offset;
1903 break;
1904 case ir_binop_interpolate_at_sample:
1905 op = nir_intrinsic_interp_deref_at_sample;
1906 break;
1907 default:
1908 unreachable("Invalid interpolation intrinsic");
1909 }
1910 } else {
1911 /* This case can happen if the vertex shader does not write the
1912 * given varying. In this case, the linker will lower it to a
1913 * global variable. Since interpolating a variable makes no
1914 * sense, we'll just turn it into a load which will probably
1915 * eventually end up as an SSA definition.
1916 */
1917 assert(this->deref->mode == nir_var_shader_temp);
1918 op = nir_intrinsic_load_deref;
1919 }
1920
1921 nir_intrinsic_instr *intrin = nir_intrinsic_instr_create(shader, op);
1922 intrin->num_components = deref->type->vector_elements;
1923 intrin->src[0] = nir_src_for_ssa(&this->deref->dest.ssa);
1924
1925 if (intrin->intrinsic == nir_intrinsic_interp_deref_at_offset ||
1926 intrin->intrinsic == nir_intrinsic_interp_deref_at_sample)
1927 intrin->src[1] = nir_src_for_ssa(evaluate_rvalue(ir->operands[1]));
1928
1929 unsigned bit_size = glsl_get_bit_size(deref->type);
1930 add_instr(&intrin->instr, deref->type->vector_elements, bit_size);
1931
1932 if (swizzle) {
1933 unsigned swiz[4] = {
1934 swizzle->mask.x, swizzle->mask.y, swizzle->mask.z, swizzle->mask.w
1935 };
1936
1937 result = nir_swizzle(&b, result, swiz,
1938 swizzle->type->vector_elements);
1939 }
1940
1941 return;
1942 }
1943
1944 case ir_unop_ssbo_unsized_array_length: {
1945 nir_intrinsic_instr *intrin =
1946 nir_intrinsic_instr_create(b.shader,
1947 nir_intrinsic_deref_buffer_array_length);
1948
1949 ir_dereference *deref = ir->operands[0]->as_dereference();
1950 intrin->src[0] = nir_src_for_ssa(&evaluate_deref(deref)->dest.ssa);
1951
1952 add_instr(&intrin->instr, 1, 32);
1953 return;
1954 }
1955
1956 default:
1957 break;
1958 }
1959
1960 nir_ssa_def *srcs[4];
1961 for (unsigned i = 0; i < ir->num_operands; i++)
1962 srcs[i] = evaluate_rvalue(ir->operands[i]);
1963
1964 glsl_base_type types[4];
1965 for (unsigned i = 0; i < ir->num_operands; i++)
1966 types[i] = ir->operands[i]->type->base_type;
1967
1968 glsl_base_type out_type = ir->type->base_type;
1969
1970 switch (ir->operation) {
1971 case ir_unop_bit_not: result = nir_inot(&b, srcs[0]); break;
1972 case ir_unop_logic_not:
1973 result = nir_inot(&b, srcs[0]);
1974 break;
1975 case ir_unop_neg:
1976 result = type_is_float(types[0]) ? nir_fneg(&b, srcs[0])
1977 : nir_ineg(&b, srcs[0]);
1978 break;
1979 case ir_unop_abs:
1980 result = type_is_float(types[0]) ? nir_fabs(&b, srcs[0])
1981 : nir_iabs(&b, srcs[0]);
1982 break;
1983 case ir_unop_saturate:
1984 assert(type_is_float(types[0]));
1985 result = nir_fsat(&b, srcs[0]);
1986 break;
1987 case ir_unop_sign:
1988 result = type_is_float(types[0]) ? nir_fsign(&b, srcs[0])
1989 : nir_isign(&b, srcs[0]);
1990 break;
1991 case ir_unop_rcp: result = nir_frcp(&b, srcs[0]); break;
1992 case ir_unop_rsq: result = nir_frsq(&b, srcs[0]); break;
1993 case ir_unop_sqrt: result = nir_fsqrt(&b, srcs[0]); break;
1994 case ir_unop_exp: unreachable("ir_unop_exp should have been lowered");
1995 case ir_unop_log: unreachable("ir_unop_log should have been lowered");
1996 case ir_unop_exp2: result = nir_fexp2(&b, srcs[0]); break;
1997 case ir_unop_log2: result = nir_flog2(&b, srcs[0]); break;
1998 case ir_unop_i2f:
1999 case ir_unop_u2f:
2000 case ir_unop_b2f:
2001 case ir_unop_f2i:
2002 case ir_unop_f2u:
2003 case ir_unop_f2b:
2004 case ir_unop_i2b:
2005 case ir_unop_b2i:
2006 case ir_unop_b2i64:
2007 case ir_unop_d2f:
2008 case ir_unop_f2d:
2009 case ir_unop_d2i:
2010 case ir_unop_d2u:
2011 case ir_unop_d2b:
2012 case ir_unop_i2d:
2013 case ir_unop_u2d:
2014 case ir_unop_i642i:
2015 case ir_unop_i642u:
2016 case ir_unop_i642f:
2017 case ir_unop_i642b:
2018 case ir_unop_i642d:
2019 case ir_unop_u642i:
2020 case ir_unop_u642u:
2021 case ir_unop_u642f:
2022 case ir_unop_u642d:
2023 case ir_unop_i2i64:
2024 case ir_unop_u2i64:
2025 case ir_unop_f2i64:
2026 case ir_unop_d2i64:
2027 case ir_unop_i2u64:
2028 case ir_unop_u2u64:
2029 case ir_unop_f2u64:
2030 case ir_unop_d2u64:
2031 case ir_unop_i2u:
2032 case ir_unop_u2i:
2033 case ir_unop_i642u64:
2034 case ir_unop_u642i64: {
2035 nir_alu_type src_type = nir_get_nir_type_for_glsl_base_type(types[0]);
2036 nir_alu_type dst_type = nir_get_nir_type_for_glsl_base_type(out_type);
2037 result = nir_build_alu(&b, nir_type_conversion_op(src_type, dst_type,
2038 nir_rounding_mode_undef),
2039 srcs[0], NULL, NULL, NULL);
2040 /* b2i and b2f don't have fixed bit-size versions so the builder will
2041 * just assume 32 and we have to fix it up here.
2042 */
2043 result->bit_size = nir_alu_type_get_type_size(dst_type);
2044 break;
2045 }
2046
2047 case ir_unop_bitcast_i2f:
2048 case ir_unop_bitcast_f2i:
2049 case ir_unop_bitcast_u2f:
2050 case ir_unop_bitcast_f2u:
2051 case ir_unop_bitcast_i642d:
2052 case ir_unop_bitcast_d2i64:
2053 case ir_unop_bitcast_u642d:
2054 case ir_unop_bitcast_d2u64:
2055 case ir_unop_subroutine_to_int:
2056 /* no-op */
2057 result = nir_mov(&b, srcs[0]);
2058 break;
2059 case ir_unop_trunc: result = nir_ftrunc(&b, srcs[0]); break;
2060 case ir_unop_ceil: result = nir_fceil(&b, srcs[0]); break;
2061 case ir_unop_floor: result = nir_ffloor(&b, srcs[0]); break;
2062 case ir_unop_fract: result = nir_ffract(&b, srcs[0]); break;
2063 case ir_unop_frexp_exp: result = nir_frexp_exp(&b, srcs[0]); break;
2064 case ir_unop_frexp_sig: result = nir_frexp_sig(&b, srcs[0]); break;
2065 case ir_unop_round_even: result = nir_fround_even(&b, srcs[0]); break;
2066 case ir_unop_sin: result = nir_fsin(&b, srcs[0]); break;
2067 case ir_unop_cos: result = nir_fcos(&b, srcs[0]); break;
2068 case ir_unop_dFdx: result = nir_fddx(&b, srcs[0]); break;
2069 case ir_unop_dFdy: result = nir_fddy(&b, srcs[0]); break;
2070 case ir_unop_dFdx_fine: result = nir_fddx_fine(&b, srcs[0]); break;
2071 case ir_unop_dFdy_fine: result = nir_fddy_fine(&b, srcs[0]); break;
2072 case ir_unop_dFdx_coarse: result = nir_fddx_coarse(&b, srcs[0]); break;
2073 case ir_unop_dFdy_coarse: result = nir_fddy_coarse(&b, srcs[0]); break;
2074 case ir_unop_pack_snorm_2x16:
2075 result = nir_pack_snorm_2x16(&b, srcs[0]);
2076 break;
2077 case ir_unop_pack_snorm_4x8:
2078 result = nir_pack_snorm_4x8(&b, srcs[0]);
2079 break;
2080 case ir_unop_pack_unorm_2x16:
2081 result = nir_pack_unorm_2x16(&b, srcs[0]);
2082 break;
2083 case ir_unop_pack_unorm_4x8:
2084 result = nir_pack_unorm_4x8(&b, srcs[0]);
2085 break;
2086 case ir_unop_pack_half_2x16:
2087 result = nir_pack_half_2x16(&b, srcs[0]);
2088 break;
2089 case ir_unop_unpack_snorm_2x16:
2090 result = nir_unpack_snorm_2x16(&b, srcs[0]);
2091 break;
2092 case ir_unop_unpack_snorm_4x8:
2093 result = nir_unpack_snorm_4x8(&b, srcs[0]);
2094 break;
2095 case ir_unop_unpack_unorm_2x16:
2096 result = nir_unpack_unorm_2x16(&b, srcs[0]);
2097 break;
2098 case ir_unop_unpack_unorm_4x8:
2099 result = nir_unpack_unorm_4x8(&b, srcs[0]);
2100 break;
2101 case ir_unop_unpack_half_2x16:
2102 result = nir_unpack_half_2x16(&b, srcs[0]);
2103 break;
2104 case ir_unop_pack_sampler_2x32:
2105 case ir_unop_pack_image_2x32:
2106 case ir_unop_pack_double_2x32:
2107 case ir_unop_pack_int_2x32:
2108 case ir_unop_pack_uint_2x32:
2109 result = nir_pack_64_2x32(&b, srcs[0]);
2110 break;
2111 case ir_unop_unpack_sampler_2x32:
2112 case ir_unop_unpack_image_2x32:
2113 case ir_unop_unpack_double_2x32:
2114 case ir_unop_unpack_int_2x32:
2115 case ir_unop_unpack_uint_2x32:
2116 result = nir_unpack_64_2x32(&b, srcs[0]);
2117 break;
2118 case ir_unop_bitfield_reverse:
2119 result = nir_bitfield_reverse(&b, srcs[0]);
2120 break;
2121 case ir_unop_bit_count:
2122 result = nir_bit_count(&b, srcs[0]);
2123 break;
2124 case ir_unop_find_msb:
2125 switch (types[0]) {
2126 case GLSL_TYPE_UINT:
2127 result = nir_ufind_msb(&b, srcs[0]);
2128 break;
2129 case GLSL_TYPE_INT:
2130 result = nir_ifind_msb(&b, srcs[0]);
2131 break;
2132 default:
2133 unreachable("Invalid type for findMSB()");
2134 }
2135 break;
2136 case ir_unop_find_lsb:
2137 result = nir_find_lsb(&b, srcs[0]);
2138 break;
2139
2140 case ir_unop_noise:
2141 switch (ir->type->vector_elements) {
2142 case 1:
2143 switch (ir->operands[0]->type->vector_elements) {
2144 case 1: result = nir_fnoise1_1(&b, srcs[0]); break;
2145 case 2: result = nir_fnoise1_2(&b, srcs[0]); break;
2146 case 3: result = nir_fnoise1_3(&b, srcs[0]); break;
2147 case 4: result = nir_fnoise1_4(&b, srcs[0]); break;
2148 default: unreachable("not reached");
2149 }
2150 break;
2151 case 2:
2152 switch (ir->operands[0]->type->vector_elements) {
2153 case 1: result = nir_fnoise2_1(&b, srcs[0]); break;
2154 case 2: result = nir_fnoise2_2(&b, srcs[0]); break;
2155 case 3: result = nir_fnoise2_3(&b, srcs[0]); break;
2156 case 4: result = nir_fnoise2_4(&b, srcs[0]); break;
2157 default: unreachable("not reached");
2158 }
2159 break;
2160 case 3:
2161 switch (ir->operands[0]->type->vector_elements) {
2162 case 1: result = nir_fnoise3_1(&b, srcs[0]); break;
2163 case 2: result = nir_fnoise3_2(&b, srcs[0]); break;
2164 case 3: result = nir_fnoise3_3(&b, srcs[0]); break;
2165 case 4: result = nir_fnoise3_4(&b, srcs[0]); break;
2166 default: unreachable("not reached");
2167 }
2168 break;
2169 case 4:
2170 switch (ir->operands[0]->type->vector_elements) {
2171 case 1: result = nir_fnoise4_1(&b, srcs[0]); break;
2172 case 2: result = nir_fnoise4_2(&b, srcs[0]); break;
2173 case 3: result = nir_fnoise4_3(&b, srcs[0]); break;
2174 case 4: result = nir_fnoise4_4(&b, srcs[0]); break;
2175 default: unreachable("not reached");
2176 }
2177 break;
2178 default:
2179 unreachable("not reached");
2180 }
2181 break;
2182 case ir_unop_get_buffer_size: {
2183 nir_intrinsic_instr *load = nir_intrinsic_instr_create(
2184 this->shader,
2185 nir_intrinsic_get_buffer_size);
2186 load->num_components = ir->type->vector_elements;
2187 load->src[0] = nir_src_for_ssa(evaluate_rvalue(ir->operands[0]));
2188 unsigned bit_size = glsl_get_bit_size(ir->type);
2189 add_instr(&load->instr, ir->type->vector_elements, bit_size);
2190 return;
2191 }
2192
2193 case ir_unop_atan:
2194 result = nir_atan(&b, srcs[0]);
2195 break;
2196
2197 case ir_binop_add:
2198 result = type_is_float(out_type) ? nir_fadd(&b, srcs[0], srcs[1])
2199 : nir_iadd(&b, srcs[0], srcs[1]);
2200 break;
2201 case ir_binop_sub:
2202 result = type_is_float(out_type) ? nir_fsub(&b, srcs[0], srcs[1])
2203 : nir_isub(&b, srcs[0], srcs[1]);
2204 break;
2205 case ir_binop_mul:
2206 if (type_is_float(out_type))
2207 result = nir_fmul(&b, srcs[0], srcs[1]);
2208 else if (out_type == GLSL_TYPE_INT64 &&
2209 (ir->operands[0]->type->base_type == GLSL_TYPE_INT ||
2210 ir->operands[1]->type->base_type == GLSL_TYPE_INT))
2211 result = nir_imul_2x32_64(&b, srcs[0], srcs[1]);
2212 else if (out_type == GLSL_TYPE_UINT64 &&
2213 (ir->operands[0]->type->base_type == GLSL_TYPE_UINT ||
2214 ir->operands[1]->type->base_type == GLSL_TYPE_UINT))
2215 result = nir_umul_2x32_64(&b, srcs[0], srcs[1]);
2216 else
2217 result = nir_imul(&b, srcs[0], srcs[1]);
2218 break;
2219 case ir_binop_div:
2220 if (type_is_float(out_type))
2221 result = nir_fdiv(&b, srcs[0], srcs[1]);
2222 else if (type_is_signed(out_type))
2223 result = nir_idiv(&b, srcs[0], srcs[1]);
2224 else
2225 result = nir_udiv(&b, srcs[0], srcs[1]);
2226 break;
2227 case ir_binop_mod:
2228 result = type_is_float(out_type) ? nir_fmod(&b, srcs[0], srcs[1])
2229 : nir_umod(&b, srcs[0], srcs[1]);
2230 break;
2231 case ir_binop_min:
2232 if (type_is_float(out_type))
2233 result = nir_fmin(&b, srcs[0], srcs[1]);
2234 else if (type_is_signed(out_type))
2235 result = nir_imin(&b, srcs[0], srcs[1]);
2236 else
2237 result = nir_umin(&b, srcs[0], srcs[1]);
2238 break;
2239 case ir_binop_max:
2240 if (type_is_float(out_type))
2241 result = nir_fmax(&b, srcs[0], srcs[1]);
2242 else if (type_is_signed(out_type))
2243 result = nir_imax(&b, srcs[0], srcs[1]);
2244 else
2245 result = nir_umax(&b, srcs[0], srcs[1]);
2246 break;
2247 case ir_binop_pow: result = nir_fpow(&b, srcs[0], srcs[1]); break;
2248 case ir_binop_bit_and: result = nir_iand(&b, srcs[0], srcs[1]); break;
2249 case ir_binop_bit_or: result = nir_ior(&b, srcs[0], srcs[1]); break;
2250 case ir_binop_bit_xor: result = nir_ixor(&b, srcs[0], srcs[1]); break;
2251 case ir_binop_logic_and:
2252 result = nir_iand(&b, srcs[0], srcs[1]);
2253 break;
2254 case ir_binop_logic_or:
2255 result = nir_ior(&b, srcs[0], srcs[1]);
2256 break;
2257 case ir_binop_logic_xor:
2258 result = nir_ixor(&b, srcs[0], srcs[1]);
2259 break;
2260 case ir_binop_lshift: result = nir_ishl(&b, srcs[0], srcs[1]); break;
2261 case ir_binop_rshift:
2262 result = (type_is_signed(out_type)) ? nir_ishr(&b, srcs[0], srcs[1])
2263 : nir_ushr(&b, srcs[0], srcs[1]);
2264 break;
2265 case ir_binop_imul_high:
2266 result = (out_type == GLSL_TYPE_INT) ? nir_imul_high(&b, srcs[0], srcs[1])
2267 : nir_umul_high(&b, srcs[0], srcs[1]);
2268 break;
2269 case ir_binop_carry: result = nir_uadd_carry(&b, srcs[0], srcs[1]); break;
2270 case ir_binop_borrow: result = nir_usub_borrow(&b, srcs[0], srcs[1]); break;
2271 case ir_binop_less:
2272 if (type_is_float(types[0]))
2273 result = nir_flt(&b, srcs[0], srcs[1]);
2274 else if (type_is_signed(types[0]))
2275 result = nir_ilt(&b, srcs[0], srcs[1]);
2276 else
2277 result = nir_ult(&b, srcs[0], srcs[1]);
2278 break;
2279 case ir_binop_gequal:
2280 if (type_is_float(types[0]))
2281 result = nir_fge(&b, srcs[0], srcs[1]);
2282 else if (type_is_signed(types[0]))
2283 result = nir_ige(&b, srcs[0], srcs[1]);
2284 else
2285 result = nir_uge(&b, srcs[0], srcs[1]);
2286 break;
2287 case ir_binop_equal:
2288 if (type_is_float(types[0]))
2289 result = nir_feq(&b, srcs[0], srcs[1]);
2290 else
2291 result = nir_ieq(&b, srcs[0], srcs[1]);
2292 break;
2293 case ir_binop_nequal:
2294 if (type_is_float(types[0]))
2295 result = nir_fne(&b, srcs[0], srcs[1]);
2296 else
2297 result = nir_ine(&b, srcs[0], srcs[1]);
2298 break;
2299 case ir_binop_all_equal:
2300 if (type_is_float(types[0])) {
2301 switch (ir->operands[0]->type->vector_elements) {
2302 case 1: result = nir_feq(&b, srcs[0], srcs[1]); break;
2303 case 2: result = nir_ball_fequal2(&b, srcs[0], srcs[1]); break;
2304 case 3: result = nir_ball_fequal3(&b, srcs[0], srcs[1]); break;
2305 case 4: result = nir_ball_fequal4(&b, srcs[0], srcs[1]); break;
2306 default:
2307 unreachable("not reached");
2308 }
2309 } else {
2310 switch (ir->operands[0]->type->vector_elements) {
2311 case 1: result = nir_ieq(&b, srcs[0], srcs[1]); break;
2312 case 2: result = nir_ball_iequal2(&b, srcs[0], srcs[1]); break;
2313 case 3: result = nir_ball_iequal3(&b, srcs[0], srcs[1]); break;
2314 case 4: result = nir_ball_iequal4(&b, srcs[0], srcs[1]); break;
2315 default:
2316 unreachable("not reached");
2317 }
2318 }
2319 break;
2320 case ir_binop_any_nequal:
2321 if (type_is_float(types[0])) {
2322 switch (ir->operands[0]->type->vector_elements) {
2323 case 1: result = nir_fne(&b, srcs[0], srcs[1]); break;
2324 case 2: result = nir_bany_fnequal2(&b, srcs[0], srcs[1]); break;
2325 case 3: result = nir_bany_fnequal3(&b, srcs[0], srcs[1]); break;
2326 case 4: result = nir_bany_fnequal4(&b, srcs[0], srcs[1]); break;
2327 default:
2328 unreachable("not reached");
2329 }
2330 } else {
2331 switch (ir->operands[0]->type->vector_elements) {
2332 case 1: result = nir_ine(&b, srcs[0], srcs[1]); break;
2333 case 2: result = nir_bany_inequal2(&b, srcs[0], srcs[1]); break;
2334 case 3: result = nir_bany_inequal3(&b, srcs[0], srcs[1]); break;
2335 case 4: result = nir_bany_inequal4(&b, srcs[0], srcs[1]); break;
2336 default:
2337 unreachable("not reached");
2338 }
2339 }
2340 break;
2341 case ir_binop_dot:
2342 switch (ir->operands[0]->type->vector_elements) {
2343 case 2: result = nir_fdot2(&b, srcs[0], srcs[1]); break;
2344 case 3: result = nir_fdot3(&b, srcs[0], srcs[1]); break;
2345 case 4: result = nir_fdot4(&b, srcs[0], srcs[1]); break;
2346 default:
2347 unreachable("not reached");
2348 }
2349 break;
2350 case ir_binop_vector_extract: {
2351 result = nir_channel(&b, srcs[0], 0);
2352 for (unsigned i = 1; i < ir->operands[0]->type->vector_elements; i++) {
2353 nir_ssa_def *swizzled = nir_channel(&b, srcs[0], i);
2354 result = nir_bcsel(&b, nir_ieq(&b, srcs[1], nir_imm_int(&b, i)),
2355 swizzled, result);
2356 }
2357 break;
2358 }
2359
2360 case ir_binop_atan2:
2361 result = nir_atan2(&b, srcs[0], srcs[1]);
2362 break;
2363
2364 case ir_binop_ldexp: result = nir_ldexp(&b, srcs[0], srcs[1]); break;
2365 case ir_triop_fma:
2366 result = nir_ffma(&b, srcs[0], srcs[1], srcs[2]);
2367 break;
2368 case ir_triop_lrp:
2369 result = nir_flrp(&b, srcs[0], srcs[1], srcs[2]);
2370 break;
2371 case ir_triop_csel:
2372 result = nir_bcsel(&b, srcs[0], srcs[1], srcs[2]);
2373 break;
2374 case ir_triop_bitfield_extract:
2375 result = (out_type == GLSL_TYPE_INT) ?
2376 nir_ibitfield_extract(&b, srcs[0], srcs[1], srcs[2]) :
2377 nir_ubitfield_extract(&b, srcs[0], srcs[1], srcs[2]);
2378 break;
2379 case ir_quadop_bitfield_insert:
2380 result = nir_bitfield_insert(&b, srcs[0], srcs[1], srcs[2], srcs[3]);
2381 break;
2382 case ir_quadop_vector:
2383 result = nir_vec(&b, srcs, ir->type->vector_elements);
2384 break;
2385
2386 default:
2387 unreachable("not reached");
2388 }
2389 }
2390
2391 void
2392 nir_visitor::visit(ir_swizzle *ir)
2393 {
2394 unsigned swizzle[4] = { ir->mask.x, ir->mask.y, ir->mask.z, ir->mask.w };
2395 result = nir_swizzle(&b, evaluate_rvalue(ir->val), swizzle,
2396 ir->type->vector_elements);
2397 }
2398
2399 void
2400 nir_visitor::visit(ir_texture *ir)
2401 {
2402 unsigned num_srcs;
2403 nir_texop op;
2404 switch (ir->op) {
2405 case ir_tex:
2406 op = nir_texop_tex;
2407 num_srcs = 1; /* coordinate */
2408 break;
2409
2410 case ir_txb:
2411 case ir_txl:
2412 op = (ir->op == ir_txb) ? nir_texop_txb : nir_texop_txl;
2413 num_srcs = 2; /* coordinate, bias/lod */
2414 break;
2415
2416 case ir_txd:
2417 op = nir_texop_txd; /* coordinate, dPdx, dPdy */
2418 num_srcs = 3;
2419 break;
2420
2421 case ir_txf:
2422 op = nir_texop_txf;
2423 if (ir->lod_info.lod != NULL)
2424 num_srcs = 2; /* coordinate, lod */
2425 else
2426 num_srcs = 1; /* coordinate */
2427 break;
2428
2429 case ir_txf_ms:
2430 op = nir_texop_txf_ms;
2431 num_srcs = 2; /* coordinate, sample_index */
2432 break;
2433
2434 case ir_txs:
2435 op = nir_texop_txs;
2436 if (ir->lod_info.lod != NULL)
2437 num_srcs = 1; /* lod */
2438 else
2439 num_srcs = 0;
2440 break;
2441
2442 case ir_lod:
2443 op = nir_texop_lod;
2444 num_srcs = 1; /* coordinate */
2445 break;
2446
2447 case ir_tg4:
2448 op = nir_texop_tg4;
2449 num_srcs = 1; /* coordinate */
2450 break;
2451
2452 case ir_query_levels:
2453 op = nir_texop_query_levels;
2454 num_srcs = 0;
2455 break;
2456
2457 case ir_texture_samples:
2458 op = nir_texop_texture_samples;
2459 num_srcs = 0;
2460 break;
2461
2462 case ir_samples_identical:
2463 op = nir_texop_samples_identical;
2464 num_srcs = 1; /* coordinate */
2465 break;
2466
2467 default:
2468 unreachable("not reached");
2469 }
2470
2471 if (ir->projector != NULL)
2472 num_srcs++;
2473 if (ir->shadow_comparator != NULL)
2474 num_srcs++;
2475 /* offsets are constants we store inside nir_tex_intrs.offsets */
2476 if (ir->offset != NULL && !ir->offset->type->is_array())
2477 num_srcs++;
2478
2479 /* Add one for the texture deref */
2480 num_srcs += 2;
2481
2482 nir_tex_instr *instr = nir_tex_instr_create(this->shader, num_srcs);
2483
2484 instr->op = op;
2485 instr->sampler_dim =
2486 (glsl_sampler_dim) ir->sampler->type->sampler_dimensionality;
2487 instr->is_array = ir->sampler->type->sampler_array;
2488 instr->is_shadow = ir->sampler->type->sampler_shadow;
2489 if (instr->is_shadow)
2490 instr->is_new_style_shadow = (ir->type->vector_elements == 1);
2491 switch (ir->type->base_type) {
2492 case GLSL_TYPE_FLOAT:
2493 instr->dest_type = nir_type_float;
2494 break;
2495 case GLSL_TYPE_INT:
2496 instr->dest_type = nir_type_int;
2497 break;
2498 case GLSL_TYPE_BOOL:
2499 case GLSL_TYPE_UINT:
2500 instr->dest_type = nir_type_uint;
2501 break;
2502 default:
2503 unreachable("not reached");
2504 }
2505
2506 nir_deref_instr *sampler_deref = evaluate_deref(ir->sampler);
2507
2508 /* check for bindless handles */
2509 if (sampler_deref->mode != nir_var_uniform ||
2510 nir_deref_instr_get_variable(sampler_deref)->data.bindless) {
2511 nir_ssa_def *load = nir_load_deref(&b, sampler_deref);
2512 instr->src[0].src = nir_src_for_ssa(load);
2513 instr->src[0].src_type = nir_tex_src_texture_handle;
2514 instr->src[1].src = nir_src_for_ssa(load);
2515 instr->src[1].src_type = nir_tex_src_sampler_handle;
2516 } else {
2517 instr->src[0].src = nir_src_for_ssa(&sampler_deref->dest.ssa);
2518 instr->src[0].src_type = nir_tex_src_texture_deref;
2519 instr->src[1].src = nir_src_for_ssa(&sampler_deref->dest.ssa);
2520 instr->src[1].src_type = nir_tex_src_sampler_deref;
2521 }
2522
2523 unsigned src_number = 2;
2524
2525 if (ir->coordinate != NULL) {
2526 instr->coord_components = ir->coordinate->type->vector_elements;
2527 instr->src[src_number].src =
2528 nir_src_for_ssa(evaluate_rvalue(ir->coordinate));
2529 instr->src[src_number].src_type = nir_tex_src_coord;
2530 src_number++;
2531 }
2532
2533 if (ir->projector != NULL) {
2534 instr->src[src_number].src =
2535 nir_src_for_ssa(evaluate_rvalue(ir->projector));
2536 instr->src[src_number].src_type = nir_tex_src_projector;
2537 src_number++;
2538 }
2539
2540 if (ir->shadow_comparator != NULL) {
2541 instr->src[src_number].src =
2542 nir_src_for_ssa(evaluate_rvalue(ir->shadow_comparator));
2543 instr->src[src_number].src_type = nir_tex_src_comparator;
2544 src_number++;
2545 }
2546
2547 if (ir->offset != NULL) {
2548 if (ir->offset->type->is_array()) {
2549 for (int i = 0; i < ir->offset->type->array_size(); i++) {
2550 const ir_constant *c =
2551 ir->offset->as_constant()->get_array_element(i);
2552
2553 for (unsigned j = 0; j < 2; ++j) {
2554 int val = c->get_int_component(j);
2555 assert(val <= 31 && val >= -32);
2556 instr->tg4_offsets[i][j] = val;
2557 }
2558 }
2559 } else {
2560 assert(ir->offset->type->is_vector() || ir->offset->type->is_scalar());
2561
2562 instr->src[src_number].src =
2563 nir_src_for_ssa(evaluate_rvalue(ir->offset));
2564 instr->src[src_number].src_type = nir_tex_src_offset;
2565 src_number++;
2566 }
2567 }
2568
2569 switch (ir->op) {
2570 case ir_txb:
2571 instr->src[src_number].src =
2572 nir_src_for_ssa(evaluate_rvalue(ir->lod_info.bias));
2573 instr->src[src_number].src_type = nir_tex_src_bias;
2574 src_number++;
2575 break;
2576
2577 case ir_txl:
2578 case ir_txf:
2579 case ir_txs:
2580 if (ir->lod_info.lod != NULL) {
2581 instr->src[src_number].src =
2582 nir_src_for_ssa(evaluate_rvalue(ir->lod_info.lod));
2583 instr->src[src_number].src_type = nir_tex_src_lod;
2584 src_number++;
2585 }
2586 break;
2587
2588 case ir_txd:
2589 instr->src[src_number].src =
2590 nir_src_for_ssa(evaluate_rvalue(ir->lod_info.grad.dPdx));
2591 instr->src[src_number].src_type = nir_tex_src_ddx;
2592 src_number++;
2593 instr->src[src_number].src =
2594 nir_src_for_ssa(evaluate_rvalue(ir->lod_info.grad.dPdy));
2595 instr->src[src_number].src_type = nir_tex_src_ddy;
2596 src_number++;
2597 break;
2598
2599 case ir_txf_ms:
2600 instr->src[src_number].src =
2601 nir_src_for_ssa(evaluate_rvalue(ir->lod_info.sample_index));
2602 instr->src[src_number].src_type = nir_tex_src_ms_index;
2603 src_number++;
2604 break;
2605
2606 case ir_tg4:
2607 instr->component = ir->lod_info.component->as_constant()->value.u[0];
2608 break;
2609
2610 default:
2611 break;
2612 }
2613
2614 assert(src_number == num_srcs);
2615
2616 unsigned bit_size = glsl_get_bit_size(ir->type);
2617 add_instr(&instr->instr, nir_tex_instr_dest_size(instr), bit_size);
2618 }
2619
2620 void
2621 nir_visitor::visit(ir_constant *ir)
2622 {
2623 /*
2624 * We don't know if this variable is an array or struct that gets
2625 * dereferenced, so do the safe thing an make it a variable with a
2626 * constant initializer and return a dereference.
2627 */
2628
2629 nir_variable *var =
2630 nir_local_variable_create(this->impl, ir->type, "const_temp");
2631 var->data.read_only = true;
2632 var->constant_initializer = constant_copy(ir, var);
2633
2634 this->deref = nir_build_deref_var(&b, var);
2635 }
2636
2637 void
2638 nir_visitor::visit(ir_dereference_variable *ir)
2639 {
2640 if (ir->variable_referenced()->data.mode == ir_var_function_out) {
2641 unsigned i = (sig->return_type != glsl_type::void_type) ? 1 : 0;
2642
2643 foreach_in_list(ir_variable, param, &sig->parameters) {
2644 if (param == ir->variable_referenced()) {
2645 break;
2646 }
2647 i++;
2648 }
2649
2650 this->deref = nir_build_deref_cast(&b, nir_load_param(&b, i),
2651 nir_var_function_temp, ir->type, 0);
2652 return;
2653 }
2654
2655 assert(ir->variable_referenced()->data.mode != ir_var_function_inout);
2656
2657 struct hash_entry *entry =
2658 _mesa_hash_table_search(this->var_table, ir->var);
2659 assert(entry);
2660 nir_variable *var = (nir_variable *) entry->data;
2661
2662 this->deref = nir_build_deref_var(&b, var);
2663 }
2664
2665 void
2666 nir_visitor::visit(ir_dereference_record *ir)
2667 {
2668 ir->record->accept(this);
2669
2670 int field_index = ir->field_idx;
2671 assert(field_index >= 0);
2672
2673 this->deref = nir_build_deref_struct(&b, this->deref, field_index);
2674 }
2675
2676 void
2677 nir_visitor::visit(ir_dereference_array *ir)
2678 {
2679 nir_ssa_def *index = evaluate_rvalue(ir->array_index);
2680
2681 ir->array->accept(this);
2682
2683 this->deref = nir_build_deref_array(&b, this->deref, index);
2684 }
2685
2686 void
2687 nir_visitor::visit(ir_barrier *)
2688 {
2689 nir_intrinsic_instr *instr =
2690 nir_intrinsic_instr_create(this->shader, nir_intrinsic_barrier);
2691 nir_builder_instr_insert(&b, &instr->instr);
2692 }
2693
2694 nir_shader *
2695 glsl_float64_funcs_to_nir(struct gl_context *ctx,
2696 const nir_shader_compiler_options *options)
2697 {
2698 /* We pretend it's a vertex shader. Ultimately, the stage shouldn't
2699 * matter because we're not optimizing anything here.
2700 */
2701 struct gl_shader *sh = _mesa_new_shader(-1, MESA_SHADER_VERTEX);
2702 sh->Source = float64_source;
2703 sh->CompileStatus = COMPILE_FAILURE;
2704 _mesa_glsl_compile_shader(ctx, sh, false, false, true);
2705
2706 if (!sh->CompileStatus) {
2707 if (sh->InfoLog) {
2708 _mesa_problem(ctx,
2709 "fp64 software impl compile failed:\n%s\nsource:\n%s\n",
2710 sh->InfoLog, float64_source);
2711 }
2712 return NULL;
2713 }
2714
2715 nir_shader *nir = nir_shader_create(NULL, MESA_SHADER_VERTEX, options, NULL);
2716
2717 nir_visitor v1(ctx, nir);
2718 nir_function_visitor v2(&v1);
2719 v2.run(sh->ir);
2720 visit_exec_list(sh->ir, &v1);
2721
2722 /* _mesa_delete_shader will try to free sh->Source but it's static const */
2723 sh->Source = NULL;
2724 _mesa_delete_shader(ctx, sh);
2725
2726 nir_validate_shader(nir, "float64_funcs_to_nir");
2727
2728 NIR_PASS_V(nir, nir_lower_constant_initializers, nir_var_function_temp);
2729 NIR_PASS_V(nir, nir_lower_returns);
2730 NIR_PASS_V(nir, nir_inline_functions);
2731 NIR_PASS_V(nir, nir_opt_deref);
2732
2733 /* Do some optimizations to clean up the shader now. By optimizing the
2734 * functions in the library, we avoid having to re-do that work every
2735 * time we inline a copy of a function. Reducing basic blocks also helps
2736 * with compile times.
2737 */
2738 NIR_PASS_V(nir, nir_lower_vars_to_ssa);
2739 NIR_PASS_V(nir, nir_copy_prop);
2740 NIR_PASS_V(nir, nir_opt_dce);
2741 NIR_PASS_V(nir, nir_opt_cse);
2742 NIR_PASS_V(nir, nir_opt_gcm, true);
2743 NIR_PASS_V(nir, nir_opt_peephole_select, 1, false, false);
2744 NIR_PASS_V(nir, nir_opt_dce);
2745
2746 return nir;
2747 }