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