2 * Copyright © 2016 Intel Corporation
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:
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
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
25 #include "nir/nir_builder.h"
28 load_frag_coord(nir_builder
*b
)
30 nir_foreach_variable(var
, &b
->shader
->inputs
) {
31 if (var
->data
.location
== VARYING_SLOT_POS
)
32 return nir_load_var(b
, var
);
35 nir_variable
*pos
= nir_variable_create(b
->shader
, nir_var_shader_in
,
36 glsl_vec4_type(), NULL
);
37 pos
->data
.location
= VARYING_SLOT_POS
;
38 pos
->data
.origin_upper_left
= true;
40 return nir_load_var(b
, pos
);
44 try_lower_input_load(nir_function_impl
*impl
, nir_intrinsic_instr
*load
)
47 const struct glsl_type
*image_type
=
48 glsl_without_array(load
->variables
[0]->var
->type
);
49 enum glsl_sampler_dim image_dim
= glsl_get_sampler_dim(image_type
);
50 if (image_dim
!= GLSL_SAMPLER_DIM_SUBPASS
&&
51 image_dim
!= GLSL_SAMPLER_DIM_SUBPASS_MS
)
54 const bool multisampled
= (image_dim
== GLSL_SAMPLER_DIM_SUBPASS_MS
);
57 nir_builder_init(&b
, impl
);
58 b
.cursor
= nir_before_instr(&load
->instr
);
60 nir_ssa_def
*frag_coord
= nir_f2i(&b
, load_frag_coord(&b
));
61 nir_ssa_def
*offset
= nir_ssa_for_src(&b
, load
->src
[0], 2);
62 nir_ssa_def
*pos
= nir_iadd(&b
, frag_coord
, offset
);
65 nir_load_system_value(&b
, nir_intrinsic_load_layer_id
, 0);
67 nir_vec3(&b
, nir_channel(&b
, pos
, 0), nir_channel(&b
, pos
, 1), layer
);
69 nir_tex_instr
*tex
= nir_tex_instr_create(b
.shader
, 2 + multisampled
);
71 tex
->op
= nir_texop_txf
;
73 switch (glsl_get_sampler_result_type(image_type
)) {
75 tex
->dest_type
= nir_type_float
;
78 tex
->dest_type
= nir_type_int
;
81 tex
->dest_type
= nir_type_uint
;
84 unreachable("Invalid image type");
87 tex
->is_shadow
= false;
90 nir_deref_as_var(nir_copy_deref(tex
, &load
->variables
[0]->deref
));
92 tex
->texture_index
= 0;
93 tex
->sampler_index
= 0;
95 tex
->src
[0].src_type
= nir_tex_src_coord
;
96 tex
->src
[0].src
= nir_src_for_ssa(coord
);
97 tex
->coord_components
= 3;
99 tex
->src
[1].src_type
= nir_tex_src_lod
;
100 tex
->src
[1].src
= nir_src_for_ssa(nir_imm_int(&b
, 0));
102 if (image_dim
== GLSL_SAMPLER_DIM_SUBPASS_MS
) {
103 tex
->op
= nir_texop_txf_ms
;
105 nir_ssa_def
*sample_id
=
106 nir_load_system_value(&b
, nir_intrinsic_load_sample_id
, 0);
107 tex
->src
[2].src_type
= nir_tex_src_ms_index
;
108 tex
->src
[2].src
= nir_src_for_ssa(sample_id
);
111 nir_ssa_dest_init(&tex
->instr
, &tex
->dest
, 4, 32, NULL
);
112 nir_builder_instr_insert(&b
, &tex
->instr
);
114 nir_ssa_def_rewrite_uses(&load
->dest
.ssa
,
115 nir_src_for_ssa(&tex
->dest
.ssa
));
119 anv_nir_lower_input_attachments(nir_shader
*shader
)
121 assert(shader
->stage
== MESA_SHADER_FRAGMENT
);
123 nir_foreach_function(function
, shader
) {
127 nir_foreach_block(block
, function
->impl
) {
128 nir_foreach_instr(instr
, block
) {
129 if (instr
->type
!= nir_instr_type_intrinsic
)
132 nir_intrinsic_instr
*load
= nir_instr_as_intrinsic(instr
);
134 if (load
->intrinsic
!= nir_intrinsic_image_load
)
137 try_lower_input_load(function
->impl
, load
);