From 6d874d0ee18b3694c49e0206fa519bd8b746ec24 Mon Sep 17 00:00:00 2001 From: Eric Anholt Date: Tue, 8 Nov 2011 19:27:46 -0800 Subject: [PATCH] i965/fs: Add support for user-defined out variables. Before, I was tracking the ir_variable * found for gl_FragColor or gl_FragData[]. Instead, when visiting those variables, set up an array of per-render-target fs_regs to copy the output data from. This cleans up the color emit path, while making handling of multiple user-defined out variables easier. v2: incorporate idr's feedback about ir->location (changes by Kenneth Graunke) Reviewed-by: Kenneth Graunke --- src/mesa/drivers/dri/i965/brw_fs.h | 8 +-- src/mesa/drivers/dri/i965/brw_fs_visitor.cpp | 66 +++++++++++--------- 2 files changed, 39 insertions(+), 35 deletions(-) diff --git a/src/mesa/drivers/dri/i965/brw_fs.h b/src/mesa/drivers/dri/i965/brw_fs.h index 3e4503028b1..de3164457c8 100644 --- a/src/mesa/drivers/dri/i965/brw_fs.h +++ b/src/mesa/drivers/dri/i965/brw_fs.h @@ -376,9 +376,8 @@ public: else this->reg_null_cmp = reg_null_f; - this->frag_color = NULL; - this->frag_data = NULL; this->frag_depth = NULL; + memset(this->outputs, 0, sizeof(this->outputs)); this->first_non_payload_grf = 0; this->current_annotation = NULL; @@ -533,7 +532,7 @@ public: void emit_if_gen6(ir_if *ir); void emit_unspill(fs_inst *inst, fs_reg reg, uint32_t spill_offset); - void emit_color_write(int index, int first_color_mrf, fs_reg color); + void emit_color_write(int target, int index, int first_color_mrf); void emit_fb_writes(); bool try_rewrite_rhs_to_dst(ir_assignment *ir, fs_reg dst, @@ -581,7 +580,8 @@ public: int *params_remap; struct hash_table *variable_ht; - ir_variable *frag_color, *frag_data, *frag_depth; + ir_variable *frag_depth; + fs_reg outputs[BRW_MAX_DRAW_BUFFERS]; int first_non_payload_grf; int urb_setup[FRAG_ATTRIB_MAX]; bool kill_emitted; diff --git a/src/mesa/drivers/dri/i965/brw_fs_visitor.cpp b/src/mesa/drivers/dri/i965/brw_fs_visitor.cpp index 3e2feafec38..a76ba967d44 100644 --- a/src/mesa/drivers/dri/i965/brw_fs_visitor.cpp +++ b/src/mesa/drivers/dri/i965/brw_fs_visitor.cpp @@ -58,14 +58,6 @@ fs_visitor::visit(ir_variable *ir) if (variable_storage(ir)) return; - if (strcmp(ir->name, "gl_FragColor") == 0) { - this->frag_color = ir; - } else if (strcmp(ir->name, "gl_FragData") == 0) { - this->frag_data = ir; - } else if (strcmp(ir->name, "gl_FragDepth") == 0) { - this->frag_depth = ir; - } - if (ir->mode == ir_var_in) { if (!strcmp(ir->name, "gl_FragCoord")) { reg = emit_fragcoord_interpolation(ir); @@ -77,9 +69,29 @@ fs_visitor::visit(ir_variable *ir) assert(reg); hash_table_insert(this->variable_ht, reg, ir); return; - } + } else if (ir->mode == ir_var_out) { + reg = new(this->mem_ctx) fs_reg(this, ir->type); - if (ir->mode == ir_var_uniform) { + if (ir->location == FRAG_RESULT_COLOR) { + /* Writing gl_FragColor outputs to all color regions. */ + for (int i = 0; i < c->key.nr_color_regions; i++) { + this->outputs[i] = *reg; + } + } else if (ir->location == FRAG_RESULT_DEPTH) { + this->frag_depth = ir; + } else { + /* gl_FragData or a user-defined FS output */ + assert(ir->location >= FRAG_RESULT_DATA0 && + ir->location < FRAG_RESULT_DATA0 + BRW_MAX_DRAW_BUFFERS); + + /* General color output. */ + for (unsigned int i = 0; i < MAX2(1, ir->type->length); i++) { + int output = ir->location - FRAG_RESULT_DATA0 + i; + this->outputs[output] = *reg; + this->outputs[output].reg_offset += 4 * i; + } + } + } else if (ir->mode == ir_var_uniform) { int param_index = c->prog_data.nr_params; if (c->dispatch_width == 16) { @@ -1830,10 +1842,18 @@ fs_visitor::emit_interpolation_setup_gen6() } void -fs_visitor::emit_color_write(int index, int first_color_mrf, fs_reg color) +fs_visitor::emit_color_write(int target, int index, int first_color_mrf) { int reg_width = c->dispatch_width / 8; fs_inst *inst; + fs_reg color = outputs[target]; + fs_reg mrf; + + /* If there's no color data to be written, skip it. */ + if (color.file == BAD_FILE) + return; + + color.reg_offset += index; if (c->dispatch_width == 8 || intel->gen >= 6) { /* SIMD8 write looks like: @@ -1959,27 +1979,12 @@ fs_visitor::emit_fb_writes() nr += reg_width; } - fs_reg color = reg_undef; - if (this->frag_color) - color = *(variable_storage(this->frag_color)); - else if (this->frag_data) { - color = *(variable_storage(this->frag_data)); - color.type = BRW_REGISTER_TYPE_F; - } - for (int target = 0; target < c->key.nr_color_regions; target++) { this->current_annotation = ralloc_asprintf(this->mem_ctx, "FB write target %d", target); - if (this->frag_color || this->frag_data) { - for (int i = 0; i < 4; i++) { - emit_color_write(i, color_mrf, color); - color.reg_offset++; - } - } - - if (this->frag_color) - color.reg_offset -= 4; + for (int i = 0; i < 4; i++) + emit_color_write(target, i, color_mrf); fs_inst *inst = emit(FS_OPCODE_FB_WRITE); inst->target = target; @@ -1991,13 +1996,12 @@ fs_visitor::emit_fb_writes() } if (c->key.nr_color_regions == 0) { - if (c->key.alpha_test && (this->frag_color || this->frag_data)) { + if (c->key.alpha_test) { /* If the alpha test is enabled but there's no color buffer, * we still need to send alpha out the pipeline to our null * renderbuffer. */ - color.reg_offset += 3; - emit_color_write(3, color_mrf, color); + emit_color_write(0, 3, color_mrf); } fs_inst *inst = emit(FS_OPCODE_FB_WRITE);