Merge pull request #1025 from KhronosGroup/fix-1013

MSL: Support OpImageQueryLod.
This commit is contained in:
Hans-Kristian Arntzen 2019-06-19 14:07:39 +02:00 committed by GitHub
commit 4c20c941f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 206 additions and 1 deletions

View File

@ -0,0 +1,70 @@
#include <metal_stdlib>
#include <simd/simd.h>
using namespace metal;
struct main0_out
{
float2 FragColor [[color(0)]];
};
struct main0_in
{
float3 vUV [[user(locn0)]];
};
fragment main0_out main0(main0_in in [[stage_in]], texture2d<float> uSampler2D [[texture(0)]], texture3d<float> uSampler3D [[texture(1)]], texturecube<float> uSamplerCube [[texture(2)]], texture2d<float> uTexture2D [[texture(4)]], texture3d<float> uTexture3D [[texture(5)]], texturecube<float> uTextureCube [[texture(6)]], sampler uSampler2DSmplr [[sampler(0)]], sampler uSampler3DSmplr [[sampler(1)]], sampler uSamplerCubeSmplr [[sampler(2)]], sampler uSampler [[sampler(3)]])
{
main0_out out = {};
out.FragColor = float2(0.0);
float2 _79;
_79.x = uSampler2D.calculate_clamped_lod(uSampler2DSmplr, in.vUV.xy);
_79.y = uSampler2D.calculate_unclamped_lod(uSampler2DSmplr, in.vUV.xy);
out.FragColor += _79;
float2 _84;
_84.x = uSampler3D.calculate_clamped_lod(uSampler3DSmplr, in.vUV);
_84.y = uSampler3D.calculate_unclamped_lod(uSampler3DSmplr, in.vUV);
out.FragColor += _84;
float2 _89;
_89.x = uSamplerCube.calculate_clamped_lod(uSamplerCubeSmplr, in.vUV);
_89.y = uSamplerCube.calculate_unclamped_lod(uSamplerCubeSmplr, in.vUV);
out.FragColor += _89;
float2 _97;
_97.x = uTexture2D.calculate_clamped_lod(uSampler, in.vUV.xy);
_97.y = uTexture2D.calculate_unclamped_lod(uSampler, in.vUV.xy);
out.FragColor += _97;
float2 _104;
_104.x = uTexture3D.calculate_clamped_lod(uSampler, in.vUV);
_104.y = uTexture3D.calculate_unclamped_lod(uSampler, in.vUV);
out.FragColor += _104;
float2 _111;
_111.x = uTextureCube.calculate_clamped_lod(uSampler, in.vUV);
_111.y = uTextureCube.calculate_unclamped_lod(uSampler, in.vUV);
out.FragColor += _111;
float2 _118;
_118.x = uSampler2D.calculate_clamped_lod(uSampler2DSmplr, in.vUV.xy);
_118.y = uSampler2D.calculate_unclamped_lod(uSampler2DSmplr, in.vUV.xy);
out.FragColor += _118;
float2 _123;
_123.x = uSampler3D.calculate_clamped_lod(uSampler3DSmplr, in.vUV);
_123.y = uSampler3D.calculate_unclamped_lod(uSampler3DSmplr, in.vUV);
out.FragColor += _123;
float2 _128;
_128.x = uSamplerCube.calculate_clamped_lod(uSamplerCubeSmplr, in.vUV);
_128.y = uSamplerCube.calculate_unclamped_lod(uSamplerCubeSmplr, in.vUV);
out.FragColor += _128;
float2 _136;
_136.x = uTexture2D.calculate_clamped_lod(uSampler, in.vUV.xy);
_136.y = uTexture2D.calculate_unclamped_lod(uSampler, in.vUV.xy);
out.FragColor += _136;
float2 _143;
_143.x = uTexture3D.calculate_clamped_lod(uSampler, in.vUV);
_143.y = uTexture3D.calculate_unclamped_lod(uSampler, in.vUV);
out.FragColor += _143;
float2 _150;
_150.x = uTextureCube.calculate_clamped_lod(uSampler, in.vUV);
_150.y = uTextureCube.calculate_unclamped_lod(uSampler, in.vUV);
out.FragColor += _150;
return out;
}

