/* * Copyright (c) 2014 Scott Mansell * Copyright © 2014 Broadcom * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice (including the next * paragraph) shall be included in all copies or substantial portions of the * Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS * IN THE SOFTWARE. */ #include #include #include "pipe/p_state.h" #include "util/u_format.h" #include "util/u_hash_table.h" #include "util/u_hash.h" #include "util/u_memory.h" #include "tgsi/tgsi_parse.h" #include "tgsi/tgsi_dump.h" #include "vc4_context.h" #include "vc4_qpu.h" #include "vc4_qir.h" #ifdef USE_VC4_SIMULATOR #include "simpenrose/simpenrose.h" #endif struct tgsi_to_qir { struct tgsi_parse_context parser; struct qcompile *c; struct qreg *temps; struct qreg *inputs; struct qreg *outputs; struct qreg *uniforms; struct qreg *consts; uint32_t num_consts; struct pipe_shader_state *shader_state; struct vc4_fs_key *fs_key; struct vc4_vs_key *vs_key; uint32_t *uniform_data; enum quniform_contents *uniform_contents; uint32_t num_uniforms; uint32_t num_outputs; uint32_t num_texture_samples; }; struct vc4_key { struct pipe_shader_state *shader_state; }; struct vc4_fs_key { struct vc4_key base; enum pipe_format color_format; bool depth_enabled; }; struct vc4_vs_key { struct vc4_key base; enum pipe_format attr_formats[8]; }; static struct qreg add_uniform(struct tgsi_to_qir *trans, enum quniform_contents contents, uint32_t data) { uint32_t uniform = trans->num_uniforms++; struct qreg u = { QFILE_UNIF, uniform }; trans->uniform_contents[uniform] = contents; trans->uniform_data[uniform] = data; return u; } static struct qreg get_temp_for_uniform(struct tgsi_to_qir *trans, enum quniform_contents contents, uint32_t data) { struct qcompile *c = trans->c; for (int i = 0; i < trans->num_uniforms; i++) { if (trans->uniform_contents[i] == contents && trans->uniform_data[i] == data) return trans->uniforms[i]; } struct qreg u = add_uniform(trans, contents, data); struct qreg t = qir_MOV(c, u); trans->uniforms[u.index] = t; return t; } static struct qreg qir_uniform_ui(struct tgsi_to_qir *trans, uint32_t ui) { return get_temp_for_uniform(trans, QUNIFORM_CONSTANT, ui); } static struct qreg qir_uniform_f(struct tgsi_to_qir *trans, float f) { return qir_uniform_ui(trans, fui(f)); } static struct qreg get_src(struct tgsi_to_qir *trans, struct tgsi_src_register *src, int i) { struct qcompile *c = trans->c; struct qreg r = c->undef; uint32_t s = i; switch (i) { case TGSI_SWIZZLE_X: s = src->SwizzleX; break; case TGSI_SWIZZLE_Y: s = src->SwizzleY; break; case TGSI_SWIZZLE_Z: s = src->SwizzleZ; break; case TGSI_SWIZZLE_W: s = src->SwizzleW; break; default: abort(); } assert(!src->Indirect); switch (src->File) { case TGSI_FILE_NULL: return r; case TGSI_FILE_TEMPORARY: r = trans->temps[src->Index * 4 + s]; break; case TGSI_FILE_IMMEDIATE: r = trans->consts[src->Index * 4 + s]; break; case TGSI_FILE_CONSTANT: r = get_temp_for_uniform(trans, QUNIFORM_UNIFORM, src->Index * 4 + s); break; case TGSI_FILE_INPUT: r = trans->inputs[src->Index * 4 + s]; break; case TGSI_FILE_SAMPLER: case TGSI_FILE_SAMPLER_VIEW: r = c->undef; break; default: fprintf(stderr, "unknown src file %d\n", src->File); abort(); } if (src->Absolute) r = qir_FMAXABS(c, r, r); if (src->Negate) r = qir_FSUB(c, qir_uniform_f(trans, 0), r); return r; }; static void update_dst(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, int i, struct qreg val) { struct tgsi_dst_register *tgsi_dst = &tgsi_inst->Dst[0].Register; assert(!tgsi_dst->Indirect); switch (tgsi_dst->File) { case TGSI_FILE_TEMPORARY: trans->temps[tgsi_dst->Index * 4 + i] = val; break; case TGSI_FILE_OUTPUT: trans->outputs[tgsi_dst->Index * 4 + i] = val; trans->num_outputs = MAX2(trans->num_outputs, tgsi_dst->Index * 4 + i + 1); break; default: fprintf(stderr, "unknown dst file %d\n", tgsi_dst->File); abort(); } }; static struct qreg tgsi_to_qir_alu(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; struct qreg dst = qir_get_temp(c); qir_emit(c, qir_inst4(op, dst, src[0 * 4 + i], src[1 * 4 + i], src[2 * 4 + i], c->undef)); return dst; } static struct qreg tgsi_to_qir_mad(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; return qir_FADD(c, qir_FMUL(c, src[0 * 4 + i], src[1 * 4 + i]), src[2 * 4 + i]); } static struct qreg tgsi_to_qir_lit(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; struct qreg x = src[0 * 4 + 0]; struct qreg y = src[0 * 4 + 1]; struct qreg w = src[0 * 4 + 3]; switch (i) { case 0: case 3: return qir_uniform_f(trans, 1.0); case 1: return qir_FMAX(c, src[0 * 4 + 0], qir_uniform_f(trans, 0.0)); case 2: { struct qreg zero = qir_uniform_f(trans, 0.0); /* XXX: Clamp w to -128..128 */ return qir_CMP(c, x, zero, qir_EXP2(c, qir_FMUL(c, w, qir_LOG2(c, qir_FMAX(c, y, zero))))); } default: assert(!"not reached"); return c->undef; } } static struct qreg tgsi_to_qir_lrp(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; struct qreg src0 = src[0 * 4 + i]; struct qreg src1 = src[1 * 4 + i]; struct qreg src2 = src[2 * 4 + i]; /* LRP is: * src0 * src1 + (1 - src0) * src2. * -> src0 * src1 + src2 - src0 * src2 * -> src2 + src0 * (src1 - src2) */ return qir_FADD(c, src2, qir_FMUL(c, src0, qir_FSUB(c, src1, src2))); } static void tgsi_to_qir_tex(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src) { struct qcompile *c = trans->c; assert(!tgsi_inst->Instruction.Saturate); struct qreg s = src[0 * 4 + 0]; struct qreg t = src[0 * 4 + 1]; if (tgsi_inst->Instruction.Opcode == TGSI_OPCODE_TXP) { struct qreg proj = qir_RCP(c, src[0 * 4 + 3]); s = qir_FMUL(c, s, proj); t = qir_FMUL(c, t, proj); } /* There is no native support for GL texture rectangle coordinates, so * we have to rescale from ([0, width], [0, height]) to ([0, 1], [0, * 1]). */ if (tgsi_inst->Texture.Texture == TGSI_TEXTURE_RECT) { uint32_t sampler = 0; /* XXX */ s = qir_FMUL(c, s, get_temp_for_uniform(trans, QUNIFORM_TEXRECT_SCALE_X, sampler)); t = qir_FMUL(c, t, get_temp_for_uniform(trans, QUNIFORM_TEXRECT_SCALE_Y, sampler)); } uint32_t tex_and_sampler = 0; /* XXX */ qir_TEX_T(c, t, add_uniform(trans, QUNIFORM_TEXTURE_CONFIG_P0, tex_and_sampler)); struct qreg sampler_p1 = add_uniform(trans, QUNIFORM_TEXTURE_CONFIG_P1, tex_and_sampler); if (tgsi_inst->Instruction.Opcode == TGSI_OPCODE_TXB) { qir_TEX_B(c, src[0 * 4 + 3], sampler_p1); qir_TEX_S(c, s, add_uniform(trans, QUNIFORM_CONSTANT, 0)); } else { qir_TEX_S(c, s, sampler_p1); } trans->num_texture_samples++; qir_emit(c, qir_inst(QOP_TEX_RESULT, c->undef, c->undef, c->undef)); for (int i = 0; i < 4; i++) { if (!(tgsi_inst->Dst[0].Register.WriteMask & (1 << i))) continue; struct qreg dst = qir_get_temp(c); qir_emit(c, qir_inst(QOP_R4_UNPACK_A + i, dst, c->undef, c->undef)); update_dst(trans, tgsi_inst, i, dst); } } static struct qreg tgsi_to_qir_pow(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; /* Note that this instruction replicates its result from the x channel */ return qir_EXP2(c, qir_FMUL(c, src[1 * 4 + 0], qir_LOG2(c, src[0 * 4 + 0]))); } static struct qreg tgsi_to_qir_trunc(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; return qir_ITOF(c, qir_FTOI(c, src[0 * 4 + i])); } /** * Computes x - floor(x), which is tricky because our FTOI truncates (rounds * to zero). */ static struct qreg tgsi_to_qir_frc(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; struct qreg trunc = qir_ITOF(c, qir_FTOI(c, src[0 * 4 + i])); struct qreg diff = qir_FSUB(c, src[0 * 4 + i], trunc); return qir_CMP(c, diff, qir_FADD(c, diff, qir_uniform_f(trans, 1.0)), diff); } /** * Computes floor(x), which is tricky because our FTOI truncates (rounds to * zero). */ static struct qreg tgsi_to_qir_flr(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; struct qreg trunc = qir_ITOF(c, qir_FTOI(c, src[0 * 4 + i])); return qir_CMP(c, src[0 * 4 + i], qir_FSUB(c, trunc, qir_uniform_f(trans, 1.0)), trunc); } static struct qreg tgsi_to_qir_dp(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, int num, struct qreg *src, int i) { struct qcompile *c = trans->c; struct qreg sum = qir_FMUL(c, src[0 * 4 + 0], src[1 * 4 + 0]); for (int j = 1; j < num; j++) { sum = qir_FADD(c, sum, qir_FMUL(c, src[0 * 4 + j], src[1 * 4 + j])); } return sum; } static struct qreg tgsi_to_qir_dp2(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { return tgsi_to_qir_dp(trans, tgsi_inst, 2, src, i); } static struct qreg tgsi_to_qir_dp3(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { return tgsi_to_qir_dp(trans, tgsi_inst, 3, src, i); } static struct qreg tgsi_to_qir_dp4(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { return tgsi_to_qir_dp(trans, tgsi_inst, 4, src, i); } static struct qreg tgsi_to_qir_abs(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; struct qreg arg = src[0 * 4 + i]; return qir_FMAXABS(c, arg, arg); } /* Note that this instruction replicates its result from the x channel */ static struct qreg tgsi_to_qir_sin(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; float coeff[] = { 2.0 * M_PI, -pow(2.0 * M_PI, 3) / (3 * 2 * 1), pow(2.0 * M_PI, 5) / (5 * 4 * 3 * 2 * 1), -pow(2.0 * M_PI, 7) / (7 * 6 * 5 * 4 * 3 * 2 * 1), }; struct qreg scaled_x = qir_FMUL(c, src[0 * 4 + 0], qir_uniform_f(trans, 1.0f / (M_PI * 2.0f))); struct qreg x = tgsi_to_qir_frc(trans, NULL, 0, &scaled_x, 0); struct qreg x2 = qir_FMUL(c, x, x); struct qreg sum = qir_FMUL(c, x, qir_uniform_f(trans, coeff[0])); for (int i = 1; i < ARRAY_SIZE(coeff); i++) { x = qir_FMUL(c, x, x2); sum = qir_FADD(c, sum, qir_FMUL(c, x, qir_uniform_f(trans, coeff[i]))); } return sum; } /* Note that this instruction replicates its result from the x channel */ static struct qreg tgsi_to_qir_cos(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i) { struct qcompile *c = trans->c; float coeff[] = { 1.0f, -pow(2.0 * M_PI, 2) / (2 * 1), pow(2.0 * M_PI, 4) / (4 * 3 * 2 * 1), -pow(2.0 * M_PI, 6) / (6 * 5 * 4 * 3 * 2 * 1), }; struct qreg scaled_x = qir_FMUL(c, src[0 * 4 + 0], qir_uniform_f(trans, 1.0f / (M_PI * 2.0f))); struct qreg x_frac = tgsi_to_qir_frc(trans, NULL, 0, &scaled_x, 0); struct qreg sum = qir_uniform_f(trans, coeff[0]); struct qreg x2 = qir_FMUL(c, x_frac, x_frac); struct qreg x = x2; /* Current x^2, x^4, or x^6 */ for (int i = 1; i < ARRAY_SIZE(coeff); i++) { if (i != 1) x = qir_FMUL(c, x, x2); struct qreg mul = qir_FMUL(c, x, qir_uniform_f(trans, coeff[i])); if (i == 0) sum = mul; else sum = qir_FADD(c, sum, mul); } return sum; } static void emit_vertex_input(struct tgsi_to_qir *trans, int attr) { enum pipe_format format = trans->vs_key->attr_formats[attr]; struct qcompile *c = trans->c; struct qreg vpm_reads[4]; /* Right now, we're setting the VPM offsets to be 16 bytes wide every * time, so we always read 4 32-bit VPM entries. */ for (int i = 0; i < 4; i++) { vpm_reads[i] = qir_get_temp(c); qir_emit(c, qir_inst(QOP_VPM_READ, vpm_reads[i], c->undef, c->undef)); c->num_inputs++; } bool format_warned = false; const struct util_format_description *desc = util_format_description(format); for (int i = 0; i < 4; i++) { uint8_t swiz = desc->swizzle[i]; switch (swiz) { case UTIL_FORMAT_SWIZZLE_NONE: if (!format_warned) { fprintf(stderr, "vtx element %d NONE swizzle: %s\n", attr, util_format_name(format)); format_warned = true; } /* FALLTHROUGH */ case UTIL_FORMAT_SWIZZLE_0: trans->inputs[attr * 4 + i] = qir_uniform_ui(trans, 0); break; case UTIL_FORMAT_SWIZZLE_1: trans->inputs[attr * 4 + i] = qir_uniform_ui(trans, fui(1.0)); break; default: if (!format_warned && (desc->channel[swiz].type != UTIL_FORMAT_TYPE_FLOAT || desc->channel[swiz].size != 32)) { fprintf(stderr, "vtx element %d unsupported type: %s\n", attr, util_format_name(format)); format_warned = true; } trans->inputs[attr * 4 + i] = vpm_reads[swiz]; break; } } } static void emit_fragcoord_input(struct tgsi_to_qir *trans, int attr) { struct qcompile *c = trans->c; trans->inputs[attr * 4 + 0] = qir_FRAG_X(c); trans->inputs[attr * 4 + 1] = qir_FRAG_Y(c); trans->inputs[attr * 4 + 2] = qir_FMUL(c, qir_FRAG_Z(c), qir_uniform_f(trans, 1.0 / 0xffffff)); trans->inputs[attr * 4 + 3] = qir_FRAG_RCP_W(c); } static void emit_fragment_input(struct tgsi_to_qir *trans, int attr) { struct qcompile *c = trans->c; for (int i = 0; i < 4; i++) { struct qreg vary = { QFILE_VARY, attr * 4 + i }; /* XXX: multiply by W */ trans->inputs[attr * 4 + i] = qir_VARY_ADD_C(c, qir_MOV(c, vary)); c->num_inputs++; } } static void emit_tgsi_declaration(struct tgsi_to_qir *trans, struct tgsi_full_declaration *decl) { struct qcompile *c = trans->c; switch (decl->Declaration.File) { case TGSI_FILE_INPUT: for (int i = decl->Range.First; i <= decl->Range.Last; i++) { if (c->stage == QSTAGE_FRAG) { if (decl->Semantic.Name == TGSI_SEMANTIC_POSITION) { emit_fragcoord_input(trans, i); } else { emit_fragment_input(trans, i); } } else { emit_vertex_input(trans, i); } } break; } } static void emit_tgsi_instruction(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst) { struct qcompile *c = trans->c; struct { enum qop op; struct qreg (*func)(struct tgsi_to_qir *trans, struct tgsi_full_instruction *tgsi_inst, enum qop op, struct qreg *src, int i); } op_trans[] = { [TGSI_OPCODE_MOV] = { QOP_MOV, tgsi_to_qir_alu }, [TGSI_OPCODE_ABS] = { 0, tgsi_to_qir_abs }, [TGSI_OPCODE_MUL] = { QOP_FMUL, tgsi_to_qir_alu }, [TGSI_OPCODE_ADD] = { QOP_FADD, tgsi_to_qir_alu }, [TGSI_OPCODE_SUB] = { QOP_FSUB, tgsi_to_qir_alu }, [TGSI_OPCODE_MIN] = { QOP_FMIN, tgsi_to_qir_alu }, [TGSI_OPCODE_MAX] = { QOP_FMAX, tgsi_to_qir_alu }, [TGSI_OPCODE_RSQ] = { QOP_RSQ, tgsi_to_qir_alu }, [TGSI_OPCODE_SEQ] = { QOP_SEQ, tgsi_to_qir_alu }, [TGSI_OPCODE_SNE] = { QOP_SNE, tgsi_to_qir_alu }, [TGSI_OPCODE_SGE] = { QOP_SGE, tgsi_to_qir_alu }, [TGSI_OPCODE_SLT] = { QOP_SLT, tgsi_to_qir_alu }, [TGSI_OPCODE_CMP] = { QOP_CMP, tgsi_to_qir_alu }, [TGSI_OPCODE_MAD] = { 0, tgsi_to_qir_mad }, [TGSI_OPCODE_DP2] = { 0, tgsi_to_qir_dp2 }, [TGSI_OPCODE_DP3] = { 0, tgsi_to_qir_dp3 }, [TGSI_OPCODE_DP4] = { 0, tgsi_to_qir_dp4 }, [TGSI_OPCODE_RCP] = { QOP_RCP, tgsi_to_qir_alu }, [TGSI_OPCODE_RSQ] = { QOP_RSQ, tgsi_to_qir_alu }, [TGSI_OPCODE_EX2] = { QOP_EXP2, tgsi_to_qir_alu }, [TGSI_OPCODE_LG2] = { QOP_LOG2, tgsi_to_qir_alu }, [TGSI_OPCODE_LIT] = { 0, tgsi_to_qir_lit }, [TGSI_OPCODE_LRP] = { 0, tgsi_to_qir_lrp }, [TGSI_OPCODE_POW] = { 0, tgsi_to_qir_pow }, [TGSI_OPCODE_TRUNC] = { 0, tgsi_to_qir_trunc }, [TGSI_OPCODE_FRC] = { 0, tgsi_to_qir_frc }, [TGSI_OPCODE_FLR] = { 0, tgsi_to_qir_flr }, [TGSI_OPCODE_SIN] = { 0, tgsi_to_qir_sin }, [TGSI_OPCODE_COS] = { 0, tgsi_to_qir_cos }, }; static int asdf = 0; uint32_t tgsi_op = tgsi_inst->Instruction.Opcode; if (tgsi_op == TGSI_OPCODE_END) return; struct qreg src_regs[12]; for (int s = 0; s < 3; s++) { for (int i = 0; i < 4; i++) { src_regs[4 * s + i] = get_src(trans, &tgsi_inst->Src[s].Register, i); } } switch (tgsi_op) { case TGSI_OPCODE_TEX: case TGSI_OPCODE_TXP: case TGSI_OPCODE_TXB: tgsi_to_qir_tex(trans, tgsi_inst, op_trans[tgsi_op].op, src_regs); return; default: break; } if (tgsi_op > ARRAY_SIZE(op_trans) || !(op_trans[tgsi_op].func)) { fprintf(stderr, "unknown tgsi inst: "); tgsi_dump_instruction(tgsi_inst, asdf++); fprintf(stderr, "\n"); abort(); } for (int i = 0; i < 4; i++) { if (!(tgsi_inst->Dst[0].Register.WriteMask & (1 << i))) continue; struct qreg result; result = op_trans[tgsi_op].func(trans, tgsi_inst, op_trans[tgsi_op].op, src_regs, i); if (tgsi_inst->Instruction.Saturate) { float low = (tgsi_inst->Instruction.Saturate == TGSI_SAT_MINUS_PLUS_ONE ? -1.0 : 0.0); result = qir_FMAX(c, qir_FMIN(c, result, qir_uniform_f(trans, 1.0)), qir_uniform_f(trans, low)); } update_dst(trans, tgsi_inst, i, result); } } static void parse_tgsi_immediate(struct tgsi_to_qir *trans, struct tgsi_full_immediate *imm) { for (int i = 0; i < 4; i++) { unsigned n = trans->num_consts++; trans->consts[n] = qir_uniform_ui(trans, imm->u[i].Uint); } } static void emit_frag_end(struct tgsi_to_qir *trans) { struct qcompile *c = trans->c; struct qreg t = qir_get_temp(c); const struct util_format_description *format_desc = util_format_description(trans->fs_key->color_format); /* Debug: Sometimes you're getting a black output and just want to see * if the FS is getting executed at all. Spam magenta into the color * output. */ if (0) { trans->outputs[format_desc->swizzle[0]] = qir_uniform_ui(trans, fui(1.0)); trans->outputs[format_desc->swizzle[1]] = qir_uniform_ui(trans, fui(0.0)); trans->outputs[format_desc->swizzle[2]] = qir_uniform_ui(trans, fui(1.0)); trans->outputs[format_desc->swizzle[3]] = qir_uniform_ui(trans, fui(0.5)); } struct qreg swizzled_outputs[4] = { trans->outputs[format_desc->swizzle[0]], trans->outputs[format_desc->swizzle[1]], trans->outputs[format_desc->swizzle[2]], trans->outputs[format_desc->swizzle[3]], }; if (trans->fs_key->depth_enabled) { qir_emit(c, qir_inst(QOP_TLB_PASSTHROUGH_Z_WRITE, c->undef, c->undef, c->undef)); } qir_emit(c, qir_inst4(QOP_PACK_COLORS, t, swizzled_outputs[0], swizzled_outputs[1], swizzled_outputs[2], swizzled_outputs[3])); qir_emit(c, qir_inst(QOP_TLB_COLOR_WRITE, c->undef, t, c->undef)); } static void emit_scaled_viewport_write(struct tgsi_to_qir *trans, struct qreg rcp_w) { struct qcompile *c = trans->c; struct qreg xyi[2]; for (int i = 0; i < 2; i++) { struct qreg scale = add_uniform(trans, QUNIFORM_VIEWPORT_X_SCALE + i, 0); xyi[i] = qir_FTOI(c, qir_FMUL(c, qir_FMUL(c, trans->outputs[i], scale), rcp_w)); } qir_VPM_WRITE(c, qir_PACK_SCALED(c, xyi[0], xyi[1])); } static void emit_zs_write(struct tgsi_to_qir *trans, struct qreg rcp_w) { struct qcompile *c = trans->c; struct qreg zscale = add_uniform(trans, QUNIFORM_VIEWPORT_Z_SCALE, 0); struct qreg zoffset = add_uniform(trans, QUNIFORM_VIEWPORT_Z_OFFSET, 0); qir_VPM_WRITE(c, qir_FMUL(c, qir_FADD(c, qir_FMUL(c, trans->outputs[2], zscale), zoffset), rcp_w)); } static void emit_rcp_wc_write(struct tgsi_to_qir *trans, struct qreg rcp_w) { struct qcompile *c = trans->c; qir_VPM_WRITE(c, rcp_w); } static void emit_vert_end(struct tgsi_to_qir *trans) { struct qcompile *c = trans->c; struct qreg rcp_w = qir_RCP(c, trans->outputs[3]); emit_scaled_viewport_write(trans, rcp_w); emit_zs_write(trans, rcp_w); emit_rcp_wc_write(trans, rcp_w); for (int i = 4; i < trans->num_outputs; i++) { qir_VPM_WRITE(c, trans->outputs[i]); } } static void emit_coord_end(struct tgsi_to_qir *trans) { struct qcompile *c = trans->c; struct qreg rcp_w = qir_RCP(c, trans->outputs[3]); for (int i = 0; i < 4; i++) qir_VPM_WRITE(c, trans->outputs[i]); emit_scaled_viewport_write(trans, rcp_w); emit_zs_write(trans, rcp_w); emit_rcp_wc_write(trans, rcp_w); } static struct tgsi_to_qir * vc4_shader_tgsi_to_qir(struct vc4_compiled_shader *shader, enum qstage stage, struct vc4_key *key) { struct tgsi_to_qir *trans = CALLOC_STRUCT(tgsi_to_qir); struct qcompile *c; int ret; c = qir_compile_init(); c->stage = stage; memset(trans, 0, sizeof(*trans)); /* XXX sizing */ trans->temps = calloc(sizeof(struct qreg), 1024); trans->inputs = calloc(sizeof(struct qreg), 8 * 4); trans->outputs = calloc(sizeof(struct qreg), 1024); trans->uniforms = calloc(sizeof(struct qreg), 1024); trans->consts = calloc(sizeof(struct qreg), 1024); trans->uniform_data = calloc(sizeof(uint32_t), 1024); trans->uniform_contents = calloc(sizeof(enum quniform_contents), 1024); trans->shader_state = key->shader_state; trans->c = c; ret = tgsi_parse_init(&trans->parser, trans->shader_state->tokens); assert(ret == TGSI_PARSE_OK); if (vc4_debug & VC4_DEBUG_TGSI) { fprintf(stderr, "TGSI:\n"); tgsi_dump(trans->shader_state->tokens, 0); } switch (stage) { case QSTAGE_FRAG: trans->fs_key = (struct vc4_fs_key *)key; break; case QSTAGE_VERT: trans->vs_key = (struct vc4_vs_key *)key; break; case QSTAGE_COORD: trans->vs_key = (struct vc4_vs_key *)key; break; } while (!tgsi_parse_end_of_tokens(&trans->parser)) { tgsi_parse_token(&trans->parser); switch (trans->parser.FullToken.Token.Type) { case TGSI_TOKEN_TYPE_DECLARATION: emit_tgsi_declaration(trans, &trans->parser.FullToken.FullDeclaration); break; case TGSI_TOKEN_TYPE_INSTRUCTION: emit_tgsi_instruction(trans, &trans->parser.FullToken.FullInstruction); break; case TGSI_TOKEN_TYPE_IMMEDIATE: parse_tgsi_immediate(trans, &trans->parser.FullToken.FullImmediate); break; } } switch (stage) { case QSTAGE_FRAG: emit_frag_end(trans); break; case QSTAGE_VERT: emit_vert_end(trans); break; case QSTAGE_COORD: emit_coord_end(trans); break; } tgsi_parse_free(&trans->parser); free(trans->temps); qir_optimize(c); if (vc4_debug & VC4_DEBUG_QIR) { fprintf(stderr, "QIR:\n"); qir_dump(c); } vc4_generate_code(c); if (vc4_debug & VC4_DEBUG_SHADERDB) { fprintf(stderr, "SHADER-DB: %s: %d instructions\n", qir_get_stage_name(c->stage), c->qpu_inst_count); fprintf(stderr, "SHADER-DB: %s: %d uniforms\n", qir_get_stage_name(c->stage), trans->num_uniforms); } return trans; } static void * vc4_shader_state_create(struct pipe_context *pctx, const struct pipe_shader_state *cso) { struct pipe_shader_state *so = CALLOC_STRUCT(pipe_shader_state); if (!so) return NULL; so->tokens = tgsi_dup_tokens(cso->tokens); return so; } static void copy_uniform_state_to_shader(struct vc4_compiled_shader *shader, int shader_index, struct tgsi_to_qir *trans) { int count = trans->num_uniforms; struct vc4_shader_uniform_info *uinfo = &shader->uniforms[shader_index]; uinfo->count = count; uinfo->data = malloc(count * sizeof(*uinfo->data)); memcpy(uinfo->data, trans->uniform_data, count * sizeof(*uinfo->data)); uinfo->contents = malloc(count * sizeof(*uinfo->contents)); memcpy(uinfo->contents, trans->uniform_contents, count * sizeof(*uinfo->contents)); uinfo->num_texture_samples = trans->num_texture_samples; } static void vc4_fs_compile(struct vc4_context *vc4, struct vc4_compiled_shader *shader, struct vc4_fs_key *key) { struct tgsi_to_qir *trans = vc4_shader_tgsi_to_qir(shader, QSTAGE_FRAG, &key->base); shader->num_inputs = trans->c->num_inputs; copy_uniform_state_to_shader(shader, 0, trans); shader->bo = vc4_bo_alloc_mem(vc4->screen, trans->c->qpu_insts, trans->c->qpu_inst_count * sizeof(uint64_t), "fs_code"); qir_compile_destroy(trans->c); free(trans); } static void vc4_vs_compile(struct vc4_context *vc4, struct vc4_compiled_shader *shader, struct vc4_vs_key *key) { struct tgsi_to_qir *vs_trans = vc4_shader_tgsi_to_qir(shader, QSTAGE_VERT, &key->base); copy_uniform_state_to_shader(shader, 0, vs_trans); struct tgsi_to_qir *cs_trans = vc4_shader_tgsi_to_qir(shader, QSTAGE_COORD, &key->base); copy_uniform_state_to_shader(shader, 1, cs_trans); uint32_t vs_size = vs_trans->c->qpu_inst_count * sizeof(uint64_t); uint32_t cs_size = cs_trans->c->qpu_inst_count * sizeof(uint64_t); shader->coord_shader_offset = vs_size; /* XXX: alignment? */ shader->bo = vc4_bo_alloc(vc4->screen, shader->coord_shader_offset + cs_size, "vs_code"); void *map = vc4_bo_map(shader->bo); memcpy(map, vs_trans->c->qpu_insts, vs_size); memcpy(map + shader->coord_shader_offset, cs_trans->c->qpu_insts, cs_size); qir_compile_destroy(vs_trans->c); qir_compile_destroy(cs_trans->c); } static void vc4_update_compiled_fs(struct vc4_context *vc4) { struct vc4_fs_key local_key; struct vc4_fs_key *key = &local_key; memset(key, 0, sizeof(*key)); key->base.shader_state = vc4->prog.bind_fs; if (vc4->framebuffer.cbufs[0]) key->color_format = vc4->framebuffer.cbufs[0]->format; key->depth_enabled = vc4->zsa->base.depth.enabled; vc4->prog.fs = util_hash_table_get(vc4->fs_cache, key); if (vc4->prog.fs) return; key = malloc(sizeof(*key)); memcpy(key, &local_key, sizeof(*key)); struct vc4_compiled_shader *shader = CALLOC_STRUCT(vc4_compiled_shader); vc4_fs_compile(vc4, shader, key); util_hash_table_set(vc4->fs_cache, key, shader); vc4->prog.fs = shader; } static void vc4_update_compiled_vs(struct vc4_context *vc4) { struct vc4_vs_key local_key; struct vc4_vs_key *key = &local_key; memset(key, 0, sizeof(*key)); key->base.shader_state = vc4->prog.bind_vs; for (int i = 0; i < ARRAY_SIZE(key->attr_formats); i++) key->attr_formats[i] = vc4->vtx->pipe[i].src_format; vc4->prog.vs = util_hash_table_get(vc4->vs_cache, key); if (vc4->prog.vs) return; key = malloc(sizeof(*key)); memcpy(key, &local_key, sizeof(*key)); struct vc4_compiled_shader *shader = CALLOC_STRUCT(vc4_compiled_shader); vc4_vs_compile(vc4, shader, key); util_hash_table_set(vc4->vs_cache, key, shader); vc4->prog.vs = shader; } void vc4_update_compiled_shaders(struct vc4_context *vc4) { vc4_update_compiled_fs(vc4); vc4_update_compiled_vs(vc4); } static unsigned fs_cache_hash(void *key) { return util_hash_crc32(key, sizeof(struct vc4_fs_key)); } static unsigned vs_cache_hash(void *key) { return util_hash_crc32(key, sizeof(struct vc4_vs_key)); } static int fs_cache_compare(void *key1, void *key2) { return memcmp(key1, key2, sizeof(struct vc4_fs_key)); } static int vs_cache_compare(void *key1, void *key2) { return memcmp(key1, key2, sizeof(struct vc4_vs_key)); } struct delete_state { struct vc4_context *vc4; struct pipe_shader_state *shader_state; }; static enum pipe_error fs_delete_from_cache(void *in_key, void *in_value, void *data) { struct delete_state *del = data; struct vc4_fs_key *key = in_key; struct vc4_compiled_shader *shader = in_value; if (key->base.shader_state == data) { util_hash_table_remove(del->vc4->fs_cache, key); vc4_bo_unreference(&shader->bo); free(shader); } return 0; } static enum pipe_error vs_delete_from_cache(void *in_key, void *in_value, void *data) { struct delete_state *del = data; struct vc4_vs_key *key = in_key; struct vc4_compiled_shader *shader = in_value; if (key->base.shader_state == data) { util_hash_table_remove(del->vc4->vs_cache, key); vc4_bo_unreference(&shader->bo); free(shader); } return 0; } static void vc4_shader_state_delete(struct pipe_context *pctx, void *hwcso) { struct vc4_context *vc4 = vc4_context(pctx); struct pipe_shader_state *so = hwcso; struct delete_state del; del.vc4 = vc4; del.shader_state = so; util_hash_table_foreach(vc4->fs_cache, fs_delete_from_cache, &del); util_hash_table_foreach(vc4->vs_cache, vs_delete_from_cache, &del); free((void *)so->tokens); free(so); } static uint32_t translate_wrap(uint32_t p_wrap) { switch (p_wrap) { case PIPE_TEX_WRAP_REPEAT: return 0; case PIPE_TEX_WRAP_CLAMP: case PIPE_TEX_WRAP_CLAMP_TO_EDGE: return 1; case PIPE_TEX_WRAP_MIRROR_REPEAT: return 2; case PIPE_TEX_WRAP_CLAMP_TO_BORDER: return 3; default: fprintf(stderr, "Unknown wrap mode %d\n", p_wrap); assert(!"not reached"); return 0; } } static void write_texture_p0(struct vc4_context *vc4, struct vc4_texture_stateobj *texstate, uint32_t tex_and_sampler) { uint32_t texi = (tex_and_sampler >> 0) & 0xff; struct pipe_sampler_view *texture = texstate->textures[texi]; struct vc4_resource *rsc = vc4_resource(texture->texture); cl_reloc(vc4, &vc4->uniforms, rsc->bo, texture->u.tex.last_level); } static void write_texture_p1(struct vc4_context *vc4, struct vc4_texture_stateobj *texstate, uint32_t tex_and_sampler) { uint32_t texi = (tex_and_sampler >> 0) & 0xff; uint32_t sampi = (tex_and_sampler >> 8) & 0xff; struct pipe_sampler_view *texture = texstate->textures[texi]; struct pipe_sampler_state *sampler = texstate->samplers[sampi]; static const uint32_t mipfilter_map[] = { [PIPE_TEX_MIPFILTER_NEAREST] = 2, [PIPE_TEX_MIPFILTER_LINEAR] = 4, [PIPE_TEX_MIPFILTER_NONE] = 0 }; static const uint32_t imgfilter_map[] = { [PIPE_TEX_FILTER_NEAREST] = 1, [PIPE_TEX_FILTER_LINEAR] = 0, }; cl_u32(&vc4->uniforms, (1 << 31) /* XXX: data type */| (texture->texture->height0 << 20) | (texture->texture->width0 << 8) | (imgfilter_map[sampler->mag_img_filter] << 7) | ((imgfilter_map[sampler->min_img_filter] + mipfilter_map[sampler->min_mip_filter]) << 4) | (translate_wrap(sampler->wrap_t) << 2) | (translate_wrap(sampler->wrap_s) << 0)); } static uint32_t get_texrect_scale(struct vc4_texture_stateobj *texstate, enum quniform_contents contents, uint32_t data) { struct pipe_sampler_view *texture = texstate->textures[data]; uint32_t dim; if (contents == QUNIFORM_TEXRECT_SCALE_X) dim = texture->texture->width0; else dim = texture->texture->height0; return fui(1.0f / dim); } void vc4_write_uniforms(struct vc4_context *vc4, struct vc4_compiled_shader *shader, struct vc4_constbuf_stateobj *cb, struct vc4_texture_stateobj *texstate, int shader_index) { struct vc4_shader_uniform_info *uinfo = &shader->uniforms[shader_index]; const uint32_t *gallium_uniforms = cb->cb[0].user_buffer; cl_start_shader_reloc(&vc4->uniforms, uinfo->num_texture_samples); for (int i = 0; i < uinfo->count; i++) { switch (uinfo->contents[i]) { case QUNIFORM_CONSTANT: cl_u32(&vc4->uniforms, uinfo->data[i]); break; case QUNIFORM_UNIFORM: cl_u32(&vc4->uniforms, gallium_uniforms[uinfo->data[i]]); break; case QUNIFORM_VIEWPORT_X_SCALE: cl_u32(&vc4->uniforms, fui(vc4->framebuffer.width * 16.0f / 2.0f)); break; case QUNIFORM_VIEWPORT_Y_SCALE: cl_u32(&vc4->uniforms, fui(vc4->framebuffer.height * -16.0f / 2.0f)); break; case QUNIFORM_VIEWPORT_Z_OFFSET: cl_u32(&vc4->uniforms, fui(vc4->viewport.translate[2])); break; case QUNIFORM_VIEWPORT_Z_SCALE: cl_u32(&vc4->uniforms, fui(vc4->viewport.scale[2])); break; case QUNIFORM_TEXTURE_CONFIG_P0: write_texture_p0(vc4, texstate, uinfo->data[i]); break; case QUNIFORM_TEXTURE_CONFIG_P1: write_texture_p1(vc4, texstate, uinfo->data[i]); break; case QUNIFORM_TEXRECT_SCALE_X: case QUNIFORM_TEXRECT_SCALE_Y: cl_u32(&vc4->uniforms, get_texrect_scale(texstate, uinfo->contents[i], uinfo->data[i])); break; } #if 0 uint32_t written_val = *(uint32_t *)(vc4->uniforms.next - 4); fprintf(stderr, "%p/%d: %d: 0x%08x (%f)\n", shader, shader_index, i, written_val, uif(written_val)); #endif } } static void vc4_fp_state_bind(struct pipe_context *pctx, void *hwcso) { struct vc4_context *vc4 = vc4_context(pctx); vc4->prog.bind_fs = hwcso; vc4->prog.dirty |= VC4_SHADER_DIRTY_FP; vc4->dirty |= VC4_DIRTY_PROG; } static void vc4_vp_state_bind(struct pipe_context *pctx, void *hwcso) { struct vc4_context *vc4 = vc4_context(pctx); vc4->prog.bind_vs = hwcso; vc4->prog.dirty |= VC4_SHADER_DIRTY_VP; vc4->dirty |= VC4_DIRTY_PROG; } void vc4_program_init(struct pipe_context *pctx) { struct vc4_context *vc4 = vc4_context(pctx); pctx->create_vs_state = vc4_shader_state_create; pctx->delete_vs_state = vc4_shader_state_delete; pctx->create_fs_state = vc4_shader_state_create; pctx->delete_fs_state = vc4_shader_state_delete; pctx->bind_fs_state = vc4_fp_state_bind; pctx->bind_vs_state = vc4_vp_state_bind; vc4->fs_cache = util_hash_table_create(fs_cache_hash, fs_cache_compare); vc4->vs_cache = util_hash_table_create(vs_cache_hash, vs_cache_compare); }