rtl/gfx: implement fragment address linearization

5 files changed, 143 insertions, 1 deletions

diff --git a/rtl/gfx/gfx.sv b/rtl/gfx/gfx.sv
index f973b73..bd797df 100644
--- a/rtl/gfx/gfx.sv
+++ b/rtl/gfx/gfx.sv
@@ -42,6 +42,24 @@ module gfx
 		.*
 	);
 
+	logic raster_ready;
+	paint_lanes raster_valid;
+	frag_xy_lanes fragments;
+
+	gfx_raster raster
+	(
+		.in_ready(raster_ready),
+		.in_valid(0), //TODO
+		.out_ready(frag_ready),
+		.out_valid(raster_valid),
+
+		.vertex_a(), //TODO
+		.vertex_b(), //TODO
+		.vertex_c(), //TODO
+
+		.*
+	);
+
 	logic frag_mask_set, frag_mask_write, frag_wait;
 	linear_coord frag_mask_write_addr;
 
@@ -52,6 +70,20 @@ module gfx
 		.*
 	);
 
+	logic frag_ready, frag_valid;
+	frag_paint frag_out;
+
+	gfx_frag frag
+	(
+		.in_ready(frag_ready),
+		.in_valid(raster_valid),
+		.out_ready(1), //TODO
+		.out_valid(frag_valid),
+
+		.out(frag_out),
+		.*
+	);
+
 	logic scanout_read_tmp, vsync;
 	linear_coord scan_mask_addr;
 
diff --git a/rtl/gfx/gfx_defs.sv b/rtl/gfx/gfx_defs.sv
index 60962e1..cc6e6f3 100644
--- a/rtl/gfx/gfx_defs.sv
+++ b/rtl/gfx/gfx_defs.sv
@@ -48,6 +48,11 @@ typedef struct packed
 	logic[9:0] r, g, b;
 } rgb30;
 
+typedef struct packed
+{
+	logic[7:0] a, r, g, b;
+} rgb32;
+
 `define FIXED_FRAC 16
 
 `define FIXED_FMA_STAGES     5
@@ -115,4 +120,12 @@ typedef struct packed
 typedef frag_xy[`GFX_FINE_LANES - 1:0] frag_xy_lanes;
 typedef logic[`GFX_FINE_LANES - 1:0]   paint_lanes;
 
+typedef struct packed
+{
+	linear_coord addr;
+	rgb32        color;
+} frag_paint;
+
+`define GFX_FRAG_ADDR_STAGES 3
+
 `endif
diff --git a/rtl/gfx/gfx_frag.sv b/rtl/gfx/gfx_frag.sv
new file mode 100644
index 0000000..21f111f
--- /dev/null
+++ b/rtl/gfx/gfx_frag.sv
@@ -0,0 +1,38 @@
+`include "gfx/gfx_defs.sv"
+
+module gfx_frag
+(
+	input  logic         clk,
+	                     rst_n,
+
+	input  frag_xy_lanes fragments,
+	input  paint_lanes   in_valid,
+	output logic         in_ready,
+
+	input  logic         out_ready,
+	output logic         out_valid,
+	output frag_paint    out
+);
+
+	logic addr_stall;
+
+	gfx_pipeline_flow #(.STAGES(`GFX_FRAG_ADDR_STAGES)) addr_flow
+	(
+		.stall(addr_stall),
+		.in_valid(0),
+		.in_ready(),
+		.out_ready(1),
+		.out_valid(),
+		.*
+	);
+
+	gfx_frag_addr addr
+	(
+		.stall(addr_stall),
+
+		.frag(),
+		.linear(),
+		.*
+	);
+
+endmodule
diff --git a/rtl/gfx/gfx_frag_addr.sv b/rtl/gfx/gfx_frag_addr.sv
new file mode 100644
index 0000000..9b817cb
--- /dev/null
+++ b/rtl/gfx/gfx_frag_addr.sv
@@ -0,0 +1,59 @@
+`include "gfx/gfx_defs.sv"
+
+module gfx_frag_addr
+(
+	input  logic        clk,
+
+	input  frag_xy      frag,
+	input  logic        stall,
+
+	output linear_coord linear
+);
+
+	/* frag está expresado en un rango normalizado con igual distribución
+	 * entre positivos y negativos. Para obtener la dirección lineal que le
+	 * corresponde, debemos corregir esto para que el mínimo sea cero en
+	 * cada coordenada. Luego de eso,
+	 *
+	 *   linear = y_corregido * `GFX_X_RES + x_corregido
+	 *
+	 * Afortunadamente, esto es trivializable y no necesita FMA:
+	 *
+	 *   y * `GFX_X_RES + x
+	 * = y * 640 + x
+	 * = y * 128 * 5 + x
+	 * = ((y * 5) << 7) + x
+	 * = ((y * (4 + 1)) << 7) + x
+	 * = (((y << 2) + y) << 7) + x
+	 * = (y << 9) + (y << 7) + x
+	 *
+	 * Para corregir x ([-320, 319]) se le suma `GFX_RES_X / 2.
+	 *
+	 * Para corregir y ([-240, 239]) se debe tomar en cuenta que las
+	 * direcciones lineales incrementan hacia abajo, así que:
+	 *   y_corregido = `GFX_RES_Y / 2 - 1 - y
+	 */
+
+	localparam ZERO_PAD = $bits(linear_coord) - $bits(xy_coord);
+
+	// Estas constantes asumen `GFX_X_RES == 640
+	localparam Y_SHIFT0 = 9, Y_SHIFT1 = 7;
+
+	xy_coord bias_x, bias_y;
+	linear_coord row_start, x_biased, x_hold, y_biased;
+
+	assign bias_x = `GFX_X_RES / 2;
+	assign bias_y = `GFX_Y_RES / 2 - 1;
+
+	always_ff @(posedge clk)
+		if (!stall) begin
+			x_biased <= {{ZERO_PAD{1'b0}}, frag.x + bias_x};
+			y_biased <= {{ZERO_PAD{1'b0}}, bias_y - frag.y};
+
+			x_hold <= x_biased;
+			row_start <= (y_biased << Y_SHIFT0) + (y_biased << Y_SHIFT1);
+
+			linear <= row_start + x_hold;
+		end
+
+endmodule
diff --git a/rtl/gfx/gfx_raster.sv b/rtl/gfx/gfx_raster.sv
index c61cbb5..a51c4bb 100644
--- a/rtl/gfx/gfx_raster.sv
+++ b/rtl/gfx/gfx_raster.sv
@@ -6,7 +6,7 @@ module gfx_raster
 	                     rst_n,
 
 	input  raster_xy     vertex_a,
-				         vertex_b,
+	                     vertex_b,
 	                     vertex_c,
 	input  logic         in_valid,
 	output logic         in_ready,