View File

@ -0,0 +1,77 @@
#pragma clang diagnostic ignored "-Wmissing-prototypes"
#include <metal_stdlib>
#include <simd/simd.h>
using namespace metal;
struct main0_out
{
float2 FragColor [[color(0)]];
};
struct main0_in
{
float3 vUV [[user(locn0)]];
};
void from_function(thread float2& FragColor, thread texture2d<float> uSampler2D, thread const sampler uSampler2DSmplr, thread float3& vUV, thread texture3d<float> uSampler3D, thread const sampler uSampler3DSmplr, thread texturecube<float> uSamplerCube, thread const sampler uSamplerCubeSmplr, thread texture2d<float> uTexture2D, thread sampler uSampler, thread texture3d<float> uTexture3D, thread texturecube<float> uTextureCube)
{
float2 _22;
_22.x = uSampler2D.calculate_clamped_lod(uSampler2DSmplr, vUV.xy);
_22.y = uSampler2D.calculate_unclamped_lod(uSampler2DSmplr, vUV.xy);
FragColor += _22;
float2 _31;
_31.x = uSampler3D.calculate_clamped_lod(uSampler3DSmplr, vUV);
_31.y = uSampler3D.calculate_unclamped_lod(uSampler3DSmplr, vUV);
FragColor += _31;
float2 _40;
_40.x = uSamplerCube.calculate_clamped_lod(uSamplerCubeSmplr, vUV);
_40.y = uSamplerCube.calculate_unclamped_lod(uSamplerCubeSmplr, vUV);
FragColor += _40;
float2 _53;
_53.x = uTexture2D.calculate_clamped_lod(uSampler, vUV.xy);
_53.y = uTexture2D.calculate_unclamped_lod(uSampler, vUV.xy);
FragColor += _53;
float2 _62;
_62.x = uTexture3D.calculate_clamped_lod(uSampler, vUV);
_62.y = uTexture3D.calculate_unclamped_lod(uSampler, vUV);
FragColor += _62;
float2 _71;
_71.x = uTextureCube.calculate_clamped_lod(uSampler, vUV);
_71.y = uTextureCube.calculate_unclamped_lod(uSampler, vUV);
FragColor += _71;
}
fragment main0_out main0(main0_in in [[stage_in]], texture2d<float> uSampler2D [[texture(0)]], texture3d<float> uSampler3D [[texture(1)]], texturecube<float> uSamplerCube [[texture(2)]], texture2d<float> uTexture2D [[texture(4)]], texture3d<float> uTexture3D [[texture(5)]], texturecube<float> uTextureCube [[texture(6)]], sampler uSampler2DSmplr [[sampler(0)]], sampler uSampler3DSmplr [[sampler(1)]], sampler uSamplerCubeSmplr [[sampler(2)]], sampler uSampler [[sampler(3)]])
{
main0_out out = {};
out.FragColor = float2(0.0);
float2 _79;
_79.x = uSampler2D.calculate_clamped_lod(uSampler2DSmplr, in.vUV.xy);
_79.y = uSampler2D.calculate_unclamped_lod(uSampler2DSmplr, in.vUV.xy);
out.FragColor += _79;
float2 _84;
_84.x = uSampler3D.calculate_clamped_lod(uSampler3DSmplr, in.vUV);
_84.y = uSampler3D.calculate_unclamped_lod(uSampler3DSmplr, in.vUV);
out.FragColor += _84;
float2 _89;
_89.x = uSamplerCube.calculate_clamped_lod(uSamplerCubeSmplr, in.vUV);
_89.y = uSamplerCube.calculate_unclamped_lod(uSamplerCubeSmplr, in.vUV);
out.FragColor += _89;
float2 _97;
_97.x = uTexture2D.calculate_clamped_lod(uSampler, in.vUV.xy);
_97.y = uTexture2D.calculate_unclamped_lod(uSampler, in.vUV.xy);
out.FragColor += _97;
float2 _104;
_104.x = uTexture3D.calculate_clamped_lod(uSampler, in.vUV);
_104.y = uTexture3D.calculate_unclamped_lod(uSampler, in.vUV);
out.FragColor += _104;
float2 _111;
_111.x = uTextureCube.calculate_clamped_lod(uSampler, in.vUV);
_111.y = uTextureCube.calculate_unclamped_lod(uSampler, in.vUV);
out.FragColor += _111;
from_function(out.FragColor, uSampler2D, uSampler2DSmplr, in.vUV, uSampler3D, uSampler3DSmplr, uSamplerCube, uSamplerCubeSmplr, uTexture2D, uSampler, uTexture3D, uTextureCube);
return out;
}

