Bind gaussian splat chunk metadata in prepare pass

This commit is contained in:
2026-04-11 13:55:39 +08:00
parent ff4e3f639a
commit 88a71a5426
7 changed files with 148 additions and 15 deletions

View File

@@ -46,6 +46,7 @@ inline bool TryBuildBuiltinPassResourceBindingPlan(
location = &outPlan.material;
break;
case BuiltinPassResourceSemantic::MaterialBuffer:
case BuiltinPassResourceSemantic::MaterialTexture:
break;
case BuiltinPassResourceSemantic::Lighting:
location = &outPlan.lighting;
@@ -80,6 +81,9 @@ inline bool TryBuildBuiltinPassResourceBindingPlan(
case BuiltinPassResourceSemantic::GaussianSplatSHBuffer:
location = &outPlan.gaussianSplatSHBuffer;
break;
case BuiltinPassResourceSemantic::GaussianSplatChunkBuffer:
location = &outPlan.gaussianSplatChunkBuffer;
break;
case BuiltinPassResourceSemantic::GaussianSplatViewDataBuffer:
location = &outPlan.gaussianSplatViewDataBuffer;
break;
@@ -109,7 +113,9 @@ inline bool TryBuildBuiltinPassResourceBindingPlan(
break;
}
if (semantic != BuiltinPassResourceSemantic::MaterialBuffer && location == nullptr) {
if (semantic != BuiltinPassResourceSemantic::MaterialBuffer &&
semantic != BuiltinPassResourceSemantic::MaterialTexture &&
location == nullptr) {
return fail("Builtin pass resource semantic could not be mapped");
}
if (location != nullptr && location->IsValid()) {
@@ -137,6 +143,9 @@ inline bool TryBuildBuiltinPassResourceBindingPlan(
if (semantic == BuiltinPassResourceSemantic::MaterialBuffer) {
outPlan.materialBufferBindings.PushBack(resolvedBinding);
outPlan.usesMaterialBuffers = true;
} else if (semantic == BuiltinPassResourceSemantic::MaterialTexture) {
outPlan.materialTextureBindings.PushBack(resolvedBinding);
outPlan.usesMaterialTextures = true;
}
outPlan.maxSetIndex = std::max(outPlan.maxSetIndex, binding.set);
@@ -593,6 +602,11 @@ inline bool TryBuildBuiltinPassSetLayouts(
setLayout.usesMaterialBuffers = true;
setLayout.materialBufferBindings.push_back(binding);
break;
case BuiltinPassResourceSemantic::MaterialTexture:
setLayout.usesTexture = true;
setLayout.usesMaterialTextures = true;
setLayout.materialTextureBindings.push_back(binding);
break;
case BuiltinPassResourceSemantic::Lighting:
setLayout.usesLighting = true;
break;
@@ -626,6 +640,9 @@ inline bool TryBuildBuiltinPassSetLayouts(
case BuiltinPassResourceSemantic::GaussianSplatSHBuffer:
setLayout.usesGaussianSplatSHBuffer = true;
break;
case BuiltinPassResourceSemantic::GaussianSplatChunkBuffer:
setLayout.usesGaussianSplatChunkBuffer = true;
break;
case BuiltinPassResourceSemantic::GaussianSplatViewDataBuffer:
setLayout.usesGaussianSplatViewDataBuffer = true;
break;

View File

@@ -183,6 +183,12 @@ inline BuiltinPassResourceSemantic ResolveBuiltinPassResourceSemantic(
return BuiltinPassResourceSemantic::GaussianSplatSHBuffer;
}
if (semantic == Containers::String("gaussiansplatchunkbuffer") ||
semantic == Containers::String("gaussiansplatchunks") ||
semantic == Containers::String("splatchunks")) {
return BuiltinPassResourceSemantic::GaussianSplatChunkBuffer;
}
if (semantic == Containers::String("gaussiansplatviewdatabuffer") ||
semantic == Containers::String("gaussiansplatviewdata") ||
semantic == Containers::String("splatviewdata")) {
@@ -221,6 +227,9 @@ inline BuiltinPassResourceSemantic ResolveBuiltinPassResourceSemantic(
}
switch (binding.type) {
case Resources::ShaderResourceType::Texture2D:
case Resources::ShaderResourceType::TextureCube:
return BuiltinPassResourceSemantic::MaterialTexture;
case Resources::ShaderResourceType::StructuredBuffer:
case Resources::ShaderResourceType::RawBuffer:
case Resources::ShaderResourceType::RWStructuredBuffer:
@@ -241,6 +250,8 @@ inline const char* BuiltinPassResourceSemanticToString(BuiltinPassResourceSemant
return "Material";
case BuiltinPassResourceSemantic::MaterialBuffer:
return "MaterialBuffer";
case BuiltinPassResourceSemantic::MaterialTexture:
return "MaterialTexture";
case BuiltinPassResourceSemantic::Lighting:
return "Lighting";
case BuiltinPassResourceSemantic::ShadowReceiver:
@@ -263,6 +274,8 @@ inline const char* BuiltinPassResourceSemanticToString(BuiltinPassResourceSemant
return "GaussianSplatColorBuffer";
case BuiltinPassResourceSemantic::GaussianSplatSHBuffer:
return "GaussianSplatSHBuffer";
case BuiltinPassResourceSemantic::GaussianSplatChunkBuffer:
return "GaussianSplatChunkBuffer";
case BuiltinPassResourceSemantic::GaussianSplatViewDataBuffer:
return "GaussianSplatViewDataBuffer";
case BuiltinPassResourceSemantic::BaseColorTexture:
@@ -350,6 +363,9 @@ inline bool IsBuiltinPassResourceTypeCompatible(
type == Resources::ShaderResourceType::RawBuffer ||
type == Resources::ShaderResourceType::RWStructuredBuffer ||
type == Resources::ShaderResourceType::RWRawBuffer;
case BuiltinPassResourceSemantic::MaterialTexture:
return type == Resources::ShaderResourceType::Texture2D ||
type == Resources::ShaderResourceType::TextureCube;
case BuiltinPassResourceSemantic::VolumeField:
return type == Resources::ShaderResourceType::StructuredBuffer ||
type == Resources::ShaderResourceType::RawBuffer;
@@ -359,6 +375,7 @@ inline bool IsBuiltinPassResourceTypeCompatible(
case BuiltinPassResourceSemantic::GaussianSplatOtherBuffer:
case BuiltinPassResourceSemantic::GaussianSplatColorBuffer:
case BuiltinPassResourceSemantic::GaussianSplatSHBuffer:
case BuiltinPassResourceSemantic::GaussianSplatChunkBuffer:
case BuiltinPassResourceSemantic::GaussianSplatViewDataBuffer:
return type == Resources::ShaderResourceType::StructuredBuffer ||
type == Resources::ShaderResourceType::RawBuffer ||

View File

@@ -42,6 +42,7 @@ enum class BuiltinPassResourceSemantic : Core::uint8 {
PerObject,
Material,
MaterialBuffer,
MaterialTexture,
Lighting,
ShadowReceiver,
Environment,
@@ -53,6 +54,7 @@ enum class BuiltinPassResourceSemantic : Core::uint8 {
GaussianSplatOtherBuffer,
GaussianSplatColorBuffer,
GaussianSplatSHBuffer,
GaussianSplatChunkBuffer,
GaussianSplatViewDataBuffer,
BaseColorTexture,
SourceColorTexture,
@@ -73,6 +75,7 @@ struct BuiltinPassResourceBindingDesc {
struct BuiltinPassResourceBindingPlan {
Containers::Array<BuiltinPassResourceBindingDesc> bindings;
Containers::Array<BuiltinPassResourceBindingDesc> materialBufferBindings;
Containers::Array<BuiltinPassResourceBindingDesc> materialTextureBindings;
Core::uint32 maxSetIndex = 0;
Core::uint32 firstDescriptorSet = 0;
Core::uint32 descriptorSetCount = 0;
@@ -80,6 +83,7 @@ struct BuiltinPassResourceBindingPlan {
bool usesTextures = false;
bool usesSamplers = false;
bool usesMaterialBuffers = false;
bool usesMaterialTextures = false;
PassResourceBindingLocation perObject = {};
PassResourceBindingLocation material = {};
PassResourceBindingLocation volumeField = {};
@@ -93,6 +97,7 @@ struct BuiltinPassResourceBindingPlan {
PassResourceBindingLocation gaussianSplatOtherBuffer = {};
PassResourceBindingLocation gaussianSplatColorBuffer = {};
PassResourceBindingLocation gaussianSplatSHBuffer = {};
PassResourceBindingLocation gaussianSplatChunkBuffer = {};
PassResourceBindingLocation gaussianSplatViewDataBuffer = {};
PassResourceBindingLocation baseColorTexture = {};
PassResourceBindingLocation sourceColorTexture = {};
@@ -116,6 +121,7 @@ struct BuiltinPassResourceBindingPlan {
struct BuiltinPassSetLayoutMetadata {
std::vector<RHI::DescriptorSetLayoutBinding> bindings;
std::vector<BuiltinPassResourceBindingDesc> materialBufferBindings;
std::vector<BuiltinPassResourceBindingDesc> materialTextureBindings;
RHI::DescriptorSetLayoutDesc layout = {};
RHI::DescriptorHeapType heapType = RHI::DescriptorHeapType::CBV_SRV_UAV;
bool shaderVisible = false;
@@ -126,6 +132,7 @@ struct BuiltinPassSetLayoutMetadata {
bool usesEnvironment = false;
bool usesPassConstants = false;
bool usesMaterialBuffers = false;
bool usesMaterialTextures = false;
bool usesVolumeField = false;
bool usesGaussianSplatSortDistanceBuffer = false;
bool usesGaussianSplatOrderBuffer = false;
@@ -133,6 +140,7 @@ struct BuiltinPassSetLayoutMetadata {
bool usesGaussianSplatOtherBuffer = false;
bool usesGaussianSplatColorBuffer = false;
bool usesGaussianSplatSHBuffer = false;
bool usesGaussianSplatChunkBuffer = false;
bool usesGaussianSplatViewDataBuffer = false;
bool usesTexture = false;
bool usesBaseColorTexture = false;
@@ -148,14 +156,16 @@ struct BuiltinPassSetLayoutMetadata {
BuiltinPassSetLayoutMetadata(const BuiltinPassSetLayoutMetadata& other)
: bindings(other.bindings)
, materialBufferBindings(other.materialBufferBindings) {
, materialBufferBindings(other.materialBufferBindings)
, materialTextureBindings(other.materialTextureBindings) {
CopyFlagsFrom(other);
SyncLayoutView();
}
BuiltinPassSetLayoutMetadata(BuiltinPassSetLayoutMetadata&& other) noexcept
: bindings(std::move(other.bindings))
, materialBufferBindings(std::move(other.materialBufferBindings)) {
, materialBufferBindings(std::move(other.materialBufferBindings))
, materialTextureBindings(std::move(other.materialTextureBindings)) {
CopyFlagsFrom(other);
SyncLayoutView();
other.layout = {};
@@ -165,6 +175,7 @@ struct BuiltinPassSetLayoutMetadata {
if (this != &other) {
bindings = other.bindings;
materialBufferBindings = other.materialBufferBindings;
materialTextureBindings = other.materialTextureBindings;
CopyFlagsFrom(other);
SyncLayoutView();
}
@@ -176,6 +187,7 @@ struct BuiltinPassSetLayoutMetadata {
if (this != &other) {
bindings = std::move(other.bindings);
materialBufferBindings = std::move(other.materialBufferBindings);
materialTextureBindings = std::move(other.materialTextureBindings);
CopyFlagsFrom(other);
SyncLayoutView();
other.layout = {};
@@ -200,6 +212,7 @@ private:
usesEnvironment = other.usesEnvironment;
usesPassConstants = other.usesPassConstants;
usesMaterialBuffers = other.usesMaterialBuffers;
usesMaterialTextures = other.usesMaterialTextures;
usesVolumeField = other.usesVolumeField;
usesGaussianSplatSortDistanceBuffer = other.usesGaussianSplatSortDistanceBuffer;
usesGaussianSplatOrderBuffer = other.usesGaussianSplatOrderBuffer;
@@ -207,6 +220,7 @@ private:
usesGaussianSplatOtherBuffer = other.usesGaussianSplatOtherBuffer;
usesGaussianSplatColorBuffer = other.usesGaussianSplatColorBuffer;
usesGaussianSplatSHBuffer = other.usesGaussianSplatSHBuffer;
usesGaussianSplatChunkBuffer = other.usesGaussianSplatChunkBuffer;
usesGaussianSplatViewDataBuffer = other.usesGaussianSplatViewDataBuffer;
usesTexture = other.usesTexture;
usesBaseColorTexture = other.usesBaseColorTexture;

View File

@@ -100,6 +100,7 @@ private:
PassResourceBindingLocation gaussianSplatOtherBuffer = {};
PassResourceBindingLocation gaussianSplatColorBuffer = {};
PassResourceBindingLocation gaussianSplatSHBuffer = {};
PassResourceBindingLocation gaussianSplatChunkBuffer = {};
PassResourceBindingLocation gaussianSplatViewDataBuffer = {};
};
@@ -142,6 +143,7 @@ private:
RHI::RHIResourceView* otherView = nullptr;
RHI::RHIResourceView* colorView = nullptr;
RHI::RHIResourceView* shView = nullptr;
RHI::RHIResourceView* chunkView = nullptr;
RHI::RHIResourceView* viewDataView = nullptr;
};