2 * Copyright © 2014-2015 Broadcom
3 * Copyright (C) 2014 Rob Clark <robclark@freedesktop.org>
5 * Permission is hereby granted, free of charge, to any person obtaining a
6 * copy of this software and associated documentation files (the "Software"),
7 * to deal in the Software without restriction, including without limitation
8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9 * and/or sell copies of the Software, and to permit persons to whom the
10 * Software is furnished to do so, subject to the following conditions:
12 * The above copyright notice and this permission notice (including the next
13 * paragraph) shall be included in all copies or substantial portions of the
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
25 #include "util/ralloc.h"
26 #include "glsl/nir/nir.h"
27 #include "glsl/nir/nir_builder.h"
28 #include "glsl/list.h"
29 #include "glsl/shader_enums.h"
31 #include "nir/tgsi_to_nir.h"
32 #include "tgsi/tgsi_parse.h"
33 #include "tgsi/tgsi_dump.h"
34 #include "tgsi/tgsi_info.h"
35 #include "tgsi/tgsi_scan.h"
37 #define SWIZ(X, Y, Z, W) (unsigned[4]){ \
45 /** nir register containing this TGSI index. */
48 /** Offset (in vec4s) from the start of var for this TGSI index. */
53 union tgsi_full_token
*token
;
55 struct tgsi_shader_info
*scan
;
57 struct ttn_reg_info
*output_regs
;
58 struct ttn_reg_info
*temp_regs
;
59 nir_ssa_def
**imm_defs
;
61 unsigned num_samp_types
;
62 nir_alu_type
*samp_types
;
64 nir_register
*addr_reg
;
67 * Stack of cf_node_lists where instructions should be pushed as we pop
68 * back out of the control flow stack.
70 * For each IF/ELSE/ENDIF block, if_stack[if_stack_pos] has where the else
71 * instructions should be placed, and if_stack[if_stack_pos - 1] has where
72 * the next instructions outside of the if/then/else block go.
74 struct exec_list
**if_stack
;
75 unsigned if_stack_pos
;
78 * Stack of cf_node_lists where instructions should be pushed as we pop
79 * back out of the control flow stack.
81 * loop_stack[loop_stack_pos - 1] contains the cf_node_list for the outside
84 struct exec_list
**loop_stack
;
85 unsigned loop_stack_pos
;
87 /* How many TGSI_FILE_IMMEDIATE vec4s have been parsed so far. */
91 #define ttn_swizzle(b, src, x, y, z, w) \
92 nir_swizzle(b, src, SWIZ(x, y, z, w), 4, false)
93 #define ttn_channel(b, src, swiz) \
94 nir_swizzle(b, src, SWIZ(swiz, swiz, swiz, swiz), 1, false)
97 ttn_src_for_dest(nir_builder
*b
, nir_alu_dest
*dest
)
100 memset(&src
, 0, sizeof(src
));
102 if (dest
->dest
.is_ssa
)
103 src
.src
= nir_src_for_ssa(&dest
->dest
.ssa
);
105 assert(!dest
->dest
.reg
.indirect
);
106 src
.src
= nir_src_for_reg(dest
->dest
.reg
.reg
);
107 src
.src
.reg
.base_offset
= dest
->dest
.reg
.base_offset
;
110 for (int i
= 0; i
< 4; i
++)
113 return nir_fmov_alu(b
, src
, 4);
117 ttn_emit_declaration(struct ttn_compile
*c
)
119 nir_builder
*b
= &c
->build
;
120 struct tgsi_full_declaration
*decl
= &c
->token
->FullDeclaration
;
121 unsigned array_size
= decl
->Range
.Last
- decl
->Range
.First
+ 1;
122 unsigned file
= decl
->Declaration
.File
;
125 if (file
== TGSI_FILE_TEMPORARY
) {
126 if (decl
->Declaration
.Array
) {
127 /* for arrays, we create variables instead of registers: */
128 nir_variable
*var
= rzalloc(b
->shader
, nir_variable
);
130 var
->type
= glsl_array_type(glsl_vec4_type(), array_size
);
131 var
->data
.mode
= nir_var_global
;
132 var
->name
= ralloc_asprintf(var
, "arr_%d", decl
->Array
.ArrayID
);
134 exec_list_push_tail(&b
->shader
->globals
, &var
->node
);
136 for (i
= 0; i
< array_size
; i
++) {
137 /* point all the matching slots to the same var,
138 * with appropriate offset set, mostly just so
139 * we know what to do when tgsi does a non-indirect
142 c
->temp_regs
[decl
->Range
.First
+ i
].reg
= NULL
;
143 c
->temp_regs
[decl
->Range
.First
+ i
].var
= var
;
144 c
->temp_regs
[decl
->Range
.First
+ i
].offset
= i
;
147 for (i
= 0; i
< array_size
; i
++) {
148 nir_register
*reg
= nir_local_reg_create(b
->impl
);
149 reg
->num_components
= 4;
150 c
->temp_regs
[decl
->Range
.First
+ i
].reg
= reg
;
151 c
->temp_regs
[decl
->Range
.First
+ i
].var
= NULL
;
152 c
->temp_regs
[decl
->Range
.First
+ i
].offset
= 0;
155 } else if (file
== TGSI_FILE_ADDRESS
) {
156 c
->addr_reg
= nir_local_reg_create(b
->impl
);
157 c
->addr_reg
->num_components
= 4;
158 } else if (file
== TGSI_FILE_SYSTEM_VALUE
) {
159 /* Nothing to record for system values. */
160 } else if (file
== TGSI_FILE_SAMPLER
) {
161 /* Nothing to record for samplers. */
162 } else if (file
== TGSI_FILE_SAMPLER_VIEW
) {
163 struct tgsi_declaration_sampler_view
*sview
= &decl
->SamplerView
;
166 assert((sview
->ReturnTypeX
== sview
->ReturnTypeY
) &&
167 (sview
->ReturnTypeX
== sview
->ReturnTypeZ
) &&
168 (sview
->ReturnTypeX
== sview
->ReturnTypeW
));
170 switch (sview
->ReturnTypeX
) {
171 case TGSI_RETURN_TYPE_SINT
:
174 case TGSI_RETURN_TYPE_UINT
:
175 type
= nir_type_unsigned
;
177 case TGSI_RETURN_TYPE_FLOAT
:
179 type
= nir_type_float
;
183 for (i
= 0; i
< array_size
; i
++) {
184 c
->samp_types
[decl
->Range
.First
+ i
] = type
;
188 assert(file
== TGSI_FILE_INPUT
||
189 file
== TGSI_FILE_OUTPUT
||
190 file
== TGSI_FILE_CONSTANT
);
192 /* nothing to do for UBOs: */
193 if ((file
== TGSI_FILE_CONSTANT
) && decl
->Declaration
.Dimension
)
196 var
= rzalloc(b
->shader
, nir_variable
);
197 var
->data
.driver_location
= decl
->Range
.First
;
199 var
->type
= glsl_vec4_type();
201 var
->type
= glsl_array_type(var
->type
, array_size
);
204 case TGSI_FILE_INPUT
:
205 var
->data
.read_only
= true;
206 var
->data
.mode
= nir_var_shader_in
;
207 var
->name
= ralloc_asprintf(var
, "in_%d", decl
->Range
.First
);
209 /* We should probably translate to a VERT_ATTRIB_* or VARYING_SLOT_*
210 * instead, but nothing in NIR core is looking at the value
211 * currently, and this is less change to drivers.
213 var
->data
.location
= decl
->Semantic
.Name
;
214 var
->data
.index
= decl
->Semantic
.Index
;
216 /* We definitely need to translate the interpolation field, because
217 * nir_print will decode it.
219 switch (decl
->Interp
.Interpolate
) {
220 case TGSI_INTERPOLATE_CONSTANT
:
221 var
->data
.interpolation
= INTERP_QUALIFIER_FLAT
;
223 case TGSI_INTERPOLATE_LINEAR
:
224 var
->data
.interpolation
= INTERP_QUALIFIER_NOPERSPECTIVE
;
226 case TGSI_INTERPOLATE_PERSPECTIVE
:
227 var
->data
.interpolation
= INTERP_QUALIFIER_SMOOTH
;
231 exec_list_push_tail(&b
->shader
->inputs
, &var
->node
);
233 case TGSI_FILE_OUTPUT
: {
234 /* Since we can't load from outputs in the IR, we make temporaries
235 * for the outputs and emit stores to the real outputs at the end of
238 nir_register
*reg
= nir_local_reg_create(b
->impl
);
239 reg
->num_components
= 4;
241 reg
->num_array_elems
= array_size
;
243 var
->data
.mode
= nir_var_shader_out
;
244 var
->name
= ralloc_asprintf(var
, "out_%d", decl
->Range
.First
);
246 var
->data
.location
= decl
->Semantic
.Name
;
247 var
->data
.index
= decl
->Semantic
.Index
;
249 for (i
= 0; i
< array_size
; i
++) {
250 c
->output_regs
[decl
->Range
.First
+ i
].offset
= i
;
251 c
->output_regs
[decl
->Range
.First
+ i
].reg
= reg
;
254 exec_list_push_tail(&b
->shader
->outputs
, &var
->node
);
257 case TGSI_FILE_CONSTANT
:
258 var
->data
.mode
= nir_var_uniform
;
259 var
->name
= ralloc_asprintf(var
, "uniform_%d", decl
->Range
.First
);
261 exec_list_push_tail(&b
->shader
->uniforms
, &var
->node
);
264 unreachable("bad declaration file");
272 ttn_emit_immediate(struct ttn_compile
*c
)
274 nir_builder
*b
= &c
->build
;
275 struct tgsi_full_immediate
*tgsi_imm
= &c
->token
->FullImmediate
;
276 nir_load_const_instr
*load_const
;
279 load_const
= nir_load_const_instr_create(b
->shader
, 4);
280 c
->imm_defs
[c
->next_imm
] = &load_const
->def
;
283 for (i
= 0; i
< 4; i
++)
284 load_const
->value
.u
[i
] = tgsi_imm
->u
[i
].Uint
;
286 nir_instr_insert_after_cf_list(b
->cf_node_list
, &load_const
->instr
);
290 ttn_src_for_indirect(struct ttn_compile
*c
, struct tgsi_ind_register
*indirect
);
292 /* generate either a constant or indirect deref chain for accessing an
295 static nir_deref_var
*
296 ttn_array_deref(struct ttn_compile
*c
, nir_intrinsic_instr
*instr
,
297 nir_variable
*var
, unsigned offset
,
298 struct tgsi_ind_register
*indirect
)
300 nir_deref_var
*deref
= nir_deref_var_create(instr
, var
);
301 nir_deref_array
*arr
= nir_deref_array_create(deref
);
303 arr
->base_offset
= offset
;
304 arr
->deref
.type
= glsl_get_array_element(var
->type
);
307 arr
->deref_array_type
= nir_deref_array_type_indirect
;
308 arr
->indirect
= ttn_src_for_indirect(c
, indirect
);
310 arr
->deref_array_type
= nir_deref_array_type_direct
;
313 deref
->deref
.child
= &arr
->deref
;
319 ttn_src_for_file_and_index(struct ttn_compile
*c
, unsigned file
, unsigned index
,
320 struct tgsi_ind_register
*indirect
,
321 struct tgsi_dimension
*dim
,
322 struct tgsi_ind_register
*dimind
)
324 nir_builder
*b
= &c
->build
;
327 memset(&src
, 0, sizeof(src
));
330 case TGSI_FILE_TEMPORARY
:
331 if (c
->temp_regs
[index
].var
) {
332 unsigned offset
= c
->temp_regs
[index
].offset
;
333 nir_variable
*var
= c
->temp_regs
[index
].var
;
334 nir_intrinsic_instr
*load
;
336 load
= nir_intrinsic_instr_create(b
->shader
,
337 nir_intrinsic_load_var
);
338 load
->num_components
= 4;
339 load
->variables
[0] = ttn_array_deref(c
, load
, var
, offset
, indirect
);
341 nir_ssa_dest_init(&load
->instr
, &load
->dest
, 4, NULL
);
342 nir_instr_insert_after_cf_list(b
->cf_node_list
, &load
->instr
);
344 src
= nir_src_for_ssa(&load
->dest
.ssa
);
348 src
.reg
.reg
= c
->temp_regs
[index
].reg
;
353 case TGSI_FILE_ADDRESS
:
354 src
.reg
.reg
= c
->addr_reg
;
358 case TGSI_FILE_IMMEDIATE
:
359 src
= nir_src_for_ssa(c
->imm_defs
[index
]);
364 case TGSI_FILE_SYSTEM_VALUE
: {
365 nir_intrinsic_instr
*load
;
372 switch (c
->scan
->system_value_semantic_name
[index
]) {
373 case TGSI_SEMANTIC_VERTEXID_NOBASE
:
374 op
= nir_intrinsic_load_vertex_id_zero_base
;
376 case TGSI_SEMANTIC_VERTEXID
:
377 op
= nir_intrinsic_load_vertex_id
;
379 case TGSI_SEMANTIC_BASEVERTEX
:
380 op
= nir_intrinsic_load_base_vertex
;
382 case TGSI_SEMANTIC_INSTANCEID
:
383 op
= nir_intrinsic_load_instance_id
;
386 unreachable("bad system value");
389 load
= nir_intrinsic_instr_create(b
->shader
, op
);
390 load
->num_components
= ncomp
;
392 nir_ssa_dest_init(&load
->instr
, &load
->dest
, ncomp
, NULL
);
393 nir_instr_insert_after_cf_list(b
->cf_node_list
, &load
->instr
);
395 src
= nir_src_for_ssa(&load
->dest
.ssa
);
399 case TGSI_FILE_INPUT
:
400 case TGSI_FILE_CONSTANT
: {
401 nir_intrinsic_instr
*load
;
406 case TGSI_FILE_INPUT
:
407 op
= indirect
? nir_intrinsic_load_input_indirect
:
408 nir_intrinsic_load_input
;
411 case TGSI_FILE_CONSTANT
:
413 op
= indirect
? nir_intrinsic_load_ubo_indirect
:
414 nir_intrinsic_load_ubo
;
415 /* convert index from vec4 to byte: */
418 op
= indirect
? nir_intrinsic_load_uniform_indirect
:
419 nir_intrinsic_load_uniform
;
423 unreachable("No other load files supported");
427 load
= nir_intrinsic_instr_create(b
->shader
, op
);
429 load
->num_components
= 4;
430 load
->const_index
[0] = index
;
434 ttn_src_for_file_and_index(c
, dimind
->File
, dimind
->Index
,
437 /* UBOs start at index 1 in TGSI: */
439 nir_src_for_ssa(nir_imm_int(b
, dim
->Index
- 1));
444 load
->src
[srcn
] = ttn_src_for_indirect(c
, indirect
);
446 assert(load
->src
[srcn
].is_ssa
);
447 /* we also need to covert vec4 to byte here too: */
449 nir_src_for_ssa(nir_ishl(b
, load
->src
[srcn
].ssa
,
454 nir_ssa_dest_init(&load
->instr
, &load
->dest
, 4, NULL
);
455 nir_instr_insert_after_cf_list(b
->cf_node_list
, &load
->instr
);
457 src
= nir_src_for_ssa(&load
->dest
.ssa
);
462 unreachable("bad src file");
470 ttn_src_for_indirect(struct ttn_compile
*c
, struct tgsi_ind_register
*indirect
)
472 nir_builder
*b
= &c
->build
;
474 memset(&src
, 0, sizeof(src
));
475 for (int i
= 0; i
< 4; i
++)
476 src
.swizzle
[i
] = indirect
->Swizzle
;
477 src
.src
= ttn_src_for_file_and_index(c
,
481 return nir_src_for_ssa(nir_imov_alu(b
, src
, 1));
485 ttn_get_dest(struct ttn_compile
*c
, struct tgsi_full_dst_register
*tgsi_fdst
)
487 struct tgsi_dst_register
*tgsi_dst
= &tgsi_fdst
->Register
;
489 unsigned index
= tgsi_dst
->Index
;
491 memset(&dest
, 0, sizeof(dest
));
493 if (tgsi_dst
->File
== TGSI_FILE_TEMPORARY
) {
494 if (c
->temp_regs
[index
].var
) {
495 nir_builder
*b
= &c
->build
;
496 nir_intrinsic_instr
*load
;
497 struct tgsi_ind_register
*indirect
=
498 tgsi_dst
->Indirect
? &tgsi_fdst
->Indirect
: NULL
;
501 /* this works, because TGSI will give us a base offset
502 * (in case of indirect index) that points back into
503 * the array. Access can be direct or indirect, we
504 * don't really care. Just create a one-shot dst reg
505 * that will get store_var'd back into the array var
506 * at the end of ttn_emit_instruction()
508 reg
= nir_local_reg_create(c
->build
.impl
);
509 reg
->num_components
= 4;
510 dest
.dest
.reg
.reg
= reg
;
511 dest
.dest
.reg
.base_offset
= 0;
513 /* since the alu op might not write to all components
514 * of the temporary, we must first do a load_var to
515 * get the previous array elements into the register.
516 * This is one area that NIR could use a bit of
517 * improvement (or opt pass to clean up the mess
518 * once things are scalarized)
521 load
= nir_intrinsic_instr_create(c
->build
.shader
,
522 nir_intrinsic_load_var
);
523 load
->num_components
= 4;
525 ttn_array_deref(c
, load
, c
->temp_regs
[index
].var
,
526 c
->temp_regs
[index
].offset
,
529 load
->dest
= nir_dest_for_reg(reg
);
531 nir_instr_insert_after_cf_list(b
->cf_node_list
, &load
->instr
);
533 assert(!tgsi_dst
->Indirect
);
534 dest
.dest
.reg
.reg
= c
->temp_regs
[index
].reg
;
535 dest
.dest
.reg
.base_offset
= c
->temp_regs
[index
].offset
;
537 } else if (tgsi_dst
->File
== TGSI_FILE_OUTPUT
) {
538 dest
.dest
.reg
.reg
= c
->output_regs
[index
].reg
;
539 dest
.dest
.reg
.base_offset
= c
->output_regs
[index
].offset
;
540 } else if (tgsi_dst
->File
== TGSI_FILE_ADDRESS
) {
542 dest
.dest
.reg
.reg
= c
->addr_reg
;
545 dest
.write_mask
= tgsi_dst
->WriteMask
;
546 dest
.saturate
= false;
548 if (tgsi_dst
->Indirect
&& (tgsi_dst
->File
!= TGSI_FILE_TEMPORARY
)) {
549 nir_src
*indirect
= ralloc(c
->build
.shader
, nir_src
);
550 *indirect
= ttn_src_for_indirect(c
, &tgsi_fdst
->Indirect
);
551 dest
.dest
.reg
.indirect
= indirect
;
557 static nir_variable
*
558 ttn_get_var(struct ttn_compile
*c
, struct tgsi_full_dst_register
*tgsi_fdst
)
560 struct tgsi_dst_register
*tgsi_dst
= &tgsi_fdst
->Register
;
561 unsigned index
= tgsi_dst
->Index
;
563 if (tgsi_dst
->File
== TGSI_FILE_TEMPORARY
) {
564 /* we should not have an indirect when there is no var! */
565 if (!c
->temp_regs
[index
].var
)
566 assert(!tgsi_dst
->Indirect
);
567 return c
->temp_regs
[index
].var
;
574 ttn_get_src(struct ttn_compile
*c
, struct tgsi_full_src_register
*tgsi_fsrc
)
576 nir_builder
*b
= &c
->build
;
577 struct tgsi_src_register
*tgsi_src
= &tgsi_fsrc
->Register
;
578 unsigned tgsi_opcode
= c
->token
->FullInstruction
.Instruction
.Opcode
;
579 unsigned tgsi_src_type
= tgsi_opcode_infer_src_type(tgsi_opcode
);
580 bool src_is_float
= !(tgsi_src_type
== TGSI_TYPE_SIGNED
||
581 tgsi_src_type
== TGSI_TYPE_UNSIGNED
);
584 memset(&src
, 0, sizeof(src
));
586 if (tgsi_src
->File
== TGSI_FILE_NULL
) {
587 return nir_imm_float(b
, 0.0);
588 } else if (tgsi_src
->File
== TGSI_FILE_SAMPLER
) {
589 /* Only the index of the sampler gets used in texturing, and it will
590 * handle looking that up on its own instead of using the nir_alu_src.
592 assert(!tgsi_src
->Indirect
);
595 struct tgsi_ind_register
*ind
= NULL
;
596 struct tgsi_dimension
*dim
= NULL
;
597 struct tgsi_ind_register
*dimind
= NULL
;
598 if (tgsi_src
->Indirect
)
599 ind
= &tgsi_fsrc
->Indirect
;
600 if (tgsi_src
->Dimension
) {
601 dim
= &tgsi_fsrc
->Dimension
;
603 dimind
= &tgsi_fsrc
->DimIndirect
;
605 src
.src
= ttn_src_for_file_and_index(c
,
611 src
.swizzle
[0] = tgsi_src
->SwizzleX
;
612 src
.swizzle
[1] = tgsi_src
->SwizzleY
;
613 src
.swizzle
[2] = tgsi_src
->SwizzleZ
;
614 src
.swizzle
[3] = tgsi_src
->SwizzleW
;
616 nir_ssa_def
*def
= nir_fmov_alu(b
, src
, 4);
618 if (tgsi_src
->Absolute
) {
620 def
= nir_fabs(b
, def
);
622 def
= nir_iabs(b
, def
);
625 if (tgsi_src
->Negate
) {
627 def
= nir_fneg(b
, def
);
629 def
= nir_ineg(b
, def
);
636 ttn_alu(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
638 unsigned num_srcs
= nir_op_infos
[op
].num_inputs
;
639 nir_alu_instr
*instr
= nir_alu_instr_create(b
->shader
, op
);
642 for (i
= 0; i
< num_srcs
; i
++)
643 instr
->src
[i
].src
= nir_src_for_ssa(src
[i
]);
646 nir_instr_insert_after_cf_list(b
->cf_node_list
, &instr
->instr
);
650 ttn_move_dest_masked(nir_builder
*b
, nir_alu_dest dest
,
651 nir_ssa_def
*def
, unsigned write_mask
)
653 if (!(dest
.write_mask
& write_mask
))
656 nir_alu_instr
*mov
= nir_alu_instr_create(b
->shader
, nir_op_imov
);
658 mov
->dest
.write_mask
&= write_mask
;
659 mov
->src
[0].src
= nir_src_for_ssa(def
);
660 for (unsigned i
= def
->num_components
; i
< 4; i
++)
661 mov
->src
[0].swizzle
[i
] = def
->num_components
- 1;
662 nir_instr_insert_after_cf_list(b
->cf_node_list
, &mov
->instr
);
666 ttn_move_dest(nir_builder
*b
, nir_alu_dest dest
, nir_ssa_def
*def
)
668 ttn_move_dest_masked(b
, dest
, def
, TGSI_WRITEMASK_XYZW
);
672 ttn_arl(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
674 ttn_move_dest(b
, dest
, nir_f2i(b
, nir_ffloor(b
, src
[0])));
677 /* EXP - Approximate Exponential Base 2
678 * dst.x = 2^{\lfloor src.x\rfloor}
679 * dst.y = src.x - \lfloor src.x\rfloor
684 ttn_exp(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
686 nir_ssa_def
*srcx
= ttn_channel(b
, src
[0], X
);
688 ttn_move_dest_masked(b
, dest
, nir_fexp2(b
, nir_ffloor(b
, srcx
)),
690 ttn_move_dest_masked(b
, dest
, nir_fsub(b
, srcx
, nir_ffloor(b
, srcx
)),
692 ttn_move_dest_masked(b
, dest
, nir_fexp2(b
, srcx
), TGSI_WRITEMASK_Z
);
693 ttn_move_dest_masked(b
, dest
, nir_imm_float(b
, 1.0), TGSI_WRITEMASK_W
);
696 /* LOG - Approximate Logarithm Base 2
697 * dst.x = \lfloor\log_2{|src.x|}\rfloor
698 * dst.y = \frac{|src.x|}{2^{\lfloor\log_2{|src.x|}\rfloor}}
699 * dst.z = \log_2{|src.x|}
703 ttn_log(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
705 nir_ssa_def
*abs_srcx
= nir_fabs(b
, ttn_channel(b
, src
[0], X
));
706 nir_ssa_def
*log2
= nir_flog2(b
, abs_srcx
);
708 ttn_move_dest_masked(b
, dest
, nir_ffloor(b
, log2
), TGSI_WRITEMASK_X
);
709 ttn_move_dest_masked(b
, dest
,
710 nir_fdiv(b
, abs_srcx
, nir_fexp2(b
, nir_ffloor(b
, log2
))),
712 ttn_move_dest_masked(b
, dest
, nir_flog2(b
, abs_srcx
), TGSI_WRITEMASK_Z
);
713 ttn_move_dest_masked(b
, dest
, nir_imm_float(b
, 1.0), TGSI_WRITEMASK_W
);
716 /* DST - Distance Vector
718 * dst.y = src0.y \times src1.y
723 ttn_dst(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
725 ttn_move_dest_masked(b
, dest
, nir_imm_float(b
, 1.0), TGSI_WRITEMASK_X
);
726 ttn_move_dest_masked(b
, dest
, nir_fmul(b
, src
[0], src
[1]), TGSI_WRITEMASK_Y
);
727 ttn_move_dest_masked(b
, dest
, nir_fmov(b
, src
[0]), TGSI_WRITEMASK_Z
);
728 ttn_move_dest_masked(b
, dest
, nir_fmov(b
, src
[1]), TGSI_WRITEMASK_W
);
731 /* LIT - Light Coefficients
733 * dst.y = max(src.x, 0.0)
734 * dst.z = (src.x > 0.0) ? max(src.y, 0.0)^{clamp(src.w, -128.0, 128.0))} : 0
738 ttn_lit(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
740 ttn_move_dest_masked(b
, dest
, nir_imm_float(b
, 1.0), TGSI_WRITEMASK_XW
);
742 ttn_move_dest_masked(b
, dest
, nir_fmax(b
, ttn_channel(b
, src
[0], X
),
743 nir_imm_float(b
, 0.0)), TGSI_WRITEMASK_Y
);
745 if (dest
.write_mask
& TGSI_WRITEMASK_Z
) {
746 nir_ssa_def
*src0_y
= ttn_channel(b
, src
[0], Y
);
747 nir_ssa_def
*wclamp
= nir_fmax(b
, nir_fmin(b
, ttn_channel(b
, src
[0], W
),
748 nir_imm_float(b
, 128.0)),
749 nir_imm_float(b
, -128.0));
750 nir_ssa_def
*pow
= nir_fpow(b
, nir_fmax(b
, src0_y
, nir_imm_float(b
, 0.0)),
753 ttn_move_dest_masked(b
, dest
,
756 nir_imm_float(b
, 0.0),
757 ttn_channel(b
, src
[0], X
)),
758 nir_imm_float(b
, 0.0),
765 * dst.x = \cos{src.x}
766 * dst.y = \sin{src.x}
771 ttn_scs(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
773 ttn_move_dest_masked(b
, dest
, nir_fcos(b
, ttn_channel(b
, src
[0], X
)),
775 ttn_move_dest_masked(b
, dest
, nir_fsin(b
, ttn_channel(b
, src
[0], X
)),
777 ttn_move_dest_masked(b
, dest
, nir_imm_float(b
, 0.0), TGSI_WRITEMASK_Z
);
778 ttn_move_dest_masked(b
, dest
, nir_imm_float(b
, 1.0), TGSI_WRITEMASK_W
);
782 ttn_sle(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
784 ttn_move_dest(b
, dest
, nir_sge(b
, src
[1], src
[0]));
788 ttn_sgt(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
790 ttn_move_dest(b
, dest
, nir_slt(b
, src
[1], src
[0]));
794 ttn_clamp(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
796 ttn_move_dest(b
, dest
, nir_fmin(b
, nir_fmax(b
, src
[0], src
[1]), src
[2]));
800 ttn_xpd(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
802 ttn_move_dest_masked(b
, dest
,
805 ttn_swizzle(b
, src
[0], Y
, Z
, X
, X
),
806 ttn_swizzle(b
, src
[1], Z
, X
, Y
, X
)),
808 ttn_swizzle(b
, src
[1], Y
, Z
, X
, X
),
809 ttn_swizzle(b
, src
[0], Z
, X
, Y
, X
))),
811 ttn_move_dest_masked(b
, dest
, nir_imm_float(b
, 1.0), TGSI_WRITEMASK_W
);
815 ttn_dp2a(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
817 ttn_move_dest(b
, dest
,
818 ttn_channel(b
, nir_fadd(b
, nir_fdot2(b
, src
[0], src
[1]),
824 ttn_dp2(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
826 ttn_move_dest(b
, dest
, nir_fdot2(b
, src
[0], src
[1]));
830 ttn_dp3(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
832 ttn_move_dest(b
, dest
, nir_fdot3(b
, src
[0], src
[1]));
836 ttn_dp4(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
838 ttn_move_dest(b
, dest
, nir_fdot4(b
, src
[0], src
[1]));
842 ttn_dph(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
844 ttn_move_dest(b
, dest
, nir_fadd(b
, nir_fdot3(b
, src
[0], src
[1]),
845 ttn_channel(b
, src
[1], W
)));
849 ttn_umad(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
851 ttn_move_dest(b
, dest
, nir_iadd(b
, nir_imul(b
, src
[0], src
[1]), src
[2]));
855 ttn_arr(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
857 ttn_move_dest(b
, dest
, nir_ffloor(b
, nir_fadd(b
, src
[0], nir_imm_float(b
, 0.5))));
861 ttn_cmp(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
863 ttn_move_dest(b
, dest
, nir_bcsel(b
,
864 nir_flt(b
, src
[0], nir_imm_float(b
, 0.0)),
869 ttn_ucmp(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
871 ttn_move_dest(b
, dest
, nir_bcsel(b
,
872 nir_ine(b
, src
[0], nir_imm_int(b
, 0)),
877 ttn_kill(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
879 nir_intrinsic_instr
*discard
=
880 nir_intrinsic_instr_create(b
->shader
, nir_intrinsic_discard
);
881 nir_instr_insert_after_cf_list(b
->cf_node_list
, &discard
->instr
);
885 ttn_kill_if(nir_builder
*b
, nir_op op
, nir_alu_dest dest
, nir_ssa_def
**src
)
887 nir_ssa_def
*cmp
= nir_bany4(b
, nir_flt(b
, src
[0], nir_imm_float(b
, 0.0)));
888 nir_intrinsic_instr
*discard
=
889 nir_intrinsic_instr_create(b
->shader
, nir_intrinsic_discard_if
);
890 discard
->src
[0] = nir_src_for_ssa(cmp
);
891 nir_instr_insert_after_cf_list(b
->cf_node_list
, &discard
->instr
);
895 ttn_if(struct ttn_compile
*c
, nir_ssa_def
*src
, bool is_uint
)
897 nir_builder
*b
= &c
->build
;
899 /* Save the outside-of-the-if-statement node list. */
900 c
->if_stack
[c
->if_stack_pos
] = b
->cf_node_list
;
903 src
= ttn_channel(b
, src
, X
);
905 nir_if
*if_stmt
= nir_if_create(b
->shader
);
907 if_stmt
->condition
= nir_src_for_ssa(nir_ine(b
, src
, nir_imm_int(b
, 0)));
909 if_stmt
->condition
= nir_src_for_ssa(nir_fne(b
, src
, nir_imm_int(b
, 0)));
911 nir_cf_node_insert_end(b
->cf_node_list
, &if_stmt
->cf_node
);
913 nir_builder_insert_after_cf_list(b
, &if_stmt
->then_list
);
915 c
->if_stack
[c
->if_stack_pos
] = &if_stmt
->else_list
;
920 ttn_else(struct ttn_compile
*c
)
922 nir_builder
*b
= &c
->build
;
924 nir_builder_insert_after_cf_list(b
, c
->if_stack
[c
->if_stack_pos
- 1]);
928 ttn_endif(struct ttn_compile
*c
)
930 nir_builder
*b
= &c
->build
;
932 c
->if_stack_pos
-= 2;
933 nir_builder_insert_after_cf_list(b
, c
->if_stack
[c
->if_stack_pos
]);
937 ttn_bgnloop(struct ttn_compile
*c
)
939 nir_builder
*b
= &c
->build
;
941 /* Save the outside-of-the-loop node list. */
942 c
->loop_stack
[c
->loop_stack_pos
] = b
->cf_node_list
;
945 nir_loop
*loop
= nir_loop_create(b
->shader
);
946 nir_cf_node_insert_end(b
->cf_node_list
, &loop
->cf_node
);
948 nir_builder_insert_after_cf_list(b
, &loop
->body
);
952 ttn_cont(nir_builder
*b
)
954 nir_jump_instr
*instr
= nir_jump_instr_create(b
->shader
, nir_jump_continue
);
955 nir_instr_insert_after_cf_list(b
->cf_node_list
, &instr
->instr
);
959 ttn_brk(nir_builder
*b
)
961 nir_jump_instr
*instr
= nir_jump_instr_create(b
->shader
, nir_jump_break
);
962 nir_instr_insert_after_cf_list(b
->cf_node_list
, &instr
->instr
);
966 ttn_endloop(struct ttn_compile
*c
)
968 nir_builder
*b
= &c
->build
;
971 nir_builder_insert_after_cf_list(b
, c
->loop_stack
[c
->loop_stack_pos
]);
975 setup_texture_info(nir_tex_instr
*instr
, unsigned texture
)
978 case TGSI_TEXTURE_1D
:
979 instr
->sampler_dim
= GLSL_SAMPLER_DIM_1D
;
981 case TGSI_TEXTURE_1D_ARRAY
:
982 instr
->sampler_dim
= GLSL_SAMPLER_DIM_1D
;
983 instr
->is_array
= true;
985 case TGSI_TEXTURE_SHADOW1D
:
986 instr
->sampler_dim
= GLSL_SAMPLER_DIM_1D
;
987 instr
->is_shadow
= true;
989 case TGSI_TEXTURE_SHADOW1D_ARRAY
:
990 instr
->sampler_dim
= GLSL_SAMPLER_DIM_1D
;
991 instr
->is_shadow
= true;
992 instr
->is_array
= true;
994 case TGSI_TEXTURE_2D
:
995 instr
->sampler_dim
= GLSL_SAMPLER_DIM_2D
;
997 case TGSI_TEXTURE_2D_ARRAY
:
998 instr
->sampler_dim
= GLSL_SAMPLER_DIM_2D
;
999 instr
->is_array
= true;
1001 case TGSI_TEXTURE_2D_MSAA
:
1002 instr
->sampler_dim
= GLSL_SAMPLER_DIM_MS
;
1004 case TGSI_TEXTURE_2D_ARRAY_MSAA
:
1005 instr
->sampler_dim
= GLSL_SAMPLER_DIM_MS
;
1006 instr
->is_array
= true;
1008 case TGSI_TEXTURE_SHADOW2D
:
1009 instr
->sampler_dim
= GLSL_SAMPLER_DIM_2D
;
1010 instr
->is_shadow
= true;
1012 case TGSI_TEXTURE_SHADOW2D_ARRAY
:
1013 instr
->sampler_dim
= GLSL_SAMPLER_DIM_2D
;
1014 instr
->is_shadow
= true;
1015 instr
->is_array
= true;
1017 case TGSI_TEXTURE_3D
:
1018 instr
->sampler_dim
= GLSL_SAMPLER_DIM_3D
;
1020 case TGSI_TEXTURE_CUBE
:
1021 instr
->sampler_dim
= GLSL_SAMPLER_DIM_CUBE
;
1023 case TGSI_TEXTURE_CUBE_ARRAY
:
1024 instr
->sampler_dim
= GLSL_SAMPLER_DIM_CUBE
;
1025 instr
->is_array
= true;
1027 case TGSI_TEXTURE_SHADOWCUBE
:
1028 instr
->sampler_dim
= GLSL_SAMPLER_DIM_CUBE
;
1029 instr
->is_shadow
= true;
1031 case TGSI_TEXTURE_SHADOWCUBE_ARRAY
:
1032 instr
->sampler_dim
= GLSL_SAMPLER_DIM_CUBE
;
1033 instr
->is_shadow
= true;
1034 instr
->is_array
= true;
1036 case TGSI_TEXTURE_RECT
:
1037 instr
->sampler_dim
= GLSL_SAMPLER_DIM_RECT
;
1039 case TGSI_TEXTURE_SHADOWRECT
:
1040 instr
->sampler_dim
= GLSL_SAMPLER_DIM_RECT
;
1041 instr
->is_shadow
= true;
1044 fprintf(stderr
, "Unknown TGSI texture target %d\n", texture
);
1050 ttn_tex(struct ttn_compile
*c
, nir_alu_dest dest
, nir_ssa_def
**src
)
1052 nir_builder
*b
= &c
->build
;
1053 struct tgsi_full_instruction
*tgsi_inst
= &c
->token
->FullInstruction
;
1054 nir_tex_instr
*instr
;
1056 unsigned num_srcs
, samp
= 1, sview
, i
;
1058 switch (tgsi_inst
->Instruction
.Opcode
) {
1059 case TGSI_OPCODE_TEX
:
1063 case TGSI_OPCODE_TXP
:
1067 case TGSI_OPCODE_TXB
:
1071 case TGSI_OPCODE_TXB2
:
1076 case TGSI_OPCODE_TXL
:
1080 case TGSI_OPCODE_TXL2
:
1085 case TGSI_OPCODE_TXF
:
1086 if (tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_2D_MSAA
||
1087 tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_2D_ARRAY_MSAA
) {
1088 op
= nir_texop_txf_ms
;
1094 case TGSI_OPCODE_TXD
:
1101 fprintf(stderr
, "unknown TGSI tex op %d\n", tgsi_inst
->Instruction
.Opcode
);
1105 if (tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_SHADOW1D
||
1106 tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_SHADOW1D_ARRAY
||
1107 tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_SHADOW2D
||
1108 tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_SHADOW2D_ARRAY
||
1109 tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_SHADOWRECT
||
1110 tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_SHADOWCUBE
||
1111 tgsi_inst
->Texture
.Texture
== TGSI_TEXTURE_SHADOWCUBE_ARRAY
) {
1115 num_srcs
+= tgsi_inst
->Texture
.NumOffsets
;
1117 instr
= nir_tex_instr_create(b
->shader
, num_srcs
);
1120 setup_texture_info(instr
, tgsi_inst
->Texture
.Texture
);
1122 switch (instr
->sampler_dim
) {
1123 case GLSL_SAMPLER_DIM_1D
:
1124 case GLSL_SAMPLER_DIM_BUF
:
1125 instr
->coord_components
= 1;
1127 case GLSL_SAMPLER_DIM_2D
:
1128 case GLSL_SAMPLER_DIM_RECT
:
1129 case GLSL_SAMPLER_DIM_EXTERNAL
:
1130 case GLSL_SAMPLER_DIM_MS
:
1131 instr
->coord_components
= 2;
1133 case GLSL_SAMPLER_DIM_3D
:
1134 case GLSL_SAMPLER_DIM_CUBE
:
1135 instr
->coord_components
= 3;
1139 if (instr
->is_array
)
1140 instr
->coord_components
++;
1142 assert(tgsi_inst
->Src
[samp
].Register
.File
== TGSI_FILE_SAMPLER
);
1143 instr
->sampler_index
= tgsi_inst
->Src
[samp
].Register
.Index
;
1145 /* TODO if we supported any opc's which take an explicit SVIEW
1146 * src, we would use that here instead. But for the "legacy"
1147 * texture opc's the SVIEW index is same as SAMP index:
1149 sview
= instr
->sampler_index
;
1151 if (sview
< c
->num_samp_types
) {
1152 instr
->dest_type
= c
->samp_types
[sview
];
1154 instr
->dest_type
= nir_type_float
;
1157 unsigned src_number
= 0;
1159 instr
->src
[src_number
].src
=
1160 nir_src_for_ssa(nir_swizzle(b
, src
[0], SWIZ(X
, Y
, Z
, W
),
1161 instr
->coord_components
, false));
1162 instr
->src
[src_number
].src_type
= nir_tex_src_coord
;
1165 if (tgsi_inst
->Instruction
.Opcode
== TGSI_OPCODE_TXP
) {
1166 instr
->src
[src_number
].src
= nir_src_for_ssa(ttn_channel(b
, src
[0], W
));
1167 instr
->src
[src_number
].src_type
= nir_tex_src_projector
;
1171 if (tgsi_inst
->Instruction
.Opcode
== TGSI_OPCODE_TXB
) {
1172 instr
->src
[src_number
].src
= nir_src_for_ssa(ttn_channel(b
, src
[0], W
));
1173 instr
->src
[src_number
].src_type
= nir_tex_src_bias
;
1177 if (tgsi_inst
->Instruction
.Opcode
== TGSI_OPCODE_TXB2
) {
1178 instr
->src
[src_number
].src
= nir_src_for_ssa(ttn_channel(b
, src
[1], X
));
1179 instr
->src
[src_number
].src_type
= nir_tex_src_bias
;
1183 if (tgsi_inst
->Instruction
.Opcode
== TGSI_OPCODE_TXL
) {
1184 instr
->src
[src_number
].src
= nir_src_for_ssa(ttn_channel(b
, src
[0], W
));
1185 instr
->src
[src_number
].src_type
= nir_tex_src_lod
;
1189 if (tgsi_inst
->Instruction
.Opcode
== TGSI_OPCODE_TXL2
) {
1190 instr
->src
[src_number
].src
= nir_src_for_ssa(ttn_channel(b
, src
[1], X
));
1191 instr
->src
[src_number
].src_type
= nir_tex_src_lod
;
1195 if (tgsi_inst
->Instruction
.Opcode
== TGSI_OPCODE_TXF
) {
1196 instr
->src
[src_number
].src
= nir_src_for_ssa(ttn_channel(b
, src
[0], W
));
1197 if (op
== nir_texop_txf_ms
)
1198 instr
->src
[src_number
].src_type
= nir_tex_src_ms_index
;
1200 instr
->src
[src_number
].src_type
= nir_tex_src_lod
;
1204 if (tgsi_inst
->Instruction
.Opcode
== TGSI_OPCODE_TXD
) {
1205 instr
->src
[src_number
].src
=
1206 nir_src_for_ssa(nir_swizzle(b
, src
[1], SWIZ(X
, Y
, Z
, W
),
1207 instr
->coord_components
, false));
1208 instr
->src
[src_number
].src_type
= nir_tex_src_ddx
;
1210 instr
->src
[src_number
].src
=
1211 nir_src_for_ssa(nir_swizzle(b
, src
[2], SWIZ(X
, Y
, Z
, W
),
1212 instr
->coord_components
, false));
1213 instr
->src
[src_number
].src_type
= nir_tex_src_ddy
;
1217 if (instr
->is_shadow
) {
1218 if (instr
->coord_components
< 3)
1219 instr
->src
[src_number
].src
= nir_src_for_ssa(ttn_channel(b
, src
[0], Z
));
1221 instr
->src
[src_number
].src
= nir_src_for_ssa(ttn_channel(b
, src
[0], W
));
1223 instr
->src
[src_number
].src_type
= nir_tex_src_comparitor
;
1227 for (i
= 0; i
< tgsi_inst
->Texture
.NumOffsets
; i
++) {
1228 struct tgsi_texture_offset
*tex_offset
= &tgsi_inst
->TexOffsets
[i
];
1229 /* since TexOffset ins't using tgsi_full_src_register we get to
1230 * do some extra gymnastics:
1234 memset(&src
, 0, sizeof(src
));
1236 src
.src
= ttn_src_for_file_and_index(c
,
1241 src
.swizzle
[0] = tex_offset
->SwizzleX
;
1242 src
.swizzle
[1] = tex_offset
->SwizzleY
;
1243 src
.swizzle
[2] = tex_offset
->SwizzleZ
;
1244 src
.swizzle
[3] = TGSI_SWIZZLE_W
;
1246 instr
->src
[src_number
].src_type
= nir_tex_src_offset
;
1247 instr
->src
[src_number
].src
= nir_src_for_ssa(
1248 nir_fmov_alu(b
, src
, nir_tex_instr_src_size(instr
, src_number
)));
1252 assert(src_number
== num_srcs
);
1254 nir_ssa_dest_init(&instr
->instr
, &instr
->dest
, 4, NULL
);
1255 nir_instr_insert_after_cf_list(b
->cf_node_list
, &instr
->instr
);
1257 /* Resolve the writemask on the texture op. */
1258 ttn_move_dest(b
, dest
, &instr
->dest
.ssa
);
1261 /* TGSI_OPCODE_TXQ is actually two distinct operations:
1263 * dst.x = texture\_width(unit, lod)
1264 * dst.y = texture\_height(unit, lod)
1265 * dst.z = texture\_depth(unit, lod)
1266 * dst.w = texture\_levels(unit)
1268 * dst.xyz map to NIR txs opcode, and dst.w maps to query_levels
1271 ttn_txq(struct ttn_compile
*c
, nir_alu_dest dest
, nir_ssa_def
**src
)
1273 nir_builder
*b
= &c
->build
;
1274 struct tgsi_full_instruction
*tgsi_inst
= &c
->token
->FullInstruction
;
1275 nir_tex_instr
*txs
, *qlv
;
1277 txs
= nir_tex_instr_create(b
->shader
, 1);
1278 txs
->op
= nir_texop_txs
;
1279 setup_texture_info(txs
, tgsi_inst
->Texture
.Texture
);
1281 qlv
= nir_tex_instr_create(b
->shader
, 0);
1282 qlv
->op
= nir_texop_query_levels
;
1283 setup_texture_info(qlv
, tgsi_inst
->Texture
.Texture
);
1285 assert(tgsi_inst
->Src
[1].Register
.File
== TGSI_FILE_SAMPLER
);
1286 txs
->sampler_index
= tgsi_inst
->Src
[1].Register
.Index
;
1287 qlv
->sampler_index
= tgsi_inst
->Src
[1].Register
.Index
;
1289 /* only single src, the lod: */
1290 txs
->src
[0].src
= nir_src_for_ssa(ttn_channel(b
, src
[0], X
));
1291 txs
->src
[0].src_type
= nir_tex_src_lod
;
1293 nir_ssa_dest_init(&txs
->instr
, &txs
->dest
, 3, NULL
);
1294 nir_instr_insert_after_cf_list(b
->cf_node_list
, &txs
->instr
);
1296 nir_ssa_dest_init(&qlv
->instr
, &qlv
->dest
, 1, NULL
);
1297 nir_instr_insert_after_cf_list(b
->cf_node_list
, &qlv
->instr
);
1299 ttn_move_dest_masked(b
, dest
, &txs
->dest
.ssa
, TGSI_WRITEMASK_XYZ
);
1300 ttn_move_dest_masked(b
, dest
, &qlv
->dest
.ssa
, TGSI_WRITEMASK_W
);
1303 static const nir_op op_trans
[TGSI_OPCODE_LAST
] = {
1304 [TGSI_OPCODE_ARL
] = 0,
1305 [TGSI_OPCODE_MOV
] = nir_op_fmov
,
1306 [TGSI_OPCODE_LIT
] = 0,
1307 [TGSI_OPCODE_RCP
] = nir_op_frcp
,
1308 [TGSI_OPCODE_RSQ
] = nir_op_frsq
,
1309 [TGSI_OPCODE_EXP
] = 0,
1310 [TGSI_OPCODE_LOG
] = 0,
1311 [TGSI_OPCODE_MUL
] = nir_op_fmul
,
1312 [TGSI_OPCODE_ADD
] = nir_op_fadd
,
1313 [TGSI_OPCODE_DP3
] = 0,
1314 [TGSI_OPCODE_DP4
] = 0,
1315 [TGSI_OPCODE_DST
] = 0,
1316 [TGSI_OPCODE_MIN
] = nir_op_fmin
,
1317 [TGSI_OPCODE_MAX
] = nir_op_fmax
,
1318 [TGSI_OPCODE_SLT
] = nir_op_slt
,
1319 [TGSI_OPCODE_SGE
] = nir_op_sge
,
1320 [TGSI_OPCODE_MAD
] = nir_op_ffma
,
1321 [TGSI_OPCODE_SUB
] = nir_op_fsub
,
1322 [TGSI_OPCODE_LRP
] = 0,
1323 [TGSI_OPCODE_SQRT
] = nir_op_fsqrt
,
1324 [TGSI_OPCODE_DP2A
] = 0,
1325 [TGSI_OPCODE_FRC
] = nir_op_ffract
,
1326 [TGSI_OPCODE_CLAMP
] = 0,
1327 [TGSI_OPCODE_FLR
] = nir_op_ffloor
,
1328 [TGSI_OPCODE_ROUND
] = nir_op_fround_even
,
1329 [TGSI_OPCODE_EX2
] = nir_op_fexp2
,
1330 [TGSI_OPCODE_LG2
] = nir_op_flog2
,
1331 [TGSI_OPCODE_POW
] = nir_op_fpow
,
1332 [TGSI_OPCODE_XPD
] = 0,
1333 [TGSI_OPCODE_ABS
] = nir_op_fabs
,
1334 [TGSI_OPCODE_DPH
] = 0,
1335 [TGSI_OPCODE_COS
] = nir_op_fcos
,
1336 [TGSI_OPCODE_DDX
] = nir_op_fddx
,
1337 [TGSI_OPCODE_DDY
] = nir_op_fddy
,
1338 [TGSI_OPCODE_KILL
] = 0,
1339 [TGSI_OPCODE_PK2H
] = 0, /* XXX */
1340 [TGSI_OPCODE_PK2US
] = 0, /* XXX */
1341 [TGSI_OPCODE_PK4B
] = 0, /* XXX */
1342 [TGSI_OPCODE_PK4UB
] = 0, /* XXX */
1343 [TGSI_OPCODE_SEQ
] = nir_op_seq
,
1344 [TGSI_OPCODE_SGT
] = 0,
1345 [TGSI_OPCODE_SIN
] = nir_op_fsin
,
1346 [TGSI_OPCODE_SNE
] = nir_op_sne
,
1347 [TGSI_OPCODE_SLE
] = 0,
1348 [TGSI_OPCODE_TEX
] = 0,
1349 [TGSI_OPCODE_TXD
] = 0,
1350 [TGSI_OPCODE_TXP
] = 0,
1351 [TGSI_OPCODE_UP2H
] = 0, /* XXX */
1352 [TGSI_OPCODE_UP2US
] = 0, /* XXX */
1353 [TGSI_OPCODE_UP4B
] = 0, /* XXX */
1354 [TGSI_OPCODE_UP4UB
] = 0, /* XXX */
1355 [TGSI_OPCODE_ARR
] = 0,
1357 /* No function calls, yet. */
1358 [TGSI_OPCODE_CAL
] = 0, /* XXX */
1359 [TGSI_OPCODE_RET
] = 0, /* XXX */
1361 [TGSI_OPCODE_SSG
] = nir_op_fsign
,
1362 [TGSI_OPCODE_CMP
] = 0,
1363 [TGSI_OPCODE_SCS
] = 0,
1364 [TGSI_OPCODE_TXB
] = 0,
1365 [TGSI_OPCODE_DIV
] = nir_op_fdiv
,
1366 [TGSI_OPCODE_DP2
] = 0,
1367 [TGSI_OPCODE_DP2A
] = 0,
1368 [TGSI_OPCODE_TXL
] = 0,
1370 [TGSI_OPCODE_BRK
] = 0,
1371 [TGSI_OPCODE_IF
] = 0,
1372 [TGSI_OPCODE_UIF
] = 0,
1373 [TGSI_OPCODE_ELSE
] = 0,
1374 [TGSI_OPCODE_ENDIF
] = 0,
1376 [TGSI_OPCODE_DDX_FINE
] = nir_op_fddx_fine
,
1377 [TGSI_OPCODE_DDY_FINE
] = nir_op_fddy_fine
,
1379 [TGSI_OPCODE_PUSHA
] = 0, /* XXX */
1380 [TGSI_OPCODE_POPA
] = 0, /* XXX */
1382 [TGSI_OPCODE_CEIL
] = nir_op_fceil
,
1383 [TGSI_OPCODE_I2F
] = nir_op_i2f
,
1384 [TGSI_OPCODE_NOT
] = nir_op_inot
,
1385 [TGSI_OPCODE_TRUNC
] = nir_op_ftrunc
,
1386 [TGSI_OPCODE_SHL
] = nir_op_ishl
,
1387 [TGSI_OPCODE_AND
] = nir_op_iand
,
1388 [TGSI_OPCODE_OR
] = nir_op_ior
,
1389 [TGSI_OPCODE_MOD
] = nir_op_umod
,
1390 [TGSI_OPCODE_XOR
] = nir_op_ixor
,
1391 [TGSI_OPCODE_SAD
] = 0, /* XXX */
1392 [TGSI_OPCODE_TXF
] = 0,
1393 [TGSI_OPCODE_TXQ
] = 0,
1395 [TGSI_OPCODE_CONT
] = 0,
1397 [TGSI_OPCODE_EMIT
] = 0, /* XXX */
1398 [TGSI_OPCODE_ENDPRIM
] = 0, /* XXX */
1400 [TGSI_OPCODE_BGNLOOP
] = 0,
1401 [TGSI_OPCODE_BGNSUB
] = 0, /* XXX: no function calls */
1402 [TGSI_OPCODE_ENDLOOP
] = 0,
1403 [TGSI_OPCODE_ENDSUB
] = 0, /* XXX: no function calls */
1405 [TGSI_OPCODE_TXQ_LZ
] = 0,
1406 [TGSI_OPCODE_NOP
] = 0,
1407 [TGSI_OPCODE_FSEQ
] = nir_op_feq
,
1408 [TGSI_OPCODE_FSGE
] = nir_op_fge
,
1409 [TGSI_OPCODE_FSLT
] = nir_op_flt
,
1410 [TGSI_OPCODE_FSNE
] = nir_op_fne
,
1412 /* No control flow yet */
1413 [TGSI_OPCODE_CALLNZ
] = 0, /* XXX */
1414 [TGSI_OPCODE_BREAKC
] = 0, /* not emitted by glsl_to_tgsi.cpp */
1416 [TGSI_OPCODE_KILL_IF
] = 0,
1418 [TGSI_OPCODE_END
] = 0,
1420 [TGSI_OPCODE_F2I
] = nir_op_f2i
,
1421 [TGSI_OPCODE_IDIV
] = nir_op_idiv
,
1422 [TGSI_OPCODE_IMAX
] = nir_op_imax
,
1423 [TGSI_OPCODE_IMIN
] = nir_op_imin
,
1424 [TGSI_OPCODE_INEG
] = nir_op_ineg
,
1425 [TGSI_OPCODE_ISGE
] = nir_op_ige
,
1426 [TGSI_OPCODE_ISHR
] = nir_op_ishr
,
1427 [TGSI_OPCODE_ISLT
] = nir_op_ilt
,
1428 [TGSI_OPCODE_F2U
] = nir_op_f2u
,
1429 [TGSI_OPCODE_U2F
] = nir_op_u2f
,
1430 [TGSI_OPCODE_UADD
] = nir_op_iadd
,
1431 [TGSI_OPCODE_UDIV
] = nir_op_udiv
,
1432 [TGSI_OPCODE_UMAD
] = 0,
1433 [TGSI_OPCODE_UMAX
] = nir_op_umax
,
1434 [TGSI_OPCODE_UMIN
] = nir_op_umin
,
1435 [TGSI_OPCODE_UMOD
] = nir_op_umod
,
1436 [TGSI_OPCODE_UMUL
] = nir_op_imul
,
1437 [TGSI_OPCODE_USEQ
] = nir_op_ieq
,
1438 [TGSI_OPCODE_USGE
] = nir_op_uge
,
1439 [TGSI_OPCODE_USHR
] = nir_op_ushr
,
1440 [TGSI_OPCODE_USLT
] = nir_op_ult
,
1441 [TGSI_OPCODE_USNE
] = nir_op_ine
,
1443 [TGSI_OPCODE_SWITCH
] = 0, /* not emitted by glsl_to_tgsi.cpp */
1444 [TGSI_OPCODE_CASE
] = 0, /* not emitted by glsl_to_tgsi.cpp */
1445 [TGSI_OPCODE_DEFAULT
] = 0, /* not emitted by glsl_to_tgsi.cpp */
1446 [TGSI_OPCODE_ENDSWITCH
] = 0, /* not emitted by glsl_to_tgsi.cpp */
1448 /* XXX: SAMPLE opcodes */
1450 [TGSI_OPCODE_UARL
] = nir_op_imov
,
1451 [TGSI_OPCODE_UCMP
] = 0,
1452 [TGSI_OPCODE_IABS
] = nir_op_iabs
,
1453 [TGSI_OPCODE_ISSG
] = nir_op_isign
,
1457 [TGSI_OPCODE_TEX2
] = 0,
1458 [TGSI_OPCODE_TXB2
] = 0,
1459 [TGSI_OPCODE_TXL2
] = 0,
1461 [TGSI_OPCODE_IMUL_HI
] = nir_op_imul_high
,
1462 [TGSI_OPCODE_UMUL_HI
] = nir_op_umul_high
,
1464 [TGSI_OPCODE_TG4
] = 0,
1465 [TGSI_OPCODE_LODQ
] = 0, /* XXX */
1467 [TGSI_OPCODE_IBFE
] = nir_op_ibitfield_extract
,
1468 [TGSI_OPCODE_UBFE
] = nir_op_ubitfield_extract
,
1469 [TGSI_OPCODE_BFI
] = nir_op_bitfield_insert
,
1470 [TGSI_OPCODE_BREV
] = nir_op_bitfield_reverse
,
1471 [TGSI_OPCODE_POPC
] = nir_op_bit_count
,
1472 [TGSI_OPCODE_LSB
] = nir_op_find_lsb
,
1473 [TGSI_OPCODE_IMSB
] = nir_op_ifind_msb
,
1474 [TGSI_OPCODE_UMSB
] = nir_op_ifind_msb
, /* XXX: signed vs unsigned */
1476 [TGSI_OPCODE_INTERP_CENTROID
] = 0, /* XXX */
1477 [TGSI_OPCODE_INTERP_SAMPLE
] = 0, /* XXX */
1478 [TGSI_OPCODE_INTERP_OFFSET
] = 0, /* XXX */
1482 ttn_emit_instruction(struct ttn_compile
*c
)
1484 nir_builder
*b
= &c
->build
;
1485 struct tgsi_full_instruction
*tgsi_inst
= &c
->token
->FullInstruction
;
1487 unsigned tgsi_op
= tgsi_inst
->Instruction
.Opcode
;
1488 struct tgsi_full_dst_register
*tgsi_dst
= &tgsi_inst
->Dst
[0];
1490 if (tgsi_op
== TGSI_OPCODE_END
)
1493 nir_ssa_def
*src
[TGSI_FULL_MAX_SRC_REGISTERS
];
1494 for (i
= 0; i
< TGSI_FULL_MAX_SRC_REGISTERS
; i
++) {
1495 src
[i
] = ttn_get_src(c
, &tgsi_inst
->Src
[i
]);
1497 nir_alu_dest dest
= ttn_get_dest(c
, tgsi_dst
);
1500 case TGSI_OPCODE_RSQ
:
1501 ttn_move_dest(b
, dest
, nir_frsq(b
, ttn_channel(b
, src
[0], X
)));
1504 case TGSI_OPCODE_SQRT
:
1505 ttn_move_dest(b
, dest
, nir_fsqrt(b
, ttn_channel(b
, src
[0], X
)));
1508 case TGSI_OPCODE_RCP
:
1509 ttn_move_dest(b
, dest
, nir_frcp(b
, ttn_channel(b
, src
[0], X
)));
1512 case TGSI_OPCODE_EX2
:
1513 ttn_move_dest(b
, dest
, nir_fexp2(b
, ttn_channel(b
, src
[0], X
)));
1516 case TGSI_OPCODE_LG2
:
1517 ttn_move_dest(b
, dest
, nir_flog2(b
, ttn_channel(b
, src
[0], X
)));
1520 case TGSI_OPCODE_POW
:
1521 ttn_move_dest(b
, dest
, nir_fpow(b
,
1522 ttn_channel(b
, src
[0], X
),
1523 ttn_channel(b
, src
[1], X
)));
1526 case TGSI_OPCODE_COS
:
1527 ttn_move_dest(b
, dest
, nir_fcos(b
, ttn_channel(b
, src
[0], X
)));
1530 case TGSI_OPCODE_SIN
:
1531 ttn_move_dest(b
, dest
, nir_fsin(b
, ttn_channel(b
, src
[0], X
)));
1534 case TGSI_OPCODE_ARL
:
1535 ttn_arl(b
, op_trans
[tgsi_op
], dest
, src
);
1538 case TGSI_OPCODE_EXP
:
1539 ttn_exp(b
, op_trans
[tgsi_op
], dest
, src
);
1542 case TGSI_OPCODE_LOG
:
1543 ttn_log(b
, op_trans
[tgsi_op
], dest
, src
);
1546 case TGSI_OPCODE_DST
:
1547 ttn_dst(b
, op_trans
[tgsi_op
], dest
, src
);
1550 case TGSI_OPCODE_LIT
:
1551 ttn_lit(b
, op_trans
[tgsi_op
], dest
, src
);
1554 case TGSI_OPCODE_CLAMP
:
1555 ttn_clamp(b
, op_trans
[tgsi_op
], dest
, src
);
1558 case TGSI_OPCODE_XPD
:
1559 ttn_xpd(b
, op_trans
[tgsi_op
], dest
, src
);
1562 case TGSI_OPCODE_DP2
:
1563 ttn_dp2(b
, op_trans
[tgsi_op
], dest
, src
);
1566 case TGSI_OPCODE_DP3
:
1567 ttn_dp3(b
, op_trans
[tgsi_op
], dest
, src
);
1570 case TGSI_OPCODE_DP4
:
1571 ttn_dp4(b
, op_trans
[tgsi_op
], dest
, src
);
1574 case TGSI_OPCODE_DP2A
:
1575 ttn_dp2a(b
, op_trans
[tgsi_op
], dest
, src
);
1578 case TGSI_OPCODE_DPH
:
1579 ttn_dph(b
, op_trans
[tgsi_op
], dest
, src
);
1582 case TGSI_OPCODE_UMAD
:
1583 ttn_umad(b
, op_trans
[tgsi_op
], dest
, src
);
1586 case TGSI_OPCODE_LRP
:
1587 ttn_move_dest(b
, dest
, nir_flrp(b
, src
[2], src
[1], src
[0]));
1590 case TGSI_OPCODE_KILL
:
1591 ttn_kill(b
, op_trans
[tgsi_op
], dest
, src
);
1594 case TGSI_OPCODE_ARR
:
1595 ttn_arr(b
, op_trans
[tgsi_op
], dest
, src
);
1598 case TGSI_OPCODE_CMP
:
1599 ttn_cmp(b
, op_trans
[tgsi_op
], dest
, src
);
1602 case TGSI_OPCODE_UCMP
:
1603 ttn_ucmp(b
, op_trans
[tgsi_op
], dest
, src
);
1606 case TGSI_OPCODE_SCS
:
1607 ttn_scs(b
, op_trans
[tgsi_op
], dest
, src
);
1610 case TGSI_OPCODE_SGT
:
1611 ttn_sgt(b
, op_trans
[tgsi_op
], dest
, src
);
1614 case TGSI_OPCODE_SLE
:
1615 ttn_sle(b
, op_trans
[tgsi_op
], dest
, src
);
1618 case TGSI_OPCODE_KILL_IF
:
1619 ttn_kill_if(b
, op_trans
[tgsi_op
], dest
, src
);
1622 case TGSI_OPCODE_TEX
:
1623 case TGSI_OPCODE_TXP
:
1624 case TGSI_OPCODE_TXL
:
1625 case TGSI_OPCODE_TXB
:
1626 case TGSI_OPCODE_TXD
:
1627 case TGSI_OPCODE_TXL2
:
1628 case TGSI_OPCODE_TXB2
:
1629 case TGSI_OPCODE_TXQ_LZ
:
1630 case TGSI_OPCODE_TXF
:
1631 case TGSI_OPCODE_TG4
:
1632 ttn_tex(c
, dest
, src
);
1635 case TGSI_OPCODE_TXQ
:
1636 ttn_txq(c
, dest
, src
);
1639 case TGSI_OPCODE_NOP
:
1642 case TGSI_OPCODE_IF
:
1643 ttn_if(c
, src
[0], false);
1646 case TGSI_OPCODE_UIF
:
1647 ttn_if(c
, src
[0], true);
1650 case TGSI_OPCODE_ELSE
:
1654 case TGSI_OPCODE_ENDIF
:
1658 case TGSI_OPCODE_BGNLOOP
:
1662 case TGSI_OPCODE_BRK
:
1666 case TGSI_OPCODE_CONT
:
1670 case TGSI_OPCODE_ENDLOOP
:
1675 if (op_trans
[tgsi_op
] != 0 || tgsi_op
== TGSI_OPCODE_MOV
) {
1676 ttn_alu(b
, op_trans
[tgsi_op
], dest
, src
);
1678 fprintf(stderr
, "unknown TGSI opcode: %s\n",
1679 tgsi_get_opcode_name(tgsi_op
));
1685 if (tgsi_inst
->Instruction
.Saturate
) {
1686 assert(!dest
.dest
.is_ssa
);
1687 ttn_move_dest(b
, dest
, nir_fsat(b
, ttn_src_for_dest(b
, &dest
)));
1690 /* if the dst has a matching var, append store_global to move
1691 * output from reg to var
1693 nir_variable
*var
= ttn_get_var(c
, tgsi_dst
);
1695 unsigned index
= tgsi_dst
->Register
.Index
;
1696 unsigned offset
= c
->temp_regs
[index
].offset
;
1697 nir_intrinsic_instr
*store
=
1698 nir_intrinsic_instr_create(b
->shader
, nir_intrinsic_store_var
);
1699 struct tgsi_ind_register
*indirect
= tgsi_dst
->Register
.Indirect
?
1700 &tgsi_dst
->Indirect
: NULL
;
1702 store
->num_components
= 4;
1703 store
->variables
[0] = ttn_array_deref(c
, store
, var
, offset
, indirect
);
1704 store
->src
[0] = nir_src_for_reg(dest
.dest
.reg
.reg
);
1706 nir_instr_insert_after_cf_list(b
->cf_node_list
, &store
->instr
);
1711 * Puts a NIR intrinsic to store of each TGSI_FILE_OUTPUT value to the output
1712 * variables at the end of the shader.
1714 * We don't generate these incrementally as the TGSI_FILE_OUTPUT values are
1715 * written, because there's no output load intrinsic, which means we couldn't
1716 * handle writemasks.
1719 ttn_add_output_stores(struct ttn_compile
*c
)
1721 nir_builder
*b
= &c
->build
;
1723 foreach_list_typed(nir_variable
, var
, node
, &b
->shader
->outputs
) {
1724 unsigned array_len
= MAX2(glsl_get_length(var
->type
), 1);
1727 for (i
= 0; i
< array_len
; i
++) {
1728 nir_intrinsic_instr
*store
=
1729 nir_intrinsic_instr_create(b
->shader
, nir_intrinsic_store_output
);
1730 unsigned loc
= var
->data
.driver_location
+ i
;
1731 store
->num_components
= 4;
1732 store
->const_index
[0] = loc
;
1733 store
->src
[0].reg
.reg
= c
->output_regs
[loc
].reg
;
1734 store
->src
[0].reg
.base_offset
= c
->output_regs
[loc
].offset
;
1735 nir_instr_insert_after_cf_list(b
->cf_node_list
, &store
->instr
);
1741 tgsi_to_nir(const void *tgsi_tokens
,
1742 const nir_shader_compiler_options
*options
)
1744 struct tgsi_parse_context parser
;
1745 struct tgsi_shader_info scan
;
1746 struct ttn_compile
*c
;
1747 struct nir_shader
*s
;
1750 c
= rzalloc(NULL
, struct ttn_compile
);
1751 s
= nir_shader_create(NULL
, options
);
1753 nir_function
*func
= nir_function_create(s
, "main");
1754 nir_function_overload
*overload
= nir_function_overload_create(func
);
1755 nir_function_impl
*impl
= nir_function_impl_create(overload
);
1757 nir_builder_init(&c
->build
, impl
);
1758 nir_builder_insert_after_cf_list(&c
->build
, &impl
->body
);
1760 tgsi_scan_shader(tgsi_tokens
, &scan
);
1763 s
->num_inputs
= scan
.file_max
[TGSI_FILE_INPUT
] + 1;
1764 s
->num_uniforms
= scan
.const_file_max
[0] + 1;
1765 s
->num_outputs
= scan
.file_max
[TGSI_FILE_OUTPUT
] + 1;
1767 c
->output_regs
= rzalloc_array(c
, struct ttn_reg_info
,
1768 scan
.file_max
[TGSI_FILE_OUTPUT
] + 1);
1769 c
->temp_regs
= rzalloc_array(c
, struct ttn_reg_info
,
1770 scan
.file_max
[TGSI_FILE_TEMPORARY
] + 1);
1771 c
->imm_defs
= rzalloc_array(c
, nir_ssa_def
*,
1772 scan
.file_max
[TGSI_FILE_IMMEDIATE
] + 1);
1774 c
->num_samp_types
= scan
.file_max
[TGSI_FILE_SAMPLER_VIEW
] + 1;
1775 c
->samp_types
= rzalloc_array(c
, nir_alu_type
, c
->num_samp_types
);
1777 c
->if_stack
= rzalloc_array(c
, struct exec_list
*,
1778 (scan
.opcode_count
[TGSI_OPCODE_IF
] +
1779 scan
.opcode_count
[TGSI_OPCODE_UIF
]) * 2);
1780 c
->loop_stack
= rzalloc_array(c
, struct exec_list
*,
1781 scan
.opcode_count
[TGSI_OPCODE_BGNLOOP
]);
1783 ret
= tgsi_parse_init(&parser
, tgsi_tokens
);
1784 assert(ret
== TGSI_PARSE_OK
);
1786 while (!tgsi_parse_end_of_tokens(&parser
)) {
1787 tgsi_parse_token(&parser
);
1788 c
->token
= &parser
.FullToken
;
1790 switch (parser
.FullToken
.Token
.Type
) {
1791 case TGSI_TOKEN_TYPE_DECLARATION
:
1792 ttn_emit_declaration(c
);
1795 case TGSI_TOKEN_TYPE_INSTRUCTION
:
1796 ttn_emit_instruction(c
);
1799 case TGSI_TOKEN_TYPE_IMMEDIATE
:
1800 ttn_emit_immediate(c
);
1805 tgsi_parse_free(&parser
);
1807 ttn_add_output_stores(c
);