View File

@ -0,0 +1,33 @@
#version 450
layout(location = 0) out vec2 FragColor;
layout(set = 0, binding = 0) uniform sampler2D uSampler2D;
layout(set = 0, binding = 1) uniform sampler3D uSampler3D;
layout(set = 0, binding = 2) uniform samplerCube uSamplerCube;
layout(set = 0, binding = 3) uniform sampler uSampler;
layout(set = 0, binding = 4) uniform texture2D uTexture2D;
layout(set = 0, binding = 5) uniform texture3D uTexture3D;
layout(set = 0, binding = 6) uniform textureCube uTextureCube;
layout(location = 0) in vec3 vUV;
void from_function()
{
FragColor += textureQueryLod(uSampler2D, vUV.xy);
FragColor += textureQueryLod(uSampler3D, vUV);
FragColor += textureQueryLod(uSamplerCube, vUV);
FragColor += textureQueryLod(sampler2D(uTexture2D, uSampler), vUV.xy);
FragColor += textureQueryLod(sampler3D(uTexture3D, uSampler), vUV);
FragColor += textureQueryLod(samplerCube(uTextureCube, uSampler), vUV);
}
void main()
{
FragColor = vec2(0.0);
FragColor += textureQueryLod(uSampler2D, vUV.xy);
FragColor += textureQueryLod(uSampler3D, vUV);
FragColor += textureQueryLod(uSamplerCube, vUV);
FragColor += textureQueryLod(sampler2D(uTexture2D, uSampler), vUV.xy);
FragColor += textureQueryLod(sampler3D(uTexture3D, uSampler), vUV);
FragColor += textureQueryLod(samplerCube(uTextureCube, uSampler), vUV);
from_function();
}

View File

@ -3904,7 +3904,32 @@ void CompilerMSL::emit_instruction(const Instruction &instruction)
} }
case OpImageQueryLod: case OpImageQueryLod:
SPIRV_CROSS_THROW("MSL does not support textureQueryLod()."); {
if (!msl_options.supports_msl_version(2, 2))
SPIRV_CROSS_THROW("ImageQueryLod is only supported on MSL 2.2 and up.");
uint32_t result_type = ops[0];
uint32_t id = ops[1];
uint32_t image_id = ops[2];
uint32_t coord_id = ops[3];
emit_uninitialized_temporary_expression(result_type, id);
auto sampler_expr = to_sampler_expression(image_id);
auto *combined = maybe_get<SPIRCombinedImageSampler>(image_id);
auto image_expr = combined ? to_expression(combined->image) : to_expression(image_id);
// TODO: It is unclear if calculcate_clamped_lod also conditionally rounds
// the reported LOD based on the sampler. NEAREST miplevel should
// round the LOD, but LINEAR miplevel should not round.
// Let's hope this does not become an issue ...
statement(to_expression(id), ".x = ",
image_expr, ".calculate_clamped_lod(",
sampler_expr, ", ", to_expression(coord_id), ");");
statement(to_expression(id), ".y = ",
image_expr, ".calculate_unclamped_lod(",
sampler_expr, ", ", to_expression(coord_id), ");");
register_control_dependent_expression(id);
break;
}
#define MSL_ImgQry(qrytype) \ #define MSL_ImgQry(qrytype) \
do \ do \