1 files changed, 312 insertions, 0 deletions
diff --git a/src/gallium/drivers/radeonsi/si_state_streamout.c b/src/gallium/drivers/radeonsi/si_state_streamout.c
new file mode 100644
index 00000000000..42a83d4bd71
--- /dev/null
+++ b/src/gallium/drivers/radeonsi/si_state_streamout.c
@@ -0,0 +1,312 @@
+/*
+ * Copyright 2013 Advanced Micro Devices, Inc.
+ *
+ * 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.
+ *
+ * Authors: Marek Olšák <[email protected]>
+ *
+ */
+
+#include "si_pipe.h"
+#include "si_state.h"
+#include "radeon/r600_cs.h"
+
+#include "util/u_memory.h"
+
+static void si_set_streamout_enable(struct si_context *sctx, bool enable);
+
+static struct pipe_stream_output_target *
+si_create_so_target(struct pipe_context *ctx,
+		    struct pipe_resource *buffer,
+		    unsigned buffer_offset,
+		    unsigned buffer_size)
+{
+	struct si_context *sctx = (struct si_context *)ctx;
+	struct si_streamout_target *t;
+	struct r600_resource *rbuffer = (struct r600_resource*)buffer;
+
+	t = CALLOC_STRUCT(si_streamout_target);
+	if (!t) {
+		return NULL;
+	}
+
+	u_suballocator_alloc(sctx->b.allocator_zeroed_memory, 4, 4,
+			     &t->buf_filled_size_offset,
+			     (struct pipe_resource**)&t->buf_filled_size);
+	if (!t->buf_filled_size) {
+		FREE(t);
+		return NULL;
+	}
+
+	t->b.reference.count = 1;
+	t->b.context = ctx;
+	pipe_resource_reference(&t->b.buffer, buffer);
+	t->b.buffer_offset = buffer_offset;
+	t->b.buffer_size = buffer_size;
+
+	util_range_add(&rbuffer->valid_buffer_range, buffer_offset,
+		       buffer_offset + buffer_size);
+	return &t->b;
+}
+
+static void si_so_target_destroy(struct pipe_context *ctx,
+				 struct pipe_stream_output_target *target)
+{
+	struct si_streamout_target *t = (struct si_streamout_target*)target;
+	pipe_resource_reference(&t->b.buffer, NULL);
+	r600_resource_reference(&t->buf_filled_size, NULL);
+	FREE(t);
+}
+
+void si_streamout_buffers_dirty(struct si_context *sctx)
+{
+	if (!sctx->streamout.enabled_mask)
+		return;
+
+	si_mark_atom_dirty(sctx, &sctx->streamout.begin_atom);
+	si_set_streamout_enable(sctx, true);
+}
+
+void si_common_set_streamout_targets(struct pipe_context *ctx,
+				     unsigned num_targets,
+				     struct pipe_stream_output_target **targets,
+				     const unsigned *offsets)
+{
+	struct si_context *sctx = (struct si_context *)ctx;
+	unsigned i;
+        unsigned enabled_mask = 0, append_bitmask = 0;
+
+	/* Stop streamout. */
+	if (sctx->streamout.num_targets && sctx->streamout.begin_emitted) {
+		si_emit_streamout_end(sctx);
+	}
+
+	/* Set the new targets. */
+	for (i = 0; i < num_targets; i++) {
+		pipe_so_target_reference((struct pipe_stream_output_target**)&sctx->streamout.targets[i], targets[i]);
+		if (!targets[i])
+			continue;
+
+		r600_context_add_resource_size(ctx, targets[i]->buffer);
+		enabled_mask |= 1 << i;
+		if (offsets[i] == ((unsigned)-1))
+			append_bitmask |= 1 << i;
+	}
+	for (; i < sctx->streamout.num_targets; i++) {
+		pipe_so_target_reference((struct pipe_stream_output_target**)&sctx->streamout.targets[i], NULL);
+	}
+
+	sctx->streamout.enabled_mask = enabled_mask;
+
+	sctx->streamout.num_targets = num_targets;
+	sctx->streamout.append_bitmask = append_bitmask;
+
+	if (num_targets) {
+		si_streamout_buffers_dirty(sctx);
+	} else {
+		si_set_atom_dirty(sctx, &sctx->streamout.begin_atom, false);
+		si_set_streamout_enable(sctx, false);
+	}
+}
+
+static void si_flush_vgt_streamout(struct si_context *sctx)
+{
+	struct radeon_winsys_cs *cs = sctx->b.gfx.cs;
+	unsigned reg_strmout_cntl;
+
+	/* The register is at different places on different ASICs. */
+	if (sctx->b.chip_class >= CIK) {
+		reg_strmout_cntl = R_0300FC_CP_STRMOUT_CNTL;
+		radeon_set_uconfig_reg(cs, reg_strmout_cntl, 0);
+	} else {
+		reg_strmout_cntl = R_0084FC_CP_STRMOUT_CNTL;
+		radeon_set_config_reg(cs, reg_strmout_cntl, 0);
+	}
+
+	radeon_emit(cs, PKT3(PKT3_EVENT_WRITE, 0, 0));
+	radeon_emit(cs, EVENT_TYPE(EVENT_TYPE_SO_VGTSTREAMOUT_FLUSH) | EVENT_INDEX(0));
+
+	radeon_emit(cs, PKT3(PKT3_WAIT_REG_MEM, 5, 0));
+	radeon_emit(cs, WAIT_REG_MEM_EQUAL); /* wait until the register is equal to the reference value */
+	radeon_emit(cs, reg_strmout_cntl >> 2);  /* register */
+	radeon_emit(cs, 0);
+	radeon_emit(cs, S_008490_OFFSET_UPDATE_DONE(1)); /* reference value */
+	radeon_emit(cs, S_008490_OFFSET_UPDATE_DONE(1)); /* mask */
+	radeon_emit(cs, 4); /* poll interval */
+}
+
+static void si_emit_streamout_begin(struct r600_common_context *rctx, struct r600_atom *atom)
+{
+	struct si_context *sctx = (struct si_context*)rctx;
+	struct radeon_winsys_cs *cs = sctx->b.gfx.cs;
+	struct si_streamout_target **t = sctx->streamout.targets;
+	uint16_t *stride_in_dw = sctx->streamout.stride_in_dw;
+	unsigned i;
+
+	si_flush_vgt_streamout(sctx);
+
+	for (i = 0; i < sctx->streamout.num_targets; i++) {
+		if (!t[i])
+			continue;
+
+		t[i]->stride_in_dw = stride_in_dw[i];
+
+		/* SI binds streamout buffers as shader resources.
+		 * VGT only counts primitives and tells the shader
+		 * through SGPRs what to do. */
+		radeon_set_context_reg_seq(cs, R_028AD0_VGT_STRMOUT_BUFFER_SIZE_0 + 16*i, 2);
+		radeon_emit(cs, (t[i]->b.buffer_offset +
+				 t[i]->b.buffer_size) >> 2);	/* BUFFER_SIZE (in DW) */
+		radeon_emit(cs, stride_in_dw[i]);		/* VTX_STRIDE (in DW) */
+
+		if (sctx->streamout.append_bitmask & (1 << i) && t[i]->buf_filled_size_valid) {
+			uint64_t va = t[i]->buf_filled_size->gpu_address +
+				      t[i]->buf_filled_size_offset;
+
+			/* Append. */
+			radeon_emit(cs, PKT3(PKT3_STRMOUT_BUFFER_UPDATE, 4, 0));
+			radeon_emit(cs, STRMOUT_SELECT_BUFFER(i) |
+				    STRMOUT_OFFSET_SOURCE(STRMOUT_OFFSET_FROM_MEM)); /* control */
+			radeon_emit(cs, 0); /* unused */
+			radeon_emit(cs, 0); /* unused */
+			radeon_emit(cs, va); /* src address lo */
+			radeon_emit(cs, va >> 32); /* src address hi */
+
+			r600_emit_reloc(&sctx->b,  &sctx->b.gfx, t[i]->buf_filled_size,
+					RADEON_USAGE_READ, RADEON_PRIO_SO_FILLED_SIZE);
+		} else {
+			/* Start from the beginning. */
+			radeon_emit(cs, PKT3(PKT3_STRMOUT_BUFFER_UPDATE, 4, 0));
+			radeon_emit(cs, STRMOUT_SELECT_BUFFER(i) |
+				    STRMOUT_OFFSET_SOURCE(STRMOUT_OFFSET_FROM_PACKET)); /* control */
+			radeon_emit(cs, 0); /* unused */
+			radeon_emit(cs, 0); /* unused */
+			radeon_emit(cs, t[i]->b.buffer_offset >> 2); /* buffer offset in DW */
+			radeon_emit(cs, 0); /* unused */
+		}
+	}
+
+	sctx->streamout.begin_emitted = true;
+}
+
+void si_emit_streamout_end(struct si_context *sctx)
+{
+	struct radeon_winsys_cs *cs = sctx->b.gfx.cs;
+	struct si_streamout_target **t = sctx->streamout.targets;
+	unsigned i;
+	uint64_t va;
+
+	si_flush_vgt_streamout(sctx);
+
+	for (i = 0; i < sctx->streamout.num_targets; i++) {
+		if (!t[i])
+			continue;
+
+		va = t[i]->buf_filled_size->gpu_address + t[i]->buf_filled_size_offset;
+		radeon_emit(cs, PKT3(PKT3_STRMOUT_BUFFER_UPDATE, 4, 0));
+		radeon_emit(cs, STRMOUT_SELECT_BUFFER(i) |
+			    STRMOUT_OFFSET_SOURCE(STRMOUT_OFFSET_NONE) |
+			    STRMOUT_STORE_BUFFER_FILLED_SIZE); /* control */
+		radeon_emit(cs, va);     /* dst address lo */
+		radeon_emit(cs, va >> 32); /* dst address hi */
+		radeon_emit(cs, 0); /* unused */
+		radeon_emit(cs, 0); /* unused */
+
+		r600_emit_reloc(&sctx->b,  &sctx->b.gfx, t[i]->buf_filled_size,
+				RADEON_USAGE_WRITE, RADEON_PRIO_SO_FILLED_SIZE);
+
+		/* Zero the buffer size. The counters (primitives generated,
+		 * primitives emitted) may be enabled even if there is not
+		 * buffer bound. This ensures that the primitives-emitted query
+		 * won't increment. */
+		radeon_set_context_reg(cs, R_028AD0_VGT_STRMOUT_BUFFER_SIZE_0 + 16*i, 0);
+
+		t[i]->buf_filled_size_valid = true;
+	}
+
+	sctx->streamout.begin_emitted = false;
+	sctx->b.flags |= R600_CONTEXT_STREAMOUT_FLUSH;
+}
+
+/* STREAMOUT CONFIG DERIVED STATE
+ *
+ * Streamout must be enabled for the PRIMITIVES_GENERATED query to work.
+ * The buffer mask is an independent state, so no writes occur if there
+ * are no buffers bound.
+ */
+
+static void si_emit_streamout_enable(struct r600_common_context *rctx,
+				     struct r600_atom *atom)
+{
+	struct si_context *sctx = (struct si_context*)rctx;
+
+	radeon_set_context_reg_seq(sctx->b.gfx.cs, R_028B94_VGT_STRMOUT_CONFIG, 2);
+	radeon_emit(sctx->b.gfx.cs,
+		    S_028B94_STREAMOUT_0_EN(si_get_strmout_en(sctx)) |
+		    S_028B94_RAST_STREAM(0) |
+		    S_028B94_STREAMOUT_1_EN(si_get_strmout_en(sctx)) |
+		    S_028B94_STREAMOUT_2_EN(si_get_strmout_en(sctx)) |
+		    S_028B94_STREAMOUT_3_EN(si_get_strmout_en(sctx)));
+	radeon_emit(sctx->b.gfx.cs,
+		    sctx->streamout.hw_enabled_mask &
+		    sctx->streamout.enabled_stream_buffers_mask);
+}
+
+static void si_set_streamout_enable(struct si_context *sctx, bool enable)
+{
+	bool old_strmout_en = si_get_strmout_en(sctx);
+	unsigned old_hw_enabled_mask = sctx->streamout.hw_enabled_mask;
+
+	sctx->streamout.streamout_enabled = enable;
+
+	sctx->streamout.hw_enabled_mask = sctx->streamout.enabled_mask |
+					  (sctx->streamout.enabled_mask << 4) |
+					  (sctx->streamout.enabled_mask << 8) |
+					  (sctx->streamout.enabled_mask << 12);
+
+	if ((old_strmout_en != si_get_strmout_en(sctx)) ||
+            (old_hw_enabled_mask != sctx->streamout.hw_enabled_mask))
+		si_mark_atom_dirty(sctx, &sctx->streamout.enable_atom);
+}
+
+void si_update_prims_generated_query_state(struct si_context *sctx,
+					   unsigned type, int diff)
+{
+	if (type == PIPE_QUERY_PRIMITIVES_GENERATED) {
+		bool old_strmout_en = si_get_strmout_en(sctx);
+
+		sctx->streamout.num_prims_gen_queries += diff;
+		assert(sctx->streamout.num_prims_gen_queries >= 0);
+
+		sctx->streamout.prims_gen_query_enabled =
+			sctx->streamout.num_prims_gen_queries != 0;
+
+		if (old_strmout_en != si_get_strmout_en(sctx))
+			si_mark_atom_dirty(sctx, &sctx->streamout.enable_atom);
+	}
+}
+
+void si_init_streamout_functions(struct si_context *sctx)
+{
+	sctx->b.b.create_stream_output_target = si_create_so_target;
+	sctx->b.b.stream_output_target_destroy = si_so_target_destroy;
+	sctx->streamout.begin_atom.emit = si_emit_streamout_begin;
+	sctx->streamout.enable_atom.emit = si_emit_streamout_enable;
+}