1 /*
2 * Copyright © 2017-2018 Rob Clark <robclark@freedesktop.org>
3 * SPDX-License-Identifier: MIT
4 *
5 * Authors:
6 * Rob Clark <robclark@freedesktop.org>
7 */
8
9 #include "ir3_image.h"
10
11 /*
12 * SSBO/Image to/from IBO/tex hw mapping table:
13 */
14
15 void
ir3_ibo_mapping_init(struct ir3_ibo_mapping * mapping,unsigned num_textures)16 ir3_ibo_mapping_init(struct ir3_ibo_mapping *mapping, unsigned num_textures)
17 {
18 memset(mapping, IBO_INVALID, sizeof(*mapping));
19 mapping->num_tex = 0;
20 mapping->tex_base = num_textures;
21 }
22
23 struct ir3_instruction *
ir3_ssbo_to_ibo(struct ir3_context * ctx,nir_src src)24 ir3_ssbo_to_ibo(struct ir3_context *ctx, nir_src src)
25 {
26 if (ir3_bindless_resource(src))
27 ctx->so->bindless_ibo = true;
28 return ir3_get_src(ctx, &src)[0];
29 }
30
31 unsigned
ir3_ssbo_to_tex(struct ir3_ibo_mapping * mapping,unsigned ssbo)32 ir3_ssbo_to_tex(struct ir3_ibo_mapping *mapping, unsigned ssbo)
33 {
34 if (mapping->ssbo_to_tex[ssbo] == IBO_INVALID) {
35 unsigned tex = mapping->num_tex++;
36 mapping->ssbo_to_tex[ssbo] = tex;
37 mapping->tex_to_image[tex] = IBO_SSBO | ssbo;
38 }
39 return mapping->ssbo_to_tex[ssbo] + mapping->tex_base;
40 }
41
42 struct ir3_instruction *
ir3_image_to_ibo(struct ir3_context * ctx,nir_src src)43 ir3_image_to_ibo(struct ir3_context *ctx, nir_src src)
44 {
45 if (ir3_bindless_resource(src)) {
46 ctx->so->bindless_ibo = true;
47 return ir3_get_src(ctx, &src)[0];
48 }
49
50 if (nir_src_is_const(src)) {
51 int image_idx = nir_src_as_uint(src);
52 return create_immed(&ctx->build, ctx->s->info.num_ssbos + image_idx);
53 } else {
54 struct ir3_instruction *image_idx = ir3_get_src(ctx, &src)[0];
55 if (ctx->s->info.num_ssbos) {
56 return ir3_ADD_U(&ctx->build, image_idx, 0,
57 create_immed(&ctx->build, ctx->s->info.num_ssbos), 0);
58 } else {
59 return image_idx;
60 }
61 }
62 }
63
64 unsigned
ir3_image_to_tex(struct ir3_ibo_mapping * mapping,unsigned image)65 ir3_image_to_tex(struct ir3_ibo_mapping *mapping, unsigned image)
66 {
67 if (mapping->image_to_tex[image] == IBO_INVALID) {
68 unsigned tex = mapping->num_tex++;
69 mapping->image_to_tex[image] = tex;
70 mapping->tex_to_image[tex] = image;
71 }
72 return mapping->image_to_tex[image] + mapping->tex_base;
73 }
74
75 /* see tex_info() for equiv logic for texture instructions.. it would be
76 * nice if this could be better unified..
77 */
78 unsigned
ir3_get_image_coords(const nir_intrinsic_instr * instr,unsigned * flagsp)79 ir3_get_image_coords(const nir_intrinsic_instr *instr, unsigned *flagsp)
80 {
81 enum glsl_sampler_dim dim = nir_intrinsic_image_dim(instr);
82 unsigned coords = nir_image_intrinsic_coord_components(instr);
83 unsigned flags = 0;
84
85 if (dim == GLSL_SAMPLER_DIM_CUBE || nir_intrinsic_image_array(instr))
86 flags |= IR3_INSTR_A;
87 else if (dim == GLSL_SAMPLER_DIM_3D)
88 flags |= IR3_INSTR_3D;
89
90 if (flagsp)
91 *flagsp = flags;
92
93 return coords;
94 }
95
96 type_t
ir3_get_type_for_image_intrinsic(const nir_intrinsic_instr * instr)97 ir3_get_type_for_image_intrinsic(const nir_intrinsic_instr *instr)
98 {
99 const nir_intrinsic_info *info = &nir_intrinsic_infos[instr->intrinsic];
100 int bit_size = info->has_dest ? instr->def.bit_size : nir_src_bit_size(instr->src[3]);
101
102 nir_alu_type type = nir_type_uint;
103 switch (instr->intrinsic) {
104 case nir_intrinsic_image_load:
105 case nir_intrinsic_bindless_image_load:
106 type = nir_alu_type_get_base_type(nir_intrinsic_dest_type(instr));
107 /* SpvOpAtomicLoad doesn't have dest type */
108 if (type == nir_type_invalid)
109 type = nir_type_uint;
110 break;
111
112 case nir_intrinsic_image_store:
113 case nir_intrinsic_bindless_image_store:
114 type = nir_alu_type_get_base_type(nir_intrinsic_src_type(instr));
115 /* SpvOpAtomicStore doesn't have src type */
116 if (type == nir_type_invalid)
117 type = nir_type_uint;
118 break;
119
120 case nir_intrinsic_image_atomic:
121 case nir_intrinsic_bindless_image_atomic:
122 case nir_intrinsic_image_atomic_swap:
123 case nir_intrinsic_bindless_image_atomic_swap:
124 type = nir_atomic_op_type(nir_intrinsic_atomic_op(instr));
125 break;
126
127 default:
128 unreachable("Unhandled NIR image intrinsic");
129 }
130
131 switch (type) {
132 case nir_type_uint:
133 return bit_size == 16 ? TYPE_U16 : TYPE_U32;
134 case nir_type_int:
135 return bit_size == 16 ? TYPE_S16 : TYPE_S32;
136 case nir_type_float:
137 return bit_size == 16 ? TYPE_F16 : TYPE_F32;
138 default:
139 unreachable("bad type");
140 }
141 }
142
143 /* Returns the number of components for the different image formats
144 * supported by the GLES 3.1 spec, plus those added by the
145 * GL_NV_image_formats extension.
146 */
147 unsigned
ir3_get_num_components_for_image_format(enum pipe_format format)148 ir3_get_num_components_for_image_format(enum pipe_format format)
149 {
150 if (format == PIPE_FORMAT_NONE)
151 return 4;
152 else
153 return util_format_get_nr_components(format);
154 }
155