GS/HW: Tweak AF shader

Author: TheLastRar

bin/resources/shaders/dx11/tfx.fx

@@ -264,7 +264,7 @@ float manual_lod(float uv_w)
 float4 sample_c_af(float2 uv, float uv_w)
 {
 	// HW sampler will reject bad UVs, match that here.
-	uv = any(isnan(uv) | isinf(uv)) ? float2(0, 0) : uv;
+	uv = any(isnan(uv) | isinf(uv)) ? float2(0.0f, 0.0f) : uv;
 
 	// Large floating point values risk NaN/Inf values.
 	// Above this value floats lose decimal precision, so seems a resonable limit for UVs.
@@ -277,119 +277,155 @@ float4 sample_c_af(float2 uv, float uv_w)
 	float2 dX = ddx(uv) * sz;
 	float2 dY = ddy(uv) * sz;
 
-	// Calculate Ellipse Transform
-	bool d_zero = length(dX) == 0 || length(dY) == 0;
-	bool d_par = (dX.x * dY.y - dY.x * dX.y) == 0;
-	bool d_per = dot(dX, dY) == 0;
-	bool d_inf_nan = any(isinf(dX) | isinf(dY) | isnan(dX) | isnan(dY));
+	// Check if sample is magnification and if so, skip anisotropic calculations.
+	// Avoid performing square root by checking squared length.
+	float squared_length_x = dot(dX, dX);
+	float squared_length_y = dot(dY, dY);
 
-	if (!(d_zero || d_par || d_per || d_inf_nan))
+	[branch]
+	if (max(squared_length_x, squared_length_y) < (0.75f * 0.75f))
 	{
-		float A = dX.y * dX.y + dY.y * dY.y;
-		float B = -2 * (dX.x * dX.y + dY.x * dY.y);
-		float C = dX.x * dX.x + dY.x * dY.x;
-		float f = (dX.x * dY.y - dY.x * dX.y);
-		float F = f * f;
-
-		float p = A - C;
-		float q = A + C;
-		float t = sqrt(p * p + B * B);
-
-		float2 new_dX = float2(
-			sqrt(F * (t + p) / (t * (q + t))),
-			sqrt(F * (t - p) / (t * (q + t))) * sign(B)
-		);
-		
-		float2 new_dY = float2(
-			sqrt(F * (t - p) / (t * (q - t))) * -sign(B),
-			sqrt(F * (t + p) / (t * (q - t)))
-		);
-		
-		d_inf_nan = any(isinf(new_dX) | isinf(new_dY) | isnan(new_dX) | isnan(new_dY));
-		if (!d_inf_nan)
-		{
-			dX = new_dX;
-			dY = new_dY;
-		}
-	}
-
-	// Compute AF values
-	float squared_length_x = dX.x * dX.x + dX.y * dX.y;
-	float squared_length_y = dY.x * dY.x + dY.y * dY.y;
-	float determinant = abs(dX.x * dY.y - dX.y * dY.x);
-	bool is_major_x = squared_length_x > squared_length_y;
-	float squared_length_major = is_major_x ? squared_length_x : squared_length_y;
-	float length_major = sqrt(squared_length_major);
-
-	float aniso_ratio;
-	float length_lod;
-	float2 aniso_line;
-	if (length_major <= 1.0f)
-	{
-		// A zero length_major would result in NaN Lod and break sampling.
-		// A small length_major would result in aniso_ratio getting clamped to 1.
-		// Perform isotropic filtering instead.
-		aniso_ratio = 1.0f;
-		length_lod = length_major;
-		aniso_line = float2(0, 0);
+#if PS_AUTOMATIC_LOD == 1
+		return Texture.Sample(TextureSampler, uv);
+#else
+#if PS_MANUAL_LOD == 1
+		float lod = manual_lod(uv_w);
+#else
+		float lod = 0.0f; // No Lod
+#endif
+		return Texture.SampleLevel(TextureSampler, uv, lod);
+#endif
 	}
 	else
 	{
-		float norm_major = 1.0f / length_major;
-	
-		float2 aniso_line_dir = float2(
-			(is_major_x ? dX.x : dY.x) * norm_major,
-			(is_major_x ? dX.y : dY.y) * norm_major
-		);
-	
-		aniso_ratio = squared_length_major / determinant;
+		// Calculate Ellipse Transform
+		bool d_zero = squared_length_x < 0.001f || squared_length_y < 0.001f;
+		float f = (dX.x * dY.y - dX.y * dY.x);
+		bool d_par = f == 0.0f < 0.001f;
+		bool d_per = dot(dX, dY) == 0.0f < 0.001f;
+		bool d_inf_nan = any(isinf(dX) | isinf(dY) | isnan(dX) | isnan(dY));
+
+		if (!(d_zero || d_par || d_per || d_inf_nan))
+		{
+			float2 dXYy = float2(dX.y, dY.y);
+			float2 dXYx = float2(dX.x, dY.x);
+			float A = dot(dXYy, dXYy);
+			float B = -2.0f * dot(dXYx, dXYy);
+			float C = dot(dXYx, dXYx);
+			float F = f * f;
+
+			float p = A - C;
+			float q = A + C;
+			float t = sqrt(p * p + B * B);
+
+			float sqrt_num_plus  = sqrt(F * (t + p));
+			float sqrt_num_minus = sqrt(F * (t - p));
+
+			float inv_sqrt_denom_plus  = rsqrt(t * (q + t));
+			float inv_sqrt_denom_minus = rsqrt(t * (q - t));
+
+			float signB = sign(B);
+
+			float2 new_dX = float2(
+				sqrt_num_plus  * inv_sqrt_denom_plus,
+				sqrt_num_minus * inv_sqrt_denom_plus * signB
+			);
+
+			float2 new_dY = float2(
+				sqrt_num_minus * inv_sqrt_denom_minus * -signB,
+				sqrt_num_plus  * inv_sqrt_denom_minus
+			);
+
+			d_inf_nan = any(isinf(new_dX) | isinf(new_dY) | isnan(new_dX) | isnan(new_dY));
+			if (!d_inf_nan)
+			{
+				dX = new_dX;
+				dY = new_dY;
+				squared_length_x = dot(dX, dX);
+				squared_length_y = dot(dY, dY);
+				f = dX.x * dY.y - dX.y * dY.x;
+			}
+		}
 
-		// Calculate the minor length of the ellipse for Lod, while also clamping the ratio of anisotropy.
-		if (aniso_ratio > PS_ANISOTROPIC_FILTERING)
+		// Compute AF values
+		float determinant = abs(f);
+		bool is_major_x = squared_length_x > squared_length_y;
+		float squared_length_major = is_major_x ? squared_length_x : squared_length_y;
+		float length_major = sqrt(squared_length_major);
+
+		float aniso_ratio;
+		float length_lod;
+		float2 aniso_line;
+		if (length_major <= 1.0f)
 		{
-			// ratio is clamped - Lod is based on ratio (preserves area)
-			aniso_ratio = PS_ANISOTROPIC_FILTERING;
-			length_lod = length_major / PS_ANISOTROPIC_FILTERING;
+			// A zero length_major would result in NaN Lod and break sampling.
+			// A small length_major would result in aniso_ratio getting clamped to 1.
+			// Perform isotropic filtering instead.
+			aniso_ratio = 1.0f;
+			length_lod = length_major;
+			aniso_line = float2(0.0f, 0.0f);
 		}
 		else
 		{
-			// ratio not clamped - Lod is based on area
-			length_lod = determinant / length_major;
-		}
+			float norm_major = 1.0f / length_major;
 
-		// clamp to top Lod
-		if (length_lod < 1.0f)
-			aniso_ratio = max(1.0f, aniso_ratio * length_lod);
+			float2 aniso_line_dir = float2(
+				(is_major_x ? dX.x : dY.x) * norm_major,
+				(is_major_x ? dX.y : dY.y) * norm_major
+			);
+
+			aniso_ratio = squared_length_major / determinant;
+
+			// Calculate the minor length of the ellipse for Lod, while also clamping the ratio of anisotropy.
+			if (aniso_ratio > PS_ANISOTROPIC_FILTERING)
+			{
+				// ratio is clamped - Lod is based on ratio (preserves area)
+				aniso_ratio = PS_ANISOTROPIC_FILTERING;
+				length_lod = length_major / PS_ANISOTROPIC_FILTERING;
+			}
+			else
+			{
+				// ratio not clamped - Lod is based on area
+				length_lod = determinant / length_major;
+			}
+
+			// clamp to top Lod
+			if (length_lod < 1.0f)
+				aniso_ratio = max(1.0f, aniso_ratio * length_lod);
+
+			aniso_ratio = round(aniso_ratio);
+			aniso_line = aniso_line_dir * 0.5f * length_major * (1.0f / sz);
+		}
 
-		aniso_ratio = round(aniso_ratio);
-		aniso_line = aniso_line_dir * 0.5f * length_major * (1.0f / sz);
-	}
-	
 #if PS_AUTOMATIC_LOD == 1
