Derive gaussian splat SH order from resource layout

This commit is contained in:
2026-04-11 06:57:47 +08:00
parent 785377bc9b
commit b3acb5afc2
5 changed files with 82 additions and 1 deletions

View File

@@ -3,6 +3,38 @@
namespace XCEngine {
namespace Resources {
Core::uint32 ResolveGaussianSplatSHOrderFromCoefficientCount(Core::uint32 coefficientCount) {
if (coefficientCount == 0u || (coefficientCount % kGaussianSplatSHColorChannelCount) != 0u) {
return 0u;
}
Core::uint32 remainingTriplets = coefficientCount / kGaussianSplatSHColorChannelCount;
Core::uint32 resolvedOrder = 0u;
for (Core::uint32 order = 1u; order <= kGaussianSplatMaxSHOrder; ++order) {
const Core::uint32 bandTriplets = (2u * order) + 1u;
if (remainingTriplets < bandTriplets) {
break;
}
remainingTriplets -= bandTriplets;
resolvedOrder = order;
}
return resolvedOrder;
}
Core::uint32 ResolveGaussianSplatSHOrderFromSectionStride(Core::uint32 elementStride) {
if (elementStride < sizeof(float)) {
return 0u;
}
const Core::uint32 floatCount = elementStride / static_cast<Core::uint32>(sizeof(float));
const Core::uint32 coefficientCount = floatCount > kGaussianSplatSHCoefficientCount
? kGaussianSplatSHCoefficientCount
: floatCount;
return ResolveGaussianSplatSHOrderFromCoefficientCount(coefficientCount);
}
GaussianSplat::GaussianSplat() = default;
GaussianSplat::~GaussianSplat() = default;
@@ -69,6 +101,11 @@ const GaussianSplatSHRecord* GaussianSplat::GetSHRecords() const {
return static_cast<const GaussianSplatSHRecord*>(GetSectionData(GaussianSplatSectionType::SH));
}
Core::uint32 GaussianSplat::GetSHOrder() const {
const GaussianSplatSection* shSection = FindSection(GaussianSplatSectionType::SH);
return shSection != nullptr ? ResolveGaussianSplatSHOrderFromSectionStride(shSection->elementStride) : 0u;
}
bool GaussianSplat::ValidateSections(const Containers::Array<GaussianSplatSection>& sections,
size_t payloadSize) const {
for (size_t index = 0; index < sections.Size(); ++index) {