storytold/spark
2
0
Fork 0
mirror of https://github.com/storytold/spark.git synced 2026-10-11 21:40:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix .ksplat decoding by implementing partial bucket decoding. (#68)

Browse Source
This commit is contained in:
Andreas Sundquist
2025-06-24 13:40:04 -07:00
committed by GitHub
parent 521fcb6a3a
commit af45f72a2b
1 changed files with 46 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ksplat.ts
+46 -4
View File
@@ -143,7 +143,8 @@ export function decodeKsplat(
(section.getUint32(24, true) ||
KSPLAT_COMPRESSION[compressionLevel]?.scaleRange) ??
1;
// const fullBucketCount = section.getUint32(32, true);
const fullBucketCount = section.getUint32(32, true);
const fullBucketSplats = fullBucketCount * bucketSize;
const partiallyFilledBucketCount = section.getUint32(36, true);
const bucketsMetaDataSizeBytes = partiallyFilledBucketCount * 4;
const bucketsStorageSizeBytes =
@@ -201,6 +202,11 @@ export function decodeKsplat(
bucketsBase,
bucketCount * 3,
);
const partiallyFilledBucketLengths = new Uint32Array(
fileBytes.buffer,
sectionBase,
partiallyFilledBucketCount,
);
function getSh(splatOffset: number, component: number) {
if (compressionLevel === 0) {
@@ -226,9 +232,24 @@ export function decodeKsplat(
);
}
let partialBucketIndex = fullBucketCount;
let partialBucketBase = fullBucketSplats;
for (let i = 0; i < sectionSplatCount; ++i) {
const splatOffset = i * bytesPerSplat;
const bucketIndex = Math.floor(i / bucketSize);
let bucketIndex: number;
if (i < fullBucketSplats) {
bucketIndex = Math.floor(i / bucketSize);
} else {
const bucketLength =
partiallyFilledBucketLengths[partialBucketIndex - fullBucketCount];
if (i >= partialBucketBase + bucketLength) {
partialBucketIndex += 1;
partialBucketBase += bucketLength;
}
bucketIndex = partialBucketIndex;
}
const x =
compressionLevel === 0
@@ -372,7 +393,8 @@ export function unpackKsplat(fileBytes: Uint8Array): {
(section.getUint32(24, true) ||
KSPLAT_COMPRESSION[compressionLevel]?.scaleRange) ??
1;
// const fullBucketCount = section.getUint32(32, true);
const fullBucketCount = section.getUint32(32, true);
const fullBucketSplats = fullBucketCount * bucketSize;
const partiallyFilledBucketCount = section.getUint32(36, true);
const bucketsMetaDataSizeBytes = partiallyFilledBucketCount * 4;
const bucketsStorageSizeBytes =
@@ -430,6 +452,11 @@ export function unpackKsplat(fileBytes: Uint8Array): {
bucketsBase,
bucketCount * 3,
);
const partiallyFilledBucketLengths = new Uint32Array(
fileBytes.buffer,
sectionBase,
partiallyFilledBucketCount,
);
function getSh(splatOffset: number, component: number) {
if (compressionLevel === 0) {
@@ -455,9 +482,24 @@ export function unpackKsplat(fileBytes: Uint8Array): {
);
}
let partialBucketIndex = fullBucketCount;
let partialBucketBase = fullBucketSplats;
for (let i = 0; i < sectionSplatCount; ++i) {
const splatOffset = i * bytesPerSplat;
const bucketIndex = Math.floor(i / bucketSize);
let bucketIndex: number;
if (i < fullBucketSplats) {
bucketIndex = Math.floor(i / bucketSize);
} else {
const bucketLength =
partiallyFilledBucketLengths[partialBucketIndex - fullBucketCount];
if (i >= partialBucketBase + bucketLength) {
partialBucketIndex += 1;
partialBucketBase += bucketLength;
}
bucketIndex = partialBucketIndex;
}
const x =
compressionLevel === 0