-	float lod = log2(length_lod);
+		float lod = log2(length_lod);
 #elif PS_MANUAL_LOD == 1
-	float lod = manual_lod(uv_w);
+		float lod = manual_lod(uv_w);
 #else
-	float lod = 0; // No Lod
+		float lod = 0.0f; // No Lod
 #endif
-	
-	float4 colour;
-	if (aniso_ratio == 1.0f)
-		colour = Texture.SampleLevel(TextureSampler, uv, lod);
-	else
-	{
-		float4 num = float4(0, 0, 0, 0);
-		for (int i = 0; i < aniso_ratio; i++)
-		{		
-			float2 d = -aniso_line + (0.5f + i) * (2.0f * aniso_line) / aniso_ratio;	
-			float2 uv_sample = uv + d;
-			float4 sample_colour = Texture.SampleLevel(TextureSampler, uv_sample, lod);
-			num += sample_colour;
-		}
 
-		colour = num / aniso_ratio;
+		float4 colour;
+		if (aniso_ratio == 1.0f)
+			colour = Texture.SampleLevel(TextureSampler, uv, lod);
+		else
+		{
+			float4 num = float4(0.0f, 0.0f, 0.0f, 0.0f);
+			float2 segment = (2.0f * aniso_line) / aniso_ratio;
+
+			int aniso_ratio_i = (int)aniso_ratio;
+			for (int i = 0; i < aniso_ratio_i; i++)
+			{
+				float2 d = -aniso_line + (0.5f + i) * segment;	
+				float2 uv_sample = uv + d;
+				float4 sample_colour = Texture.SampleLevel(TextureSampler, uv_sample, lod);
+				num += sample_colour;
+			}
+
+			colour = num / aniso_ratio;
+		}
+		return colour;
 	}
-	return colour;
 }
 #endif