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