blob: fd614dde3881f8265bcf32058f2e158c1c922042 [file] [log] [blame]
Tom Stellarda75c6162012-01-06 17:38:37 -05001
Christian Königce40e472012-08-02 12:14:59 +02002/*
3 * Copyright 2012 Advanced Micro Devices, Inc.
4 *
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 * on the rights to use, copy, modify, merge, publish, distribute, sub
9 * license, and/or sell copies of the Software, and to permit persons to whom
10 * the Software is furnished to do so, subject to the following conditions:
11 *
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
14 * Software.
15 *
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 NON-INFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
20 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
21 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
22 * USE OR OTHER DEALINGS IN THE SOFTWARE.
23 *
24 * Authors:
25 * Tom Stellard <thomas.stellard@amd.com>
26 * Michel Dänzer <michel.daenzer@amd.com>
27 * Christian König <christian.koenig@amd.com>
28 */
29
Tom Stellarda75c6162012-01-06 17:38:37 -050030#include "gallivm/lp_bld_tgsi_action.h"
31#include "gallivm/lp_bld_const.h"
Michel Dänzerc2bae6b2012-08-02 17:19:22 +020032#include "gallivm/lp_bld_gather.h"
Tom Stellarda75c6162012-01-06 17:38:37 -050033#include "gallivm/lp_bld_intr.h"
34#include "gallivm/lp_bld_tgsi.h"
35#include "radeon_llvm.h"
Tom Stellard509ddb02012-04-16 17:48:44 -040036#include "radeon_llvm_emit.h"
Tom Stellarda75c6162012-01-06 17:38:37 -050037#include "tgsi/tgsi_info.h"
38#include "tgsi/tgsi_parse.h"
39#include "tgsi/tgsi_scan.h"
40#include "tgsi/tgsi_dump.h"
41
42#include "radeonsi_pipe.h"
43#include "radeonsi_shader.h"
Christian Königf67fae02012-07-17 23:43:00 +020044#include "si_state.h"
Tom Stellarda75c6162012-01-06 17:38:37 -050045#include "sid.h"
46
47#include <assert.h>
48#include <errno.h>
49#include <stdio.h>
50
51/*
52static ps_remap_inputs(
53 struct tgsi_llvm_context * tl_ctx,
54 unsigned tgsi_index,
55 unsigned tgsi_chan)
56{
57 :
58}
59
60struct si_input
61{
62 struct list_head head;
63 unsigned tgsi_index;
64 unsigned tgsi_chan;
65 unsigned order;
66};
67*/
68
69
70struct si_shader_context
71{
72 struct radeon_llvm_context radeon_bld;
73 struct r600_context *rctx;
74 struct tgsi_parse_context parse;
75 struct tgsi_token * tokens;
76 struct si_pipe_shader *shader;
77 unsigned type; /* TGSI_PROCESSOR_* specifies the type of shader. */
78/* unsigned num_inputs; */
79/* struct list_head inputs; */
80/* unsigned * input_mappings *//* From TGSI to SI hw */
81/* struct tgsi_shader_info info;*/
82};
83
84static struct si_shader_context * si_shader_context(
85 struct lp_build_tgsi_context * bld_base)
86{
87 return (struct si_shader_context *)bld_base;
88}
89
90
91#define PERSPECTIVE_BASE 0
92#define LINEAR_BASE 9
93
94#define SAMPLE_OFFSET 0
95#define CENTER_OFFSET 2
96#define CENTROID_OFSET 4
97
98#define USE_SGPR_MAX_SUFFIX_LEN 5
Tom Stellard467f5162012-05-16 15:15:35 -040099#define CONST_ADDR_SPACE 2
Tom Stellard89ece082012-05-29 11:36:29 -0400100#define USER_SGPR_ADDR_SPACE 8
Tom Stellarda75c6162012-01-06 17:38:37 -0500101
102enum sgpr_type {
Tom Stellard467f5162012-05-16 15:15:35 -0400103 SGPR_CONST_PTR_F32,
104 SGPR_CONST_PTR_V4I32,
105 SGPR_CONST_PTR_V8I32,
Tom Stellarda75c6162012-01-06 17:38:37 -0500106 SGPR_I32,
Tom Stellard467f5162012-05-16 15:15:35 -0400107 SGPR_I64
Tom Stellarda75c6162012-01-06 17:38:37 -0500108};
109
Tom Stellard467f5162012-05-16 15:15:35 -0400110/**
111 * Build an LLVM bytecode indexed load using LLVMBuildGEP + LLVMBuildLoad
112 *
113 * @param offset The offset parameter specifies the number of
114 * elements to offset, not the number of bytes or dwords. An element is the
115 * the type pointed to by the base_ptr parameter (e.g. int is the element of
116 * an int* pointer)
117 *
118 * When LLVM lowers the load instruction, it will convert the element offset
119 * into a dword offset automatically.
120 *
121 */
122static LLVMValueRef build_indexed_load(
123 struct gallivm_state * gallivm,
124 LLVMValueRef base_ptr,
125 LLVMValueRef offset)
126{
127 LLVMValueRef computed_ptr = LLVMBuildGEP(
128 gallivm->builder, base_ptr, &offset, 1, "");
129
130 return LLVMBuildLoad(gallivm->builder, computed_ptr, "");
131}
132
Tom Stellard89ece082012-05-29 11:36:29 -0400133/**
134 * Load a value stored in one of the user SGPRs
135 *
136 * @param sgpr This is the sgpr to load the value from. If you need to load a
137 * value that is stored in consecutive SGPR registers (e.g. a 64-bit pointer),
138 * then you should pass the index of the first SGPR that holds the value. For
139 * example, if you want to load a pointer that is stored in SGPRs 2 and 3, then
140 * use pass 2 for the sgpr parameter.
141 *
142 * The value of the sgpr parameter must also be aligned to the width of the type
143 * being loaded, so that the sgpr parameter is divisible by the dword width of the
144 * type. For example, if the value being loaded is two dwords wide, then the sgpr
145 * parameter must be divisible by two.
Tom Stellard467f5162012-05-16 15:15:35 -0400146 */
Tom Stellarda75c6162012-01-06 17:38:37 -0500147static LLVMValueRef use_sgpr(
148 struct gallivm_state * gallivm,
149 enum sgpr_type type,
150 unsigned sgpr)
151{
152 LLVMValueRef sgpr_index;
Tom Stellarda75c6162012-01-06 17:38:37 -0500153 LLVMTypeRef ret_type;
Tom Stellard89ece082012-05-29 11:36:29 -0400154 LLVMValueRef ptr;
Tom Stellarda75c6162012-01-06 17:38:37 -0500155
156 sgpr_index = lp_build_const_int32(gallivm, sgpr);
157
Tom Stellard467f5162012-05-16 15:15:35 -0400158 switch (type) {
159 case SGPR_CONST_PTR_F32:
Tom Stellard89ece082012-05-29 11:36:29 -0400160 assert(sgpr % 2 == 0);
Tom Stellard467f5162012-05-16 15:15:35 -0400161 ret_type = LLVMFloatTypeInContext(gallivm->context);
162 ret_type = LLVMPointerType(ret_type, CONST_ADDR_SPACE);
Tom Stellard89ece082012-05-29 11:36:29 -0400163 break;
164
Tom Stellard467f5162012-05-16 15:15:35 -0400165 case SGPR_I32:
Tom Stellarda75c6162012-01-06 17:38:37 -0500166 ret_type = LLVMInt32TypeInContext(gallivm->context);
Tom Stellard89ece082012-05-29 11:36:29 -0400167 break;
168
Tom Stellard467f5162012-05-16 15:15:35 -0400169 case SGPR_I64:
Tom Stellard89ece082012-05-29 11:36:29 -0400170 assert(sgpr % 2 == 0);
Tom Stellard467f5162012-05-16 15:15:35 -0400171 ret_type= LLVMInt64TypeInContext(gallivm->context);
Tom Stellard89ece082012-05-29 11:36:29 -0400172 break;
173
Tom Stellard467f5162012-05-16 15:15:35 -0400174 case SGPR_CONST_PTR_V4I32:
Tom Stellard89ece082012-05-29 11:36:29 -0400175 assert(sgpr % 2 == 0);
Tom Stellarda75c6162012-01-06 17:38:37 -0500176 ret_type = LLVMInt32TypeInContext(gallivm->context);
177 ret_type = LLVMVectorType(ret_type, 4);
Tom Stellard467f5162012-05-16 15:15:35 -0400178 ret_type = LLVMPointerType(ret_type, CONST_ADDR_SPACE);
Tom Stellard89ece082012-05-29 11:36:29 -0400179 break;
180
Tom Stellard467f5162012-05-16 15:15:35 -0400181 case SGPR_CONST_PTR_V8I32:
Tom Stellard89ece082012-05-29 11:36:29 -0400182 assert(sgpr % 2 == 0);
Tom Stellarda75c6162012-01-06 17:38:37 -0500183 ret_type = LLVMInt32TypeInContext(gallivm->context);
184 ret_type = LLVMVectorType(ret_type, 8);
Tom Stellard467f5162012-05-16 15:15:35 -0400185 ret_type = LLVMPointerType(ret_type, CONST_ADDR_SPACE);
Tom Stellard89ece082012-05-29 11:36:29 -0400186 break;
187
Tom Stellarda75c6162012-01-06 17:38:37 -0500188 default:
189 assert(!"Unsupported SGPR type in use_sgpr()");
190 return NULL;
191 }
Tom Stellard89ece082012-05-29 11:36:29 -0400192
193 ret_type = LLVMPointerType(ret_type, USER_SGPR_ADDR_SPACE);
194 ptr = LLVMBuildIntToPtr(gallivm->builder, sgpr_index, ret_type, "");
195 return LLVMBuildLoad(gallivm->builder, ptr, "");
Tom Stellarda75c6162012-01-06 17:38:37 -0500196}
197
198static void declare_input_vs(
199 struct si_shader_context * si_shader_ctx,
200 unsigned input_index,
201 const struct tgsi_full_declaration *decl)
202{
203 LLVMValueRef t_list_ptr;
204 LLVMValueRef t_offset;
Tom Stellard467f5162012-05-16 15:15:35 -0400205 LLVMValueRef t_list;
Tom Stellarda75c6162012-01-06 17:38:37 -0500206 LLVMValueRef attribute_offset;
207 LLVMValueRef buffer_index_reg;
Tom Stellard467f5162012-05-16 15:15:35 -0400208 LLVMValueRef args[3];
Tom Stellarda75c6162012-01-06 17:38:37 -0500209 LLVMTypeRef vec4_type;
210 LLVMValueRef input;
211 struct lp_build_context * uint = &si_shader_ctx->radeon_bld.soa.bld_base.uint_bld;
212 struct lp_build_context * base = &si_shader_ctx->radeon_bld.soa.bld_base.base;
213 struct r600_context *rctx = si_shader_ctx->rctx;
Christian Königb15e3ae2012-07-25 11:22:59 +0200214 //struct pipe_vertex_element *velem = &rctx->vertex_elements->elements[input_index];
Tom Stellarda75c6162012-01-06 17:38:37 -0500215 unsigned chan;
216
Tom Stellard467f5162012-05-16 15:15:35 -0400217 /* Load the T list */
Tom Stellarda75c6162012-01-06 17:38:37 -0500218 /* XXX: Communicate with the rest of the driver about which SGPR the T#
Michel Dänzer9918fbd2012-04-19 11:41:48 +0200219 * list pointer is going to be stored in. Hard code to SGPR[6:7] for
Tom Stellarda75c6162012-01-06 17:38:37 -0500220 * now */
Tom Stellard89ece082012-05-29 11:36:29 -0400221 t_list_ptr = use_sgpr(base->gallivm, SGPR_CONST_PTR_V4I32, 6);
Tom Stellarda75c6162012-01-06 17:38:37 -0500222
Christian Königb15e3ae2012-07-25 11:22:59 +0200223 t_offset = lp_build_const_int32(base->gallivm, input_index);
Tom Stellard467f5162012-05-16 15:15:35 -0400224
225 t_list = build_indexed_load(base->gallivm, t_list_ptr, t_offset);
226
227 /* Build the attribute offset */
Christian Königb15e3ae2012-07-25 11:22:59 +0200228 attribute_offset = lp_build_const_int32(base->gallivm, 0);
Tom Stellarda75c6162012-01-06 17:38:37 -0500229
230 /* Load the buffer index is always, which is always stored in VGPR0
231 * for Vertex Shaders */
Tom Stellard40c41fe2012-07-25 19:17:16 +0000232 buffer_index_reg = build_intrinsic(base->gallivm->builder,
233 "llvm.SI.vs.load.buffer.index", uint->elem_type, NULL, 0,
234 LLVMReadNoneAttribute);
Tom Stellarda75c6162012-01-06 17:38:37 -0500235
236 vec4_type = LLVMVectorType(base->elem_type, 4);
Tom Stellard467f5162012-05-16 15:15:35 -0400237 args[0] = t_list;
238 args[1] = attribute_offset;
239 args[2] = buffer_index_reg;
Tom Stellarda75c6162012-01-06 17:38:37 -0500240 input = lp_build_intrinsic(base->gallivm->builder,
Tom Stellard467f5162012-05-16 15:15:35 -0400241 "llvm.SI.vs.load.input", vec4_type, args, 3);
Tom Stellarda75c6162012-01-06 17:38:37 -0500242
243 /* Break up the vec4 into individual components */
244 for (chan = 0; chan < 4; chan++) {
245 LLVMValueRef llvm_chan = lp_build_const_int32(base->gallivm, chan);
246 /* XXX: Use a helper function for this. There is one in
247 * tgsi_llvm.c. */
248 si_shader_ctx->radeon_bld.inputs[radeon_llvm_reg_index_soa(input_index, chan)] =
249 LLVMBuildExtractElement(base->gallivm->builder,
250 input, llvm_chan, "");
251 }
252}
253
254static void declare_input_fs(
255 struct si_shader_context * si_shader_ctx,
256 unsigned input_index,
257 const struct tgsi_full_declaration *decl)
258{
259 const char * intr_name;
260 unsigned chan;
261 struct lp_build_context * base =
262 &si_shader_ctx->radeon_bld.soa.bld_base.base;
263 struct gallivm_state * gallivm = base->gallivm;
264
265 /* This value is:
266 * [15:0] NewPrimMask (Bit mask for each quad. It is set it the
267 * quad begins a new primitive. Bit 0 always needs
268 * to be unset)
269 * [32:16] ParamOffset
270 *
271 */
Michel Dänzer9918fbd2012-04-19 11:41:48 +0200272 /* XXX: This register number must be identical to the S_00B02C_USER_SGPR
273 * register field value
274 */
Tom Stellarda75c6162012-01-06 17:38:37 -0500275 LLVMValueRef params = use_sgpr(base->gallivm, SGPR_I32, 6);
276
277
278 /* XXX: Is this the input_index? */
279 LLVMValueRef attr_number = lp_build_const_int32(gallivm, input_index);
280
281 /* XXX: Handle all possible interpolation modes */
Francisco Jerez12799232012-04-30 18:27:52 +0200282 switch (decl->Interp.Interpolate) {
Tom Stellarda75c6162012-01-06 17:38:37 -0500283 case TGSI_INTERPOLATE_COLOR:
Michel Dänzer90c6eac2012-06-08 17:15:21 +0200284 /* XXX: Flat shading hangs the GPU */
Michel Dänzerd1e40b32012-08-23 17:10:37 +0200285 if (si_shader_ctx->rctx->queued.named.rasterizer &&
286 si_shader_ctx->rctx->queued.named.rasterizer->flatshade) {
Michel Dänzer90c6eac2012-06-08 17:15:21 +0200287#if 0
Tom Stellarda75c6162012-01-06 17:38:37 -0500288 intr_name = "llvm.SI.fs.interp.constant";
Michel Dänzer90c6eac2012-06-08 17:15:21 +0200289#else
290 intr_name = "llvm.SI.fs.interp.linear.center";
291#endif
Michel Dänzer1deb2be2012-05-14 16:26:19 +0200292 } else {
293 if (decl->Interp.Centroid)
294 intr_name = "llvm.SI.fs.interp.persp.centroid";
295 else
296 intr_name = "llvm.SI.fs.interp.persp.center";
297 }
Tom Stellarda75c6162012-01-06 17:38:37 -0500298 break;
299 case TGSI_INTERPOLATE_CONSTANT:
Michel Dänzer90c6eac2012-06-08 17:15:21 +0200300 /* XXX: Flat shading hangs the GPU */
301#if 0
Tom Stellarda75c6162012-01-06 17:38:37 -0500302 intr_name = "llvm.SI.fs.interp.constant";
303 break;
Michel Dänzer90c6eac2012-06-08 17:15:21 +0200304#endif
Tom Stellarda75c6162012-01-06 17:38:37 -0500305 case TGSI_INTERPOLATE_LINEAR:
Michel Dänzer1deb2be2012-05-14 16:26:19 +0200306 if (decl->Interp.Centroid)
307 intr_name = "llvm.SI.fs.interp.linear.centroid";
308 else
309 intr_name = "llvm.SI.fs.interp.linear.center";
310 break;
311 case TGSI_INTERPOLATE_PERSPECTIVE:
312 if (decl->Interp.Centroid)
313 intr_name = "llvm.SI.fs.interp.persp.centroid";
314 else
315 intr_name = "llvm.SI.fs.interp.persp.center";
Tom Stellarda75c6162012-01-06 17:38:37 -0500316 break;
317 default:
318 fprintf(stderr, "Warning: Unhandled interpolation mode.\n");
319 return;
320 }
321
322 /* XXX: Could there be more than TGSI_NUM_CHANNELS (4) ? */
323 for (chan = 0; chan < TGSI_NUM_CHANNELS; chan++) {
324 LLVMValueRef args[3];
325 LLVMValueRef llvm_chan = lp_build_const_int32(gallivm, chan);
326 unsigned soa_index = radeon_llvm_reg_index_soa(input_index, chan);
327 LLVMTypeRef input_type = LLVMFloatTypeInContext(gallivm->context);
328 args[0] = llvm_chan;
329 args[1] = attr_number;
330 args[2] = params;
331 si_shader_ctx->radeon_bld.inputs[soa_index] =
332 lp_build_intrinsic(gallivm->builder, intr_name,
333 input_type, args, 3);
334 }
335}
336
337static void declare_input(
338 struct radeon_llvm_context * radeon_bld,
339 unsigned input_index,
340 const struct tgsi_full_declaration *decl)
341{
342 struct si_shader_context * si_shader_ctx =
343 si_shader_context(&radeon_bld->soa.bld_base);
344 if (si_shader_ctx->type == TGSI_PROCESSOR_VERTEX) {
345 declare_input_vs(si_shader_ctx, input_index, decl);
346 } else if (si_shader_ctx->type == TGSI_PROCESSOR_FRAGMENT) {
347 declare_input_fs(si_shader_ctx, input_index, decl);
348 } else {
349 fprintf(stderr, "Warning: Unsupported shader type,\n");
350 }
351}
352
353static LLVMValueRef fetch_constant(
354 struct lp_build_tgsi_context * bld_base,
355 const struct tgsi_full_src_register *reg,
356 enum tgsi_opcode_type type,
357 unsigned swizzle)
358{
359 struct lp_build_context * base = &bld_base->base;
360
361 LLVMValueRef const_ptr;
362 LLVMValueRef offset;
Tom Stellard022b5432012-07-25 08:23:52 -0400363 LLVMValueRef load;
Tom Stellarda75c6162012-01-06 17:38:37 -0500364
365 /* XXX: Assume the pointer to the constant buffer is being stored in
Michel Dänzer9918fbd2012-04-19 11:41:48 +0200366 * SGPR[0:1] */
Tom Stellard467f5162012-05-16 15:15:35 -0400367 const_ptr = use_sgpr(base->gallivm, SGPR_CONST_PTR_F32, 0);
Tom Stellarda75c6162012-01-06 17:38:37 -0500368
369 /* XXX: This assumes that the constant buffer is not packed, so
370 * CONST[0].x will have an offset of 0 and CONST[1].x will have an
371 * offset of 4. */
372 offset = lp_build_const_int32(base->gallivm,
373 (reg->Register.Index * 4) + swizzle);
374
Tom Stellard022b5432012-07-25 08:23:52 -0400375 load = build_indexed_load(base->gallivm, const_ptr, offset);
376 return bitcast(bld_base, type, load);
Tom Stellarda75c6162012-01-06 17:38:37 -0500377}
378
Michel Dänzer26c71392012-08-24 12:03:11 +0200379/* Initialize arguments for the shader export intrinsic */
380static void si_llvm_init_export_args(struct lp_build_tgsi_context *bld_base,
381 struct tgsi_full_declaration *d,
382 unsigned index,
383 unsigned target,
384 LLVMValueRef *args)
385{
386 struct si_shader_context *si_shader_ctx = si_shader_context(bld_base);
387 struct lp_build_context *uint =
388 &si_shader_ctx->radeon_bld.soa.bld_base.uint_bld;
389 struct lp_build_context *base = &bld_base->base;
390 unsigned compressed = 0;
391 unsigned chan;
392
393 for (chan = 0; chan < 4; chan++ ) {
394 LLVMValueRef out_ptr =
395 si_shader_ctx->radeon_bld.soa.outputs[index][chan];
396 /* +5 because the first output value will be
397 * the 6th argument to the intrinsic. */
398 args[chan + 5] = LLVMBuildLoad(base->gallivm->builder,
399 out_ptr, "");
400 }
401
402 /* XXX: This controls which components of the output
403 * registers actually get exported. (e.g bit 0 means export
404 * X component, bit 1 means export Y component, etc.) I'm
405 * hard coding this to 0xf for now. In the future, we might
406 * want to do something else. */
407 args[0] = lp_build_const_int32(base->gallivm, 0xf);
408
409 /* Specify whether the EXEC mask represents the valid mask */
410 args[1] = uint->zero;
411
412 /* Specify whether this is the last export */
413 args[2] = uint->zero;
414
415 /* Specify the target we are exporting */
416 args[3] = lp_build_const_int32(base->gallivm, target);
417
418 /* Set COMPR flag */
419 args[4] = uint->zero;
420
421 /* XXX: We probably need to keep track of the output
422 * values, so we know what we are passing to the next
423 * stage. */
424}
425
Tom Stellarda75c6162012-01-06 17:38:37 -0500426/* XXX: This is partially implemented for VS only at this point. It is not complete */
427static void si_llvm_emit_epilogue(struct lp_build_tgsi_context * bld_base)
428{
429 struct si_shader_context * si_shader_ctx = si_shader_context(bld_base);
Christian König3c09f112012-07-18 17:39:15 +0200430 struct si_shader * shader = &si_shader_ctx->shader->shader;
Tom Stellarda75c6162012-01-06 17:38:37 -0500431 struct lp_build_context * base = &bld_base->base;
432 struct lp_build_context * uint =
433 &si_shader_ctx->radeon_bld.soa.bld_base.uint_bld;
434 struct tgsi_parse_context *parse = &si_shader_ctx->parse;
435 LLVMValueRef last_args[9] = { 0 };
Christian König35088152012-08-01 22:35:24 +0200436 unsigned color_count = 0;
437 unsigned param_count = 0;
Tom Stellarda75c6162012-01-06 17:38:37 -0500438
439 while (!tgsi_parse_end_of_tokens(parse)) {
Tom Stellarda75c6162012-01-06 17:38:37 -0500440 struct tgsi_full_declaration *d =
441 &parse->FullToken.FullDeclaration;
442 LLVMValueRef args[9];
443 unsigned target;
444 unsigned index;
Tom Stellarda75c6162012-01-06 17:38:37 -0500445 int i;
446
447 tgsi_parse_token(parse);
448 if (parse->FullToken.Token.Type != TGSI_TOKEN_TYPE_DECLARATION)
449 continue;
450
451 switch (d->Declaration.File) {
452 case TGSI_FILE_INPUT:
453 i = shader->ninput++;
454 shader->input[i].name = d->Semantic.Name;
455 shader->input[i].sid = d->Semantic.Index;
Francisco Jerez12799232012-04-30 18:27:52 +0200456 shader->input[i].interpolate = d->Interp.Interpolate;
457 shader->input[i].centroid = d->Interp.Centroid;
Christian König35088152012-08-01 22:35:24 +0200458 continue;
459
Tom Stellarda75c6162012-01-06 17:38:37 -0500460 case TGSI_FILE_OUTPUT:
461 i = shader->noutput++;
462 shader->output[i].name = d->Semantic.Name;
463 shader->output[i].sid = d->Semantic.Index;
Francisco Jerez12799232012-04-30 18:27:52 +0200464 shader->output[i].interpolate = d->Interp.Interpolate;
Tom Stellarda75c6162012-01-06 17:38:37 -0500465 break;
Tom Stellarda75c6162012-01-06 17:38:37 -0500466
Christian König35088152012-08-01 22:35:24 +0200467 default:
Tom Stellarda75c6162012-01-06 17:38:37 -0500468 continue;
Christian König35088152012-08-01 22:35:24 +0200469 }
Tom Stellarda75c6162012-01-06 17:38:37 -0500470
471 for (index = d->Range.First; index <= d->Range.Last; index++) {
Tom Stellarda75c6162012-01-06 17:38:37 -0500472 /* Select the correct target */
473 switch(d->Semantic.Name) {
474 case TGSI_SEMANTIC_POSITION:
475 target = V_008DFC_SQ_EXP_POS;
476 break;
477 case TGSI_SEMANTIC_COLOR:
478 if (si_shader_ctx->type == TGSI_PROCESSOR_VERTEX) {
479 target = V_008DFC_SQ_EXP_PARAM + param_count;
Michel Dänzerdd9d6192012-05-18 15:01:10 +0200480 shader->output[i].param_offset = param_count;
Tom Stellarda75c6162012-01-06 17:38:37 -0500481 param_count++;
482 } else {
483 target = V_008DFC_SQ_EXP_MRT + color_count;
484 color_count++;
485 }
486 break;
487 case TGSI_SEMANTIC_GENERIC:
488 target = V_008DFC_SQ_EXP_PARAM + param_count;
Michel Dänzerdd9d6192012-05-18 15:01:10 +0200489 shader->output[i].param_offset = param_count;
Tom Stellarda75c6162012-01-06 17:38:37 -0500490 param_count++;
491 break;
492 default:
493 target = 0;
494 fprintf(stderr,
495 "Warning: SI unhandled output type:%d\n",
496 d->Semantic.Name);
497 }
498
Michel Dänzer26c71392012-08-24 12:03:11 +0200499 si_llvm_init_export_args(bld_base, d, index, target, args);
Tom Stellarda75c6162012-01-06 17:38:37 -0500500
501 if (si_shader_ctx->type == TGSI_PROCESSOR_VERTEX ?
502 (d->Semantic.Name == TGSI_SEMANTIC_POSITION) :
503 (d->Semantic.Name == TGSI_SEMANTIC_COLOR)) {
504 if (last_args[0]) {
505 lp_build_intrinsic(base->gallivm->builder,
506 "llvm.SI.export",
507 LLVMVoidTypeInContext(base->gallivm->context),
508 last_args, 9);
509 }
510
511 memcpy(last_args, args, sizeof(args));
512 } else {
513 lp_build_intrinsic(base->gallivm->builder,
514 "llvm.SI.export",
515 LLVMVoidTypeInContext(base->gallivm->context),
516 args, 9);
517 }
518
519 }
520 }
521
Christian Königf18fd252012-07-25 21:58:46 +0200522 if (!last_args[0]) {
523 assert(si_shader_ctx->type == TGSI_PROCESSOR_FRAGMENT);
524
525 /* Specify which components to enable */
526 last_args[0] = lp_build_const_int32(base->gallivm, 0x0);
527
528 /* Specify the target we are exporting */
529 last_args[3] = lp_build_const_int32(base->gallivm, V_008DFC_SQ_EXP_MRT);
530
531 /* Set COMPR flag to zero to export data as 32-bit */
532 last_args[4] = uint->zero;
533
534 /* dummy bits */
535 last_args[5]= uint->zero;
536 last_args[6]= uint->zero;
537 last_args[7]= uint->zero;
538 last_args[8]= uint->zero;
539 }
540
Tom Stellarda75c6162012-01-06 17:38:37 -0500541 /* Specify whether the EXEC mask represents the valid mask */
542 last_args[1] = lp_build_const_int32(base->gallivm,
543 si_shader_ctx->type == TGSI_PROCESSOR_FRAGMENT);
544
545 /* Specify that this is the last export */
546 last_args[2] = lp_build_const_int32(base->gallivm, 1);
547
548 lp_build_intrinsic(base->gallivm->builder,
549 "llvm.SI.export",
550 LLVMVoidTypeInContext(base->gallivm->context),
551 last_args, 9);
552
553/* XXX: Look up what this function does */
554/* ctx->shader->output[i].spi_sid = r600_spi_sid(&ctx->shader->output[i]);*/
555}
556
557static void tex_fetch_args(
558 struct lp_build_tgsi_context * bld_base,
559 struct lp_build_emit_data * emit_data)
560{
Michel Dänzerc2bae6b2012-08-02 17:19:22 +0200561 const struct tgsi_full_instruction * inst = emit_data->inst;
Tom Stellard467f5162012-05-16 15:15:35 -0400562 LLVMValueRef ptr;
563 LLVMValueRef offset;
564
Tom Stellarda75c6162012-01-06 17:38:37 -0500565 /* WriteMask */
Christian König250b7fd2012-08-01 23:18:14 +0200566 /* XXX: should be optimized using emit_data->inst->Dst[0].Register.WriteMask*/
567 emit_data->args[0] = lp_build_const_int32(bld_base->base.gallivm, 0xf);
Tom Stellarda75c6162012-01-06 17:38:37 -0500568
569 /* Coordinates */
570 /* XXX: Not all sample instructions need 4 address arguments. */
Michel Dänzerc2bae6b2012-08-02 17:19:22 +0200571 if (inst->Instruction.Opcode == TGSI_OPCODE_TXP) {
572 LLVMValueRef src_w;
573 unsigned chan;
574 LLVMValueRef coords[4];
575
576 emit_data->dst_type = LLVMVectorType(bld_base->base.elem_type, 4);
577 src_w = lp_build_emit_fetch(bld_base, emit_data->inst, 0, TGSI_CHAN_W);
578
579 for (chan = 0; chan < 3; chan++ ) {
580 LLVMValueRef arg = lp_build_emit_fetch(bld_base,
581 emit_data->inst, 0, chan);
582 coords[chan] = lp_build_emit_llvm_binary(bld_base,
583 TGSI_OPCODE_DIV,
584 arg, src_w);
585 }
586 coords[3] = bld_base->base.one;
587 emit_data->args[1] = lp_build_gather_values(bld_base->base.gallivm,
588 coords, 4);
589 } else
590 emit_data->args[1] = lp_build_emit_fetch(bld_base, emit_data->inst,
591 0, LP_CHAN_ALL);
Tom Stellarda75c6162012-01-06 17:38:37 -0500592
593 /* Resource */
Tom Stellard89ece082012-05-29 11:36:29 -0400594 ptr = use_sgpr(bld_base->base.gallivm, SGPR_CONST_PTR_V8I32, 4);
Tom Stellard467f5162012-05-16 15:15:35 -0400595 offset = lp_build_const_int32(bld_base->base.gallivm,
Christian König92b96a82012-08-01 15:20:07 +0200596 emit_data->inst->Src[1].Register.Index);
Tom Stellard467f5162012-05-16 15:15:35 -0400597 emit_data->args[2] = build_indexed_load(bld_base->base.gallivm,
598 ptr, offset);
Tom Stellarda75c6162012-01-06 17:38:37 -0500599
600 /* Sampler */
Tom Stellard89ece082012-05-29 11:36:29 -0400601 ptr = use_sgpr(bld_base->base.gallivm, SGPR_CONST_PTR_V4I32, 2);
Tom Stellard467f5162012-05-16 15:15:35 -0400602 offset = lp_build_const_int32(bld_base->base.gallivm,
Christian König92b96a82012-08-01 15:20:07 +0200603 emit_data->inst->Src[1].Register.Index);
Tom Stellard467f5162012-05-16 15:15:35 -0400604 emit_data->args[3] = build_indexed_load(bld_base->base.gallivm,
605 ptr, offset);
Tom Stellarda75c6162012-01-06 17:38:37 -0500606
607 /* Dimensions */
608 /* XXX: We might want to pass this information to the shader at some. */
609/* emit_data->args[4] = lp_build_const_int32(bld_base->base.gallivm,
610 emit_data->inst->Texture.Texture);
611*/
612
Tom Stellard467f5162012-05-16 15:15:35 -0400613 emit_data->arg_count = 4;
Tom Stellarda75c6162012-01-06 17:38:37 -0500614 /* XXX: To optimize, we could use a float or v2f32, if the last bits of
615 * the writemask are clear */
616 emit_data->dst_type = LLVMVectorType(
617 LLVMFloatTypeInContext(bld_base->base.gallivm->context),
618 4);
619}
620
621static const struct lp_build_tgsi_action tex_action = {
622 .fetch_args = tex_fetch_args,
623 .emit = lp_build_tgsi_intrinsic,
624 .intr_name = "llvm.SI.sample"
625};
626
627
628int si_pipe_shader_create(
629 struct pipe_context *ctx,
630 struct si_pipe_shader *shader)
631{
632 struct r600_context *rctx = (struct r600_context*)ctx;
Michel Dänzerd1e40b32012-08-23 17:10:37 +0200633 struct si_pipe_shader_selector *sel = shader->selector;
Tom Stellarda75c6162012-01-06 17:38:37 -0500634 struct si_shader_context si_shader_ctx;
635 struct tgsi_shader_info shader_info;
636 struct lp_build_tgsi_context * bld_base;
637 LLVMModuleRef mod;
638 unsigned char * inst_bytes;
639 unsigned inst_byte_count;
640 unsigned i;
Christian Königd51b9b72012-07-24 18:50:49 +0200641 uint32_t *ptr;
Michel Dänzer4c4ef9c2012-06-07 19:30:47 +0200642 bool dump;
643
644 dump = debug_get_bool_option("RADEON_DUMP_SHADERS", FALSE);
Tom Stellarda75c6162012-01-06 17:38:37 -0500645
Tom Stellarda5ac8ee2012-08-02 13:21:30 -0400646 memset(&si_shader_ctx.radeon_bld, 0, sizeof(si_shader_ctx.radeon_bld));
Tom Stellarda75c6162012-01-06 17:38:37 -0500647 radeon_llvm_context_init(&si_shader_ctx.radeon_bld);
648 bld_base = &si_shader_ctx.radeon_bld.soa.bld_base;
649
Michel Dänzerd1e40b32012-08-23 17:10:37 +0200650 tgsi_scan_shader(sel->tokens, &shader_info);
Tom Stellarda75c6162012-01-06 17:38:37 -0500651 bld_base->info = &shader_info;
652 bld_base->emit_fetch_funcs[TGSI_FILE_CONSTANT] = fetch_constant;
Tom Stellarda75c6162012-01-06 17:38:37 -0500653 bld_base->emit_epilogue = si_llvm_emit_epilogue;
654
655 bld_base->op_actions[TGSI_OPCODE_TEX] = tex_action;
Michel Dänzerc2bae6b2012-08-02 17:19:22 +0200656 bld_base->op_actions[TGSI_OPCODE_TXP] = tex_action;
Tom Stellarda75c6162012-01-06 17:38:37 -0500657
658 si_shader_ctx.radeon_bld.load_input = declare_input;
Michel Dänzerd1e40b32012-08-23 17:10:37 +0200659 si_shader_ctx.tokens = sel->tokens;
Tom Stellarda75c6162012-01-06 17:38:37 -0500660 tgsi_parse_init(&si_shader_ctx.parse, si_shader_ctx.tokens);
661 si_shader_ctx.shader = shader;
662 si_shader_ctx.type = si_shader_ctx.parse.FullHeader.Processor.Processor;
663 si_shader_ctx.rctx = rctx;
664
Christian König835098a2012-07-17 21:28:10 +0200665 shader->shader.nr_cbufs = rctx->framebuffer.nr_cbufs;
Tom Stellarda75c6162012-01-06 17:38:37 -0500666
Tom Stellard185fc9a2012-07-12 10:40:47 -0400667 /* Dump TGSI code before doing TGSI->LLVM conversion in case the
668 * conversion fails. */
669 if (dump) {
Michel Dänzerd1e40b32012-08-23 17:10:37 +0200670 tgsi_dump(sel->tokens, 0);
Tom Stellard185fc9a2012-07-12 10:40:47 -0400671 }
672
Michel Dänzerd1e40b32012-08-23 17:10:37 +0200673 if (!lp_build_tgsi_llvm(bld_base, sel->tokens)) {
Michel Dänzer82cd9c02012-08-08 15:35:42 +0200674 fprintf(stderr, "Failed to translate shader from TGSI to LLVM\n");
675 return -EINVAL;
676 }
Tom Stellarda75c6162012-01-06 17:38:37 -0500677
678 radeon_llvm_finalize_module(&si_shader_ctx.radeon_bld);
679
680 mod = bld_base->base.gallivm->module;
Michel Dänzer4c4ef9c2012-06-07 19:30:47 +0200681 if (dump) {
Michel Dänzer4c4ef9c2012-06-07 19:30:47 +0200682 LLVMDumpModule(mod);
683 }
684 radeon_llvm_compile(mod, &inst_bytes, &inst_byte_count, "SI", dump);
685 if (dump) {
686 fprintf(stderr, "SI CODE:\n");
687 for (i = 0; i < inst_byte_count; i+=4 ) {
688 fprintf(stderr, "%02x%02x%02x%02x\n", inst_bytes[i + 3],
689 inst_bytes[i + 2], inst_bytes[i + 1],
690 inst_bytes[i]);
691 }
Tom Stellarda75c6162012-01-06 17:38:37 -0500692 }
693
694 shader->num_sgprs = util_le32_to_cpu(*(uint32_t*)inst_bytes);
695 shader->num_vgprs = util_le32_to_cpu(*(uint32_t*)(inst_bytes + 4));
696 shader->spi_ps_input_ena = util_le32_to_cpu(*(uint32_t*)(inst_bytes + 8));
697
Michel Dänzer4b64fa22012-08-15 18:22:46 +0200698 radeon_llvm_dispose(&si_shader_ctx.radeon_bld);
Tom Stellarda75c6162012-01-06 17:38:37 -0500699 tgsi_parse_free(&si_shader_ctx.parse);
700
701 /* copy new shader */
Christian Königd51b9b72012-07-24 18:50:49 +0200702 si_resource_reference(&shader->bo, NULL);
703 shader->bo = si_resource_create_custom(ctx->screen, PIPE_USAGE_IMMUTABLE,
704 inst_byte_count - 12);
Tom Stellarda75c6162012-01-06 17:38:37 -0500705 if (shader->bo == NULL) {
Christian Königd51b9b72012-07-24 18:50:49 +0200706 return -ENOMEM;
Tom Stellarda75c6162012-01-06 17:38:37 -0500707 }
708
Christian Königd51b9b72012-07-24 18:50:49 +0200709 ptr = (uint32_t*)rctx->ws->buffer_map(shader->bo->cs_buf, rctx->cs, PIPE_TRANSFER_WRITE);
710 if (0 /*R600_BIG_ENDIAN*/) {
711 for (i = 0; i < (inst_byte_count-12)/4; ++i) {
712 ptr[i] = util_bswap32(*(uint32_t*)(inst_bytes+12 + i*4));
713 }
714 } else {
715 memcpy(ptr, inst_bytes + 12, inst_byte_count - 12);
716 }
717 rctx->ws->buffer_unmap(shader->bo->cs_buf);
718
Tom Stellarda75c6162012-01-06 17:38:37 -0500719 free(inst_bytes);
720
721 return 0;
722}
723
724void si_pipe_shader_destroy(struct pipe_context *ctx, struct si_pipe_shader *shader)
725{
Christian Königfe412872012-07-24 18:47:19 +0200726 si_resource_reference(&shader->bo, NULL);
Tom Stellarda75c6162012-01-06 17:38:37 -0500727}