glsl: Implement Images
This commit is contained in:
parent
6577a63d36
commit
af9696059c
2 changed files with 74 additions and 9 deletions
|
@ -91,11 +91,42 @@ std::string_view SamplerType(TextureType type, bool is_depth) {
|
||||||
case TextureType::Buffer:
|
case TextureType::Buffer:
|
||||||
return "samplerBuffer";
|
return "samplerBuffer";
|
||||||
default:
|
default:
|
||||||
fmt::print("Texture type: {}", type);
|
|
||||||
throw NotImplementedException("Texture type: {}", type);
|
throw NotImplementedException("Texture type: {}", type);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::string_view ImageType(TextureType type) {
|
||||||
|
switch (type) {
|
||||||
|
case TextureType::Color2D:
|
||||||
|
return "uimage2D";
|
||||||
|
default:
|
||||||
|
throw NotImplementedException("Image type: {}", type);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string_view ImageFormatString(ImageFormat format) {
|
||||||
|
switch (format) {
|
||||||
|
case ImageFormat::Typeless:
|
||||||
|
return "";
|
||||||
|
case ImageFormat::R8_UINT:
|
||||||
|
return ",r8ui";
|
||||||
|
case ImageFormat::R8_SINT:
|
||||||
|
return ",r8i";
|
||||||
|
case ImageFormat::R16_UINT:
|
||||||
|
return ",r16ui";
|
||||||
|
case ImageFormat::R16_SINT:
|
||||||
|
return ",r16i";
|
||||||
|
case ImageFormat::R32_UINT:
|
||||||
|
return ",r32ui";
|
||||||
|
case ImageFormat::R32G32_UINT:
|
||||||
|
return ",rg32ui";
|
||||||
|
case ImageFormat::R32G32B32A32_UINT:
|
||||||
|
return ",rgba32ui";
|
||||||
|
default:
|
||||||
|
throw NotImplementedException("Image format: {}", format);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
std::string_view GetTessMode(TessPrimitive primitive) {
|
std::string_view GetTessMode(TessPrimitive primitive) {
|
||||||
switch (primitive) {
|
switch (primitive) {
|
||||||
case TessPrimitive::Triangles:
|
case TessPrimitive::Triangles:
|
||||||
|
@ -250,6 +281,7 @@ void EmitContext::SetupExtensions(std::string&) {
|
||||||
// TODO: track this usage
|
// TODO: track this usage
|
||||||
header += "#extension GL_ARB_sparse_texture2 : enable\n";
|
header += "#extension GL_ARB_sparse_texture2 : enable\n";
|
||||||
header += "#extension GL_EXT_texture_shadow_lod : enable\n";
|
header += "#extension GL_EXT_texture_shadow_lod : enable\n";
|
||||||
|
header += "#extension GL_EXT_shader_image_load_formatted : enable\n";
|
||||||
if (info.uses_int64) {
|
if (info.uses_int64) {
|
||||||
header += "#extension GL_ARB_gpu_shader_int64 : enable\n";
|
header += "#extension GL_ARB_gpu_shader_int64 : enable\n";
|
||||||
}
|
}
|
||||||
|
@ -396,15 +428,25 @@ void EmitContext::DefineHelperFunctions() {
|
||||||
void EmitContext::SetupImages(Bindings& bindings) {
|
void EmitContext::SetupImages(Bindings& bindings) {
|
||||||
image_buffer_bindings.reserve(info.image_buffer_descriptors.size());
|
image_buffer_bindings.reserve(info.image_buffer_descriptors.size());
|
||||||
for (const auto& desc : info.image_buffer_descriptors) {
|
for (const auto& desc : info.image_buffer_descriptors) {
|
||||||
throw NotImplementedException("image_buffer_descriptors");
|
const auto indices{bindings.image + desc.count};
|
||||||
|
for (u32 index = bindings.image; index < indices; ++index) {
|
||||||
|
header += fmt::format("layout(binding={}) uniform uimageBuffer img{};", bindings.image,
|
||||||
|
index);
|
||||||
|
}
|
||||||
image_buffer_bindings.push_back(bindings.image);
|
image_buffer_bindings.push_back(bindings.image);
|
||||||
bindings.image += desc.count;
|
bindings.image += desc.count;
|
||||||
}
|
}
|
||||||
image_bindings.reserve(info.image_descriptors.size());
|
image_bindings.reserve(info.image_descriptors.size());
|
||||||
for (const auto& desc : info.image_descriptors) {
|
for (const auto& desc : info.image_descriptors) {
|
||||||
throw NotImplementedException("image_bindings");
|
|
||||||
|
|
||||||
image_bindings.push_back(bindings.image);
|
image_bindings.push_back(bindings.image);
|
||||||
|
const auto format{ImageFormatString(desc.format)};
|
||||||
|
const auto image_type{ImageType(desc.type)};
|
||||||
|
const auto qualifier{desc.is_written ? "" : "readonly "};
|
||||||
|
const auto indices{bindings.image + desc.count};
|
||||||
|
for (u32 index = bindings.image; index < indices; ++index) {
|
||||||
|
header += fmt::format("layout(binding={}{})uniform {}{} img{};", bindings.image, format,
|
||||||
|
qualifier, image_type, index);
|
||||||
|
}
|
||||||
bindings.image += desc.count;
|
bindings.image += desc.count;
|
||||||
}
|
}
|
||||||
texture_buffer_bindings.reserve(info.texture_buffer_descriptors.size());
|
texture_buffer_bindings.reserve(info.texture_buffer_descriptors.size());
|
||||||
|
|
|
@ -14,15 +14,25 @@ namespace {
|
||||||
std::string Texture(EmitContext& ctx, const IR::TextureInstInfo& info,
|
std::string Texture(EmitContext& ctx, const IR::TextureInstInfo& info,
|
||||||
[[maybe_unused]] const IR::Value& index) {
|
[[maybe_unused]] const IR::Value& index) {
|
||||||
if (info.type == TextureType::Buffer) {
|
if (info.type == TextureType::Buffer) {
|
||||||
throw NotImplementedException("TextureType::Buffer");
|
return fmt::format("tex{}", ctx.texture_buffer_bindings.at(info.descriptor_index));
|
||||||
} else {
|
} else {
|
||||||
return fmt::format("tex{}", ctx.texture_bindings.at(info.descriptor_index));
|
return fmt::format("tex{}", ctx.texture_bindings.at(info.descriptor_index));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::string Image(EmitContext& ctx, const IR::TextureInstInfo& info,
|
||||||
|
[[maybe_unused]] const IR::Value& index) {
|
||||||
|
if (info.type == TextureType::Buffer) {
|
||||||
|
return fmt::format("img{}", ctx.image_buffer_bindings.at(info.descriptor_index));
|
||||||
|
} else {
|
||||||
|
return fmt::format("img{}", ctx.image_bindings.at(info.descriptor_index));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
std::string CastToIntVec(std::string_view value, const IR::TextureInstInfo& info) {
|
std::string CastToIntVec(std::string_view value, const IR::TextureInstInfo& info) {
|
||||||
switch (info.type) {
|
switch (info.type) {
|
||||||
case TextureType::Color1D:
|
case TextureType::Color1D:
|
||||||
|
case TextureType::Buffer:
|
||||||
return fmt::format("int({})", value);
|
return fmt::format("int({})", value);
|
||||||
case TextureType::ColorArray1D:
|
case TextureType::ColorArray1D:
|
||||||
case TextureType::Color2D:
|
case TextureType::Color2D:
|
||||||
|
@ -41,6 +51,7 @@ std::string CastToIntVec(std::string_view value, const IR::TextureInstInfo& info
|
||||||
std::string TexelFetchCastToInt(std::string_view value, const IR::TextureInstInfo& info) {
|
std::string TexelFetchCastToInt(std::string_view value, const IR::TextureInstInfo& info) {
|
||||||
switch (info.type) {
|
switch (info.type) {
|
||||||
case TextureType::Color1D:
|
case TextureType::Color1D:
|
||||||
|
case TextureType::Buffer:
|
||||||
return fmt::format("int({})", value);
|
return fmt::format("int({})", value);
|
||||||
case TextureType::ColorArray1D:
|
case TextureType::ColorArray1D:
|
||||||
case TextureType::Color2D:
|
case TextureType::Color2D:
|
||||||
|
@ -348,10 +359,14 @@ void EmitImageFetch([[maybe_unused]] EmitContext& ctx, [[maybe_unused]] IR::Inst
|
||||||
if (!offset.empty()) {
|
if (!offset.empty()) {
|
||||||
ctx.Add("{}=texelFetchOffset({},{},int({}),{});", texel, texture,
|
ctx.Add("{}=texelFetchOffset({},{},int({}),{});", texel, texture,
|
||||||
TexelFetchCastToInt(coords, info), lod, TexelFetchCastToInt(offset, info));
|
TexelFetchCastToInt(coords, info), lod, TexelFetchCastToInt(offset, info));
|
||||||
|
} else {
|
||||||
|
if (info.type == TextureType::Buffer) {
|
||||||
|
ctx.Add("{}=texelFetch({},int({}));", texel, texture, coords);
|
||||||
} else {
|
} else {
|
||||||
ctx.Add("{}=texelFetch({},{},int({}));", texel, texture,
|
ctx.Add("{}=texelFetch({},{},int({}));", texel, texture,
|
||||||
TexelFetchCastToInt(coords, info), lod);
|
TexelFetchCastToInt(coords, info), lod);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
// TODO: Query sparseTexels extension support
|
// TODO: Query sparseTexels extension support
|
||||||
|
@ -434,14 +449,22 @@ void EmitImageGradient([[maybe_unused]] EmitContext& ctx, [[maybe_unused]] IR::I
|
||||||
void EmitImageRead([[maybe_unused]] EmitContext& ctx, [[maybe_unused]] IR::Inst& inst,
|
void EmitImageRead([[maybe_unused]] EmitContext& ctx, [[maybe_unused]] IR::Inst& inst,
|
||||||
[[maybe_unused]] const IR::Value& index,
|
[[maybe_unused]] const IR::Value& index,
|
||||||
[[maybe_unused]] std::string_view coords) {
|
[[maybe_unused]] std::string_view coords) {
|
||||||
NotImplemented();
|
const auto info{inst.Flags<IR::TextureInstInfo>()};
|
||||||
|
const auto sparse_inst{PrepareSparse(inst)};
|
||||||
|
if (sparse_inst) {
|
||||||
|
throw NotImplementedException("EmitImageRead Sparse");
|
||||||
|
}
|
||||||
|
const auto image{Image(ctx, info, index)};
|
||||||
|
ctx.AddU32x4("{}=uvec4(imageLoad({},{}));", inst, image, TexelFetchCastToInt(coords, info));
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitImageWrite([[maybe_unused]] EmitContext& ctx, [[maybe_unused]] IR::Inst& inst,
|
void EmitImageWrite([[maybe_unused]] EmitContext& ctx, [[maybe_unused]] IR::Inst& inst,
|
||||||
[[maybe_unused]] const IR::Value& index,
|
[[maybe_unused]] const IR::Value& index,
|
||||||
[[maybe_unused]] std::string_view coords,
|
[[maybe_unused]] std::string_view coords,
|
||||||
[[maybe_unused]] std::string_view color) {
|
[[maybe_unused]] std::string_view color) {
|
||||||
NotImplemented();
|
const auto info{inst.Flags<IR::TextureInstInfo>()};
|
||||||
|
const auto image{Image(ctx, info, index)};
|
||||||
|
ctx.Add("imageStore({},{},{});", image, TexelFetchCastToInt(coords, info), color);
|
||||||
}
|
}
|
||||||
|
|
||||||
void EmitBindlessImageSampleImplicitLod(EmitContext&) {
|
void EmitBindlessImageSampleImplicitLod(EmitContext&) {
|
||||||
|
|
Loading…
Reference in a new issue