Support SampleId and SamplePosition builtins

src/Pipeline/PixelProgram.cpp

@@ -170,6 +170,22 @@ void PixelProgram::applyShader(Int cMask[4], Int sMask[4], Int zMask[4], int sam
 			routine.getVariable(it->second.Id)[it->second.FirstComponent + i] = Float4(0);
 	}
 
+	it = spirvShader->inputBuiltins.find(spv::BuiltInSampleId);
+	if(it != spirvShader->inputBuiltins.end())
+	{
+		routine.getVariable(it->second.Id)[it->second.FirstComponent] =
+		    As<SIMD::Float>(SIMD::Int((sampleId >= 0) ? sampleId : 0));
+	}
+
+	it = spirvShader->inputBuiltins.find(spv::BuiltInSamplePosition);
+	if(it != spirvShader->inputBuiltins.end())
+	{
+		routine.getVariable(it->second.Id)[it->second.FirstComponent + 0] =
+		    SIMD::Float(((sampleId >= 0) && (state.multiSampleCount > 1)) ? Constants::VkSampleLocations4[sampleId][0] : 0.5f);
+		routine.getVariable(it->second.Id)[it->second.FirstComponent + 1] =
+		    SIMD::Float(((sampleId >= 0) && (state.multiSampleCount > 1)) ? Constants::VkSampleLocations4[sampleId][1] : 0.5f);
+	}
+
 	// Note: all lanes initially active to facilitate derivatives etc. Actual coverage is
 	// handled separately, through the cMask.
 	auto activeLaneMask = SIMD::Int(0xFFFFFFFF);

src/Pipeline/PixelRoutine.cpp

@@ -62,9 +62,28 @@ void PixelRoutine::quad(Pointer<Byte> cBuffer[RENDERTARGETS], Pointer<Byte> &zBu
 	Int zMask[4];  // Depth mask
 	Int sMask[4];  // Stencil mask
 
+	bool sampleShadingEnabled = state.sampleShadingEnabled;
+	float minSampleShading = state.minSampleShading;
+	if(spirvShader)
+	{
+		// SampleId and SamplePosition built-ins require the sampleRateShading feature, so the Vulkan spec
+		// requires turning on per sample shading if either of them is present in the shader.
+
+		// "If a fragment shader entry point's interface includes an input variable decorated with SampleId,
+		//  Sample Shading is considered enabled with a minSampleShading value of 1.0."
+
+		// "If a fragment shader entry point's interface includes an input variable decorated with SamplePosition,
+		//  Sample Shading is considered enabled with a minSampleShading value of 1.0."
+		if(spirvShader->hasBuiltinInput(spv::BuiltInSampleId) || spirvShader->hasBuiltinInput(spv::BuiltInSamplePosition))
+		{
+			sampleShadingEnabled = true;
+			minSampleShading = 1.0f;
+		}
+	}
+
 	bool shaderContainsInterpolation = spirvShader && spirvShader->getUsedCapabilities().InterpolationFunction;
 	bool shaderContainsSampleQualifier = spirvShader && spirvShader->getModes().ContainsSampleQualifier;
-	bool perSampleShading = (state.sampleShadingEnabled && (state.minSampleShading > 0.0f)) ||
+	bool perSampleShading = (sampleShadingEnabled && (minSampleShading > 0.0f)) ||
 	                        shaderContainsInterpolation || shaderContainsSampleQualifier;
 	unsigned int numSampleRenders = perSampleShading ? state.multiSampleCount : 1;